]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliESDtrack.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Added QA for digits during reconstruction (Yves)
[u/mrichter/AliRoot.git] / STEER / AliESDtrack.cxx
CommitLineData
ae982df3 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 **************************************************************************/
ae982df3 15//-----------------------------------------------------------------
16// Implementation of the ESD track class
4427806c 17// ESD = Event Summary Data
15614b8b 18// This is the class to deal with during the phisics analysis of data
4427806c 19// Origin: Iouri Belikov, CERN
20// e-mail: Jouri.Belikov@cern.ch
4df45162 21//
22//
23//
24// What do you need to know before starting analysis
25// (by Marian Ivanov: marian.ivanov@cern.ch)
26//
27//
28// AliESDtrack:
29// 1. What is the AliESDtrack
30// 2. What informations do we store
31// 3. How to use the information for analysis
32//
33//
34// 1.AliESDtrack is the container of the information about the track/particle
35// reconstructed during Barrel Tracking.
36// The track information is propagated from one tracking detector to
37// other using the functionality of AliESDtrack - Current parameters.
38//
39// No global fit model is used.
40// Barrel tracking use Kalman filtering technique, it gives optimal local
41// track parameters at given point under certian assumptions.
42//
43// Kalman filter take into account additional effect which are
44// difficult to handle using global fit.
45// Effects:
46// a.) Multiple scattering
47// b.) Energy loss
48// c.) Non homogenous magnetic field
49//
50// In general case, following barrel detectors are contributing to
51// the Kalman track information:
52// a. TPC
53// b. ITS
54// c. TRD
55//
56// In general 3 reconstruction itteration are performed:
57// 1. Find tracks - sequence TPC->ITS
58// 2. PropagateBack - sequence ITS->TPC->TRD -> Outer PID detectors
59// 3. Refit invward - sequence TRD->TPC->ITS
60// The current tracks are updated after each detector (see bellow).
61// In specical cases a track sanpshots are stored.
62//
63//
64// For some type of analysis (+visualization) track local parameters at
65// different position are neccesary. A snapshots during the track
66// propagation are created.
67// (See AliExternalTrackParam class for desctiption of variables and
68// functionality)
69// Snapshots:
70// a. Current parameters - class itself (AliExternalTrackParam)
71// Contributors: general case TRD->TPC->ITS
72// Preferable usage: Decission - primary or secondary track
73// NOTICE - By default the track parameters are stored at the DCA point
74// to the primary vertex. optimal for primary tracks,
75// far from optimal for secondary tracks.
76// b. Constrained parameters - Kalman information updated with
77// the Primary vertex information
78// Contributors: general case TRD->TPC->ITS
79// Preferable usage: Use only for tracks selected as primary
80// NOTICE - not real constrain - taken as additional measurement
81// with corresponding error
82// Function:
83// const AliExternalTrackParam *GetConstrainedParam() const {return fCp;}
84// c. Inner parameters - Track parameters at inner wall of the TPC
85// Contributors: general case TRD->TPC
86// function:
87// const AliExternalTrackParam *GetInnerParam() const { return fIp;}
88//
89// d. TPCinnerparam - contributors - TPC only
90// Contributors: TPC
91// Preferable usage: Requested for HBT study
92// (smaller correlations as using also ITS information)
93// NOTICE - the track parameters are propagated to the DCA to
94// to primary vertex
95// Optimal for primary, far from optimal for secondary tracks
96// Function:
97// const AliExternalTrackParam *GetTPCInnerParam() const {return fTPCInner;}
98//
99// e. Outer parameters -
100// Contributors- general case - ITS-> TPC -> TRD
101// The last point - Outer parameters radius is determined
102// e.a) Local inclination angle bigger than threshold -
103// Low momenta tracks
104// e.a) Catastrofic energy losss in material
105// e.b) Not further improvement (no space points)
106// Usage:
107// a.) Tracking: Starting parameter for Refit inward
108// b.) Visualization
109// c.) QA
110// NOTICE: Should be not used for the physic analysis
111// Function:
112// const AliExternalTrackParam *GetOuterParam() const { return fOp;}
113//
ae982df3 114//-----------------------------------------------------------------
115
e1e6896f 116#include <TMath.h>
49edfa08 117#include <TParticle.h>
ae982df3 118
49d13e89 119#include "AliESDVertex.h"
ae982df3 120#include "AliESDtrack.h"
121#include "AliKalmanTrack.h"
4f6e22bd 122#include "AliVTrack.h"
5f7789fc 123#include "AliLog.h"
15e85efa 124#include "AliTrackPointArray.h"
0c19adf7 125#include "TPolyMarker3D.h"
ae982df3 126
127ClassImp(AliESDtrack)
128
562dd0b4 129void SetPIDValues(Double_t * dest, const Double_t * src, Int_t n) {
d27bbc79 130 // This function copies "n" PID weights from "scr" to "dest"
131 // and normalizes their sum to 1 thus producing conditional probabilities.
132 // The negative weights are set to 0.
133 // In case all the weights are non-positive they are replaced by
134 // uniform probabilities
135
136 if (n<=0) return;
137
138 Float_t uniform = 1./(Float_t)n;
139
140 Float_t sum = 0;
141 for (Int_t i=0; i<n; i++)
142 if (src[i]>=0) {
143 sum+=src[i];
144 dest[i] = src[i];
145 }
146 else {
147 dest[i] = 0;
148 }
149
150 if(sum>0)
151 for (Int_t i=0; i<n; i++) dest[i] /= sum;
152 else
153 for (Int_t i=0; i<n; i++) dest[i] = uniform;
154}
155
ae982df3 156//_______________________________________________________________________
157AliESDtrack::AliESDtrack() :
c9ec41e8 158 AliExternalTrackParam(),
562dd0b4 159 fCp(0),
160 fIp(0),
161 fTPCInner(0),
162 fOp(0),
163 fFriendTrack(new AliESDfriendTrack()),
164 fTPCClusterMap(159),//number of padrows
165 fTPCSharedMap(159),//number of padrows
90e48c0c 166 fFlags(0),
90e48c0c 167 fID(0),
562dd0b4 168 fLabel(0),
169 fITSLabel(0),
170 fTPCLabel(0),
171 fTRDLabel(0),
172 fTOFCalChannel(0),
ce3f4882 173 fTOFindex(-1),
562dd0b4 174 fHMPIDqn(0),
81aa7a0d 175 fHMPIDcluIdx(-1),
562dd0b4 176 fEMCALindex(kEMCALNoMatch),
177 fHMPIDtrkTheta(0),
178 fHMPIDtrkPhi(0),
179 fHMPIDsignal(0),
90e48c0c 180 fTrackLength(0),
d7ddf1e9 181 fdTPC(0),fzTPC(0),
182 fCddTPC(0),fCdzTPC(0),fCzzTPC(0),
436dfe39 183 fCchi2TPC(0),
49d13e89 184 fD(0),fZ(0),
185 fCdd(0),fCdz(0),fCzz(0),
562dd0b4 186 fCchi2(0),
90e48c0c 187 fITSchi2(0),
90e48c0c 188 fTPCchi2(0),
562dd0b4 189 fTRDchi2(0),
190 fTOFchi2(0),
191 fHMPIDchi2(0),
b5b2b4db 192 fGlobalChi2(0),
562dd0b4 193 fITSsignal(0),
90e48c0c 194 fTPCsignal(0),
e1d4c1b5 195 fTPCsignalS(0),
90e48c0c 196 fTRDsignal(0),
90e48c0c 197 fTRDQuality(0),
23d49657 198 fTRDBudget(0),
562dd0b4 199 fTOFsignal(0),
85324138 200 fTOFsignalToT(0),
d321691a 201 fTOFsignalRaw(0),
202 fTOFsignalDz(0),
562dd0b4 203 fHMPIDtrkX(0),
204 fHMPIDtrkY(0),
205 fHMPIDmipX(0),
206 fHMPIDmipY(0),
207 fTPCncls(0),
208 fTPCnclsF(0),
209 fTPCsignalN(0),
210 fITSncls(0),
211 fITSClusterMap(0),
212 fTRDncls(0),
213 fTRDncls0(0),
ed15ef4f 214 fTRDntracklets(0),
6984f7c1 215 fTRDnSlices(0),
6dc21f57 216 fTRDslices(0x0),
217 fVertexID(-2) // -2 means an orphan track
ae982df3 218{
219 //
220 // The default ESD constructor
221 //
6984f7c1 222 Int_t i;
6238d7a9 223 for (i=0; i<AliPID::kSPECIES; i++) {
4a78b8c5 224 fTrackTime[i]=0.;
562dd0b4 225 fR[i]=0.;
226 fITSr[i]=0.;
227 fTPCr[i]=0.;
228 fTRDr[i]=0.;
229 fTOFr[i]=0.;
230 fHMPIDr[i]=0.;
2bad268c 231 }
ac2f7574 232
ef7253ac 233 for (i=0; i<3; i++) { fKinkIndexes[i]=0;}
562dd0b4 234 for (i=0; i<3; i++) { fV0Indexes[i]=0;}
6984f7c1 235 for (i=0;i<kTRDnPlanes;i++) {
562dd0b4 236 fTRDTimBin[i]=0;
6d45eaef 237 }
562dd0b4 238 for (i=0;i<4;i++) {fTPCPoints[i]=0;}
239 for (i=0;i<3;i++) {fTOFLabel[i]=0;}
240 for (i=0;i<10;i++) {fTOFInfo[i]=0;}
89f1b176 241 for (i=0;i<12;i++) {fITSModule[i]=-1;}
c4d11b15 242}
243
244//_______________________________________________________________________
90e48c0c 245AliESDtrack::AliESDtrack(const AliESDtrack& track):
c9ec41e8 246 AliExternalTrackParam(track),
562dd0b4 247 fCp(0),
248 fIp(0),
249 fTPCInner(0),
250 fOp(0),
251 fFriendTrack(0),
252 fTPCClusterMap(track.fTPCClusterMap),
253 fTPCSharedMap(track.fTPCSharedMap),
90e48c0c 254 fFlags(track.fFlags),
90e48c0c 255 fID(track.fID),
562dd0b4 256 fLabel(track.fLabel),
257 fITSLabel(track.fITSLabel),
258 fTPCLabel(track.fTPCLabel),
259 fTRDLabel(track.fTRDLabel),
260 fTOFCalChannel(track.fTOFCalChannel),
261 fTOFindex(track.fTOFindex),
262 fHMPIDqn(track.fHMPIDqn),
263 fHMPIDcluIdx(track.fHMPIDcluIdx),
264 fEMCALindex(track.fEMCALindex),
265 fHMPIDtrkTheta(track.fHMPIDtrkTheta),
266 fHMPIDtrkPhi(track.fHMPIDtrkPhi),
267 fHMPIDsignal(track.fHMPIDsignal),
90e48c0c 268 fTrackLength(track.fTrackLength),
d7ddf1e9 269 fdTPC(track.fdTPC),fzTPC(track.fzTPC),
270 fCddTPC(track.fCddTPC),fCdzTPC(track.fCdzTPC),fCzzTPC(track.fCzzTPC),
436dfe39 271 fCchi2TPC(track.fCchi2TPC),
49d13e89 272 fD(track.fD),fZ(track.fZ),
273 fCdd(track.fCdd),fCdz(track.fCdz),fCzz(track.fCzz),
90e48c0c 274 fCchi2(track.fCchi2),
90e48c0c 275 fITSchi2(track.fITSchi2),
90e48c0c 276 fTPCchi2(track.fTPCchi2),
562dd0b4 277 fTRDchi2(track.fTRDchi2),
278 fTOFchi2(track.fTOFchi2),
279 fHMPIDchi2(track.fHMPIDchi2),
b5b2b4db 280 fGlobalChi2(track.fGlobalChi2),
562dd0b4 281 fITSsignal(track.fITSsignal),
90e48c0c 282 fTPCsignal(track.fTPCsignal),
e1d4c1b5 283 fTPCsignalS(track.fTPCsignalS),
90e48c0c 284 fTRDsignal(track.fTRDsignal),
90e48c0c 285 fTRDQuality(track.fTRDQuality),
23d49657 286 fTRDBudget(track.fTRDBudget),
90e48c0c 287 fTOFsignal(track.fTOFsignal),
85324138 288 fTOFsignalToT(track.fTOFsignalToT),
d321691a 289 fTOFsignalRaw(track.fTOFsignalRaw),
290 fTOFsignalDz(track.fTOFsignalDz),
f4b3bbb7 291 fHMPIDtrkX(track.fHMPIDtrkX),
292 fHMPIDtrkY(track.fHMPIDtrkY),
293 fHMPIDmipX(track.fHMPIDmipX),
294 fHMPIDmipY(track.fHMPIDmipY),
562dd0b4 295 fTPCncls(track.fTPCncls),
296 fTPCnclsF(track.fTPCnclsF),
297 fTPCsignalN(track.fTPCsignalN),
298 fITSncls(track.fITSncls),
299 fITSClusterMap(track.fITSClusterMap),
300 fTRDncls(track.fTRDncls),
301 fTRDncls0(track.fTRDncls0),
ed15ef4f 302 fTRDntracklets(track.fTRDntracklets),
6984f7c1 303 fTRDnSlices(track.fTRDnSlices),
6dc21f57 304 fTRDslices(0x0),
305 fVertexID(track.fVertexID)
90e48c0c 306{
c4d11b15 307 //
308 //copy constructor
309 //
ef7253ac 310 for (Int_t i=0;i<AliPID::kSPECIES;i++) fTrackTime[i]=track.fTrackTime[i];
311 for (Int_t i=0;i<AliPID::kSPECIES;i++) fR[i]=track.fR[i];
c4d11b15 312 //
304864ab 313 for (Int_t i=0;i<AliPID::kSPECIES;i++) fITSr[i]=track.fITSr[i];
c4d11b15 314 //
304864ab 315 for (Int_t i=0;i<AliPID::kSPECIES;i++) fTPCr[i]=track.fTPCr[i];
51ad6848 316 for (Int_t i=0;i<4;i++) {fTPCPoints[i]=track.fTPCPoints[i];}
317 for (Int_t i=0; i<3;i++) { fKinkIndexes[i]=track.fKinkIndexes[i];}
318 for (Int_t i=0; i<3;i++) { fV0Indexes[i]=track.fV0Indexes[i];}
c4d11b15 319 //
6984f7c1 320 for (Int_t i=0;i<kTRDnPlanes;i++) {
6d45eaef 321 fTRDTimBin[i]=track.fTRDTimBin[i];
eab5961e 322 }
6984f7c1 323
324 if (fTRDnSlices) {
325 fTRDslices=new Double32_t[fTRDnSlices];
326 for (Int_t i=0; i<fTRDnSlices; i++) fTRDslices[i]=track.fTRDslices[i];
327 }
328
304864ab 329 for (Int_t i=0;i<AliPID::kSPECIES;i++) fTRDr[i]=track.fTRDr[i];
304864ab 330 for (Int_t i=0;i<AliPID::kSPECIES;i++) fTOFr[i]=track.fTOFr[i];
51ad6848 331 for (Int_t i=0;i<3;i++) fTOFLabel[i]=track.fTOFLabel[i];
332 for (Int_t i=0;i<10;i++) fTOFInfo[i]=track.fTOFInfo[i];
89f1b176 333 for (Int_t i=0;i<12;i++) fITSModule[i]=track.fITSModule[i];
f4b3bbb7 334 for (Int_t i=0;i<AliPID::kSPECIES;i++) fHMPIDr[i]=track.fHMPIDr[i];
c9ec41e8 335
336 if (track.fCp) fCp=new AliExternalTrackParam(*track.fCp);
337 if (track.fIp) fIp=new AliExternalTrackParam(*track.fIp);
4aeb9470 338 if (track.fTPCInner) fTPCInner=new AliExternalTrackParam(*track.fTPCInner);
c9ec41e8 339 if (track.fOp) fOp=new AliExternalTrackParam(*track.fOp);
15e85efa 340
8497bca0 341 if (track.fFriendTrack) fFriendTrack=new AliESDfriendTrack(*(track.fFriendTrack));
ae982df3 342}
15e85efa 343
4f6e22bd 344//_______________________________________________________________________
345AliESDtrack::AliESDtrack(const AliVTrack *track) :
346 AliExternalTrackParam(track),
347 fCp(0),
348 fIp(0),
349 fTPCInner(0),
350 fOp(0),
351 fFriendTrack(0),
352 fTPCClusterMap(159),//number of padrows
353 fTPCSharedMap(159),//number of padrows
354 fFlags(0),
355 fID(),
356 fLabel(0),
357 fITSLabel(0),
358 fTPCLabel(0),
359 fTRDLabel(0),
360 fTOFCalChannel(0),
361 fTOFindex(-1),
362 fHMPIDqn(0),
81aa7a0d 363 fHMPIDcluIdx(-1),
4f6e22bd 364 fEMCALindex(kEMCALNoMatch),
365 fHMPIDtrkTheta(0),
366 fHMPIDtrkPhi(0),
367 fHMPIDsignal(0),
368 fTrackLength(0),
369 fdTPC(0),fzTPC(0),
370 fCddTPC(0),fCdzTPC(0),fCzzTPC(0),
371 fCchi2TPC(0),
372 fD(0),fZ(0),
373 fCdd(0),fCdz(0),fCzz(0),
374 fCchi2(0),
375 fITSchi2(0),
376 fTPCchi2(0),
377 fTRDchi2(0),
378 fTOFchi2(0),
379 fHMPIDchi2(0),
b5b2b4db 380 fGlobalChi2(0),
4f6e22bd 381 fITSsignal(0),
382 fTPCsignal(0),
383 fTPCsignalS(0),
384 fTRDsignal(0),
385 fTRDQuality(0),
386 fTRDBudget(0),
387 fTOFsignal(0),
388 fTOFsignalToT(0),
389 fTOFsignalRaw(0),
390 fTOFsignalDz(0),
391 fHMPIDtrkX(0),
392 fHMPIDtrkY(0),
393 fHMPIDmipX(0),
394 fHMPIDmipY(0),
395 fTPCncls(0),
396 fTPCnclsF(0),
397 fTPCsignalN(0),
398 fITSncls(0),
399 fITSClusterMap(0),
400 fTRDncls(0),
401 fTRDncls0(0),
ed15ef4f 402 fTRDntracklets(0),
4f6e22bd 403 fTRDnSlices(0),
6dc21f57 404 fTRDslices(0x0),
405 fVertexID(-2) // -2 means an orphan track
4f6e22bd 406{
407 //
610e3088 408 // ESD track from AliVTrack.
409 // This is not a copy constructor !
4f6e22bd 410 //
411
610e3088 412 if (track->InheritsFrom("AliExternalTrackParam")) {
413 AliError("This is not a copy constructor. Use AliESDtrack(const AliESDtrack &) !");
414 AliWarning("Calling the default constructor...");
415 AliESDtrack();
416 return;
417 }
418
4f6e22bd 419 // Reset all the arrays
420 Int_t i;
421 for (i=0; i<AliPID::kSPECIES; i++) {
422 fTrackTime[i]=0.;
423 fR[i]=0.;
424 fITSr[i]=0.;
425 fTPCr[i]=0.;
426 fTRDr[i]=0.;
427 fTOFr[i]=0.;
428 fHMPIDr[i]=0.;
429 }
430
431 for (i=0; i<3; i++) { fKinkIndexes[i]=0;}
432 for (i=0; i<3; i++) { fV0Indexes[i]=-1;}
433 for (i=0;i<kTRDnPlanes;i++) {
434 fTRDTimBin[i]=0;
435 }
436 for (i=0;i<4;i++) {fTPCPoints[i]=0;}
437 for (i=0;i<3;i++) {fTOFLabel[i]=0;}
438 for (i=0;i<10;i++) {fTOFInfo[i]=0;}
439 for (i=0;i<12;i++) {fITSModule[i]=-1;}
440
441 // Set the ID
442 SetID(track->GetID());
443
444 // Set ITS cluster map
445 fITSClusterMap=track->GetITSClusterMap();
446
d577eec9 447 fITSncls=0;
448 for(i=0; i<6; i++) {
449 if(HasPointOnITSLayer(i)) fITSncls++;
450 }
451
4f6e22bd 452 // Set the combined PID
453 const Double_t *pid = track->PID();
67be2d29 454 if(pid){
455 for (i=0; i<AliPID::kSPECIES; i++) fR[i]=pid[i];
456 }
4f6e22bd 457 // AliESD track label
458 SetLabel(track->GetLabel());
39ca41b3 459 // Set the status
460 SetStatus(track->GetStatus());
4f6e22bd 461}
462
49edfa08 463//_______________________________________________________________________
464AliESDtrack::AliESDtrack(TParticle * part) :
465 AliExternalTrackParam(),
562dd0b4 466 fCp(0),
467 fIp(0),
468 fTPCInner(0),
469 fOp(0),
470 fFriendTrack(0),
471 fTPCClusterMap(159),//number of padrows
472 fTPCSharedMap(159),//number of padrows
49edfa08 473 fFlags(0),
49edfa08 474 fID(0),
562dd0b4 475 fLabel(0),
476 fITSLabel(0),
477 fTPCLabel(0),
478 fTRDLabel(0),
479 fTOFCalChannel(0),
ce3f4882 480 fTOFindex(-1),
562dd0b4 481 fHMPIDqn(0),
81aa7a0d 482 fHMPIDcluIdx(-1),
562dd0b4 483 fEMCALindex(kEMCALNoMatch),
484 fHMPIDtrkTheta(0),
485 fHMPIDtrkPhi(0),
486 fHMPIDsignal(0),
49edfa08 487 fTrackLength(0),
d7ddf1e9 488 fdTPC(0),fzTPC(0),
489 fCddTPC(0),fCdzTPC(0),fCzzTPC(0),
436dfe39 490 fCchi2TPC(0),
49edfa08 491 fD(0),fZ(0),
492 fCdd(0),fCdz(0),fCzz(0),
562dd0b4 493 fCchi2(0),
49edfa08 494 fITSchi2(0),
49edfa08 495 fTPCchi2(0),
562dd0b4 496 fTRDchi2(0),
497 fTOFchi2(0),
498 fHMPIDchi2(0),
b5b2b4db 499 fGlobalChi2(0),
562dd0b4 500 fITSsignal(0),
49edfa08 501 fTPCsignal(0),
49edfa08 502 fTPCsignalS(0),
49edfa08 503 fTRDsignal(0),
49edfa08 504 fTRDQuality(0),
505 fTRDBudget(0),
562dd0b4 506 fTOFsignal(0),
49edfa08 507 fTOFsignalToT(0),
d321691a 508 fTOFsignalRaw(0),
509 fTOFsignalDz(0),
562dd0b4 510 fHMPIDtrkX(0),
511 fHMPIDtrkY(0),
512 fHMPIDmipX(0),
513 fHMPIDmipY(0),
514 fTPCncls(0),
515 fTPCnclsF(0),
516 fTPCsignalN(0),
517 fITSncls(0),
518 fITSClusterMap(0),
519 fTRDncls(0),
520 fTRDncls0(0),
ed15ef4f 521 fTRDntracklets(0),
6984f7c1 522 fTRDnSlices(0),
6dc21f57 523 fTRDslices(0x0),
524 fVertexID(-2) // -2 means an orphan track
49edfa08 525{
526 //
527 // ESD track from TParticle
528 //
529
530 // Reset all the arrays
6984f7c1 531 Int_t i;
49edfa08 532 for (i=0; i<AliPID::kSPECIES; i++) {
533 fTrackTime[i]=0.;
534 fR[i]=0.;
535 fITSr[i]=0.;
536 fTPCr[i]=0.;
537 fTRDr[i]=0.;
538 fTOFr[i]=0.;
f4b3bbb7 539 fHMPIDr[i]=0.;
49edfa08 540 }
541
542 for (i=0; i<3; i++) { fKinkIndexes[i]=0;}
543 for (i=0; i<3; i++) { fV0Indexes[i]=-1;}
6984f7c1 544 for (i=0;i<kTRDnPlanes;i++) {
562dd0b4 545 fTRDTimBin[i]=0;
49edfa08 546 }
562dd0b4 547 for (i=0;i<4;i++) {fTPCPoints[i]=0;}
548 for (i=0;i<3;i++) {fTOFLabel[i]=0;}
549 for (i=0;i<10;i++) {fTOFInfo[i]=0;}
89f1b176 550 for (i=0;i<12;i++) {fITSModule[i]=-1;}
49edfa08 551
552 // Calculate the AliExternalTrackParam content
553
554 Double_t xref;
555 Double_t alpha;
556 Double_t param[5];
557 Double_t covar[15];
558
559 // Calculate alpha: the rotation angle of the corresponding local system (TPC sector)
560 alpha = part->Phi()*180./TMath::Pi();
561 if (alpha<0) alpha+= 360.;
562 if (alpha>360) alpha -= 360.;
563
564 Int_t sector = (Int_t)(alpha/20.);
565 alpha = 10. + 20.*sector;
566 alpha /= 180;
567 alpha *= TMath::Pi();
568
569 // Covariance matrix: no errors, the parameters are exact
6c27b212 570 for (i=0; i<15; i++) covar[i]=0.;
49edfa08 571
572 // Get the vertex of origin and the momentum
573 TVector3 ver(part->Vx(),part->Vy(),part->Vz());
574 TVector3 mom(part->Px(),part->Py(),part->Pz());
575
576 // Rotate to the local coordinate system (TPC sector)
577 ver.RotateZ(-alpha);
578 mom.RotateZ(-alpha);
579
580 // X of the referense plane
581 xref = ver.X();
582
583 Int_t pdgCode = part->GetPdgCode();
584
585 Double_t charge =
586 TDatabasePDG::Instance()->GetParticle(pdgCode)->Charge();
587
588 param[0] = ver.Y();
589 param[1] = ver.Z();
590 param[2] = TMath::Sin(mom.Phi());
591 param[3] = mom.Pz()/mom.Pt();
592 param[4] = TMath::Sign(1/mom.Pt(),charge);
593
594 // Set AliExternalTrackParam
595 Set(xref, alpha, param, covar);
596
597 // Set the PID
598 Int_t indexPID = 99;
599
600 switch (TMath::Abs(pdgCode)) {
601
602 case 11: // electron
603 indexPID = 0;
604 break;
605
606 case 13: // muon
607 indexPID = 1;
608 break;
609
610 case 211: // pion
611 indexPID = 2;
612 break;
613
614 case 321: // kaon
615 indexPID = 3;
616 break;
617
618 case 2212: // proton
619 indexPID = 4;
620 break;
621
622 default:
623 break;
624 }
625
626 // If the particle is not e,mu,pi,K or p the PID probabilities are set to 0
627 if (indexPID < AliPID::kSPECIES) {
628 fR[indexPID]=1.;
629 fITSr[indexPID]=1.;
630 fTPCr[indexPID]=1.;
631 fTRDr[indexPID]=1.;
632 fTOFr[indexPID]=1.;
f4b3bbb7 633 fHMPIDr[indexPID]=1.;
49edfa08 634
635 }
636 // AliESD track label
637 SetLabel(part->GetUniqueID());
638
639}
640
c4d11b15 641//_______________________________________________________________________
642AliESDtrack::~AliESDtrack(){
643 //
644 // This is destructor according Coding Conventrions
645 //
646 //printf("Delete track\n");
c9ec41e8 647 delete fIp;
4aeb9470 648 delete fTPCInner;
c9ec41e8 649 delete fOp;
650 delete fCp;
15e85efa 651 delete fFriendTrack;
6984f7c1 652 if(fTRDnSlices)
653 delete[] fTRDslices;
c4d11b15 654}
ae982df3 655
732a24fe 656AliESDtrack &AliESDtrack::operator=(const AliESDtrack &source){
657
658
659 if(&source == this) return *this;
660 AliExternalTrackParam::operator=(source);
661
662
663 if(source.fCp){
664 // we have the trackparam: assign or copy construct
665 if(fCp)*fCp = *source.fCp;
666 else fCp = new AliExternalTrackParam(*source.fCp);
667 }
668 else{
669 // no track param delete the old one
670 if(fCp)delete fCp;
671 fCp = 0;
672 }
673
674 if(source.fIp){
675 // we have the trackparam: assign or copy construct
676 if(fIp)*fIp = *source.fIp;
677 else fIp = new AliExternalTrackParam(*source.fIp);
678 }
679 else{
680 // no track param delete the old one
681 if(fIp)delete fIp;
682 fIp = 0;
683 }
684
685
686 if(source.fTPCInner){
687 // we have the trackparam: assign or copy construct
688 if(fTPCInner) *fTPCInner = *source.fTPCInner;
689 else fTPCInner = new AliExternalTrackParam(*source.fTPCInner);
690 }
691 else{
692 // no track param delete the old one
693 if(fTPCInner)delete fTPCInner;
694 fTPCInner = 0;
695 }
696
697
698 if(source.fOp){
699 // we have the trackparam: assign or copy construct
700 if(fOp) *fOp = *source.fOp;
701 else fOp = new AliExternalTrackParam(*source.fOp);
702 }
703 else{
704 // no track param delete the old one
705 if(fOp)delete fOp;
706 fOp = 0;
707 }
708
709 // copy also the friend track
710 // use copy constructor
711 if(source.fFriendTrack){
712 // we have the trackparam: assign or copy construct
713 delete fFriendTrack; fFriendTrack=new AliESDfriendTrack(*source.fFriendTrack);
714 }
715 else{
716 // no track param delete the old one
717 delete fFriendTrack; fFriendTrack= 0;
718 }
719
720 fTPCClusterMap = source.fTPCClusterMap;
721 fTPCSharedMap = source.fTPCSharedMap;
722 // the simple stuff
723 fFlags = source.fFlags;
724 fID = source.fID;
725 fLabel = source.fLabel;
726 fITSLabel = source.fITSLabel;
727 for(int i = 0; i< 12;++i){
728 fITSModule[i] = source.fITSModule[i];
729 }
730 fTPCLabel = source.fTPCLabel;
731 fTRDLabel = source.fTRDLabel;
732 for(int i = 0; i< 3;++i){
733 fTOFLabel[i] = source.fTOFLabel[i];
734 }
735 fTOFCalChannel = source.fTOFCalChannel;
736 fTOFindex = source.fTOFindex;
737 fHMPIDqn = source.fHMPIDqn;
738 fHMPIDcluIdx = source.fHMPIDcluIdx;
739 fEMCALindex = source.fEMCALindex;
740
741 for(int i = 0; i< 3;++i){
742 fKinkIndexes[i] = source.fKinkIndexes[i];
743 fV0Indexes[i] = source.fV0Indexes[i];
744 }
745
746 for(int i = 0; i< AliPID::kSPECIES;++i){
747 fR[i] = source.fR[i];
748 fITSr[i] = source.fITSr[i];
749 fTPCr[i] = source.fTPCr[i];
750 fTRDr[i] = source.fTRDr[i];
751 fTOFr[i] = source.fTOFr[i];
752 fHMPIDr[i] = source.fHMPIDr[i];
753 fTrackTime[i] = source.fTrackTime[i];
754 }
755
756 fHMPIDtrkTheta = source.fHMPIDtrkTheta;
757 fHMPIDtrkPhi = source.fHMPIDtrkPhi;
758 fHMPIDsignal = source.fHMPIDsignal;
759
760
761 fTrackLength = source. fTrackLength;
d7ddf1e9 762 fdTPC = source.fdTPC;
763 fzTPC = source.fzTPC;
764 fCddTPC = source.fCddTPC;
765 fCdzTPC = source.fCdzTPC;
766 fCzzTPC = source.fCzzTPC;
436dfe39 767 fCchi2TPC = source.fCchi2TPC;
768
732a24fe 769 fD = source.fD;
770 fZ = source.fZ;
771 fCdd = source.fCdd;
772 fCdz = source.fCdz;
773 fCzz = source.fCzz;
732a24fe 774 fCchi2 = source.fCchi2;
436dfe39 775
732a24fe 776 fITSchi2 = source.fITSchi2;
777 fTPCchi2 = source.fTPCchi2;
778 fTRDchi2 = source.fTRDchi2;
779 fTOFchi2 = source.fTOFchi2;
780 fHMPIDchi2 = source.fHMPIDchi2;
781
b5b2b4db 782 fGlobalChi2 = source.fGlobalChi2;
732a24fe 783
784 fITSsignal = source.fITSsignal;
785 fTPCsignal = source.fTPCsignal;
786 fTPCsignalS = source.fTPCsignalS;
787 for(int i = 0; i< 4;++i){
788 fTPCPoints[i] = source.fTPCPoints[i];
789 }
790 fTRDsignal = source.fTRDsignal;
791
6984f7c1 792 for(int i = 0;i < kTRDnPlanes;++i){
732a24fe 793 fTRDTimBin[i] = source.fTRDTimBin[i];
732a24fe 794 }
6984f7c1 795
796 if(fTRDnSlices)
797 delete[] fTRDslices;
798 fTRDslices=0;
799 fTRDnSlices=source.fTRDnSlices;
800 if (fTRDnSlices) {
801 fTRDslices=new Double32_t[fTRDnSlices];
802 for(int j = 0;j < fTRDnSlices;++j) fTRDslices[j] = source.fTRDslices[j];
803 }
804
732a24fe 805 fTRDQuality = source.fTRDQuality;
806 fTRDBudget = source.fTRDBudget;
807 fTOFsignal = source.fTOFsignal;
808 fTOFsignalToT = source.fTOFsignalToT;
809 fTOFsignalRaw = source.fTOFsignalRaw;
810 fTOFsignalDz = source.fTOFsignalDz;
811
812 for(int i = 0;i<10;++i){
813 fTOFInfo[i] = source.fTOFInfo[i];
814 }
815
816 fHMPIDtrkX = source.fHMPIDtrkX;
817 fHMPIDtrkY = source.fHMPIDtrkY;
818 fHMPIDmipX = source.fHMPIDmipX;
819 fHMPIDmipY = source.fHMPIDmipY;
820
821 fTPCncls = source.fTPCncls;
822 fTPCnclsF = source.fTPCnclsF;
823 fTPCsignalN = source.fTPCsignalN;
824
825 fITSncls = source.fITSncls;
826 fITSClusterMap = source.fITSClusterMap;
827 fTRDncls = source.fTRDncls;
828 fTRDncls0 = source.fTRDncls0;
ed15ef4f 829 fTRDntracklets = source.fTRDntracklets;
6dc21f57 830 fVertexID = source.fVertexID;
732a24fe 831 return *this;
832}
833
834
835
836void AliESDtrack::Copy(TObject &obj) const {
837
838 // this overwrites the virtual TOBject::Copy()
839 // to allow run time copying without casting
840 // in AliESDEvent
841
842 if(this==&obj)return;
843 AliESDtrack *robj = dynamic_cast<AliESDtrack*>(&obj);
844 if(!robj)return; // not an AliESDtrack
845 *robj = *this;
846
847}
848
849
850
00dce61a 851void AliESDtrack::AddCalibObject(TObject * object){
852 //
853 // add calib object to the list
854 //
855 if (!fFriendTrack) fFriendTrack = new AliESDfriendTrack;
856 fFriendTrack->AddCalibObject(object);
857}
858
859TObject * AliESDtrack::GetCalibObject(Int_t index){
860 //
861 // return calib objct at given position
862 //
863 if (!fFriendTrack) return 0;
864 return fFriendTrack->GetCalibObject(index);
865}
866
867
f12d42ce 868Bool_t AliESDtrack::FillTPCOnlyTrack(AliESDtrack &track){
b9ca886f 869
870 // Fills the information of the TPC-only first reconstruction pass
871 // into the passed ESDtrack object. For consistency fTPCInner is also filled
872 // again
873
5b305f70 874
875
876 // For data produced before r26675
877 // RelateToVertexTPC was not properly called during reco
878 // so you'll have to call it again, before FillTPCOnlyTrack
879 // Float_t p[2],cov[3];
880 // track->GetImpactParametersTPC(p,cov);
881 // if(p[0]==0&&p[1]==0) // <- Default values
882 // track->RelateToVertexTPC(esd->GetPrimaryVertexTPC(),esd->GetMagneticField(),kVeryBig);
883
884
b9ca886f 885 if(!fTPCInner)return kFALSE;
886
887 // fill the TPC track params to the global track parameters
888 track.Set(fTPCInner->GetX(),fTPCInner->GetAlpha(),fTPCInner->GetParameter(),fTPCInner->GetCovariance());
889 track.fD = fdTPC;
890 track.fZ = fzTPC;
891 track.fCdd = fCddTPC;
892 track.fCdz = fCdzTPC;
893 track.fCzz = fCzzTPC;
894
895 // copy the TPCinner parameters
896 if(track.fTPCInner) *track.fTPCInner = *fTPCInner;
897 else track.fTPCInner = new AliExternalTrackParam(*fTPCInner);
898 track.fdTPC = fdTPC;
899 track.fzTPC = fzTPC;
900 track.fCddTPC = fCddTPC;
901 track.fCdzTPC = fCdzTPC;
902 track.fCzzTPC = fCzzTPC;
436dfe39 903 track.fCchi2TPC = fCchi2TPC;
b9ca886f 904
905
906 // copy all other TPC specific parameters
907
908 // replace label by TPC label
909 track.fLabel = fTPCLabel;
910 track.fTPCLabel = fTPCLabel;
911
912 track.fTPCchi2 = fTPCchi2;
913 track.fTPCsignal = fTPCsignal;
914 track.fTPCsignalS = fTPCsignalS;
915 for(int i = 0;i<4;++i)track.fTPCPoints[i] = fTPCPoints[i];
916
917 track.fTPCncls = fTPCncls;
918 track.fTPCnclsF = fTPCnclsF;
919 track.fTPCsignalN = fTPCsignalN;
920
921 // PID
922 for(int i=0;i<AliPID::kSPECIES;++i){
923 track.fTPCr[i] = fTPCr[i];
924 // combined PID is TPC only!
925 track.fR[i] = fTPCr[i];
926 }
927 track.fTPCClusterMap = fTPCClusterMap;
928 track.fTPCSharedMap = fTPCSharedMap;
929
930
931 // reset the flags
932 track.fFlags = kTPCin;
933 track.fID = fID;
934
b1cfce51 935 track.fFlags |= fFlags & kTPCpid; //copy the TPCpid status flag
b9ca886f 936
937 for (Int_t i=0;i<3;i++) track.fKinkIndexes[i] = fKinkIndexes[i];
938
939 return kTRUE;
940
941}
942
9559cbc4 943//_______________________________________________________________________
944void AliESDtrack::MakeMiniESDtrack(){
945 // Resets everything except
946 // fFlags: Reconstruction status flags
947 // fLabel: Track label
948 // fID: Unique ID of the track
d7ddf1e9 949 // Impact parameter information
9559cbc4 950 // fR[AliPID::kSPECIES]: combined "detector response probability"
8497bca0 951 // Running track parameters in the base class (AliExternalTrackParam)
9559cbc4 952
953 fTrackLength = 0;
562dd0b4 954
9559cbc4 955 for (Int_t i=0;i<AliPID::kSPECIES;i++) fTrackTime[i] = 0;
9559cbc4 956
957 // Reset track parameters constrained to the primary vertex
562dd0b4 958 delete fCp;fCp = 0;
9559cbc4 959
960 // Reset track parameters at the inner wall of TPC
562dd0b4 961 delete fIp;fIp = 0;
962 delete fTPCInner;fTPCInner=0;
9559cbc4 963 // Reset track parameters at the inner wall of the TRD
562dd0b4 964 delete fOp;fOp = 0;
965
9559cbc4 966
967 // Reset ITS track related information
968 fITSchi2 = 0;
9559cbc4 969 fITSncls = 0;
62665e7f 970 fITSClusterMap=0;
9559cbc4 971 fITSsignal = 0;
ef7253ac 972 for (Int_t i=0;i<AliPID::kSPECIES;i++) fITSr[i]=0;
9559cbc4 973 fITSLabel = 0;
9559cbc4 974
975 // Reset TPC related track information
976 fTPCchi2 = 0;
977 fTPCncls = 0;
e1d4c1b5 978 fTPCnclsF = 0;
9559cbc4 979 fTPCClusterMap = 0;
eb7f6854 980 fTPCSharedMap = 0;
9559cbc4 981 fTPCsignal= 0;
e1d4c1b5 982 fTPCsignalS= 0;
983 fTPCsignalN= 0;
9559cbc4 984 for (Int_t i=0;i<AliPID::kSPECIES;i++) fTPCr[i]=0;
985 fTPCLabel=0;
986 for (Int_t i=0;i<4;i++) fTPCPoints[i] = 0;
987 for (Int_t i=0; i<3;i++) fKinkIndexes[i] = 0;
988 for (Int_t i=0; i<3;i++) fV0Indexes[i] = 0;
989
990 // Reset TRD related track information
991 fTRDchi2 = 0;
992 fTRDncls = 0;
993 fTRDncls0 = 0;
9559cbc4 994 fTRDsignal = 0;
6984f7c1 995 for (Int_t i=0;i<kTRDnPlanes;i++) {
6d45eaef 996 fTRDTimBin[i] = 0;
9559cbc4 997 }
998 for (Int_t i=0;i<AliPID::kSPECIES;i++) fTRDr[i] = 0;
999 fTRDLabel = 0;
9559cbc4 1000 fTRDQuality = 0;
ed15ef4f 1001 fTRDntracklets = 0;
6984f7c1 1002 if(fTRDnSlices)
1003 delete[] fTRDslices;
1004 fTRDslices=0x0;
1005 fTRDnSlices=0;
23d49657 1006 fTRDBudget = 0;
9559cbc4 1007
1008 // Reset TOF related track information
1009 fTOFchi2 = 0;
ce3f4882 1010 fTOFindex = -1;
9559cbc4 1011 fTOFsignal = 0;
562dd0b4 1012 fTOFCalChannel = 0;
85324138 1013 fTOFsignalToT = 0;
d321691a 1014 fTOFsignalRaw = 0;
1015 fTOFsignalDz = 0;
9559cbc4 1016 for (Int_t i=0;i<AliPID::kSPECIES;i++) fTOFr[i] = 0;
1017 for (Int_t i=0;i<3;i++) fTOFLabel[i] = 0;
1018 for (Int_t i=0;i<10;i++) fTOFInfo[i] = 0;
1019
f4b3bbb7 1020 // Reset HMPID related track information
1021 fHMPIDchi2 = 0;
562dd0b4 1022 fHMPIDqn = 0;
81aa7a0d 1023 fHMPIDcluIdx = -1;
f4b3bbb7 1024 fHMPIDsignal = 0;
1025 for (Int_t i=0;i<AliPID::kSPECIES;i++) fHMPIDr[i] = 0;
562dd0b4 1026 fHMPIDtrkTheta = 0;
1027 fHMPIDtrkPhi = 0;
1028 fHMPIDtrkX = 0;
1029 fHMPIDtrkY = 0;
1030 fHMPIDmipX = 0;
1031 fHMPIDmipY = 0;
2e1dcd14 1032 fEMCALindex = kEMCALNoMatch;
9559cbc4 1033
b5b2b4db 1034 // reset global track chi2
1035 fGlobalChi2 = 0;
1036
6dc21f57 1037 fVertexID = -2; // an orphan track
1038
15e85efa 1039 delete fFriendTrack; fFriendTrack = 0;
9559cbc4 1040}
ae982df3 1041//_______________________________________________________________________
4a78b8c5 1042Double_t AliESDtrack::GetMass() const {
4427806c 1043 // Returns the mass of the most probable particle type
ae982df3 1044 Float_t max=0.;
1045 Int_t k=-1;
304864ab 1046 for (Int_t i=0; i<AliPID::kSPECIES; i++) {
ae982df3 1047 if (fR[i]>max) {k=i; max=fR[i];}
1048 }
db3989b3 1049 if (k==0) { // dE/dx "crossing points" in the TPC
1050 Double_t p=GetP();
1051 if ((p>0.38)&&(p<0.48))
304864ab 1052 if (fR[0]<fR[3]*10.) return AliPID::ParticleMass(AliPID::kKaon);
db3989b3 1053 if ((p>0.75)&&(p<0.85))
304864ab 1054 if (fR[0]<fR[4]*10.) return AliPID::ParticleMass(AliPID::kProton);
db3989b3 1055 return 0.00051;
1056 }
304864ab 1057 if (k==1) return AliPID::ParticleMass(AliPID::kMuon);
1058 if (k==2||k==-1) return AliPID::ParticleMass(AliPID::kPion);
1059 if (k==3) return AliPID::ParticleMass(AliPID::kKaon);
1060 if (k==4) return AliPID::ParticleMass(AliPID::kProton);
5f7789fc 1061 AliWarning("Undefined mass !");
304864ab 1062 return AliPID::ParticleMass(AliPID::kPion);
ae982df3 1063}
1064
aad8d435 1065//______________________________________________________________________________
1066Double_t AliESDtrack::E() const
1067{
1068 // Returns the energy of the particle given its assumed mass.
1069 // Assumes the pion mass if the particle can't be identified properly.
1070
1071 Double_t m = M();
1072 Double_t p = P();
1073 return TMath::Sqrt(p*p + m*m);
1074}
1075
1076//______________________________________________________________________________
1077Double_t AliESDtrack::Y() const
1078{
1079 // Returns the rapidity of a particle given its assumed mass.
1080 // Assumes the pion mass if the particle can't be identified properly.
1081
1082 Double_t e = E();
1083 Double_t pz = Pz();
e03e4544 1084 if (e != TMath::Abs(pz)) { // energy was not equal to pz
aad8d435 1085 return 0.5*TMath::Log((e+pz)/(e-pz));
1086 } else { // energy was equal to pz
1087 return -999.;
1088 }
1089}
1090
ae982df3 1091//_______________________________________________________________________
c9ec41e8 1092Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
ae982df3 1093 //
1094 // This function updates track's running parameters
1095 //
15e85efa 1096 Int_t *index=0;
15614b8b 1097 Bool_t rc=kTRUE;
1098
9b859005 1099 SetStatus(flags);
1100 fLabel=t->GetLabel();
1101
1102 if (t->IsStartedTimeIntegral()) {
1103 SetStatus(kTIME);
1104 Double_t times[10];t->GetIntegratedTimes(times); SetIntegratedTimes(times);
1105 SetIntegratedLength(t->GetIntegratedLength());
1106 }
1107
6c94f330 1108 Set(t->GetX(),t->GetAlpha(),t->GetParameter(),t->GetCovariance());
ded25cc6 1109 if (flags==kITSout) fFriendTrack->SetITSOut(*t);
1110 if (flags==kTPCout) fFriendTrack->SetTPCOut(*t);
1111 if (flags==kTRDin || flags==kTRDrefit) fFriendTrack->SetTRDIn(*t);
e1d4c1b5 1112
ae982df3 1113 switch (flags) {
ad2f1f2b 1114
9b859005 1115 case kITSin: case kITSout: case kITSrefit:
48704648 1116 fITSClusterMap=0;
ae982df3 1117 fITSncls=t->GetNumberOfClusters();
62665e7f 1118 index=fFriendTrack->GetITSindices();
1119 for (Int_t i=0;i<AliESDfriendTrack::kMaxITScluster;i++) {
1120 index[i]=t->GetClusterIndex(i);
1121 if (i<fITSncls) {
1122 Int_t l=(index[i] & 0xf0000000) >> 28;
1123 SETBIT(fITSClusterMap,l);
1124 }
1125 }
ae982df3 1126 fITSchi2=t->GetChi2();
ae982df3 1127 fITSsignal=t->GetPIDsignal();
6e5b1b04 1128 fITSLabel = t->GetLabel();
57483eb1 1129 // keep in fOp the parameters outside ITS for ITS stand-alone tracks
1130 if (flags==kITSout) {
1131 if (!fOp) fOp=new AliExternalTrackParam(*t);
1132 else
1133 fOp->Set(t->GetX(),t->GetAlpha(),t->GetParameter(),t->GetCovariance());
ded25cc6 1134 }
ae982df3 1135 break;
ad2f1f2b 1136
9b859005 1137 case kTPCin: case kTPCrefit:
6e5b1b04 1138 fTPCLabel = t->GetLabel();
4aeb9470 1139 if (flags==kTPCin) fTPCInner=new AliExternalTrackParam(*t);
c9ec41e8 1140 if (!fIp) fIp=new AliExternalTrackParam(*t);
6c94f330 1141 else
1142 fIp->Set(t->GetX(),t->GetAlpha(),t->GetParameter(),t->GetCovariance());
9b859005 1143 case kTPCout:
15e85efa 1144 index=fFriendTrack->GetTPCindices();
1d303a24 1145 if (flags & kTPCout){
1146 if (!fOp) fOp=new AliExternalTrackParam(*t);
6c94f330 1147 else
1148 fOp->Set(t->GetX(),t->GetAlpha(),t->GetParameter(),t->GetCovariance());
1d303a24 1149 }
e1d4c1b5 1150 fTPCncls=t->GetNumberOfClusters();
ae982df3 1151 fTPCchi2=t->GetChi2();
a866ac60 1152
1153 {//prevrow must be declared in separate namespace, otherwise compiler cries:
1154 //"jump to case label crosses initialization of `Int_t prevrow'"
1155 Int_t prevrow = -1;
6e5b1b04 1156 // for (Int_t i=0;i<fTPCncls;i++)
15e85efa 1157 for (Int_t i=0;i<AliESDfriendTrack::kMaxTPCcluster;i++)
a866ac60 1158 {
15e85efa 1159 index[i]=t->GetClusterIndex(i);
1160 Int_t idx = index[i];
a866ac60 1161
15e85efa 1162 if (idx<0) continue;
9fe5b2ff 1163
a866ac60 1164 // Piotr's Cluster Map for HBT
1165 // ### please change accordingly if cluster array is changing
1166 // to "New TPC Tracking" style (with gaps in array)
a866ac60 1167 Int_t sect = (idx&0xff000000)>>24;
1168 Int_t row = (idx&0x00ff0000)>>16;
1169 if (sect > 18) row +=63; //if it is outer sector, add number of inner sectors
1170
1171 fTPCClusterMap.SetBitNumber(row,kTRUE);
1172
1173 //Fill the gap between previous row and this row with 0 bits
1174 //In case ### pleas change it as well - just set bit 0 in case there
1175 //is no associated clusters for current "i"
1176 if (prevrow < 0)
1177 {
1178 prevrow = row;//if previous bit was not assigned yet == this is the first one
1179 }
1180 else
1181 { //we don't know the order (inner to outer or reverse)
1182 //just to be save in case it is going to change
1183 Int_t n = 0, m = 0;
1184 if (prevrow < row)
1185 {
1186 n = prevrow;
1187 m = row;
1188 }
1189 else
1190 {
1191 n = row;
1192 m = prevrow;
1193 }
1194
1195 for (Int_t j = n+1; j < m; j++)
1196 {
1197 fTPCClusterMap.SetBitNumber(j,kFALSE);
1198 }
1199 prevrow = row;
1200 }
1201 // End Of Piotr's Cluster Map for HBT
1202 }
1203 }
ae982df3 1204 fTPCsignal=t->GetPIDsignal();
ae982df3 1205 break;
9b859005 1206
64130601 1207 case kTRDin: case kTRDrefit:
1208 break;
1209 case kTRDout:
2f83b7a6 1210 index = fFriendTrack->GetTRDindices();
51ad6848 1211 fTRDLabel = t->GetLabel();
2f83b7a6 1212 fTRDchi2 = t->GetChi2();
1213 fTRDncls = t->GetNumberOfClusters();
5bc3e158 1214 for (Int_t i=0;i<6;i++) index[i]=t->GetTrackletIndex(i);
1215
79e94bf8 1216 fTRDsignal=t->GetPIDsignal();
1217 break;
c4d11b15 1218 case kTRDbackup:
c9ec41e8 1219 if (!fOp) fOp=new AliExternalTrackParam(*t);
6c94f330 1220 else
1221 fOp->Set(t->GetX(),t->GetAlpha(),t->GetParameter(),t->GetCovariance());
c4d11b15 1222 fTRDncls0 = t->GetNumberOfClusters();
1223 break;
1224 case kTOFin:
1225 break;
1226 case kTOFout:
1227 break;
d0862fea 1228 case kTRDStop:
1229 break;
ae982df3 1230 default:
5f7789fc 1231 AliError("Wrong flag !");
ae982df3 1232 return kFALSE;
1233 }
1234
15614b8b 1235 return rc;
ae982df3 1236}
1237
1238//_______________________________________________________________________
1239void AliESDtrack::GetExternalParameters(Double_t &x, Double_t p[5]) const {
1240 //---------------------------------------------------------------------
1241 // This function returns external representation of the track parameters
1242 //---------------------------------------------------------------------
c9ec41e8 1243 x=GetX();
1244 for (Int_t i=0; i<5; i++) p[i]=GetParameter()[i];
15614b8b 1245}
1246
67c3dcbe 1247//_______________________________________________________________________
a866ac60 1248void AliESDtrack::GetExternalCovariance(Double_t cov[15]) const {
67c3dcbe 1249 //---------------------------------------------------------------------
1250 // This function returns external representation of the cov. matrix
1251 //---------------------------------------------------------------------
c9ec41e8 1252 for (Int_t i=0; i<15; i++) cov[i]=AliExternalTrackParam::GetCovariance()[i];
67c3dcbe 1253}
1254
67c3dcbe 1255//_______________________________________________________________________
c0b978f0 1256Bool_t AliESDtrack::GetConstrainedExternalParameters
1257 (Double_t &alpha, Double_t &x, Double_t p[5]) const {
67c3dcbe 1258 //---------------------------------------------------------------------
1259 // This function returns the constrained external track parameters
1260 //---------------------------------------------------------------------
c0b978f0 1261 if (!fCp) return kFALSE;
1262 alpha=fCp->GetAlpha();
c9ec41e8 1263 x=fCp->GetX();
1264 for (Int_t i=0; i<5; i++) p[i]=fCp->GetParameter()[i];
c0b978f0 1265 return kTRUE;
67c3dcbe 1266}
c9ec41e8 1267
67c3dcbe 1268//_______________________________________________________________________
c0b978f0 1269Bool_t
67c3dcbe 1270AliESDtrack::GetConstrainedExternalCovariance(Double_t c[15]) const {
1271 //---------------------------------------------------------------------
1272 // This function returns the constrained external cov. matrix
1273 //---------------------------------------------------------------------
c0b978f0 1274 if (!fCp) return kFALSE;
c9ec41e8 1275 for (Int_t i=0; i<15; i++) c[i]=fCp->GetCovariance()[i];
c0b978f0 1276 return kTRUE;
67c3dcbe 1277}
1278
c0b978f0 1279Bool_t
1280AliESDtrack::GetInnerExternalParameters
1281 (Double_t &alpha, Double_t &x, Double_t p[5]) const {
1282 //---------------------------------------------------------------------
c9ec41e8 1283 // This function returns external representation of the track parameters
1284 // at the inner layer of TPC
9b859005 1285 //---------------------------------------------------------------------
c0b978f0 1286 if (!fIp) return kFALSE;
1287 alpha=fIp->GetAlpha();
c9ec41e8 1288 x=fIp->GetX();
1289 for (Int_t i=0; i<5; i++) p[i]=fIp->GetParameter()[i];
c0b978f0 1290 return kTRUE;
9b859005 1291}
1292
c0b978f0 1293Bool_t
1294AliESDtrack::GetInnerExternalCovariance(Double_t cov[15]) const {
c9ec41e8 1295 //---------------------------------------------------------------------
1296 // This function returns external representation of the cov. matrix
1297 // at the inner layer of TPC
1298 //---------------------------------------------------------------------
c0b978f0 1299 if (!fIp) return kFALSE;
c9ec41e8 1300 for (Int_t i=0; i<15; i++) cov[i]=fIp->GetCovariance()[i];
c0b978f0 1301 return kTRUE;
9b859005 1302}
1303
d61ca12d 1304void
1305AliESDtrack::SetOuterParam(const AliExternalTrackParam *p, ULong_t flags) {
1306 //
1307 // This is a direct setter for the outer track parameters
1308 //
1309 SetStatus(flags);
1310 if (fOp) delete fOp;
1311 fOp=new AliExternalTrackParam(*p);
1312}
1313
c0b978f0 1314Bool_t
1315AliESDtrack::GetOuterExternalParameters
1316 (Double_t &alpha, Double_t &x, Double_t p[5]) const {
1317 //---------------------------------------------------------------------
c9ec41e8 1318 // This function returns external representation of the track parameters
1319 // at the inner layer of TRD
a866ac60 1320 //---------------------------------------------------------------------
c0b978f0 1321 if (!fOp) return kFALSE;
1322 alpha=fOp->GetAlpha();
c9ec41e8 1323 x=fOp->GetX();
1324 for (Int_t i=0; i<5; i++) p[i]=fOp->GetParameter()[i];
c0b978f0 1325 return kTRUE;
a866ac60 1326}
c9ec41e8 1327
c0b978f0 1328Bool_t
1329AliESDtrack::GetOuterExternalCovariance(Double_t cov[15]) const {
a866ac60 1330 //---------------------------------------------------------------------
c9ec41e8 1331 // This function returns external representation of the cov. matrix
1332 // at the inner layer of TRD
a866ac60 1333 //---------------------------------------------------------------------
c0b978f0 1334 if (!fOp) return kFALSE;
c9ec41e8 1335 for (Int_t i=0; i<15; i++) cov[i]=fOp->GetCovariance()[i];
c0b978f0 1336 return kTRUE;
a866ac60 1337}
1338
98937d93 1339Int_t AliESDtrack::GetNcls(Int_t idet) const
1340{
1341 // Get number of clusters by subdetector index
1342 //
1343 Int_t ncls = 0;
1344 switch(idet){
1345 case 0:
1346 ncls = fITSncls;
1347 break;
1348 case 1:
1349 ncls = fTPCncls;
1350 break;
1351 case 2:
1352 ncls = fTRDncls;
1353 break;
1354 case 3:
ce3f4882 1355 if (fTOFindex != -1)
98937d93 1356 ncls = 1;
1357 break;
81aa7a0d 1358 case 4: //PHOS
1359 break;
1360 case 5: //HMPID
1361 if ((fHMPIDcluIdx >= 0) && (fHMPIDcluIdx < 7000000)) {
1362 if ((fHMPIDcluIdx%1000000 != 9999) && (fHMPIDcluIdx%1000000 != 99999)) {
1363 ncls = 1;
1364 }
1365 }
1366 break;
98937d93 1367 default:
1368 break;
1369 }
1370 return ncls;
1371}
1372
ef7253ac 1373Int_t AliESDtrack::GetClusters(Int_t idet, Int_t *idx) const
98937d93 1374{
1375 // Get cluster index array by subdetector index
1376 //
1377 Int_t ncls = 0;
1378 switch(idet){
1379 case 0:
1380 ncls = GetITSclusters(idx);
1381 break;
1382 case 1:
ef7253ac 1383 ncls = GetTPCclusters(idx);
98937d93 1384 break;
1385 case 2:
1386 ncls = GetTRDclusters(idx);
1387 break;
1388 case 3:
ce3f4882 1389 if (fTOFindex != -1) {
1390 idx[0] = fTOFindex;
98937d93 1391 ncls = 1;
1392 }
1393 break;
313af949 1394 case 4: //PHOS
1395 break;
1396 case 5:
81aa7a0d 1397 if ((fHMPIDcluIdx >= 0) && (fHMPIDcluIdx < 7000000)) {
1398 if ((fHMPIDcluIdx%1000000 != 9999) && (fHMPIDcluIdx%1000000 != 99999)) {
1399 idx[0] = GetHMPIDcluIdx();
1400 ncls = 1;
1401 }
313af949 1402 }
1403 break;
1404 case 6: //EMCAL
1405 break;
98937d93 1406 default:
1407 break;
1408 }
1409 return ncls;
1410}
1411
ae982df3 1412//_______________________________________________________________________
1413void AliESDtrack::GetIntegratedTimes(Double_t *times) const {
4427806c 1414 // Returns the array with integrated times for each particle hypothesis
304864ab 1415 for (Int_t i=0; i<AliPID::kSPECIES; i++) times[i]=fTrackTime[i];
ae982df3 1416}
1417
1418//_______________________________________________________________________
1419void AliESDtrack::SetIntegratedTimes(const Double_t *times) {
4427806c 1420 // Sets the array with integrated times for each particle hypotesis
304864ab 1421 for (Int_t i=0; i<AliPID::kSPECIES; i++) fTrackTime[i]=times[i];
ae982df3 1422}
1423
c630aafd 1424//_______________________________________________________________________
4427806c 1425void AliESDtrack::SetITSpid(const Double_t *p) {
1426 // Sets values for the probability of each particle type (in ITS)
d27bbc79 1427 SetPIDValues(fITSr,p,AliPID::kSPECIES);
c630aafd 1428 SetStatus(AliESDtrack::kITSpid);
1429}
1430
1431//_______________________________________________________________________
1432void AliESDtrack::GetITSpid(Double_t *p) const {
4427806c 1433 // Gets the probability of each particle type (in ITS)
304864ab 1434 for (Int_t i=0; i<AliPID::kSPECIES; i++) p[i]=fITSr[i];
c630aafd 1435}
1436
ae982df3 1437//_______________________________________________________________________
562dd0b4 1438Char_t AliESDtrack::GetITSclusters(Int_t *idx) const {
ae982df3 1439 //---------------------------------------------------------------------
1440 // This function returns indices of the assgined ITS clusters
1441 //---------------------------------------------------------------------
15e85efa 1442 if (idx!=0) {
1443 Int_t *index=fFriendTrack->GetITSindices();
dbfebd7d 1444 for (Int_t i=0; i<AliESDfriendTrack::kMaxITScluster; i++) {
1445 if ( (i>=fITSncls) && (i<6) ) idx[i]=-1;
1446 else idx[i]=index[i];
1447 }
15e85efa 1448 }
ae982df3 1449 return fITSncls;
1450}
1451
89f1b176 1452//_______________________________________________________________________
1453Bool_t AliESDtrack::GetITSModuleIndexInfo(Int_t ilayer,Int_t &idet,Int_t &status,
1454 Float_t &xloc,Float_t &zloc) const {
1455 //----------------------------------------------------------------------
1456 // This function encodes in the module number also the status of cluster association
1457 // "status" can have the following values:
1458 // 1 "found" (cluster is associated),
1459 // 2 "dead" (module is dead from OCDB),
1460 // 3 "skipped" (module or layer forced to be skipped),
1461 // 4 "outinz" (track out of z acceptance),
1462 // 5 "nocls" (no clusters in the road),
1463 // 6 "norefit" (cluster rejected during refit),
1464 // 7 "deadzspd" (holes in z in SPD)
1465 // Also given are the coordinates of the crossing point of track and module
1466 // (in the local module ref. system)
1467 // WARNING: THIS METHOD HAS TO BE SYNCHRONIZED WITH AliITStrackV2::GetModuleIndexInfo()!
1468 //----------------------------------------------------------------------
1469
1470 if(fITSModule[ilayer]==-1) {
89f1b176 1471 idet = -1;
1472 status=0;
1473 xloc=-99.; zloc=-99.;
1474 return kFALSE;
1475 }
1476
1477 Int_t module = fITSModule[ilayer];
1478
1479 idet = Int_t(module/1000000);
1480
1481 module -= idet*1000000;
1482
1483 status = Int_t(module/100000);
1484
1485 module -= status*100000;
1486
1487 Int_t signs = Int_t(module/10000);
1488
1489 module-=signs*10000;
1490
1491 Int_t xInt = Int_t(module/100);
1492 module -= xInt*100;
1493
1494 Int_t zInt = module;
1495
1496 if(signs==1) { xInt*=1; zInt*=1; }
1497 if(signs==2) { xInt*=1; zInt*=-1; }
1498 if(signs==3) { xInt*=-1; zInt*=1; }
1499 if(signs==4) { xInt*=-1; zInt*=-1; }
1500
1501 xloc = 0.1*(Float_t)xInt;
1502 zloc = 0.1*(Float_t)zInt;
1503
1504 if(status==4) idet = -1;
1505
1506 return kTRUE;
1507}
1508
ae982df3 1509//_______________________________________________________________________
562dd0b4 1510UShort_t AliESDtrack::GetTPCclusters(Int_t *idx) const {
ae982df3 1511 //---------------------------------------------------------------------
1512 // This function returns indices of the assgined ITS clusters
1513 //---------------------------------------------------------------------
15e85efa 1514 if (idx!=0) {
1515 Int_t *index=fFriendTrack->GetTPCindices();
1516 for (Int_t i=0; i<AliESDfriendTrack::kMaxTPCcluster; i++) idx[i]=index[i];
1517 }
ae982df3 1518 return fTPCncls;
1519}
8c6a71ab 1520
562dd0b4 1521Double_t AliESDtrack::GetTPCdensity(Int_t row0, Int_t row1) const{
81e97e0d 1522 //
1523 // GetDensity of the clusters on given region between row0 and row1
1524 // Dead zone effect takin into acoount
1525 //
1526 Int_t good = 0;
1527 Int_t found = 0;
1528 //
15e85efa 1529 Int_t *index=fFriendTrack->GetTPCindices();
81e97e0d 1530 for (Int_t i=row0;i<=row1;i++){
15e85efa 1531 Int_t idx = index[i];
1532 if (idx!=-1) good++; // track outside of dead zone
1533 if (idx>0) found++;
81e97e0d 1534 }
1535 Float_t density=0.5;
1536 if (good>(row1-row0)*0.5) density = Float_t(found)/Float_t(good);
1537 return density;
1538}
c84a5e9e 1539
8c6a71ab 1540//_______________________________________________________________________
1541void AliESDtrack::SetTPCpid(const Double_t *p) {
4427806c 1542 // Sets values for the probability of each particle type (in TPC)
d27bbc79 1543 SetPIDValues(fTPCr,p,AliPID::kSPECIES);
8c6a71ab 1544 SetStatus(AliESDtrack::kTPCpid);
1545}
1546
1547//_______________________________________________________________________
1548void AliESDtrack::GetTPCpid(Double_t *p) const {
4427806c 1549 // Gets the probability of each particle type (in TPC)
304864ab 1550 for (Int_t i=0; i<AliPID::kSPECIES; i++) p[i]=fTPCr[i];
8c6a71ab 1551}
1552
bb2ceb1f 1553//_______________________________________________________________________
562dd0b4 1554UChar_t AliESDtrack::GetTRDclusters(Int_t *idx) const {
bb2ceb1f 1555 //---------------------------------------------------------------------
1556 // This function returns indices of the assgined TRD clusters
1557 //---------------------------------------------------------------------
15e85efa 1558 if (idx!=0) {
1559 Int_t *index=fFriendTrack->GetTRDindices();
1560 for (Int_t i=0; i<AliESDfriendTrack::kMaxTRDcluster; i++) idx[i]=index[i];
1561 }
bb2ceb1f 1562 return fTRDncls;
1563}
1564
5bc3e158 1565//_______________________________________________________________________
1566UChar_t AliESDtrack::GetTRDtracklets(Int_t *idx) const {
1567 //---------------------------------------------------------------------
1568 // This function returns indices of the assigned TRD tracklets
1569 //---------------------------------------------------------------------
1570 if (idx!=0) {
1571 Int_t *index=fFriendTrack->GetTRDindices();
1572 for (Int_t i=0; i<6/*AliESDfriendTrack::kMaxTRDcluster*/; i++) idx[i]=index[i];
1573 }
1574 return fTRDncls;
1575}
1576
c630aafd 1577//_______________________________________________________________________
1578void AliESDtrack::SetTRDpid(const Double_t *p) {
4427806c 1579 // Sets values for the probability of each particle type (in TRD)
d27bbc79 1580 SetPIDValues(fTRDr,p,AliPID::kSPECIES);
c630aafd 1581 SetStatus(AliESDtrack::kTRDpid);
1582}
1583
1584//_______________________________________________________________________
1585void AliESDtrack::GetTRDpid(Double_t *p) const {
4427806c 1586 // Gets the probability of each particle type (in TRD)
304864ab 1587 for (Int_t i=0; i<AliPID::kSPECIES; i++) p[i]=fTRDr[i];
c630aafd 1588}
1589
79e94bf8 1590//_______________________________________________________________________
1591void AliESDtrack::SetTRDpid(Int_t iSpecies, Float_t p)
1592{
4427806c 1593 // Sets the probability of particle type iSpecies to p (in TRD)
79e94bf8 1594 fTRDr[iSpecies] = p;
1595}
1596
562dd0b4 1597Double_t AliESDtrack::GetTRDpid(Int_t iSpecies) const
79e94bf8 1598{
4427806c 1599 // Returns the probability of particle type iSpecies (in TRD)
79e94bf8 1600 return fTRDr[iSpecies];
1601}
1602
6984f7c1 1603void AliESDtrack::SetNumberOfTRDslices(Int_t n) {
1604 //Sets the number of slices used for PID
1605 if (fTRDnSlices != 0) return;
1606 fTRDnSlices=kTRDnPlanes*n;
1607 fTRDslices=new Double32_t[fTRDnSlices];
1608 for (Int_t i=0; i<fTRDnSlices; i++) fTRDslices[i]=-1.;
1609}
1610
1611void AliESDtrack::SetTRDslice(Double_t q, Int_t plane, Int_t slice) {
1612 //Sets the charge q in the slice of the plane
1613 Int_t ns=GetNumberOfTRDslices();
1614 if (ns==0) {
1615 AliError("No TRD slices allocated for this track !");
1616 return;
1617 }
1618
1619 if ((plane<0) || (plane>=kTRDnPlanes)) {
1620 AliError("Wrong TRD plane !");
1621 return;
1622 }
1623 if ((slice<0) || (slice>=ns)) {
1624 AliError("Wrong TRD slice !");
1625 return;
1626 }
1627 Int_t n=plane*ns + slice;
1628 fTRDslices[n]=q;
1629}
1630
1631Double_t AliESDtrack::GetTRDslice(Int_t plane, Int_t slice) const {
1632 //Gets the charge from the slice of the plane
1633 Int_t ns=GetNumberOfTRDslices();
1634 if (ns==0) {
1635 //AliError("No TRD slices allocated for this track !");
1636 return -1.;
1637 }
1638
1639 if ((plane<0) || (plane>=kTRDnPlanes)) {
1640 AliError("Wrong TRD plane !");
1641 return -1.;
1642 }
1643 if ((slice<-1) || (slice>=ns)) {
1644 //AliError("Wrong TRD slice !");
1645 return -1.;
1646 }
1647
1648 if (slice==-1) {
1649 Double_t q=0.;
1650 for (Int_t i=0; i<ns; i++) q+=fTRDslices[plane*ns + i];
1651 return q/ns;
1652 }
1653
1654 return fTRDslices[plane*ns + slice];
1655}
1656
1657
c630aafd 1658//_______________________________________________________________________
1659void AliESDtrack::SetTOFpid(const Double_t *p) {
4427806c 1660 // Sets the probability of each particle type (in TOF)
d27bbc79 1661 SetPIDValues(fTOFr,p,AliPID::kSPECIES);
c630aafd 1662 SetStatus(AliESDtrack::kTOFpid);
1663}
1664
51ad6848 1665//_______________________________________________________________________
1666void AliESDtrack::SetTOFLabel(const Int_t *p) {
1667 // Sets (in TOF)
1668 for (Int_t i=0; i<3; i++) fTOFLabel[i]=p[i];
1669}
1670
c630aafd 1671//_______________________________________________________________________
1672void AliESDtrack::GetTOFpid(Double_t *p) const {
4427806c 1673 // Gets probabilities of each particle type (in TOF)
304864ab 1674 for (Int_t i=0; i<AliPID::kSPECIES; i++) p[i]=fTOFr[i];
c630aafd 1675}
1676
51ad6848 1677//_______________________________________________________________________
1678void AliESDtrack::GetTOFLabel(Int_t *p) const {
1679 // Gets (in TOF)
1680 for (Int_t i=0; i<3; i++) p[i]=fTOFLabel[i];
1681}
1682
1683//_______________________________________________________________________
1684void AliESDtrack::GetTOFInfo(Float_t *info) const {
1685 // Gets (in TOF)
1686 for (Int_t i=0; i<10; i++) info[i]=fTOFInfo[i];
1687}
1688
1689//_______________________________________________________________________
1690void AliESDtrack::SetTOFInfo(Float_t*info) {
1691 // Gets (in TOF)
1692 for (Int_t i=0; i<10; i++) fTOFInfo[i]=info[i];
1693}
1694
4a78b8c5 1695
1696
4a78b8c5 1697//_______________________________________________________________________
f4b3bbb7 1698void AliESDtrack::SetHMPIDpid(const Double_t *p) {
1699 // Sets the probability of each particle type (in HMPID)
1700 SetPIDValues(fHMPIDr,p,AliPID::kSPECIES);
1701 SetStatus(AliESDtrack::kHMPIDpid);
4a78b8c5 1702}
1703
1704//_______________________________________________________________________
f4b3bbb7 1705void AliESDtrack::GetHMPIDpid(Double_t *p) const {
1706 // Gets probabilities of each particle type (in HMPID)
1707 for (Int_t i=0; i<AliPID::kSPECIES; i++) p[i]=fHMPIDr[i];
4a78b8c5 1708}
1709
1710
1711
8c6a71ab 1712//_______________________________________________________________________
1713void AliESDtrack::SetESDpid(const Double_t *p) {
4427806c 1714 // Sets the probability of each particle type for the ESD track
d27bbc79 1715 SetPIDValues(fR,p,AliPID::kSPECIES);
8c6a71ab 1716 SetStatus(AliESDtrack::kESDpid);
1717}
1718
1719//_______________________________________________________________________
1720void AliESDtrack::GetESDpid(Double_t *p) const {
4427806c 1721 // Gets probability of each particle type for the ESD track
304864ab 1722 for (Int_t i=0; i<AliPID::kSPECIES; i++) p[i]=fR[i];
8c6a71ab 1723}
1724
d7ddf1e9 1725//_______________________________________________________________________
436dfe39 1726Bool_t AliESDtrack::RelateToVertexTPC(const AliESDVertex *vtx,
1727Double_t b, Double_t maxd, AliExternalTrackParam *cParam) {
d7ddf1e9 1728 //
436dfe39 1729 // Try to relate the TPC-only track parameters to the vertex "vtx",
d7ddf1e9 1730 // if the (rough) transverse impact parameter is not bigger then "maxd".
1731 // Magnetic field is "b" (kG).
1732 //
1733 // a) The TPC-only paramters are extapolated to the DCA to the vertex.
1734 // b) The impact parameters and their covariance matrix are calculated.
436dfe39 1735 // c) An attempt to constrain the TPC-only params to the vertex is done.
1736 // The constrained params are returned via "cParam".
d7ddf1e9 1737 //
436dfe39 1738 // In the case of success, the returned value is kTRUE
1739 // otherwise, it's kFALSE)
1740 //
d7ddf1e9 1741
1742 if (!fTPCInner) return kFALSE;
1743 if (!vtx) return kFALSE;
1744
1745 Double_t dz[2],cov[3];
1746 if (!fTPCInner->PropagateToDCA(vtx, b, maxd, dz, cov)) return kFALSE;
1747
1748 fdTPC = dz[0];
1749 fzTPC = dz[1];
1750 fCddTPC = cov[0];
1751 fCdzTPC = cov[1];
1752 fCzzTPC = cov[2];
1753
436dfe39 1754 Double_t covar[6]; vtx->GetCovMatrix(covar);
1755 Double_t p[2]={GetParameter()[0]-dz[0],GetParameter()[1]-dz[1]};
1756 Double_t c[3]={covar[2],0.,covar[5]};
1757
1758 Double_t chi2=GetPredictedChi2(p,c);
1759 if (chi2>kVeryBig) return kFALSE;
1760
1761 fCchi2TPC=chi2;
1762
1763 if (!cParam) return kTRUE;
1764
1765 *cParam = *fTPCInner;
1766 if (!cParam->Update(p,c)) return kFALSE;
1767
d7ddf1e9 1768 return kTRUE;
1769}
1770
49d13e89 1771//_______________________________________________________________________
436dfe39 1772Bool_t AliESDtrack::RelateToVertex(const AliESDVertex *vtx,
1773Double_t b, Double_t maxd, AliExternalTrackParam *cParam) {
49d13e89 1774 //
1775 // Try to relate this track to the vertex "vtx",
1776 // if the (rough) transverse impact parameter is not bigger then "maxd".
1777 // Magnetic field is "b" (kG).
1778 //
1779 // a) The track gets extapolated to the DCA to the vertex.
1780 // b) The impact parameters and their covariance matrix are calculated.
1781 // c) An attempt to constrain this track to the vertex is done.
436dfe39 1782 // The constrained params are returned via "cParam".
49d13e89 1783 //
436dfe39 1784 // In the case of success, the returned value is kTRUE
1785 // (otherwise, it's kFALSE)
49d13e89 1786 //
b5d34a4c 1787
1788 if (!vtx) return kFALSE;
1789
e99a34df 1790 Double_t dz[2],cov[3];
1791 if (!PropagateToDCA(vtx, b, maxd, dz, cov)) return kFALSE;
1792
1793 fD = dz[0];
1794 fZ = dz[1];
1795 fCdd = cov[0];
1796 fCdz = cov[1];
1797 fCzz = cov[2];
49d13e89 1798
e99a34df 1799 Double_t covar[6]; vtx->GetCovMatrix(covar);
1800 Double_t p[2]={GetParameter()[0]-dz[0],GetParameter()[1]-dz[1]};
1801 Double_t c[3]={covar[2],0.,covar[5]};
3231f9e5 1802
e99a34df 1803 Double_t chi2=GetPredictedChi2(p,c);
436dfe39 1804 if (chi2>kVeryBig) return kFALSE;
1805
1806 fCchi2=chi2;
49d13e89 1807
436dfe39 1808
1809 //--- Could now these lines be removed ? ---
e99a34df 1810 delete fCp;
1811 fCp=new AliExternalTrackParam(*this);
49d13e89 1812
e99a34df 1813 if (!fCp->Update(p,c)) {delete fCp; fCp=0; return kFALSE;}
436dfe39 1814 //----------------------------------------
1815
6dc21f57 1816 fVertexID = vtx->GetID();
436dfe39 1817
1818 if (!cParam) return kTRUE;
1819
1820 *cParam = *this;
1821 if (!cParam->Update(p,c)) return kFALSE;
1822
49d13e89 1823 return kTRUE;
1824}
1825
ac2f7574 1826//_______________________________________________________________________
1827void AliESDtrack::Print(Option_t *) const {
1828 // Prints info on the track
b9ca886f 1829 AliExternalTrackParam::Print();
5f7789fc 1830 printf("ESD track info\n") ;
304864ab 1831 Double_t p[AliPID::kSPECIESN] ;
ac2f7574 1832 Int_t index = 0 ;
1833 if( IsOn(kITSpid) ){
1834 printf("From ITS: ") ;
1835 GetITSpid(p) ;
304864ab 1836 for(index = 0 ; index < AliPID::kSPECIES; index++)
ac2f7574 1837 printf("%f, ", p[index]) ;
1838 printf("\n signal = %f\n", GetITSsignal()) ;
1839 }
1840 if( IsOn(kTPCpid) ){
1841 printf("From TPC: ") ;
1842 GetTPCpid(p) ;
304864ab 1843 for(index = 0 ; index < AliPID::kSPECIES; index++)
ac2f7574 1844 printf("%f, ", p[index]) ;
1845 printf("\n signal = %f\n", GetTPCsignal()) ;
1846 }
1847 if( IsOn(kTRDpid) ){
1848 printf("From TRD: ") ;
1849 GetTRDpid(p) ;
304864ab 1850 for(index = 0 ; index < AliPID::kSPECIES; index++)
ac2f7574 1851 printf("%f, ", p[index]) ;
6984f7c1 1852 printf("\n signal = %f\n", GetTRDsignal()) ;
ac2f7574 1853 }
1854 if( IsOn(kTOFpid) ){
1855 printf("From TOF: ") ;
1856 GetTOFpid(p) ;
304864ab 1857 for(index = 0 ; index < AliPID::kSPECIES; index++)
ac2f7574 1858 printf("%f, ", p[index]) ;
1859 printf("\n signal = %f\n", GetTOFsignal()) ;
1860 }
f4b3bbb7 1861 if( IsOn(kHMPIDpid) ){
1862 printf("From HMPID: ") ;
1863 GetHMPIDpid(p) ;
304864ab 1864 for(index = 0 ; index < AliPID::kSPECIES; index++)
ac2f7574 1865 printf("%f, ", p[index]) ;
f4b3bbb7 1866 printf("\n signal = %f\n", GetHMPIDsignal()) ;
ac2f7574 1867 }
ac2f7574 1868}
6c94f330 1869
0c19adf7 1870
1871//
1872// Draw functionality
1873// Origin: Marian Ivanov, Marian.Ivanov@cern.ch
1874//
1875void AliESDtrack::FillPolymarker(TPolyMarker3D *pol, Float_t magF, Float_t minR, Float_t maxR, Float_t stepR){
1876 //
1877 // Fill points in the polymarker
1878 //
1879 TObjArray arrayRef;
1880 arrayRef.AddLast(new AliExternalTrackParam(*this));
1881 if (fIp) arrayRef.AddLast(new AliExternalTrackParam(*fIp));
1882 if (fOp) arrayRef.AddLast(new AliExternalTrackParam(*fOp));
1883 //
1884 Double_t mpos[3]={0,0,0};
1885 Int_t entries=arrayRef.GetEntries();
1886 for (Int_t i=0;i<entries;i++){
1887 Double_t pos[3];
1888 ((AliExternalTrackParam*)arrayRef.At(i))->GetXYZ(pos);
1889 mpos[0]+=pos[0]/entries;
1890 mpos[1]+=pos[1]/entries;
1891 mpos[2]+=pos[2]/entries;
1892 }
1893 // Rotate to the mean position
1894 //
1895 Float_t fi= TMath::ATan2(mpos[1],mpos[0]);
1896 for (Int_t i=0;i<entries;i++){
1897 Bool_t res = ((AliExternalTrackParam*)arrayRef.At(i))->Rotate(fi);
1898 if (!res) delete arrayRef.RemoveAt(i);
1899 }
1900 Int_t counter=0;
1901 for (Double_t r=minR; r<maxR; r+=stepR){
1902 Double_t sweight=0;
1903 Double_t mlpos[3]={0,0,0};
1904 for (Int_t i=0;i<entries;i++){
1905 Double_t point[3]={0,0,0};
1906 AliExternalTrackParam *param = ((AliExternalTrackParam*)arrayRef.At(i));
1907 if (!param) continue;
1908 if (param->GetXYZAt(r,magF,point)){
1909 Double_t weight = 1./(10.+(r-param->GetX())*(r-param->GetX()));
1910 sweight+=weight;
1911 mlpos[0]+=point[0]*weight;
1912 mlpos[1]+=point[1]*weight;
1913 mlpos[2]+=point[2]*weight;
1914 }
1915 }
1916 if (sweight>0){
1917 mlpos[0]/=sweight;
1918 mlpos[1]/=sweight;
1919 mlpos[2]/=sweight;
1920 pol->SetPoint(counter,mlpos[0],mlpos[1], mlpos[2]);
1921 printf("xyz\t%f\t%f\t%f\n",mlpos[0], mlpos[1],mlpos[2]);
1922 counter++;
1923 }
1924 }
1925}