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

//$Id$

// Author: Anders Vestbo <mailto:vestbo@fi.uib.no>, Uli Frankenfeld <mailto:franken@fi.uib.no>
//*-- Copyright &copy ASV 

#include "AliL3Defs.h"
#include "AliL3RootTypes.h"
#include "AliL3Logging.h"
#include "AliL3Vertex.h"
#include "AliL3ConfMapPoint.h"
#include "AliL3ConfMapFit.h"
#include "AliL3ConfMapTrack.h"
#include <math.h>

//_____________________________________________________________
// AliL3ConfMapTrack
//
// Track class for conformal mapper

ClassImp(AliL3ConfMapTrack)


AliL3ConfMapTrack::AliL3ConfMapTrack()
{
  //Constructor

  fChiSq[0] = 0.;
  fChiSq[1] = 0.;

}

AliL3ConfMapTrack::~AliL3ConfMapTrack()
{

}

void AliL3ConfMapTrack::DeleteCandidate()
{
  //Deletes this track by resetting all its parameters. Does not delete
  //the object itself.

  AliL3ConfMapPoint *curHit = (AliL3ConfMapPoint*)firstHit;
  AliL3ConfMapPoint *nextHit;
  
  while(curHit != 0)
    {
      nextHit = (AliL3ConfMapPoint*)curHit->nextTrackHit;
      curHit->nextTrackHit = 0;
      curHit = nextHit;
    }
  
  UInt_t *hit_numbers = GetHitNumbers();
  for(Int_t i=0; i<GetNHits(); i++)
    {
      //fHitNumbers[i] = 0;
      hit_numbers[i]=0;
    }
    
  SetRadius(0.);
  SetCenterX(0.);
  SetCenterY(0.);
  
  ComesFromMainVertex(false);

  SetNHits(0);
  SetCharge(0);
  fChiSq[0] = 0.;
  fChiSq[1] = 0.;
}


void AliL3ConfMapTrack::SetProperties(Bool_t usage)
{
  //Set the hits to this track to 'usage'
  
  for(StartLoop(); LoopDone(); GetNextHit())
    {
      AliL3ConfMapPoint *p = (AliL3ConfMapPoint*)currentHit;
      p->SetUsage(usage);
    }
  return;
}

void AliL3ConfMapTrack::Reset()
{
  //Resets the fit parameters of this track.

  //xy-plane
  s11Xy   = 0;
  s12Xy   = 0;
  s22Xy   = 0;
  g1Xy    = 0;
  g2Xy    = 0;
  fChiSq[0]  = 0.;
    
  //sz-plane
  s11Sz = 0;
  s12Sz = 0;
  s22Sz = 0;
  g1Sz  = 0;
  g2Sz  = 0;
  fChiSq[1] = 0; 
  SetLength(0);
  
  
}

void AliL3ConfMapTrack::UpdateParam(AliL3ConfMapPoint *thisHit)
{
  //Function to update fit parameters of track
  //Also, it updates the hit pointers.
  

  //Increment the number of hits assigned to this track:

  //fNHits++;
  Int_t nhits = GetNHits();
  nhits++;
  SetNHits(nhits); //SetNHits(nhits++);

  //Set the hit pointers:
  //if(fNHits == 1)
  if(GetNHits()==1)  
    firstHit = thisHit;
  else
    ((AliL3ConfMapPoint*)lastHit)->nextTrackHit = thisHit;
  lastHit = thisHit;

  
  s11Xy = s11Xy + thisHit->GetXYWeight() ;
  s12Xy = s12Xy + thisHit->GetXYWeight() * thisHit->GetXprime() ;
  s22Xy = s22Xy + thisHit->GetXYWeight() * pow((thisHit->GetXprime()),2) ;
  g1Xy  = g1Xy  + thisHit->GetXYWeight() * thisHit->GetYprime() ;
  g2Xy  = g2Xy  + thisHit->GetXYWeight() * thisHit->GetXprime() * thisHit->GetYprime() ;
    
  ddXy  = s11Xy * s22Xy - pow((s12Xy),2) ;
  if ( ddXy != 0 ) 
    {
      a1Xy  = ( g1Xy * s22Xy - g2Xy * s12Xy ) / ddXy ;
      a2Xy  = ( g2Xy * s11Xy - g1Xy * s12Xy ) / ddXy ;
    }

  //     Now in the sz plane
  s11Sz = s11Sz + thisHit->GetZWeight() ;
  s12Sz = s12Sz + thisHit->GetZWeight() * thisHit->GetS() ;
  s22Sz = s22Sz + thisHit->GetZWeight() * thisHit->GetS() * thisHit->GetS() ;
  g1Sz  = g1Sz  + thisHit->GetZWeight() * thisHit->GetZ() ;
  g2Sz  = g2Sz  + thisHit->GetZWeight() * thisHit->GetS() * thisHit->GetZ() ;
  
        
  ddSz  = s11Sz * s22Sz -  s12Sz * s12Sz ;
  if ( ddSz != 0 ) {
    a1Sz  = ( g1Sz * s22Sz - g2Sz * s12Sz ) / ddSz ;
    a2Sz  = ( g2Sz * s11Sz - g1Sz * s12Sz ) / ddSz ;
  }
      
}


