trigger filling of AOD extensions
[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//-----------------------------------------------------------------------
17// This class compares the global reconstruction with the MC information
18// Author : Marta Verweij - UU
19//-----------------------------------------------------------------------
20
21#ifndef ALIPWG4HIGHPTTRACKQA_H
22#define ALIPWG4HIGHPTTRACKQA_H
23
24#include "AliAnalysisTaskSE.h"
25
26class TH1F;
27class TH2F;
28class TH3F;
29class TProfile;
30class TList;
31class TArrayF;
32
33class AliVEvent;
34class AliESDEvent;
35class AliESDtrackCuts;
36class AliESDVertex;
37class AliAODTrack;
38
39class AliGenPythiaEventHeader;
40class AliMCEvent;
41//class AliAnalysisHelperJetTasks;
42
43class AliPWG4HighPtTrackQA: public AliAnalysisTaskSE {
44
45 public:
46 AliPWG4HighPtTrackQA();
47 AliPWG4HighPtTrackQA(const char *name);
48 virtual ~AliPWG4HighPtTrackQA() {;}
49
50 // virtual void ConnectInputData(Option_t *);
51 virtual void UserCreateOutputObjects();
52 virtual void UserExec(Option_t *option);
53 virtual void Terminate(Option_t *);
54 virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2
55
56 enum DataType {kESD,kAOD};
57
58 Bool_t IsPbPb() {return fIsPbPb;} //is PbPb data?
59 Bool_t SelectEvent(); //decides if event is used for analysis
60 Int_t CalculateCentrality(AliVEvent *ev);
61 Int_t CalculateCentrality(AliESDEvent *esd);
62 Int_t CalculateCentrality(AliAODEvent *aod);
63 void DoAnalysisESD();
64 void DoAnalysisAOD();
65 void FillHistograms();
66
67
68 //Setters
69 void SetDataType(DataType d) {fDataType = d;}
70 void SetIsPbPb(Bool_t cs) {fIsPbPb = cs;}
71 void SetCentralityClass(int cent) {fCentClass=cent;}
72 void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
73 void SetTrackType(Int_t trackType) {fTrackType = trackType;}
74 void SetFilterMask(UInt_t filterMask) {fFilterMask = filterMask;}
75
76 void SetPtMax(Float_t ptmax) {fPtMax = ptmax;}
77 void SetNVariables(Int_t nv) {fNVariables = nv;}
78
79 Float_t GetPtMax() {return fPtMax;}
80 Float_t GetTPCClusterInfo(AliAODTrack *tr,Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159) const;
81
82 static AliGenPythiaEventHeader* GetPythiaEventHeader(AliMCEvent *mcEvent);
83 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
84
85 protected:
86
87 private:
88 AliPWG4HighPtTrackQA(const AliPWG4HighPtTrackQA&);
89 AliPWG4HighPtTrackQA& operator=(const AliPWG4HighPtTrackQA&);
90
91 DataType fDataType; //! kESD or kAOD
92
93 AliVEvent *fEvent;
94 AliESDEvent *fESD; //! ESD object
95 const AliESDVertex *fVtx; //! vertex object
96
97 AliESDtrackCuts *fTrackCuts; // TrackCuts
98 Int_t fTrackType; // 0: global track; 1:TPConly track 2: TPConly constrained track 3: global ITSrefit
99 UInt_t fFilterMask; //! Select tracks from specific track cuts belonging to certain filter mask for AOD analysis
100
101 Float_t fPtMax; // Maximum pT for histograms
102
103 Bool_t fIsPbPb; // kTRUE if PbPb
104 Int_t fCentClass; // Select only events from predefined centrality class
105
106 /*
107 0: pt
108 1: phi
109 2: eta
110 3: dca2D
111 4: dcaZ
112 5: nClustersTPC
113 6: nPointITS
114 7: chi2C
115 8: nSigmaToVertex
116 9: relUncertainty1Pt
117 10: chi2PerClusterTPC
118 11: #crossed rows
119 12: (#crossed rows)/(#findable clusters)
120 */
121 Int_t fNVariables; // Number of variables
122 TArrayF *fVariables; // QA variables
123
124 Float_t fAvgTrials; // Average number of trials
125
126 TH1F *fNEventAll; //! Event counter
127 TH1F *fNEventSel; //! Event counter
128 TH1F *fNEventReject; //! Book keeping of reason of rejecting events
129
130 TH1F *fh1Centrality; //! Centrality
131
132 TProfile* fh1Xsec; //! pythia cross section and trials
133 TH1F* fh1Trials; //! trials which are added
134 TH1F* fh1PtHard; //! pt hard of the event
135 TH1F* fh1PtHardTrials; //! pt hard of the event
136
137 TH1F *fh1NTracksAll; //! All tracks
138 TH1F *fh1NTracksReject; //! Reason why track was rejected
139 TH1F *fh1NTracksSel; //! Number of accepted tracks
140
141 TH1F *fPtAll; //! Pt spectrum all charged particles
142 TH1F *fPtSel; //! Pt spectrum all selected charged particles by fTrackCuts
143 TH2F *fPtPhi; //! Pt vs Phi
144 TH2F *fPtEta; //! Pt vs Eta
145 TH2F *fPtDCA2D; //! Pt vs DCA2D
146 TH2F *fPtDCAZ; //! Pt vs DCAZ
147 TH2F *fPtNClustersTPC; //! Pt vs nClustersTPC
148 TH2F *fPtNPointITS; //! Pt vs nPointITS
149 TH2F *fPtChi2C; //! Pt vs Chi2C
150 TH2F *fPtNSigmaToVertex; //! Pt vs nSigmaToVertex
151 TH2F *fPtRelUncertainty1Pt; //! Pt vs relUncertainty1Pt
152 TH2F *fPtChi2PerClusterTPC; //! Pt vs Chi2PerClusterTPC
153 TH2F *fPtNCrossedRows; //! Pt vs NCrossedRows
154 TH2F *fPtNCrossedRowsNClusF; //! Pt vs NCrossedRows/NClusF
155 TH3F *fPtNCrRNCrRNClusF; //! Pt vs NCrossedRows vs NCrossedRows/NClusF
156
157
158 TList *fHistList; //! List of Histograms
159
160
161 ClassDef(AliPWG4HighPtTrackQA,1)
162
163};
164#endif