CbmMuchTransportQa.h

Go to the documentation of this file.

 
#ifndef CbmMuchTransportQa_H
#define CbmMuchTransportQa_H
 
#include "CbmGeoMuchPar.h"
#include "CbmMCTrack.h"
#include "CbmMuchGeoScheme.h"
#include "CbmMuchPoint.h"
#include "CbmMuchStation.h"
#include "CbmQaCanvas.h"
#include "FairLogger.h"
#include "FairRootFileSink.h"
#include "FairRootManager.h"
#include "FairRun.h"
#include "FairRuntimeDb.h"
#include "FairTask.h"
#include "TClonesArray.h"
#include "TDatabasePDG.h"
#include "TFolder.h"
#include "TH1.h"
#include "TH2.h"
#include "TLegend.h"
#include "TParameter.h"
#include "TParticlePDG.h"
#include "TPie.h"
#include "TPieSlice.h"
#include "TString.h"
#include "TStyle.h"
#include <cassert>
#include <vector>
 
#define BINS_STA fNstations, 0, fNstations
 
class CbmMuchTransportQa : public FairTask {
 
public:
  CbmMuchTransportQa(const char* name = "MuchHitFinderQa", Int_t verbose = 1);
  CbmMuchTransportQa(const CbmMuchTransportQa&) = delete;
  CbmMuchTransportQa& operator=(const CbmMuchTransportQa&) = delete;
 
  virtual ~CbmMuchTransportQa();
 
  void SetParContainers();
  InitStatus Init();
  InitStatus ReInit();
  void Exec(Option_t* option);
  void Finish();
 
  TFolder& GetQa();
 
private:
  void InitCountingHistos();
  void InitFractionHistos();
  void Init2dSpatialDistributionHistos();
  void InitRatioPieCharts();
  void InitCanvases();
  void FillCountingHistos(Int_t stId, Int_t motherId, Int_t pdgCode);
  void Fill2dSpatialDistributionHistos(CbmMuchPoint* point, Int_t stId);
 
  void DeInit();
 
  void MakePRatioPieCharts();
  void MakePrimRatioPieCharts();
  void DrawCanvases();
 
  Int_t fNstations = 0;
 
  TClonesArray* fPoints   = nullptr;
  TClonesArray* fMcTracks = nullptr;
 
  TFolder* histFolder;        
  TFolder fOutFolder;         
  TParameter<int> fhNevents;  
 
  TH1F* fhUsNtraAll  = nullptr;  
  TH1F* fhUsNtraPrim = nullptr;  
  TH1F* fhUsNtraSec  = nullptr;  
  TH1F* fhUsNtraPr   = nullptr;  
  TH1F* fhUsNtraPi   = nullptr;  
  TH1F* fhUsNtraEl   = nullptr;  
  TH1F* fhUsNtraMu   = nullptr;  
  TH1F* fhUsNtraKa   = nullptr;  
  std::vector<TH1F*> fvUsNtra;   
 
  std::vector<TH2F*> fvMcPointXY;    
  std::vector<TH2F*> fvMcPointPhiZ;  
  std::vector<TH2F*> fvMcPointRZ;    
 
  TH1F* fhNtracks      = nullptr;  
  TH1F* fhFractionPrim = nullptr;  
  TH1F* fhFractionSec  = nullptr;  
  TH1F* fhFractionPr   = nullptr;  
  TH1F* fhFractionPi   = nullptr;  
  TH1F* fhFractionEl   = nullptr;  
  TH1F* fhFractionMu   = nullptr;  
  TH1F* fhFractionKa   = nullptr;  
  std::vector<TH1F*> fvFraction;   
 
  std::vector<TPie*>
    fvMcPointPRatio;  
  std::vector<TPie*>
    fvMcPointPrimRatio;  
 
  // output canvaces with histogramm collections
  CbmQaCanvas* fCanvStationXY   = nullptr;
  CbmQaCanvas* fCanvStationPhiZ = nullptr;
  CbmQaCanvas* fCanvStationRZ   = nullptr;
  CbmQaCanvas* fCanvUsNtra      = nullptr;
 
  // output canvaces with pie chart collections
  CbmQaCanvas* fCanvStationPRatio    = nullptr;
  CbmQaCanvas* fCanvStationPrimRatio = nullptr;
 
  ClassDef(CbmMuchTransportQa, 0)
};
 
#endif