void AliL3ConfMapTrack::Fill(AliL3Vertex *vertex,Double_t max_Dca)
{
  //Fill track variables with or without fit.
  
  //fRadius = sqrt(a2Xy*a2Xy+1)/(2*fabs(a1Xy));
  Double_t radius = sqrt(a2Xy*a2Xy+1)/(2*fabs(a1Xy));
  SetRadius(radius);

  //fPt = (Double_t)(BFACT * BField * fRadius);
  Double_t pt = (Double_t)(BFACT * BField * GetRadius());
  SetPt(pt);

  if(GetPt() > max_Dca) //go for fit of helix in real space
    {
      AliL3ConfMapFit *fit = new AliL3ConfMapFit(this,vertex);
      fit->FitHelix();
      UpdateToFirstPoint();
      
      delete fit;
    }
  else if(GetPt() == 0)
    LOG(AliL3Log::kError,"AliL3ConfMapTrack::Fill","Tracks")<<AliL3Log::kDec<<
      "Found track with Pt=0!!!"<<ENDLOG;
  else
    {
      LOG(AliL3Log::kError,"AliL3ConfMapTrack::Fill","Tracks")<<AliL3Log::kDec<<
	"Track with pt<max_Dca :"<<GetPt()<<ENDLOG;
    }
  
}

void AliL3ConfMapTrack::UpdateToFirstPoint()
{
  //Update track parameters to the innermost point on the track.
  //Basically it justs calculates the intersection of the track, and a cylinder
  //with radius = r(innermost point). Then the parameters are updated to this point.
  //Should be called after the helixfit (in FillTracks).
  
  //AliL3ConfMapPoint *lHit = (AliL3ConfMapPoint*)fPoints->Last();
  AliL3ConfMapPoint *lHit = (AliL3ConfMapPoint*)lastHit;
  Double_t radius = sqrt(lHit->GetX()*lHit->GetX()+lHit->GetY()*lHit->GetY());
  
  //Get the track parameters
  
  Double_t tPhi0 = GetPsi() + GetCharge() * 0.5 * pi / fabs(GetCharge()) ;
  Double_t x0    = GetR0() * cos(GetPhi0()) ;
  Double_t y0    = GetR0() * sin(GetPhi0()) ;
  Double_t rc    = fabs(GetPt()) / ( BFACT * BField )  ;
  Double_t xc    = x0 - rc * cos(tPhi0) ;
  Double_t yc    = y0 - rc * sin(tPhi0) ;
  
  //Check helix and cylinder intersect
  
  Double_t fac1 = xc*xc + yc*yc ;
  Double_t sfac = sqrt( fac1 ) ;
    
  if ( fabs(sfac-rc) > radius || fabs(sfac+rc) < radius ) {
    LOG(AliL3Log::kError,"AliL3ConfMapTrack::UpdateToLastPoint","Tracks")<<AliL3Log::kDec<<
      "Track does not intersect"<<ENDLOG;
    return;
  }
  
  //Find intersection
  
  Double_t fac2   = ( radius*radius + fac1 - rc*rc) / (2.00 * radius * sfac ) ;
  Double_t phi    = atan2(yc,xc) + GetCharge()*acos(fac2) ;
  Double_t td     = atan2(radius*sin(phi) - yc,radius*cos(phi) - xc) ;
  
  //Intersection in z
  
  if ( td < 0 ) td = td + 2. * pi ;
  Double_t deltat = fmod((-GetCharge()*td + GetCharge()*tPhi0),2*pi) ;
  if ( deltat < 0.      ) deltat += 2. * pi ;
  if ( deltat > 2.*pi ) deltat -= 2. * pi ;
  Double_t z = GetZ0() + rc * GetTgl() * deltat ;
 
  
  Double_t xExtra = radius * cos(phi) ;
  Double_t yExtra = radius * sin(phi) ;
  
  Double_t tPhi = atan2(yExtra-yc,xExtra-xc);
  
  //if ( tPhi < 0 ) tPhi += 2. * M_PI ;
  
  Double_t tPsi = tPhi - GetCharge() * 0.5 * pi / fabs(GetCharge()) ;
  if ( tPsi > 2. * pi ) tPsi -= 2. * pi ;
  if ( tPsi < 0.        ) tPsi += 2. * pi ;
  
  //And finally, update the track parameters
  
  SetCenterX(xc);
  SetCenterY(yc);
  SetR0(radius);
  SetPhi0(phi);
  SetZ0(z);
  SetPsi(tPsi);
}

