[u/mrichter/AliRoot.git] / HLT / src / AliL3Merger.cxx

//$Id$

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

#include "AliL3StandardIncludes.h"

#include "AliL3Logging.h"
#include "AliL3Merger.h"
#include "AliL3Track.h"
#include "AliL3TrackSegmentData.h"
#include "AliL3Transform.h"
#include "AliL3TrackArray.h"

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

//_____________________________________________________________
// AliL3Merger
//
// The L3 merger base class
//

ClassImp(AliL3Merger)

AliL3Merger::AliL3Merger(){
  //Default constructor
  SetArray(0);
}


AliL3Merger::AliL3Merger(Int_t ntrackarrays,Char_t *tracktype){
  //Constructor.
  
  if(strcmp(tracktype,"AliL3Track")==0) fTrackType='t';
  else if(strcmp(tracktype,"AliL3ConfMapTrack")==0) fTrackType='c';
  else if(strcmp(tracktype,"AliL3HoughTrack")==0) fTrackType='h';
  else
    LOG(AliL3Log::kError,"AliL3Merger::AliL3Merger","Track types")
      <<"Unknown tracktype"<<ENDLOG;
  SetArray(ntrackarrays);
  fCurrentTracks=0;

}

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

void AliL3Merger::DeleteArray(){
  for(Int_t i=0; i<fNIn;i++)
    delete fInTrack[i];
  delete[] (Byte_t*) fInTrack;
  delete fOutTrack;
}

void AliL3Merger::SetArray(Int_t nin){
  fNIn = nin;
  fInTrack = (AliL3TrackArray **) new Byte_t[fNIn*sizeof(AliL3TrackArray *)];
  for(Int_t i=0; i<fNIn;i++){
    if(fTrackType=='h')
      fInTrack[i] = new AliL3TrackArray("AliL3HoughTrack");
    else
      fInTrack[i] = new AliL3TrackArray("AliL3Track");
    
  }
  if(fTrackType=='h')
    fOutTrack= new AliL3TrackArray("AliL3HoughTrack");
  else
    fOutTrack= new AliL3TrackArray("AliL3Track");
}

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

void AliL3Merger::FillTracks(Int_t ntracks, AliL3TrackSegmentData* tr){
  //Read tracks from shared memory (or memory)
  AliL3TrackArray *destination = GetInTracks(fCurrentTracks);
  if(Is2Global())
    destination->FillTracks(ntracks, tr, fSlice);
    //destination->FillTracks(ntracks, tr, fSlice, fTransformer);
  else
    destination->FillTracks(ntracks, tr);
}

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

void AliL3Merger::SortGlobalTracks(AliL3Track **tracks, Int_t ntrack){
  AliL3Track **tmp = new AliL3Track*[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 AliL3Merger::SortTracks(AliL3Track **tracks, Int_t ntrack){
  AliL3Track **tmp = new  AliL3Track*[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 AliL3Merger::AddTrack(AliL3TrackArray *mergedtrack,AliL3Track *track){
  AliL3Track *t[1];
  t[0] = track;
  MultiMerge(mergedtrack,t,1);
}

AliL3Track * AliL3Merger::MergeTracks(AliL3TrackArray *mergedtrack,AliL3Track *t0,AliL3Track *t1){
  AliL3Track *t[2];
  t[0] = t0; 
  t[1] = t1;
  SortTracks(t,2);
  return MultiMerge(mergedtrack,t,2);
}

AliL3Track * AliL3Merger::MultiMerge(AliL3TrackArray *mergedtracks,AliL3Track **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>AliL3Transform::GetNRows()){
    LOG(AliL3Log::kWarning,"AliL3Merger::MultiMerge","Adding Points")
    <<AliL3Log::kDec<<"Too many Points: "<<nps<<ENDLOG;
    return 0;
  }

  //create new track
  AliL3Track *newtrack = mergedtracks->NextTrack();
  //copy points
  UInt_t nn[AliL3Transform::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();
  }
  AliL3Track *tpf=tracks[0];
  AliL3Track *tpl=tracks[ntrack-1];
  AliL3Track *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* AliL3Merger::GetNtuple(char *varlist){
#ifdef use_root
  TNtuple* nt = new TNtuple("ntuple","ntuple",varlist);
  return (void*) nt;
#else
  return 0;
#endif
}

void* AliL3Merger::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 AliL3Merger::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 AliL3Merger::FillNtuple(void *nt,AliL3Track *innertrack,AliL3Track *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 AliL3Merger::FillNtuple(void *nt,Float_t *data){
#ifdef use_root
    TNtuple *ntuple = (TNtuple *) nt;
    ntuple->Fill(data);
#endif
}

Double_t AliL3Merger::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 AliL3Merger::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 AliL3Merger::IsTrack(AliL3Track *innertrack,AliL3Track *outertrack){

  if(innertrack->GetCharge()!=outertrack->GetCharge()) return kFALSE;
  if( (!innertrack->IsPoint()) || (!outertrack->IsPoint()) )  return kFALSE; 
  if(innertrack->GetNHits()+outertrack->GetNHits()>AliL3Transform::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 AliL3Merger::IsRTrack(AliL3Track *innertrack,AliL3Track *outertrack){
  return IsTrack(innertrack,outertrack);
}

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