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