Int_t AliL3ConfMapTrack::GetMCLabel()
{
  //For evaluation study.
  //Returns the MCtrackID of the belonging clusters.
  //If MCLabel < 0, means that track is fake.

  return 0;
  /*
  Int_t num_of_clusters = GetNumberOfPoints();
  S *s=new S[num_of_clusters];
  Int_t i;
  for (i=0; i<num_of_clusters; i++) s[i].lab=s[i].max=0;
  
  Int_t lab=123456789;
  for (i=0; i<num_of_clusters; i++) {
    AliL3ConfMapPoint *c=(AliL3ConfMapPoint*)fPoints->UncheckedAt(i);
    lab=fabs(c->fMCTrackID[0]);
    Int_t j;
    for (j=0; j<num_of_clusters; j++)
      if (s[j].lab==lab || s[j].max==0) break;
    s[j].lab=lab;
    s[j].max++;
  }
  
  Int_t max=0;
  for (i=0; i<num_of_clusters; i++) 
    if (s[i].max>max) {max=s[i].max; lab=s[i].lab;}
    
  delete[] s;
  
  for (i=0; i<num_of_clusters; i++) {
    AliL3ConfMapPoint *c=(AliL3ConfMapPoint*)fPoints->UncheckedAt(i);
    if (fabs(c->fMCTrackID[1]) == lab ||
	fabs(c->fMCTrackID[2]) == lab ) max++;
  }
  
  //check if more than 10% of the clusters are incorrectly assigned (fake track):
  
  if (1.-Float_t(max)/num_of_clusters > 0.10) 
    {
      return -lab;
    }
  Int_t tail=Int_t(0.08*174);
  if (num_of_clusters < tail) return lab;
  
  max=0;
  for (i=1; i<=tail; i++) {
    AliL3ConfMapPoint *c = (AliL3ConfMapPoint*)fPoints->UncheckedAt(num_of_clusters-i);
    if (lab == fabs(c->fMCTrackID[0]) ||
	lab == fabs(c->fMCTrackID[1]) ||
	lab == fabs(c->fMCTrackID[2])) max++;
  }
  if (max < Int_t(0.5*tail)) 
    {
      //printf("Wrong innermost clusters\n");
      return -lab;
    }
  return lab;
  */
}
Commit	Line	Data
9392f4fd	1	//$Id$
9392f4fd	2
b661165c	3	// Author: Anders Vestbo <mailto:vestbo@fi.uib.no>, Uli Frankenfeld <mailto:franken@fi.uib.no>
b661165c	4	//*-- Copyright &copy ASV
108615fc	5
9392f4fd	6	#include "AliL3Defs.h"
108615fc	7	#include "AliL3RootTypes.h"
	8	#include "AliL3Logging.h"
	9	#include "AliL3Vertex.h"
	10	#include "AliL3ConfMapPoint.h"
	11	#include "AliL3ConfMapFit.h"
	12	#include "AliL3ConfMapTrack.h"
	13	#include <math.h>
	14
