]> git.uio.no Git - u/mrichter/AliRoot.git/blame_incremental - TRD/AliTRDseedV1.h
Analysis now deriving from AliAnalysisTaskSE. The subtasks are still there but now...
[u/mrichter/AliRoot.git] / TRD / AliTRDseedV1.h
... / ...
CommitLineData
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
30class TTreeSRedirector;
31
32class AliRieman;
33
34class AliTRDtrackingChamber;
35class AliTRDcluster;
36class AliTRDtrackV1;
37class AliTRDReconstructor;
38class 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(
58 AliTRDtrackingChamber *chamber, Float_t quality,
59 Bool_t kZcorr = kFALSE, AliTRDcluster *c=0x0);
60 Bool_t AttachClusters(
61 AliTRDtrackingChamber *chamber, Bool_t kZcorr = kFALSE);
62 void Bootstrap(const AliTRDReconstructor *rec);
63 void CookdEdx(Int_t nslices);
64 Bool_t Fit(Bool_t tilt=kTRUE);
65
66 Bool_t Init(AliTRDtrackV1 *track);
67 inline void Init(const AliRieman *fit);
68 Bool_t IsEqual(const TObject *inTracklet) const;
69 Bool_t IsOwner() const { return TestBit(kOwner);}
70 Bool_t IsRowCross() const { return TestBit(kRowCross);}
71
72 inline Float_t GetChi2Z(const Float_t z = 999.) const;
73 inline Float_t GetChi2Y(const Float_t y = 999.) const;
74 void GetCovAt(Double_t x, Double_t *cov) const;
75 void GetCovRef(Double_t *cov) const { memcpy(cov, &fRefCov[0], 3*sizeof(Double_t));}
76 Double_t* GetCrossXYZ() { return &fCross[0];}
77 Double_t GetCrossSz2() const { return fCross[3];}
78 Float_t GetdX() const { return fdX;}
79 Float_t* GetdEdx() { return &fdEdx[0];}
80 Float_t GetdQdl(Int_t ic) const;
81 Int_t GetDetector() const { return fDet;}
82 Double_t GetMomentum() const { return fMom;}
83 Int_t GetN() const { return fN2;}
84 Float_t GetQuality(Bool_t kZcorr) const;
85 Int_t GetPlane() const { return AliTRDgeometry::GetLayer(fDet); }
86
87 Double_t* GetProbability();
88 Double_t GetSnp() const { return fSnp;}
89 Double_t GetTgl() const { return fTgl;}
90 Float_t GetXref() const { return fXref;}
91 Double_t GetYat(Double_t x) const { return fYfit[0] + fYfit[1] * (x-fX0);}
92 Double_t GetZat(Double_t x) const { return fZfit[0] + fZfit[1] * (x-fX0);}
93
94 inline AliTRDcluster* NextCluster();
95 inline AliTRDcluster* PrevCluster();
96 void Print(Option_t *o = "") const;
97 inline void ResetClusterIter(Bool_t forward = kTRUE);
98
99 void SetMomentum(Double_t mom){ fMom = mom;}
100 void SetOwner();
101 void SetDetector(Int_t d) { fDet = d; }
102 void SetDX(Float_t inDX) { fdX = inDX;}
103 void SetSnp(Double_t snp) { fSnp = snp;}
104 void SetTgl(Double_t tgl) { fTgl = tgl;}
105 void SetReconstructor(const AliTRDReconstructor *rec) {fReconstructor = rec;}
106protected:
107
108 void Copy(TObject &ref) const;
109
110private:
111 const AliTRDReconstructor *fReconstructor;//! local reconstructor
112 AliTRDcluster **fClusterIter; //! clusters iterator
113 Char_t fClusterIdx; //! clusters iterator
114 Int_t fDet; // TRD detector
115 Float_t fMom; // Momentum estimate for tracklet [GeV/c]
116 Float_t fSnp; // sin of track with respect to x direction in XY plane
117 Float_t fTgl; // tg of track with respect to x direction in XZ plane
118 Float_t fdX; // length of time bin
119 Float_t fXref; // average radial position of clusters
120 Float_t fdEdx[knSlices]; // dE/dx measurements for tracklet
121 Double_t fCross[4]; // spatial parameters of the pad row crossing
122 Double_t fRefCov[3]; // covariance matrix of the track in the yz plane
123 Double_t fProb[AliPID::kSPECIES]; // PID probabilities
124
125 ClassDef(AliTRDseedV1, 3) // New TRD seed
126
127};
128
129//____________________________________________________________
130inline Float_t AliTRDseedV1::GetChi2Z(const Float_t z) const
131{
132 Float_t z1 = (z == 999.) ? fMeanz : z;
133 Float_t chi = fZref[0] - z1;
134 return chi*chi;
135}
136
137//____________________________________________________________
138inline Float_t AliTRDseedV1::GetChi2Y(const Float_t y) const
139{
140 Float_t y1 = (y == 999.) ? fYfitR[0] : y;
141 Float_t chi = fYref[0] - y1;
142 return chi*chi;
143}
144
145//____________________________________________________________
146inline void AliTRDseedV1::Init(const AliRieman *rieman)
147{
148 fZref[0] = rieman->GetZat(fX0);
149 fZref[1] = rieman->GetDZat(fX0);
150 fYref[0] = rieman->GetYat(fX0);
151 fYref[1] = rieman->GetDYat(fX0);
152 fC = rieman->GetC();
153 fChi2 = rieman->GetChi2();
154}
155
156//____________________________________________________________
157inline AliTRDcluster* AliTRDseedV1::NextCluster()
158{
159// Mimic the usage of STL iterators.
160// Forward iterator
161
162 fClusterIdx++; fClusterIter++;
163 while(fClusterIdx < AliTRDseed::knTimebins){
164 if(!(*fClusterIter)){
165 fClusterIdx++;
166 fClusterIter++;
167 continue;
168 }
169 return *fClusterIter;
170 }
171 return 0x0;
172}
173
174//____________________________________________________________
175inline AliTRDcluster* AliTRDseedV1::PrevCluster()
176{
177// Mimic the usage of STL iterators.
178// Backward iterator
179
180 fClusterIdx--; fClusterIter--;
181 while(fClusterIdx >= 0){
182 if(!(*fClusterIter)){
183 fClusterIdx--;
184 fClusterIter--;
185 continue;
186 }
187 return *fClusterIter;
188 }
189 return 0x0;
190}
191
192//____________________________________________________________
193inline void AliTRDseedV1::ResetClusterIter(Bool_t forward)
194{
195// Mimic the usage of STL iterators.
196// Facilitate the usage of NextCluster for forward like
197// iterator (kTRUE) and PrevCluster for backward like iterator (kFALSE)
198
199 if(forward){
200 fClusterIter = &fClusters[0]; fClusterIter--;
201 fClusterIdx=-1;
202 } else {
203 fClusterIter = &fClusters[AliTRDseed::knTimebins-1]; fClusterIter++;
204 fClusterIdx=AliTRDseed::knTimebins;
205 }
206}
207
208#endif
209
210
211