Fixed copy/paste error
[u/mrichter/AliRoot.git] / TRD / AliTRDtrackV1.h
CommitLineData
d9950a5a 1#ifndef ALITRDTRACKV1_H
2#define ALITRDTRACKV1_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id$ */
7
0906e73e 8///////////////////////////////////////////////////////////////////////////////
9// //
10// Represents a reconstructed TRD track //
11// //
12///////////////////////////////////////////////////////////////////////////////
d9950a5a 13
3b57a3f7 14#ifndef ALIKALMANTRACK_H
15#include "AliKalmanTrack.h"
d9950a5a 16#endif
17
3b57a3f7 18#ifndef ALIESDTRACK_H
19#include "AliESDtrack.h"
20#endif
21
22#ifndef ALITRDSEEDV1_H
23#include "AliTRDseedV1.h"
24#endif
d9950a5a 25
3b57a3f7 26class AliTRDcluster;
27class AliESDtrack;
28class AliTRDtrackV1 : public AliKalmanTrack
d9950a5a 29{
3b57a3f7 30public:
31 enum { kMAXCLUSTERSPERTRACK = 210 };
32
33 enum { kNdet = 540
34 , kNstacks = 90
35 , kNplane = AliESDtrack::kTRDnPlanes
36 , kNcham = 5
37 , kNsect = 18
38 , kNslice = 3
39 , kNMLPslice = 8 };
40
41 enum AliTRDPIDMethod {
42 kNN = 0
43 , kLQ = 1 };
0906e73e 44
3b57a3f7 45 AliTRDtrackV1();
46 AliTRDtrackV1(AliTRDseedV1 *trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
47 AliTRDtrackV1(const AliESDtrack &ref);
48 AliTRDtrackV1(const AliTRDtrackV1 &ref);
49 virtual ~AliTRDtrackV1();
50 AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref) { *(new(this) AliTRDtrackV1(ref)); return *this; }
51
52 Bool_t CookPID();
53 Bool_t CookLabel(Float_t wrong);
54 AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
55 Double_t GetBudget(Int_t i) const { return fBudget[i];}
56 Double_t GetC() const { return AliExternalTrackParam::GetC(GetBz());}
57 Int_t GetClusterIndex(Int_t id) const;
58 Float_t GetEdep() const {return fDE;}
59 inline Float_t GetMomentum(Int_t plane) const;
60 inline Int_t GetNCross();
61 Double_t GetPIDsignal() const {return 0.;}
62 Double_t GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;}
63 Double_t GetPredictedChi2(const AliTRDseedV1 *tracklet) const;
64 Double_t GetPredictedChi2(const AliCluster* /*c*/) const { return 0.0; }
65 Int_t GetProlongation(Double_t xk, Double_t &y, Double_t &z);
66 AliTRDseedV1* GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : 0x0;}
67 Int_t GetTrackletIndex(Int_t plane) const { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;}
68 UShort_t* GetTrackletIndexes() {return &fTrackletIndex[0];}
69
70 Bool_t IsOwner() const {return TestBit(1);};
71 Bool_t IsStopped() const {return TestBit(2);};
72
87a7fa94 73 void MakeBackupTrack();
3b57a3f7 74 Bool_t PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
75 Int_t PropagateToR(Double_t xr, Double_t step);
76 Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE);
77 void SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
78 void SetNumberOfClusters();
79 void SetOwner();
80 void SetStopped(Bool_t stop) {SetBit(2, stop);}
81 void SetTracklet(AliTRDseedV1 *trklt, Int_t index);
82 inline Float_t StatusForTOF();
83 Bool_t Update(AliTRDseedV1 *tracklet, Double_t chi2);
84 //Bool_t Update(const AliTRDcluster *c, Double_t chi2, Int_t index, Double_t h01){ return AliTRDtrack::Update(c,chi2,index,h01); };
85 Bool_t Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
86 void UpdateESDtrack(AliESDtrack *t);
87
88protected:
89 Double_t GetBz() const;
90
91private:
92 UChar_t fPIDquality; // No of planes used for PID calculation
93 UShort_t fTrackletIndex[kNplane];// Tracklets index in the tracker list
94 Double32_t fPID[AliPID::kSPECIES];// PID probabilities
95 Double32_t fBudget[3]; // Integrated material budget
96 Double32_t fDE; // Integrated delta energy
97 AliTRDseedV1 *fTracklet[kNplane]; // Tracklets array defining the track
98 AliTRDtrackV1 *fBackupTrack; // Backup track
d9950a5a 99
0906e73e 100
3b57a3f7 101 ClassDef(AliTRDtrackV1, 2) // new TRD track
d9950a5a 102};
103
3b57a3f7 104//____________________________________________________
105inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
106{
107 return plane >=0 && plane < kNplane && fTrackletIndex[plane] != 0xff ? fTracklet[plane]->GetMomentum() : -1.;
108}
109
110//____________________________________________________
111inline Int_t AliTRDtrackV1::GetNCross()
112{
113 Int_t ncross = 0;
114 for(Int_t ip=0; ip<kNplane; ip++){
115 if(!fTracklet[ip]) continue;
116 ncross += fTracklet[ip]->IsRowCross();
117 }
118 return ncross;
119}
120
121//____________________________________________________________________________
122inline Float_t AliTRDtrackV1::StatusForTOF()
123{
124 //
125 // Defines the status of the TOF extrapolation
126 //
127
128 if(!fTracklet[5]) return 0.;
129
130 // Definition of res ????
131 Float_t res = /*(0.2 + 0.8 * (fN / (fNExpected + 5.0))) **/ (0.4 + 0.6 * fTracklet[5]->GetN() / 20.0);
132 res *= (0.25 + 0.8 * 40.0 / (40.0 + fBudget[2]));
133 return res;
134}
d9950a5a 135
136#endif
137
138