[u/mrichter/AliRoot.git] / HLT / TPCLib / Ref / AliHLTTPCMerger.cxx

//$Id$

// Author: Uli Frankenfeld <mailto:franken@fi.uib.no>
//*-- Copyright &copy Uli 

#include "AliHLTTPCStandardIncludes.h"

#include "AliHLTTPCLogging.h"
#include "AliHLTTPCMerger.h"
#include "AliHLTTPCTrack.h"
#include "AliHLTTPCTrackSegmentData.h"
#include "AliHLTTPCTransform.h"
#include "AliHLTTPCTrackArray.h"

#ifdef use_root //use root ntuple for slow merge
#include <TNtuple.h>
#include <TTree.h>
#include <TFile.h>
#endif

/** \class AliHLTTPCMerger
<pre>
//_____________________________________________________________
// AliHLTTPCMerger
//
// The L3 merger base class
//
</pre>
*/

ClassImp(AliHLTTPCMerger)

AliHLTTPCMerger::AliHLTTPCMerger()
{
  //Default constructor
  fInTrack=0;
  fOutTrack=0;
  fCurrentTracks=0;
  fNIn=0;
}

AliHLTTPCMerger::~AliHLTTPCMerger()
{
  //Destructor
  DeleteArray();
}

void AliHLTTPCMerger::InitMerger(Int_t ntrackarrays,Char_t *tracktype)
{
  //Used to setup all arrays
  
  if(strcmp(tracktype,"AliHLTTPCTrack")==0) fTrackType='t';
  else if(strcmp(tracktype,"AliHLTTPCConfMapTrack")==0) fTrackType='c';
  else if(strcmp(tracktype,"AliHLTTPCHoughTrack")==0) fTrackType='h';
  else
    LOG(AliHLTTPCLog::kError,"AliHLTTPCMerger::AliHLTTPCMerger","Track types")
      <<"Unknown tracktype"<<ENDLOG;
  SetArray(ntrackarrays);
  fCurrentTracks=0;

}

void AliHLTTPCMerger::DeleteArray()
{
  for(Int_t i=0; i<fNIn;i++)
    {
      if(!fInTrack[i]) continue;
      delete fInTrack[i];
      fInTrack[i]=0;
    }
  if(fInTrack)
    delete[] fInTrack;
  if(fOutTrack)
    delete fOutTrack;
  fInTrack=0;
  fOutTrack=0;
}

void AliHLTTPCMerger::SetArray(Int_t nin)
{
  DeleteArray();//Make sure arrays are cleaned 
  
  fNIn = nin;
  fInTrack = new AliHLTTPCTrackArray*[fNIn];
  for(Int_t i=0; i<fNIn;i++)
    {
      if(fTrackType=='h')
	fInTrack[i] = new AliHLTTPCTrackArray("AliHLTTPCHoughTrack");
      else
	fInTrack[i] = new AliHLTTPCTrackArray("AliHLTTPCTrack");
      
    }
  if(fTrackType=='h')
    fOutTrack= new AliHLTTPCTrackArray("AliHLTTPCHoughTrack");
  else
    fOutTrack= new AliHLTTPCTrackArray("AliHLTTPCTrack");
}

void AliHLTTPCMerger::Reset()
{
  for(Int_t i=0; i<fNIn;i++)
    {
      fInTrack[i]->Reset();
    }
  fOutTrack->Reset();
}

void AliHLTTPCMerger::FillTracks(Int_t ntracks, AliHLTTPCTrackSegmentData* tr)
{
  //Read tracks from shared memory (or memory)

  AliHLTTPCTrackArray *destination = GetInTracks(fCurrentTracks);
  if(Is2Global())
    destination->FillTracks(ntracks, tr, fSlice);
  else
    destination->FillTracks(ntracks, tr);
}

void AliHLTTPCMerger::AddAllTracks()
{
  for(Int_t i=0; i<GetNIn();i++)
    {
      AliHLTTPCTrackArray *in = GetInTracks(i);
      AliHLTTPCTrackArray *out = GetOutTracks();
      out->AddTracks(in);
    }
}