b661165c	15	//_____________________________________________________________
	16	// AliL3ConfMapTrack
	17	//
	18	// Track class for conformal mapper
	19
108615fc	20	ClassImp(AliL3ConfMapTrack)
	21
	22
	23	AliL3ConfMapTrack::AliL3ConfMapTrack()
	24	{
	25	//Constructor
	26
	27	fChiSq[0] = 0.;
	28	fChiSq[1] = 0.;
	29
	30	}
	31
	32	AliL3ConfMapTrack::~AliL3ConfMapTrack()
	33	{
	34
	35	}
	36
	37	void AliL3ConfMapTrack::DeleteCandidate()
	38	{
	39	//Deletes this track by resetting all its parameters. Does not delete
	40	//the object itself.
	41
	42	AliL3ConfMapPoint curHit = (AliL3ConfMapPoint)firstHit;
	43	AliL3ConfMapPoint *nextHit;
	44
	45	while(curHit != 0)
	46	{
	47	nextHit = (AliL3ConfMapPoint*)curHit->nextTrackHit;
	48	curHit->nextTrackHit = 0;
	49	curHit = nextHit;
	50	}
	51
	52	UInt_t *hit_numbers = GetHitNumbers();
	53	for(Int_t i=0; i<GetNHits(); i++)
	54	{
	55	//fHitNumbers[i] = 0;
	56	hit_numbers[i]=0;
	57	}
	58
	59	SetRadius(0.);
	60	SetCenterX(0.);
	61	SetCenterY(0.);
	62
	63	ComesFromMainVertex(false);
	64
	65	SetNHits(0);
	66	SetCharge(0);
	67	fChiSq[0] = 0.;
	68	fChiSq[1] = 0.;
	69	}
	70
	71
	72	void AliL3ConfMapTrack::SetProperties(Bool_t usage)
	73	{
	74	//Set the hits to this track to 'usage'
	75
	76	for(StartLoop(); LoopDone(); GetNextHit())
	77	{
	78	AliL3ConfMapPoint p = (AliL3ConfMapPoint)currentHit;
	79	p->SetUsage(usage);
	80	}
	81	return;
	82	}
	83
