]>
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 | |
86 | AliESDtrackCuts* esdTrackCutsH1 = new AliESDtrackCuts("OldStandardTrackCuts+ITSRefit", "High pT ESD Track Cuts"); | |
87 | esdTrackCutsH1->SetMinNClustersTPC(50); | |
88 | esdTrackCutsH1->SetMaxChi2PerClusterTPC(3.5); | |
89 | esdTrackCutsH1->SetRequireTPCRefit(kTRUE); | |
90 | esdTrackCutsH1->SetMaxDCAToVertexXY(2.4); | |
91 | esdTrackCutsH1->SetMaxDCAToVertexZ(3.2); | |
92 | esdTrackCutsH1->SetDCAToVertex2D(kTRUE); | |
93 | esdTrackCutsH1->SetRequireSigmaToVertex(kFALSE); | |
94 | esdTrackCutsH1->SetAcceptKinkDaughters(kFALSE); | |
95 | esdTrackCutsH1->SetRequireITSRefit(kTRUE); // additional cut | |
96 | ||
97 | AliESDtrackCuts* esdTrackCutsH2 = AliESDtrackCuts::GetStandardITSTPCTrackCuts2009(); | |
98 | esdTrackCutsH2->SetName("StandardFromAliESDTrackCutsNoSPDRequirement"); | |
99 | esdTrackCutsH2->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kOff); | |
100 | ||
101 | ||
102 | // Compose the filter | |
103 | AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter"); | |
104 | // 1 | |
105 | trackFilter->AddCuts(esdTrackCutsL); | |
106 | // 2 | |
107 | trackFilter->AddCuts(esdTrackCutsITSsa); | |
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 | |
114 | // 16 | |
115 | trackFilter->AddCuts(esdTrackCutsH); | |
116 | // 32 | |
117 | trackFilter->AddCuts(esdTrackCutsH1); | |
118 | // 64 | |
119 | trackFilter->AddCuts(esdTrackCutsH2); | |
120 | ||
121 | // Filter with cuts on V0s | |
122 | AliESDv0Cuts* esdV0Cuts = new AliESDv0Cuts("Standard V0 Cuts pp", "ESD V0 Cuts"); | |
123 | esdV0Cuts->SetMinRadius(0.2); | |
124 | esdV0Cuts->SetMaxRadius(200); | |
125 | esdV0Cuts->SetMinDcaPosToVertex(0.05); | |
126 | esdV0Cuts->SetMinDcaNegToVertex(0.05); | |
127 | esdV0Cuts->SetMaxDcaV0Daughters(1.5); | |
128 | esdV0Cuts->SetMinCosinePointingAngle(0.99); | |
129 | AliAnalysisFilter* v0Filter = new AliAnalysisFilter("v0Filter"); | |
130 | // v0Filter->AddCuts(esdV0Cuts); | |
131 | ||
132 | esdfilter->SetTrackFilter(trackFilter); | |
133 | // esdfilter->SetV0Filter(v0Filter); | |
134 | ||
135 | // Enable writing of Muon AODs | |
136 | esdmuonfilter->SetWriteMuonAOD(writeMuonAOD); | |
137 | ||
138 | // Enable writing of Dimuon AODs | |
139 | esdmuonfilter->SetWriteDimuonAOD(writeDimuonAOD); | |
140 | ||
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 | ||
145 | mgr->ConnectInput (esdfilter, 0, mgr->GetCommonInputContainer()); | |
146 | mgr->ConnectOutput (esdfilter, 0, mgr->GetCommonOutputContainer()); | |
147 | ||
148 | ||
149 | // mgr->ConnectInput (esdmuonfilter, 0, mgr->GetCommonInputContainer()); | |
150 | if (useKineFilter) { | |
151 | mgr->ConnectInput (kinefilter, 0, mgr->GetCommonInputContainer()); | |
152 | mgr->ConnectOutput (kinefilter, 0, mgr->GetCommonOutputContainer()); | |
153 | AliAnalysisDataContainer *coutputEx = mgr->CreateContainer("cFilterList", TList::Class(), | |
154 | AliAnalysisManager::kOutputContainer,"pyxsec_hists.root"); | |
155 | mgr->ConnectOutput (kinefilter, 1,coutputEx); | |
156 | } | |
157 | return esdfilter; | |
158 | } | |
159 |