void AliHLTTPCMerger::SortGlobalTracks(AliHLTTPCTrack **tracks, Int_t ntrack)
{
  AliHLTTPCTrack **tmp = new AliHLTTPCTrack*[ntrack]; 
  for(Int_t i=0;i<ntrack;i++) tmp[i] = tracks[i];
  Int_t *t = new Int_t[ntrack];
  for(Int_t i=0;i<ntrack;i++) t[i]=-1;
  
  for(Int_t j=0;j<ntrack;j++)
    {
      Double_t minr=300;
      Int_t    mini=0;
      for(Int_t i=0;i<ntrack;i++)
	{
	  if(!tracks[i]) continue;
	  Double_t rr=pow(tracks[i]->GetFirstPointX(),2)+pow(tracks[i]->GetFirstPointY(),2);
	  Double_t r=sqrt(rr);
	  if(r<minr){
	    minr=r;
	    mini=i;
	  }
	}
      t[j]=mini;
      tracks[mini]=0;
    }
  for(Int_t i=0;i<ntrack;i++) tracks[i] = tmp[t[i]];
  delete[] t;
  delete[] tmp;
}


void AliHLTTPCMerger::SortTracks(AliHLTTPCTrack **tracks, Int_t ntrack)
{
  AliHLTTPCTrack **tmp = new  AliHLTTPCTrack*[ntrack];
  for(Int_t i=0;i<ntrack;i++) tmp[i] = tracks[i];
  Int_t *t = new Int_t[ntrack];
  for(Int_t i=0;i<ntrack;i++) t[i]=-1;
  
  for(Int_t j=0;j<ntrack;j++)
    {
      Double_t minx=300; 
      Int_t    mini=0;
      for(Int_t i=0;i<ntrack;i++)
	{
	  if(!tracks[i]) continue;
	  if(tracks[i]->GetFirstPointX()<minx)
	    {
	      minx=tracks[i]->GetFirstPointX();
	      mini=i;
	    }     
	}
      t[j]=mini;  
      tracks[mini]=0;
    }
  for(Int_t i=0;i<ntrack;i++) tracks[i] = tmp[t[i]];
  delete[] t;
  delete[] tmp;
}

void AliHLTTPCMerger::AddTrack(AliHLTTPCTrackArray *mergedtrack,AliHLTTPCTrack *track)
{
  AliHLTTPCTrack *t[1];
  t[0] = track;
  MultiMerge(mergedtrack,t,1);
}

AliHLTTPCTrack * AliHLTTPCMerger::MergeTracks(AliHLTTPCTrackArray *mergedtrack,AliHLTTPCTrack *t0,AliHLTTPCTrack *t1)
{
  AliHLTTPCTrack *t[2];
  t[0] = t0; 
  t[1] = t1;
  SortTracks(t,2);
  return MultiMerge(mergedtrack,t,2);
}

