2 #ifndef ALIPERFORMANCEPTCALIB_H
3 #define ALIPERFORMANCEPTCALIB_H
4 //----------------------------------------------------------------------------------------------------
5 // Class to study systematic shifts in pt and charge/pt respectively.Furthermore a comparison between
6 // ESD and TPC track momenta is included.
7 // Track cuts and a user defined shift in 1/pt can be switched on and off by user.
9 // Analysis with class AliPerfAnalyzeInvPt via AliPerformancePtCalib::Analyse(). :
10 // Projection of charge/pt vs theta and vs phi resp. Histograms will be fitted with either
11 // polynomial or gaussian fit function to extract minimum position of 1/pt.
12 // Fit options and theta, phi bins can be set by user.
13 // Attention: use the Set* functions of AliPerformancePtCalib when running AliPerformancePtCalib::Analyse().
15 // Author: S. Schuchmann 11/13/2009
16 //----------------------------------------------------------------------------------------------------
30 class AliTrackReference;
33 class AliESDfriendTrack;
39 class AliESDtrackCuts;
40 class AliAnalysisTask;
42 #include "AliPerformanceObject.h"
44 class AliPerformancePtCalib : public AliPerformanceObject {
46 AliPerformancePtCalib();
47 AliPerformancePtCalib(Char_t* name, Char_t* title);//, Int_t analysisMode, Bool_t hptGenerator);
48 virtual ~AliPerformancePtCalib();
54 virtual void Exec(AliMCEvent* const mcEvent, AliESDEvent *const esdEvent, AliESDfriend *const esdFriend, const Bool_t bUseMC, const Bool_t bUseESDfriend);
56 // Merge output objects (needed by PROOF)
57 virtual Long64_t Merge(TCollection* const list);
59 // Analyse output histograms
60 virtual void Analyse();
62 // Get analysis folder
63 virtual TFolder* GetAnalysisFolder() const {return fAnalysisFolder;}
65 Bool_t AddTPCcuts(const AliESDtrack *esdTrack);// applies TPC cuts
66 Bool_t AddITScuts(const AliESDtrack *esdTrack);// applies ITS cuts
67 Bool_t AddDCAcuts(const AliESDtrack *esdTrack);// applies DCA cuts
69 void SetReadTPCTracks(const Bool_t readTPC) {fOptTPC = readTPC;}//read only ESD tracks
70 void SetTPCRefit(const Bool_t refitTPC) {fRefitTPC = refitTPC;} //switch TPC refit flag on/off
71 void SetITSRefit(const Bool_t refitITS) {fRefitITS = refitITS;} //switch ITS refit flag on/off
72 void SetESDCuts(const Bool_t esdCuts) {fESDcuts = esdCuts;} //switch ESD track cuts on/off
73 void SetDCACuts(const Bool_t dcaCut) {fDCAcut = dcaCut;} //switch DCA cut off
74 void SetAcceptKinkDaughters(const Bool_t kink) {fAcceptKinkDaughters = kink;} //switch accept kink daughters on/off
75 void SetRequireSigmaToVertex(const Bool_t sigmaTo) {fRequireSigmaToVertex = sigmaTo;}//switch require SigmaToVertex on/off
76 void SetfDCAToVertex2D(const Bool_t dcaTo) {fDCAToVertex2D = dcaTo;}//switch DCA to vertex2D on/off
77 void SetEtaRange(const Double_t eta) {fEtaAcceptance = eta ;}//sets eta window
78 void SetESDcutValues(const Double_t * esdCutValues);// set ESD cut values as an array with size 6
79 // fMinPt = esdCutValues[0];
80 // fMaxPt = esdCutValues[1];
81 // fMinNClustersTPC = esdCutValues[2];
82 // fMaxChi2PerClusterTPC = esdCutValues[3];
83 // fMaxDCAtoVertexXY = esdCutValues[4];
84 // fMaxDCAtoVertexZ = esdCutValues[5];
86 //user defined shift in charge/pt
87 void SetPtShift(const Double_t shiftVal); // set user defined shift in charge/pt
89 // for analysis with AliPerformancePtCalibMC::Analyse()
90 void SetProjBinsPhi(const Double_t *pBins,const Int_t sizep);// set phi bins for projection
91 void SetProjBinsTheta(const Double_t *tBins, const Int_t sizet);// set theta bins for projection
92 void SetMakeFitOption(const Bool_t setGausFit, const Double_t exclusionR,const Double_t fitR );//set fit options
93 void SetDoRebin(const Int_t rebin){if(rebin) {fDoRebin = kTRUE; fRebin = rebin;}}
94 const TList *GetHistoList() {return fList;}// get list of histograms for analysis
96 // Create folder for analysed histograms
97 TFolder *CreateFolder(TString folder = "folderPtCalib",TString title = "Analysed PtCalib histograms");
99 // Export objects to folder
100 TFolder *ExportToFolder(TObjArray * array=0);
103 void SetAliRecInfoCuts(AliRecInfoCuts* const cuts=0) {fCutsRC = cuts;}
104 void SetAliMCInfoCuts(AliMCInfoCuts* const cuts=0) {fCutsMC = cuts;}
106 AliRecInfoCuts* GetAliRecInfoCuts() const {return fCutsRC;}
107 AliMCInfoCuts* GetAliMCInfoCuts() const {return fCutsMC;}
110 Double_t fThetaBins[100];// array of theta bins for projection of 1/pt vs theta
111 Double_t fPhiBins[100]; // array of phi bins for projection of 1/pt vs theta
113 Int_t fNThetaBins;// sets number of theta bins
114 Int_t fNPhiBins ;// sets number of phi bins
115 Double_t fRange;// sets fit range
116 Double_t fExclRange ;// sets range of rejection of points around 0
117 Bool_t fFitGaus ;// flag for usage of gaussian fit function
123 Bool_t fShift;//flag for shift in charge/pt
124 Double_t fDeltaInvP;// shift value of charge/pt
127 Bool_t fOptTPC;// flag for reading of TPC tracks in Exec
128 Bool_t fESDcuts;//flag for usage of esd track cuts
129 Bool_t fRefitTPC;//flag for TPC refit
130 Bool_t fRefitITS;// flag for ITS refit
131 Bool_t fDCAcut;//flag for usage of DCA cut
133 Double_t fEtaAcceptance;//sets value of eta window
134 Double_t fMinPt;//sets minimum pt for esd track cuts
135 Double_t fMaxPt;//sets maximum pt for esd track cuts
136 Double_t fMinNClustersTPC;// set minimum number of clusters in TPC for esd track cuts
137 Double_t fMaxChi2PerClusterTPC;//set maximum of chi2 per cluster in TPC for esd track cuts
138 Double_t fMaxDCAtoVertexXY;//set maximum of dca to vertex in xy direction for esd track cuts
139 Double_t fMaxDCAtoVertexZ;//set maximum of dca to vertex in z for esd track cuts
141 Bool_t fAcceptKinkDaughters; // flag for acception of kink daughters
142 Bool_t fRequireSigmaToVertex;// flag for requirering sigma to vertex
143 Bool_t fDCAToVertex2D;//flag for dca to vertex in 2d cut
146 AliRecInfoCuts* fCutsRC; // selection cuts for reconstructed tracks
147 AliMCInfoCuts* fCutsMC; // selection cuts for MC tracks
150 TList *fList;// list of histograms
151 TH2F *fHistInvPtTheta;//theta vs charge/pt
152 TH2F *fHistInvPtPhi;//phi vs charge/pt
153 TH2F *fHistPtTheta;//theta vs pt
154 TH2F *fHistPtPhi;//phi vs pt
156 TH1F *fHistPtShift0;//if shift in 1/pt is set by user, this histogram shows pt wihtout shift
157 TH1F *fHistPrimaryVertexPosX;// primary vertex position x
158 TH1F *fHistPrimaryVertexPosY;// primary vertex position y
159 TH1F *fHistPrimaryVertexPosZ;// primary vertex position z
160 TH1F *fHistTrackMultiplicity; // track multiplicity
161 TH1F *fHistTrackMultiplicityCuts;//track multiplicity after all cuts are applied
163 TH2F *fHistTPCMomentaPosP;//TPC p vs global esd track p for positive tracks
164 TH2F *fHistTPCMomentaNegP;//TPC p vs global esd track p for negative tracks
165 TH2F *fHistTPCMomentaPosPt;//TPC pt vs global esd track p positive tracks
166 TH2F *fHistTPCMomentaNegPt;//TPC pt vs global esd track p for negative tracks
169 TH1F *fHistUserPtShift;// shows the shift value if set by user
171 AliESDtrackCuts* fESDTrackCuts;// esd track cuts
174 TFolder *fAnalysisFolder; // folder for analysed histograms
176 AliPerformancePtCalib(const AliPerformancePtCalib&); // not implemented
177 AliPerformancePtCalib& operator=(const AliPerformancePtCalib&); // not implemented
179 ClassDef(AliPerformancePtCalib, 1);