]>
Commit | Line | Data |
---|---|---|
a6c02c85 | 1 | // @(#) $Id$ |
2 | ||
3 | // Author: Anders Vestbo <mailto:vestbo@fi.uib.no>, Uli Frankenfeld <mailto:franken@fi.uib.no> | |
4 | //*-- Copyright © ALICE HLT Group | |
5 | ||
6 | #include "AliHLTTPCStandardIncludes.h" | |
7 | ||
8 | #include "AliHLTTPCRootTypes.h" | |
9 | #include "AliHLTTPCLogging.h" | |
10 | #include "AliHLTTPCVertex.h" | |
11 | #include "AliHLTTPCConfMapPoint.h" | |
12 | #include "AliHLTTPCConfMapFit.h" | |
13 | #include "AliHLTTPCConfMapTrack.h" | |
14 | #include "AliHLTTPCTransform.h" | |
15 | #include "AliHLTTPC.h" | |
16 | ||
17 | /** \class AliHLTTPCConfMapTrack | |
18 | <pre> | |
19 | //_____________________________________________________________ | |
20 | // AliHLTTPCConfMapTrack | |
21 | // | |
22 | // Track class for conformal mapper | |
23 | </pre> | |
24 | */ | |
25 | ||
26 | #if __GNUC__ >= 3 | |
27 | using namespace std; | |
28 | #endif | |
29 | ||
30 | ClassImp(AliHLTTPCConfMapTrack) | |
31 | ||
32 | ||
33 | AliHLTTPCConfMapTrack::AliHLTTPCConfMapTrack() | |
34 | { | |
35 | //Constructor | |
36 | fChiSq[0] = 0.; | |
37 | fChiSq[1] = 0.; | |
38 | } | |
39 | ||
40 | AliHLTTPCConfMapTrack::~AliHLTTPCConfMapTrack() | |
41 | { | |
42 | //deconstructor | |
43 | } | |
44 | ||
45 | void AliHLTTPCConfMapTrack::DeleteCandidate() | |
46 | { | |
47 | //Deletes this track by resetting all its parameters. Does not delete | |
48 | //the object itself. | |
49 | ||
50 | AliHLTTPCConfMapPoint *curHit = (AliHLTTPCConfMapPoint*)fFirstHit; | |
51 | AliHLTTPCConfMapPoint *nextHit; | |
52 | ||
53 | while(curHit != 0) | |
54 | { | |
55 | nextHit = (AliHLTTPCConfMapPoint*)curHit->GetNextTrackHit(); | |
56 | curHit->SetNextTrackHit(0); | |
57 | curHit = nextHit; | |
58 | } | |
59 | ||
60 | UInt_t *hit_numbers = GetHitNumbers(); | |
61 | for(Int_t i=0; i<GetNHits(); i++) | |
62 | { | |
63 | //fHitNumbers[i] = 0; | |
64 | hit_numbers[i]=0; | |
65 | } | |
66 | ||
67 | SetRadius(0.); | |
68 | SetCenterX(0.); | |
69 | SetCenterY(0.); | |
70 | ||
71 | ComesFromMainVertex(false); | |
72 | ||
73 | SetNHits(0); | |
74 | SetCharge(0); | |
75 | fChiSq[0] = 0.; | |
76 | fChiSq[1] = 0.; | |
77 | } | |
78 | ||
79 | ||
80 | void AliHLTTPCConfMapTrack::SetProperties(Bool_t usage) | |
81 | { | |
82 | //Set the hits to this track to 'usage' | |
83 | for(StartLoop(); LoopDone(); GetNextHit()) | |
84 | { | |
85 | AliHLTTPCConfMapPoint *p = (AliHLTTPCConfMapPoint*)fCurrentHit; | |
86 | p->SetUsage(usage); | |
87 | } | |
88 | return; | |
89 | } | |
90 | ||
91 | void AliHLTTPCConfMapTrack::Reset() | |
92 | { | |
93 | //Resets the fit parameters of this track. | |
94 | ||
95 | //xy-plane | |
96 | fs11Xy = 0; | |
97 | fs12Xy = 0; | |
98 | fs22Xy = 0; | |
99 | fg1Xy = 0; | |
100 | fg2Xy = 0; | |
101 | fChiSq[0] = 0.; | |
102 | ||
103 | //sz-plane | |
104 | fs11Sz = 0; | |
105 | fs12Sz = 0; | |
106 | fs22Sz = 0; | |
107 | fg1Sz = 0; | |
108 | fg2Sz = 0; | |
109 | fChiSq[1] = 0; | |
110 | SetLength(0); | |
111 | SetNHits(0); | |
112 | } | |
113 | ||
114 | void AliHLTTPCConfMapTrack::UpdateParam(AliHLTTPCConfMapPoint *thisHit) | |
115 | { | |
116 | //Function to update fit parameters of track | |
117 | //Also, it updates the hit pointers. | |
118 | ||
119 | ||
120 | //Increment the number of hits assigned to this track: | |
121 | ||
122 | //fNHits++; | |
123 | Int_t nhits = GetNHits(); | |
124 | nhits++; | |
125 | SetNHits(nhits); //SetNHits(nhits++); | |
126 | ||
127 | //Set the hit pointers: | |
128 | //if(fNHits == 1) | |
129 | if(GetNHits()==1) | |
130 | fFirstHit = thisHit; | |
131 | else | |
132 | ((AliHLTTPCConfMapPoint*)fLastHit)->SetNextTrackHit(thisHit); | |
133 | fLastHit = thisHit; | |
134 | ||
135 | ||
136 | fs11Xy = fs11Xy + thisHit->GetXYWeight() ; | |
137 | fs12Xy = fs12Xy + thisHit->GetXYWeight() * thisHit->GetXprime() ; | |
138 | fs22Xy = fs22Xy + thisHit->GetXYWeight() * pow((thisHit->GetXprime()),2) ; | |
139 | fg1Xy = fg1Xy + thisHit->GetXYWeight() * thisHit->GetYprime() ; | |
140 | fg2Xy = fg2Xy + thisHit->GetXYWeight() * thisHit->GetXprime() * thisHit->GetYprime() ; | |
141 | ||
142 | fddXy = fs11Xy * fs22Xy - pow((fs12Xy),2) ; | |
143 | if ( fddXy != 0 ) | |
144 | { | |
145 | fa1Xy = ( fg1Xy * fs22Xy - fg2Xy * fs12Xy ) / fddXy ; | |
146 | fa2Xy = ( fg2Xy * fs11Xy - fg1Xy * fs12Xy ) / fddXy ; | |
147 | } | |
148 | ||
149 | // Now in the sz plane | |
150 | fs11Sz = fs11Sz + thisHit->GetZWeight() ; | |
151 | fs12Sz = fs12Sz + thisHit->GetZWeight() * thisHit->GetS() ; | |
152 | fs22Sz = fs22Sz + thisHit->GetZWeight() * thisHit->GetS() * thisHit->GetS() ; | |
153 | fg1Sz = fg1Sz + thisHit->GetZWeight() * thisHit->GetZ() ; | |
154 | fg2Sz = fg2Sz + thisHit->GetZWeight() * thisHit->GetS() * thisHit->GetZ() ; | |
155 | ||
156 | ||
157 | fddSz = fs11Sz * fs22Sz - fs12Sz * fs12Sz ; | |
158 | if ( fddSz != 0 ) { | |
159 | fa1Sz = ( fg1Sz * fs22Sz - fg2Sz * fs12Sz ) / fddSz ; | |
160 | fa2Sz = ( fg2Sz * fs11Sz - fg1Sz * fs12Sz ) / fddSz ; | |
161 | } | |
162 | } | |
163 | ||
164 | ||
165 | void AliHLTTPCConfMapTrack::Fill(AliHLTTPCVertex *vertex,Double_t max_Dca) | |
166 | { | |
167 | //Fill track variables with or without fit. | |
168 | ||
169 | //fRadius = sqrt(fa2Xy*fa2Xy+1)/(2*fabs(fa1Xy)); | |
170 | Double_t radius = sqrt(fa2Xy*fa2Xy+1)/(2*fabs(fa1Xy)); | |
171 | SetRadius(radius); | |
172 | ||
173 | //fPt = (Double_t)(AliHLTTPCTransform::GetBFieldValue() * fRadius); | |
174 | Double_t pt = (Double_t)(AliHLTTPCTransform::GetBFieldValue() * GetRadius()); | |
175 | SetPt(pt); | |
176 | ||
177 | if(GetPt() > max_Dca) //go for fit of helix in real space | |
178 | { | |
179 | AliHLTTPCConfMapFit *fit = new AliHLTTPCConfMapFit(this,vertex); | |
180 | ComesFromMainVertex(AliHLTTPC::DoVertexFit()); | |
181 | fit->FitHelix(); | |
182 | ||
183 | //AliHLTTPCConfMapPoint *lHit = (AliHLTTPCConfMapPoint*)fLastHit; | |
184 | AliHLTTPCConfMapPoint *fHit = (AliHLTTPCConfMapPoint*)fFirstHit; | |
185 | SetLastPoint(fHit->GetX(),fHit->GetY(),fHit->GetZ()); | |
186 | ||
187 | UpdateToFirstPoint(); | |
188 | ||
189 | delete fit; | |
190 | } | |
191 | else if(GetPt() == 0) | |
192 | LOG(AliHLTTPCLog::kError,"AliHLTTPCConfMapTrack::Fill","Tracks")<<AliHLTTPCLog::kDec<< | |
193 | "Found track with Pt=0!!!"<<ENDLOG; | |
194 | else | |
195 | { | |
196 | LOG(AliHLTTPCLog::kError,"AliHLTTPCConfMapTrack::Fill","Tracks")<<AliHLTTPCLog::kDec<< | |
197 | "Track with pt<max_Dca :"<<GetPt()<<ENDLOG; | |
198 | } | |
199 | } | |
200 | ||
201 | Int_t AliHLTTPCConfMapTrack::GetMCLabel() | |
202 | { | |
203 | //For evaluation study. | |
204 | //Returns the MCtrackID of the belonging clusters. | |
205 | //If MCLabel < 0, means that track is fake. | |
206 | ||
207 | return 0; | |
208 | /* | |
209 | Int_t num_of_clusters = GetNumberOfPoints(); | |
210 | S *s=new S[num_of_clusters]; | |
211 | Int_t i; | |
212 | for (i=0; i<num_of_clusters; i++) s[i].lab=s[i].max=0; | |
213 | ||
214 | Int_t lab=123456789; | |
215 | for (i=0; i<num_of_clusters; i++) { | |
216 | AliHLTTPCConfMapPoint *c=(AliHLTTPCConfMapPoint*)fPoints->UncheckedAt(i); | |
217 | lab=fabs(c->fMCTrackID[0]); | |
218 | Int_t j; | |
219 | for (j=0; j<num_of_clusters; j++) | |
220 | if (s[j].lab==lab || s[j].max==0) break; | |
221 | s[j].lab=lab; | |
222 | s[j].max++; | |
223 | } | |
224 | ||
225 | Int_t max=0; | |
226 | for (i=0; i<num_of_clusters; i++) | |
227 | if (s[i].max>max) {max=s[i].max; lab=s[i].lab;} | |
228 | ||
229 | delete[] s; | |
230 | ||
231 | for (i=0; i<num_of_clusters; i++) { | |
232 | AliHLTTPCConfMapPoint *c=(AliHLTTPCConfMapPoint*)fPoints->UncheckedAt(i); | |
233 | if (fabs(c->fMCTrackID[1]) == lab || | |
234 | fabs(c->fMCTrackID[2]) == lab ) max++; | |
235 | } | |
236 | ||
237 | //check if more than 10% of the clusters are incorrectly assigned (fake track): | |
238 | ||
239 | if (1.-Float_t(max)/num_of_clusters > 0.10) | |
240 | { | |
241 | return -lab; | |
242 | } | |
243 | Int_t tail=Int_t(0.08*174); | |
244 | if (num_of_clusters < tail) return lab; | |
245 | ||
246 | max=0; | |
247 | for (i=1; i<=tail; i++) { | |
248 | AliHLTTPCConfMapPoint *c = (AliHLTTPCConfMapPoint*)fPoints->UncheckedAt(num_of_clusters-i); | |
249 | if (lab == fabs(c->fMCTrackID[0]) || | |
250 | lab == fabs(c->fMCTrackID[1]) || | |
251 | lab == fabs(c->fMCTrackID[2])) max++; | |
252 | } | |
253 | if (max < Int_t(0.5*tail)) | |
254 | { | |
255 | //printf("Wrong innermost clusters\n"); | |
256 | return -lab; | |
257 | } | |
258 | return lab; | |
259 | */ | |
260 | } | |
261 |