FileFinder.h

Go to the documentation of this file.

#ifndef ROOTUTIL_FILEFINDER_H
#define ROOTUTIL_FILEFINDER_H

#include <fstream>
#include <iostream>
#include <string>
#include <vector>

namespace RootUtil
{
   class FileFinder
   {
      
   public:
      
      FileFinder(); 
      ~FileFinder();  
      
      // file name must contain substring = value
      void SetKey(const std::string &value);
      
      // select run numbers in this range
      void SetRunRange(int first_run, int last_run);
      
      // in string filename run number starts at beg_pos for length characters
      void SetRunPosition(int beg_pos, int length);

      // recursively search path for matching files, return # of matching files
      int SearchPath(const std::string &path) { return Search(path, 0); }

      void SetRecursiveDepth(short depth);
      
      // search only for files from index file
      void SetIndexFile(const std::string &index_file);
      
      //---------------------------------------------------------------------------
      // const methods below this line
      //---------------------------------------------------------------------------

      // number of found files that meet search criteria
      unsigned int GetNFiles() const { return fFiles.size(); }
      
      // get full file path at this index
      std::string GetFile(int index) const;
      
      // get file size at this index, if index < 0 return size of all files
      double FileSize(int index = -1) const;
      
      // print config parameters
      void Print() const;
      
   private:

      // recursively search path for matching files
      int Search(std::string path, short depth);
      
      bool FindMatch(const std::string &file_name);
      
      bool CheckFileName(const std::string &name, const std::string &path);

      int GetRunNumber(const std::string &filename) const;
      
   private:
      
      int fFirstRun;
      int fLastRun;
      
      int fRunNumberPos;
      int fRunNumberLen;

      int fRecursiveDepth;

      bool fUseIndex;

      // value is required matching substring
      std::vector<std::string> fKeys;
      
      // file names
      std::vector<std::string> fNames;

      // full file path
      std::vector<std::string> fFiles;

      // file names
      std::vector<std::string> fIndex;
   };
}
#endif

---