]>
Commit | Line | Data |
---|---|---|
d756027f | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
16 | //----------------------------------------------------------------------- | |
2b553e6f | 17 | // This class stores QA variables as function of pT for different type |
18 | // of tracks and track selection criteria | |
d756027f | 19 | // Author : Marta Verweij - UU |
20 | //----------------------------------------------------------------------- | |
21 | ||
22 | #ifndef ALIPWG4HIGHPTTRACKQA_H | |
23 | #define ALIPWG4HIGHPTTRACKQA_H | |
24 | ||
25 | #include "AliAnalysisTaskSE.h" | |
26 | ||
27 | class TH1F; | |
28 | class TH2F; | |
29 | class TH3F; | |
30 | class TProfile; | |
31 | class TList; | |
32 | class TArrayF; | |
33 | ||
34 | class AliVEvent; | |
35 | class AliESDEvent; | |
36 | class AliESDtrackCuts; | |
37 | class AliESDVertex; | |
491bba9d | 38 | class AliAODVertex; |
d756027f | 39 | class AliAODTrack; |
aa3ba8d2 | 40 | class AliESDtrack; |
d756027f | 41 | |
42 | class AliGenPythiaEventHeader; | |
43 | class AliMCEvent; | |
44 | //class AliAnalysisHelperJetTasks; | |
45 | ||
46 | class AliPWG4HighPtTrackQA: public AliAnalysisTaskSE { | |
47 | ||
48 | public: | |
49 | AliPWG4HighPtTrackQA(); | |
50 | AliPWG4HighPtTrackQA(const char *name); | |
51 | virtual ~AliPWG4HighPtTrackQA() {;} | |
52 | ||
53 | // virtual void ConnectInputData(Option_t *); | |
54 | virtual void UserCreateOutputObjects(); | |
55 | virtual void UserExec(Option_t *option); | |
56 | virtual void Terminate(Option_t *); | |
57 | virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2 | |
58 | ||
59 | enum DataType {kESD,kAOD}; | |
60 | ||
61 | Bool_t IsPbPb() {return fIsPbPb;} //is PbPb data? | |
62 | Bool_t SelectEvent(); //decides if event is used for analysis | |
63 | Int_t CalculateCentrality(AliVEvent *ev); | |
64 | Int_t CalculateCentrality(AliESDEvent *esd); | |
006b2a30 | 65 | Int_t CalculateCentrality(const AliAODEvent *aod); |
66 | Int_t GetCentralityClass(Float_t cent=-1.) const; | |
d756027f | 67 | void DoAnalysisESD(); |
68 | void DoAnalysisAOD(); | |
69 | void FillHistograms(); | |
70 | ||
d756027f | 71 | //Setters |
72 | void SetDataType(DataType d) {fDataType = d;} | |
73 | void SetIsPbPb(Bool_t cs) {fIsPbPb = cs;} | |
74 | void SetCentralityClass(int cent) {fCentClass=cent;} | |
584095b1 | 75 | void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;} |
44684f3b | 76 | void SetCutsITSLoose(AliESDtrackCuts* trackCuts) {fTrackCutsITSLoose = trackCuts;} |
584095b1 | 77 | void SetCutsTPConly(AliESDtrackCuts* trackCuts) {fTrackCutsTPConly = trackCuts;} |
d756027f | 78 | void SetTrackType(Int_t trackType) {fTrackType = trackType;} |
584095b1 | 79 | void SetFilterMask(UInt_t filterMask) {fFilterMask = filterMask ;} |
80 | void SetIncludeNoITS(Bool_t f) {fIncludeNoITS = f ; } | |
d756027f | 81 | |
2b553e6f | 82 | void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;} |
d756027f | 83 | void SetPtMax(Float_t ptmax) {fPtMax = ptmax;} |
aa3ba8d2 | 84 | void SetPtBinEdges(Int_t region, Double_t ptmax, Double_t ptBinWidth); |
d756027f | 85 | void SetNVariables(Int_t nv) {fNVariables = nv;} |
86 | ||
87 | Float_t GetPtMax() {return fPtMax;} | |
e49861e4 | 88 | Float_t GetTPCClusterInfo(const AliAODTrack *tr,Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159, Bool_t useFitMap=kFALSE) const; |
006b2a30 | 89 | Float_t GetTPCClusterInfoFitMap(const AliESDtrack *tr,Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159) const; |
90 | Int_t GetTrackLengthTPC(const AliESDtrack *track) const; | |
91 | Int_t GetTrackLengthTPC(const AliAODTrack *track) const; | |
42881dab | 92 | Float_t GetGoldenChi2(AliESDtrack *origtrack); |
93 | Float_t GetGGCChi2(AliESDtrack *origtrack); | |
d756027f | 94 | |
006b2a30 | 95 | static AliGenPythiaEventHeader* GetPythiaEventHeader(const AliMCEvent *mcEvent); |
d756027f | 96 | static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root |
97 | ||
98 | protected: | |
99 | ||
100 | private: | |
101 | AliPWG4HighPtTrackQA(const AliPWG4HighPtTrackQA&); | |
102 | AliPWG4HighPtTrackQA& operator=(const AliPWG4HighPtTrackQA&); | |
103 | ||
d480da92 | 104 | DataType fDataType; // kESD or kAOD |
d756027f | 105 | |
aa3ba8d2 | 106 | AliVEvent *fEvent; //! AliVEvent object |
107 | AliESDEvent *fESD; //! ESD object | |
d756027f | 108 | const AliESDVertex *fVtx; //! vertex object |
491bba9d | 109 | const AliAODVertex *fVtxAOD; //! vertex object AOD |
d756027f | 110 | |
44684f3b | 111 | AliESDtrackCuts *fTrackCuts; // TrackCuts |
05cb235d | 112 | AliESDtrackCuts *fTrackCutsITSLoose; // Loose ITS track cuts |
113 | AliESDtrackCuts *fTrackCutsTPConly; // TPC track cuts | |
44684f3b | 114 | Int_t fTrackType; // 0: global track; 1:TPConly track 2: TPConly constrained track 3: global ITSrefit 4: TPConly constrained track with QA selection based on global track |
1ac752c8 | 115 | UInt_t fFilterMask; // Select tracks from specific track cuts belonging to certain filter mask for AOD analysis |
584095b1 | 116 | Bool_t fIncludeNoITS; // includes tracks with failed ITS refit |
d756027f | 117 | |
2b553e6f | 118 | Double_t fSigmaConstrainedMax; // max sigma on constrained fit |
d756027f | 119 | Float_t fPtMax; // Maximum pT for histograms |
aa3ba8d2 | 120 | Float_t fPtBinEdges[3][2]; // 3 regions total with different binning for pT axis of histos |
d756027f | 121 | |
669e2312 | 122 | Bool_t fIsPbPb; // kTRUE if PbPb |
d756027f | 123 | Int_t fCentClass; // Select only events from predefined centrality class |
584095b1 | 124 | |
d756027f | 125 | |
126 | /* | |
a5e0d236 | 127 | QA variables stored in TArrayF *fVariables |
d756027f | 128 | 0: pt |
129 | 1: phi | |
130 | 2: eta | |
131 | 3: dca2D | |
132 | 4: dcaZ | |
133 | 5: nClustersTPC | |
134 | 6: nPointITS | |
135 | 7: chi2C | |
136 | 8: nSigmaToVertex | |
137 | 9: relUncertainty1Pt | |
138 | 10: chi2PerClusterTPC | |
139 | 11: #crossed rows | |
140 | 12: (#crossed rows)/(#findable clusters) | |
a5e0d236 | 141 | 13: SigmaY2 |
142 | 14: SigmaZ2 | |
143 | 15: SigmaSnp2 | |
144 | 16: SigmaTgl2 | |
145 | 17: Sigma1Pt2 | |
146 | 18: NClustersTPCIter1 | |
147 | 19: TPCChi2Iter1 | |
148 | 20: NClustersTPCShared | |
149 | 21: Chi2Gold (TPC constrained vs global) | |
150 | 22: Chi2GGC (global constrained vs global) | |
151 | 23: NCrossed rows from fit map | |
152 | 24: (#crossed rows)/(#findable clusters) from fit map | |
d756027f | 153 | */ |
a5e0d236 | 154 | |
d756027f | 155 | Int_t fNVariables; // Number of variables |
156 | TArrayF *fVariables; // QA variables | |
157 | ||
b8f6bad5 | 158 | UChar_t fITSClusterMap; // map of clusters, one bit per a layer |
159 | ||
d756027f | 160 | Float_t fAvgTrials; // Average number of trials |
161 | ||
162 | TH1F *fNEventAll; //! Event counter | |
163 | TH1F *fNEventSel; //! Event counter | |
164 | TH1F *fNEventReject; //! Book keeping of reason of rejecting events | |
165 | ||
166 | TH1F *fh1Centrality; //! Centrality | |
167 | ||
168 | TProfile* fh1Xsec; //! pythia cross section and trials | |
169 | TH1F* fh1Trials; //! trials which are added | |
170 | TH1F* fh1PtHard; //! pt hard of the event | |
171 | TH1F* fh1PtHardTrials; //! pt hard of the event | |
172 | ||
173 | TH1F *fh1NTracksAll; //! All tracks | |
174 | TH1F *fh1NTracksReject; //! Reason why track was rejected | |
175 | TH1F *fh1NTracksSel; //! Number of accepted tracks | |
176 | ||
177 | TH1F *fPtAll; //! Pt spectrum all charged particles | |
178 | TH1F *fPtSel; //! Pt spectrum all selected charged particles by fTrackCuts | |
179 | TH2F *fPtPhi; //! Pt vs Phi | |
180 | TH2F *fPtEta; //! Pt vs Eta | |
584095b1 | 181 | TH3F *fPtEtaPhi; //! Pt vs Eta vs Phi |
d756027f | 182 | TH2F *fPtDCA2D; //! Pt vs DCA2D |
183 | TH2F *fPtDCAZ; //! Pt vs DCAZ | |
184 | TH2F *fPtNClustersTPC; //! Pt vs nClustersTPC | |
c3db108c | 185 | TH2F *fPtNClustersTPCPhi; //! Phi vs nClustersTPC |
0f76d8ae | 186 | TH2F *fPtNClustersTPCIter1; //! Pt vs nClustersTPCIter1 |
a5e0d236 | 187 | TH3F *fPtNClustersTPCIter1Phi; //! Pt vs nClustersTPCIter1 vs Phi |
d889ce29 | 188 | TH2F *fPtNClustersTPCShared; //! Pt vs nClustersTPCShared |
189 | TH2F *fPtNClustersTPCSharedFrac; //! Pt vs nClustersTPCSharedFrac | |
d756027f | 190 | TH2F *fPtNPointITS; //! Pt vs nPointITS |
0714a353 | 191 | TH3F *fPtNPointITSPhi; //! Pt vs nPointITS vs phi |
d756027f | 192 | TH2F *fPtChi2C; //! Pt vs Chi2C |
193 | TH2F *fPtNSigmaToVertex; //! Pt vs nSigmaToVertex | |
b8f6bad5 | 194 | |
d756027f | 195 | TH2F *fPtRelUncertainty1Pt; //! Pt vs relUncertainty1Pt |
05cb235d | 196 | TH3F *fPtRelUncertainty1PtNClus; //! Pt vs relUncertainty1Pt vs NClustersTPC |
0f76d8ae | 197 | TH3F *fPtRelUncertainty1PtNClusIter1; //! Pt vs relUncertainty1Pt vs NClustersTPCIter1 |
b8f6bad5 | 198 | TH3F *fPtRelUncertainty1PtNPointITS; //! Pt vs relUncertainty1Pt vs NPointITS |
199 | TH3F *fPtRelUncertainty1PtITSClusterMap; //! Pt vs relUncertainty1Pt vs byte of ITS clustermap | |
200 | ||
05cb235d | 201 | TH3F *fPtRelUncertainty1PtChi2; //! Pt vs relUncertainty1Pt vs Chi2TPC/NClus |
0f76d8ae | 202 | TH3F *fPtRelUncertainty1PtChi2Iter1; //! Pt vs relUncertainty1Pt vs Chi2TPC/NClusIter1 |
203 | TH3F *fPtRelUncertainty1PtPhi; //! Pt vs relUncertainty1PtPhi | |
0e8a59df | 204 | |
d756027f | 205 | TH2F *fPtChi2PerClusterTPC; //! Pt vs Chi2PerClusterTPC |
dae7dd67 | 206 | TH2F *fPtChi2PerClusterTPCIter1; //! Pt vs Chi2PerClusterTPCIter1 |
d756027f | 207 | TH2F *fPtNCrossedRows; //! Pt vs NCrossedRows |
a337a5a9 | 208 | TH3F *fPtNCrossedRowsPhi; //! Pt vs NCrossedRows vs Phi |
b8f6bad5 | 209 | TH3F *fPtNCrossedRowsNClusFPhi; //! Pt vs NCrossedRows/NClusF vs Phi |
d756027f | 210 | TH3F *fPtNCrRNCrRNClusF; //! Pt vs NCrossedRows vs NCrossedRows/NClusF |
a5e0d236 | 211 | TH2F *fPtNCrossedRowsFit; //! Pt vs NCrossedRows from NClusterFitMap |
a337a5a9 | 212 | TH3F *fPtNCrossedRowsFitPhi; //! Pt vs NCrossedRows from FitMap vs Phi |
851fbc03 | 213 | TH3F *fPtNCrossedRowsNClusFFitPhi; //! Pt vs NCrossedRows/NClusF from NClusterFitMap |
a337a5a9 | 214 | TH2F *fNCrossedRowsNCrossedRowsFit; //! NCrossedRows from ClusterMap vs NCrossedRows from NClusterFitMap |
215 | TH2F *fNClustersNCrossedRows; //! NClusters vs NCrossedRows | |
216 | TH2F *fNClustersNCrossedRowsFit; //! NClusters vs NCrossedRows from fit map | |
851fbc03 | 217 | TH3F *fPtNClustersNClustersFitMap; //! pT vs ncls vs ncls from fit map |
218 | TH3F *fPtRelUncertainty1PtNCrossedRows; //! Pt vs relUncertainty1Pt vs NCrossedRows | |
219 | TH3F *fPtRelUncertainty1PtNCrossedRowsFit; //! Pt vs relUncertainty1Pt vs NCrossedRowsFitMap | |
d756027f | 220 | |
dae7dd67 | 221 | TH2F *fPtChi2Gold; //! Pt vs Chi2 between global and TPC constrained track |
222 | TH2F *fPtChi2GGC; //! Pt vs Chi2 between global and global constrained track | |
223 | TH3F *fPtChi2GoldPhi; //! Pt vs Chi2 between global and TPC constrained track vs phi | |
224 | TH3F *fPtChi2GGCPhi; //! Pt vs Chi2 between global and global constrained track vs phi | |
225 | TH2F *fChi2GoldChi2GGC; //! Correlations between gold chi2 and GGC chi2 | |
226 | ||
0e8a59df | 227 | TH3F *fPtChi2ITSPhi; //! Pt vs ITS chi2 vs phi |
228 | ||
2b553e6f | 229 | //histos for covariance matrix elements |
aa3ba8d2 | 230 | TH2F *fPtSigmaY2; //! 1/Pt vs sigma(y) extCov[0] |
231 | TH2F *fPtSigmaZ2; //! 1/Pt vs sigma(z) extCov[2] | |
232 | TH2F *fPtSigmaSnp2; //! 1/Pt vs sigma(Snp) extCov[5] | |
233 | TH2F *fPtSigmaTgl2; //! 1/Pt vs sigma(Tgl) extCov[9] | |
234 | TH2F *fPtSigma1Pt2; //! 1/Pt vs sigma(1/pT) extCov[14] | |
235 | ||
236 | //profiles for covariance matrix elements | |
237 | TProfile *fProfPtSigmaY2; //! 1/Pt vs sigma(y) extCov[0] | |
238 | TProfile *fProfPtSigmaZ2; //! 1/Pt vs sigma(z) extCov[2] | |
239 | TProfile *fProfPtSigmaSnp2; //! 1/Pt vs sigma(Snp) extCov[5] | |
240 | TProfile *fProfPtSigmaTgl2; //! 1/Pt vs sigma(Tgl) extCov[9] | |
241 | TProfile *fProfPtSigma1Pt2; //! 1/Pt vs sigma(1/pT) extCov[14] | |
242 | ||
243 | TProfile *fProfPtSigma1Pt; //! pT vs sigma(1/Pt) | |
244 | TProfile *fProfPtPtSigma1Pt; //! pT vs pT*sigma(1/Pt) | |
245 | ||
d756027f | 246 | TList *fHistList; //! List of Histograms |
d756027f | 247 | |
c3db108c | 248 | ClassDef(AliPWG4HighPtTrackQA,8) |
d756027f | 249 | }; |
250 | #endif |