217a36dbafd2afc41db68f2adeaf998c14fe9437
[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskESDFilterPWG4Train.C
1
2
3 AliAnalysisTaskESDfilter *AddTaskESDFilter(Bool_t useKineFilter=kTRUE, 
4                                            Bool_t writeMuonAOD=kFALSE,
5                                            Bool_t writeDimuonAOD=kFALSE,
6                                            Bool_t usePhysicsSelection=kFALSE,
7                                            Bool_t useCentralityTask=kFALSE)
8 {
9 // Creates a filter task and adds it to the analysis manager.
10
11    // Get the pointer to the existing analysis manager via the static access method.
12    //==============================================================================
13    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
14    if (!mgr) {
15       ::Error("AddTaskESDFilter", "No analysis manager to connect to.");
16       return NULL;
17    }   
18    
19    // This task requires an ESD input handler and an AOD output handler.
20    // Check this using the analysis manager.
21    //===============================================================================
22    TString type = mgr->GetInputEventHandler()->GetDataType();
23    if (!type.Contains("ESD")) {
24       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an ESD input handler.");
25       return NULL;
26    }   
27    // Check if AOD output handler exist.
28    AliAODHandler *aod_h = (AliAODHandler*)mgr->GetOutputEventHandler();
29    if (!aod_h) {
30       ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an AOD output handler.");
31       return NULL;
32    }
33    // Check if MC handler is connected in case kine filter requested
34    AliMCEventHandler *mcH = (AliMCEventHandler*)mgr->GetMCtruthEventHandler();
35    if (!mcH && useKineFilter) {
36       ::Error("AddTaskESDFilter", "No MC handler connected while kine filtering requested");
37       return NULL;
38    }   
39    
40    // Create the task, add it to the manager and configure it.
41    //===========================================================================   
42    // Barrel tracks filter
43    AliAnalysisTaskESDfilter *esdfilter = new AliAnalysisTaskESDfilter("ESD Filter");
44
45    // Make the AOD a little bit lighter and filtering faster
46    
47    esdfilter->DisableCascades();
48    //   esdfilter->DisableV0s();
49    esdfilter->DisableKinks();
50    //   esdfilter->DisableTracks();
51    esdfilter->DisablePmdClusters();
52    //   esdfilter->DisableCaloClusters();
53    //   esdfilter->DisableCells(); 
54    esdfilter->DisableTracklets();
55    esdfilter->SetWriteHybridGlobalConstrainedOnly(kTRUE);
56
57    AliAnalysisTaskAODCentralityMaker* ctask = 0;
58    if (useCentralityTask) {
59        ctask = new AliAnalysisTaskAODCentralityMaker("AODCentralityMaker");
60        ctask->SetDeltaAODFileName("AliAODCentrality.root");
61        mgr->AddTask(ctask);
62    }
63
64
65    mgr->AddTask(esdfilter);
66    // Muons
67    //   AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");
68    //   mgr->AddTask(esdmuonfilter);
69    if(usePhysicsSelection){
70      esdfilter->SelectCollisionCandidates(AliVEvent::kAny);
71      //     esdmuonfilter->SelectCollisionCandidates(AliVEvent::kAny);
72    }  
73
74    // Filtering of MC particles (decays conversions etc)
75    // this task has to go AFTER all other filter tasks
76    // since it fills the AODMC array with all
77    // selected MC Particles, only this way we have the 
78    // AODMCparticle information available for following tasks
79    AliAnalysisTaskMCParticleFilter *kinefilter = 0;
80    if (useKineFilter) {
81       kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Kine Filter");
82       if(usePhysicsSelection)kinefilter->SelectCollisionCandidates(AliVEvent::kAny);
83       mgr->AddTask(kinefilter);
84    }   
85
86    // Cuts on primary tracks
87    AliESDtrackCuts* esdTrackCutsL = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
88
89    // ITS stand-alone tracks
90    AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("ITS stand-alone Track Cuts", "ESD Track Cuts");
91    esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE);
92
93    // Pixel OR necessary for the electrons
94    AliESDtrackCuts *itsStrong = new AliESDtrackCuts("ITSorSPD", "pixel requirement for ITS");
95    itsStrong->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny);
96
97
98    // PID for the electrons
99    AliESDpidCuts *electronID = new AliESDpidCuts("Electrons", "Electron PID cuts");
100    electronID->SetTPCnSigmaCut(AliPID::kElectron, 3.);
101
102    // tighter cuts on primary particles for high pT tracks
103    // take the standard cuts, which include already 
104    // ITSrefit and use only primaries...
105
106    // ITS cuts for new jet analysis 
107    gROOT->LoadMacro("$ALICE_ROOT/PWG4/macros/CreateTrackCutsPWG4.C");
108    AliESDtrackCuts* esdTrackCutsHG0 = CreateTrackCutsPWG4(10001001);
109
110    // throw out tracks with too low number of clusters in
111    // the first pass (be consistent with TPC only tracks)
112    // N.B. the number off crossed rows still acts on the tracks after
113    // all iterations if we require tpc standalone, number of clusters
114    // and chi2 TPC cuts act on track after the first iteration
115    //   esdTrackCutsH0->SetRequireTPCStandAlone(kTRUE);
116    //   esdTrackCutsH0->SetMinNClustersTPC(80); // <--- first pass
117
118
119    // the complement to the one with SPD requirement
120    AliESDtrackCuts* esdTrackCutsHG1 = CreateTrackCutsPWG4(10011001);
121
122    // the tracks that must not be taken pass this cut and
123    // non HGC1 and HG
124    AliESDtrackCuts* esdTrackCutsHG2 = CreateTrackCutsPWG4(10021001);
125
126    
127
128
129    // standard cuts also used in R_AA analysis
130    gROOT->LoadMacro("$ALICE_ROOT/PWG0/dNdPt/macros/CreatedNdPtTrackCuts.C");
131    AliESDtrackCuts* esdTrackCutsH2 = CreatedNdPtTrackCuts(200);
132    esdTrackCutsH2->SetMaxChi2PerClusterITS(36.);
133    esdTrackCutsH2->SetPtRange(0.15,1E10);
134
135    // TPC only tracks
136    AliESDtrackCuts* esdTrackCutsGCOnly = CreateTrackCutsPWG4(10041001);
137
138    // Compose the filter
139    AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");
140    // 1, 1<<0
141    trackFilter->AddCuts(esdTrackCutsL);
142    // 2 1<<1
143    trackFilter->AddCuts(esdTrackCutsITSsa);
144    // 4 1<<2
145    trackFilter->AddCuts(itsStrong);
146    itsStrong->SetFilterMask(1);        // AND with Standard track cuts 
147    // 8 1<<3
148    trackFilter->AddCuts(electronID);
149    electronID->SetFilterMask(4);       // AND with Pixel Cuts
150    // 16 1<<4
151    trackFilter->AddCuts(esdTrackCutsHG0);
152    // 32 1<<5
153    trackFilter->AddCuts(esdTrackCutsHG1);
154    // 64 1<<6
155    trackFilter->AddCuts(esdTrackCutsHG2);
156    // 128 1<<7
157    trackFilter->AddCuts(esdTrackCutsHG0); // add once more for tpc only tracks
158    // 256 1<<8
159    trackFilter->AddCuts(esdTrackCutsGCOnly);
160    // 512 1<<9                         
161    trackFilter->AddCuts(esdTrackCutsHG1); // add once more for tpc only tracks
162    // 512 1<<10                        
163    trackFilter->AddCuts(esdTrackCutsH2); // add r_aa cuts
164
165    esdfilter->SetGlobalConstrainedFilterMask(1<<8|1<<9); // these tracks are written out as global constrained tracks
166    esdfilter->SetHybridFilterMaskGlobalConstrainedGlobal((1<<4)); // these normal global tracks will be marked as hybrid
167
168
169    // Filter with cuts on V0s
170    AliESDv0Cuts*   esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts");
171    esdV0Cuts->SetMinRadius(0.2);
172    esdV0Cuts->SetMaxRadius(200);
173    esdV0Cuts->SetMinDcaPosToVertex(0.05);
174    esdV0Cuts->SetMinDcaNegToVertex(0.05);
175    esdV0Cuts->SetMaxDcaV0Daughters(1.5);
176    esdV0Cuts->SetMinCosinePointingAngle(0.99);
177    AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter");
178    v0Filter->AddCuts(esdV0Cuts);
179
180    esdfilter->SetTrackFilter(trackFilter);
181    esdfilter->SetV0Filter(v0Filter);
182
183    // Enable writing of Muon AODs
184    //   esdmuonfilter->SetWriteMuonAOD(writeMuonAOD);
185    
186    // Enable writing of Dimuon AODs
187    //   esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD);
188  
189    // Create ONLY the output containers for the data produced by the task.
190    // Get and connect other common input/output containers via the manager as below
191    //==============================================================================
192
193    mgr->ConnectInput  (esdfilter,  0, mgr->GetCommonInputContainer());
194    mgr->ConnectOutput (esdfilter,  0, mgr->GetCommonOutputContainer());
195    
196
197    //   mgr->ConnectInput  (esdmuonfilter, 0, mgr->GetCommonInputContainer());
198    if (useKineFilter) {
199       mgr->ConnectInput  (kinefilter,  0, mgr->GetCommonInputContainer());
200       mgr->ConnectOutput (kinefilter,  0, mgr->GetCommonOutputContainer());
201       AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(),
202                                                                    AliAnalysisManager::kOutputContainer,"pyxsec_hists.root");
203       mgr->ConnectOutput (kinefilter,  1,coutputEx);
204    }   
205   if (useCentralityTask) {
206        mgr->ConnectInput (ctask, 0, mgr->GetCommonInputContainer());
207        mgr->ConnectOutput(ctask, 0, mgr->GetCommonOutputContainer());
208    }
209
210    return esdfilter;
211  }
212