TMultiFileLooper Class Reference

Base class to loop over all the keys in several ROOT files. More...

#include <TMultiFileLooper.h>

Inheritance diagram for TMultiFileLooper:

Collaboration diagram for TMultiFileLooper:

Public Member Functions
	TMultiFileLooper ()
virtual	~TMultiFileLooper ()=default
virtual Int_t	run (const char *fileListName, const char *rootDir=0)
	Start processing.
Int_t	run ()
void	addFile (const char *filename)
virtual void	processKey (TDirectory &dir, TKey &key)
	Method called for every key.

Private Attributes
std::vector< TString >	m_fileList

Processing hooks
virtual void	beginJob ()
virtual void	endJob ()
virtual void	beforeFile ()
virtual void	afterFile ()
virtual void	beforeDir ()
virtual void	afterDir ()
virtual void	beforeObj ()
virtual void	afterObj ()
void	setDirsToSkip (const char *skipDirs)
	Skip this comma separated list of directories.
void	addFailRegexp (const char *regexp)
	Skip keys that match this regexp.
void	addPassRegexp (const char *regexp)
	Never skip keys that match this regexp.
void	passBeforeFailRegexp (Bool_t passBeforeFail=kTRUE)
	Reverse pass/fail logic. First check on pass then on fail.
void	setVerbose (Bool_t verbose=kTRUE)
	Set verbose mode.
Bool_t	verbose () const
	Query verbose mode.
const TFile *	file () const
	Current ROOT file.
TString	rootDir () const
	Current directory.
TFile *	m_file {nullptr}
TString	m_rootDir
TString	m_skipDirs
Bool_t	m_verbose {kFALSE}
Bool_t	m_passBeforeFail {kFALSE}
Int_t	m_errorCode {0}
std::vector< TPRegexp >	m_failRE
std::vector< TPRegexp >	m_passRE
std::vector< std::string >	m_skippedObjects
TString	getPathFromDir (const TDirectory &dir)
TString	getKeyPath (const TDirectory &dir, const TKey &key)
void	processDir (TDirectory &dir)
void	processFile (const char *filename, const char *rootDir=0)
Bool_t	skipDir (const TDirectory &dir)
Bool_t	skipObject (const char *name)
TString	getDrawOptions (const TH1 &h)

Detailed Description

Base class to loop over all the keys in several ROOT files.

Author

Frank Winklmeier

Same as TFileLooper but can loop over multiple ROOT files.

Definition at line 28 of file TMultiFileLooper.h.

Constructor & Destructor Documentation

TMultiFileLooper()

TMultiFileLooper::TMultiFileLooper ( )

inline

Definition at line 30 of file TMultiFileLooper.h.

{}

~TMultiFileLooper()

virtual TMultiFileLooper::~TMultiFileLooper ( )

virtualdefault

Member Function Documentation

addFailRegexp()

void TFileLooper::addFailRegexp ( const char * regexp )

inherited

Skip keys that match this regexp.

Definition at line 160 of file TFileLooper.cxx.

{
  if (regexp) {
    m_failRE.emplace_back(regexp);
  }
}

addFile()

void TMultiFileLooper::addFile

(

const char *

filename

)

Definition at line 62 of file TMultiFileLooper.cxx.

{
  if (filename) m_fileList.push_back(TString(filename));
}

addPassRegexp()

void TFileLooper::addPassRegexp ( const char * regexp )

inherited

Never skip keys that match this regexp.

Definition at line 169 of file TFileLooper.cxx.

{
  if (regexp) {
    m_passRE.emplace_back(regexp);
  }
}

afterDir()

virtual void TFileLooper::afterDir ( )

inlinevirtualinherited

Definition at line 59 of file TFileLooper.h.

{}

afterFile()

virtual void TFileLooper::afterFile ( )

inlinevirtualinherited

Definition at line 57 of file TFileLooper.h.

{}

afterObj()

virtual void TFileLooper::afterObj ( )

inlinevirtualinherited

Definition at line 61 of file TFileLooper.h.

{}

beforeDir()

virtual void TFileLooper::beforeDir ( )

inlinevirtualinherited

Definition at line 58 of file TFileLooper.h.

{}

beforeFile()

virtual void TFileLooper::beforeFile ( )

inlinevirtualinherited

Definition at line 56 of file TFileLooper.h.

{}

beforeObj()

virtual void TFileLooper::beforeObj ( )

