HLT/src/AliL3Merger.cxx

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