Rename AliVirtualParticle to AliVParticle. Add better evaluation of muons tracks...
[u/mrichter/AliRoot.git] / STEER / AliAODTrack.cxx
CommitLineData
df9db588 1/**************************************************************************
2 * Copyright(c) 1998-2007, 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
18//-------------------------------------------------------------------------
19// AOD track base class
20// Author: Markus Oldenburg, CERN
21//-------------------------------------------------------------------------
22
23#include "AliAODTrack.h"
24
9861edc0 25//#include <TPDGCode.h>
26//#include <TDatabasePDG.h>
27
df9db588 28ClassImp(AliAODTrack)
29
30//______________________________________________________________________________
31AliAODTrack::AliAODTrack() :
9861edc0 32 AliVParticle(),
1912763f 33 fChi2perNDF(-999.),
df9db588 34 fID(-999),
35 fLabel(-999),
36 fCovMatrix(NULL),
37 fProdVertex(0x0),
14b34be5 38 fCharge(-99),
1912763f 39 fITSMuonClusterMap(0),
e1c744ca 40 fType(kUndef),
ec40c484 41 fChi2MatchTrigger(0.),
42 fFilterMap(0)
df9db588 43{
44 // default constructor
45
46 SetP();
47 SetPosition((Float_t*)NULL);
48 SetPID((Float_t*)NULL);
49}
50
51//______________________________________________________________________________
52AliAODTrack::AliAODTrack(Int_t id,
53 Int_t label,
54 Double_t p[3],
55 Bool_t cartesian,
56 Double_t x[3],
57 Bool_t isDCA,
58 Double_t covMatrix[21],
59 Short_t charge,
60 UChar_t itsClusMap,
61 Double_t pid[10],
62 AliAODVertex *prodVertex,
1912763f 63 Bool_t usedForVtxFit,
dc825b15 64 Bool_t usedForPrimVtxFit,
ec40c484 65 AODTrk_t ttype,
66 UInt_t selectInfo) :
9861edc0 67 AliVParticle(),
1912763f 68 fChi2perNDF(-999.),
df9db588 69 fID(id),
70 fLabel(label),
71 fCovMatrix(NULL),
72 fProdVertex(prodVertex),
73 fCharge(charge),
1912763f 74 fITSMuonClusterMap(itsClusMap),
e1c744ca 75 fType(ttype),
ec40c484 76 fChi2MatchTrigger(0.),
77 fFilterMap(selectInfo)
df9db588 78{
79 // constructor
80
81 SetP(p, cartesian);
82 SetPosition(x, isDCA);
1912763f 83 SetUsedForVtxFit(usedForVtxFit);
dc825b15 84 SetUsedForPrimVtxFit(usedForPrimVtxFit);
df9db588 85 if(covMatrix) SetCovMatrix(covMatrix);
86 SetPID(pid);
87
88}
89
90//______________________________________________________________________________
91AliAODTrack::AliAODTrack(Int_t id,
92 Int_t label,
93 Float_t p[3],
94 Bool_t cartesian,
95 Float_t x[3],
96 Bool_t isDCA,
97 Float_t covMatrix[21],
98 Short_t charge,
99 UChar_t itsClusMap,
100 Float_t pid[10],
101 AliAODVertex *prodVertex,
1912763f 102 Bool_t usedForVtxFit,
dc825b15 103 Bool_t usedForPrimVtxFit,
ec40c484 104 AODTrk_t ttype,
105 UInt_t selectInfo) :
9861edc0 106 AliVParticle(),
1912763f 107 fChi2perNDF(-999.),
df9db588 108 fID(id),
109 fLabel(label),
110 fCovMatrix(NULL),
111 fProdVertex(prodVertex),
112 fCharge(charge),
1912763f 113 fITSMuonClusterMap(itsClusMap),
e1c744ca 114 fType(ttype),
ec40c484 115 fChi2MatchTrigger(0.),
116 fFilterMap(selectInfo)
df9db588 117{
118 // constructor
119
120 SetP(p, cartesian);
121 SetPosition(x, isDCA);
1912763f 122 SetUsedForVtxFit(usedForVtxFit);
dc825b15 123 SetUsedForPrimVtxFit(usedForPrimVtxFit);
df9db588 124 if(covMatrix) SetCovMatrix(covMatrix);
125 SetPID(pid);
df9db588 126}
127
df9db588 128//______________________________________________________________________________
129AliAODTrack::~AliAODTrack()
130{
131 // destructor
132 delete fCovMatrix;
133}
134
135
136//______________________________________________________________________________
137AliAODTrack::AliAODTrack(const AliAODTrack& trk) :
9861edc0 138 AliVParticle(trk),
1912763f 139 fChi2perNDF(trk.fChi2perNDF),
df9db588 140 fID(trk.fID),
141 fLabel(trk.fLabel),
142 fCovMatrix(NULL),
143 fProdVertex(trk.fProdVertex),
144 fCharge(trk.fCharge),
1912763f 145 fITSMuonClusterMap(trk.fITSMuonClusterMap),
e1c744ca 146 fType(trk.fType),
ec40c484 147 fChi2MatchTrigger(0.),
148 fFilterMap(trk.fFilterMap)
df9db588 149{
150 // Copy constructor
151
152 trk.GetP(fMomentum);
153 trk.GetPosition(fPosition);
1912763f 154 SetUsedForVtxFit(trk.GetUsedForVtxFit());
dc825b15 155 SetUsedForPrimVtxFit(trk.GetUsedForPrimVtxFit());
5d62ce04 156 if(trk.fCovMatrix) fCovMatrix=new AliAODRedCov<6>(*trk.fCovMatrix);
df9db588 157 SetPID(trk.fPID);
158
159}
160
161//______________________________________________________________________________
162AliAODTrack& AliAODTrack::operator=(const AliAODTrack& trk)
163{
164 // Assignment operator
165 if(this!=&trk) {
166
9861edc0 167 AliVParticle::operator=(trk);
df9db588 168
169 trk.GetP(fMomentum);
170 trk.GetPosition(fPosition);
171 trk.GetPID(fPID);
172
1912763f 173 fChi2perNDF = trk.fChi2perNDF;
df9db588 174
175 fID = trk.fID;
176 fLabel = trk.fLabel;
177
178 delete fCovMatrix;
5d62ce04 179 if(trk.fCovMatrix) fCovMatrix=new AliAODRedCov<6>(*trk.fCovMatrix);
df9db588 180 else fCovMatrix=NULL;
181 fProdVertex = trk.fProdVertex;
182
183 fCharge = trk.fCharge;
1912763f 184 fITSMuonClusterMap = trk.fITSMuonClusterMap;
185 SetUsedForVtxFit(trk.GetUsedForVtxFit());
dc825b15 186 SetUsedForPrimVtxFit(trk.GetUsedForPrimVtxFit());
df9db588 187 fType = trk.fType;
ec40c484 188 fFilterMap = trk.fFilterMap;
df9db588 189 }
190
191 return *this;
192}
193
4697e4fb 194//______________________________________________________________________________
195Double_t AliAODTrack::M(AODTrkPID_t pid) const
196{
197 // Returns the mass.
9861edc0 198 // Masses for nuclei don't exist in the PDG tables, therefore they were put by hand.
4697e4fb 199
200 switch (pid) {
201
202 case kElectron :
9861edc0 203 return 0.000510999; //TDatabasePDG::Instance()->GetParticle(11/*::kElectron*/)->Mass();
4697e4fb 204 break;
205
206 case kMuon :
9861edc0 207 return 0.1056584; //TDatabasePDG::Instance()->GetParticle(13/*::kMuonMinus*/)->Mass();
4697e4fb 208 break;
209
210 case kPion :
9861edc0 211 return 0.13957; //TDatabasePDG::Instance()->GetParticle(211/*::kPiPlus*/)->Mass();
4697e4fb 212 break;
213
214 case kKaon :
9861edc0 215 return 0.4937; //TDatabasePDG::Instance()->GetParticle(321/*::kKPlus*/)->Mass();
4697e4fb 216 break;
217
218 case kProton :
9861edc0 219 return 0.9382720; //TDatabasePDG::Instance()->GetParticle(2212/*::kProton*/)->Mass();
4697e4fb 220 break;
221
222 case kDeuteron :
9861edc0 223 return 1.8756; //TDatabasePDG::Instance()->GetParticle(1000010020)->Mass();
4697e4fb 224 break;
225
226 case kTriton :
9861edc0 227 return 2.8089; //TDatabasePDG::Instance()->GetParticle(1000010030)->Mass();
4697e4fb 228 break;
229
230 case kHelium3 :
9861edc0 231 return 2.8084; //TDatabasePDG::Instance()->GetParticle(1000020030)->Mass();
4697e4fb 232 break;
233
234 case kAlpha :
9861edc0 235 return 3.7274; //TDatabasePDG::Instance()->GetParticle(1000020040)->Mass();
4697e4fb 236 break;
237
238 case kUnknown :
239 return -999.;
240 break;
241
242 default :
243 return -999.;
244 }
245}
246
247//______________________________________________________________________________
248Double_t AliAODTrack::E(AODTrkPID_t pid) const
249{
250 // Returns the energy of the particle of a given pid.
251
252 if (pid != kUnknown) { // particle was identified
253 Double_t m = M(pid);
254 return TMath::Sqrt(P()*P() + m*m);
255 } else { // pid unknown
256 return -999.;
257 }
258}
259
260//______________________________________________________________________________
261Double_t AliAODTrack::Y(AODTrkPID_t pid) const
262{
9861edc0 263 // Returns the rapidity of a particle of a given pid.
4697e4fb 264
265 if (pid != kUnknown) { // particle was identified
266 Double_t e = E(pid);
267 Double_t pz = Pz();
268 if (e>=0 && e!=pz) { // energy was positive (e.g. not -999.) and not equal to pz
269 return 0.5*TMath::Log((e+pz)/(e-pz));
270 } else { // energy not known or equal to pz
271 return -999.;
272 }
273 } else { // pid unknown
274 return -999.;
275 }
276}
277
278//______________________________________________________________________________
279Double_t AliAODTrack::Y(Double_t m) const
280{
9861edc0 281 // Returns the rapidity of a particle of a given mass.
4697e4fb 282
283 if (m >= 0.) { // mass makes sense
284 Double_t e = E(m);
285 Double_t pz = Pz();
286 if (e>=0 && e!=pz) { // energy was positive (e.g. not -999.) and not equal to pz
287 return 0.5*TMath::Log((e+pz)/(e-pz));
288 } else { // energy not known or equal to pz
289 return -999.;
290 }
291 } else { // pid unknown
292 return -999.;
293 }
294}
295
296//______________________________________________________________________________
297AliAODTrack::AODTrkPID_t AliAODTrack::GetMostProbablePID() const
298{
299 // Returns the most probable PID array element.
300
301 Int_t nPID = 10;
302 if (fPID) {
303 AODTrkPID_t loc = kUnknown;
304 Double_t max = 0.;
305 Bool_t allTheSame = kTRUE;
306
307 for (Int_t iPID = 0; iPID < nPID; iPID++) {
308 if (fPID[iPID] >= max) {
309 if (fPID[iPID] > max) {
310 allTheSame = kFALSE;
311 max = fPID[iPID];
312 loc = (AODTrkPID_t)iPID;
313 } else {
314 allTheSame = kTRUE;
315 }
316 }
317 }
318
319 return allTheSame ? kUnknown : loc;
320 } else {
321 return kUnknown;
322 }
323}
324
325//______________________________________________________________________________
326void AliAODTrack::ConvertAliPIDtoAODPID()
327{
328 // Converts AliPID array.
329 // The numbering scheme is the same for electrons, muons, pions, kaons, and protons.
330 // Everything else has to be set to zero.
331
332 fPID[kDeuteron] = 0.;
8a1418dc 333 fPID[kTriton] = 0.;
334 fPID[kHelium3] = 0.;
335 fPID[kAlpha] = 0.;
336 fPID[kUnknown] = 0.;
4697e4fb 337
338 return;
339}
340
341
df9db588 342//______________________________________________________________________________
343template <class T> void AliAODTrack::SetP(const T *p, const Bool_t cartesian)
344{
8a1418dc 345 // Set the momentum
df9db588 346
347 if (p) {
348 if (cartesian) {
16b65f2a 349 Double_t pt2 = p[0]*p[0] + p[1]*p[1];
350 Double_t P = TMath::Sqrt(pt2 + p[2]*p[2]);
df9db588 351
16b65f2a 352 fMomentum[0] = TMath::Sqrt(pt2); // pt
353 fMomentum[1] = (pt2 != 0.) ? TMath::ATan2(p[1], p[0]) : -999; // phi
354 fMomentum[2] = (P != 0.) ? TMath::ACos(p[2]/P) : -999.; // theta
df9db588 355 } else {
16b65f2a 356 fMomentum[0] = p[0]; // pt
df9db588 357 fMomentum[1] = p[1]; // phi
358 fMomentum[2] = p[2]; // theta
359 }
360 } else {
361 fMomentum[0] = -999.;
362 fMomentum[1] = -999.;
363 fMomentum[2] = -999.;
364 }
365}
366
367//______________________________________________________________________________
368template <class T> void AliAODTrack::SetPosition(const T *x, const Bool_t dca)
369{
370 // set the position
371
372 if (x) {
373 if (!dca) {
374 ResetBit(kIsDCA);
375
376 fPosition[0] = x[0];
377 fPosition[1] = x[1];
378 fPosition[2] = x[2];
379 } else {
380 SetBit(kIsDCA);
381 // don't know any better yet
382 fPosition[0] = -999.;
383 fPosition[1] = -999.;
384 fPosition[2] = -999.;
385 }
386 } else {
387 ResetBit(kIsDCA);
388
389 fPosition[0] = -999.;
390 fPosition[1] = -999.;
391 fPosition[2] = -999.;
392 }
393}
394
395//______________________________________________________________________________
396void AliAODTrack::SetDCA(Double_t d, Double_t z)
397{
398 // set the dca
399 fPosition[0] = d;
400 fPosition[1] = z;
401 fPosition[2] = 0.;
402 SetBit(kIsDCA);
403}
404
405//______________________________________________________________________________
406void AliAODTrack::Print(Option_t* /* option */) const
407{
408 // prints information about AliAODTrack
409
410 printf("Object name: %s Track type: %s\n", GetName(), GetTitle());
411 printf(" px = %f\n", Px());
412 printf(" py = %f\n", Py());
413 printf(" pz = %f\n", Pz());
414 printf(" pt = %f\n", Pt());
415 printf(" 1/pt = %f\n", OneOverPt());
416 printf(" theta = %f\n", Theta());
417 printf(" phi = %f\n", Phi());
1912763f 418 printf(" chi2/NDF = %f\n", Chi2perNDF());
df9db588 419 printf(" charge = %d\n", Charge());
420 printf(" PID object: %p\n", PID());
421}
422
e1c744ca 423void AliAODTrack::SetMatchTrigger(Int_t MatchTrigger){
8a1418dc 424//
425// Set the MUON trigger information
e1c744ca 426 switch(MatchTrigger){
427 case 0: // 0 track does not match trigger
428 fITSMuonClusterMap=fITSMuonClusterMap&0x3fffffff;
429 break;
430 case 1: // 1 track match but does not pass pt cut
431 fITSMuonClusterMap=(fITSMuonClusterMap&0x3fffffff)|0x40000000;
432 break;
433 case 2: // 2 track match Low pt cut
434 fITSMuonClusterMap=(fITSMuonClusterMap&0x3fffffff)|0x80000000;
435 break;
436 case 3: // 3 track match High pt cut
437 fITSMuonClusterMap=fITSMuonClusterMap|0xc0000000;
438 break;
439 default:
440 fITSMuonClusterMap=fITSMuonClusterMap&0x3fffffff;
441 printf("AliAODTrack::SetMatchTrigger unknown case for MatchTrigger: %d\n",MatchTrigger);
442 }
443}
444
445void AliAODTrack::SetHitsPatternInTrigCh(UShort_t hitsPatternInTrigCh){
8a1418dc 446//
447// Set the MUON hit pattern (1 bit per chamber)
e1c744ca 448 fITSMuonClusterMap=(fITSMuonClusterMap&0xffff00ff)|(hitsPatternInTrigCh<<8);
449}
450
451Int_t AliAODTrack::HitsMT(Int_t istation, Int_t iplane, Char_t *cathode){
8a1418dc 452//
453// Retrieve hit information for MUON identified by (station, plane, cathode)
e1c744ca 454 if(cathode){
455 if(cathode[0]=='x'||cathode[0]=='X'){
456 if(istation==1){
457 if(iplane==1)
458 return (fITSMuonClusterMap&0x8000)?1:0;
459 else if(iplane==2)
460 return (fITSMuonClusterMap&0x4000)?1:0;
461 else
462 return 0;
463 }else if(istation==2){
464 if(iplane==1)
465 return (fITSMuonClusterMap&0x2000)?1:0;
466 else if(iplane==2)
467 return (fITSMuonClusterMap&0x1000)?1:0;
468 else
469 return 0;
470 }else{
471 return 0;
472 }
473 }else if(cathode[0]=='y'||cathode[0]=='Y'){
474 if(istation==1){
475 if(iplane==1)
476 return (fITSMuonClusterMap&0x0800)?1:0;
477 else if(iplane==2)
478 return (fITSMuonClusterMap&0x0400)?1:0;
479 else
480 return 0;
481 }else if(istation==2){
482 if(iplane==1)
483 return (fITSMuonClusterMap&0x0200)?1:0;
484 else if(iplane==2)
485 return (fITSMuonClusterMap&0x0100)?1:0;
486 else
487 return 0;
488 }else{
489 return 0;
490 }
491 }else{
492 return 0;
493 }
494 }else{
495 if(istation==1){
496 if(iplane==1)
497 return (HitsMT(1,1,"X")||HitsMT(1,1,"Y"))?1:0;
498 else if(iplane==2)
499 return (HitsMT(1,2,"X")||HitsMT(1,2,"Y"))?1:0;
500 else
501 return 0;
502 }else if(istation==2){
503 if(iplane==1)
504 return (HitsMT(2,1,"X")||HitsMT(2,1,"Y"))?1:0;
505 else if(iplane==2)
506 return (HitsMT(2,2,"X")||HitsMT(2,2,"Y"))?1:0;
507 else
508 return 0;
509 }else{
510 return 0;
511 }
512 }
513}
514
515Int_t AliAODTrack::HitsMuonChamber(Int_t MuonChamber){
8a1418dc 516// Retrieve hit information for MUON Chamber
e1c744ca 517 switch(MuonChamber){
518 case 11:
519 return HitsMT(1,1);
520 case 12:
521 return HitsMT(1,2);
522 case 13:
523 return HitsMT(2,1);
524 case 14:
525 return HitsMT(2,2);
526 default:
527 printf("Unknown MUON chamber: %d\n",MuonChamber);
528 return 0;
529 }
530}