]>
Commit | Line | Data |
---|---|---|
d9950a5a | 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 | /* $Id$ */ | |
17 | ||
203967fc | 18 | #include "AliLog.h" |
eb38ed55 | 19 | #include "AliESDtrack.h" |
41880fac | 20 | #include "AliTracker.h" |
eb38ed55 | 21 | |
d9950a5a | 22 | #include "AliTRDtrackV1.h" |
23 | #include "AliTRDcluster.h" | |
24 | #include "AliTRDcalibDB.h" | |
3afdab72 | 25 | #include "AliTRDReconstructor.h" |
d9950a5a | 26 | #include "AliTRDrecoParam.h" |
27 | ||
d9950a5a | 28 | ClassImp(AliTRDtrackV1) |
29 | ||
0906e73e | 30 | /////////////////////////////////////////////////////////////////////////////// |
31 | // // | |
32 | // Represents a reconstructed TRD track // | |
33 | // Local TRD Kalman track // | |
34 | // // | |
35 | // Authors: // | |
36 | // Alex Bercuci <A.Bercuci@gsi.de> // | |
37 | // Markus Fasel <M.Fasel@gsi.de> // | |
38 | // // | |
39 | /////////////////////////////////////////////////////////////////////////////// | |
d9950a5a | 40 | |
41 | //_______________________________________________________________ | |
3b57a3f7 | 42 | AliTRDtrackV1::AliTRDtrackV1() : AliKalmanTrack() |
eb2b4f91 | 43 | ,fStatus(0) |
352cef8f | 44 | ,fESDid(0) |
3b57a3f7 | 45 | ,fDE(0.) |
4d6aee34 | 46 | ,fkReconstructor(NULL) |
47 | ,fBackupTrack(NULL) | |
48 | ,fTrackLow(NULL) | |
49 | ,fTrackHigh(NULL) | |
0906e73e | 50 | { |
d9950a5a | 51 | // |
52 | // Default constructor | |
53 | // | |
6e49cfdb | 54 | //printf("AliTRDtrackV1::AliTRDtrackV1()\n"); |
e44586fb | 55 | |
3b57a3f7 | 56 | for(int i =0; i<3; i++) fBudget[i] = 0.; |
57 | ||
58 | Float_t pid = 1./AliPID::kSPECIES; | |
59 | for(int is =0; is<AliPID::kSPECIES; is++) fPID[is] = pid; | |
60 | ||
61 | for(int ip=0; ip<kNplane; ip++){ | |
17896e82 | 62 | fTrackletIndex[ip] = -1; |
4d6aee34 | 63 | fTracklet[ip] = NULL; |
3b57a3f7 | 64 | } |
d9950a5a | 65 | } |
66 | ||
67 | //_______________________________________________________________ | |
3b57a3f7 | 68 | AliTRDtrackV1::AliTRDtrackV1(const AliTRDtrackV1 &ref) : AliKalmanTrack(ref) |
eb2b4f91 | 69 | ,fStatus(ref.fStatus) |
352cef8f | 70 | ,fESDid(ref.fESDid) |
3b57a3f7 | 71 | ,fDE(ref.fDE) |
4d6aee34 | 72 | ,fkReconstructor(ref.fkReconstructor) |
73 | ,fBackupTrack(NULL) | |
74 | ,fTrackLow(NULL) | |
75 | ,fTrackHigh(NULL) | |
d9950a5a | 76 | { |
77 | // | |
3b57a3f7 | 78 | // Copy constructor |
d9950a5a | 79 | // |
80 | ||
6e49cfdb | 81 | //printf("AliTRDtrackV1::AliTRDtrackV1(const AliTRDtrackV1 &)\n"); |
29b87567 | 82 | SetBit(kOwner, kFALSE); |
3b57a3f7 | 83 | for(int ip=0; ip<kNplane; ip++){ |
84 | fTrackletIndex[ip] = ref.fTrackletIndex[ip]; | |
85 | fTracklet[ip] = ref.fTracklet[ip]; | |
86 | } | |
e3cf3d02 | 87 | if(ref.fTrackLow) fTrackLow = new AliExternalTrackParam(*ref.fTrackLow); |
88 | if(ref.fTrackHigh) fTrackHigh = new AliExternalTrackParam(*ref.fTrackHigh); | |
89 | ||
3b57a3f7 | 90 | for (Int_t i = 0; i < 3;i++) fBudget[i] = ref.fBudget[i]; |
91 | ||
92 | for(Int_t is = 0; is<AliPID::kSPECIES; is++) fPID[is] = ref.fPID[is]; | |
93 | ||
94 | AliKalmanTrack::SetNumberOfClusters(ref.GetNumberOfClusters()); | |
d9950a5a | 95 | } |
96 | ||
97 | //_______________________________________________________________ | |
3b57a3f7 | 98 | AliTRDtrackV1::AliTRDtrackV1(const AliESDtrack &t) : AliKalmanTrack() |
eb2b4f91 | 99 | ,fStatus(0) |
352cef8f | 100 | ,fESDid(0) |
3b57a3f7 | 101 | ,fDE(0.) |
4d6aee34 | 102 | ,fkReconstructor(NULL) |
103 | ,fBackupTrack(NULL) | |
104 | ,fTrackLow(NULL) | |
105 | ,fTrackHigh(NULL) | |
d9950a5a | 106 | { |
107 | // | |
3b57a3f7 | 108 | // Constructor from AliESDtrack |
d9950a5a | 109 | // |
110 | ||
352cef8f | 111 | SetESDid(t.GetID()); |
3b57a3f7 | 112 | SetLabel(t.GetLabel()); |
113 | SetChi2(0.0); | |
0b433f72 | 114 | |
115 | SetMass(t.GetMass()/*0.000510*/); | |
3b57a3f7 | 116 | AliKalmanTrack::SetNumberOfClusters(t.GetTRDncls()); |
17896e82 | 117 | Int_t ti[]={-1, -1, -1, -1, -1, -1}; t.GetTRDtracklets(&ti[0]); |
3b57a3f7 | 118 | for(int ip=0; ip<kNplane; ip++){ |
17896e82 | 119 | fTrackletIndex[ip] = ti[ip]; |
4d6aee34 | 120 | fTracklet[ip] = NULL; |
3b57a3f7 | 121 | } |
122 | for(int i =0; i<3; i++) fBudget[i] = 0.; | |
123 | ||
124 | Float_t pid = 1./AliPID::kSPECIES; | |
125 | for(int is =0; is<AliPID::kSPECIES; is++) fPID[is] = pid; | |
126 | ||
127 | const AliExternalTrackParam *par = &t; | |
128 | if (t.GetStatus() & AliESDtrack::kTRDbackup) { | |
129 | par = t.GetOuterParam(); | |
130 | if (!par) { | |
131 | AliError("No backup info!"); | |
132 | par = &t; | |
133 | } | |
134 | } | |
135 | Set(par->GetX() | |
136 | ,par->GetAlpha() | |
137 | ,par->GetParameter() | |
138 | ,par->GetCovariance()); | |
139 | ||
140 | if(t.GetStatus() & AliESDtrack::kTIME) { | |
141 | StartTimeIntegral(); | |
142 | Double_t times[10]; | |
143 | t.GetIntegratedTimes(times); | |
144 | SetIntegratedTimes(times); | |
145 | SetIntegratedLength(t.GetIntegratedLength()); | |
146 | } | |
d9950a5a | 147 | } |
148 | ||
d9950a5a | 149 | //_______________________________________________________________ |
e17f4785 | 150 | AliTRDtrackV1::AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], const Double_t cov[15] |
3b57a3f7 | 151 | , Double_t x, Double_t alpha) : AliKalmanTrack() |
eb2b4f91 | 152 | ,fStatus(0) |
352cef8f | 153 | ,fESDid(0) |
3b57a3f7 | 154 | ,fDE(0.) |
4d6aee34 | 155 | ,fkReconstructor(NULL) |
156 | ,fBackupTrack(NULL) | |
157 | ,fTrackLow(NULL) | |
158 | ,fTrackHigh(NULL) | |
d9950a5a | 159 | { |
0906e73e | 160 | // |
161 | // The stand alone tracking constructor | |
162 | // TEMPORARY !!!!!!!!!!! | |
163 | // to check : | |
164 | // 1. covariance matrix | |
165 | // 2. dQdl calculation | |
166 | // | |
d9950a5a | 167 | |
68f9b6bd | 168 | Double_t b(GetBz()); |
169 | Double_t cnv = (TMath::Abs(b) < 1.e-5) ? 1.e5 : 1./GetBz()/kB2C; | |
170 | ||
d9950a5a | 171 | Double_t pp[5] = { p[0] |
68f9b6bd | 172 | , p[1] |
173 | , p[2] | |
174 | , p[3] | |
175 | , p[4]*cnv }; | |
176 | ||
d9950a5a | 177 | Double_t c22 = x*x*cov[14] - 2*x*cov[12] + cov[ 5]; |
178 | Double_t c32 = x*cov[13] - cov[ 8]; | |
179 | Double_t c20 = x*cov[10] - cov[ 3]; | |
180 | Double_t c21 = x*cov[11] - cov[ 4]; | |
181 | Double_t c42 = x*cov[14] - cov[12]; | |
68f9b6bd | 182 | |
d9950a5a | 183 | Double_t cc[15] = { cov[ 0] |
184 | , cov[ 1], cov[ 2] | |
185 | , c20, c21, c22 | |
186 | , cov[ 6], cov[ 7], c32, cov[ 9] | |
187 | , cov[10]*cnv, cov[11]*cnv, c42*cnv, cov[13]*cnv, cov[14]*cnv*cnv }; | |
68f9b6bd | 188 | |
ae3fbe1f | 189 | Double_t mostProbablePt=AliExternalTrackParam::GetMostProbablePt(); |
190 | Double_t p0=TMath::Sign(1/mostProbablePt,pp[4]); | |
191 | Double_t w0=cc[14]/(cc[14] + p0*p0), w1=p0*p0/(cc[14] + p0*p0); | |
68f9b6bd | 192 | AliDebug(3, Form("Pt mixing : w0[%4.2f] pt0[%5.3f] w1[%4.2f] pt[%5.3f]", w0, 1./p0, w1, 1./pp[4])); |
193 | ||
ae3fbe1f | 194 | pp[4] = w0*p0 + w1*pp[4]; |
195 | cc[10]*=w1; cc[11]*=w1; cc[12]*=w1; cc[13]*=w1; cc[14]*=w1; | |
68f9b6bd | 196 | Set(x,alpha,pp,cc); |
0217fcd0 | 197 | AliDebug(2, Form("Init @ x[%6.2f] pt[%5.3f]", x, 1./pp[4])); |
41702fec | 198 | Int_t ncls = 0; |
68f9b6bd | 199 | for(int iplane=0; iplane<kNplane; iplane++){ |
17896e82 | 200 | fTrackletIndex[iplane] = -1; |
68f9b6bd | 201 | if(!trklts[iplane].IsOK()) fTracklet[iplane] = NULL; |
41702fec | 202 | else{ |
e17f4785 | 203 | fTracklet[iplane] = &trklts[iplane]; |
41702fec | 204 | ncls += fTracklet[iplane]->GetN(); |
205 | } | |
68f9b6bd | 206 | } |
41702fec | 207 | AliKalmanTrack::SetNumberOfClusters(ncls); |
208 | for(int i =0; i<3; i++) fBudget[i] = 0.; | |
209 | ||
210 | Float_t pid = 1./AliPID::kSPECIES; | |
211 | for(int is =0; is<AliPID::kSPECIES; is++) fPID[is] = pid; | |
d9950a5a | 212 | } |
213 | ||
bb56afff | 214 | //_______________________________________________________________ |
3b57a3f7 | 215 | AliTRDtrackV1::~AliTRDtrackV1() |
bb56afff | 216 | { |
e44586fb | 217 | //AliInfo(""); |
6e49cfdb | 218 | //printf("I-AliTRDtrackV1::~AliTRDtrackV1() : Owner[%s]\n", TestBit(kOwner)?"YES":"NO"); |
76b60503 | 219 | |
4d6aee34 | 220 | if(fBackupTrack) delete fBackupTrack; fBackupTrack = NULL; |
e3cf3d02 | 221 | |
4d6aee34 | 222 | if(fTrackLow) delete fTrackLow; fTrackLow = NULL; |
223 | if(fTrackHigh) delete fTrackHigh; fTrackHigh = NULL; | |
bb56afff | 224 | |
76b60503 | 225 | for(Int_t ip=0; ip<kNplane; ip++){ |
226 | if(TestBit(kOwner) && fTracklet[ip]) delete fTracklet[ip]; | |
4d6aee34 | 227 | fTracklet[ip] = NULL; |
17896e82 | 228 | fTrackletIndex[ip] = -1; |
3b57a3f7 | 229 | } |
230 | } | |
231 | ||
e08a5492 | 232 | //_______________________________________________________________ |
233 | AliTRDtrackV1 &AliTRDtrackV1::operator=(const AliTRDtrackV1 &t) | |
234 | { | |
235 | // | |
236 | // Assignment operator | |
237 | // | |
238 | ||
239 | if (this != &t) { | |
240 | ((AliTRDtrackV1 &) t).Copy(*this); | |
241 | } | |
242 | ||
243 | return *this; | |
244 | ||
245 | } | |
246 | ||
247 | //_____________________________________________________________________________ | |
248 | void AliTRDtrackV1::Copy(TObject &t) const | |
249 | { | |
250 | // | |
251 | // Copy function | |
252 | // | |
253 | ||
254 | ((AliTRDtrackV1 &) t).fStatus = fStatus; | |
255 | ((AliTRDtrackV1 &) t).fESDid = fESDid; | |
256 | ((AliTRDtrackV1 &) t).fDE = fDE; | |
257 | ((AliTRDtrackV1 &) t).fkReconstructor = fkReconstructor; | |
258 | ((AliTRDtrackV1 &) t).fBackupTrack = 0x0; | |
259 | ((AliTRDtrackV1 &) t).fTrackLow = 0x0; | |
260 | ((AliTRDtrackV1 &) t).fTrackHigh = 0x0; | |
261 | ||
262 | for(Int_t ip = 0; ip < kNplane; ip++) { | |
263 | ((AliTRDtrackV1 &) t).fTrackletIndex[ip] = fTrackletIndex[ip]; | |
264 | ((AliTRDtrackV1 &) t).fTracklet[ip] = fTracklet[ip]; | |
265 | } | |
266 | if (fTrackLow) { | |
267 | ((AliTRDtrackV1 &) t).fTrackLow = new AliExternalTrackParam(*fTrackLow); | |
268 | } | |
269 | if (fTrackHigh){ | |
270 | ((AliTRDtrackV1 &) t).fTrackHigh = new AliExternalTrackParam(*fTrackHigh); | |
271 | } | |
272 | ||
273 | for (Int_t i = 0; i < 3; i++) { | |
274 | ((AliTRDtrackV1 &) t).fBudget[i] = fBudget[i]; | |
275 | } | |
276 | for (Int_t is = 0; is < AliPID::kSPECIES; is++) { | |
277 | ((AliTRDtrackV1 &) t).fPID[is] = fPID[is]; | |
278 | } | |
279 | ||
280 | } | |
281 | ||
3b57a3f7 | 282 | //_______________________________________________________________ |
283 | Bool_t AliTRDtrackV1::CookLabel(Float_t wrong) | |
284 | { | |
285 | // set MC label for this track | |
bb2db46c | 286 | if(!GetNumberOfClusters()) return kFALSE; |
287 | ||
bb56afff | 288 | Int_t s[kMAXCLUSTERSPERTRACK][2]; |
289 | for (Int_t i = 0; i < kMAXCLUSTERSPERTRACK; i++) { | |
290 | s[i][0] = -1; | |
291 | s[i][1] = 0; | |
292 | } | |
3b57a3f7 | 293 | |
bb56afff | 294 | Bool_t labelAdded; |
3b57a3f7 | 295 | Int_t label; |
4d6aee34 | 296 | AliTRDcluster *c = NULL; |
3b57a3f7 | 297 | for (Int_t ip = 0; ip < kNplane; ip++) { |
68f9b6bd | 298 | if(fTrackletIndex[ip]<0 || !fTracklet[ip]) continue; |
8d2bec9e | 299 | for (Int_t ic = 0; ic < AliTRDseedV1::kNclusters; ic++) { |
3b57a3f7 | 300 | if(!(c = fTracklet[ip]->GetClusters(ic))) continue; |
301 | for (Int_t k = 0; k < 3; k++) { | |
302 | label = c->GetLabel(k); | |
303 | labelAdded = kFALSE; | |
304 | Int_t j = 0; | |
305 | if (label >= 0) { | |
306 | while ((!labelAdded) && (j < kMAXCLUSTERSPERTRACK)) { | |
307 | if ((s[j][0] == label) || | |
308 | (s[j][1] == 0)) { | |
309 | s[j][0] = label; | |
310 | s[j][1]++; | |
311 | labelAdded = kTRUE; | |
312 | } | |
313 | j++; | |
314 | } | |
315 | } | |
316 | } | |
317 | } | |
318 | } | |
319 | ||
bb56afff | 320 | Int_t max = 0; |
321 | label = -123456789; | |
322 | for (Int_t i = 0; i < kMAXCLUSTERSPERTRACK; i++) { | |
323 | if (s[i][1] <= max) continue; | |
3b57a3f7 | 324 | max = s[i][1]; |
325 | label = s[i][0]; | |
bb56afff | 326 | } |
bb56afff | 327 | if ((1. - Float_t(max)/GetNumberOfClusters()) > wrong) label = -label; |
3b57a3f7 | 328 | |
bb56afff | 329 | SetLabel(label); |
3b57a3f7 | 330 | |
331 | return kTRUE; | |
bb56afff | 332 | } |
333 | ||
d9950a5a | 334 | //_______________________________________________________________ |
335 | Bool_t AliTRDtrackV1::CookPID() | |
336 | { | |
2a3191bb | 337 | // |
338 | // Cook the PID information for the track by delegating the omonim function of the tracklets. | |
339 | // Computes the number of tracklets used. The tracklet information are considered independent. | |
340 | // For the moment no global track measurement of PID is performed as for example to estimate | |
341 | // bremsstrahlung probability based on global chi2 of the track. | |
342 | // | |
343 | // The status bit AliESDtrack::kTRDpid is set during the call of AliTRDtrackV1::UpdateESDtrack().The PID performance of the | |
344 | //TRD for tracks with 6 tacklets is displayed below. | |
345 | //Begin_Html | |
346 | //<img src="TRD/trackPID.gif"> | |
347 | //End_Html | |
348 | // | |
3b57a3f7 | 349 | |
2a3191bb | 350 | /*Reset the a priori probabilities*/ |
3b57a3f7 | 351 | Double_t pid = 1. / AliPID::kSPECIES; |
eb2b4f91 | 352 | for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) fPID[ispec] = pid; |
353 | ||
354 | UChar_t fPIDquality = SetNumberOfTrackletsPID(kTRUE); | |
355 | // no tracklet found for PID calculations | |
356 | if(!fPIDquality) return kFALSE; | |
357 | ||
2a3191bb | 358 | // slot for PID calculation @ track level for bremsstrahlung TODO |
eb2b4f91 | 359 | |
360 | // normalize probabilities | |
361 | Double_t probTotal = 0.0; | |
362 | for (Int_t is = 0; is < AliPID::kSPECIES; is++) probTotal += fPID[is]; | |
363 | ||
364 | ||
365 | if (probTotal <= 0.0) { | |
366 | AliWarning("The total probability over all species <= 0. This may be caused by some error in the reference data."); | |
367 | return kFALSE; | |
3b57a3f7 | 368 | } |
3b57a3f7 | 369 | |
eb2b4f91 | 370 | for (Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++) fPID[iSpecies] /= probTotal; |
371 | ||
372 | return kTRUE; | |
373 | } | |
374 | ||
375 | //___________________________________________________________ | |
376 | UChar_t AliTRDtrackV1::GetNumberOfTrackletsPID() const | |
377 | { | |
378 | // Retrieve number of tracklets used for PID calculation. | |
379 | ||
e20bef2b | 380 | UChar_t nPID = 0; |
4d6aee34 | 381 | Float_t *prob = NULL; |
3b57a3f7 | 382 | for(int ip=0; ip<kNplane; ip++){ |
68f9b6bd | 383 | if(fTrackletIndex[ip]<0 || !fTracklet[ip]) continue; |
3b57a3f7 | 384 | if(!fTracklet[ip]->IsOK()) continue; |
e20bef2b | 385 | if(!(prob = fTracklet[ip]->GetProbability(kFALSE))) continue; |
68f9b6bd | 386 | |
3b57a3f7 | 387 | Int_t nspec = 0; // quality check of tracklet dEdx |
388 | for(int ispec=0; ispec<AliPID::kSPECIES; ispec++){ | |
389 | if(prob[ispec] < 0.) continue; | |
3b57a3f7 | 390 | nspec++; |
391 | } | |
392 | if(!nspec) continue; | |
393 | ||
e20bef2b | 394 | fTracklet[ip]->SetPID(); |
395 | nPID++; | |
0906e73e | 396 | } |
e20bef2b | 397 | return nPID; |
eb2b4f91 | 398 | } |
399 | ||
400 | //___________________________________________________________ | |
401 | UChar_t AliTRDtrackV1::SetNumberOfTrackletsPID(Bool_t recalc) | |
402 | { | |
403 | // Retrieve number of tracklets used for PID calculation. // Recalculated PID at tracklet level by quering the PID DB. | |
404 | ||
68f9b6bd | 405 | UChar_t fPIDquality(0); |
0906e73e | 406 | |
eb2b4f91 | 407 | // steer PID calculation @ tracklet level |
68f9b6bd | 408 | Float_t *prob(NULL); |
eb2b4f91 | 409 | for(int ip=0; ip<kNplane; ip++){ |
68f9b6bd | 410 | if(fTrackletIndex[ip]<0 || !fTracklet[ip]) continue; |
eb2b4f91 | 411 | if(!fTracklet[ip]->IsOK()) continue; |
412 | if(!(prob = fTracklet[ip]->GetProbability(recalc))) return 0; | |
68f9b6bd | 413 | |
eb2b4f91 | 414 | Int_t nspec = 0; // quality check of tracklet dEdx |
415 | for(int ispec=0; ispec<AliPID::kSPECIES; ispec++){ | |
416 | if(prob[ispec] < 0.) continue; | |
417 | fPID[ispec] *= prob[ispec]; | |
418 | nspec++; | |
419 | } | |
420 | if(!nspec) continue; | |
421 | ||
422 | fPIDquality++; | |
d9950a5a | 423 | } |
eb2b4f91 | 424 | return fPIDquality; |
d9950a5a | 425 | } |
426 | ||
0349cc67 | 427 | //_______________________________________________________________ |
428 | AliTRDcluster* AliTRDtrackV1::GetCluster(Int_t id) | |
429 | { | |
4d6aee34 | 430 | // Get the cluster at a certain position in the track |
0349cc67 | 431 | Int_t n = 0; |
432 | for(Int_t ip=0; ip<kNplane; ip++){ | |
433 | if(!fTracklet[ip]) continue; | |
434 | if(n+fTracklet[ip]->GetN() <= id){ | |
435 | n+=fTracklet[ip]->GetN(); | |
436 | continue; | |
437 | } | |
4d6aee34 | 438 | AliTRDcluster *c = NULL; |
8d2bec9e | 439 | for(Int_t ic=AliTRDseedV1::kNclusters; ic--;){ |
0349cc67 | 440 | if(!(c = fTracklet[ip]->GetClusters(ic))) continue; |
441 | ||
442 | if(n<id){n++; continue;} | |
443 | return c; | |
444 | } | |
445 | } | |
4d6aee34 | 446 | return NULL; |
0349cc67 | 447 | } |
448 | ||
3b57a3f7 | 449 | //_______________________________________________________________ |
450 | Int_t AliTRDtrackV1::GetClusterIndex(Int_t id) const | |
451 | { | |
4d6aee34 | 452 | // Get the cluster index at a certain position in the track |
3b57a3f7 | 453 | Int_t n = 0; |
454 | for(Int_t ip=0; ip<kNplane; ip++){ | |
455 | if(!fTracklet[ip]) continue; | |
76b60503 | 456 | if(n+fTracklet[ip]->GetN() <= id){ |
3b57a3f7 | 457 | n+=fTracklet[ip]->GetN(); |
458 | continue; | |
459 | } | |
4d6aee34 | 460 | AliTRDcluster *c = NULL; |
8d2bec9e | 461 | for(Int_t ic=AliTRDseedV1::kNclusters; ic--;){ |
76b60503 | 462 | if(!(c = fTracklet[ip]->GetClusters(ic))) continue; |
463 | ||
464 | if(n<id){n++; continue;} | |
3b57a3f7 | 465 | return fTracklet[ip]->GetIndexes(ic); |
466 | } | |
467 | } | |
468 | return -1; | |
0906e73e | 469 | } |
d9950a5a | 470 | |
471 | //_______________________________________________________________ | |
b72f4eaf | 472 | Double_t AliTRDtrackV1::GetPredictedChi2(const AliTRDseedV1 *trklt, Double_t *cov) const |
d9950a5a | 473 | { |
ed15ef4f | 474 | // Compute chi2 between tracklet and track. The value is calculated at the radial position of the track |
475 | // equal to the reference radial position of the tracklet (see AliTRDseedV1) | |
476 | // | |
477 | // The chi2 estimator is computed according to the following formula | |
478 | // BEGIN_LATEX | |
479 | // #chi^{2}=(X_{trklt}-X_{track})(C_{trklt}+C_{track})^{-1}(X_{trklt}-X_{track})^{T} | |
480 | // END_LATEX | |
481 | // where X=(y z), the position of the track/tracklet in the yz plane | |
482 | // | |
483 | ||
b72f4eaf | 484 | Double_t p[2] = { trklt->GetY(), trklt->GetZ()}; |
485 | trklt->GetCovAt(trklt->GetX(), cov); | |
ed15ef4f | 486 | return AliExternalTrackParam::GetPredictedChi2(p, cov); |
3b57a3f7 | 487 | } |
d9950a5a | 488 | |
eb2b4f91 | 489 | //_______________________________________________________________ |
490 | Int_t AliTRDtrackV1::GetSector() const | |
491 | { | |
492 | return Int_t(GetAlpha()/AliTRDgeometry::GetAlpha() + (GetAlpha()>0. ? 0 : AliTRDgeometry::kNsector)); | |
493 | } | |
494 | ||
203967fc | 495 | //_______________________________________________________________ |
496 | Bool_t AliTRDtrackV1::IsEqual(const TObject *o) const | |
497 | { | |
4d6aee34 | 498 | // Checks whether two tracks are equal |
203967fc | 499 | if (!o) return kFALSE; |
500 | const AliTRDtrackV1 *inTrack = dynamic_cast<const AliTRDtrackV1*>(o); | |
501 | if (!inTrack) return kFALSE; | |
502 | ||
eb2b4f91 | 503 | //if ( fPIDquality != inTrack->GetPIDquality() ) return kFALSE; |
203967fc | 504 | |
505 | for(Int_t i = 0; i < AliPID::kSPECIES; i++){ | |
506 | if ( fPID[i] != inTrack->GetPID(i) ) return kFALSE; | |
507 | } | |
508 | ||
509 | for (Int_t i = 0; i < 3; i++){ | |
510 | if ( fBudget[i] != inTrack->GetBudget(i) ) return kFALSE; | |
511 | } | |
512 | if ( fDE != inTrack->GetEdep() ) return kFALSE; | |
513 | if ( fFakeRatio != inTrack->GetFakeRatio() ) return kFALSE; | |
514 | if ( fChi2 != inTrack->GetChi2() ) return kFALSE; | |
515 | if ( fMass != inTrack->GetMass() ) return kFALSE; | |
516 | if ( fLab != inTrack->GetLabel() ) return kFALSE; | |
517 | if ( fN != inTrack->GetNumberOfClusters() ) return kFALSE; | |
518 | if ( AliKalmanTrack::GetIntegratedLength() != inTrack->GetIntegratedLength() ) return kFALSE; | |
519 | ||
520 | if ( GetX() != inTrack->GetX() ) return kFALSE; | |
521 | if ( GetAlpha() != inTrack->GetAlpha() ) return kFALSE; | |
522 | const Double_t *inP = inTrack->GetParameter(); | |
523 | const Double_t *curP = GetParameter(); | |
524 | for (Int_t i = 0; i < 5; i++){ | |
525 | if ( curP[i] != inP[i]) return kFALSE; | |
526 | } | |
527 | const Double_t *inC = inTrack->GetCovariance(); | |
528 | const Double_t *curC = GetCovariance(); | |
529 | for (Int_t i = 0; i < 15; i++){ | |
530 | if ( curC[i] != inC[i]) return kFALSE; | |
531 | } | |
532 | ||
533 | for (Int_t iTracklet = 0; iTracklet < kNplane; iTracklet++){ | |
534 | AliTRDseedV1 *curTracklet = fTracklet[iTracklet]; | |
535 | AliTRDseedV1 *inTracklet = inTrack->GetTracklet(iTracklet); | |
536 | if (curTracklet && inTracklet){ | |
537 | if (! curTracklet->IsEqual(inTracklet) ) { | |
538 | curTracklet->Print(); | |
539 | inTracklet->Print(); | |
540 | return kFALSE; | |
541 | } | |
542 | } else { | |
543 | // if one tracklet exists, and corresponding | |
544 | // in other track doesn't - return kFALSE | |
545 | if(inTracklet || curTracklet) return kFALSE; | |
546 | } | |
547 | } | |
548 | ||
549 | return kTRUE; | |
550 | } | |
551 | ||
22a4ab0c | 552 | //_______________________________________________________________ |
553 | Bool_t AliTRDtrackV1::IsElectron() const | |
554 | { | |
4d6aee34 | 555 | if(GetPID(0) > fkReconstructor->GetRecoParam()->GetPIDThreshold(GetP())) return kTRUE; |
22a4ab0c | 556 | return kFALSE; |
557 | } | |
558 | ||
3b57a3f7 | 559 | |
560 | //_____________________________________________________________________________ | |
b453ef55 | 561 | Int_t AliTRDtrackV1::MakeBackupTrack() |
3b57a3f7 | 562 | { |
b453ef55 | 563 | // |
564 | // Creates a backup track | |
565 | // TO DO update quality check of the track. | |
566 | // | |
567 | ||
568 | Float_t occupancy(0.); Int_t n(0), ncls(0); | |
569 | for(Int_t il(AliTRDgeometry::kNlayer); il--;){ | |
570 | if(!fTracklet[il]) continue; | |
571 | n++; | |
572 | occupancy+=fTracklet[il]->GetOccupancyTB(); | |
573 | ncls += fTracklet[il]->GetN(); | |
574 | } | |
575 | if(!n) return -1; | |
576 | occupancy/=n; | |
577 | ||
578 | //Float_t ratio1 = Float_t(t.GetNumberOfClusters()+1) / Float_t(t.GetNExpected()+1); | |
579 | ||
580 | Int_t failedCutId(0); | |
581 | if(GetChi2()/n > 5.0) failedCutId=1; | |
582 | if(occupancy < 0.7) failedCutId=2; | |
583 | //if(ratio1 > 0.6) && | |
584 | //if(ratio0+ratio1 > 1.5) && | |
585 | if(GetNCross() != 0) failedCutId=3; | |
586 | if(TMath::Abs(GetSnp()) > 0.85) failedCutId=4; | |
587 | if(ncls < 20) failedCutId=5; | |
588 | ||
589 | if(failedCutId){ | |
590 | AliDebug(2, Form("\n" | |
591 | "chi2/tracklet < 5.0 [%c] %5.2f\n" | |
592 | "occupancy > 0.7 [%c] %4.2f\n" | |
593 | "NCross == 0 [%c] %d\n" | |
594 | "Abs(snp) < 0.85 [%c] %4.2f\n" | |
595 | "NClusters > 20 [%c] %d" | |
596 | ,(GetChi2()/n<5.0)?'y':'n', GetChi2()/n | |
597 | ,(occupancy>0.7)?'y':'n', occupancy | |
598 | ,(GetNCross()==0)?'y':'n', GetNCross() | |
599 | ,(TMath::Abs(GetSnp())<0.85)?'y':'n', TMath::Abs(GetSnp()) | |
600 | ,(ncls>20)?'y':'n', ncls | |
601 | )); | |
602 | return failedCutId; | |
603 | } | |
3b57a3f7 | 604 | |
605 | if(fBackupTrack) { | |
606 | fBackupTrack->~AliTRDtrackV1(); | |
607 | new(fBackupTrack) AliTRDtrackV1((AliTRDtrackV1&)(*this)); | |
b453ef55 | 608 | return 0; |
3b57a3f7 | 609 | } |
610 | fBackupTrack = new AliTRDtrackV1((AliTRDtrackV1&)(*this)); | |
b453ef55 | 611 | return 0; |
d9950a5a | 612 | } |
613 | ||
3b57a3f7 | 614 | //_____________________________________________________________________________ |
f8a9723f | 615 | Int_t AliTRDtrackV1::GetProlongation(Double_t xk, Double_t &y, Double_t &z) const |
3b57a3f7 | 616 | { |
617 | // | |
618 | // Find a prolongation at given x | |
530fb4e2 | 619 | // Return -1 if it does not exist |
3b57a3f7 | 620 | // |
621 | ||
622 | Double_t bz = GetBz(); | |
530fb4e2 | 623 | if (!AliExternalTrackParam::GetYAt(xk,bz,y)) return -1; |
624 | if (!AliExternalTrackParam::GetZAt(xk,bz,z)) return -1; | |
3b57a3f7 | 625 | |
626 | return 1; | |
627 | ||
628 | } | |
629 | ||
630 | //_____________________________________________________________________________ | |
631 | Bool_t AliTRDtrackV1::PropagateTo(Double_t xk, Double_t xx0, Double_t xrho) | |
0906e73e | 632 | { |
633 | // | |
3b57a3f7 | 634 | // Propagates this track to a reference plane defined by "xk" [cm] |
635 | // correcting for the mean crossed material. | |
0906e73e | 636 | // |
3b57a3f7 | 637 | // "xx0" - thickness/rad.length [units of the radiation length] |
638 | // "xrho" - thickness*density [g/cm^2] | |
639 | // | |
0906e73e | 640 | |
dc5e390a | 641 | if (xk == GetX()) return kTRUE; |
642 | ||
307aac71 | 643 | Double_t xyz0[3] = {GetX(), GetY(), GetZ()}, // track position BEFORE propagation |
dc5e390a | 644 | b[3]; // magnetic field |
307aac71 | 645 | GetBxByBz(b); |
dc5e390a | 646 | if(!AliExternalTrackParam::PropagateToBxByBz(xk,b)) return kFALSE; |
647 | ||
307aac71 | 648 | // local track position AFTER propagation |
649 | Double_t xyz1[3] = {GetX(), GetY(), GetZ()}; | |
dc5e390a | 650 | if(xyz0[0] < xk) { |
3b57a3f7 | 651 | xrho = -xrho; |
652 | if (IsStartedTimeIntegral()) { | |
dc5e390a | 653 | Double_t l2 = TMath::Sqrt((xyz1[0]-xyz0[0])*(xyz1[0]-xyz0[0]) |
654 | + (xyz1[1]-xyz0[1])*(xyz1[1]-xyz0[1]) | |
655 | + (xyz1[2]-xyz0[2])*(xyz1[2]-xyz0[2])); | |
6790f7d7 | 656 | Double_t crv = AliExternalTrackParam::GetC(b[2]); |
3b57a3f7 | 657 | if (TMath::Abs(l2*crv) > 0.0001) { |
658 | // Make correction for curvature if neccesary | |
dc5e390a | 659 | l2 = 0.5 * TMath::Sqrt((xyz1[0]-xyz0[0])*(xyz1[0]-xyz0[0]) |
660 | + (xyz1[1]-xyz0[1])*(xyz1[1]-xyz0[1])); | |
3b57a3f7 | 661 | l2 = 2.0 * TMath::ASin(l2 * crv) / crv; |
dc5e390a | 662 | l2 = TMath::Sqrt(l2*l2 + (xyz1[2]-xyz0[2])*(xyz1[2]-xyz0[2])); |
3b57a3f7 | 663 | } |
664 | AddTimeStep(l2); | |
665 | } | |
666 | } | |
667 | ||
fd40f855 | 668 | if (!AliExternalTrackParam::CorrectForMeanMaterial(xx0, xrho, GetMass())) return kFALSE; |
669 | ||
3b57a3f7 | 670 | |
671 | { | |
672 | ||
673 | // Energy losses | |
674 | Double_t p2 = (1.0 + GetTgl()*GetTgl()) / (GetSigned1Pt()*GetSigned1Pt()); | |
675 | Double_t beta2 = p2 / (p2 + GetMass()*GetMass()); | |
676 | if ((beta2 < 1.0e-10) || | |
677 | ((5940.0 * beta2/(1.0 - beta2 + 1.0e-10) - beta2) < 0.0)) { | |
678 | return kFALSE; | |
679 | } | |
680 | ||
681 | Double_t dE = 0.153e-3 / beta2 | |
682 | * (TMath::Log(5940.0 * beta2/(1.0 - beta2 + 1.0e-10)) - beta2) | |
683 | * xrho; | |
684 | fBudget[0] += xrho; | |
685 | ||
686 | /* | |
687 | // Suspicious part - think about it ? | |
688 | Double_t kinE = TMath::Sqrt(p2); | |
689 | if (dE > 0.8*kinE) dE = 0.8 * kinE; // | |
690 | if (dE < 0) dE = 0.0; // Not valid region for Bethe bloch | |
691 | */ | |
692 | ||
693 | fDE += dE; | |
694 | ||
695 | /* | |
696 | // Suspicious ! I.B. | |
697 | Double_t sigmade = 0.07 * TMath::Sqrt(TMath::Abs(dE)); // Energy loss fluctuation | |
698 | Double_t sigmac2 = sigmade*sigmade*fC*fC*(p2+GetMass()*GetMass())/(p2*p2); | |
699 | fCcc += sigmac2; | |
700 | fCee += fX*fX * sigmac2; | |
701 | */ | |
702 | ||
703 | } | |
704 | ||
705 | return kTRUE; | |
0906e73e | 706 | } |
3b57a3f7 | 707 | |
87a7fa94 | 708 | //_____________________________________________________________________________ |
3b57a3f7 | 709 | Int_t AliTRDtrackV1::PropagateToR(Double_t r,Double_t step) |
87a7fa94 | 710 | { |
711 | // | |
3b57a3f7 | 712 | // Propagate track to the radial position |
713 | // Rotation always connected to the last track position | |
87a7fa94 | 714 | // |
715 | ||
3b57a3f7 | 716 | Double_t xyz0[3]; |
717 | Double_t xyz1[3]; | |
718 | Double_t y; | |
719 | Double_t z; | |
720 | ||
721 | Double_t radius = TMath::Sqrt(GetX()*GetX() + GetY()*GetY()); | |
722 | // Direction +- | |
723 | Double_t dir = (radius > r) ? -1.0 : 1.0; | |
724 | ||
725 | for (Double_t x = radius+dir*step; dir*x < dir*r; x += dir*step) { | |
726 | ||
727 | GetXYZ(xyz0); | |
728 | Double_t alpha = TMath::ATan2(xyz0[1],xyz0[0]); | |
729 | Rotate(alpha,kTRUE); | |
730 | GetXYZ(xyz0); | |
e20bef2b | 731 | if(GetProlongation(x,y,z)<0) return -1; |
3b57a3f7 | 732 | xyz1[0] = x * TMath::Cos(alpha) + y * TMath::Sin(alpha); |
733 | xyz1[1] = x * TMath::Sin(alpha) - y * TMath::Cos(alpha); | |
734 | xyz1[2] = z; | |
735 | Double_t param[7]; | |
83dea92e | 736 | if(AliTracker::MeanMaterialBudget(xyz0,xyz1,param)<=0.) return -1; |
3b57a3f7 | 737 | if (param[1] <= 0) { |
738 | param[1] = 100000000; | |
739 | } | |
740 | PropagateTo(x,param[1],param[0]*param[4]); | |
741 | ||
742 | } | |
743 | ||
744 | GetXYZ(xyz0); | |
745 | Double_t alpha = TMath::ATan2(xyz0[1],xyz0[0]); | |
746 | Rotate(alpha,kTRUE); | |
747 | GetXYZ(xyz0); | |
530fb4e2 | 748 | if(GetProlongation(r,y,z)<0) return -1; |
3b57a3f7 | 749 | xyz1[0] = r * TMath::Cos(alpha) + y * TMath::Sin(alpha); |
750 | xyz1[1] = r * TMath::Sin(alpha) - y * TMath::Cos(alpha); | |
751 | xyz1[2] = z; | |
752 | Double_t param[7]; | |
83dea92e | 753 | if(AliTracker::MeanMaterialBudget(xyz0,xyz1,param) <= 0.) return -1; |
3b57a3f7 | 754 | |
755 | if (param[1] <= 0) { | |
756 | param[1] = 100000000; | |
87a7fa94 | 757 | } |
3b57a3f7 | 758 | PropagateTo(r,param[1],param[0]*param[4]); |
759 | ||
760 | return 0; | |
761 | ||
87a7fa94 | 762 | } |
763 | ||
203967fc | 764 | //_____________________________________________________________________________ |
765 | void AliTRDtrackV1::Print(Option_t *o) const | |
766 | { | |
4d6aee34 | 767 | // Print track status |
eb2b4f91 | 768 | AliInfo(Form("PID [%4.1f %4.1f %4.1f %4.1f %4.1f]", 1.E2*fPID[0], 1.E2*fPID[1], 1.E2*fPID[2], 1.E2*fPID[3], 1.E2*fPID[4])); |
203967fc | 769 | AliInfo(Form("Material[%5.2f %5.2f %5.2f]", fBudget[0], fBudget[1], fBudget[2])); |
770 | ||
771 | AliInfo(Form("x[%7.2f] t[%7.4f] alpha[%f] mass[%f]", GetX(), GetIntegratedLength(), GetAlpha(), fMass)); | |
eb2b4f91 | 772 | AliInfo(Form("Ntr[%1d] NtrPID[%1d] Ncl[%3d] lab[%3d]", GetNumberOfTracklets(), GetNumberOfTrackletsPID(), fN, fLab)); |
203967fc | 773 | |
203967fc | 774 | printf("|X| = ("); |
775 | const Double_t *curP = GetParameter(); | |
776 | for (Int_t i = 0; i < 5; i++) printf("%7.2f ", curP[i]); | |
777 | printf(")\n"); | |
778 | ||
779 | printf("|V| = \n"); | |
780 | const Double_t *curC = GetCovariance(); | |
781 | for (Int_t i = 0, j=4, k=0; i<15; i++, k++){ | |
782 | printf("%7.2f ", curC[i]); | |
783 | if(k==j){ | |
784 | printf("\n"); | |
785 | k=-1; j--; | |
786 | } | |
787 | } | |
0217fcd0 | 788 | if(strcmp(o, "a")!=0) return; |
203967fc | 789 | |
790 | for(Int_t ip=0; ip<kNplane; ip++){ | |
791 | if(!fTracklet[ip]) continue; | |
792 | fTracklet[ip]->Print(o); | |
793 | } | |
794 | } | |
795 | ||
796 | ||
3b57a3f7 | 797 | //_____________________________________________________________________________ |
798 | Bool_t AliTRDtrackV1::Rotate(Double_t alpha, Bool_t absolute) | |
799 | { | |
800 | // | |
801 | // Rotates track parameters in R*phi plane | |
802 | // if absolute rotation alpha is in global system | |
803 | // otherwise alpha rotation is relative to the current rotation angle | |
804 | // | |
805 | ||
806 | if (absolute) alpha -= GetAlpha(); | |
807 | //else fNRotate++; | |
808 | ||
809 | return AliExternalTrackParam::Rotate(GetAlpha()+alpha); | |
810 | } | |
87a7fa94 | 811 | |
eb38ed55 | 812 | //___________________________________________________________ |
813 | void AliTRDtrackV1::SetNumberOfClusters() | |
814 | { | |
815 | // Calculate the number of clusters attached to this track | |
816 | ||
3b57a3f7 | 817 | Int_t ncls = 0; |
818 | for(int ip=0; ip<kNplane; ip++){ | |
68f9b6bd | 819 | if(fTracklet[ip] && fTrackletIndex[ip] >= 0) ncls += fTracklet[ip]->GetN(); |
3b57a3f7 | 820 | } |
821 | AliKalmanTrack::SetNumberOfClusters(ncls); | |
eb38ed55 | 822 | } |
823 | ||
824 | ||
0906e73e | 825 | //_______________________________________________________________ |
3b57a3f7 | 826 | void AliTRDtrackV1::SetOwner() |
0906e73e | 827 | { |
828 | // | |
829 | // Toggle ownership of tracklets | |
830 | // | |
831 | ||
e44586fb | 832 | if(TestBit(kOwner)) return; |
3b57a3f7 | 833 | for (Int_t ip = 0; ip < kNplane; ip++) { |
68f9b6bd | 834 | if(fTrackletIndex[ip]<0 || !fTracklet[ip]) continue; |
3b57a3f7 | 835 | fTracklet[ip] = new AliTRDseedV1(*fTracklet[ip]); |
836 | fTracklet[ip]->SetOwner(); | |
837 | } | |
e44586fb | 838 | SetBit(kOwner); |
0906e73e | 839 | } |
840 | ||
d9950a5a | 841 | //_______________________________________________________________ |
4d6aee34 | 842 | void AliTRDtrackV1::SetTracklet(AliTRDseedV1 *const trklt, Int_t index) |
d9950a5a | 843 | { |
844 | // | |
0906e73e | 845 | // Set the tracklets |
d9950a5a | 846 | // |
3b57a3f7 | 847 | Int_t plane = trklt->GetPlane(); |
76b60503 | 848 | |
3b57a3f7 | 849 | fTracklet[plane] = trklt; |
850 | fTrackletIndex[plane] = index; | |
d9950a5a | 851 | } |
852 | ||
e3cf3d02 | 853 | //_______________________________________________________________ |
a310e49b | 854 | void AliTRDtrackV1::SetTrackIn() |
e3cf3d02 | 855 | { |
fc6d4c2d | 856 | // Save location of birth for the TRD track |
857 | // If the pointer is not valid allocate memory | |
858 | // | |
e3cf3d02 | 859 | const AliExternalTrackParam *op = dynamic_cast<const AliExternalTrackParam*>(this); |
fc6d4c2d | 860 | |
861 | if(fTrackLow){ | |
862 | fTrackLow->~AliExternalTrackParam(); | |
863 | new(fTrackLow) AliExternalTrackParam(*op); | |
864 | } else fTrackLow = new AliExternalTrackParam(*op); | |
e3cf3d02 | 865 | } |
866 | ||
867 | //_______________________________________________________________ | |
a310e49b | 868 | void AliTRDtrackV1::SetTrackOut(const AliExternalTrackParam *op) |
e3cf3d02 | 869 | { |
fc6d4c2d | 870 | // Save location of death for the TRD track |
871 | // If the pointer is not valid allocate memory | |
872 | // | |
e3cf3d02 | 873 | if(!op) op = dynamic_cast<const AliExternalTrackParam*>(this); |
fc6d4c2d | 874 | if(fTrackHigh){ |
875 | fTrackHigh->~AliExternalTrackParam(); | |
876 | new(fTrackHigh) AliExternalTrackParam(*op); | |
877 | } else fTrackHigh = new AliExternalTrackParam(*op); | |
e3cf3d02 | 878 | } |
879 | ||
181d2c97 | 880 | //_______________________________________________________________ |
881 | void AliTRDtrackV1::UnsetTracklet(Int_t plane) | |
882 | { | |
883 | if(plane<0 && plane >= kNplane) return; | |
17896e82 | 884 | fTrackletIndex[plane] = -1; |
4d6aee34 | 885 | fTracklet[plane] = NULL; |
181d2c97 | 886 | } |
887 | ||
888 | ||
d9950a5a | 889 | //_______________________________________________________________ |
ef867f55 | 890 | void AliTRDtrackV1::UpdateChi2(Float_t chi2) |
d9950a5a | 891 | { |
ef867f55 | 892 | // Update chi2/track with one tracklet contribution |
b72f4eaf | 893 | SetChi2(GetChi2() + chi2); |
d9950a5a | 894 | } |
895 | ||
896 | //_______________________________________________________________ | |
0906e73e | 897 | void AliTRDtrackV1::UpdateESDtrack(AliESDtrack *track) |
d9950a5a | 898 | { |
899 | // | |
6984f7c1 | 900 | // Update the TRD PID information in the ESD track |
d9950a5a | 901 | // |
6984f7c1 | 902 | |
a2fbb6ec | 903 | Int_t nslices = fkReconstructor->GetRecoParam()->IsEightSlices() ? (Int_t)AliTRDpidUtil::kNNslices : (Int_t)AliTRDpidUtil::kLQslices; |
e20bef2b | 904 | // number of tracklets used for PID calculation |
905 | UChar_t nPID = GetNumberOfTrackletsPID(); | |
906 | // number of tracklets attached to the track | |
907 | UChar_t nTrk = GetNumberOfTracklets(); | |
908 | // pack the two numbers together and store them in the ESD | |
909 | track->SetTRDntracklets(nPID | (nTrk<<3)); | |
910 | // allocate space to store raw PID signals dEdx & momentum | |
b48e9cbb | 911 | track->SetNumberOfTRDslices((nslices+2)*AliTRDgeometry::kNlayer); |
e20bef2b | 912 | // store raw signals |
7b23a4e1 | 913 | Float_t p, sp; Double_t spd; |
6984f7c1 | 914 | for (Int_t ip = 0; ip < kNplane; ip++) { |
68f9b6bd | 915 | if(fTrackletIndex[ip]<0 || !fTracklet[ip]) continue; |
e20bef2b | 916 | if(!fTracklet[ip]->HasPID()) continue; |
4d6aee34 | 917 | const Float_t *dedx = fTracklet[ip]->GetdEdx(); |
215f7116 | 918 | for (Int_t js = 0; js < nslices; js++, dedx++) track->SetTRDslice(*dedx, ip, js); |
7b23a4e1 | 919 | p = fTracklet[ip]->GetMomentum(&sp); spd = sp; |
920 | track->SetTRDmomentum(p, ip, &spd); | |
6984f7c1 | 921 | } |
e20bef2b | 922 | // store PID probabilities |
923 | track->SetTRDpid(fPID); | |
d9950a5a | 924 | } |