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 <memory>
#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:
    std::regex m_filePattern;

  private:
    std::regex m_directoryPattern;

  private:
    std::regex m_samplePattern;

  private:
    std::regex m_samplePostfix;

  private:
    bool m_samplePostfixEmpty;

  private:
    typedef std::vector<std::pair<std::regex,std::string> >::const_iterator SampleRenameIter;
    std::vector<std::pair<std::regex,std::string> > m_sampleRename;

  private:
    int m_extraNameComponent;
 

  private:
    void recurse (std::map<std::string,std::shared_ptr<Sample>>& samples,
          DiskList& list,
          const std::vector<std::string>& hierarchy) const;
 

  private:
    void addSampleFile (std::map<std::string,std::shared_ptr<Sample>>& 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