]>
Commit | Line | Data |
---|---|---|
ebad39a9 | 1 | |
2 | AliAnalysisTaskESDfilter *AddTaskESDFilter(Bool_t useKineFilter=kTRUE, | |
3 | Bool_t writeMuonAOD=kFALSE, | |
4 | Bool_t writeDimuonAOD=kFALSE, | |
5 | Bool_t usePhysicsSelection=kFALSE) | |
6 | { | |
7 | // Creates a filter task and adds it to the analysis manager. | |
8 | ||
9 | // Get the pointer to the existing analysis manager via the static access method. | |
10 | //============================================================================== | |
11 | AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); | |
12 | if (!mgr) { | |
13 | ::Error("AddTaskESDFilter", "No analysis manager to connect to."); | |
14 | return NULL; | |
15 | } | |
16 | ||
17 | // This task requires an ESD input handler and an AOD output handler. | |
18 | // Check this using the analysis manager. | |
19 | //=============================================================================== | |
20 | TString type = mgr->GetInputEventHandler()->GetDataType(); | |
21 | if (!type.Contains("ESD")) { | |
22 | ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an ESD input handler."); | |
23 | return NULL; | |
24 | } | |
25 | // Check if AOD output handler exist. | |
26 | AliAODHandler *aod_h = (AliAODHandler*)mgr->GetOutputEventHandler(); | |
27 | if (!aod_h) { | |
28 | ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an AOD output handler."); | |
29 | return NULL; | |
30 | } | |
31 | // Check if MC handler is connected in case kine filter requested | |
32 | AliMCEventHandler *mcH = (AliMCEventHandler*)mgr->GetMCtruthEventHandler(); | |
33 | if (!mcH && useKineFilter) { | |
34 | ::Error("AddTaskESDFilter", "No MC handler connected while kine filtering requested"); | |
35 | return NULL; | |
36 | } | |
37 | ||
38 | // Create the task, add it to the manager and configure it. | |
39 | //=========================================================================== | |
40 | // Barrel tracks filter | |
41 | AliAnalysisTaskESDfilter *esdfilter = new AliAnalysisTaskESDfilter("ESD Filter"); | |
3095177c | 42 | |
43 | // Make the AOD a little bit lighter and filtering faster | |
44 | ||
45 | esdfilter->DisableCascades(); | |
46 | // esdfilter->DisableV0s(); | |
47 | esdfilter->DisableKinks(); | |
48 | // esdfilter->DisableTracks(); | |
49 | esdfilter->DisablePmdClusters(); | |
50 | // esdfilter->DisableCaloClusters(); | |
51 | // esdfilter->DisableCells(); | |
52 | esdfilter->DisableTracklets(); | |
53 | ||
54 | ||
ebad39a9 | 55 | mgr->AddTask(esdfilter); |
56 | // Muons | |
48602d36 | 57 | // AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter"); |
58 | // mgr->AddTask(esdmuonfilter); | |
ebad39a9 | 59 | if(usePhysicsSelection){ |
60 | esdfilter->SelectCollisionCandidates(AliVEvent::kAny); | |
48602d36 | 61 | // esdmuonfilter->SelectCollisionCandidates(AliVEvent::kAny); |
ebad39a9 | 62 | } |
63 | ||
64 | // Filtering of MC particles (decays conversions etc) | |
65 | // this task has to go AFTER all other filter tasks | |
66 | // since it fills the AODMC array with all | |
67 | // selected MC Particles, only this way we have the | |
68 | // AODMCparticle information available for following tasks | |
69 | AliAnalysisTaskMCParticleFilter *kinefilter = 0; | |
70 | if (useKineFilter) { | |
71 | kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Kine Filter"); | |
48602d36 | 72 | if(usePhysicsSelection)kinefilter->SelectCollisionCandidates(AliVEvent::kAny); |
ebad39a9 | 73 | mgr->AddTask(kinefilter); |
74 | } | |
75 | ||
76 | // Cuts on primary tracks | |
77 | AliESDtrackCuts* esdTrackCutsL = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts(); | |
78 | ||
79 | // ITS stand-alone tracks | |
80 | AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("ITS stand-alone Track Cuts", "ESD Track Cuts"); | |
81 | esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE); | |
82 | ||
83 | // Pixel OR necessary for the electrons | |
84 | AliESDtrackCuts *itsStrong = new AliESDtrackCuts("ITSorSPD", "pixel requirement for ITS"); | |
85 | itsStrong->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny); | |
86 | ||
87 | ||
88 | // PID for the electrons | |
89 | AliESDpidCuts *electronID = new AliESDpidCuts("Electrons", "Electron PID cuts"); | |
90 | electronID->SetTPCnSigmaCut(AliPID::kElectron, 3.); | |
91 | ||
92 | // tighter cuts on primary particles for high pT tracks | |
93 | // take the standard cuts, which include already | |
94 | // ITSrefit and use only primaries... | |
3456f527 | 95 | |
39e7e8ab | 96 | // loose DCA cuts |
3456f527 | 97 | AliESDtrackCuts* esdTrackCutsH = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE); |
98 | esdTrackCutsH->SetMaxDCAToVertexXY(2.4); | |
99 | esdTrackCutsH->SetMaxDCAToVertexZ(3.2); | |
100 | esdTrackCutsH->SetDCAToVertex2D(kTRUE); | |
f4132e7d | 101 | esdTrackCutsH->SetDCAToVertex2D(kTRUE); |
102 | esdTrackCutsH->SetPtRange(0.15,1E10); | |
3456f527 | 103 | |
104 | // standard cuts with tight DCA cut | |
105 | AliESDtrackCuts* esdTrackCutsH2 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(); | |
f4132e7d | 106 | esdTrackCutsH2->SetPtRange(0.15,1E10); |
3456f527 | 107 | |
108 | // standard cuts with tight DCA but with requiring the first SDD cluster instead of an SPD cluster | |
109 | // tracks selected by this cut are exclusive to those selected by the previous cut | |
110 | AliESDtrackCuts* esdTrackCutsH3 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(); | |
111 | esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone); | |
112 | esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSDD, AliESDtrackCuts::kFirst); | |
113 | ||
ebad39a9 | 114 | |
1c79997e | 115 | AliESDtrackCuts* esdTrackCutsH4 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE); |
116 | esdTrackCutsH4->SetMaxDCAToVertexXY(2.4); | |
117 | esdTrackCutsH4->SetMaxDCAToVertexZ(3.2); | |
118 | esdTrackCutsH4->SetDCAToVertex2D(kTRUE); | |
119 | esdTrackCutsH4->SetDCAToVertex2D(kTRUE); | |
120 | esdTrackCutsH4->SetPtRange(0.15,1E10); | |
121 | esdTrackCutsH4->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone); | |
122 | esdTrackCutsH4->SetClusterRequirementITS(AliESDtrackCuts::kSDD, AliESDtrackCuts::kFirst); | |
58a3693a | 123 | |
070f06d2 | 124 | // TPC only tracks |
125 | AliESDtrackCuts* esdTrackCutsTPCOnly = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts(); | |
126 | esdTrackCutsTPCOnly->SetMinNClustersTPC(70); | |
127 | ||
ebad39a9 | 128 | // Compose the filter |
129 | AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter"); | |
39e7e8ab | 130 | // 1, 1<<0 |
ebad39a9 | 131 | trackFilter->AddCuts(esdTrackCutsL); |
39e7e8ab | 132 | // 2 1<<1 |
ebad39a9 | 133 | trackFilter->AddCuts(esdTrackCutsITSsa); |
39e7e8ab | 134 | // 4 1<<2 |
ebad39a9 | 135 | trackFilter->AddCuts(itsStrong); |
136 | itsStrong->SetFilterMask(1); // AND with Standard track cuts | |
39e7e8ab | 137 | // 8 1<<3 |
ebad39a9 | 138 | trackFilter->AddCuts(electronID); |
139 | electronID->SetFilterMask(4); // AND with Pixel Cuts | |
39e7e8ab | 140 | // 16 1<<4 |
ebad39a9 | 141 | trackFilter->AddCuts(esdTrackCutsH); |
39e7e8ab | 142 | // 32 1<<5 |
ebad39a9 | 143 | trackFilter->AddCuts(esdTrackCutsH2); |
39e7e8ab | 144 | // 64 1<<6 |
3456f527 | 145 | trackFilter->AddCuts(esdTrackCutsH3); |
39e7e8ab | 146 | // 128 1<<7 |
1c79997e | 147 | trackFilter->AddCuts(esdTrackCutsH4); |
070f06d2 | 148 | // 256 1<<8 |
149 | trackFilter->AddCuts(esdTrackCutsTPCOnly); | |
150 | esdfilter->SetTPCOnlyFilterMask(1<<8); | |
ebad39a9 | 151 | // Filter with cuts on V0s |
152 | AliESDv0Cuts* esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts"); | |
153 | esdV0Cuts->SetMinRadius(0.2); | |
154 | esdV0Cuts->SetMaxRadius(200); | |
155 | esdV0Cuts->SetMinDcaPosToVertex(0.05); | |
156 | esdV0Cuts->SetMinDcaNegToVertex(0.05); | |
157 | esdV0Cuts->SetMaxDcaV0Daughters(1.5); | |
158 | esdV0Cuts->SetMinCosinePointingAngle(0.99); | |
159 | AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter"); | |
070f06d2 | 160 | v0Filter->AddCuts(esdV0Cuts); |
ebad39a9 | 161 | |
162 | esdfilter->SetTrackFilter(trackFilter); | |
070f06d2 | 163 | esdfilter->SetV0Filter(v0Filter); |
ebad39a9 | 164 | |
165 | // Enable writing of Muon AODs | |
48602d36 | 166 | // esdmuonfilter->SetWriteMuonAOD(writeMuonAOD); |
ebad39a9 | 167 | |
168 | // Enable writing of Dimuon AODs | |
48602d36 | 169 | // esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD); |
ebad39a9 | 170 | |
171 | // Create ONLY the output containers for the data produced by the task. | |
172 | // Get and connect other common input/output containers via the manager as below | |
173 | //============================================================================== | |
174 | ||
175 | mgr->ConnectInput (esdfilter, 0, mgr->GetCommonInputContainer()); | |
176 | mgr->ConnectOutput (esdfilter, 0, mgr->GetCommonOutputContainer()); | |
177 | ||
178 | ||
179 | // mgr->ConnectInput (esdmuonfilter, 0, mgr->GetCommonInputContainer()); | |
180 | if (useKineFilter) { | |
181 | mgr->ConnectInput (kinefilter, 0, mgr->GetCommonInputContainer()); | |
182 | mgr->ConnectOutput (kinefilter, 0, mgr->GetCommonOutputContainer()); | |
183 | AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(), | |
184 | AliAnalysisManager::kOutputContainer,"pyxsec_hists.root"); | |
185 | mgr->ConnectOutput (kinefilter, 1,coutputEx); | |
186 | } | |
187 | return esdfilter; | |
188 | } | |
189 |