84	void AliL3ConfMapTrack::Reset()
85	{
86	//Resets the fit parameters of this track.
87
88	//xy-plane
89	s11Xy = 0;
90	s12Xy = 0;
91	s22Xy = 0;
92	g1Xy = 0;
93	g2Xy = 0;
94	fChiSq[0] = 0.;
95
96	//sz-plane
97	s11Sz = 0;
98	s12Sz = 0;
99	s22Sz = 0;
100	g1Sz = 0;
101	g2Sz = 0;
102	fChiSq[1] = 0;
103	SetLength(0);
104
105
106	}
107
108	void AliL3ConfMapTrack::UpdateParam(AliL3ConfMapPoint *thisHit)
109	{
110	//Function to update fit parameters of track
111	//Also, it updates the hit pointers.
112
113
114	//Increment the number of hits assigned to this track:
115
116	//fNHits++;
117	Int_t nhits = GetNHits();
118	nhits++;
119	SetNHits(nhits); //SetNHits(nhits++);
120
121	//Set the hit pointers:
122	//if(fNHits == 1)
123	if(GetNHits()==1)
124	firstHit = thisHit;
125	else
126	((AliL3ConfMapPoint*)lastHit)->nextTrackHit = thisHit;
127	lastHit = thisHit;
128
129
130	s11Xy = s11Xy + thisHit->GetXYWeight() ;
131	s12Xy = s12Xy + thisHit->GetXYWeight() * thisHit->GetXprime() ;
132	s22Xy = s22Xy + thisHit->GetXYWeight() * pow((thisHit->GetXprime()),2) ;
133	g1Xy = g1Xy + thisHit->GetXYWeight() * thisHit->GetYprime() ;
134	g2Xy = g2Xy + thisHit->GetXYWeight() * thisHit->GetXprime() * thisHit->GetYprime() ;
135
136	ddXy = s11Xy * s22Xy - pow((s12Xy),2) ;
137	if ( ddXy != 0 )
138	{
139	a1Xy = ( g1Xy * s22Xy - g2Xy * s12Xy ) / ddXy ;
140	a2Xy = ( g2Xy * s11Xy - g1Xy * s12Xy ) / ddXy ;
141	}
142
143	// Now in the sz plane
144	s11Sz = s11Sz + thisHit->GetZWeight() ;
145	s12Sz = s12Sz + thisHit->GetZWeight() * thisHit->GetS() ;
146	s22Sz = s22Sz + thisHit->GetZWeight() * thisHit->GetS() * thisHit->GetS() ;
147	g1Sz = g1Sz + thisHit->GetZWeight() * thisHit->GetZ() ;
148	g2Sz = g2Sz + thisHit->GetZWeight() * thisHit->GetS() * thisHit->GetZ() ;
149
150
151	ddSz = s11Sz * s22Sz - s12Sz * s12Sz ;
152	if ( ddSz != 0 ) {
153	a1Sz = ( g1Sz * s22Sz - g2Sz * s12Sz ) / ddSz ;
154	a2Sz = ( g2Sz * s11Sz - g1Sz * s12Sz ) / ddSz ;
155	}
156
157	}
158
159
160	void AliL3ConfMapTrack::Fill(AliL3Vertex *vertex,Double_t max_Dca)
161	{
162	//Fill track variables with or without fit.
163
164	//fRadius = sqrt(a2Xya2Xy+1)/(2fabs(a1Xy));
165	Double_t radius = sqrt(a2Xya2Xy+1)/(2fabs(a1Xy));
166	SetRadius(radius);
167
9392f4fd	168	//fPt = (Double_t)(BFACT * BField * fRadius);
9392f4fd	169	Double_t pt = (Double_t)(BFACT * BField * GetRadius());
108615fc	170	SetPt(pt);
	171
	172	if(GetPt() > max_Dca) //go for fit of helix in real space
	173	{
	174	AliL3ConfMapFit *fit = new AliL3ConfMapFit(this,vertex);
	175	fit->FitHelix();
	176	UpdateToFirstPoint();
	177
	178	delete fit;
	179	}
	180	else if(GetPt() == 0)
	181	LOG(AliL3Log::kError,"AliL3ConfMapTrack::Fill","Tracks")<<AliL3Log::kDec<<
	182	"Found track with Pt=0!!!"<<ENDLOG;
	183	else
	184	{
	185	LOG(AliL3Log::kError,"AliL3ConfMapTrack::Fill","Tracks")<<AliL3Log::kDec<<
	186	"Track with pt<max_Dca :"<<GetPt()<<ENDLOG;
	187	}
	188
	189	}
	190
	191	void AliL3ConfMapTrack::UpdateToFirstPoint()
	192	{
	193	//Update track parameters to the innermost point on the track.
	194	//Basically it justs calculates the intersection of the track, and a cylinder
	195	//with radius = r(innermost point). Then the parameters are updated to this point.
	196	//Should be called after the helixfit (in FillTracks).
	197
	198	//AliL3ConfMapPoint lHit = (AliL3ConfMapPoint)fPoints->Last();
	199	AliL3ConfMapPoint lHit = (AliL3ConfMapPoint)lastHit;
	200	Double_t radius = sqrt(lHit->GetX()lHit->GetX()+lHit->GetY()lHit->GetY());
	201
	202	//Get the track parameters
	203
	204	Double_t tPhi0 = GetPsi() + GetCharge() * 0.5 * pi / fabs(GetCharge()) ;
	205	Double_t x0 = GetR0() * cos(GetPhi0()) ;
	206	Double_t y0 = GetR0() * sin(GetPhi0()) ;
9392f4fd	207	Double_t rc = fabs(GetPt()) / ( BFACT * BField ) ;
108615fc	208	Double_t xc = x0 - rc * cos(tPhi0) ;
	209	Double_t yc = y0 - rc * sin(tPhi0) ;
	210
	211	//Check helix and cylinder intersect
	212
	213	Double_t fac1 = xcxc + ycyc ;
	214	Double_t sfac = sqrt( fac1 ) ;
	215
	216	if ( fabs(sfac-rc) > radius \|\| fabs(sfac+rc) < radius ) {
	217	LOG(AliL3Log::kError,"AliL3ConfMapTrack::UpdateToLastPoint","Tracks")<<AliL3Log::kDec<<
	218	"Track does not intersect"<<ENDLOG;
	219	return;
	220	}
	221
	222	//Find intersection
	223
	224	Double_t fac2 = ( radiusradius + fac1 - rcrc) / (2.00 * radius * sfac ) ;
	225	Double_t phi = atan2(yc,xc) + GetCharge()*acos(fac2) ;
	226	Double_t td = atan2(radiussin(phi) - yc,radiuscos(phi) - xc) ;
	227
	228	//Intersection in z
	229
	230	if ( td < 0 ) td = td + 2. * pi ;
	231	Double_t deltat = fmod((-GetCharge()td + GetCharge()tPhi0),2*pi) ;
	232	if ( deltat < 0. ) deltat += 2. * pi ;
	233	if ( deltat > 2.pi ) deltat -= 2. pi ;
	234	Double_t z = GetZ0() + rc * GetTgl() * deltat ;
	235
	236
	237	Double_t xExtra = radius * cos(phi) ;
	238	Double_t yExtra = radius * sin(phi) ;
	239
	240	Double_t tPhi = atan2(yExtra-yc,xExtra-xc);
	241
	242	//if ( tPhi < 0 ) tPhi += 2. * M_PI ;
	243
	244	Double_t tPsi = tPhi - GetCharge() * 0.5 * pi / fabs(GetCharge()) ;
	245	if ( tPsi > 2. * pi ) tPsi -= 2. * pi ;
	246	if ( tPsi < 0. ) tPsi += 2. * pi ;
	247
	248	//And finally, update the track parameters
	249
	250	SetCenterX(xc);
	251	SetCenterY(yc);
	252	SetR0(radius);
	253	SetPhi0(phi);
	254	SetZ0(z);
	255	SetPsi(tPsi);
	256	}
	257
	258	Int_t AliL3ConfMapTrack::GetMCLabel()
	259	{
	260	//For evaluation study.
	261	//Returns the MCtrackID of the belonging clusters.
	262	//If MCLabel < 0, means that track is fake.
	263
	264	return 0;
	265	/*
	266	Int_t num_of_clusters = GetNumberOfPoints();
	267	S *s=new S[num_of_clusters];
	268	Int_t i;
	269	for (i=0; i<num_of_clusters; i++) s[i].lab=s[i].max=0;
	270
	271	Int_t lab=123456789;
272	for (i=0; i<num_of_clusters; i++) {
273	AliL3ConfMapPoint c=(AliL3ConfMapPoint)fPoints->UncheckedAt(i);
274	lab=fabs(c->fMCTrackID[0]);
275	Int_t j;
276	for (j=0; j<num_of_clusters; j++)
277	if (s[j].lab==lab \|\| s[j].max==0) break;
278	s[j].lab=lab;
279	s[j].max++;
280	}
281
282	Int_t max=0;
283	for (i=0; i<num_of_clusters; i++)
284	if (s[i].max>max) {max=s[i].max; lab=s[i].lab;}
285
286	delete[] s;
287
288	for (i=0; i<num_of_clusters; i++) {
289	AliL3ConfMapPoint c=(AliL3ConfMapPoint)fPoints->UncheckedAt(i);
290	if (fabs(c->fMCTrackID[1]) == lab \|\|
291	fabs(c->fMCTrackID[2]) == lab ) max++;
292	}
293
294	//check if more than 10% of the clusters are incorrectly assigned (fake track):
295
296	if (1.-Float_t(max)/num_of_clusters > 0.10)
297	{
298	return -lab;
299	}
300	Int_t tail=Int_t(0.08*174);
301	if (num_of_clusters < tail) return lab;
302
303	max=0;
304	for (i=1; i<=tail; i++) {
305	AliL3ConfMapPoint c = (AliL3ConfMapPoint)fPoints->UncheckedAt(num_of_clusters-i);
306	if (lab == fabs(c->fMCTrackID[0]) \|\|
307	lab == fabs(c->fMCTrackID[1]) \|\|
308	lab == fabs(c->fMCTrackID[2])) max++;
309	}
310	if (max < Int_t(0.5*tail))
311	{
312	//printf("Wrong innermost clusters\n");
313	return -lab;
314	}
315	return lab;
316	*/
317	}
318