Changes for #89817: Commit calorimeter AOD
[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskESDfilter.h
1 #ifndef ALIANALYSISTASKESDFILTER_H
2 #define ALIANALYSISTASKESDFILTER_H
3  
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 /* $Id$ */
8
9 #include <TList.h> 
10 #include <TF1.h> 
11 #include "AliAnalysisTaskSE.h"
12 #include "AliESDtrack.h"
13 #include "AliAODTrack.h"
14 #include "AliAODPid.h"
15 #include "AliESDpid.h"
16
17 class AliAnalysisFilter;
18 class AliStack;
19 class AliESDtrack;
20 class AliMCEventHandler;
21 class TRefArray;
22 class AliAODHeader;
23 class AliESDtrackCuts;
24
25 class AliAnalysisTaskESDfilter : public AliAnalysisTaskSE
26 {
27  public:
28     AliAnalysisTaskESDfilter();
29     AliAnalysisTaskESDfilter(const char* name);
30     virtual ~AliAnalysisTaskESDfilter();
31     // Implementation of interface methods
32     virtual void   UserCreateOutputObjects();
33     virtual void   Init();
34     virtual void   LocalInit() {Init();}
35     virtual void   UserExec(Option_t *option);
36     virtual void   Terminate(Option_t *option);
37
38     virtual void ConvertESDtoAOD();
39     // Setters
40     virtual void SetTrackFilter   (AliAnalysisFilter*   trackF) {fTrackFilter    =   trackF;}
41     virtual void SetTPCOnlyFilterMask (UInt_t filterMask)       {SetTPCConstrainedFilterMask(filterMask);}
42
43     virtual void SetTPCConstrainedFilterMask (UInt_t filterMask)       {fTPCConstrainedFilterMask    =  filterMask;}
44     virtual void SetHybridFilterMaskTPCConstrainedGlobal(UInt_t filterMask)  {fHybridFilterMaskTPCCG    =  filterMask;}
45     virtual void SetWriteHybridTPCConstrainedOnly(bool b){fWriteHybridTPCCOnly = b;}
46
47     virtual void SetGlobalConstrainedFilterMask (UInt_t filterMask)    {fGlobalConstrainedFilterMask    =  filterMask;}
48     virtual void SetHybridFilterMaskGlobalConstrainedGlobal(UInt_t filterMask)  {fHybridFilterMaskGCG    =  filterMask;}
49     virtual void SetWriteHybridGlobalConstrainedOnly(bool b){fWriteHybridGCOnly = b;}
50
51
52     virtual void SetKinkFilter    (AliAnalysisFilter*    KinkF) {fKinkFilter     =    KinkF;}
53     virtual void SetV0Filter      (AliAnalysisFilter*      V0F) {fV0Filter       =      V0F;}
54     virtual void SetCascadeFilter (AliAnalysisFilter* CascadeF) {fCascadeFilter  = CascadeF;}
55     virtual void SetPthreshold    (Double_t p)                  {fHighPthreshold =        p;}
56     virtual void SetPshape        (TF1 *func)                   {fPtshape        =     func;}
57     virtual void SetEnableFillAOD (Bool_t b)                    {fEnableFillAOD  =     b;}
58
59     virtual void SetAODPID(AliESDtrack *esdtrack, AliAODTrack *aodtrack, AliAODPid *detpid);
60     void SetDetectorRawSignals(AliAODPid *aodpid, AliESDtrack *track);
61
62   void PrintTask(Option_t *option="all", Int_t indent=0) const;
63   
64   void DisableVZERO() { fIsVZEROEnabled = kFALSE; }
65   void DisableTZERO() { fIsTZEROEnabled = kFALSE; }
66   void DisableZDC()   { fIsZDCEnabled   = kFALSE; }
67   void DisableCascades() { fAreCascadesEnabled = kFALSE; }
68   void DisableV0s() { fAreV0sEnabled = kFALSE; }
69   void DisableKinks() { fAreKinksEnabled = kFALSE; }
70   void DisableTracks() { fAreTracksEnabled = kFALSE; }
71   void DisablePmdClusters() { fArePmdClustersEnabled = kFALSE; }
72   void DisableCaloClusters() { fAreCaloClustersEnabled = kFALSE; }
73   void DisableCells() { fAreEMCALCellsEnabled = fArePHOSCellsEnabled = kFALSE; }
74   void DisableCaloTrigger(TString calo = "PHOS") { if (calo.Contains("EMCAL")) fAreEMCALTriggerEnabled = kFALSE; else fArePHOSTriggerEnabled = kFALSE; }
75   void DisableTracklets() { fAreTrackletsEnabled = kFALSE; }
76
77   virtual void SetTimeZeroType(AliESDpid::EStartTimeType_t tofTimeZeroType) {fTimeZeroType = tofTimeZeroType;}
78   
79 private:
80     AliAnalysisTaskESDfilter(const AliAnalysisTaskESDfilter&);
81     AliAnalysisTaskESDfilter& operator=(const AliAnalysisTaskESDfilter&);
82     void PrintMCInfo(AliStack *pStack,Int_t label); // for debugging
83     Double_t Chi2perNDF(AliESDtrack* track);
84     
85   AliAODHeader* ConvertHeader(const AliESDEvent& esd);
86   void ConvertCascades(const AliESDEvent& esd);
87   void ConvertV0s(const AliESDEvent& esd);
88   void ConvertKinks(const AliESDEvent& esd);
89   void ConvertPrimaryVertices(const AliESDEvent& esd);
90   void ConvertTracks(const AliESDEvent& esd);
91   void ConvertPmdClusters(const AliESDEvent& esd);
92   void ConvertCaloClusters(const AliESDEvent& esd);
93   void ConvertEMCALCells(const AliESDEvent& esd);
94   void ConvertPHOSCells(const AliESDEvent& esd);
95   void ConvertCaloTrigger(TString calo, const AliESDEvent& esd);
96   void ConvertTracklets(const AliESDEvent& esd);
97   void ConvertTPCOnlyTracks(const AliESDEvent& esd);
98   void ConvertGlobalConstrainedTracks(const AliESDEvent& esd);
99   void ConvertVZERO(const AliESDEvent& esd);
100   void ConvertTZERO(const AliESDEvent& esd);
101   void ConvertZDC(const AliESDEvent& esd);
102   
103   TClonesArray& Tracks();
104   TClonesArray& V0s();
105   TClonesArray& Vertices();
106   TClonesArray& Cascades();
107   
108   // Filtering
109   AliAnalysisFilter* fTrackFilter;      //  Track   Filter
110   AliAnalysisFilter* fKinkFilter;       //  Kink    Filter
111   AliAnalysisFilter* fV0Filter;         //  V0      Filter
112   AliAnalysisFilter* fCascadeFilter;    //  Cascade Filter
113   // PID
114   Double_t     fHighPthreshold;    //  Pt threshold for detector signal setting
115   TF1 *        fPtshape;           //  Pt spectrum distribution
116   Bool_t       fEnableFillAOD;     //  value that decides if this task activates AOD filling
117   Bool_t* fUsedTrack; //! indices of used tracks
118   Bool_t* fUsedKink; //! indices of used kinks
119   Bool_t* fUsedV0; //! indices of used V0s
120   TRefArray* fAODTrackRefs; // array of track references
121   TRefArray* fAODV0VtxRefs; // array of v0 vertices references
122   TRefArray* fAODV0Refs ; // array of v0s references
123   AliMCEventHandler* fMChandler; // pointer to MC handler (if any)
124   Int_t fNumberOfTracks; // current number of tracks
125   Int_t fNumberOfPositiveTracks; // current number of positive tracks
126   Int_t fNumberOfV0s; // current number of v0s
127   Int_t fNumberOfVertices; // current number of vertices
128   Int_t fNumberOfCascades; // current number of cascades
129   Int_t fNumberOfKinks; // current number of kinks
130   Bool_t fOldESDformat; // is the ESD in old format ?
131   AliAODVertex* fPrimaryVertex; // pointer to primary vertex of the event
132   UInt_t fTPCConstrainedFilterMask;      //  Filter Mask used to select and store refitted TPC only tracks
133   UInt_t fHybridFilterMaskTPCCG; //  Filter Mask used to mark global tracks as hybrid
134   Bool_t fWriteHybridTPCCOnly;// write only the complent tracks not all global constrained
135
136   UInt_t fGlobalConstrainedFilterMask; //  Filter Mask used to select and store refitted TPC only tracks
137   UInt_t fHybridFilterMaskGCG;  //  Filter Mask used to mark global tracks as hybrid
138   Bool_t fWriteHybridGCOnly;// write only the complent tracks not all global constrained
139
140   Bool_t fIsVZEROEnabled; // whether or not to fill the vzero branch (true by default)
141   Bool_t fIsTZEROEnabled; // whether or not to fill the tzero branch (true by default)
142   Bool_t fIsZDCEnabled; // whether or not to fill the zdc branch (true by default)
143   Bool_t fAreCascadesEnabled; // whether or not to fill the cascades branch (true by default)
144   Bool_t fAreV0sEnabled; // whether or not to fill the v0 branch (true by default)
145   Bool_t fAreKinksEnabled; // whether or not to fill the kinks (true by default)
146   Bool_t fAreTracksEnabled; // whether or not to fill the (central) tracks branch (true by default)
147   Bool_t fArePmdClustersEnabled; // whether or not to fill the pmd clusters (true by default)
148   Bool_t fAreCaloClustersEnabled; // whether or not to fill the calo clusters (true by default)
149   Bool_t fAreEMCALCellsEnabled; // whether or not to fill the emcal cells (true by default)
150   Bool_t fArePHOSCellsEnabled; // whether or not to fill the phos cells (true by default)
151   Bool_t fAreEMCALTriggerEnabled; // whether or not to fill the emcal trigger (true by default)
152   Bool_t fArePHOSTriggerEnabled; // whether or not to fill the phos trigger (true by default)
153   Bool_t fAreTrackletsEnabled; // whether or not to fill the tracklets (true by default)
154   AliESDpid* fESDpid; // esd pid
155   Bool_t fIsPidOwner; // whether we own fESDpid
156   Int_t fTimeZeroType;  //  time zero type 
157   AliESDtrackCuts* fTPCaloneTrackCuts; // TPC stand-alone track cuts
158   
159   ClassDef(AliAnalysisTaskESDfilter, 12); // Analysis task for standard ESD filtering
160 };
161  
162 #endif