ScanDir.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
 
 
#ifndef SAMPLE_HANDLER__SCAN_DIR_H
#define SAMPLE_HANDLER__SCAN_DIR_H
 
 
#include <SampleHandler/Global.h>
 
#include <RootCoreUtils/StringUtil.h>
#include <map>
#include <vector>
 
namespace SH
{
 
  struct ScanDir
  {
    //
    // public interface
    //
 
  public:
    ScanDir ();
 
 
  public:
    const ScanDir& scan (SampleHandler& sh, const std::string& dir) const;
 
 
  public:
    const ScanDir& scanEOS (SampleHandler& sh, const std::string& eosDir) const;
 
 
  public:
    const ScanDir& scan (SampleHandler& sh, DiskList& list) const;
 
 
  public:
    ScanDir& sampleDepth (int val_sampleDepth);
 
  public:
    ScanDir& absSampleDepth (int val_absSampleDepth);
 
  public:
    ScanDir& sampleName (const std::string& val_sampleName);
 
  public:
    ScanDir& minDepth (std::size_t val_minDepth);
 
  public:
    ScanDir& maxDepth (std::size_t val_maxDepth);
 
  public:
    ScanDir& filePattern (const std::string& val_filePattern);
 
  public:
    ScanDir& fileRegex (const std::string& val_fileRegex);
 
  public:
    ScanDir& directoryPattern (const std::string& val_directoryPattern);
 
  public:
    ScanDir& directoryRegex (const std::string& val_directoryRegex);
 
  public:
    ScanDir& samplePattern (const std::string& val_samplePattern);
 
  public:
    ScanDir& samplePostfix (const std::string& val_samplePostfix);
 
  public:
    ScanDir& sampleRename (const std::string& pattern, const std::string& name);
 
  public:
    ScanDir& extraNameComponent (int val_relSampleDepth);
 
 
 
    //
    // private interface
    //
 
  private:
    int m_relSampleDepth;
 
  private:
    int m_absSampleDepth;
 
  private:
    std::string m_sampleName;
 
  private:
    std::size_t m_minDepth;
 
  private:
    std::size_t m_maxDepth;
 
  private:
    boost::regex m_filePattern;
 
  private:
    boost::regex m_directoryPattern;
 
  private:
    boost::regex m_samplePattern;
 
  private:
    boost::regex m_samplePostfix;
 
  private:
    bool m_samplePostfixEmpty;
 
  private:
    typedef std::vector<std::pair<boost::regex,std::string> >::const_iterator SampleRenameIter;
    std::vector<std::pair<boost::regex,std::string> > m_sampleRename;
 
  private:
    int m_extraNameComponent;
 
 
  private:
    void recurse (std::map<std::string,SamplePtr>& samples, DiskList& list,
          const std::vector<std::string>& hierarchy) const;
 
 
  private:
    void addSampleFile (std::map<std::string,SamplePtr>& samples,
            const std::vector<std::string>& hierarchy,
            const std::string& path) const;
 
 
  private:
    std::string
    findPathComponent (const std::vector<std::string>& hierarchy,
               int absSampleDepth,
               int relSampleDepth) const;
  };
}
 
#endif