CbmTrdCheckUtil.h

Go to the documentation of this file.

/* Copyright (C) 2020 Institut fuer Kernphysik, Goethe-Universitaet Frankfurt, Frankfurt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Etienne Bechtel [committer] */
 
#ifndef CBMTRDCHECKUTIL_H
#define CBMTRDCHECKUTIL_H
 
#include "CbmTrdDigi.h"
 
#include <FairRootManager.h>
 
#include <TFile.h>
#include <TH1D.h>
#include <TH2D.h>
#include <TH3D.h>
#include <TProfile.h>
#include <TProfile2D.h>
#include <TProfile3D.h>
 
#include <map>
#include <string>
#include <vector>
 
 
class CbmTrdCheckUtil : public TObject {
 
 
public:
  CbmTrdCheckUtil();
 
  CbmTrdCheckUtil(std::string readfile);
 
  CbmTrdCheckUtil(Double_t cal, Double_t tau, Int_t mode);
 
  CbmTrdCheckUtil(Double_t cal, Double_t tau, std::vector<Int_t> mask);
 
  virtual ~CbmTrdCheckUtil() { ; }
 
  //  CbmTrdDigi* MakeDigi(CbmSpadicRawMessage22* raw);
  CbmTrdDigi* MakeDigi(std::vector<Int_t> samples, Int_t channel, Int_t module, Int_t layer, ULong64_t time);
 
  Bool_t fSet = false;  // Boolean for module initialisation in simulation
  std::map<TString, TH1D*> f1D;
  std::map<TString, TH2D*> f2D;
  std::map<TString, TH3D*> f3D;
  std::map<TString, TProfile*> fProfile1D;
  std::map<TString, TProfile2D*> fProfile2D;
  std::map<TString, TProfile3D*> fProfile3D;
 
  void SetSetter(Bool_t set) { fSet = set; }
  void DumpPlots()
  {
    TDirectory* oldir = gDirectory;
    TFile* outFile    = FairRootManager::Instance()->GetOutFile();
    if (outFile != NULL) {
      outFile->cd();
      if (!f1D.empty())
        for (auto const& x : f1D)
          if (x.second) x.second->Write(x.first);
      if (!f2D.empty())
        for (auto const& x : f2D)
          if (x.second) x.second->Write(x.first);
      if (!f3D.empty())
        for (auto const& x : f3D)
          if (x.second) x.second->Write(x.first);
      if (!fProfile1D.empty())
        for (auto const& x : fProfile1D)
          if (x.second) x.second->Write(x.first);
      if (!fProfile2D.empty())
        for (auto const& x : fProfile2D)
          if (x.second) x.second->Write(x.first);
      if (!fProfile3D.empty())
        for (auto const& x : fProfile3D)
          if (x.second) x.second->Write(x.first);
    }
    gDirectory->cd(oldir->GetPath());
  }
 
  void CreateHist(std::string name, Int_t xbins, Double_t xlow, Double_t xhigh, Int_t ybins = 0, Double_t ylow = 1.,
                  Double_t yhigh = 1.);
  void CreateHist3D(std::string name, Int_t xbins, Double_t xlow, Double_t xhigh, Int_t ybins, Double_t ylow,
                    Double_t yhigh, Int_t zbins, Double_t zlow, Double_t zhigh);
  void CreateProfile(std::string name, Int_t xbins, Double_t xlow, Double_t xhigh, Int_t ybins = 0, Double_t ylow = 1.,
                     Double_t yhigh = 1.);
  void CreateProfile3D(std::string name, Int_t xbins, Double_t xlow, Double_t xhigh, Int_t ybins, Double_t ylow,
                       Double_t yhigh, Int_t zbins, Double_t zlow, Double_t zhigh);
  void Fill(std::string name, Double_t x, Double_t y = 9999.);
  //  void FillProfile(std::string name,Double_t x,Double_t y=9999.);
  void FillProfile(std::string name, Double_t x, Double_t y, Double_t z = 9999.);
  void FillProfile3D(std::string name, Double_t x, Double_t y, Double_t z, Double_t w = 1.);
  void Fill(std::string name, Double_t x, Double_t y, Double_t z);
  void Fill3D(std::string name, Double_t x, Double_t y, Double_t z);
  void FillW(std::string name, Double_t x, Double_t w);
  Double_t GetCont2D(std::string name, Double_t x, Double_t y)
  {
    if (!f2D[name]) return 0.;
    return f2D[name]->GetBinContent(x, y);
  }
  Bool_t GetSetter() { return fSet; }
 
private:
  CbmTrdCheckUtil(const CbmTrdCheckUtil&);
  CbmTrdCheckUtil operator=(const CbmTrdCheckUtil&);
 
 
  ClassDef(CbmTrdCheckUtil, 1);
};
 
#endif