TRootCompare.h

Go to the documentation of this file.

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

 
#include "TFileLooper.h"
#include "TFile.h"
#include "TCanvas.h"
#include "TKey.h"

class TRootCompare : public TFileLooper {
 public:
  enum ALG {BIN, CHI2, AXIS};
 
 public:
  TRootCompare();
  virtual ~TRootCompare();
  
  virtual void processKey(TDirectory& dir, TKey& key);
  virtual void beginJob();
  virtual void endJob();
  
  Bool_t setReferenceFile(const char* filename, const char* baseDir = 0);
  Bool_t setOutputFile(const char* filename);
  Bool_t setPsFile(const char* filename);
  void setAlg(TRootCompare::ALG alg, Double_t threshold) {
    m_alg = alg; m_threshold = threshold;
  }
  void drawNormalized(Bool_t norm = kTRUE) { m_drawNormalized = norm; }
  void drawDiff(Bool_t diff = kTRUE) { m_drawDiff = diff; }
  void sortLabels(Bool_t sort = kTRUE) { m_sortLabels = sort; }
  
  Int_t matchingHist() const { return m_histMatch; }
  Int_t totalHist() const { return m_histTotal; }
  Int_t missingHist() const { return m_histMissing; }
  
 private:
  Bool_t compareHist(const TH1& h, const TH1& href);
  void sortAndDeflate(TH1& h);
  void createDirectory(TFile* f, const char* dirpath);
  void printCanvas(const char* filename);
  
 private:
  TFile* m_refFile;
  TFile* m_outFile;
  TString m_refRootDir;
  TString m_psFile;
  TCanvas* m_can;
  
  ALG m_alg;
  Double_t m_threshold;
  Int_t m_histMatch;
  Int_t m_histTotal;
  Int_t m_histMissing;
  Bool_t m_drawNormalized;
  Bool_t m_drawDiff;
  Bool_t m_sortLabels;
  
  std::vector<std::string> m_noMatch;
 
};
 
#endif