CbmLitFitQa.h

Go to the documentation of this file.

/* Copyright (C) 2011-2017 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */
 
#ifndef CBMLITFITQA_H_
#define CBMLITFITQA_H_
 
#include "CbmDefs.h"
#include "CbmStsKFTrackFitter.h"
#include "FairTask.h"
#include "cbm/base/CbmLitDetectorSetup.h"
 
#include <string>
 
class CbmHistManager;
class CbmVertex;
class FairTrackParam;
class CbmLitMCPoint;
class CbmLitMCTrackCreator;
class CbmMCDataArray;
class CbmEvent;
 
using std::string;
 
class CbmLitFitQa : public FairTask {
 public:
  CbmLitFitQa();
 
  virtual ~CbmLitFitQa();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* opt);
 
  virtual void Finish();
 
  /* Setters */
  void SetMvdMinNofHits(Int_t minNofHits) { fMvdMinNofHits = minNofHits; }
  void SetStsMinNofHits(Int_t minNofHits) { fStsMinNofHits = minNofHits; }
  void SetTrdMinNofHits(Int_t minNofHits) { fTrdMinNofHits = minNofHits; }
  void SetMuchMinNofHits(Int_t minNofHits) { fMuchMinNofHits = minNofHits; }
  void SetOutputDir(const string& dir) { fOutputDir = dir; }
  void SetFixedBounds(Bool_t isFixedBounds) { fIsFixedBounds = isFixedBounds; }
 
  void SetPRange(Int_t bins, Int_t min, Int_t max)
  {
    fPRangeBins = bins;
    fPRangeMin  = min;
    fPRangeMax  = max;
  }
 
 private:
  void ReadDataBranches();
 
  void ProcessGlobalTracks();
 
  void ProcessStsTrack(Int_t trackId);
 
  void ProcessTrdTrack(Int_t trackId);
 
  void ProcessMuchTrack(Int_t trackId);
 
  void FillResidualsAndPulls(const FairTrackParam* par, const CbmLitMCPoint* mcPoint, const string& histName,
                             Float_t wrongPar, ECbmModuleId detId);
 
  void FillTrackParamHistogramm(const string& histName, const FairTrackParam* par);
 
  void ProcessTrackParamsAtVertex();
  void ProcessTrackParamsAtVertex(CbmEvent* event);
 
  void ProcessTrackMomentumAtVertex();
 
  void CreateHistograms();
 
  void CreateResidualAndPullHistograms(ECbmModuleId detId, const string& detName);
 
  void CreateTrackParamHistograms(ECbmModuleId detId, const string& detName);
 
  Bool_t fIsFixedBounds;  // if true than fixed bounds are used for histograms
 
  Int_t fMvdMinNofHits;   // Cut on minimum number of hits in track in MVD
  Int_t fStsMinNofHits;   // Cut on minimum number of hits in track in STS
  Int_t fTrdMinNofHits;   // Cut on minimum number of hits in track in TRD
  Int_t fMuchMinNofHits;  // Cut on minimum number of hits in track in MUCH
 
  string fOutputDir;  // Output directory for images
 
  Double_t fPRangeMin;  // Min momentum
  Double_t fPRangeMax;  // Max momentum
  Int_t fPRangeBins;    // Number of bins
 
  CbmHistManager* fHM;  // Histogram manager
 
  // Data branches
  TClonesArray* fGlobalTracks;      // CbmGlobalTrack array
  TClonesArray* fStsTracks;         // CbmStsTrack array
  TClonesArray* fStsTrackMatches;   // CbmTrackMatchNew array
  TClonesArray* fStsHits;           // CbmStsHit
  TClonesArray* fMvdHits;           // CbmMvdHit
  TClonesArray* fTrdTracks;         // CbmTrdTrack array
  TClonesArray* fTrdTrackMatches;   // CbmTrackMatchNew array
  TClonesArray* fTrdHits;           // CbmTrdHit array
  TClonesArray* fMuchTracks;        // CbmStsTrack array
  TClonesArray* fMuchTrackMatches;  // CbmTrackMatchNew array
  TClonesArray* fMuchPixelHits;     // CbmMuchPixelHit array
  TClonesArray* fMuchStripHits;     // CbmMuchStripHit array
  CbmMCDataArray* fMCTracks;        // CbmMCTrack array
  TClonesArray* fEvents;
 
  Double_t fQuota;  // percent of correctly attached hits
 
  CbmVertex* fPrimVertex;         // Pointer to the primary vertex
  CbmStsKFTrackFitter fKFFitter;  // Pointer to the Kalman Filter Fitter algorithm
 
  CbmLitMCTrackCreator* fMCTrackCreator;  // MC track creator tool
 
  CbmLitDetectorSetup fDet;  // For detector setup determination
 
  CbmLitFitQa(const CbmLitFitQa&);
  CbmLitFitQa& operator=(const CbmLitFitQa&);
 
  ClassDef(CbmLitFitQa, 1)
};
 
#endif /* CBMLITFITQA_H_ */