Fix: add enable/disable module to streamer.
[u/mrichter/AliRoot.git] / PWGGA / PHOSTasks / PHOS_PbPb / AliAnalysisTaskPi0Flow.h
1 #ifndef AliAnalysisTaskPi0Flow_cxx
2 #define AliAnalysisTaskPi0Flow_cxx
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 // Analysis task to fill histograms with PHOS ESD or AOD clusters and cells
8 // Authors : Dmitri Peressounko
9 // Date    : 28.05.2011
10 // Modified: 03.08.2012 Henrik Qvigstad
11 /* $Id$ */
12
13 class TObjArray;
14 class TH1F;
15 class TH2I;
16 class TH2F;
17 class TH3F;
18 class TF1 ;
19 class AliStack ;
20 class AliESDtrackCuts;
21 class AliPHOSGeometry;
22 class AliESDEvent ;
23 class AliPHOSCalibData;
24 class AliESDtrack ;
25 class AliESDCaloCluster ;
26 class AliEPFlattener;
27
28 #include "TArrayD.h"
29
30 #include "AliAnalysisTaskSE.h"
31
32 class AliAnalysisTaskPi0Flow : public AliAnalysisTaskSE {
33 public:
34     enum Period { kUndefinedPeriod, kLHC10h, kLHC11h };
35
36 public:
37     AliAnalysisTaskPi0Flow(const char *name = "AliAnalysisTaskPi0Flow", Period period = kUndefinedPeriod);
38     virtual ~AliAnalysisTaskPi0Flow();
39
40     virtual void   UserCreateOutputObjects();
41     virtual void   UserExec(Option_t *option);
42     /* virtual void   Terminate(Option_t *); */
43
44     void SetPeriod(Period period) { fPeriod = period;}
45     
46     void SetCentralityBinning(const TArrayD& edges, const TArrayI& nMixed);
47     void SetEventMixingRPBinning(UInt_t nBins) { fNEMRPBins = nBins; }
48     void SetMaxAbsVertexZ(Float_t z) { fMaxAbsVertexZ = z; }
49     void SetManualV0EPCalc(Bool_t manCalc = true) {fManualV0EPCalc = manCalc;}
50     void SetEnablePHOSModule(int module, Bool_t enable = true);
51     
52     void SetPHOSBadMap(Int_t mod,TH2I * badMapHist);
53     //Where to read AODB object with EP calibration if not default
54     void SetEPcalibFileName(const TString filename) {fEPcalibFileName = filename; }   
55
56 private:
57     AliAnalysisTaskPi0Flow(const AliAnalysisTaskPi0Flow&); // not implemented
58     AliAnalysisTaskPi0Flow& operator=(const AliAnalysisTaskPi0Flow&); // not implemented
59
60     // Step 0:
61     AliVEvent* GetEvent();
62     AliStack* GetMCStack();
63
64     // Step 1:
65     void SetGeometry();
66     void SetMisalignment();
67     void SetV0Calibration(); //V0 calibration
68     void SetESDTrackCuts(); // AliESDtrack cuts ( for esd data )
69     void SetPHOSCalibData(); // phos re-calibration ( for esd data)
70     void SetFlatteningData(); // phos flattening
71
72     // Step 2:
73     void SetVertex();
74     Bool_t RejectEventVertex();
75
76     // Step 4:
77     void SetCentrality();
78     Bool_t RejectCentrality();
79
80     // Step 5:
81     void EvalReactionPlane();
82     void EvalV0ReactionPlane();
83
84     // Step 7: QA PHOS cells
85     void FillPHOSCellQAHists();
86
87     // Step 8: Event Photons (PHOS Clusters) selection
88     void SelectPhotonClusters();
89     void FillSelectedClusterHistograms();
90
91     // Step 9: Consider pi0 (photon/cluster) pairs.
92     void ConsiderPi0s();
93
94     // Step 10; Mixing
95     void ConsiderPi0sMix();
96
97     // Step 11: Update lists
98     void UpdateLists();
99
100     Bool_t AreNeibors(Int_t id1,Int_t id2) ;
101     Double_t ApplyFlattening(Double_t phi, Double_t c) ; //Apply centrality-dependent flattening
102     Double_t ApplyFlatteningV0A(Double_t phi, Double_t c) ; //Apply centrality-dependent flattening
103     Double_t ApplyFlatteningV0C(Double_t phi, Double_t c) ; //Apply centrality-dependent flattening
104     Int_t ConvertToInternalRunNumber(Int_t run) ;
105     Double_t CoreEnergy(AliVCluster * clu, AliVCaloCells * cells);
106     void EvalCoreLambdas(AliVCluster * clu, AliVCaloCells * cells, Double_t &m02, Double_t &m20) ; 
107     Bool_t TestCoreLambda(Double_t pt,Double_t l1,Double_t l2) ;
108
109
110
111
112     void FillHistogram(const char * key,Double_t x) const ; //Fill 1D histogram witn name key
113     void FillHistogram(const char * key,Double_t x, Double_t y) const ; //Fill 2D histogram witn name key
114     void FillHistogram(const char * key,Double_t x, Double_t y, Double_t z) const ; //Fill 3D histogram witn name key
115
116     TVector3 GetVertexVector(const AliVVertex* vertex);
117     Int_t GetCentralityBin(Float_t centralityV0M);
118     Int_t GetRPBin();
119
120     void LogProgress(int step);
121     void LogSelection(int step, int internalRunNumber);
122
123     Bool_t IsGoodChannel(const char * det, Int_t mod,Int_t ix, Int_t iz); //Use addisional bad map for PHOS
124
125
126     void Reclusterize(AliVCluster * clu) ;
127     Double_t TestCPV(Double_t dx, Double_t dz, Double_t pt, Int_t charge);
128     Bool_t TestLambda(Double_t pt,Double_t l1,Double_t l2) ;  //Evaluate Dispersion cuts for photons
129     Bool_t TestLambda2(Double_t pt,Double_t l1,Double_t l2) ;  //Evaluate Dispersion cuts for photons
130     
131     UInt_t GetNumberOfCentralityBins() { return fCentEdges.GetSize()-1; }
132     TList* GetCaloPhotonsPHOSList(UInt_t vtxBin, UInt_t centBin, UInt_t rpBin);
133     
134
135
136
137 private:
138     // transiant constants
139     static const Int_t kNMod = 5;
140
141     // constants:
142     static const Double_t kLogWeight= 4.5 ; // log weight for recalibration.
143     static const Double_t kAlphaCut=0.7 ;
144     static const Bool_t doESDReCalibration = kTRUE;
145     static const Int_t kNCenBins = 9; // see EvalV0ReactionPlane()
146
147     // cluster cut variables:
148     static const Double_t kMinClusterEnergy = 0.3;
149     static const Double_t kMinBCDistance = 2.5;  //distance to nearest bad channel
150     static const Int_t kMinNCells = 3;
151     static const Double_t kMinM02 = 0.2;
152
153     // Binning, [vtx, centrality, reaction-plane]
154     static const Int_t kNVtxZBins = 1;
155     static const Double_t kCentCutoff = 90.; // Ignore Centrality over 90%
156     TArrayD fCentEdges;  // Centrality Bin Lower edges
157     TArrayI fCentNMixed; // Number of mixed events for each centrality bin
158     UInt_t fNEMRPBins;
159     
160
161     // Behavior / cuts
162     Period fPeriod;
163     Float_t fMaxAbsVertexZ; // in cm
164     Bool_t fManualV0EPCalc;
165     Bool_t fModuleEnabled[kNMod]; //[kNMod]
166
167
168     TList * fOutputContainer;        //final histogram container
169
170     TF1 *fNonLinCorr;          // Non-linearity correction
171 //TF1 * fRecent[5][12] ;//Recentering corrections
172     TH2I *fPHOSBadMap[6] ;    //Container for PHOS bad channels map
173
174     // Run variables
175
176     // Step 0: Event Objects
177     // fEvent, fMCStack
178     AliVEvent* fEvent; //! Current event
179     AliESDEvent* fEventESD; //! Current event, if ESD.
180     AliAODEvent* fEventAOD; //! Current event, if AOD.
181     AliStack * fMCStack ;
182
183     // Step 1: Run Number, Misalignment Matrix, and Calibration
184     Int_t fRunNumber; // run number
185     Int_t fInternalRunNumber ;    //Current internal run number
186     AliPHOSGeometry  *fPHOSGeo;  //! PHOS geometry
187     TProfile *fMultV0;                  // object containing VZERO calibration information
188     Float_t fV0Cpol,fV0Apol;            // loaded by OADB
189     Float_t fMeanQ[kNCenBins][2][2];    // and recentering
190     Float_t fWidthQ[kNCenBins][2][2];   // ...
191     AliESDtrackCuts *fESDtrackCuts; // Track cut
192     AliPHOSCalibData *fPHOSCalibData; // PHOS calibration object
193     TString fEPcalibFileName; 
194     AliEPFlattener * fTPCFlat ; //Object for flattening of TPC
195     AliEPFlattener * fV0AFlat ; //Object for flattening of V0A
196     AliEPFlattener * fV0CFlat ; //Object for flattening of V0C
197     
198     
199     // Step 2: Vertex
200     Double_t fVertex[3];
201     TVector3 fVertexVector;
202     Int_t fVtxBin;
203
204     // Step 4: Centrality
205     Float_t fCentralityV0M ; //!Centrality of the currecnt event
206     Int_t fCentBin ;       //! Current centrality bin
207
208     // Step 5: Reaction Plane
209     Bool_t fHaveTPCRP ; //! Is TPC RP defined?
210     Float_t fRP ;       //!Reaction plane calculated with full TPC
211     Float_t fRPV0A ;    //!Reaction plain calculated with A-side TPC: eta>0.15
212     Float_t fRPV0C ;    //!Reaction plain calculated with C-side TPC: eta<-0.15
213     Int_t fEMRPBin;       //! Event Mixing Reaction Plane Bin
214
215     // Step 8: Event Photons (PHOS Clusters) selection
216     TObjArray * fCaloPhotonsPHOS ;      //PHOS photons in current event
217
218     // Step 11: Update lists for mixing.
219     TObjArray* fCaloPhotonsPHOSLists; //! array of TList, Containers for events with PHOS photons
220
221
222     ClassDef(AliAnalysisTaskPi0Flow, 1); // PHOS analysis task
223 };
224
225 #endif