AliHLTTPCTrack * AliHLTTPCMerger::MultiMerge(AliHLTTPCTrackArray *mergedtracks,AliHLTTPCTrack **tracks, Int_t ntrack)
{
  // merge the tracks!!
  
  //check npoints
  Int_t nps = 0;
  for(Int_t i=0;i<ntrack;i++)
    {
      nps+=tracks[i]->GetNHits();
    }
  if(nps>AliHLTTPCTransform::GetNRows())
    {
      LOG(AliHLTTPCLog::kWarning,"AliHLTTPCMerger::MultiMerge","Adding Points")
	<<AliHLTTPCLog::kDec<<"Too many Points: "<<nps<<ENDLOG;
      return 0;
    }
  
  //create new track
  AliHLTTPCTrack *newtrack = mergedtracks->NextTrack();
  //copy points
  UInt_t nn[AliHLTTPCTransform::GetNRows()];
  nps = 0;
  
  //  for(Int_t i=0;i<ntrack;i++){
  for(Int_t i=ntrack-1;i>=0;i--)
    {
      memcpy(&nn[nps],tracks[i]->GetHitNumbers(),tracks[i]->GetNHits()*sizeof(UInt_t));
      nps+=tracks[i]->GetNHits();
    }
  AliHLTTPCTrack *tpf=tracks[0];
  AliHLTTPCTrack *tpl=tracks[ntrack-1];
  AliHLTTPCTrack *best = tpf;
  if(tpf->GetNHits()<tpl->GetNHits() && Is2Global())
    best = tpl;//Best means = most points and therefore best fit (in global case)
  
  newtrack->SetNHits(nps);
  newtrack->SetHits(nps,nn);
  newtrack->SetFirstPoint(tpf->GetFirstPointX(),tpf->GetFirstPointY(),tpf->GetFirstPointZ());
  newtrack->SetLastPoint(tpl->GetLastPointX(),tpl->GetLastPointY(),tpl->GetLastPointZ());
  newtrack->SetPt(best->GetPt());
  newtrack->SetPsi(best->GetPsi());
  newtrack->SetTgl(best->GetTgl());
  newtrack->SetCharge(tpf->GetCharge());
  return newtrack;
}

void* AliHLTTPCMerger::GetNtuple(char *varlist)
{
#ifdef use_root
  TNtuple* nt = new TNtuple("ntuple","ntuple",varlist);
  return (void*) nt;
#else
  return 0;
#endif
}

void* AliHLTTPCMerger::GetNtuple()
{
#ifdef use_root
  TNtuple* nt = new TNtuple("ntuple","ntuple",
			    "dx:dy:dz:dk:dpsi:dtgl:dq:disx:disy:disz:dis:n0:n1:diff:drx:dry:drz");
  return (void*) nt;
#else
  return 0;
#endif
}

Bool_t AliHLTTPCMerger::WriteNtuple(char *filename, void* nt)
{
#ifdef use_root
  TNtuple *ntuple=(TNtuple *) nt;
  TFile *f = new TFile(filename,"RECREATE");
  ntuple->Write();
  f->Close();
  delete ntuple; 
  return kTRUE; 
#else
  return kFALSE;
#endif
}

void AliHLTTPCMerger::FillNtuple(void *nt,AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack)
{
  Float_t data[17];
  if(outertrack->IsPoint()&&innertrack->IsPoint())
    {
      data[0] =Float_t(innertrack->GetPointX()-outertrack->GetPointX());
      data[1] =Float_t(innertrack->GetPointY()-outertrack->GetPointY());
      data[2] =Float_t(innertrack->GetPointZ()-outertrack->GetPointZ());
      data[3] =Float_t(innertrack->GetKappa()-outertrack->GetKappa());
      Double_t psi= innertrack->GetPointPsi() - outertrack->GetPointPsi();
      if(psi>PI) psi-=2*PI;
      if(psi<-PI)psi+=2*PI;
      data[4] =Float_t(psi);
      data[5] =Float_t(innertrack->GetTgl()-outertrack->GetTgl());
      data[6] =Float_t(innertrack->GetCharge()-outertrack->GetCharge());
      data[7] =Float_t(innertrack->GetLastPointX()-outertrack->GetFirstPointX());
      data[8] =Float_t(innertrack->GetLastPointY()-outertrack->GetFirstPointY());
      data[9] =Float_t(innertrack->GetLastPointZ()-outertrack->GetFirstPointZ());
      data[10] =sqrt(pow(data[7],2)+pow(data[8],2)+pow(data[9],2));
      data[11]= outertrack->GetNHits();
      data[12]= innertrack->GetNHits();
      data[13] = Float_t(TrackDiff(innertrack,outertrack));
      data[14]=0;
      data[15]=0;
      data[16]=0;
#ifdef use_root
      TNtuple *ntuple = (TNtuple *) nt;
      ntuple->Fill(data);
#endif
    }
}

