HLT/src/AliHLTInterMerger.cxx

   1 // @(#) $Id$
   2
   3 // Author: Uli Frankenfeld <mailto:franken@fi.uib.no>
   4 //*-- Copyright &copy ALICE HLT Group
   5
   6 #include "AliHLTStandardIncludes.h"
   7
   8 #include "AliHLTLogging.h"
   9 #include "AliHLTInterMerger.h"
  10 #include "AliHLTTrack.h"
  11 #include "AliHLTTrackSegmentData.h"
  12 #include "AliHLTTransform.h"
  13 #include "AliHLTTrackArray.h"
  14
  15 /** \class AliHLTInterMerger
  16 <pre>
  17 //_____________________________________________________________
  18 // AliHLTInterMerger
  19 //
  20 // The L3 track segment merger
  21 //
  22 </pre>
  23 */
  24
  25 #if __GNUC__ >= 3
  26 using namespace std;
  27 #endif
  28
  29 ClassImp(AliHLTInterMerger)
  30
  31 AliHLTInterMerger::AliHLTInterMerger()
  32 {
  33   //Default constructor
  34   InitMerger(1);
  35   Is2Global(kFALSE);
  36 //  SetParameter(2,2,.3,.3,.3);
  37   SetParameter(1,0.5,0.0005,0.05,0.1);
  38   fRowMax = fRowMin = 0;
  39 }
  40
  41
  42 AliHLTInterMerger::~AliHLTInterMerger(){
  43   //Destructor
  44
  45 }
  46
  47 void AliHLTInterMerger::SlowMerge(){
  48   Int_t nrow= fRowMax-fRowMin+1;
  49   void *ntuple=GetNtuple();
  50   AliHLTTrackArray * tracks = GetInTracks(0);
  51   const Int_t  kNIn =tracks->GetNTracks();
  52   AliHLTTrack *tr[2];
  53   Bool_t merge = kTRUE;
  54   for(Int_t in=0;in<kNIn;in++)
  55     tracks->GetCheckedTrack(in)->CalculateHelix();
  56   while(merge){
  57     Int_t inmin=-1,outmin=-1;
  58     Double_t min=10;
  59     for(Int_t out=0;out<kNIn;out++){
  60     AliHLTTrack *outertrack=tracks->GetCheckedTrack(out);
  61     if(!outertrack) continue;
  62       for(Int_t in=0;in<kNIn;in++){
  63         if(in==out) continue;
  64         AliHLTTrack *innertrack=tracks->GetCheckedTrack(in);
  65         if(!innertrack) continue;
  66         if(outertrack->GetNHits()+innertrack->GetNHits()>nrow) continue;
  67
  68         Double_t diff = TrackDiff(innertrack,outertrack);
  69
  70         if(diff>=0&&diff<min){
  71           min=diff;
  72           inmin=in;
  73           outmin=out;
  74         }
  75       }
  76     }
  77     if(inmin>=0&&outmin>=0){
  78       AliHLTTrack *outertrack=tracks->GetTrack(outmin);
  79       AliHLTTrack *innertrack=tracks->GetTrack(inmin);
  80       tr[0]=innertrack;
  81       tr[1]=outertrack;
  82       SortTracks(tr,2);
  83       MultiMerge(tracks,tr,2);
  84       outertrack->CalculatePoint(tr[0]->GetLastPointX());
  85       innertrack->CalculatePoint(tr[0]->GetLastPointX());
  86       PrintDiff(innertrack,outertrack);
  87       FillNtuple(ntuple,innertrack,outertrack);
  88       tracks->Remove(outmin);
  89       tracks->Remove(inmin);
  90     }
  91     else merge = kFALSE;
  92   }
  93   LOG(AliHLTLog::kInformational,"AliHLTInterMerger::SlowMerge","Result")
  94   <<AliHLTLog::kDec<<"Merged Tracks: "<<tracks->GetNTracks()-kNIn<<ENDLOG;
  95
  96   char name[256];
  97   sprintf(name,"ntuple_i_%d.root",fPatch);
  98   WriteNtuple(name,ntuple);
  99 }
 100
 101 void AliHLTInterMerger::MMerge(){
 102   while(Merge());
 103   GetOutTracks()->AddTracks(GetInTracks(0));
 104 }
 105
 106 Int_t AliHLTInterMerger::Merge(){
 107   Int_t nrow= fRowMax-fRowMin+1;
 108   Double_t xval =AliHLTTransform::Row2X((fRowMax+fRowMin)/2);
 109   AliHLTTrackArray * tracks = GetInTracks(0);
 110   const Int_t  kNIn =tracks->GetNTracks();
 111   AliHLTTrack *tr[2];
 112   for(Int_t in=0;in<kNIn;in++){
 113     AliHLTTrack *t = tracks->GetCheckedTrack(in);
 114     if(t){
 115       t->CalculateHelix();
 116       t->CalculatePoint(xval);
 117     }
 118   }
 119   for(Int_t out=0;out<kNIn;out++){
 120   AliHLTTrack *outertrack=tracks->GetCheckedTrack(out);
 121   if(!outertrack) continue;
 122     for(Int_t in=0;in<kNIn;in++){
 123       if(in==out) continue;
 124       AliHLTTrack *innertrack=tracks->GetCheckedTrack(in);
 125       if(!innertrack) continue;
 126       if(outertrack->GetNHits()+innertrack->GetNHits()>nrow) continue;
 127
 128       if(IsTrack(innertrack,outertrack)){
 129         tr[0]=innertrack;
 130         tr[1]=outertrack;
 131         SortGlobalTracks(tr,2);
 132
 133         Double_t r0 = pow(tr[0]->GetLastPointX(),2)+
 134                       pow(tr[0]->GetLastPointY(),2);
 135         Double_t r1 = pow(tr[1]->GetFirstPointX(),2)+
 136                       pow(tr[1]->GetFirstPointY(),2);
 137         if(r0<r1){
 138           MultiMerge(tracks,tr,2);
 139           tracks->Remove(out);
 140           tracks->Remove(in);
 141           break;
 142         }
 143       }
 144     }
 145   }
 146   Int_t nmerged = tracks->GetNTracks()-kNIn;
 147   LOG(AliHLTLog::kInformational,"AliHLTInterMerger::Merge","Result")
 148   <<AliHLTLog::kDec<<"Merged Tracks: "<<nmerged<<ENDLOG;
 149   //add in tracks
 150 //  GetOutTracks()->AddTracks(GetInTracks(0));
 151
 152   return nmerged;
 153 }
 154
 155