introduced backward iterator for clusters
[u/mrichter/AliRoot.git] / TRD / AliTRDseedV1.h
1 #ifndef ALITRDSEEDV1_H
2 #define ALITRDSEEDV1_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
8 ////////////////////////////////////////////////////////////////////////////
9 //                                                                        //
10 //  The TRD track seed                                                    //
11 //                                                                        //
12 ////////////////////////////////////////////////////////////////////////////
13
14 #ifndef ALITRDSEED_H
15 #include "AliTRDseed.h"
16 #endif
17
18 #ifndef ALITRDGEOMETRY_H
19 #include "AliTRDgeometry.h"
20 #endif
21
22 #ifndef ALIPID_H
23 #include "AliPID.h"
24 #endif
25
26 #ifndef ALIRIEMAN_H
27 #include "AliRieman.h"
28 #endif
29
30 class TTreeSRedirector;
31
32 class AliRieman;
33
34 class AliTRDtrackingChamber;
35 class AliTRDcluster;
36 class AliTRDtrackV1;
37 class AliTRDReconstructor;
38 class AliTRDseedV1 : public AliTRDseed
39 {
40
41   public:
42
43   enum {
44     knSlices = 10
45   };
46   // bits from 0-13 are reserved by ROOT (see TObject.h)
47   enum AliTRDtrackletStatus {
48     kOwner    = BIT(14)
49   , kRowCross = BIT(15) 
50   };
51
52   AliTRDseedV1(Int_t det = -1);
53   ~AliTRDseedV1();
54   AliTRDseedV1(const AliTRDseedV1 &ref);
55   AliTRDseedV1& operator=(const AliTRDseedV1 &ref);
56
57   Bool_t        AttachClustersIter(AliTRDtrackingChamber *chamber, Float_t quality, Bool_t kZcorr = kFALSE
58                                 , AliTRDcluster *c=0x0);
59   Bool_t        AttachClusters(AliTRDtrackingChamber *chamber, Bool_t kZcorr = kFALSE);
60   void    CookdEdx(Int_t nslices);
61   void    Draw(Option_t* o = "");
62   Bool_t  Fit(Bool_t tilt=kTRUE);
63
64   Bool_t  Init(AliTRDtrackV1 *track);
65   inline void      Init(const AliRieman *fit);
66   Bool_t    IsOwner() const          { return TestBit(kOwner);}
67   Bool_t    IsRowCross() const       { return TestBit(kRowCross);}
68
69   inline Float_t   GetChi2Z(const Float_t z = 999.) const;
70   inline Float_t   GetChi2Y(const Float_t y = 999.) const;
71   void      GetCovAt(Double_t x, Double_t *cov) const;
72   Double_t* GetCrossXYZ() { return &fCross[0];}
73   Double_t  GetCrossSz2() const { return fCross[3];}
74   Float_t*  GetdEdx() {return &fdEdx[0];}
75   Float_t   GetdQdl(Int_t ic) const;
76   Int_t     GetDetector() const {return fDet;}
77   Double_t  GetMomentum() const {return fMom;}
78   Int_t     GetN() const {return fN2;}
79   Float_t   GetQuality(Bool_t kZcorr) const;
80   Int_t     GetPlane() const         { return AliTRDgeometry::GetLayer(fDet);    }
81
82   Double_t* GetProbability();
83   Double_t  GetSnp() const           { return fSnp;}
84   Double_t  GetTgl() const           { return fTgl;}
85   Double_t  GetYat(Double_t x) const { return fYfit[0] + fYfit[1] * (x-fX0);}
86   Double_t  GetZat(Double_t x) const { return fZfit[0] + fZfit[1] * (x-fX0);}
87   
88   inline AliTRDcluster* NextCluster();
89   inline AliTRDcluster* PrevCluster();
90   void      Print(Option_t *o = "") const;
91   inline void ResetClusterIter(Bool_t forward = kTRUE);
92
93   void      SetMomentum(Double_t mom) {fMom = mom;}
94   void      SetOwner();
95   void      SetDetector(Int_t d) {fDet = d;  }
96   void      SetSnp(Double_t snp) {fSnp = snp;}
97   void      SetTgl(Double_t tgl) {fTgl = tgl;}
98   void      SetReconstructor(const AliTRDReconstructor *rec) {fReconstructor = rec;}
99 protected:
100
101   void Copy(TObject &ref) const;
102
103 private:
104   const AliTRDReconstructor *fReconstructor;//! local reconstructor
105   AliTRDcluster    **fClusterIter;          //! clusters iterator
106   Char_t           fClusterIdx;             //! clusters iterator
107   Int_t            fDet;                    //  TRD detector
108   Float_t          fMom;                    //  Momentum estimate for tracklet [GeV/c]
109   Float_t          fSnp;                    // sin of track with respect to x direction in XY plane     
110   Float_t          fTgl;                    // tg of track with respect to x direction in XZ plane      
111   Float_t          fdX;                     // length of time bin
112   Float_t          fdEdx[knSlices];         //  dE/dx measurements for tracklet
113   Double_t         fCross[4];            // spatial parameters of the pad row crossing
114   Double_t         fProb[AliPID::kSPECIES]; //  PID probabilities
115
116   ClassDef(AliTRDseedV1, 2)                 //  New TRD seed 
117
118 };
119
120 //____________________________________________________________
121 inline Float_t AliTRDseedV1::GetChi2Z(const Float_t z) const
122 {
123   Float_t z1  = (z == 999.) ? fMeanz : z;
124   Float_t chi = fZref[0] - z1;
125   return chi*chi;
126 }
127
128 //____________________________________________________________
129 inline Float_t AliTRDseedV1::GetChi2Y(const Float_t y) const
130 {
131   Float_t y1  = (y == 999.) ? fYfitR[0] : y;
132   Float_t chi = fYref[0] - y1;
133   return chi*chi;
134 }
135
136 //____________________________________________________________
137 inline void AliTRDseedV1::Init(const AliRieman *rieman)
138 {
139   fZref[0] = rieman->GetZat(fX0);
140   fZref[1] = rieman->GetDZat(fX0);
141   fYref[0] = rieman->GetYat(fX0);
142   fYref[1] = rieman->GetDYat(fX0);
143   fC       = rieman->GetC(); 
144   fChi2    = rieman->GetChi2();
145 }
146
147 //____________________________________________________________
148 inline AliTRDcluster* AliTRDseedV1::NextCluster()
149 {
150 // Mimic the usage of STL iterators.
151 // Forward iterator
152
153   fClusterIdx++; fClusterIter++;
154   while(fClusterIdx < AliTRDseed::knTimebins){
155     if(!(*fClusterIter)){ 
156       fClusterIdx++; 
157       fClusterIter++;
158       continue;
159     }
160     return *fClusterIter;
161   }
162   return 0x0;
163 }
164
165 //____________________________________________________________
166 inline AliTRDcluster* AliTRDseedV1::PrevCluster()
167 {
168 // Mimic the usage of STL iterators.
169 // Backward iterator
170
171   fClusterIdx--; fClusterIter--;
172   while(fClusterIdx >= 0){
173     if(!(*fClusterIter)){ 
174       fClusterIdx--; 
175       fClusterIter--;
176       continue;
177     }
178     return *fClusterIter;
179   }
180   return 0x0;
181 }
182
183 //____________________________________________________________
184 inline void AliTRDseedV1::ResetClusterIter(Bool_t forward) 
185 {
186 // Mimic the usage of STL iterators.
187 // Facilitate the usage of NextCluster for forward like 
188 // iterator (kTRUE) and PrevCluster for backward like iterator (kFALSE)
189
190   if(forward){
191     fClusterIter = &fClusters[0]; fClusterIter--; 
192     fClusterIdx=-1;
193   } else {
194     fClusterIter = &fClusters[AliTRDseed::knTimebins-1]; fClusterIter++; 
195     fClusterIdx=AliTRDseed::knTimebins;
196   }
197 }
198
199 #endif
200
201
202