void AliHLTTPCMerger::FillNtuple(void *nt,Float_t *data)
{
#ifdef use_root
  TNtuple *ntuple = (TNtuple *) nt;
  ntuple->Fill(data);
#endif
}

Double_t AliHLTTPCMerger::GetAngle(Double_t a1,Double_t a2)
{
  Double_t da = a1 - a2 +4*PI;
  da = fmod(da,2*PI);
  if(da>PI) da = 2*PI -da;
  return da;
}

void AliHLTTPCMerger::SetParameter(Double_t maxy, Double_t maxz, Double_t maxkappa, Double_t maxpsi, Double_t maxtgl)
{
  fMaxY = maxy;
  fMaxZ = maxz;
  fMaxKappa = maxkappa;
  fMaxPsi = maxpsi;
  fMaxTgl = maxtgl;
}

Bool_t AliHLTTPCMerger::IsTrack(AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack)
{
  
  if(innertrack->GetCharge()!=outertrack->GetCharge()) return kFALSE;
  if( (!innertrack->IsPoint()) || (!outertrack->IsPoint()) )  return kFALSE; 
  if(innertrack->GetNHits()+outertrack->GetNHits()>AliHLTTPCTransform::GetNRows()) return kFALSE;
  
  if(fabs(innertrack->GetPointY()-outertrack->GetPointY()) >fMaxY) return kFALSE;
  if(fabs(innertrack->GetPointZ()-outertrack->GetPointZ()) >fMaxZ) return kFALSE;
  if(fabs(innertrack->GetKappa()-outertrack->GetKappa())   >fMaxKappa) return kFALSE;
  if(GetAngle(innertrack->GetPointPsi(),outertrack->GetPointPsi()) >fMaxPsi) return kFALSE;
  if(fabs(innertrack->GetTgl()-outertrack->GetTgl()) >fMaxTgl) return kFALSE;
  //if no rejection up to this point: merge!!
  return kTRUE;
}

Bool_t AliHLTTPCMerger::IsRTrack(AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack)
{
  return IsTrack(innertrack,outertrack);
}

Double_t AliHLTTPCMerger::TrackDiff(AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack)
{
  Double_t diff =-1;
  Double_t x[4],y[4],z[4],dy[4],dz[4];
  AliHLTTPCTrack *tracks[2]; 
  
  tracks[0] = innertrack;
  tracks[1] = outertrack;
  SortGlobalTracks(tracks,2);
  innertrack = tracks[0]; 
  outertrack = tracks[1];
  
  x[0] = innertrack->GetFirstPointX();
  x[1] = innertrack->GetLastPointX();
  x[2] = outertrack->GetFirstPointX();
  x[3] = outertrack->GetLastPointX();
  
  y[0] = innertrack->GetFirstPointY();
  y[1] = innertrack->GetLastPointY();
  y[2] = outertrack->GetFirstPointY();
  y[3] = outertrack->GetLastPointY();

  z[0] = innertrack->GetFirstPointZ();
  z[1] = innertrack->GetLastPointZ();
  z[2] = outertrack->GetFirstPointZ();
  z[3] = outertrack->GetLastPointZ();

  
  outertrack->CalculatePoint(x[0]);
  if(!outertrack->IsPoint()) return diff;
  dy[0] = fabs(y[0] - outertrack->GetPointY());
  dz[0] = fabs(z[0] - outertrack->GetPointZ());
  
  outertrack->CalculatePoint(x[1]);
  if(!outertrack->IsPoint()) return diff;
  dy[1] = fabs(y[1] - outertrack->GetPointY());
  dz[1] = fabs(z[1] - outertrack->GetPointZ());
  
  innertrack->CalculatePoint(x[2]);
  if(!innertrack->IsPoint()) return diff;
  dy[2] = fabs(y[2] - innertrack->GetPointY());
  dz[2] = fabs(z[2] - innertrack->GetPointZ());
  
  innertrack->CalculatePoint(x[3]);
  if(!innertrack->IsPoint()) return diff;
  dy[3] = fabs(y[3] - innertrack->GetPointY());
  dz[3] = fabs(z[3] - innertrack->GetPointZ());

  diff=0;
  for(Int_t i=0;i<4;i++)
    diff+=sqrt(dy[i]*dy[i]+dz[i]*dz[i]);
  return diff; 
}

