PndHistoCombiner.h

Go to the documentation of this file.

#ifndef TOOLS_PNDHISTOCOMBINER_H_
#define TOOLS_PNDHISTOCOMBINER_H_

#include <TObject.h>
#include <TString.h>

#include <vector>
#include <map>

#include "PndMultiCanvasCreator.h"

class TFile;
class TH1;

class PndHistoCombiner : public TObject {
 public:
  struct fileData {
    fileData(){};
    fileData(TString fileName, TString shortName) : fFileName(fileName), fShortName(shortName){};
    TString fFileName;
    TString fShortName;
    TFile *fFile = nullptr;
    Double_t fP = -1.;
  };

  PndHistoCombiner();
  virtual ~PndHistoCombiner();

  void AddFileNamesAndShortName(TString fileName, TString shortName)
  {
    fileData myData(fileName, shortName);
    fFileData.push_back(myData);
  }

  int GenerateHistos();

  int GenerateBranches();

  void AddHistoSelector(TString histoName) { fHistoSelector.push_back(histoName); }

  void AddBranchSelector(TString branchName) { fBranchSelector.push_back(branchName); }

  void SetPicPerCan(int val) { fCanCreator.SetPicPerCan(val); }

  void SetCanvasPrefix(TString val) { fCanCreator.SetPrefix(val); }

  void SetCanvasOutputDir(TString val) { fCanCreator.SetOutputDir(val); }

  void SetCanvasOutputFormat(TString val) { fCanCreator.SetOutputFormat(val); }

  void SetPulls(bool dopulls = true)
  {
    fDoPulls = dopulls;
    SetDimensions(400, 400);
  }

  void SetDimensions(int x = 400, int y = 300) { fCanCreator.SetDimensions(x, y); }

  void DoTest(int val = 0, double threshold = 0.03)
  {
    fTestType = val;
    fThreshold = threshold;
  }

  void SetCDashOutput(bool val = true) { fCreateCDashOutput = val; }
  void CreateCDashOutput();

 protected:
  void InitFiles();
  bool CheckHistoName(TString histoName);
  bool CheckBranchName(TString branchName);
  bool CheckBranchType(TString branchType);
  void DrawAndTestHistoStack(std::vector<TH1 *> &histos);
  std::pair<TPad *, TPad *> SplitPadForPulls();
  void DrawPulls(std::vector<TH1 *> &histos);
  void CreateLegend(std::vector<TH1 *> &histos);
  double PerformTest(TH1 *h1, TH1 *h2);
  bool TestPassed(double result, double threshold)
  {
    if (result > threshold)
      return true;
    return false;
  }

  void AddCDashOutput(TString prefix, TString histoName, double value);

  TString RemoveSpecialCharacters(TString input);

 private:
  std::vector<fileData> fFileData;
  std::vector<TString> fHistoSelector;
  std::vector<TString> fBranchSelector;
  std::vector<TString> fCDashOutput;
  std::vector<TH1 *> fTmpHistos;
  PndMultiCanvasCreator fCanCreator;
  int fTestType = 0;
  bool fCreateCDashOutput = false;
  double fThreshold = 0;
  int fFailCount = 0;
  int fPadNumber = 0;
  bool fDoPulls = false;

  ClassDef(PndHistoCombiner, 1);
};

#endif /* TOOLS_PNDHISTOCOMBINER_H_ */