HLT/src/AliL3Merger.cxx

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