inlinevirtualinherited

Definition at line 60 of file TFileLooper.h.

{}

beginJob()

virtual void TFileLooper::beginJob ( )

inlinevirtualinherited

Reimplemented in TRootCompare.

Definition at line 54 of file TFileLooper.h.

{}

endJob()

virtual void TFileLooper::endJob ( )

inlinevirtualinherited

Reimplemented in TRootCompare.

Definition at line 55 of file TFileLooper.h.

{}

file()

const TFile * TFileLooper::file ( ) const

inlineinherited

Current ROOT file.

Definition at line 83 of file TFileLooper.h.

{ return m_file; }

getDrawOptions()

TString TFileLooper::getDrawOptions ( const TH1 & h )

protectedinherited

getKeyPath()

TString TFileLooper::getKeyPath ( const TDirectory & dir, const TKey & key )

protectedinherited

Definition at line 188 of file TFileLooper.cxx.

{
  TString s(dir.GetPath()+TString("/")+key.GetName());
  // Remove all chars from beginning including ":/"
  s.Replace(0,s.First(":")+2,0);
 
  return s;
}

getPathFromDir()

TString TFileLooper::getPathFromDir ( const TDirectory & dir )

protectedinherited

Definition at line 178 of file TFileLooper.cxx.

{
  TString s(dir.GetPath());
  if (s.Index("../")==0) s.Remove(0,3);
  s.ReplaceAll(":","");
  return s;
}

passBeforeFailRegexp()

void TFileLooper::passBeforeFailRegexp ( Bool_t passBeforeFail = kTRUE )

inlineinherited

Reverse pass/fail logic. First check on pass then on fail.

Definition at line 74 of file TFileLooper.h.

{ m_passBeforeFail = passBeforeFail; }

processDir()

void TFileLooper::processDir ( TDirectory & dir )

protectedinherited

Definition at line 73 of file TFileLooper.cxx.

{
  TString s(getPathFromDir(dir));
 
  if (skipDir(dir)) {
    cout << "Skipping " << s << endl;
    return;
  }
 
  if (m_verbose) cout << "Reading directory "<< dir.GetPath() << endl;
        
  // Sort directory content
  TList* dirList = dir.GetListOfKeys();
  dirList->Sort();
 
  TIter next(dirList);
  TKey* key;
  while ((key = (TKey*)next())) {
    //    if (skipObject(getKeyPath(dir,*key))) continue;
 
    TString className(key->GetClassName());
    if (className=="TDirectoryFile" || className=="TDirectory") {
      dir.cd(key->GetName());
      beforeDir();      
      processDir(*gDirectory);
      afterDir();
    }
    else {
      if (skipObject(getKeyPath(dir,*key))) {
        m_skippedObjects.push_back(getKeyPath(dir,*key).Data());
        continue;
      }
      processKey(dir, *key);
    }
  }
}

processFile()

void TFileLooper::processFile ( const char * filename, const char * rootDir = 0 )

protectedinherited

Definition at line 44 of file TFileLooper.cxx.

{    
  m_file = new TFile(filename);
  if (m_file->IsZombie()) {
    cout << "Cannot open "<<filename << endl;
    m_errorCode = 1;
    delete m_file;    
    return;
  }
 
  if (rootDir) {
    if (!m_file->cd(rootDir)) {
      cout << "Cannot change to directory " << rootDir << endl;
      m_errorCode = 1;
      return;
    }
    m_rootDir = rootDir;
  }
  else m_file->cd();
  
  beforeFile();
  processDir(*gDirectory);
  afterFile();
  
  delete m_file;   // calls Close()
}

processKey()

void TFileLooper::processKey ( TDirectory & dir, TKey & key )

virtualinherited

Method called for every key.

Only dummy implementation. Needs to be implemented in derived classes.

Reimplemented in TRootCompare.

Definition at line 110 of file TFileLooper.cxx.

{
  cout << "--- processKey: " << key.GetName()
       << " in directory " << dir.GetPath() << endl;
}

rootDir()

TString TFileLooper::rootDir ( ) const

inlineinherited

Current directory.

Definition at line 86 of file TFileLooper.h.

{ return m_rootDir; }

run() [1/2]

Int_t TMultiFileLooper::run

(

)

Definition at line 48 of file TMultiFileLooper.cxx.

