BmnToCbmHitConverter.h

Go to the documentation of this file.

#ifndef BMNTOCBMHITCONVERTER_H
#define BMNTOCBMHITCONVERTER_H 1
 
#include "BmnGemStripHit.h"
#include "BmnGemStripStationSet.h"
#include "BmnSiliconHit.h"
#include "BmnSiliconStationSet.h"
#include "BmnTask.h"
#include "BmnVSPStationSet.h"
#include "CbmStsHit.h"
#include "FairMCPoint.h"
 
#include <TClonesArray.h>
#include <TH2.h>
#include <TSystem.h>
#include <iostream>
 
class BmnToCbmHitConverter : public BmnTask
{
  public:
    BmnToCbmHitConverter();
    // BmnToCbmHitConverter(Int_t iVerbose);
    BmnToCbmHitConverter(Int_t iVerbose, Bool_t isExp = kTRUE);
 
    virtual ~BmnToCbmHitConverter();
 
    virtual InitStatus Init();
 
    virtual void Exec(Option_t* opt);
 
    virtual void Finish();
 
    void SetGemConfigFile(TString file) { fGemConfigFile = file; }
    void SetSilConfigFile(TString file) { fSilConfigFile = file; }
    void SetVspConfigFile(TString file) { fVspConfigFile = file; }   // AZ-070625
 
    void SetFixedErrors(Float_t dXgem = 0.015, Float_t dYgem = 0.058, Float_t dXsil = 0.003, Float_t dYsil = 0.021);
    void ApplyAlignment(CbmStsHit* hit);
    void ApplyAlignment9(CbmStsHit* hit);    // AZ-250726
    void FillLorentzHistos(int irun);        // AZ-260726
    void ApplyLorentzCorr(CbmStsHit* hit);   // AZ-260726
 
    // to disable alignment (used by default), call EnableAlignment(kFALSE)
    void EnableAlignment(Bool_t is_align_enabled = kTRUE) { fAlignmentEnabled = is_align_enabled; }
 
    void ReadCorrections(int irun,
                         int igem,
                         std::map<int, std::vector<Double_t>>* locCorX,
                         std::map<int, std::vector<Double_t>>* locCorY,
                         int& nx,
                         int& ny);
    // AZ-300526 Bool_t CorrectEffic(CbmStsHit* hit, int iq);                  // AZ-211223
    Bool_t CorrectEffic(CbmStsHit* hit,
                        int iq,
                        Double_t tx = 0,
                        Double_t ty = 0,
                        Double_t pmom = 0,
                        int ikill = 0);                                              // AZ-180626
    void CorrectHitMC(CbmStsHit* hit, Double_t pmom, int iq);                        // AZ-270824
    void AddVspHits();                                                               // AZ-060625
    void KillHits(set<int>* hitSet, set<int>* killedSet);                            // AZ-280526
    void SetMCcorr(Bool_t mccorr = kTRUE) { fMCcorr = mccorr; }                      // AZ-100625
    void SetEffCorr(Bool_t effcorr = kTRUE) { fEffCorr = effcorr; }                  // AZ-100625
    void SetEffFile(TString effFile, int ifile = 0) { fEffFile[ifile] = effFile; }   // AZ-180626
 
    virtual void OnlineWrite(const std::unique_ptr<TTree>& resultTree);
 
  private:
    TString fBmnGemHitsBranchName;
    TString fBmnSilHitsBranchName;
    TString fBmnVspHitsBranchName;
    TString fCbmHitsBranchName;
    TString fEffFile[2];   // AZ-180626
 
    /** Input array of BNM GEM Hits **/
    TClonesArray* fBmnGemHitsArray;       //!
    TClonesArray* fBmnGemUpperClusters;   //!
    TClonesArray* fBmnGemLowerClusters;   //!
    TClonesArray* fGemPoints;             //!
 
    /** Input array of BNM Silicon Hits **/
    TClonesArray* fBmnSilHitsArray;       //!
    TClonesArray* fBmnSilUpperClusters;   //!
    TClonesArray* fBmnSilLowerClusters;   //!
    TClonesArray* fSilPoints;             //!
 
    /** AZ-060625 Input array of VSP Silicon Hits **/
    TClonesArray* fBmnVspHitsArray;       //!
    TClonesArray* fBmnVspUpperClusters;   //!
    TClonesArray* fBmnVspLowerClusters;   //!
    TClonesArray* fVspPoints;             //!
 
    TClonesArray* fMCTracks;   //!
 
    /** Output array of CBM Hits **/
    TClonesArray* fCbmHitsArray;   //!
 
    TString fGemConfigFile;
    TString fSilConfigFile;
    TString fVspConfigFile;
 
    BmnGemStripStationSet* GemStationSet;   //! Entire GEM detector
    BmnSiliconStationSet* SilStationSet;    //! Entire SILICON detector
    BmnVSPStationSet* fVspStationSet;       //! Entire VSP detector
 
    Bool_t fUseFixedErrors;
    Float_t fDXgem;
    Float_t fDYgem;
    Float_t fDXsil;
    Float_t fDYsil;
    Bool_t fExp;
    Bool_t fMCcorr;    // AZ-100625
    Bool_t fEffCorr;   // AZ-100625
    Bool_t fieldFlag;
    // whether alignment enabled
    Bool_t fAlignmentEnabled;
    int fPeriod;   // AZ-250726
    // TH2D *fhLorCorX[2][17], *fhLorCorY[2][17];   // AZ-260726
    std::map<int, TH2D*> fhLorCorX[17], fhLorCorY[17];   // AZ-220826
 
    ClassDef(BmnToCbmHitConverter, 0);
};
 
#endif