[u/mrichter/AliRoot.git] / HLT / TPCLib / AliHLTTPCInterMerger.cxx

// @(#) $Id$
// Original: AliL3InterMerger.cxx,v 1.8 2005/06/14 10:55:21 cvetan

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

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

/** \class AliHLTTPCInterMerger
<pre>
//_____________________________________________________________
// AliHLTTPCInterMerger
//
// The HLTTPC track segment merger
//
</pre>
*/

#if __GNUC__ >= 3
using namespace std;
#endif

ClassImp(AliHLTTPCInterMerger)

AliHLTTPCInterMerger::AliHLTTPCInterMerger()
{
  //Default constructor
  InitMerger(1);
  Is2Global(kFALSE);
//  SetParameter(2,2,.3,.3,.3);
  SetParameter(1,0.5,0.0005,0.05,0.1);
  fRowMax = fRowMin = 0;
}


AliHLTTPCInterMerger::~AliHLTTPCInterMerger(){
  //Destructor
  
}

void AliHLTTPCInterMerger::SlowMerge(){
  Int_t nrow= fRowMax-fRowMin+1;
  void *ntuple=GetNtuple();
  AliHLTTPCTrackArray * tracks = GetInTracks(0);
  const Int_t  kNIn =tracks->GetNTracks();
  AliHLTTPCTrack *tr[2];
  Bool_t merge = kTRUE;
  for(Int_t in=0;in<kNIn;in++)
    tracks->GetCheckedTrack(in)->CalculateHelix();
  while(merge){
    Int_t inmin=-1,outmin=-1;
    Double_t min=10;
    for(Int_t out=0;out<kNIn;out++){
    AliHLTTPCTrack *outertrack=tracks->GetCheckedTrack(out);
    if(!outertrack) continue;
      for(Int_t in=0;in<kNIn;in++){
        if(in==out) continue;
        AliHLTTPCTrack *innertrack=tracks->GetCheckedTrack(in);
        if(!innertrack) continue;
        if(outertrack->GetNHits()+innertrack->GetNHits()>nrow) continue;

        Double_t diff = TrackDiff(innertrack,outertrack);

        if(diff>=0&&diff<min){
          min=diff;
          inmin=in;
          outmin=out; 
        }
      } 
    }
    if(inmin>=0&&outmin>=0){
      AliHLTTPCTrack *outertrack=tracks->GetTrack(outmin);
      AliHLTTPCTrack *innertrack=tracks->GetTrack(inmin);
      tr[0]=innertrack;
      tr[1]=outertrack;
      SortTracks(tr,2);
      MultiMerge(tracks,tr,2);
      outertrack->CalculatePoint(tr[0]->GetLastPointX());
      innertrack->CalculatePoint(tr[0]->GetLastPointX());
      PrintDiff(innertrack,outertrack);
      FillNtuple(ntuple,innertrack,outertrack);
      tracks->Remove(outmin);
      tracks->Remove(inmin);
    }
    else merge = kFALSE;
  }
  LOG(AliHLTTPCLog::kInformational,"AliHLTTPCInterMerger::SlowMerge","Result")
  <<AliHLTTPCLog::kDec<<"Merged Tracks: "<<tracks->GetNTracks()-kNIn<<ENDLOG;

  char name[256];
  sprintf(name,"ntuple_i_%d.root",fPatch);
  WriteNtuple(name,ntuple);
}

void AliHLTTPCInterMerger::MMerge(){
  while(Merge());
  GetOutTracks()->AddTracks(GetInTracks(0));
}

Int_t AliHLTTPCInterMerger::Merge(){
  Int_t nrow= fRowMax-fRowMin+1;
  Double_t xval =AliHLTTPCTransform::Row2X((fRowMax+fRowMin)/2);
  AliHLTTPCTrackArray * tracks = GetInTracks(0);
  const Int_t  kNIn =tracks->GetNTracks();
  AliHLTTPCTrack *tr[2];
  for(Int_t in=0;in<kNIn;in++){
    AliHLTTPCTrack *t = tracks->GetCheckedTrack(in);
    if(t){
      t->CalculateHelix();
      t->CalculatePoint(xval);
    }
  }
  for(Int_t out=0;out<kNIn;out++){
  AliHLTTPCTrack *outertrack=tracks->GetCheckedTrack(out);
  if(!outertrack) continue;
    for(Int_t in=0;in<kNIn;in++){
      if(in==out) continue;
      AliHLTTPCTrack *innertrack=tracks->GetCheckedTrack(in);
      if(!innertrack) continue;
      if(outertrack->GetNHits()+innertrack->GetNHits()>nrow) continue;

      if(IsTrack(innertrack,outertrack)){
        tr[0]=innertrack;
        tr[1]=outertrack;
        SortGlobalTracks(tr,2);

        Double_t r0 = pow(tr[0]->GetLastPointX(),2)+
                      pow(tr[0]->GetLastPointY(),2);
        Double_t r1 = pow(tr[1]->GetFirstPointX(),2)+
                      pow(tr[1]->GetFirstPointY(),2);
        if(r0<r1){
          MultiMerge(tracks,tr,2);
          tracks->Remove(out);
          tracks->Remove(in);
          break;
        }
      }
    } 
  }
  Int_t nmerged = tracks->GetNTracks()-kNIn; 
  LOG(AliHLTTPCLog::kInformational,"AliHLTTPCInterMerger::Merge","Result")
  <<AliHLTTPCLog::kDec<<"Merged Tracks: "<<nmerged<<ENDLOG;
  //add in tracks
//  GetOutTracks()->AddTracks(GetInTracks(0)); 

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