CbmLitClusteringQa.h

Go to the documentation of this file.

#
#ifndef CBMLITCLUSTERINGQA_H
#define CBMLITCLUSTERINGQA_H
 
#include <FairTask.h>
#include <string>
#include <vector>
 
#include "CbmMCEventList.h"
#include "CbmModuleList.h"
#include "CbmMuchGeoScheme.h"
#include "CbmTimeSlice.h"
#include "cbm/base/CbmLitDetectorSetup.h"
 
class CbmHistManager;
class CbmMCDataArray;
class CbmDigiManager;
 
class CbmLitClusteringQa : public FairTask {
public:
  CbmLitClusteringQa();
 
  virtual ~CbmLitClusteringQa();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* opt);
 
  virtual void Finish();
 
  void SetOutputDir(const string& outputDir) { fOutputDir = outputDir; }
  void SetMuchDigiFileName(const string& digiFileName) {
    fMuchDigiFileName = digiFileName;
  }
 
private:
  Int_t GetStationId(Int_t address, ECbmModuleId detId);
 
  void ReadDataBranches();
 
  void InitMuchGeoScheme(const string& digiFileName);
 
  void ProcessPoints(Int_t iEvent,
                     CbmMCDataArray* points,
                     const string& detName,
                     ECbmModuleId detId);
 
  template<class Digi>
  void ProcessDigis(const string& detName);
 
  void ProcessClusters(const TClonesArray* clusters,
                       const TClonesArray* clusterMatches,
                       const string& detName,
                       ECbmModuleId detId);
 
  void ProcessHits(const TClonesArray* hits,
                   const TClonesArray* hitMatches,
                   const string& detName,
                   ECbmModuleId detId);
 
  void FillEventCounterHistograms(Int_t iEvent);
 
  void CreateHistograms();
 
  void CreateNofObjectsHistograms(ECbmModuleId detId, const string& detName);
 
  void CreateNofObjectsHistograms(ECbmModuleId detId,
                                  const string& detName,
                                  const string& parameter,
                                  const string& xTitle);
 
  void CreateClusterParametersHistograms(ECbmModuleId detId,
                                         const string& detName);
 
  void FillResidualAndPullHistograms(CbmMCDataArray* points,
                                     const TClonesArray* hits,
                                     const TClonesArray* hitMatches,
                                     const string& detName,
                                     ECbmModuleId detId);
 
  void FillHitEfficiencyHistograms(Int_t iEvent,
                                   CbmMCDataArray* points,
                                   const TClonesArray* hits,
                                   const TClonesArray* hitMatches,
                                   const string& detName,
                                   ECbmModuleId detId);
 
  void CreateHitEfficiencyHistograms(ECbmModuleId detId,
                                     const string& detName,
                                     const string& parameter,
                                     const string& xTitle,
                                     Int_t nofBins,
                                     Double_t minBin,
                                     Double_t maxBin);
 
  CbmHistManager* fHM    = nullptr;    // Histogram manager
  std::string fOutputDir = "";         // Output directory for results
  CbmLitDetectorSetup fDet {};         // For detector setup determination
  CbmDigiManager* fDigiMan = nullptr;  
 
  // Pointers to data arrays
  CbmMCDataArray* fMCTracks = nullptr;  
 
  CbmMCDataArray* fMvdPoints = nullptr;  
  TClonesArray* fMvdClusters = nullptr;  
  TClonesArray* fMvdHits     = nullptr;  
 
  CbmMCDataArray* fStsPoints = nullptr;
  ;  
  TClonesArray* fStsClusters = nullptr;
  ;                                            
  TClonesArray* fStsHits           = nullptr;  
  TClonesArray* fStsClusterMatches = nullptr;  
  TClonesArray* fStsHitMatches     = nullptr;  
 
  CbmMCDataArray* fRichPoints = nullptr;  
  TClonesArray* fRichHits     = nullptr;  
 
  CbmMCDataArray* fMuchPoints        = nullptr;  
  TClonesArray* fMuchClusters        = nullptr;  
  TClonesArray* fMuchPixelHits       = nullptr;  
  TClonesArray* fMuchClusterMatches  = nullptr;  
  TClonesArray* fMuchPixelHitMatches = nullptr;  
 
  CbmMCDataArray* fTrdPoints       = nullptr;  
  TClonesArray* fTrdClusters       = nullptr;  
  TClonesArray* fTrdHits           = nullptr;  
  TClonesArray* fTrdClusterMatches = nullptr;  
  TClonesArray* fTrdHitMatches     = nullptr;  
 
  CbmMCDataArray* fTofPoints = nullptr;  
  TClonesArray* fTofHits     = nullptr;  
 
  string fMuchDigiFileName = "";
 
  CbmTimeSlice* fTimeSlice   = nullptr;
  CbmMCEventList* fEventList = nullptr;
 
  CbmLitClusteringQa(const CbmLitClusteringQa&) = delete;
  CbmLitClusteringQa& operator=(const CbmLitClusteringQa&) = delete;
 
  ClassDef(CbmLitClusteringQa, 2);
};
 
#endif /* CBMLITCLUSTERINGQA_H */