switchable response matrix
[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliPWG4HighPtTrackQA.h
CommitLineData
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
27class TH1F;
28class TH2F;
29class TH3F;
30class TProfile;
31class TList;
32class TArrayF;
33
34class AliVEvent;
35class AliESDEvent;
36class AliESDtrackCuts;
37class AliESDVertex;
38class AliAODTrack;
aa3ba8d2 39class AliESDtrack;
d756027f 40
41class AliGenPythiaEventHeader;
42class AliMCEvent;
43//class AliAnalysisHelperJetTasks;
44
45class AliPWG4HighPtTrackQA: public AliAnalysisTaskSE {
46
47 public:
48 AliPWG4HighPtTrackQA();
49 AliPWG4HighPtTrackQA(const char *name);
50 virtual ~AliPWG4HighPtTrackQA() {;}
51
52 // virtual void ConnectInputData(Option_t *);
53 virtual void UserCreateOutputObjects();
54 virtual void UserExec(Option_t *option);
55 virtual void Terminate(Option_t *);
56 virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2
57
58 enum DataType {kESD,kAOD};
59
60 Bool_t IsPbPb() {return fIsPbPb;} //is PbPb data?
61 Bool_t SelectEvent(); //decides if event is used for analysis
62 Int_t CalculateCentrality(AliVEvent *ev);
63 Int_t CalculateCentrality(AliESDEvent *esd);
64 Int_t CalculateCentrality(AliAODEvent *aod);
65 void DoAnalysisESD();
66 void DoAnalysisAOD();
67 void FillHistograms();
68
aa3ba8d2 69 void FillSystematicCutHist(AliESDtrack *track);
d756027f 70
71 //Setters
72 void SetDataType(DataType d) {fDataType = d;}
73 void SetIsPbPb(Bool_t cs) {fIsPbPb = cs;}
74 void SetCentralityClass(int cent) {fCentClass=cent;}
75 void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
76 void SetTrackType(Int_t trackType) {fTrackType = trackType;}
77 void SetFilterMask(UInt_t filterMask) {fFilterMask = filterMask;}
78
2b553e6f 79 void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;}
d756027f 80 void SetPtMax(Float_t ptmax) {fPtMax = ptmax;}
aa3ba8d2 81 void SetPtBinEdges(Int_t region, Double_t ptmax, Double_t ptBinWidth);
d756027f 82 void SetNVariables(Int_t nv) {fNVariables = nv;}
83
84 Float_t GetPtMax() {return fPtMax;}
85 Float_t GetTPCClusterInfo(AliAODTrack *tr,Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159) const;
86
87 static AliGenPythiaEventHeader* GetPythiaEventHeader(AliMCEvent *mcEvent);
88 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
89
90 protected:
91
92 private:
93 AliPWG4HighPtTrackQA(const AliPWG4HighPtTrackQA&);
94 AliPWG4HighPtTrackQA& operator=(const AliPWG4HighPtTrackQA&);
95
96 DataType fDataType; //! kESD or kAOD
97
aa3ba8d2 98 AliVEvent *fEvent; //! AliVEvent object
99 AliESDEvent *fESD; //! ESD object
d756027f 100 const AliESDVertex *fVtx; //! vertex object
101
102 AliESDtrackCuts *fTrackCuts; // TrackCuts
aa3ba8d2 103 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
d756027f 104 UInt_t fFilterMask; //! Select tracks from specific track cuts belonging to certain filter mask for AOD analysis
105
2b553e6f 106 Double_t fSigmaConstrainedMax; // max sigma on constrained fit
d756027f 107 Float_t fPtMax; // Maximum pT for histograms
aa3ba8d2 108 Float_t fPtBinEdges[3][2]; // 3 regions total with different binning for pT axis of histos
d756027f 109
110 Bool_t fIsPbPb; // kTRUE if PbPb
111 Int_t fCentClass; // Select only events from predefined centrality class
112
113 /*
114 0: pt
115 1: phi
116 2: eta
117 3: dca2D
118 4: dcaZ
119 5: nClustersTPC
120 6: nPointITS
121 7: chi2C
122 8: nSigmaToVertex
123 9: relUncertainty1Pt
124 10: chi2PerClusterTPC
125 11: #crossed rows
126 12: (#crossed rows)/(#findable clusters)
127 */
128 Int_t fNVariables; // Number of variables
129 TArrayF *fVariables; // QA variables
130
131 Float_t fAvgTrials; // Average number of trials
132
133 TH1F *fNEventAll; //! Event counter
134 TH1F *fNEventSel; //! Event counter
135 TH1F *fNEventReject; //! Book keeping of reason of rejecting events
136
137 TH1F *fh1Centrality; //! Centrality
138
139 TProfile* fh1Xsec; //! pythia cross section and trials
140 TH1F* fh1Trials; //! trials which are added
141 TH1F* fh1PtHard; //! pt hard of the event
142 TH1F* fh1PtHardTrials; //! pt hard of the event
143
144 TH1F *fh1NTracksAll; //! All tracks
145 TH1F *fh1NTracksReject; //! Reason why track was rejected
146 TH1F *fh1NTracksSel; //! Number of accepted tracks
147
148 TH1F *fPtAll; //! Pt spectrum all charged particles
149 TH1F *fPtSel; //! Pt spectrum all selected charged particles by fTrackCuts
150 TH2F *fPtPhi; //! Pt vs Phi
151 TH2F *fPtEta; //! Pt vs Eta
152 TH2F *fPtDCA2D; //! Pt vs DCA2D
153 TH2F *fPtDCAZ; //! Pt vs DCAZ
154 TH2F *fPtNClustersTPC; //! Pt vs nClustersTPC
155 TH2F *fPtNPointITS; //! Pt vs nPointITS
156 TH2F *fPtChi2C; //! Pt vs Chi2C
157 TH2F *fPtNSigmaToVertex; //! Pt vs nSigmaToVertex
158 TH2F *fPtRelUncertainty1Pt; //! Pt vs relUncertainty1Pt
aa3ba8d2 159 TH2F *fPtUncertainty1Pt; //! Pt vs Uncertainty1Pt
d756027f 160 TH2F *fPtChi2PerClusterTPC; //! Pt vs Chi2PerClusterTPC
161 TH2F *fPtNCrossedRows; //! Pt vs NCrossedRows
162 TH2F *fPtNCrossedRowsNClusF; //! Pt vs NCrossedRows/NClusF
163 TH3F *fPtNCrRNCrRNClusF; //! Pt vs NCrossedRows vs NCrossedRows/NClusF
164
2b553e6f 165 //histos for covariance matrix elements
aa3ba8d2 166 TH2F *fPtSigmaY2; //! 1/Pt vs sigma(y) extCov[0]
167 TH2F *fPtSigmaZ2; //! 1/Pt vs sigma(z) extCov[2]
168 TH2F *fPtSigmaSnp2; //! 1/Pt vs sigma(Snp) extCov[5]
169 TH2F *fPtSigmaTgl2; //! 1/Pt vs sigma(Tgl) extCov[9]
170 TH2F *fPtSigma1Pt2; //! 1/Pt vs sigma(1/pT) extCov[14]
171
172 //profiles for covariance matrix elements
173 TProfile *fProfPtSigmaY2; //! 1/Pt vs sigma(y) extCov[0]
174 TProfile *fProfPtSigmaZ2; //! 1/Pt vs sigma(z) extCov[2]
175 TProfile *fProfPtSigmaSnp2; //! 1/Pt vs sigma(Snp) extCov[5]
176 TProfile *fProfPtSigmaTgl2; //! 1/Pt vs sigma(Tgl) extCov[9]
177 TProfile *fProfPtSigma1Pt2; //! 1/Pt vs sigma(1/pT) extCov[14]
178
179 TProfile *fProfPtSigma1Pt; //! pT vs sigma(1/Pt)
180 TProfile *fProfPtPtSigma1Pt; //! pT vs pT*sigma(1/Pt)
181
182 TH1F *fSystTrackCuts; //! Bookkeeping for track cuts
d756027f 183
184 TList *fHistList; //! List of Histograms
185
186
187 ClassDef(AliPWG4HighPtTrackQA,1)
188
189};
190#endif