]>
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"); | |
42 | mgr->AddTask(esdfilter); | |
43 | // Muons | |
44 | AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter"); | |
45 | mgr->AddTask(esdmuonfilter); | |
46 | if(usePhysicsSelection){ | |
47 | esdfilter->SelectCollisionCandidates(AliVEvent::kAny); | |
48 | esdmuonfilter->SelectCollisionCandidates(AliVEvent::kAny); | |
49 | } | |
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 | ||
62 | // Cuts on primary tracks | |
63 | AliESDtrackCuts* esdTrackCutsL = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts(); | |
64 | ||
65 | // ITS stand-alone tracks | |
66 | AliESDtrackCuts* esdTrackCutsITSsa = new AliESDtrackCuts("ITS stand-alone Track Cuts", "ESD Track Cuts"); | |
67 | esdTrackCutsITSsa->SetRequireITSStandAlone(kTRUE); | |
68 | ||
69 | // Pixel OR necessary for the electrons | |
70 | AliESDtrackCuts *itsStrong = new AliESDtrackCuts("ITSorSPD", "pixel requirement for ITS"); | |
71 | itsStrong->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny); | |
72 | ||
73 | ||
74 | // PID for the electrons | |
75 | AliESDpidCuts *electronID = new AliESDpidCuts("Electrons", "Electron PID cuts"); | |
76 | electronID->SetTPCnSigmaCut(AliPID::kElectron, 3.); | |
77 | ||
78 | // tighter cuts on primary particles for high pT tracks | |
79 | // take the standard cuts, which include already | |
80 | // ITSrefit and use only primaries... | |
81 | AliESDtrackCuts* esdTrackCutsH = AliESDtrackCuts::GetStandardITSTPCTrackCuts2009(kTRUE); | |
82 | esdTrackCutsH->SetName("StandardFromAliESDTrackCuts"); | |
83 | ||
84 | ||
85 | // these are needed for older MC | |
58a3693a | 86 | /* |
ebad39a9 | 87 | AliESDtrackCuts* esdTrackCutsH1 = new AliESDtrackCuts("OldStandardTrackCuts+ITSRefit", "High pT ESD Track Cuts"); |
88 | esdTrackCutsH1->SetMinNClustersTPC(50); | |
89 | esdTrackCutsH1->SetMaxChi2PerClusterTPC(3.5); | |
90 | esdTrackCutsH1->SetRequireTPCRefit(kTRUE); | |
91 | esdTrackCutsH1->SetMaxDCAToVertexXY(2.4); | |
92 | esdTrackCutsH1->SetMaxDCAToVertexZ(3.2); | |
93 | esdTrackCutsH1->SetDCAToVertex2D(kTRUE); | |
94 | esdTrackCutsH1->SetRequireSigmaToVertex(kFALSE); | |
95 | esdTrackCutsH1->SetAcceptKinkDaughters(kFALSE); | |
96 | esdTrackCutsH1->SetRequireITSRefit(kTRUE); // additional cut | |
58a3693a | 97 | */ |
98 | AliESDtrackCuts* esdTrackCutsH1 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2009(); | |
99 | esdTrackCutsH1->SetName("StandardFromAliESDTrackCutsMaxRelPt"); | |
100 | esdTrackCutsH1->SetMaxRel1PtUncertainty(0.08); // new | |
101 | ||
ebad39a9 | 102 | |
103 | AliESDtrackCuts* esdTrackCutsH2 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2009(); | |
104 | esdTrackCutsH2->SetName("StandardFromAliESDTrackCutsNoSPDRequirement"); | |
105 | esdTrackCutsH2->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kOff); | |
106 | ||
107 | ||
108 | // Compose the filter | |
109 | AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter"); | |
110 | // 1 | |
111 | trackFilter->AddCuts(esdTrackCutsL); | |
112 | // 2 | |
113 | trackFilter->AddCuts(esdTrackCutsITSsa); | |
114 | // 4 | |
115 | trackFilter->AddCuts(itsStrong); | |
116 | itsStrong->SetFilterMask(1); // AND with Standard track cuts | |
117 | // 8 | |
118 | trackFilter->AddCuts(electronID); | |
119 | electronID->SetFilterMask(4); // AND with Pixel Cuts | |
120 | // 16 | |
121 | trackFilter->AddCuts(esdTrackCutsH); | |
122 | // 32 | |
123 | trackFilter->AddCuts(esdTrackCutsH1); | |
124 | // 64 | |
125 | trackFilter->AddCuts(esdTrackCutsH2); | |
126 | ||
127 | // Filter with cuts on V0s | |
128 | AliESDv0Cuts* esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts"); | |
129 | esdV0Cuts->SetMinRadius(0.2); | |
130 | esdV0Cuts->SetMaxRadius(200); | |
131 | esdV0Cuts->SetMinDcaPosToVertex(0.05); | |
132 | esdV0Cuts->SetMinDcaNegToVertex(0.05); | |
133 | esdV0Cuts->SetMaxDcaV0Daughters(1.5); | |
134 | esdV0Cuts->SetMinCosinePointingAngle(0.99); | |
135 | AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter"); | |
136 | // v0Filter->AddCuts(esdV0Cuts); | |
137 | ||
138 | esdfilter->SetTrackFilter(trackFilter); | |
139 | // esdfilter->SetV0Filter(v0Filter); | |
140 | ||
141 | // Enable writing of Muon AODs | |
142 | esdmuonfilter->SetWriteMuonAOD(writeMuonAOD); | |
143 | ||
144 | // Enable writing of Dimuon AODs | |
145 | esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD); | |
146 | ||
147 | // Create ONLY the output containers for the data produced by the task. | |
148 | // Get and connect other common input/output containers via the manager as below | |
149 | //============================================================================== | |
150 | ||
151 | mgr->ConnectInput (esdfilter, 0, mgr->GetCommonInputContainer()); | |
152 | mgr->ConnectOutput (esdfilter, 0, mgr->GetCommonOutputContainer()); | |
153 | ||
154 | ||
155 | // mgr->ConnectInput (esdmuonfilter, 0, mgr->GetCommonInputContainer()); | |
156 | if (useKineFilter) { | |
157 | mgr->ConnectInput (kinefilter, 0, mgr->GetCommonInputContainer()); | |
158 | mgr->ConnectOutput (kinefilter, 0, mgr->GetCommonOutputContainer()); | |
159 | AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(), | |
160 | AliAnalysisManager::kOutputContainer,"pyxsec_hists.root"); | |
161 | mgr->ConnectOutput (kinefilter, 1,coutputEx); | |
162 | } | |
163 | return esdfilter; | |
164 | } | |
165 |