CbmAnaDimuonAnalysis.h

Go to the documentation of this file.

//----------------------------------------
//
// 2019 A. Senger a.senger@gsi.de
//
//----------------------------------------
 
 
#ifndef CBMANADIMUONANALYSIS_H_
#define CBMANADIMUONANALYSIS_H_ 1
 
#include "FairTask.h"
#include <vector>
 
class CbmAnaDimuonAnalysis;
class CbmAnaMuonCandidate;
class CbmTrackMatch;
class CbmMatch;
class CbmMuchTrack;
class CbmTofHit;
class CbmMuchGeoScheme;
class CbmStsKFTrackFitter;
class CbmTrdTrackFitterKF;
class CbmGlobalTrackFitterKF;
class CbmVertex;
class TClonesArray;
class TLorentzVector;
class TString;
class TH1D;
class TH2D;
class TH3D;
class TProfile;
 
class CbmAnaDimuonAnalysis : public FairTask {
public:
  CbmAnaDimuonAnalysis(TString name, TString setup);
 
  virtual ~CbmAnaDimuonAnalysis() {}
  virtual InitStatus Init();
  virtual void Exec(Option_t* opt);
  virtual void Finish();
  virtual void SetParContainers();
 
  void SetChi2StsCut(Double_t cut) { fChi2StsCut = cut; }
  void SetChi2MuchCut(Double_t cut) { fChi2MuchCut = cut; }
  void SetChi2VertexCut(Double_t cut) { fChi2VertexCut = cut; }
 
  void SetNofMuchCut(Int_t cut) { fNofMuchCut = cut; }
  void SetNofStsCut(Int_t cut) { fNofStsCut = cut; }
  void SetNofTrdCut(Int_t cut) { fNofTrdCut = cut; }
 
  void SetSigmaTofCut(Int_t cut) { fSigmaTofCut = cut; }
 
  void UseCuts(Bool_t cut) { fUseCuts = cut; }
  void UseMC(Bool_t useMC) { fUseMC = useMC; }
 
  //  void SetHistoFileName(TString name) {fFileName = name; }
  //  void SetEffFileName(TString name)   {fEffFileName = name; }
 
  void FillProfile(TProfile* profile, Double_t param, Bool_t trigger);
 
private:
  Int_t fEvent;
  TClonesArray* fMCTracks;
  TClonesArray* fStsTracks;
  TClonesArray* fStsTrackMatches;
  TClonesArray* fMuchTracks;
  TClonesArray* fMuchTrackMatches;
  TClonesArray* fGlobalTracks;
  TClonesArray* fTrdTracks;
  TClonesArray* fTofHit;
  TClonesArray* fMuPlus;
  TClonesArray* fMuMinus;
  TClonesArray* fParticles;
 
  TTree* fInputTree;
  TFile* fPlutoFile;
 
  CbmStsKFTrackFitter* fFitter;
  CbmTrdTrackFitterKF* fFitterTRD;
  CbmGlobalTrackFitterKF* fFitterGlobal;
 
  CbmVertex* fVertex;
 
  Double_t fChi2StsCut;
  Double_t fChi2MuchCut;
  Double_t fChi2VertexCut;
  Int_t fSigmaTofCut;
 
  Double_t fMass;
 
  Bool_t fUseCuts;
  Bool_t fUseMC;
 
  Int_t fNofMuchCut;
  Int_t fNofStsCut;
  Int_t fNofTrdCut;
 
  Double_t p0min, p1min, p2min;
  Double_t p0max, p1max, p2max;
 
  //  TString fFileName;
  //  TString fEffFileName;
  TString fPlutoFileName;
  TString fSetupName;
 
  CbmMuchGeoScheme* fGeoScheme;
 
  TH2D *YPt_pluto, *YPt_StsAcc, *YPt_StsMuchAcc, *YPt_StsMuchTrdAcc,
    *YPt_StsMuchTrdTofAcc;
  TH2D *YPt_VtxReco, *YPt_VtxStsReco, *YPt_VtxStsMuchReco,
    *YPt_VtxStsMuchTrdReco, *YPt_VtxStsMuchTrdTofReco;
  TH3D* YPtM;
 
  TProfile *acc_P[4][3], *acc_Theta[4][3];
  TProfile *effReco_P[4][3], *effReco_Theta[4][3];
  TProfile *eff4pi_P[5][3], *eff4pi_Theta[5][3];
 
  TH1D* BgSup[6];
 
  CbmAnaDimuonAnalysis(const CbmAnaDimuonAnalysis&);
  CbmAnaDimuonAnalysis operator=(const CbmAnaDimuonAnalysis&);
 
  ClassDef(CbmAnaDimuonAnalysis, 2);
};
 
#endif