trigger differentiated QA histos
[u/mrichter/AliRoot.git] / PWGJE / AliAnalysisTaskJetServices.h
1 #ifndef ALIANALYSISTASKJETSERVICES_H
2 #define ALIANALYSISTASKJETSERVICES_H
3  
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 // **************************************
8 // Task used for the correction of determiantion of reconstructed jet spectra
9 // Compares input (gen) and output (rec) jets   
10 // *******************************************
11
12 #include  "AliAnalysisTaskSE.h"
13 #include  "THnSparse.h" // cannot forward declare ThnSparseF
14
15 ////////////////
16 class AliJetHeader;
17 class AliESDEvent;
18 class AliESDVertex;
19 class AliAODEvent;
20 class AliAODVertex;
21 class AliAODVZERO;
22 class AliAODJet;
23 class AliGenPythiaEventHeader;
24 class AliCFManager;
25 class AliTriggerAnalysis;
26
27 class TList;
28 class TClonesArray;
29 class TObjArray;
30 class TChain;
31 class TH1F;
32 class TH2F;
33 class TH3F;
34 class TProfile;
35 class TRandom3;
36
37
38
39 class AliAnalysisTaskJetServices : public AliAnalysisTaskSE
40 {
41  public:
42     AliAnalysisTaskJetServices();
43     AliAnalysisTaskJetServices(const char* name);
44     virtual ~AliAnalysisTaskJetServices();
45     // Implementation of interface methods
46     virtual void UserCreateOutputObjects();
47     virtual void Init();
48     virtual void LocalInit() { Init(); }
49     virtual void UserExec(Option_t *option);
50     virtual void Terminate(Option_t *option);
51     virtual void SetZVertexCut(Float_t f){fVtxZCut = f;}
52     virtual void SetPtMinCosmic(Float_t ptMin) {fPtMinCosmic = ptMin;}
53     virtual void SetRMinCosmic(Float_t rMin) {fRIsolMinCosmic = rMin;}
54     virtual void SetMaxCosmicAngle(Float_t angle)   {fMaxCosmicAngle = angle;}
55     virtual Bool_t Notify();
56
57     virtual void SetAODInput(Bool_t b){fUseAODInput = b;}
58     virtual void SetRunRange(Float_t fLo,Float_t fUp){fRunRange[0] = fLo;fRunRange[1] = fUp;}
59     virtual void SetMCData(Bool_t b){fMC = b;}
60     virtual void SetCollisionType(Int_t iType){fCollisionType = iType;}
61     virtual void SetUsePhysicsSelection(Bool_t b){fUsePhysicsSelection = b;}
62     virtual void SetPhysicsSelectionFlag(Int_t i){fPhysicsSelectionFlag = i;}
63     virtual void SetFilterAODCollisions(Bool_t b){fFilterAODCollisions = b;}
64
65     virtual void SetNonStdFile(char *c){fNonStdFile = c;}
66     Bool_t IsEventSelected(const AliESDEvent* esd);
67     Bool_t IsEventSelected(const AliAODEvent* aod) const;
68
69     Bool_t IsEventPileUp(const AliESDEvent* esd) const;
70
71     Bool_t IsEventCosmic(const AliESDEvent* esd) const;
72
73     Bool_t IsVertexValid(const AliESDVertex *vtx);
74     Bool_t IsVertexValid(const AliAODVertex *vtx) const;
75
76     Bool_t IsVertexIn(const AliESDVertex *vtx);
77     Bool_t IsVertexIn(const AliAODVertex *vtx) const;
78     Int_t GetEventClass(AliESDEvent *esd);
79     Int_t GetEventClass(AliAODEvent *aod);
80
81
82     virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
83     virtual void SetMinTrackPt(Float_t f){fMinTrackPt = f;}
84     virtual void SetTrackEtaWindow(Float_t f){fTrackRecEtaWindow = f;}
85     virtual void SetRPMethod(Int_t i){fRPMethod = i;}
86     virtual void SetV0Centroids(TProfile *xa,TProfile *ya,
87                                TProfile *xc,TProfile *yc);
88
89     virtual void SetNTrigger(Int_t n);
90     virtual void SetTrigger(Int_t i,UInt_t it,const char* c = "");
91
92
93     Bool_t   CalculateReactionPlaneAngleVZERO(AliAODEvent *aod);
94     Int_t   GetListOfTracks(TList *list);
95
96
97     enum { kAllTriggered = 0,kTriggeredVertex,kTriggeredVertexIn,kSelectedALICE,kSelectedALICEVertexValid,kSelectedALICEVertexIn,kSelected,kConstraints};
98     enum { kRPTracks = 0, kRPVZEROA,kRPVZEROC,kRPMethods};
99
100     enum { kNoEventCut=1<<0,
101            kPhysicsSelectionCut=1<<1,
102            kContributorsCut1=1<<2,
103            kContributorsCut2=1<<3,
104            kContributorsCut3=1<<4,
105            kVertexTPC=1<<5,
106            kVertexSPD=1<<6,
107            kVertexGlobal=1<<7,
108            kSPDDispersionCut=1<<8,
109            kVertexZCut=1<<9,
110            kVertexRCut=1<<10,
111            kTotalEventCuts=(1<<11)-1};
112     enum {kPbPb = 0,kPP,kPbP};
113
114  private:
115
116     AliAnalysisTaskJetServices(const AliAnalysisTaskJetServices&);
117     AliAnalysisTaskJetServices& operator=(const AliAnalysisTaskJetServices&);
118
119     Bool_t        fUseAODInput;        // take jet from input AOD not from ouptu AOD
120     Bool_t        fUsePhysicsSelection;// decide wether we take into account physicsselction task
121     Bool_t        fMC;                 // true for MC data to allow correct trigger slection
122     Bool_t        fFilterAODCollisions; // filter out collision canditates to the AOD
123     UInt_t        fPhysicsSelectionFlag; // defines the glag for acceptance of events from physics selection
124     UInt_t        fSelectionInfoESD;   // slection info bit mask
125     UInt_t        fEventCutInfoESD;   // event selection info of what is cutted after physics selection
126     UInt_t        fFilterMask;         // filter bit for slecected tracks
127     Int_t         fRPMethod;           // method for subevent calculation
128     Int_t         fCollisionType;      // type of collisions
129     Int_t         fNTrigger;           // Number of different triggers
130     Float_t       fAvgTrials;          // Average number of trials
131     Float_t       fVtxXMean;           // mean x for cuts
132     Float_t       fVtxYMean;           // mean y for cuts
133     Float_t       fVtxZMean;           // mean z for cuts
134     Float_t       fVtxRCut;            // vtx cut in R
135     Float_t       fVtxZCut;            // vtzx cut in Z
136     Float_t       fPtMinCosmic;        // Minimum pT to be considered as cosmic candidate
137     Float_t       fRIsolMinCosmic;     // Minimum R = sqrt{deltaPhi^2 + deltaEta^2} to be considered as cosmic candidate
138     Float_t       fMaxCosmicAngle;     // Max deviation from pi (angle between two tracks) in case of cosmic candidate
139     Float_t       fRunRange[2];        // only important for real data for 
140     Float_t       fCentrality;         // ! centrality
141     Float_t       fTrackRecEtaWindow;  // eta window for rec tracks
142     Float_t       fMinTrackPt;         // limits the track p_T 
143     Float_t       fRPAngle;            // ! RP angle of the reaction plane
144     Float_t       fPsiVZEROA;          // ! RP angle from vzeroa
145     Float_t       fPsiVZEROC;          // ! RP angle from vzeroc
146     UInt_t        *fTriggerBit  ;      //[fNTrigger]
147
148     TRandom3      *fRandomizer;        // ! randomizer
149
150     TString       fNonStdFile;         // outputName for replication
151     TString       *fTriggerName;  //[fNTrigger] array of trigger names
152     TProfile*     fh1Xsec;             //! pythia cross section and trials
153     TH1F*         fh1Trials;           //! trials are added
154     TH1F*         fh1PtHard;           //! Pt har of the event...       
155     TH1F*         fh1PtHardTrials;     //! Number of trials 
156     TH1F*         fh1SelectionInfoESD; //! Masks that satisfy fSelectionInfo
157     TH1F*         fh1EventCutInfoESD;  //! Masks that satisfy fSelectionInfo
158     TH1F*         fh1CentralityESD;    //! centrality 
159     TH1F*         fh1Centrality;       //! centrality 
160     TH1F*         fh1RP;               //! RP distribution
161     TH2F*         fh2ReducedTrigger;   //! reduced trigger count 
162     TH2F*         fh2CentralityTriggerESD;    //! centrality 
163     TH2F*         fh2CentralityTrigger;       //! centrality 
164     TH2F*         fh2TriggerCount;     //! number of fire triggers in each case
165     TH2F*         fh2ESDTriggerCount;  //! number of fire triggers in each case
166     TH2F*         fh2TriggerVtx;       //! vtx. position vs. trigger decision
167     TH2F*         fh2ESDTriggerVtx;    //! vtx. position vs. trigger decision 
168     TH2F*         fh2ESDTriggerRun;    //! fired triggers vs. run number
169     TH2F*         fh2VtxXY;            //! XY position of VTX were available
170     TH1F*         fh1NCosmicsPerEvent; //! Number of coscmic candidates found in event
171     TProfile*     fp1RPXA;              //! mean XA vs run
172     TProfile*     fp1RPYA;              //! mean YA vs run
173     TProfile*     fp1RPXC;              //! mean XA vs run
174     TProfile*     fp1RPYC;              //! mean YA vs run
175     TProfile*     fp1CalibRPXA;              // calib mean XA vs run
176     TProfile*     fp1CalibRPYA;              // calib YA vs run
177     TProfile*     fp1CalibRPXC;              // calib XA vs run
178     TProfile*     fp1CalibRPYC;              // calib YA vs run
179     TH2F*         fh2RPAC;              //! RP A vs C 
180     TH2F*         fh2RPAT;              //! RP A vs tracks 
181     TH2F*         fh2RPCT;              //! RP C vs tracks 
182     TH2F*         fh2XYA;               //! XY correlations VZERO C
183     TH2F*         fh2XYC;               //! XY correlations VZERO C
184     TH2F*         fh2RPCentrality;      //! RP vs centrality
185     TH2F*         fh2RPACentrality;     //! RP vs centrality
186     TH2F*         fh2RPCCentrality;     //! RP vs centrality
187
188     AliTriggerAnalysis *fTriggerAnalysis; //! Trigger Analysis to get the background rates etc.
189     TList *fHistList; //! Output list
190
191         // Provisions for replication
192     static AliAODHeader*    fgAODHeader;        //! Header for replication
193     static AliAODVZERO*    fgAODVZERO;        //! vzero for replication
194     static TClonesArray*  fgAODVertices;        //! primary vertex for replication
195     ClassDef(AliAnalysisTaskJetServices,15)
196 };
197  
198 #endif