CbmTrdCheckUtil.cxx

Go to the documentation of this file.

// Includes from TRD
#include "CbmTrdCheckUtil.h"
#include "CbmTrdDigi.h"
#include "TMath.h"
#include <string>
 
#include "assert.h"
#include <chrono>
#include <fstream>
#include <iostream>
#include <vector>
 
#include <TH1D.h>
#include <TH2D.h>
#include <TH3D.h>
#include <TProfile.h>
#include <TProfile2D.h>
#include <TProfile3D.h>
 
//_________________________________________________________________________________
CbmTrdCheckUtil::CbmTrdCheckUtil()
  : TObject()
  , f1D()
  , f2D()
  , f3D()
  , fProfile1D()
  , fProfile2D()
  , fProfile3D()
 
{}
 
 
void CbmTrdCheckUtil::CreateHist(std::string name,
                                 Int_t xbins,
                                 Double_t xlow,
                                 Double_t xhigh,
                                 Int_t ybins,
                                 Double_t ylow,
                                 Double_t yhigh) {
  if (ybins == 0 && f1D[name]) return;
  if (f2D[name]) return;
  if (ybins == 0)
    f1D[name] = new TH1D(name.data(), name.data(), xbins, xlow, xhigh);
  else
    f2D[name] = new TH2D(
      name.data(), name.data(), xbins, xlow, xhigh, ybins, ylow, yhigh);
}
 
void CbmTrdCheckUtil::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) {
  if (f3D[name]) return;
  f3D[name] = new TH3D(name.data(),
                       name.data(),
                       xbins,
                       xlow,
                       xhigh,
                       ybins,
                       ylow,
                       yhigh,
                       zbins,
                       zlow,
                       zhigh);
}
 
void CbmTrdCheckUtil::CreateProfile(std::string name,
                                    Int_t xbins,
                                    Double_t xlow,
                                    Double_t xhigh,
                                    Int_t ybins,
                                    Double_t ylow,
                                    Double_t yhigh) {
  if (ybins == 0 && fProfile1D[name]) return;
  if (fProfile2D[name]) return;
  if (ybins == 0)
    fProfile1D[name] =
      new TProfile(name.data(), name.data(), xbins, xlow, xhigh);
  else
    fProfile2D[name] = new TProfile2D(
      name.data(), name.data(), xbins, xlow, xhigh, ybins, ylow, yhigh);
}
 
void CbmTrdCheckUtil::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) {
  if (fProfile3D[name]) return;
  fProfile3D[name] = new TProfile3D(name.data(),
                                    name.data(),
                                    xbins,
                                    xlow,
                                    xhigh,
                                    ybins,
                                    ylow,
                                    yhigh,
                                    zbins,
                                    zlow,
                                    zhigh);
}
 
void CbmTrdCheckUtil::Fill(std::string name, Double_t x, Double_t y) {
  if (y == 9999.) {
    if (!f1D[name]) return;
    f1D[name]->Fill(x);
  } else {
    if (!f2D[name]) return;
    f2D[name]->Fill(x, y);
  }
}
 
void CbmTrdCheckUtil::Fill3D(std::string name,
                             Double_t x,
                             Double_t y,
                             Double_t z) {
  if (!f3D[name]) return;
  f3D[name]->Fill(x, y, z);
}
 
// void CbmTrdCheckUtil::FillProfile(std::string name,Double_t x, Double_t y){
//   if(y == 9999.)   {if(!fProfile1D[name]) return;fProfile1D[name]->Fill(x);}
//   else             {if(!fProfile2D[name]) return;fProfile2D[name]->Fill(x,y);}
// }
 
void CbmTrdCheckUtil::FillProfile(std::string name,
                                  Double_t x,
                                  Double_t y,
                                  Double_t z) {
  if (z == 9999.) {
    if (!fProfile1D[name]) return;
    fProfile1D[name]->Fill(x, y);
  } else {
    if (!fProfile2D[name]) return;
    fProfile2D[name]->Fill(x, y, z);
  }
}
 
void CbmTrdCheckUtil::FillProfile3D(std::string name,
                                    Double_t x,
                                    Double_t y,
                                    Double_t z,
                                    Double_t w) {
  if (!fProfile3D[name]) return;
  fProfile3D[name]->Fill(x, y, z, w);
}
 
void CbmTrdCheckUtil::Fill(std::string name,
                           Double_t x,
                           Double_t y,
                           Double_t z) {
  if (!f2D[name]) return;
  f2D[name]->Fill(x, y, z);
}
 
void CbmTrdCheckUtil::FillW(std::string name, Double_t x, Double_t w) {
  if (!f1D[name]) return;
  f1D[name]->Fill(x, w);
}