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"

#include "FairLogger.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.;
  };

  struct histoComparisonValues{
      Double_t fChi2 = -1.; //< result/threshold for Chi2
      Double_t fKol = -1.;  //< result/threshold for Kolmogorov-Smirnov test
      Double_t fAD = -1.;   //< result/threshold for Anderson-Darling test
  };

  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 SetGlobalThresholds(double thresholdKol = 0.03, double thresholdChi2 = -1.0, double thresholdAD = -1.0)
  {
      fThresholds.clear();
      AddHistoThresholds("", thresholdKol, thresholdChi2, thresholdAD);
  }

  void AddHistoThresholds(TString histoName, double thresholdKol = 0.03, double thresholdChi2 = -1.0, double thresholdAD = -1.0){
      histoComparisonValues thresholds;
      thresholds.fKol = thresholdKol;
      thresholds.fChi2 = thresholdChi2;
      thresholds.fAD = thresholdAD;
      fThresholds.push_back(std::make_pair(histoName, thresholds));
  }

  void AddHistoThresholds(TString histoName, std::vector<double> thresholds){
      if (thresholds.size() == 3)
          AddHistoThresholds(histoName, thresholds[0], thresholds[1], thresholds[2]);
      else {
          LOG(warning) << "Wrong number of threshold values: " << histoName << " : " << thresholds.size();
      }
  }

  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);
  void DrawAndTest2DHistoStack(std::vector<TH1 *> &histos);
  std::pair<TPad *, TPad *> SplitPadForPulls();
  void DrawPulls(std::vector<TH1 *> &histos);
  void CreateLegend(std::vector<TH1 *> &histos);
  histoComparisonValues PerformTest(TH1 *h1, TH1 *h2);
  bool TestPassed(TString& histoTitle, histoComparisonValues results, histoComparisonValues thresholds)
  {
    bool result = true;
    if (thresholds.fKol > 0){
        if (results.fKol < thresholds.fKol)
            result = false;
        histoTitle += " P(K)= ";
        histoTitle += (TString::Format("%.2f", results.fKol));
    }
    if (thresholds.fChi2 > 0){
        if (results.fChi2 < thresholds.fChi2)
            result = false;
        histoTitle += " /P(#Chi^{2})= ";
        histoTitle += (TString::Format("%.2f", results.fChi2));
    }
    if (thresholds.fAD > 0){
        if (results.fAD < thresholds.fAD)
            result = false;
        histoTitle += " /P(AD)= ";
        histoTitle += (TString::Format("%.2f", results.fAD));
    }
    return result;
  }

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

  TString RemoveSpecialCharacters(TString input);

 private:
  std::vector<std::pair<TString, histoComparisonValues>> fThresholds;  //< (part of) histogram name, thresholds for comaprison
  std::vector<fileData> fFileData;
  std::vector<TString> fHistoSelector;
  std::vector<TString> fBranchSelector;
  std::vector<TString> fCDashOutput;
  PndMultiCanvasCreator fCanCreator;
  bool fCreateCDashOutput = false;
  int fFailCount = 0;
  int fPadNumber = 0;
  bool fDoPulls = false;

  ClassDef(PndHistoCombiner, 1);
};

#endif /* TOOLS_PNDHISTOCOMBINER_H_ */