]>
Commit | Line | Data |
---|---|---|
79bd4a7a | 1 | AliAnalysisTaskESDfilter *AddTaskESDFilter(Bool_t useKineFilter=kTRUE, |
2 | Bool_t writeMuonAOD=kFALSE, | |
daf100e1 | 3 | Bool_t writeDimuonAOD=kFALSE, |
216c8759 | 4 | Bool_t usePhysicsSelection=kFALSE, |
5 | Bool_t useCentralityTask=kFALSE) | |
3ef9ce84 | 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) { | |
c85963a0 | 13 | ::Error("AddTaskESDFilter", "No analysis manager to connect to."); |
3ef9ce84 | 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")) { | |
c85963a0 | 22 | ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an ESD input handler."); |
3ef9ce84 | 23 | return NULL; |
24 | } | |
25 | // Check if AOD output handler exist. | |
26 | AliAODHandler *aod_h = (AliAODHandler*)mgr->GetOutputEventHandler(); | |
27 | if (!aod_h) { | |
c85963a0 | 28 | ::Error("AddTaskESDFilter", "ESD filtering task needs the manager to have an AOD output handler."); |
3ef9ce84 | 29 | return NULL; |
099d8f33 | 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; | |
3ef9ce84 | 36 | } |
37 | ||
38 | // Create the task, add it to the manager and configure it. | |
099d8f33 | 39 | //=========================================================================== |
40 | // Barrel tracks filter | |
3ef9ce84 | 41 | AliAnalysisTaskESDfilter *esdfilter = new AliAnalysisTaskESDfilter("ESD Filter"); |
42 | mgr->AddTask(esdfilter); | |
099d8f33 | 43 | // Muons |
44 | AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter"); | |
45 | mgr->AddTask(esdmuonfilter); | |
daf100e1 | 46 | if(usePhysicsSelection){ |
0c6c629b | 47 | esdfilter->SelectCollisionCandidates(AliVEvent::kAny); |
48 | esdmuonfilter->SelectCollisionCandidates(AliVEvent::kAny); | |
daf100e1 | 49 | } |
3827f93a | 50 | |
51 | // Filtering of MC particles (decays conversions etc) | |
52 | // this task has to go AFTER all other filter tasks | |
53 | // since it fills the AODMC array with all | |
54 | // selected MC Particles, only this way we have the | |
55 | // AODMCparticle information available for following tasks | |
56 | AliAnalysisTaskMCParticleFilter *kinefilter = 0; | |
57 | if (useKineFilter) { | |
58 | kinefilter = new AliAnalysisTaskMCParticleFilter("Particle Kine Filter"); | |
59 | mgr->AddTask(kinefilter); | |
60 | } | |
61 | ||
216c8759 | 62 | AliAnalysisTaskAODCentralityMaker* ctask = 0; |
63 | ||
64 | if (useCentralityTask) { | |
65 | ctask = new AliAnalysisTaskAODCentralityMaker("AODCentralityMaker"); | |
66 | ctask->SetDeltaAODFileName("AliAODCentrality.root"); | |
67 | mgr->AddTask(ctask); | |
68 | } | |
69 | ||
099d8f33 | 70 | // Cuts on primary tracks |
31a3bbb0 | 71 | AliESDtrackCuts* esdTrackCutsL = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts(); |
a550fcbd | 72 | |
099d8f33 | 73 | // ITS stand-alone tracks |
04f2cd11 | 74 | AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("ITS stand-alone Track Cuts", "ESD Track Cuts"); |
31a3bbb0 | 75 | esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE); |
099d8f33 | 76 | |
a550fcbd | 77 | // Pixel OR necessary for the electrons |
78 | AliESDtrackCuts *itsStrong = new AliESDtrackCuts("ITSorSPD", "pixel requirement for ITS"); | |
79 | itsStrong->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny); | |
80 | ||
81 | ||
82 | // PID for the electrons | |
83 | AliESDpidCuts *electronID = new AliESDpidCuts("Electrons", "Electron PID cuts"); | |
84 | electronID->SetTPCnSigmaCut(AliPID::kElectron, 3.); | |
85 | ||
103431ad | 86 | // standard cuts with very loose DCA |
87 | AliESDtrackCuts* esdTrackCutsH = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE); | |
cfd26fc5 | 88 | esdTrackCutsH->SetMaxDCAToVertexXY(2.4); |
89 | esdTrackCutsH->SetMaxDCAToVertexZ(3.2); | |
90 | esdTrackCutsH->SetDCAToVertex2D(kTRUE); | |
cfd26fc5 | 91 | |
103431ad | 92 | // standard cuts with tight DCA cut |
93 | AliESDtrackCuts* esdTrackCutsH2 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(); | |
94 | ||
95 | // standard cuts with tight DCA but with requiring the first SDD cluster instead of an SPD cluster | |
96 | // tracks selected by this cut are exclusive to those selected by the previous cut | |
97 | AliESDtrackCuts* esdTrackCutsH3 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(); | |
98 | esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kNone); | |
99 | esdTrackCutsH3->SetClusterRequirementITS(AliESDtrackCuts::kSDD, AliESDtrackCuts::kFirst); | |
cfd26fc5 | 100 | |
a550fcbd | 101 | |
102 | // Compose the filter | |
3ef9ce84 | 103 | AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter"); |
a550fcbd | 104 | // 1 |
3ef9ce84 | 105 | trackFilter->AddCuts(esdTrackCutsL); |
a550fcbd | 106 | // 2 |
099d8f33 | 107 | trackFilter->AddCuts(esdTrackCutsITSsa); |
a550fcbd | 108 | // 4 |
109 | trackFilter->AddCuts(itsStrong); | |
110 | itsStrong->SetFilterMask(1); // AND with Standard track cuts | |
111 | // 8 | |
112 | trackFilter->AddCuts(electronID); | |
113 | electronID->SetFilterMask(4); // AND with Pixel Cuts | |
cfd26fc5 | 114 | // 16 |
115 | trackFilter->AddCuts(esdTrackCutsH); | |
103431ad | 116 | // 32 |
117 | trackFilter->AddCuts(esdTrackCutsH2); | |
118 | // 64 | |
119 | trackFilter->AddCuts(esdTrackCutsH3); | |
cfd26fc5 | 120 | |
a550fcbd | 121 | // Filter with cuts on V0s |
04f2cd11 | 122 | AliESDv0Cuts* esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts"); |
099d8f33 | 123 | esdV0Cuts->SetMinRadius(0.2); |
36cf8bb5 | 124 | esdV0Cuts->SetMaxRadius(200); |
099d8f33 | 125 | esdV0Cuts->SetMinDcaPosToVertex(0.05); |
126 | esdV0Cuts->SetMinDcaNegToVertex(0.05); | |
36cf8bb5 | 127 | esdV0Cuts->SetMaxDcaV0Daughters(1.5); |
099d8f33 | 128 | esdV0Cuts->SetMinCosinePointingAngle(0.99); |
129 | AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter"); | |
130 | v0Filter->AddCuts(esdV0Cuts); | |
131 | ||
3ef9ce84 | 132 | esdfilter->SetTrackFilter(trackFilter); |
099d8f33 | 133 | esdfilter->SetV0Filter(v0Filter); |
3ef9ce84 | 134 | |
f785fc59 | 135 | // Enable writing of Muon AODs |
48f1c230 | 136 | esdmuonfilter->SetWriteMuonAOD(writeMuonAOD); |
79bd4a7a | 137 | |
138 | // Enable writing of Dimuon AODs | |
139 | esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD); | |
cfd26fc5 | 140 | |
3ef9ce84 | 141 | // Create ONLY the output containers for the data produced by the task. |
142 | // Get and connect other common input/output containers via the manager as below | |
143 | //============================================================================== | |
144 | mgr->ConnectInput (esdfilter, 0, mgr->GetCommonInputContainer()); | |
145 | mgr->ConnectOutput (esdfilter, 0, mgr->GetCommonOutputContainer()); | |
099d8f33 | 146 | mgr->ConnectInput (esdmuonfilter, 0, mgr->GetCommonInputContainer()); |
147 | if (useKineFilter) { | |
148 | mgr->ConnectInput (kinefilter, 0, mgr->GetCommonInputContainer()); | |
149 | mgr->ConnectOutput (kinefilter, 0, mgr->GetCommonOutputContainer()); | |
a3b51fd2 | 150 | AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(), |
151 | AliAnalysisManager::kOutputContainer,"pyxsec_hists.root"); | |
84c04254 | 152 | mgr->ConnectOutput (kinefilter, 1,coutputEx); |
216c8759 | 153 | if (useCentralityTask) { |
154 | mgr->ConnectInput (ctask, 0, mgr->GetCommonInputContainer()); | |
155 | mgr->ConnectOutput(ctask, 0, mgr->GetCommonOutputContainer()); | |
156 | } | |
3827f93a | 157 | } |
3ef9ce84 | 158 | return esdfilter; |
cfd26fc5 | 159 | } |
160 |