CbmRichMirrorSortingCorrection.h

Go to the documentation of this file.

/* Copyright (C) 2016-2019 Justus-Liebig-Universitaet Giessen, Giessen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Jordan Bendarouach [committer] */
 
#ifndef CBMRICHMIRRORSORTINGCORRECTION_H
#define CBMRICHMIRRORSORTINGCORRECTION_H
 
#include "CbmHistManager.h"
#include "CbmMCTrack.h"
#include "CbmRichRing.h"
#include "CbmRichRingFitterCOP.h"
#include "CbmRichRingFitterEllipseTau.h"
#include "CbmRichRingLight.h"
#include "FairTask.h"
#include "FairTrackParam.h"
#include "TGeoNavigator.h"
#include "TVector3.h"
 
using namespace std;
 
 
class CbmRichMirrorSortingCorrection : public FairTask {
 public:
  /*
         * Constructor.
         */
  CbmRichMirrorSortingCorrection();
 
  /*
         * Destructor.
         */
  virtual ~CbmRichMirrorSortingCorrection();
 
  virtual InitStatus Init();
 
  void InitHistProjection();
 
  void InitHistoMap();
 
  virtual void Exec(Option_t* option);
 
  void GetPmtNormal(Int_t NofPMTPoints, vector<Double_t>& normalPMT, Double_t& normalCste);
 
  void ComputeR2(vector<Double_t>& ptR2Center, vector<Double_t>& ptR2Mirr, vector<Double_t> ptM, vector<Double_t> ptC,
                 vector<Double_t> ptR1, TGeoNavigator* navi, TString option, TString mirrorTileName);
 
  void ComputeP(vector<Double_t>& ptPMirr, vector<Double_t>& ptPR2, vector<Double_t> normalPMT, vector<Double_t> ptM,
                vector<Double_t> ptR2Mirr, Double_t constantePMT);
 
  /*
         * Function filling the diffX, diffY and distance histograms, from the outPos vector.
         */
  void FillHistProjection(TVector3 outPosIdeal, TVector3 outPosUnCorr, TVector3 outPos, CbmRichRingLight ringLight,
                          vector<Double_t> normalPMT, Double_t constantePMT, string str);
 
  void DrawHistProjection();
 
  void DrawMap(Int_t strX, Int_t strY);
 
  void FillRingTrackDistance();
 
  bool IsMcPrimaryElectron(const CbmMCTrack* mctrack);
 
  void DrawRingTrackDistance(Int_t k);
 
  void FillRingTrackDistanceCorr(const CbmRichRing* richRing, const FairTrackParam* pTrack, const CbmMCTrack* mcTrack);
 
  void DrawRingTrackDistanceCorr();
 
  void DrawDistanceComp();
 
  void setOutputDir(TString s) { fOutputDir = s; }
 
  void setCorrectionTableDir(TString s) { fCorrectionTableDir = s; }
 
  void setStudyName(TString s) { fStudyName = s; }
 
  void setThreshold(Int_t t) { fThreshold = t; }
 
  virtual void Finish();
 
 
 private:
  UInt_t fEventNb;
  CbmRichRingFitterCOP* fCopFit;
  CbmRichRingFitterEllipseTau* fTauFit;
  TString fOutputDir;
  TString fCorrectionTableDir;
  TString fStudyName;
  CbmHistManager *fHM, *fHM2;
  std::map<string, TH1D*> fDiffHistoMap;
  Double_t fTrackCenterDistanceIdeal;
  Double_t fTrackCenterDistanceCorrected;
  Double_t fTrackCenterDistanceUncorrected;
  TString fCorrectionMatching;
  Int_t fThreshold;
 
  TClonesArray* fGlobalTracks;
  TClonesArray* fRichRings;
  TClonesArray* fMCTracks;
  TClonesArray* fMirrorPoints;
  TClonesArray* fRefPlanePoints;
  TClonesArray* fPmtPoints;
  TClonesArray* fRichProjections;
  TClonesArray* fTrackParams;
  TClonesArray* fRichRingMatches;
  TClonesArray* fStsTrackMatches;
 
  ClassDef(CbmRichMirrorSortingCorrection, 1);
};
 
#endif