]> git.uio.no Git - u/mrichter/AliRoot.git/blame - HLT/src/AliL3Merger.cxx
Merged HLT tag v1-2 with ALIROOT tag v3-09-Release.
[u/mrichter/AliRoot.git] / HLT / src / AliL3Merger.cxx
CommitLineData
4603aa2f 1//$Id$
2
b661165c 3// Author: Uli Frankenfeld <mailto:franken@fi.uib.no>
4//*-- Copyright &copy 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
31ClassImp(AliL3Merger)
32
33AliL3Merger::AliL3Merger(){
34 //Default constructor
108615fc 35 SetArray(0);
36}
37
38
5dcd65ea 39AliL3Merger::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
53AliL3Merger::~AliL3Merger(){
54 //Destructor
55 DeleteArray();
56}
57
58void 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
65void 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
81void AliL3Merger::Reset(){
82 for(Int_t i=0; i<fNIn;i++){
83 fInTrack[i]->Reset();
84 }
85 fOutTrack->Reset();
86}
87
88void 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
98void 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
106void 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
133void 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
157void AliL3Merger::AddTrack(AliL3TrackArray *mergedtrack,AliL3Track *track){
158 AliL3Track *t[1];
159 t[0] = track;
160 MultiMerge(mergedtrack,t,1);
161}
162
163AliL3Track * 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
171AliL3Track * 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
213void* 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
222void* 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
232Bool_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
245void 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
275void AliL3Merger::FillNtuple(void *nt,Float_t *data){
276#ifdef use_root
277 TNtuple *ntuple = (TNtuple *) nt;
278 ntuple->Fill(data);
279#endif
280}
281
282Double_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
289void 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
297Bool_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
312Bool_t AliL3Merger::IsRTrack(AliL3Track *innertrack,AliL3Track *outertrack){
313 return IsTrack(innertrack,outertrack);
314}
315
316Double_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 369void 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
395void 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
425void 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}