]>
Commit | Line | Data |
---|---|---|
18856a77 | 1 | #ifndef ALIITSTRACKERV2_H |
2 | #define ALIITSTRACKERV2_H | |
006b5f7f | 3 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * |
4 | * See cxx source for full Copyright notice */ | |
5 | ||
6 | //------------------------------------------------------------------------- | |
7 | // ITS tracker | |
18856a77 | 8 | // reads AliITSclusterV2 clusters and creates AliITStrackV2 tracks |
9 | // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch | |
006b5f7f | 10 | //------------------------------------------------------------------------- |
64ea0cbc | 11 | |
12 | #include <TObjArray.h> | |
13 | ||
006b5f7f | 14 | #include "AliTracker.h" |
15 | #include "AliITSrecoV2.h" | |
16 | #include "AliITStrackV2.h" | |
17 | ||
18 | class AliITSclusterV2; | |
c630aafd | 19 | class AliESD; |
006b5f7f | 20 | class AliITSgeom; |
c630aafd | 21 | class TTree; |
006b5f7f | 22 | |
23 | ||
24 | //------------------------------------------------------------------------- | |
25 | class AliITStrackerV2 : public AliTracker { | |
26 | public: | |
27 | AliITStrackerV2():AliTracker(){} | |
c630aafd | 28 | AliITStrackerV2(const AliITSgeom *geom); |
006b5f7f | 29 | AliCluster *GetCluster(Int_t index) const; |
235b705f | 30 | AliITSclusterV2 *GetClusterLayer(Int_t layn, Int_t ncl) const |
31 | {return fgLayers[layn].GetCluster(ncl);} | |
32 | Int_t GetNumberOfClustersLayer(Int_t layn) const | |
33 | {return fgLayers[layn].GetNumberOfClusters();} | |
c630aafd | 34 | Int_t LoadClusters(TTree *cf); |
61ab8ea8 | 35 | void UnloadClusters(); |
c630aafd | 36 | Int_t Clusters2Tracks(TTree *in, TTree *out); |
37 | Int_t Clusters2Tracks(AliESD *event); | |
c630aafd | 38 | Int_t PropagateBack(AliESD *event); |
c630aafd | 39 | Int_t RefitInward(AliESD *event); |
40 | Bool_t RefitAt(Double_t x, AliITStrackV2 *seed, const AliITStrackV2 *t); | |
7f6ddf58 | 41 | void SetupFirstPass(Int_t *flags, Double_t *cuts=0); |
42 | void SetupSecondPass(Int_t *flags, Double_t *cuts=0); | |
43 | ||
c0dd5278 | 44 | void SetLastLayerToTrackTo(Int_t l=0) {fLastLayerToTrackTo=l;} |
45 | void SetLayersNotToSkip(Int_t *l); | |
46 | ||
b87bd7cd | 47 | void UseClusters(const AliKalmanTrack *t, Int_t from=0) const; |
48 | ||
babd135a | 49 | class AliITSdetector { |
7f6ddf58 | 50 | public: |
51 | AliITSdetector(){} | |
52 | AliITSdetector(Double_t r,Double_t phi) {fR=r; fPhi=phi;} | |
7f6ddf58 | 53 | Double_t GetR() const {return fR;} |
54 | Double_t GetPhi() const {return fPhi;} | |
55 | private: | |
56 | Double_t fR; // polar coordinates | |
57 | Double_t fPhi; // of this detector | |
58 | }; | |
59 | ||
60 | class AliITSlayer { | |
babd135a | 61 | friend class AliITStrackerV2; |
7f6ddf58 | 62 | public: |
63 | AliITSlayer(); | |
64 | AliITSlayer(Double_t r, Double_t p, Double_t z, Int_t nl, Int_t nd); | |
65 | ~AliITSlayer(); | |
66 | Int_t InsertCluster(AliITSclusterV2 *c); | |
61ab8ea8 | 67 | void ResetClusters(); |
babd135a | 68 | void ResetWeights(); |
7f6ddf58 | 69 | void SelectClusters(Double_t zmi,Double_t zma,Double_t ymi,Double_t yma); |
70 | const AliITSclusterV2 *GetNextCluster(Int_t &ci); | |
b87bd7cd | 71 | void ResetRoad(); |
72 | Double_t GetRoad() const {return fRoad;} | |
7f6ddf58 | 73 | Double_t GetR() const {return fR;} |
de5d3595 | 74 | AliITSclusterV2 *GetCluster(Int_t i) const {return i<fN? fClusters[i]:0;} |
babd135a | 75 | Float_t *GetWeight(Int_t i) {return i<fN ?&fClusterWeight[i]:0;} |
7f6ddf58 | 76 | AliITSdetector &GetDetector(Int_t n) const { return fDetectors[n]; } |
77 | Int_t FindDetectorIndex(Double_t phi, Double_t z) const; | |
61ab8ea8 | 78 | Double_t GetThickness(Double_t y, Double_t z, Double_t &x0) const; |
7f6ddf58 | 79 | Int_t InRoad() const ; |
80 | Int_t GetNumberOfClusters() const {return fN;} | |
61ab8ea8 | 81 | Int_t GetNladders() const {return fNladders;} |
82 | Int_t GetNdetectors() const {return fNdetectors;} | |
235b705f | 83 | protected: |
7f6ddf58 | 84 | Double_t fR; // mean radius of this layer |
85 | Double_t fPhiOffset; // offset of the first detector in Phi | |
86 | Int_t fNladders; // number of ladders | |
87 | Double_t fZOffset; // offset of the first detector in Z | |
88 | Int_t fNdetectors; // detectors/ladder | |
89 | AliITSdetector *fDetectors; // array of detectors | |
90 | Int_t fN; // number of clusters | |
91 | AliITSclusterV2 *fClusters[kMaxClusterPerLayer]; // pointers to clusters | |
babd135a | 92 | Float_t fClusterWeight[kMaxClusterPerLayer]; // probabilistic weight of the cluster |
7f6ddf58 | 93 | Double_t fZmax; // edges |
94 | Double_t fYmin; // of the | |
95 | Double_t fYmax; // "window" | |
96 | Int_t fI; // index of the current cluster within the "window" | |
b87bd7cd | 97 | Double_t fRoad; // road defined by the cluster density |
7f6ddf58 | 98 | Int_t FindClusterIndex(Double_t z) const; |
99 | }; | |
de5d3595 | 100 | |
235b705f | 101 | protected: |
7f6ddf58 | 102 | void CookLabel(AliKalmanTrack *t,Float_t wrong) const; |
a9a2d814 | 103 | Double_t GetEffectiveThickness(Double_t y, Double_t z) const; |
006b5f7f | 104 | void FollowProlongation(); |
105 | Int_t TakeNextProlongation(); | |
006b5f7f | 106 | void ResetBestTrack() { |
107 | fBestTrack.~AliITStrackV2(); | |
108 | new(&fBestTrack) AliITStrackV2(fTrackToFollow); | |
109 | } | |
006b5f7f | 110 | void ResetTrackToFollow(const AliITStrackV2 &t) { |
111 | fTrackToFollow.~AliITStrackV2(); | |
112 | new(&fTrackToFollow) AliITStrackV2(t); | |
113 | } | |
babd135a | 114 | Float_t *GetWeight(Int_t index); |
de5d3595 | 115 | void AddTrackHypothesys(AliITStrackV2 * track, Int_t esdindex); |
babd135a | 116 | void SortTrackHypothesys(Int_t esdindex, Float_t likelihoodlevel); |
117 | void CompressTrackHypothesys(Int_t esdindex, Float_t likelihoodlevel, Int_t maxsize); | |
118 | AliITStrackV2 * GetBestHypothesys(Int_t esdindex, AliITStrackV2 * original, Int_t checkmax); | |
119 | AliITStrackV2 * GetBestHypothesysMIP(Int_t esdindex, AliITStrackV2 * original); | |
7f6ddf58 | 120 | Int_t fI; // index of the current layer |
18856a77 | 121 | static AliITSlayer fgLayers[kMaxLayer];// ITS layers |
7f6ddf58 | 122 | AliITStrackV2 fTracks[kMaxLayer]; // track estimations at the ITS layers |
123 | AliITStrackV2 fBestTrack; // "best" track | |
124 | AliITStrackV2 fTrackToFollow; // followed track | |
de5d3595 | 125 | TObjArray fTrackHypothesys; // ! array with track hypothesys- ARRAY is the owner of tracks- MI |
126 | Int_t fCurrentEsdTrack; // ! current esd track - MI | |
7f6ddf58 | 127 | Int_t fPass; // current pass through the data |
128 | Int_t fConstraint[2]; // constraint flags | |
8676d691 | 129 | |
c0dd5278 | 130 | Int_t fLayersNotToSkip[kMaxLayer]; // layer masks |
131 | Int_t fLastLayerToTrackTo; // the innermost layer to track to | |
c630aafd | 132 | ClassDef(AliITStrackerV2,1) //ITS tracker V2 |
006b5f7f | 133 | }; |
134 | ||
135 | ||
7f6ddf58 | 136 | inline void AliITStrackerV2::SetupFirstPass(Int_t *flags, Double_t *cuts) { |
137 | // This function sets up flags and cuts for the first tracking pass | |
138 | // | |
139 | // flags[0] - vertex constaint flag | |
140 | // negative means "skip the pass" | |
141 | // 0 means "no constraint" | |
142 | // positive means "normal constraint" | |
143 | ||
144 | fConstraint[0]=flags[0]; | |
145 | if (cuts==0) return; | |
146 | } | |
147 | ||
148 | inline void AliITStrackerV2::SetupSecondPass(Int_t *flags, Double_t *cuts) { | |
149 | // This function sets up flags and cuts for the second tracking pass | |
150 | // | |
151 | // flags[0] - vertex constaint flag | |
152 | // negative means "skip the pass" | |
153 | // 0 means "no constraint" | |
154 | // positive means "normal constraint" | |
155 | ||
156 | fConstraint[1]=flags[0]; | |
157 | if (cuts==0) return; | |
158 | } | |
159 | ||
160 | inline void AliITStrackerV2::CookLabel(AliKalmanTrack *t,Float_t wrong) const { | |
161 | //-------------------------------------------------------------------- | |
162 | //This function "cooks" a track label. If label<0, this track is fake. | |
163 | //-------------------------------------------------------------------- | |
164 | Int_t tpcLabel=t->GetLabel(); | |
165 | if (tpcLabel<0) return; | |
166 | AliTracker::CookLabel(t,wrong); | |
babd135a | 167 | if (tpcLabel!=TMath::Abs(t->GetLabel())){ |
168 | t->SetFakeRatio(1.); | |
169 | } | |
170 | if (tpcLabel !=t->GetLabel()) { | |
171 | t->SetLabel(-tpcLabel); | |
172 | } | |
7f6ddf58 | 173 | } |
006b5f7f | 174 | |
175 | #endif |