]>
Commit | Line | Data |
---|---|---|
809a4336 | 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 | **************************************************************************/ | |
50685501 | 15 | // |
16 | // Class for TRD PID | |
17 | // Implements the abstract base class AliHFEpidbase | |
18 | // Make PID does the PID decision | |
19 | // Class further contains TRD specific cuts and QA histograms | |
20 | // | |
21 | // Authors: | |
22 | // Markus Fasel <M.Fasel@gsi.de> | |
23 | // | |
809a4336 | 24 | #include <TH1F.h> |
75d81601 | 25 | #include <TH2F.h> |
e3fc062d | 26 | #include <THnSparse.h> |
50685501 | 27 | #include <TList.h> |
809a4336 | 28 | #include <TString.h> |
809a4336 | 29 | |
6555e2ad | 30 | #include "AliAODPid.h" |
722347d8 | 31 | #include "AliAODTrack.h" |
32 | #include "AliAODMCParticle.h" | |
809a4336 | 33 | #include "AliESDtrack.h" |
75d81601 | 34 | #include "AliLog.h" |
722347d8 | 35 | #include "AliMCParticle.h" |
8c1c76e9 | 36 | #include "AliOADBContainer.h" |
809a4336 | 37 | #include "AliPID.h" |
8c1c76e9 | 38 | #include "AliPIDResponse.h" |
809a4336 | 39 | |
8c1c76e9 | 40 | #include "AliHFEOADBThresholdsTRD.h" |
3a72645a | 41 | #include "AliHFEpidQAmanager.h" |
809a4336 | 42 | #include "AliHFEpidTRD.h" |
43 | ||
44 | ClassImp(AliHFEpidTRD) | |
45 | ||
faee3b18 | 46 | //___________________________________________________________________ |
47 | AliHFEpidTRD::AliHFEpidTRD() : | |
48 | AliHFEpidBase() | |
8c1c76e9 | 49 | , fOADBThresholds(NULL) |
50 | , fMinP(0.5) | |
51 | , fNTracklets(6) | |
e17c1f86 | 52 | , fCutNTracklets(0) |
8c1c76e9 | 53 | , fRunNumber(0) |
54 | , fElectronEfficiency(0.90) | |
e156c3bb | 55 | , fTotalChargeInSlice0(kFALSE) |
b593c849 | 56 | , fTRD2DPID(kFALSE) |
faee3b18 | 57 | { |
58 | // | |
59 | // default constructor | |
60 | // | |
bf892a6a | 61 | memset(fThreshParams, 0, sizeof(Double_t) * kThreshParams); |
faee3b18 | 62 | } |
63 | ||
809a4336 | 64 | //___________________________________________________________________ |
65 | AliHFEpidTRD::AliHFEpidTRD(const char* name) : | |
66 | AliHFEpidBase(name) | |
8c1c76e9 | 67 | , fOADBThresholds(NULL) |
68 | , fMinP(0.5) | |
69 | , fNTracklets(6) | |
e17c1f86 | 70 | , fCutNTracklets(0) |
8c1c76e9 | 71 | , fRunNumber(0) |
67fe7bd0 | 72 | , fElectronEfficiency(0.91) |
e156c3bb | 73 | , fTotalChargeInSlice0(kFALSE) |
b593c849 | 74 | , fTRD2DPID(kFALSE) |
809a4336 | 75 | { |
76 | // | |
77 | // default constructor | |
78 | // | |
dbe3abbe | 79 | memset(fThreshParams, 0, sizeof(Double_t) * kThreshParams); |
809a4336 | 80 | } |
81 | ||
82 | //___________________________________________________________________ | |
83 | AliHFEpidTRD::AliHFEpidTRD(const AliHFEpidTRD &ref): | |
84 | AliHFEpidBase("") | |
8c1c76e9 | 85 | , fOADBThresholds(NULL) |
67fe7bd0 | 86 | , fMinP(ref.fMinP) |
8c1c76e9 | 87 | , fNTracklets(ref.fNTracklets) |
e17c1f86 | 88 | , fCutNTracklets(ref.fCutNTracklets) |
8c1c76e9 | 89 | , fRunNumber(ref.fRunNumber) |
67fe7bd0 | 90 | , fElectronEfficiency(ref.fElectronEfficiency) |
e156c3bb | 91 | , fTotalChargeInSlice0(ref.fTotalChargeInSlice0) |
b593c849 | 92 | , fTRD2DPID(ref.fTRD2DPID) |
809a4336 | 93 | { |
94 | // | |
95 | // Copy constructor | |
96 | // | |
dbe3abbe | 97 | memset(fThreshParams, 0, sizeof(Double_t) * kThreshParams); |
809a4336 | 98 | ref.Copy(*this); |
99 | } | |
100 | ||
101 | //___________________________________________________________________ | |
102 | AliHFEpidTRD &AliHFEpidTRD::operator=(const AliHFEpidTRD &ref){ | |
103 | // | |
104 | // Assignment operator | |
105 | // | |
106 | if(this != &ref){ | |
107 | ref.Copy(*this); | |
108 | } | |
109 | return *this; | |
110 | } | |
111 | ||
112 | //___________________________________________________________________ | |
113 | void AliHFEpidTRD::Copy(TObject &ref) const { | |
114 | // | |
115 | // Performs the copying of the object | |
116 | // | |
117 | AliHFEpidTRD &target = dynamic_cast<AliHFEpidTRD &>(ref); | |
bf892a6a | 118 | |
e3fc062d | 119 | target.fMinP = fMinP; |
8c1c76e9 | 120 | target.fNTracklets = fNTracklets; |
e17c1f86 | 121 | target.fCutNTracklets = fCutNTracklets; |
8c1c76e9 | 122 | target.fRunNumber = fRunNumber; |
e156c3bb | 123 | target.fTotalChargeInSlice0 = fTotalChargeInSlice0; |
b593c849 | 124 | target.fTRD2DPID = fTRD2DPID; |
67fe7bd0 | 125 | target.fElectronEfficiency = fElectronEfficiency; |
dbe3abbe | 126 | memcpy(target.fThreshParams, fThreshParams, sizeof(Double_t) * kThreshParams); |
809a4336 | 127 | AliHFEpidBase::Copy(ref); |
128 | } | |
129 | ||
130 | //___________________________________________________________________ | |
131 | AliHFEpidTRD::~AliHFEpidTRD(){ | |
132 | // | |
133 | // Destructor | |
134 | // | |
809a4336 | 135 | } |
136 | ||
137 | //______________________________________________________ | |
8c1c76e9 | 138 | Bool_t AliHFEpidTRD::InitializePID(Int_t run){ |
b593c849 | 139 | // |
140 | // InitializePID call different init function depending on TRD PID method | |
141 | // | |
142 | // | |
143 | ||
144 | //if(fTRD2DPID) return Initialize2D(run); | |
145 | if(fTRD2DPID) return kTRUE; | |
146 | else return Initialize1D(run); | |
147 | ||
148 | ||
149 | ||
150 | } | |
151 | ||
152 | //______________________________________________________ | |
153 | Bool_t AliHFEpidTRD::Initialize1D(Int_t run){ | |
809a4336 | 154 | // |
155 | // InitializePID: Load TRD thresholds and create the electron efficiency axis | |
156 | // to navigate | |
157 | // | |
8c1c76e9 | 158 | AliDebug(1, Form("Initializing TRD PID for run %d", run)); |
159 | if(InitParamsFromOADB(run)){ | |
160 | SetBit(kThresholdsInitialized); | |
161 | return kTRUE; | |
bf892a6a | 162 | } |
8c1c76e9 | 163 | AliDebug(1, Form("Threshold Parameters for %d tracklets and an electron efficiency %f loaded:", fNTracklets, fElectronEfficiency)); |
164 | AliDebug(1, Form("Params: [%f|%f|%f|%f]", fThreshParams[0], fThreshParams[1], fThreshParams[2], fThreshParams[3])); | |
165 | fRunNumber = run; | |
166 | return kFALSE; | |
809a4336 | 167 | } |
168 | ||
b593c849 | 169 | /* |
170 | //______________________________________________________ | |
171 | Bool_t AliHFEpidTRD::Initialize2D(Int_t run){ | |
172 | // | |
173 | // Initialize2DimPID | |
174 | // | |
175 | // | |
176 | ||
177 | return kTRUE; | |
178 | ||
179 | } | |
180 | */ | |
181 | ||
809a4336 | 182 | //______________________________________________________ |
6555e2ad | 183 | Int_t AliHFEpidTRD::IsSelected(const AliHFEpidObject *track, AliHFEpidQAmanager *pidqa) const { |
b593c849 | 184 | // |
185 | // Does PID for TRD alone: | |
186 | // PID algorithm selected according to flag | |
187 | // | |
188 | // | |
189 | ||
190 | if(fTRD2DPID) return IsSelected2D(track, pidqa); | |
191 | else return IsSelected1D(track, pidqa); | |
192 | ||
193 | ||
194 | } | |
195 | ||
196 | //______________________________________________________ | |
197 | Int_t AliHFEpidTRD::IsSelected1D(const AliHFEpidObject *track, AliHFEpidQAmanager *pidqa) const { | |
809a4336 | 198 | // |
199 | // Does PID for TRD alone: | |
200 | // PID thresholds based on 90% Electron Efficiency level approximated by a linear | |
201 | // step function | |
202 | // | |
8c1c76e9 | 203 | if(!TestBit(kThresholdsInitialized)) { |
204 | AliDebug(1,"Threshold Parameters not available"); | |
205 | return 0; | |
206 | } | |
207 | AliDebug(2, "Applying TRD PID"); | |
208 | if(!fkPIDResponse){ | |
209 | AliDebug(2, "Cannot process track"); | |
3a72645a | 210 | return 0; |
211 | } | |
722347d8 | 212 | |
e17c1f86 | 213 | |
8c1c76e9 | 214 | /* |
215 | const AliESDtrack *esdt = dynamic_cast<const AliESDtrack *>(track->GetRecTrack()); | |
216 | printf("checking IdentifiedAsElectronTRD, number of Tracklets: %d\n", esdt->GetTRDntrackletsPID()); | |
217 | if(fkPIDResponse->IdentifiedAsElectronTRD(dynamic_cast<const AliVTrack *>(track->GetRecTrack()), 0.8)) printf("Track identified as electron\n"); | |
218 | else printf("Track rejected\n"); | |
219 | */ | |
3a72645a | 220 | AliHFEpidObject::AnalysisType_t anatype = track->IsESDanalysis() ? AliHFEpidObject::kESDanalysis: AliHFEpidObject::kAODanalysis; |
221 | Double_t p = GetP(track->GetRecTrack(), anatype); | |
222 | if(p < fMinP){ | |
8c1c76e9 | 223 | AliDebug(2, Form("Track momentum below %f", fMinP)); |
3a72645a | 224 | return 0; |
225 | } | |
809a4336 | 226 | |
3a72645a | 227 | if(pidqa) pidqa->ProcessTrack(track, AliHFEpid::kTRDpid, AliHFEdetPIDqa::kBeforePID); |
e17c1f86 | 228 | |
229 | if(fCutNTracklets > 0){ | |
230 | AliDebug(1, Form("Number of tracklets cut applied: %d\n", fCutNTracklets)); | |
231 | Int_t ntracklets = track->GetRecTrack() ? track->GetRecTrack()->GetTRDntrackletsPID() : 0; | |
232 | if(TestBit(kExactTrackletCut)){ | |
233 | AliDebug(1, Form("Exact cut applied: %d tracklets found\n", ntracklets)); | |
234 | if(ntracklets != fCutNTracklets) return 0; | |
235 | } else { | |
236 | AliDebug(1, Form("Greater Equal cut applied: %d tracklets found\n", ntracklets)); | |
237 | if(ntracklets < fCutNTracklets) return 0; | |
238 | } | |
239 | } | |
240 | AliDebug(1,"Track selected\n"); | |
241 | ||
242 | Double_t electronLike = GetElectronLikelihood(track->GetRecTrack(), anatype); | |
8c1c76e9 | 243 | Double_t threshold; |
244 | if(TestBit(kSelectCutOnTheFly)){ | |
e17c1f86 | 245 | threshold = GetTRDthresholds(p, track->GetRecTrack()->GetTRDntrackletsPID()); |
8c1c76e9 | 246 | } else { |
247 | threshold = GetTRDthresholds(p); | |
248 | } | |
249 | AliDebug(2, Form("Threshold: %f\n", threshold)); | |
3a72645a | 250 | if(electronLike > threshold){ |
251 | if(pidqa) pidqa->ProcessTrack(track, AliHFEpid::kTRDpid, AliHFEdetPIDqa::kAfterPID); | |
0792aa82 | 252 | return 11; |
253 | } | |
254 | return 211; | |
3a72645a | 255 | |
809a4336 | 256 | } |
257 | ||
b593c849 | 258 | //______________________________________________________ |
259 | Int_t AliHFEpidTRD::IsSelected2D(const AliHFEpidObject *track, AliHFEpidQAmanager *pidqa) const { | |
260 | // | |
261 | // 2D TRD PID | |
262 | // | |
263 | // | |
264 | // | |
265 | AliDebug(2, "Applying TRD PID"); | |
266 | if(!fkPIDResponse){ | |
267 | AliDebug(2, "Cannot process track"); | |
268 | return 0; | |
269 | } | |
270 | ||
271 | AliHFEpidObject::AnalysisType_t anatype = track->IsESDanalysis() ? AliHFEpidObject::kESDanalysis: AliHFEpidObject::kAODanalysis; | |
272 | Double_t p = GetP(track->GetRecTrack(), anatype); | |
273 | if(p < fMinP){ | |
274 | AliDebug(2, Form("Track momentum below %f", fMinP)); | |
275 | return 0; | |
276 | } | |
277 | ||
278 | if(pidqa) pidqa->ProcessTrack(track, AliHFEpid::kTRDpid, AliHFEdetPIDqa::kBeforePID); | |
279 | ||
280 | if(fCutNTracklets > 0){ | |
281 | AliDebug(1, Form("Number of tracklets cut applied: %d\n", fCutNTracklets)); | |
282 | Int_t ntracklets = track->GetRecTrack() ? track->GetRecTrack()->GetTRDntrackletsPID() : 0; | |
283 | if(TestBit(kExactTrackletCut)){ | |
284 | AliDebug(1, Form("Exact cut applied: %d tracklets found\n", ntracklets)); | |
285 | if(ntracklets != fCutNTracklets) return 0; | |
286 | } else { | |
287 | AliDebug(1, Form("Greater Equal cut applied: %d tracklets found\n", ntracklets)); | |
288 | if(ntracklets < fCutNTracklets) return 0; | |
289 | } | |
290 | } | |
291 | AliDebug(1,"Track selected\n"); | |
292 | ||
293 | Int_t centralitybin = track->IsPbPb() ? track->GetCentrality() + 1 : 0; | |
294 | Float_t fCentralityLimitsdefault[12]= {0.,5.,10., 20., 30., 40., 50., 60.,70.,80., 90., 100.}; | |
295 | Float_t centrality=fCentralityLimitsdefault[centralitybin]+1; | |
296 | ||
297 | ||
298 | if(fkPIDResponse->IdentifiedAsElectronTRD(track->GetRecTrack(),fElectronEfficiency,centrality,AliTRDPIDResponse::kLQ2D)){ | |
299 | AliDebug(2, Form("Electron effi: %f\n", fElectronEfficiency)); | |
300 | return 11; | |
301 | } else return 211; | |
302 | ||
303 | } | |
304 | ||
809a4336 | 305 | //___________________________________________________________________ |
8c1c76e9 | 306 | Double_t AliHFEpidTRD::GetTRDthresholds(Double_t p, UInt_t nTracklets) const { |
dbe3abbe | 307 | // |
308 | // Return momentum dependent and electron efficiency dependent TRD thresholds | |
8c1c76e9 | 309 | // Determine threshold based on the number of tracklets on the fly, electron efficiency not modified |
dbe3abbe | 310 | // |
8c1c76e9 | 311 | Double_t threshParams[4]; |
e17c1f86 | 312 | AliDebug(1, Form("Select cut for %d tracklets\n", nTracklets)); |
8c1c76e9 | 313 | // Get threshold paramters for the given number of tracklets from OADB container |
314 | AliHFEOADBThresholdsTRD *thresholds = dynamic_cast<AliHFEOADBThresholdsTRD *>(fOADBThresholds->GetObject(fRunNumber)); | |
315 | if(!thresholds){ | |
316 | AliDebug(1, Form("Thresholds for run %d not in the OADB", fRunNumber)); | |
317 | return 0.; | |
318 | } | |
e17c1f86 | 319 | if(!thresholds->GetThresholdParameters(nTracklets, fElectronEfficiency, threshParams)){ |
320 | AliDebug(1, "loading thresholds failed\n"); | |
321 | return 0.; | |
322 | } | |
8c1c76e9 | 323 | Double_t threshold = 1. - threshParams[0] - threshParams[1] * p - threshParams[2] * TMath::Exp(-threshParams[3] * p); |
0792aa82 | 324 | return TMath::Max(TMath::Min(threshold, 0.99), 0.2); // truncate the threshold upperwards to 0.999 and lowerwards to 0.2 and exclude unphysical values |
809a4336 | 325 | } |
326 | ||
bf892a6a | 327 | //___________________________________________________________________ |
8c1c76e9 | 328 | Double_t AliHFEpidTRD::GetTRDthresholds(Double_t p) const { |
bf892a6a | 329 | // |
8c1c76e9 | 330 | // Return momentum dependent and electron efficiency dependent TRD thresholds |
331 | // | |
332 | Double_t threshold = 1. - fThreshParams[0] - fThreshParams[1] * p - fThreshParams[2] * TMath::Exp(-fThreshParams[3] * p); | |
333 | return TMath::Max(TMath::Min(threshold, 0.99), 0.2); // truncate the threshold upperwards to 0.999 and lowerwards to 0.2 and exclude unphysical values | |
bf892a6a | 334 | } |
335 | ||
809a4336 | 336 | |
e3fc062d | 337 | //___________________________________________________________________ |
8c1c76e9 | 338 | Bool_t AliHFEpidTRD::InitParamsFromOADB(Int_t run){ |
e3fc062d | 339 | // |
8c1c76e9 | 340 | // The name of the function says it all |
e3fc062d | 341 | // |
8c1c76e9 | 342 | AliHFEOADBThresholdsTRD *thresholds = dynamic_cast<AliHFEOADBThresholdsTRD *>(fOADBThresholds->GetObject(run)); |
343 | if(!thresholds){ | |
344 | AliDebug(1, Form("Thresholds for run %d not in the OADB", run)); | |
345 | return kFALSE; | |
346 | } | |
347 | return thresholds->GetThresholdParameters(fNTracklets, fElectronEfficiency, fThreshParams); | |
e3fc062d | 348 | } |
349 | ||
c2690925 | 350 | //___________________________________________________________________ |
351 | void AliHFEpidTRD::RenormalizeElPi(const Double_t * const likein, Double_t * const likeout) const { | |
352 | // | |
353 | // Renormalize likelihoods for electrons and pions neglecting the | |
354 | // likelihoods for protons, kaons and muons | |
355 | // | |
356 | memset(likeout, 0, sizeof(Double_t) * AliPID::kSPECIES); | |
357 | Double_t norm = likein[AliPID::kElectron] + likein[AliPID::kPion]; | |
358 | if(norm == 0.) norm = 1.; // Safety | |
359 | likeout[AliPID::kElectron] = likein[AliPID::kElectron] / norm; | |
360 | likeout[AliPID::kPion] = likein[AliPID::kPion] / norm; | |
361 | } | |
362 | ||
809a4336 | 363 | //___________________________________________________________________ |
8c1c76e9 | 364 | Double_t AliHFEpidTRD::GetElectronLikelihood(const AliVTrack *track, AliHFEpidObject::AnalysisType_t anaType) const { |
3a72645a | 365 | // |
366 | // Get TRD likelihoods for ESD respectively AOD tracks | |
367 | // | |
368 | Double_t pidProbs[AliPID::kSPECIES]; memset(pidProbs, 0, sizeof(Double_t) * AliPID::kSPECIES); | |
369 | if(anaType == AliHFEpidObject::kESDanalysis){ | |
370 | const AliESDtrack *esdtrack = dynamic_cast<const AliESDtrack *>(track); | |
bf892a6a | 371 | if(esdtrack) esdtrack->GetTRDpid(pidProbs); |
3a72645a | 372 | } else { |
8c1c76e9 | 373 | fkPIDResponse->ComputeTRDProbability(track, AliPID::kSPECIES, pidProbs); |
3a72645a | 374 | } |
c2690925 | 375 | if(!IsRenormalizeElPi()) return pidProbs[AliPID::kElectron]; |
376 | Double_t probsNew[AliPID::kSPECIES]; | |
377 | RenormalizeElPi(pidProbs, probsNew); | |
378 | return probsNew[AliPID::kElectron]; | |
3a72645a | 379 | } |
380 | ||
381 | //___________________________________________________________________ | |
6555e2ad | 382 | Double_t AliHFEpidTRD::GetP(const AliVParticle *track, AliHFEpidObject::AnalysisType_t anaType) const { |
3a72645a | 383 | // |
384 | // Get the Momentum in the TRD | |
385 | // | |
386 | Double_t p = 0.; | |
387 | if(anaType == AliHFEpidObject::kESDanalysis){ | |
388 | const AliESDtrack *esdtrack = dynamic_cast<const AliESDtrack *>(track); | |
bf892a6a | 389 | if(esdtrack) p = esdtrack->GetOuterParam() ? esdtrack->GetOuterParam()->P() : esdtrack->P(); |
3a72645a | 390 | } else { |
391 | const AliAODTrack *aodtrack = dynamic_cast<const AliAODTrack *>(track); | |
bf892a6a | 392 | if(aodtrack) p = aodtrack->P(); |
3a72645a | 393 | } |
394 | return p; | |
395 | } | |
396 | ||
397 | //___________________________________________________________________ | |
6555e2ad | 398 | Double_t AliHFEpidTRD::GetChargeLayer(const AliVParticle *track, UInt_t layer, AliHFEpidObject::AnalysisType_t anaType) const { |
399 | // | |
400 | // Get the Charge in a single TRD layer | |
401 | // | |
402 | if(layer >= 6) return 0.; | |
403 | Double_t charge = 0.; | |
404 | if(anaType == AliHFEpidObject::kESDanalysis){ | |
405 | const AliESDtrack *esdtrack = dynamic_cast<const AliESDtrack *>(track); | |
e156c3bb | 406 | if(esdtrack){ |
407 | // Distinction between old and new reconstruction: in the new reconstruction, the total charge is stored in slice 0, slices 1 to 8 are used for the slices for | |
408 | // the neural network. | |
409 | if(fTotalChargeInSlice0) | |
410 | charge = esdtrack->GetTRDslice(static_cast<UInt_t>(layer), 0); | |
411 | else | |
412 | for(Int_t islice = 0; islice < esdtrack->GetNumberOfTRDslices(); islice++) charge += esdtrack->GetTRDslice(static_cast<UInt_t>(layer), islice); | |
413 | } | |
6555e2ad | 414 | } else { |
415 | const AliAODTrack *aodtrack = dynamic_cast<const AliAODTrack *>(track); | |
bf892a6a | 416 | AliAODPid *aoddetpid = aodtrack ? aodtrack->GetDetPid() : NULL; |
e156c3bb | 417 | if(aoddetpid){ |
418 | if(fTotalChargeInSlice0) | |
6736efd5 | 419 | charge = aoddetpid->GetTRDslices()[layer * aoddetpid->GetTRDnSlices()]; |
e156c3bb | 420 | else |
6736efd5 | 421 | for(Int_t islice = 0; islice < aoddetpid->GetTRDnSlices(); islice++) charge += aoddetpid->GetTRDslices()[layer * aoddetpid->GetTRDnSlices() + islice]; |
e156c3bb | 422 | } |
6555e2ad | 423 | } |
424 | return charge; | |
425 | } | |
426 | ||
427 | //___________________________________________________________________ | |
8c1c76e9 | 428 | void AliHFEpidTRD::GetTRDmomenta(const AliVTrack *track, Double_t *mom) const { |
bf892a6a | 429 | // |
430 | // Fill Array with momentum information at the TRD tracklet | |
431 | // | |
8c1c76e9 | 432 | for(Int_t itl = 0; itl < 6; itl++) |
433 | mom[itl] = track->GetTRDmomentum(itl); | |
bf892a6a | 434 | } |
435 | ||
436 | //___________________________________________________________________ | |
437 | Double_t AliHFEpidTRD::GetTRDSignalV1(const AliESDtrack *track, Float_t truncation) const { | |
75d81601 | 438 | // |
439 | // Calculation of the TRD Signal via truncated mean | |
440 | // Method 1: Take all Slices available | |
441 | // cut out 0s | |
442 | // Order them in increasing order | |
443 | // Cut out the upper third | |
444 | // Calculate mean over the last 2/3 slices | |
445 | // | |
446 | const Int_t kNSlices = 48; | |
11ff28c5 | 447 | const Int_t kLastSlice = 6; // Slice 7 is taken out from the truncated mean calculation |
3ccf8f4c | 448 | const Double_t kVerySmall = 1e-12; |
bf892a6a | 449 | // Weight the slice to equalize the MPV of the dQ/dl-distribution per slice to the one in the first slice |
450 | // Pions are used as reference for the equalization | |
451 | const Double_t kWeightSlice[8] = {1., 2.122, 1.8, 1.635, 1.595, 1.614, 1.16, 7.0}; | |
11ff28c5 | 452 | const Double_t kWeightSliceNo0[8] = {1., 1., 1.271, 1.451, 1.531, 1.543, 1.553, 2.163}; // Weighting factors in case slice 0 stores the total charge |
453 | const Double_t *kWeightFactor = fTotalChargeInSlice0 ? kWeightSliceNo0 : kWeightSlice; | |
bf892a6a | 454 | AliDebug(3, Form("Number of Tracklets: %d\n", track->GetTRDntrackletsPID())); |
75d81601 | 455 | Double_t trdSlices[kNSlices], tmp[kNSlices]; |
456 | Int_t indices[48]; | |
457 | Int_t icnt = 0; | |
458 | for(Int_t idet = 0; idet < 6; idet++) | |
11ff28c5 | 459 | for(Int_t islice = fTotalChargeInSlice0 ? 1 : 0 ; islice <= kLastSlice; islice++){ |
75d81601 | 460 | AliDebug(2, Form("Chamber[%d], Slice[%d]: TRDSlice = %f", idet, islice, track->GetTRDslice(idet, islice))); |
3ccf8f4c | 461 | if(TMath::Abs(track->GetTRDslice(idet, islice)) < kVerySmall) continue;; |
11ff28c5 | 462 | trdSlices[icnt++] = track->GetTRDslice(idet, islice) * kWeightFactor[islice]; |
75d81601 | 463 | } |
464 | AliDebug(1, Form("Number of Slices: %d\n", icnt)); | |
465 | if(icnt < 6) return 0.; // We need at least 6 Slices for the truncated mean | |
466 | TMath::Sort(icnt, trdSlices, indices, kFALSE); | |
467 | memcpy(tmp, trdSlices, sizeof(Double_t) * icnt); | |
468 | for(Int_t ien = 0; ien < icnt; ien++) | |
469 | trdSlices[ien] = tmp[indices[ien]]; | |
bf892a6a | 470 | Double_t trdSignal = TMath::Mean(static_cast<Int_t>(static_cast<Float_t>(icnt) * truncation), trdSlices); |
75d81601 | 471 | Double_t mom = track->GetOuterParam() ? track->GetOuterParam()->P() : -1; |
e3fc062d | 472 | AliDebug(3, Form("PID Meth. 1: p[%f], TRDSignal[%f]", mom, trdSignal)); |
75d81601 | 473 | return trdSignal; |
474 | } | |
475 | ||
476 | //___________________________________________________________________ | |
bf892a6a | 477 | Double_t AliHFEpidTRD::GetTRDSignalV2(const AliESDtrack *track, Float_t truncation) const { |
75d81601 | 478 | // |
479 | // Calculation of the TRD Signal via truncated mean | |
480 | // Method 2: Take only first 5 slices per chamber | |
481 | // Order them in increasing order | |
482 | // Cut out upper half | |
483 | // Now do mean with the reamining 3 slices per chamber | |
484 | // | |
bf892a6a | 485 | const Double_t kWeightSlice[8] = {1., 2.122, 1.8, 1.635, 1.595, 1.614, 1.16, 7.0}; |
486 | const Int_t kLayers = 6; | |
487 | const Int_t kSlicesLow = 6; | |
488 | const Int_t kSlicesHigh = 1; | |
3ccf8f4c | 489 | const Double_t kVerySmall = 1e-12; |
bf892a6a | 490 | Double_t trdSlicesLowTime[kLayers*kSlicesLow], trdSlicesRemaining[kLayers*(kSlicesHigh + kSlicesLow)]; |
491 | Int_t indices[kLayers*kSlicesLow]; | |
75d81601 | 492 | Int_t cntLowTime=0, cntRemaining = 0; |
493 | for(Int_t idet = 0; idet < 6; idet++) | |
e156c3bb | 494 | for(Int_t islice = fTotalChargeInSlice0 ? 1 : 0; islice < kSlicesLow+kSlicesHigh; islice++){ |
3ccf8f4c | 495 | if(TMath::Abs(track->GetTRDslice(idet, islice)) < kVerySmall) continue;; |
bf892a6a | 496 | if(islice < kSlicesLow){ |
e3fc062d | 497 | AliDebug(3, Form("Part 1, Det[%d], Slice[%d], TRDSlice: %f", idet, islice, track->GetTRDslice(idet, islice))); |
bf892a6a | 498 | trdSlicesLowTime[cntLowTime++] = track->GetTRDslice(idet, islice) * kWeightSlice[islice]; |
75d81601 | 499 | } else{ |
e3fc062d | 500 | AliDebug(3, Form("Part 1, Det[%d], Slice[%d], TRDSlice: %f", idet, islice, track->GetTRDslice(idet, islice))); |
bf892a6a | 501 | trdSlicesRemaining[cntRemaining++] = track->GetTRDslice(idet, islice) * kWeightSlice[islice]; |
75d81601 | 502 | } |
503 | } | |
504 | if(cntLowTime < 4 || cntRemaining < 2) return 0.; // Min. Number of Slices at high time is 2 (matches with 1 layer), for the truncated mean we need at least 4 Slices | |
505 | TMath::Sort(cntLowTime, trdSlicesLowTime, indices, kFALSE); | |
506 | // Fill the second array with the lower half of the first time bins | |
bf892a6a | 507 | for(Int_t ien = 0; ien < static_cast<Int_t>(static_cast<Float_t>(cntLowTime) * truncation); ien++) |
75d81601 | 508 | trdSlicesRemaining[cntRemaining++] = trdSlicesLowTime[indices[ien]]; |
509 | Double_t trdSignal = TMath::Mean(cntRemaining, trdSlicesRemaining); | |
510 | Double_t mom = track->GetOuterParam() ? track->GetOuterParam()->P() : -1; | |
e3fc062d | 511 | AliDebug(3, Form("PID Meth. 2: p[%f], TRDSignal[%f]", mom, trdSignal)); |
75d81601 | 512 | return trdSignal; |
513 | } |