{
  m_errorCode = 0;
  beginJob();
  
  vector<TString>::iterator iter;
  for (iter=m_fileList.begin(); iter!=m_fileList.end(); ++iter) {
    processFile(*iter);
  }
 
  endJob();
  return m_errorCode;
}

run() [2/2]

Int_t TMultiFileLooper::run ( const char * filename, const char * rootDir = 0 )

virtual

Start processing.

Reimplemented from TFileLooper.

Definition at line 20 of file TMultiFileLooper.cxx.

{
  m_errorCode = 0;
 
  if (fileListName==0) {
    cout << "Invalid file list (0)" << endl;
    m_errorCode = 1;
    return m_errorCode;
  }
  
  ifstream ifs;
  ifs.open(fileListName);
  if (!ifs) {
    cout << "Cannot open file " << fileListName << endl;
    m_errorCode = 1;
    return m_errorCode;
  }
  
  while (!ifs.eof()) {
    string filename;
    ifs >> filename;
    if (filename!="") addFile(filename.c_str());
  }
 
  return run();
}

setDirsToSkip()

void TFileLooper::setDirsToSkip ( const char * skipDirs )

inlineinherited

Skip this comma separated list of directories.

Definition at line 65 of file TFileLooper.h.

{if (skipDirs) m_skipDirs = skipDirs;}

setVerbose()

void TFileLooper::setVerbose ( Bool_t verbose = kTRUE )

inlineinherited

Set verbose mode.

Definition at line 77 of file TFileLooper.h.

{m_verbose = verbose;}

skipDir()

Bool_t TFileLooper::skipDir ( const TDirectory & dir )

protectedinherited

Definition at line 118 of file TFileLooper.cxx.

{
  if (m_skipDirs.Contains(dir.GetName())) return kTRUE;
  else return kFALSE;
}

skipObject()

Bool_t TFileLooper::skipObject ( const char * name )

protectedinherited

Definition at line 129 of file TFileLooper.cxx.

{
  Bool_t failMatch(kFALSE);
  Bool_t passMatch(kFALSE);
  
  for (auto& re : m_failRE) {
    if (re.Match(name)>0) {
      failMatch = kTRUE;
      break;
    }
  }
 
  // give object another chance to match any of the m_passRE
  for (auto& re : m_passRE) {
    if (re.Match(name)>0) {
      passMatch = kTRUE;
      break;
    }
  }
 
  bool result;
  if (m_passBeforeFail) result = (!passMatch || failMatch);
  else result = (failMatch && !passMatch);
 
  if (m_verbose && result) cout << "Skipping " << name << endl;
  return result;
    
}

verbose()

Bool_t TFileLooper::verbose ( ) const

inlineinherited

Query verbose mode.

Definition at line 80 of file TFileLooper.h.

{return m_verbose;}

Member Data Documentation

m_errorCode

Int_t TFileLooper::m_errorCode {0}

protectedinherited

Definition at line 94 of file TFileLooper.h.

{0};

m_failRE

std::vector<TPRegexp> TFileLooper::m_failRE

protectedinherited

Definition at line 96 of file TFileLooper.h.

m_file

TFile* TFileLooper::m_file {nullptr}

protectedinherited

Definition at line 89 of file TFileLooper.h.

{nullptr};

m_fileList

std::vector<TString> TMultiFileLooper::m_fileList

private

Definition at line 39 of file TMultiFileLooper.h.

m_passBeforeFail

Bool_t TFileLooper::m_passBeforeFail {kFALSE}

protectedinherited

Definition at line 93 of file TFileLooper.h.

{kFALSE};

m_passRE

std::vector<TPRegexp> TFileLooper::m_passRE

protectedinherited

Definition at line 97 of file TFileLooper.h.

m_rootDir

TString TFileLooper::m_rootDir

protectedinherited

Definition at line 90 of file TFileLooper.h.

m_skipDirs

TString TFileLooper::m_skipDirs

protectedinherited

Definition at line 91 of file TFileLooper.h.

m_skippedObjects

std::vector<std::string> TFileLooper::m_skippedObjects

protectedinherited

Definition at line 99 of file TFileLooper.h.

m_verbose

Bool_t TFileLooper::m_verbose {kFALSE}

protectedinherited

Definition at line 92 of file TFileLooper.h.

{kFALSE};

---

The documentation for this class was generated from the following files:

• TMultiFileLooper.h
• TMultiFileLooper.cxx