void AliHLTTPCMerger::PrintDiff(AliHLTTPCTrack *innertrack,AliHLTTPCTrack *outertrack)
{
  if(!innertrack->IsPoint()||!outertrack->IsPoint())
    {
      LOG(AliHLTTPCLog::kInformational,"AliHLTTPCMerger::PrintDiff","No Points")<<ENDLOG;
      //cerr<<"AliHLTTPCMerger::PrintDiff: No Points"<<endl;
      //cerr<<"---------------------------"<<endl;
      return;
    } 
  
  Double_t dx = innertrack->GetPointX()-outertrack->GetPointX();
  Double_t dy = innertrack->GetPointY()-outertrack->GetPointY();
  Double_t dz = innertrack->GetPointZ()-outertrack->GetPointZ();
  Double_t dk = innertrack->GetKappa()-outertrack->GetKappa();
  Double_t dpsi= innertrack->GetPointPsi() - outertrack->GetPointPsi();
  if(dpsi>PI) dpsi-=2*PI;
  if(dpsi<-PI)dpsi+=2*PI;
  //Double_t dpsi = GetAngle(innertrack->GetPointPsi(),outertrack->GetPointPsi());
  Double_t dtgl= innertrack->GetTgl()-outertrack->GetTgl();
  Double_t dq =innertrack->GetCharge()-outertrack->GetCharge();
  
  LOG(AliHLTTPCLog::kInformational,"AliHLTTPCMerger::PrintDiff","Points") <<"dx: "<<dx<<" dy: "<<dy<<" dz: "<<dz<<" dk: "<<dk<<" dpsi: "<<dpsi<<" dtgl: "<<dtgl<<" dq: "<<dq<<ENDLOG;
  //fprintf(stderr,"dx: %4f dy: %4f dz: %4f dk: %4f dpsi: %4f dtgl: %4f dq: %4f\n",dx,dy,dz,dk,dpsi,dtgl,dq);
  //cerr<<"---------------------------"<<endl;
  
}

void AliHLTTPCMerger::Print()
{
  // print some infos
  for(Int_t i=0; i<fNIn; i++)
    {
      AliHLTTPCTrackArray *ttt= GetInTracks(i);
      for(Int_t j =0;j<ttt->GetNTracks();j++)
	{
	  AliHLTTPCTrack *track=ttt->GetCheckedTrack(j);
	  if(!track) continue;
	  track->CalculateHelix();
	  //      Double_t angle = atan2(track->GetLastPointY(),track->GetLastPointX());
	  //      if(angle<0) angle+=PI;
	  if(track->CalculatePoint(135))
	    //      if(!track->CalculateEdgePoint(angle)) cerr<<"**************"<<endl;     
	    //      if(track->CalculatePoint(track->GetLastPointX()))
	    //      if(track->CalculatePoint(0))
	    {
	      //      PrintTrack(track);
	      //      track->CalculateReferencePoint(PI/180.);
	      track->CalculateReferencePoint(0.001);
	      Float_t dx=(float)track->GetPointX()-track->GetPointX();
	      Float_t dy=(float)track->GetPointY()-track->GetPointY();
	      Float_t dz=(float)track->GetPointZ()-track->GetPointZ();
	      LOG(AliHLTTPCLog::kInformational,"AliHLTTPCMerger::Print","RefPoint") <<"npt: "<<track->GetNHits()<<" dx: "<<dx<<" dy: "<<dy<<" dz: "<<dz<<ENDLOG;
	      
	      //fprintf(stderr,"npt: %3d dx: %8.5f dy: %8.5f dz: %8.5f\n",track->GetNHits(),dx,dy,dz);
	      //cerr<<"---------------------------"<<endl;
	    }
	}  
    }
}

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