3ed2a9c3e5ba98b499ac1df74a6200cfd695b542
[u/mrichter/AliRoot.git] / HLT / TPCLib / AliHLTTPCConfMapPoint.cxx
1 // @(#) $Id$
2 // Original: AliL3ConfMapPoint.cxx,v 1.10 2005/06/23 17:46:55 hristov 
3
4 // Author: Anders Vestbo <mailto:vestbo$fi.uib.no>
5 //*-- Copyright &copy ALICE HLT Group
6
7 #include "AliHLTTPCLogging.h"
8 #include "AliHLTTPCConfMapPoint.h"
9 #include "AliHLTTPCSpacePointData.h"
10 #include "AliHLTTPCVertex.h"
11 #include "AliHLTTPCConfMapTrack.h"
12
13 /**
14 <pre>
15 //_____________________________________________________________
16 // AliHLTTPCConfMapPoint
17 //
18 // Hit class for conformal mapper
19 </pre
20 */
21
22 ClassImp(AliHLTTPCConfMapPoint)
23
24 Bool_t AliHLTTPCConfMapPoint::fgDontMap=kFALSE;
25
26 AliHLTTPCConfMapPoint::AliHLTTPCConfMapPoint()
27 {
28   //Constructor
29   
30   SetUsage(false);
31   SetHitNumber(-1);
32   SetX(0);
33   SetY(0);
34   SetZ(0);
35   SetXerr(0);
36   SetYerr(0);
37   SetZerr(0);
38
39   SetPhi(0.);
40   SetEta(0.);
41   
42   SetXprime(0.);
43   SetYprime(0.);
44   SetXprimeerr(0.);
45   SetYprimeerr(0.);
46   SetIntPoint(0., 0., 0., 0., 0., 0.);
47   SetShiftedCoord();
48   SetMCTrackID(0,0,0);
49 }
50
51 AliHLTTPCConfMapPoint::~AliHLTTPCConfMapPoint()
52 {
53   // Destructor.
54   ;
55 }
56
57 Bool_t AliHLTTPCConfMapPoint::ReadHits(AliHLTTPCSpacePointData* hits )
58 {
59   //read the hits
60   SetHitNumber(hits->fID);
61   SetPadRow(hits->fPadRow);
62   Int_t slice = (hits->fID>>25) & 0x7f;
63   SetSector(slice);
64   SetX(hits->fX);
65   SetY(hits->fY);
66   SetZ(hits->fZ);
67   SetXerr(sqrt(hits->fSigmaY2));
68   SetYerr(sqrt(hits->fSigmaY2));
69   SetZerr(sqrt(hits->fSigmaZ2));
70   return kTRUE;
71 }
72
73 void AliHLTTPCConfMapPoint::Reset()
74 {
75   //Reset this point.
76   SetUsage(kFALSE);
77   SetS(0);
78   fNextRowHit = 0;
79   fNextVolumeHit=0;
80   fNextTrackHit=0;
81 }
82
83 void AliHLTTPCConfMapPoint::Setup(AliHLTTPCVertex *vertex)
84 {
85   //Setup. Sets the vertex, conformal coordinates, 
86   //and phi and eta of each hit.
87   
88   SetIntPoint(vertex->GetX(),    vertex->GetY(),    vertex->GetZ(),
89               vertex->GetXErr(), vertex->GetYErr(), vertex->GetZErr());
90   SetShiftedCoord();
91   SetConfCoord();
92   // The angles are set properly if they are set after 
93   // the interaction point and the shifted coordinates
94   SetAngles();
95   //SetDist(0., 0.);
96   return;
97 }
98
99 void AliHLTTPCConfMapPoint::SetIntPoint(Double_t inx, Double_t iny, Double_t inz,
100                                     Double_t inxerr, Double_t inyerr, Double_t inzerr)
101 {
102   // Defines a new interaction point. This point is needed to calculate
103   // the conformal coordinates.
104
105   SetXt(inx);
106   SetYt(iny);
107   SetZt(inz);
108   SetXterr(inxerr);
109   SetYterr(inyerr);
110   SetZterr(inzerr);
111
112   return;
113 }
114
115 void AliHLTTPCConfMapPoint::SetAllCoord(const AliHLTTPCConfMapPoint *precedinghit)
116 {
117   // Sets the interaction point, the shifted coordinates, and the conformal mapping coordinates.
118   // These values are calculated from the interaction point of the given cluster which should be a
119   // already found cluster on the same track.
120
121   if (this == precedinghit) {
122     SetIntPoint(precedinghit->GetX(),    precedinghit->GetY(),    precedinghit->GetZ(),
123                 precedinghit->GetXerr(), precedinghit->GetYerr(), precedinghit->GetZerr());
124   }
125
126   else {
127     SetIntPoint(precedinghit->GetXt(),    precedinghit->GetYt(),    precedinghit->GetZt(),
128                 precedinghit->GetXterr(), precedinghit->GetYterr(), precedinghit->GetZterr());
129   }
130
131   SetShiftedCoord();
132   SetConfCoord();
133
134   return;
135 }
136
137 void AliHLTTPCConfMapPoint::SetShiftedCoord()
138 {
139   // Sets the coordinates with resepct to the given vertex point
140
141   SetXv(GetX() - fXt);
142   SetYv(GetY() - fYt);
143   SetZv(GetZ() - fZt);
144   /*
145   SetXverr(TMath::Sqrt(GetXerr()*GetXerr() + fXterr*fXterr));
146   SetYverr(TMath::Sqrt(GetYerr()*GetYerr() + fYterr*fYterr));
147   SetZverr(TMath::Sqrt(GetZerr()*GetZerr() + fZterr*fZterr));
148   */
149   return;
150 }
151
152 void AliHLTTPCConfMapPoint::SetConfCoord()
153 {
154   // Calculates the conformal coordinates of one cluster.
155   // If the option "vertex_constraint" applies the interaction point is
156   // assumed to be at (0, 0, 0). Otherwise the function will use the
157   // interaction point specified by fXt and fYt.
158
159   if(fgDontMap){
160     fXprime = fx;
161     fYprime = fy;
162     fWxy = 0;
163     fs = 0; //track trajectory
164     fWz = 0;
165     return;
166   }
167
168   Double_t r2;
169   Double_t xyErrorScale = 1;
170   Double_t szErrorScale = 1;
171
172   if ((r2 = fXv*fXv + fYv*fYv)) 
173     {
174       fXprime =  fXv / r2;
175       fYprime = -fYv / r2;
176       
177       //set weights:
178       fWxy = r2*r2 / ((xyErrorScale*xyErrorScale)*((fxerr*fxerr)+(fyerr*fyerr)));
179       fs = 0; //track trajectory
180       fWz = (Double_t)(1./(szErrorScale*fzerr*fzerr));
181     } else {
182     fXprime    = 0.;
183     fYprime    = 0.;
184     fXprimeerr = 0.;
185     fYprimeerr = 0.;
186     fWxy = 0;
187     fWz = 0;
188     fs = 0;
189   }
190
191   return;
192 }
193
194 void AliHLTTPCConfMapPoint::SetAngles()
195 {
196   // Calculates the angle phi and the pseudorapidity eta for each cluster.
197   /*
198   Double_t r = TMath::Sqrt(x*x + y*y);
199
200   fPhi = TMath::ATan2(y,x);
201   if(fPhi<0) fPhi = fPhi + 2*TMath::Pi();
202   fEta = 3.*z/(TMath::Abs(z)+2.*r);
203   return;
204   */
205   //  Double_t r3dim = TMath::Sqrt(fXv*fXv + fYv*fYv + fZv*fZv);
206   Double_t r3dim = sqrt(fXv*fXv + fYv*fYv + fZv*fZv);
207   //Double_t r2dim = TMath::Sqrt(fXv*fXv + fYv*fYv);
208
209   /*if (r2dim == 0.) {
210   // If r2dim == 0 the pseudorapidity eta cannot be calculated (division by zero)!
211   // This can only happen if the point is lying on the z-axis and this should never be possible.
212     cerr << "The pseudorapidity eta cannot be calculated (division by zero)! Set to 1.e-10." << endl;
213     r2dim = 1.e-10;
214   }
215
216   if (fXv == 0.) {
217     fPhi = (fYv > 0.) ? TMath::Pi() / 2. : - TMath::Pi() / 2.;
218   }
219
220   else {
221     fPhi = (fXv > 0.) ? TMath::ASin(fYv/r2dim) : TMath::Pi() - TMath::ASin(fYv/r2dim);
222   }
223
224   if (fPhi < 0.) {
225     fPhi += 2. * TMath::Pi();
226   }
227   */
228   //fPhi = TMath::ATan2(y,x);
229   fPhi = atan2(fy,fx);
230   //if(fPhi<0) fPhi = fPhi + 2*TMath::Pi();
231   
232   //fEta = 0.5 * TMath::Log((r3dim + fZv)/(r3dim - fZv));
233   fEta = 0.5 * log((r3dim + fZv)/(r3dim - fZv));
234   return;
235 }
236
237 /*
238 AliHLTTPCConfMapTrack *AliHLTTPCConfMapPoint::GetTrack(TClonesArray *tracks) const
239 {
240   // Returns the pointer to the track to which this hit belongs.
241   
242   return (AliHLTTPCConfMapTrack*)tracks->At(this->GetTrackNumber());
243 }
244 */