CbmMuchTransportQa.h

Go to the documentation of this file.

/* Copyright (C) 2020-2021 Facility for Antiproton and Ion Research in Europe, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergey Gorbunov, Evgeny Kryshen, Vikas Singhal, Ekata Nandy, Dominik Smith [committer] */
 
#ifndef CbmMuchTransportQa_H
#define CbmMuchTransportQa_H
 
#include "FairTask.h"
 
#include "TParameter.h"
#include <Rtypes.h>
#include <RtypesCore.h>
#include <TFolder.h>
 
#include <vector>
 
class CbmMCDataArray;
class CbmMCDataManager;
class CbmTimeSlice;
class CbmMuchPoint;
class CbmQaCanvas;
class TBuffer;
class TClass;
class TClonesArray;
class TH1F;
class TH2F;
class TMemberInspector;
class CbmQaPie;
class FairRootManager;
 
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;
 
  // setup
  FairRootManager* fManager    = nullptr;
  CbmMCDataManager* fMcManager = nullptr;
  CbmTimeSlice* fTimeSlice     = nullptr;
 
  CbmMCDataArray* fPoints   = nullptr;
  CbmMCDataArray* fMcTracks = nullptr;
 
 
  TFolder* histFolder;       
  TFolder fOutFolder;        
  TParameter<int> fNevents;  
 
  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<CbmQaPie*> fvMcPointPRatio;     
  std::vector<CbmQaPie*> fvMcPointPrimRatio;  
 
  // output canvaces with histogramm collections
  CbmQaCanvas* fCanvStationXY   = nullptr;
  CbmQaCanvas* fCanvStationPhiZ = nullptr;
  CbmQaCanvas* fCanvStationRZ   = nullptr;
  CbmQaCanvas* fCanvNtra        = nullptr;
 
  // output canvaces with pie chart collections
  CbmQaCanvas* fCanvStationPRatio    = nullptr;
  CbmQaCanvas* fCanvStationPrimRatio = nullptr;
 
  ClassDef(CbmMuchTransportQa, 0)
};
 
#endif