]>
Commit | Line | Data |
---|---|---|
4603aa2f | 1 | //$Id$ |
2 | ||
b661165c | 3 | // Author: Uli Frankenfeld <mailto:franken@fi.uib.no> |
4 | //*-- Copyright © Uli | |
108615fc | 5 | |
118c26c3 | 6 | #include "AliL3StandardIncludes.h" |
7 | ||
108615fc | 8 | #include "AliL3Logging.h" |
108615fc | 9 | #include "AliL3Merger.h" |
10 | #include "AliL3Track.h" | |
11 | #include "AliL3TrackSegmentData.h" | |
12 | #include "AliL3Transform.h" | |
13 | #include "AliL3TrackArray.h" | |
14 | ||
15 | #ifdef use_root //use root ntuple for slow merge | |
16 | #include <TNtuple.h> | |
17 | #include <TTree.h> | |
18 | #include <TFile.h> | |
19 | #endif | |
b661165c | 20 | |
3e87ef69 | 21 | /** \class AliL3Merger |
22 | <pre> | |
108615fc | 23 | //_____________________________________________________________ |
b661165c | 24 | // AliL3Merger |
108615fc | 25 | // |
26 | // The L3 merger base class | |
27 | // | |
3e87ef69 | 28 | </pre> |
29 | */ | |
108615fc | 30 | |
31 | ClassImp(AliL3Merger) | |
32 | ||
33 | AliL3Merger::AliL3Merger(){ | |
34 | //Default constructor | |
108615fc | 35 | SetArray(0); |
36 | } | |
37 | ||
38 | ||
5dcd65ea | 39 | AliL3Merger::AliL3Merger(Int_t ntrackarrays,Char_t *tracktype){ |
108615fc | 40 | //Constructor. |
5dcd65ea | 41 | |
42 | if(strcmp(tracktype,"AliL3Track")==0) fTrackType='t'; | |
43 | else if(strcmp(tracktype,"AliL3ConfMapTrack")==0) fTrackType='c'; | |
44 | else if(strcmp(tracktype,"AliL3HoughTrack")==0) fTrackType='h'; | |
45 | else | |
46 | LOG(AliL3Log::kError,"AliL3Merger::AliL3Merger","Track types") | |
47 | <<"Unknown tracktype"<<ENDLOG; | |
108615fc | 48 | SetArray(ntrackarrays); |
49 | fCurrentTracks=0; | |
5dcd65ea | 50 | |
108615fc | 51 | } |
52 | ||
53 | AliL3Merger::~AliL3Merger(){ | |
54 | //Destructor | |
55 | DeleteArray(); | |
56 | } | |
57 | ||
58 | void AliL3Merger::DeleteArray(){ | |
59 | for(Int_t i=0; i<fNIn;i++) | |
60 | delete fInTrack[i]; | |
61 | delete[] (Byte_t*) fInTrack; | |
62 | delete fOutTrack; | |
63 | } | |
64 | ||
65 | void AliL3Merger::SetArray(Int_t nin){ | |
66 | fNIn = nin; | |
67 | fInTrack = (AliL3TrackArray **) new Byte_t[fNIn*sizeof(AliL3TrackArray *)]; | |
68 | for(Int_t i=0; i<fNIn;i++){ | |
5dcd65ea | 69 | if(fTrackType=='h') |
70 | fInTrack[i] = new AliL3TrackArray("AliL3HoughTrack"); | |
71 | else | |
72 | fInTrack[i] = new AliL3TrackArray("AliL3Track"); | |
73 | ||
108615fc | 74 | } |
5dcd65ea | 75 | if(fTrackType=='h') |
76 | fOutTrack= new AliL3TrackArray("AliL3HoughTrack"); | |
77 | else | |
78 | fOutTrack= new AliL3TrackArray("AliL3Track"); | |
108615fc | 79 | } |
80 | ||
81 | void AliL3Merger::Reset(){ | |
82 | for(Int_t i=0; i<fNIn;i++){ | |
83 | fInTrack[i]->Reset(); | |
84 | } | |
85 | fOutTrack->Reset(); | |
86 | } | |
87 | ||
88 | void AliL3Merger::FillTracks(Int_t ntracks, AliL3TrackSegmentData* tr){ | |
89 | //Read tracks from shared memory (or memory) | |
90 | AliL3TrackArray *destination = GetInTracks(fCurrentTracks); | |
91 | if(Is2Global()) | |
494fad94 | 92 | destination->FillTracks(ntracks, tr, fSlice); |
93 | //destination->FillTracks(ntracks, tr, fSlice, fTransformer); | |
108615fc | 94 | else |
95 | destination->FillTracks(ntracks, tr); | |
96 | } | |
97 | ||
98 | void AliL3Merger::AddAllTracks(){ | |
99 | for(Int_t i=0; i<GetNIn();i++){ | |
100 | AliL3TrackArray *in = GetInTracks(i); | |
101 | AliL3TrackArray *out = GetOutTracks(); | |
102 | out->AddTracks(in); | |
103 | } | |
104 | } | |
105 | ||
106 | void AliL3Merger::SortGlobalTracks(AliL3Track **tracks, Int_t ntrack){ | |
107 | AliL3Track **tmp = new AliL3Track*[ntrack]; | |
108 | for(Int_t i=0;i<ntrack;i++) tmp[i] = tracks[i]; | |
109 | Int_t *t = new Int_t[ntrack]; | |
110 | for(Int_t i=0;i<ntrack;i++) t[i]=-1; | |
111 | ||
112 | for(Int_t j=0;j<ntrack;j++){ | |
113 | Double_t minr=300; | |
114 | Int_t mini=0; | |
115 | for(Int_t i=0;i<ntrack;i++){ | |
116 | if(!tracks[i]) continue; | |
117 | Double_t rr=pow(tracks[i]->GetFirstPointX(),2)+pow(tracks[i]->GetFirstPointY(),2); | |
118 | Double_t r=sqrt(rr); | |
119 | if(r<minr){ | |
120 | minr=r; | |
121 | mini=i; | |
122 | } | |
123 | } | |
124 | t[j]=mini; | |
125 | tracks[mini]=0; | |
126 | } | |
127 | for(Int_t i=0;i<ntrack;i++) tracks[i] = tmp[t[i]]; | |
128 | delete[] t; | |
129 | delete[] tmp; | |
130 | } | |
131 | ||
132 | ||
133 | void AliL3Merger::SortTracks(AliL3Track **tracks, Int_t ntrack){ | |
134 | AliL3Track **tmp = new AliL3Track*[ntrack]; | |
135 | for(Int_t i=0;i<ntrack;i++) tmp[i] = tracks[i]; | |
136 | Int_t *t = new Int_t[ntrack]; | |
137 | for(Int_t i=0;i<ntrack;i++) t[i]=-1; | |
138 | ||
139 | for(Int_t j=0;j<ntrack;j++){ | |
140 | Double_t minx=300; | |
141 | Int_t mini=0; | |
142 | for(Int_t i=0;i<ntrack;i++){ | |
143 | if(!tracks[i]) continue; | |
144 | if(tracks[i]->GetFirstPointX()<minx){ | |
145 | minx=tracks[i]->GetFirstPointX(); | |
146 | mini=i; | |
147 | } | |
148 | } | |
149 | t[j]=mini; | |
150 | tracks[mini]=0; | |
151 | } | |
152 | for(Int_t i=0;i<ntrack;i++) tracks[i] = tmp[t[i]]; | |
153 | delete[] t; | |
154 | delete[] tmp; | |
155 | } | |
156 | ||
157 | void AliL3Merger::AddTrack(AliL3TrackArray *mergedtrack,AliL3Track *track){ | |
158 | AliL3Track *t[1]; | |
159 | t[0] = track; | |
160 | MultiMerge(mergedtrack,t,1); | |
161 | } | |
162 | ||
163 | AliL3Track * AliL3Merger::MergeTracks(AliL3TrackArray *mergedtrack,AliL3Track *t0,AliL3Track *t1){ | |
164 | AliL3Track *t[2]; | |
165 | t[0] = t0; | |
166 | t[1] = t1; | |
167 | SortTracks(t,2); | |
168 | return MultiMerge(mergedtrack,t,2); | |
169 | } | |
170 | ||
171 | AliL3Track * AliL3Merger::MultiMerge(AliL3TrackArray *mergedtracks,AliL3Track **tracks, Int_t ntrack){ | |
172 | // merge the tracks!! | |
173 | ||
174 | //check npoints | |
175 | Int_t nps = 0; | |
176 | for(Int_t i=0;i<ntrack;i++){ | |
177 | nps+=tracks[i]->GetNHits(); | |
178 | } | |
68a27388 | 179 | if(nps>AliL3Transform::GetNRows()){ |
108615fc | 180 | LOG(AliL3Log::kWarning,"AliL3Merger::MultiMerge","Adding Points") |
181 | <<AliL3Log::kDec<<"Too many Points: "<<nps<<ENDLOG; | |
182 | return 0; | |
183 | } | |
184 | ||
185 | //create new track | |
186 | AliL3Track *newtrack = mergedtracks->NextTrack(); | |
187 | //copy points | |
68a27388 | 188 | UInt_t nn[AliL3Transform::GetNRows()]; |
108615fc | 189 | nps = 0; |
190 | ||
191 | // for(Int_t i=0;i<ntrack;i++){ | |
192 | for(Int_t i=ntrack-1;i>=0;i--){ | |
193 | memcpy(&nn[nps],tracks[i]->GetHitNumbers(),tracks[i]->GetNHits()*sizeof(UInt_t)); | |
194 | nps+=tracks[i]->GetNHits(); | |
195 | } | |
196 | AliL3Track *tpf=tracks[0]; | |
197 | AliL3Track *tpl=tracks[ntrack-1]; | |
4603aa2f | 198 | AliL3Track *best = tpf; |
199 | if(tpf->GetNHits()<tpl->GetNHits() && Is2Global()) | |
200 | best = tpl;//Best means = most points and therefore best fit (in global case) | |
201 | ||
108615fc | 202 | newtrack->SetNHits(nps); |
203 | newtrack->SetHits(nps,nn); | |
204 | newtrack->SetFirstPoint(tpf->GetFirstPointX(),tpf->GetFirstPointY(),tpf->GetFirstPointZ()); | |
205 | newtrack->SetLastPoint(tpl->GetLastPointX(),tpl->GetLastPointY(),tpl->GetLastPointZ()); | |
4603aa2f | 206 | newtrack->SetPt(best->GetPt()); |
207 | newtrack->SetPsi(best->GetPsi()); | |
208 | newtrack->SetTgl(best->GetTgl()); | |
108615fc | 209 | newtrack->SetCharge(tpf->GetCharge()); |
210 | return newtrack; | |
211 | } | |
212 | ||
213 | void* AliL3Merger::GetNtuple(char *varlist){ | |
214 | #ifdef use_root | |
215 | TNtuple* nt = new TNtuple("ntuple","ntuple",varlist); | |
216 | return (void*) nt; | |
217 | #else | |
218 | return 0; | |
219 | #endif | |
220 | } | |
221 | ||
222 | void* AliL3Merger::GetNtuple(){ | |
223 | #ifdef use_root | |
224 | TNtuple* nt = new TNtuple("ntuple","ntuple", | |
68a27388 | 225 | "dx:dy:dz:dk:dpsi:dtgl:dq:disx:disy:disz:dis:n0:n1:diff:drx:dry:drz"); |
108615fc | 226 | return (void*) nt; |
227 | #else | |
228 | return 0; | |
229 | #endif | |
230 | } | |
231 | ||
232 | Bool_t AliL3Merger::WriteNtuple(char *filename, void* nt){ | |
233 | #ifdef use_root | |
234 | TNtuple *ntuple=(TNtuple *) nt; | |
235 | TFile *f = new TFile(filename,"RECREATE"); | |
236 | ntuple->Write(); | |
237 | f->Close(); | |
238 | delete ntuple; | |
239 | return kTRUE; | |
240 | #else | |
241 | return kFALSE; | |
242 | #endif | |
243 | } | |
244 | ||
245 | void AliL3Merger::FillNtuple(void *nt,AliL3Track *innertrack,AliL3Track *outertrack){ | |
246 | Float_t data[17]; | |
247 | if(outertrack->IsPoint()&&innertrack->IsPoint()){ | |
248 | data[0] =Float_t(innertrack->GetPointX()-outertrack->GetPointX()); | |
249 | data[1] =Float_t(innertrack->GetPointY()-outertrack->GetPointY()); | |
250 | data[2] =Float_t(innertrack->GetPointZ()-outertrack->GetPointZ()); | |
251 | data[3] =Float_t(innertrack->GetKappa()-outertrack->GetKappa()); | |
252 | Double_t psi= innertrack->GetPointPsi() - outertrack->GetPointPsi(); | |
253 | if(psi>PI) psi-=2*PI; | |
254 | if(psi<-PI)psi+=2*PI; | |
255 | data[4] =Float_t(psi); | |
256 | data[5] =Float_t(innertrack->GetTgl()-outertrack->GetTgl()); | |
257 | data[6] =Float_t(innertrack->GetCharge()-outertrack->GetCharge()); | |
258 | data[7] =Float_t(innertrack->GetLastPointX()-outertrack->GetFirstPointX()); | |
259 | data[8] =Float_t(innertrack->GetLastPointY()-outertrack->GetFirstPointY()); | |
260 | data[9] =Float_t(innertrack->GetLastPointZ()-outertrack->GetFirstPointZ()); | |
261 | data[10] =sqrt(pow(data[7],2)+pow(data[8],2)+pow(data[9],2)); | |
262 | data[11]= outertrack->GetNHits(); | |
263 | data[12]= innertrack->GetNHits(); | |
264 | data[13] = Float_t(TrackDiff(innertrack,outertrack)); | |
68a27388 | 265 | data[14]=0; |
266 | data[15]=0; | |
267 | data[16]=0; | |
108615fc | 268 | #ifdef use_root |
269 | TNtuple *ntuple = (TNtuple *) nt; | |
270 | ntuple->Fill(data); | |
271 | #endif | |
272 | } | |
273 | } | |
274 | ||
275 | void AliL3Merger::FillNtuple(void *nt,Float_t *data){ | |
276 | #ifdef use_root | |
277 | TNtuple *ntuple = (TNtuple *) nt; | |
278 | ntuple->Fill(data); | |
279 | #endif | |
280 | } | |
281 | ||
282 | Double_t AliL3Merger::GetAngle(Double_t a1,Double_t a2){ | |
283 | Double_t da = a1 - a2 +4*PI; | |
284 | da = fmod(da,2*PI); | |
285 | if(da>PI) da = 2*PI -da; | |
286 | return da; | |
287 | } | |
288 | ||
289 | void AliL3Merger::SetParameter(Double_t maxy, Double_t maxz, Double_t maxkappa, Double_t maxpsi, Double_t maxtgl){ | |
290 | fMaxY = maxy; | |
291 | fMaxZ = maxz; | |
292 | fMaxKappa = maxkappa; | |
293 | fMaxPsi = maxpsi; | |
294 | fMaxTgl = maxtgl; | |
295 | } | |
296 | ||
297 | Bool_t AliL3Merger::IsTrack(AliL3Track *innertrack,AliL3Track *outertrack){ | |
298 | ||
299 | if(innertrack->GetCharge()!=outertrack->GetCharge()) return kFALSE; | |
300 | if( (!innertrack->IsPoint()) || (!outertrack->IsPoint()) ) return kFALSE; | |
68a27388 | 301 | if(innertrack->GetNHits()+outertrack->GetNHits()>AliL3Transform::GetNRows()) return kFALSE; |
108615fc | 302 | |
303 | if(fabs(innertrack->GetPointY()-outertrack->GetPointY()) >fMaxY) return kFALSE; | |
304 | if(fabs(innertrack->GetPointZ()-outertrack->GetPointZ()) >fMaxZ) return kFALSE; | |
305 | if(fabs(innertrack->GetKappa()-outertrack->GetKappa()) >fMaxKappa) return kFALSE; | |
306 | if(GetAngle(innertrack->GetPointPsi(),outertrack->GetPointPsi()) >fMaxPsi) return kFALSE; | |
307 | if(fabs(innertrack->GetTgl()-outertrack->GetTgl()) >fMaxTgl) return kFALSE; | |
308 | //if no rejection up to this point: merge!! | |
309 | return kTRUE; | |
310 | } | |
311 | ||
312 | Bool_t AliL3Merger::IsRTrack(AliL3Track *innertrack,AliL3Track *outertrack){ | |
313 | return IsTrack(innertrack,outertrack); | |
314 | } | |
315 | ||
316 | Double_t AliL3Merger::TrackDiff(AliL3Track *innertrack,AliL3Track *outertrack){ | |
317 | Double_t diff =-1; | |
318 | Double_t x[4],y[4],z[4],dy[4],dz[4]; | |
319 | AliL3Track *tracks[2]; | |
320 | ||
321 | tracks[0] = innertrack; | |
322 | tracks[1] = outertrack; | |
323 | SortGlobalTracks(tracks,2); | |
324 | innertrack = tracks[0]; | |
325 | outertrack = tracks[1]; | |
4603aa2f | 326 | |
108615fc | 327 | x[0] = innertrack->GetFirstPointX(); |
328 | x[1] = innertrack->GetLastPointX(); | |
329 | x[2] = outertrack->GetFirstPointX(); | |
330 | x[3] = outertrack->GetLastPointX(); | |
331 | ||
332 | y[0] = innertrack->GetFirstPointY(); | |
333 | y[1] = innertrack->GetLastPointY(); | |
334 | y[2] = outertrack->GetFirstPointY(); | |
335 | y[3] = outertrack->GetLastPointY(); | |
336 | ||
337 | z[0] = innertrack->GetFirstPointZ(); | |
338 | z[1] = innertrack->GetLastPointZ(); | |
339 | z[2] = outertrack->GetFirstPointZ(); | |
340 | z[3] = outertrack->GetLastPointZ(); | |
341 | ||
4603aa2f | 342 | |
108615fc | 343 | outertrack->CalculatePoint(x[0]); |
344 | if(!outertrack->IsPoint()) return diff; | |
345 | dy[0] = fabs(y[0] - outertrack->GetPointY()); | |
346 | dz[0] = fabs(z[0] - outertrack->GetPointZ()); | |
4603aa2f | 347 | |
108615fc | 348 | outertrack->CalculatePoint(x[1]); |
349 | if(!outertrack->IsPoint()) return diff; | |
350 | dy[1] = fabs(y[1] - outertrack->GetPointY()); | |
351 | dz[1] = fabs(z[1] - outertrack->GetPointZ()); | |
4603aa2f | 352 | |
108615fc | 353 | innertrack->CalculatePoint(x[2]); |
354 | if(!innertrack->IsPoint()) return diff; | |
355 | dy[2] = fabs(y[2] - innertrack->GetPointY()); | |
356 | dz[2] = fabs(z[2] - innertrack->GetPointZ()); | |
4603aa2f | 357 | |
108615fc | 358 | innertrack->CalculatePoint(x[3]); |
359 | if(!innertrack->IsPoint()) return diff; | |
360 | dy[3] = fabs(y[3] - innertrack->GetPointY()); | |
361 | dz[3] = fabs(z[3] - innertrack->GetPointZ()); | |
362 | ||
363 | diff=0; | |
364 | for(Int_t i=0;i<4;i++) | |
365 | diff+=sqrt(dy[i]*dy[i]+dz[i]*dz[i]); | |
366 | return diff; | |
367 | } | |
368 | ||
c3dd27a3 | 369 | void AliL3Merger::PrintDiff(AliL3Track *innertrack,AliL3Track *outertrack) |
370 | { | |
108615fc | 371 | if(!innertrack->IsPoint()||!outertrack->IsPoint()){ |
c3dd27a3 | 372 | LOG(AliL3Log::kInformational,"AliL3Merger::PrintDiff","No Points")<<ENDLOG; |
373 | //cerr<<"AliL3Merger::PrintDiff: No Points"<<endl; | |
374 | //cerr<<"---------------------------"<<endl; | |
108615fc | 375 | return; |
376 | } | |
c3dd27a3 | 377 | |
108615fc | 378 | Double_t dx = innertrack->GetPointX()-outertrack->GetPointX(); |
379 | Double_t dy = innertrack->GetPointY()-outertrack->GetPointY(); | |
380 | Double_t dz = innertrack->GetPointZ()-outertrack->GetPointZ(); | |
381 | Double_t dk = innertrack->GetKappa()-outertrack->GetKappa(); | |
c3dd27a3 | 382 | Double_t dpsi= innertrack->GetPointPsi() - outertrack->GetPointPsi(); |
383 | if(dpsi>PI) dpsi-=2*PI; | |
384 | if(dpsi<-PI)dpsi+=2*PI; | |
385 | //Double_t dpsi = GetAngle(innertrack->GetPointPsi(),outertrack->GetPointPsi()); | |
108615fc | 386 | Double_t dtgl= innertrack->GetTgl()-outertrack->GetTgl(); |
387 | Double_t dq =innertrack->GetCharge()-outertrack->GetCharge(); | |
388 | ||
c3dd27a3 | 389 | LOG(AliL3Log::kInformational,"AliL3Merger::PrintDiff","Points") <<"dx: "<<dx<<" dy: "<<dy<<" dz: "<<dz<<" dk: "<<dk<<" dpsi: "<<dpsi<<" dtgl: "<<dtgl<<" dq: "<<dq<<ENDLOG; |
390 | //fprintf(stderr,"dx: %4f dy: %4f dz: %4f dk: %4f dpsi: %4f dtgl: %4f dq: %4f\n",dx,dy,dz,dk,dpsi,dtgl,dq); | |
391 | //cerr<<"---------------------------"<<endl; | |
392 | ||
108615fc | 393 | } |
394 | ||
395 | void AliL3Merger::Print(){ | |
396 | // print some infos | |
397 | for(Int_t i=0; i<fNIn; i++){ | |
398 | AliL3TrackArray *ttt= GetInTracks(i); | |
399 | for(Int_t j =0;j<ttt->GetNTracks();j++){ | |
400 | AliL3Track *track=ttt->GetCheckedTrack(j); | |
401 | if(!track) continue; | |
402 | track->CalculateHelix(); | |
403 | // Double_t angle = atan2(track->GetLastPointY(),track->GetLastPointX()); | |
404 | // if(angle<0) angle+=PI; | |
405 | if(track->CalculatePoint(135)) | |
406 | // if(!track->CalculateEdgePoint(angle)) cerr<<"**************"<<endl; | |
407 | // if(track->CalculatePoint(track->GetLastPointX())) | |
408 | // if(track->CalculatePoint(0)) | |
409 | { | |
410 | // PrintTrack(track); | |
411 | // track->CalculateReferencePoint(PI/180.); | |
412 | track->CalculateReferencePoint(0.001); | |
c3dd27a3 | 413 | Float_t dx=(float)track->GetPointX()-track->GetPointX(); |
414 | Float_t dy=(float)track->GetPointY()-track->GetPointY(); | |
415 | Float_t dz=(float)track->GetPointZ()-track->GetPointZ(); | |
416 | LOG(AliL3Log::kInformational,"AliL3Merger::Print","RefPoint") <<"npt: "<<track->GetNHits()<<" dx: "<<dx<<" dy: "<<dy<<" dz: "<<dz<<ENDLOG; | |
108615fc | 417 | |
c3dd27a3 | 418 | //fprintf(stderr,"npt: %3d dx: %8.5f dy: %8.5f dz: %8.5f\n",track->GetNHits(),dx,dy,dz); |
419 | //cerr<<"---------------------------"<<endl; | |
108615fc | 420 | } |
421 | } | |
422 | } | |
423 | } | |
424 | ||
425 | void AliL3Merger::PrintTrack(AliL3Track *track){ | |
426 | fprintf(stderr,"npt: %3d pt: %.2f psi: %.2f tgl: %5.2f q: %2d\n", | |
427 | track->GetNHits(),track->GetPt(),track->GetPsi(), | |
428 | track->GetTgl(),track->GetCharge()); | |
429 | fprintf(stderr, | |
430 | "x1: %6.2f y1: %6.2f z1: %6.2f xl: %6.2f yl: %6.2f zl: %6.2f\n", | |
431 | track->GetFirstPointX(),track->GetFirstPointY(),track->GetFirstPointZ(), | |
432 | track->GetLastPointX(),track->GetLastPointY(),track->GetLastPointZ()); | |
433 | if(track->IsPoint()){ | |
434 | fprintf(stderr, | |
435 | "R: %.2f Xc: %.2f Yc: %.2f Xp: %.2f Yp: %.2f Zp: %.2f Psip: %.2f\n", | |
436 | track->GetRadius(),track->GetCenterX(),track->GetCenterY(), | |
437 | track->GetPointX(),track->GetPointY(),track->GetPointZ(), | |
438 | track->GetPointPsi()); | |
439 | } | |
440 | } |