CbmRichMirrorSortingAlignment.h

Go to the documentation of this file.

/* Copyright (C) 2016-2020 Justus-Liebig-Universitaet Giessen, Giessen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Jordan Bendarouach [committer] */
 
#ifndef CBMRICHMIRRORSORTINGALIGNMENT_H
#define CBMRICHMIRRORSORTINGALIGNMENT_H
 
#include "CbmRichMirror.h"
#include "CbmRichRingFitterCOP.h"
#include "CbmRichRingFitterEllipseTau.h"
#include "FairTask.h"
#include "TGeoNavigator.h"
 
#include <fstream>
#include <iomanip>
#include <vector>
 
using namespace std;
 
class TClonesArray;
class TH1D;
class TH2D;
 
 
class CbmRichMirrorSortingAlignment : public FairTask {
 public:
  /*
         * Constructor.
         */
  CbmRichMirrorSortingAlignment();
 
  /*
         * Destructor.
         */
  virtual ~CbmRichMirrorSortingAlignment();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* option);
 
  void ComputeAngles();
 
  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 s);
 
  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);
 
  void CreateHistoMap(std::map<string, vector<CbmRichMirror*>> mirrorMap, std::map<string, TH2D*>& histoMap);
 
  void DrawFitAndExtractAngles(std::map<string, vector<Double_t>>& anglesMap, std::map<string, TH2D*> histoMap);
 
  void setOutputDir(TString s) { fOutputDir = 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 fStudyName;
  Int_t fThreshold;
  std::map<string, vector<CbmRichMirror*>> fMirrorMap;
 
  TClonesArray* fGlobalTracks;
  TClonesArray* fRichRings;
  TClonesArray* fMCTracks;
  TClonesArray* fMirrorPoints;
  TClonesArray* fRefPlanePoints;
  TClonesArray* fPmtPoints;
  TClonesArray* fRichProjections;
  TClonesArray* fTrackParams;
  TClonesArray* fRichRingMatches;
  TClonesArray* fStsTrackMatches;
 
  CbmRichMirrorSortingAlignment(const CbmRichMirrorSortingAlignment&);
  CbmRichMirrorSortingAlignment operator=(const CbmRichMirrorSortingAlignment&);
 
  ClassDef(CbmRichMirrorSortingAlignment, 1);
};
 
#endif