TFileLooper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRIGVALTOOLS_TFILELOOPER_H
#define TRIGVALTOOLS_TFILELOOPER_H

 
#include <fstream>
#include <vector>
 
#include "TDirectory.h"
#include "TFile.h"
#include "TString.h"
#include "TH1.h"
#include "TPRegexp.h"
 

 
class TFileLooper {
 public:
  TFileLooper() = default;
  virtual ~TFileLooper() = default;
 
  TFileLooper (const TFileLooper& other) = delete;
  TFileLooper& operator= (const TFileLooper& other) = delete;

  virtual Int_t run(const char* filename, const char* rootDir = 0);

  virtual void processKey(TDirectory& dir, TKey& key);


  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) {if (skipDirs) m_skipDirs = skipDirs;}

  void addFailRegexp(const char* regexp);

  void addPassRegexp(const char* regexp);

  void passBeforeFailRegexp(Bool_t passBeforeFail = kTRUE) { m_passBeforeFail = passBeforeFail; }

  void setVerbose(Bool_t verbose = kTRUE) {m_verbose = verbose;}
  
  Bool_t verbose() const {return m_verbose;}

  const TFile* file() const { return m_file; }

  TString rootDir() const { return m_rootDir; }
 
protected:
  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);    
};
 
#endif