]>
Commit | Line | Data |
---|---|---|
3a72645a | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | // | |
16 | // Signal cuts | |
17 | // Checks whether a particle (reconstructed or MC) is coming from MC Signal | |
18 | // For more information see implementation file | |
19 | // | |
20 | // Autor: | |
21 | // Markus Fasel <M.Fasel@gsi.de> | |
22 | // | |
23 | #include <TClass.h> | |
24 | #include <TMath.h> | |
25 | #include <TParticle.h> | |
26 | #include <TString.h> | |
27 | ||
e156c3bb | 28 | #include "AliAODTrack.h" |
3a72645a | 29 | #include "AliAODMCParticle.h" |
e156c3bb | 30 | #include "AliESDtrack.h" |
3a72645a | 31 | #include "AliLog.h" |
32 | #include "AliMCEvent.h" | |
33 | #include "AliMCParticle.h" | |
34 | #include "AliVParticle.h" | |
35 | ||
36 | #include "AliHFEsignalCuts.h" | |
37 | #include "AliHFEmcQA.h" | |
38 | ||
39 | ClassImp(AliHFEsignalCuts) | |
40 | ||
41 | //____________________________________________________________ | |
42 | AliHFEsignalCuts::AliHFEsignalCuts(): | |
43 | AliAnalysisCuts(), | |
44 | fMC(NULL), | |
45 | fMCQA(NULL) | |
46 | { | |
47 | // | |
48 | // Dummy constructor | |
49 | // | |
50 | } | |
51 | ||
52 | //____________________________________________________________ | |
53 | AliHFEsignalCuts::AliHFEsignalCuts(const Char_t *name, const Char_t *title): | |
54 | AliAnalysisCuts(name, title), | |
55 | fMC(NULL), | |
56 | fMCQA(NULL) | |
57 | { | |
58 | // | |
59 | // Default constructor | |
60 | // | |
61 | fMCQA = new AliHFEmcQA; | |
62 | if(fMCQA) fMCQA->Init(); | |
63 | } | |
64 | ||
65 | //____________________________________________________________ | |
66 | AliHFEsignalCuts::AliHFEsignalCuts(const AliHFEsignalCuts &ref): | |
67 | AliAnalysisCuts(ref), | |
68 | fMC(ref.fMC), | |
69 | fMCQA(ref.fMCQA) | |
70 | { | |
71 | // | |
72 | // Copy constructor | |
73 | // | |
74 | } | |
75 | ||
76 | //____________________________________________________________ | |
77 | AliHFEsignalCuts &AliHFEsignalCuts::operator=(const AliHFEsignalCuts &ref){ | |
78 | // | |
79 | // Assignment operator | |
80 | // | |
81 | if(this != &ref){ | |
82 | fMC = ref.fMC; | |
83 | fMCQA = ref.fMCQA; | |
84 | } | |
85 | return *this; | |
86 | } | |
87 | ||
88 | //____________________________________________________________ | |
89 | AliHFEsignalCuts::~AliHFEsignalCuts(){ | |
90 | // | |
91 | // Destructor | |
92 | // | |
93 | if(fMCQA) delete fMCQA; | |
94 | } | |
95 | ||
96 | //____________________________________________________________ | |
97 | void AliHFEsignalCuts::SetMCEvent(AliMCEvent *mc){ | |
98 | // | |
99 | // Set mc event | |
100 | // | |
101 | fMC = mc; | |
102 | if(fMCQA) fMCQA->SetMCEvent(mc); | |
103 | } | |
104 | ||
105 | //____________________________________________________________ | |
106 | Bool_t AliHFEsignalCuts::IsSelected(TObject *o){ | |
107 | // | |
108 | // Define signal as electron coming from charm or beauty | |
109 | // @TODO: Implement setter so that also either of them can be defined | |
110 | // as signal alone | |
111 | ||
112 | ||
113 | return IsCharmElectron(o) || IsBeautyElectron(o); | |
114 | /* | |
115 | //saving time? | |
116 | Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o)); | |
117 | if(esources>0)printf("esources= %d\n",esources); | |
118 | if(esources == AliHFEmcQA::kDirectCharm || esources == AliHFEmcQA::kDirectBeauty || esources == AliHFEmcQA::kBeautyCharm) // 1: direct D->e, 2: B->e 3: B->D->e | |
119 | return kTRUE; | |
120 | else | |
121 | return kFALSE; | |
122 | */ | |
123 | ||
124 | } | |
125 | ||
126 | //____________________________________________________________ | |
127 | Bool_t AliHFEsignalCuts::IsCharmElectron(const TObject * const o) const { | |
128 | // | |
129 | // Check if mother is coming from Charm | |
130 | // | |
bf892a6a | 131 | if(!dynamic_cast<const AliVParticle *>(o)) return kFALSE; |
3a72645a | 132 | Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o)); |
133 | if(esources == AliHFEmcQA::kDirectCharm) // 1: direct D->e | |
134 | return kTRUE; | |
135 | else | |
136 | return kFALSE; | |
137 | } | |
138 | ||
139 | //____________________________________________________________ | |
140 | Bool_t AliHFEsignalCuts::IsBeautyElectron(const TObject * const o) const { | |
141 | // | |
142 | // Check if mother is coming from Beauty | |
143 | // | |
bf892a6a | 144 | if(!dynamic_cast<const AliVParticle *>(o)) return kFALSE; |
3a72645a | 145 | Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o)); |
146 | if(esources == AliHFEmcQA::kDirectBeauty || esources == AliHFEmcQA::kBeautyCharm) // 2: B->e 3: B->D->e | |
147 | return kTRUE; | |
148 | else | |
149 | return kFALSE; | |
150 | } | |
151 | ||
152 | //____________________________________________________________ | |
153 | Bool_t AliHFEsignalCuts::IsGammaElectron(const TObject * const o) const { | |
154 | // | |
c2690925 | 155 | // Check for MC if the electron is coming from Gamma |
3a72645a | 156 | // |
bf892a6a | 157 | if(!dynamic_cast<const AliVParticle *>(o)) return kFALSE; |
3a72645a | 158 | Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o)); |
c2690925 | 159 | if(esources >= AliHFEmcQA::kGammaPi0 && esources <= AliHFEmcQA::kGammaRho0 ) // 4: conversion electrons |
160 | //if(esources == AliHFEmcQA::kGammaPi0 || esources == AliHFEmcQA::kGammaEta || esources == AliHFEmcQA::kGammaOmega || esources == AliHFEmcQA::kGammaPhi || esources == AliHFEmcQA::kGammaEtaPrime || esources == AliHFEmcQA::kGammaRho0 ) // 4: conversion electrons | |
161 | return kTRUE; | |
162 | else | |
163 | return kFALSE; | |
164 | } | |
165 | ||
166 | //____________________________________________________________ | |
167 | Bool_t AliHFEsignalCuts::IsNonHFElectron(const TObject * const o) const { | |
168 | // | |
169 | // Check for MC if the electron is coming from NonHFE except for conversion | |
170 | // | |
171 | if(!dynamic_cast<const AliVParticle *>(o)) return kFALSE; | |
172 | Int_t esources = GetElecSource(dynamic_cast<const AliVParticle *>(o)); | |
173 | if(esources == AliHFEmcQA:: kPi0 || esources == AliHFEmcQA::kEta || esources == AliHFEmcQA::kOmega || esources == AliHFEmcQA::kPhi || esources == AliHFEmcQA::kEtaPrime || esources == AliHFEmcQA::kRho0) // 4: conversion electrons | |
3a72645a | 174 | return kTRUE; |
175 | else | |
176 | return kFALSE; | |
177 | } | |
c2690925 | 178 | //____________________________________________________________ |
3a72645a | 179 | |
180 | /* | |
181 | //____________________________________________________________ | |
182 | Bool_t AliHFEsignalCuts::IsCharmElectron(const TObject * const o) const { | |
183 | // | |
184 | // Check if mother is coming from Charm | |
185 | // | |
186 | if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE; | |
187 | Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o))); | |
188 | AliDebug(1, Form("Mother PDG %d\n", motherpdg)); | |
189 | ||
190 | if((motherpdg % 1000) / 100 == 4) return kTRUE; // charmed meson, 3rd position in pdg code == 4 | |
191 | if(motherpdg / 1000 == 4) return kTRUE; // charmed baryon, 4th position in pdg code == 4 | |
192 | AliDebug(1, "No Charm\n"); | |
193 | return kFALSE; | |
194 | } | |
195 | ||
196 | //____________________________________________________________ | |
197 | Bool_t AliHFEsignalCuts::IsBeautyElectron(const TObject * const o) const { | |
198 | // | |
199 | // Check if mother is coming from Beauty | |
200 | // | |
201 | if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE; | |
202 | Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o))); | |
203 | AliDebug(1, Form("Mother PDG %d\n", motherpdg)); | |
204 | ||
205 | if((motherpdg % 1000) / 100 == 5) return kTRUE; // beauty meson, 3rd position in pdg code == 5 | |
206 | if(motherpdg / 1000 == 5) return kTRUE; // beauty baryon, 4th position in pdg code == 5 | |
207 | AliDebug(1, "No Beauty\n"); | |
208 | return kFALSE; | |
209 | } | |
210 | ||
211 | //____________________________________________________________ | |
212 | Bool_t AliHFEsignalCuts::IsGammaElectron(const TObject * const o) const { | |
213 | // | |
214 | // Check for MC if the electron is coming from Gamma | |
215 | // | |
216 | if(TMath::Abs(GetTrackPDG(dynamic_cast<const AliVParticle *>(o))) != 11) return kFALSE; | |
217 | Int_t motherpdg = TMath::Abs(GetMotherPDG(dynamic_cast<const AliVParticle *>(o))); | |
218 | AliDebug(1, Form("Mother PDG %d\n", motherpdg)); | |
219 | ||
220 | if(motherpdg!=22){ | |
221 | AliDebug(1, "No Gamma"); | |
222 | return kFALSE; | |
223 | } else { | |
224 | AliDebug(1, "Gamma"); | |
225 | return kTRUE; | |
226 | } | |
227 | } | |
228 | */ | |
229 | ||
230 | //____________________________________________________________ | |
231 | Int_t AliHFEsignalCuts::GetMotherPDG(const AliVParticle * const track) const { | |
232 | // | |
233 | // Get Mother Pdg code for reconstructed respectively MC tracks | |
234 | // | |
235 | if(!fMC){ | |
236 | AliDebug(1, "No MC Event Available\n"); | |
237 | return 0; | |
238 | } | |
239 | const AliVParticle *motherParticle = NULL, *mctrack = NULL; | |
240 | TString objectType = track->IsA()->GetName(); | |
241 | if(objectType.CompareTo("AliESDtrack") == 0 || objectType.CompareTo("AliAODTrack") == 0){ | |
242 | // Reconstructed track | |
243 | if(track->GetLabel()) | |
244 | mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel())); | |
245 | } else { | |
246 | // MCParticle | |
247 | mctrack = track; | |
248 | } | |
249 | ||
250 | if(!mctrack) return 0; | |
251 | ||
252 | Int_t motherPDG = 0; | |
253 | if(TString(mctrack->IsA()->GetName()).CompareTo("AliMCParticle") == 0){ | |
254 | // case MC Particle | |
bf892a6a | 255 | const AliMCParticle *esdmctrack = dynamic_cast<const AliMCParticle *>(mctrack); |
256 | if(esdmctrack) motherParticle = fMC->GetTrack(esdmctrack->Particle()->GetFirstMother()); | |
257 | if(motherParticle){ | |
258 | const AliMCParticle *esdmcmother = dynamic_cast<const AliMCParticle *>(motherParticle); | |
259 | if(esdmcmother) motherPDG = TMath::Abs(esdmcmother->Particle()->GetPdgCode()); | |
260 | } | |
3a72645a | 261 | } else { |
262 | // case AODMCParticle | |
bf892a6a | 263 | const AliAODMCParticle *aodmctrack = dynamic_cast<const AliAODMCParticle *>(mctrack); |
264 | if(aodmctrack) motherParticle = fMC->GetTrack(aodmctrack->GetMother()); | |
265 | if(motherParticle){ | |
266 | const AliAODMCParticle *aodmcmother = dynamic_cast<const AliAODMCParticle *>(motherParticle); | |
267 | if(aodmcmother) motherPDG = TMath::Abs(aodmcmother->GetPdgCode()); | |
268 | } | |
3a72645a | 269 | } |
270 | return motherPDG; | |
271 | } | |
272 | ||
273 | //____________________________________________________________ | |
274 | Int_t AliHFEsignalCuts::GetTrackPDG(const AliVParticle * const track) const { | |
275 | // | |
276 | // Return PDG code of a particle itself | |
277 | // | |
278 | if(!fMC){ | |
279 | AliDebug(1, "No MC Event Available\n"); | |
280 | return 0; | |
281 | } | |
282 | TString sourcetype = track->IsA()->GetName(); | |
283 | const AliVParticle *mctrack = NULL; | |
284 | if(!sourcetype.CompareTo("AliESDtrack") || !sourcetype.CompareTo("AliAODTrack")){ | |
285 | mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel())); | |
286 | } else mctrack = track; | |
287 | if(!mctrack) return 0; | |
288 | ||
289 | TString mctype = mctrack->IsA()->GetName(); | |
290 | Int_t trackPdg = 0; | |
291 | if(!mctype.CompareTo("AliMCParticle")){ | |
292 | const AliMCParticle *esdmc = dynamic_cast<const AliMCParticle *>(mctrack); | |
bf892a6a | 293 | if(esdmc) trackPdg = esdmc->Particle()->GetPdgCode(); |
3a72645a | 294 | } else { |
295 | const AliAODMCParticle *aodmc = dynamic_cast< const AliAODMCParticle *>(mctrack); | |
bf892a6a | 296 | if(aodmc) trackPdg = aodmc->GetPdgCode(); |
3a72645a | 297 | } |
298 | return trackPdg; | |
299 | } | |
300 | ||
301 | //____________________________________________________________ | |
302 | Int_t AliHFEsignalCuts::GetElecSource(const AliVParticle * const track) const { | |
303 | // | |
304 | // Return PDG code of a particle itself | |
305 | // | |
306 | ||
e3ae862b | 307 | if(!fMC){ |
308 | AliDebug(1, "No MC Event Available\n"); | |
309 | return 0; | |
310 | } | |
311 | if(!fMCQA){ | |
312 | AliDebug(1, "No MCQA Available\n"); | |
313 | return 0; | |
314 | } | |
315 | if(!track){ | |
316 | AliDebug(1, "Track not Available\n"); | |
317 | return 0; | |
318 | } | |
3a72645a | 319 | |
e156c3bb | 320 | TClass *tracktype; |
e3ae862b | 321 | const AliVParticle *mctrack = NULL; |
322 | TParticle *mcpart = NULL; | |
e156c3bb | 323 | if((tracktype = track->IsA()) == AliESDtrack::Class() || tracktype == AliAODTrack::Class()){ |
e3ae862b | 324 | mctrack = fMC->GetTrack(TMath::Abs(track->GetLabel())); |
325 | } else mctrack = track; | |
326 | if(!mctrack) return 0; | |
3a72645a | 327 | |
e3ae862b | 328 | Int_t eSource = 0; |
e156c3bb | 329 | if(mctrack->IsA() == AliMCParticle::Class()){ |
e3ae862b | 330 | const AliMCParticle *esdmc = dynamic_cast<const AliMCParticle *>(mctrack); |
331 | if(esdmc){ | |
332 | mcpart = esdmc->Particle(); | |
333 | eSource=fMCQA->GetElecSource(mcpart); | |
334 | } | |
335 | } else { | |
336 | return -1; | |
337 | } | |
338 | return eSource; | |
3a72645a | 339 | } |