[u/mrichter/AliRoot.git] / TRD / AliTRDtrackV1.h

#ifndef ALITRDTRACKV1_H
#define ALITRDTRACKV1_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  Represents a reconstructed TRD track                                     //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#ifndef ALIKALMANTRACK_H
#include "AliKalmanTrack.h"
#endif

#ifndef ALIESDTRACK_H
#include "AliESDtrack.h"
#endif

#ifndef ALITRDSEEDV1_H
#include "AliTRDseedV1.h"
#endif

class AliTRDcluster;
class AliESDtrack;
class AliTRDReconstructor;
class AliTRDtrackV1 : public AliKalmanTrack
{
public:
  enum ETRDtrackV1Size { 
    kNdet      = AliTRDgeometry::kNdet
   ,kNstacks   = AliTRDgeometry::kNstack*AliTRDgeometry::kNsector
   ,kNplane    = AliTRDgeometry::kNlayer
   ,kNcham     = AliTRDgeometry::kNstack
   ,kNsect     = AliTRDgeometry::kNsector
   ,kNslice    =   3
   ,kNMLPslice =   8 
   ,kMAXCLUSTERSPERTRACK = 210
  };
  
  // bits from 0-13 are reserved by ROOT (see TObject.h)
  enum ETRDtrackV1Status {
    kOwner   = BIT(14)
   ,kStopped = BIT(15) 
  };

  AliTRDtrackV1();
  AliTRDtrackV1(AliTRDseedV1 *trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
  AliTRDtrackV1(const AliESDtrack &ref);
  AliTRDtrackV1(const AliTRDtrackV1 &ref);
  virtual ~AliTRDtrackV1();
  AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref) { *(new(this) AliTRDtrackV1(ref)); return *this; }
  
  Bool_t         CookPID();
  Bool_t         CookLabel(Float_t wrong);
  AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
  Double_t       GetBudget(Int_t i) const { return fBudget[i];}
  AliTRDcluster* GetCluster(Int_t id);
  Int_t          GetClusterIndex(Int_t id) const;
  Float_t        GetEdep() const {return fDE;}
  inline Float_t GetMomentum(Int_t plane) const;
  inline Int_t   GetNCross();
  inline Int_t   GetNumberOfTracklets() const;
  Double_t       GetPIDsignal() const   { return 0.;}
  Double_t       GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;}
  UChar_t        GetPIDquality() const  { return fPIDquality;}
  Double_t       GetPredictedChi2(const AliTRDseedV1 *tracklet) const;
  Double_t       GetPredictedChi2(const AliCluster* /*c*/) const                   { return 0.0; }
  Int_t          GetProlongation(Double_t xk, Double_t &y, Double_t &z);
  AliTRDseedV1*  GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : 0x0;}
  Int_t          GetTrackletIndex(Int_t plane) const          { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;}
  AliExternalTrackParam*
                 GetTrackLow() const  { return fTrackLow;} 
  AliExternalTrackParam*
                 GetTrackHigh() const  { return fTrackHigh;} 
  UShort_t*      GetTrackletIndexes() { return &fTrackletIndex[0];}
  
  Bool_t         IsEqual(const TObject *inTrack) const;
  Bool_t         IsOwner() const {return TestBit(kOwner);};
  Bool_t         IsStopped() const {return TestBit(kStopped);};
  Bool_t         IsElectron() const;
  
  void           MakeBackupTrack();
  void           Print(Option_t *o="") const;

  Bool_t         PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
  Int_t          PropagateToR(Double_t xr, Double_t step);
  Bool_t         Rotate(Double_t angle, Bool_t absolute = kFALSE);
  void           SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
  void           SetEdep(Double32_t inDE){fDE = inDE;};
  void           SetNumberOfClusters();
  void           SetOwner();
  void           SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
  void           SetPIDquality(UChar_t inPIDquality){fPIDquality = inPIDquality;};
  void           SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
  void           SetTracklet(AliTRDseedV1 *trklt,  Int_t index);
  void           SetTrackLow();
  void           SetTrackHigh(const AliExternalTrackParam *op=0x0);
  inline void    SetReconstructor(const AliTRDReconstructor *rec);
  inline Float_t StatusForTOF();
  void           UnsetTracklet(Int_t plane);
  Bool_t         Update(AliTRDseedV1 *tracklet, Double_t chi2);
  //Bool_t         Update(const AliTRDcluster *c, Double_t chi2, Int_t index, Double_t h01){ return AliTRDtrack::Update(c,chi2,index,h01); };
  Bool_t         Update(const AliCluster *, Double_t, Int_t)                        { return kFALSE; };
  void           UpdateESDtrack(AliESDtrack *t);

private:
  UChar_t      fPIDquality;           //  No of planes used for PID calculation	
  UShort_t     fTrackletIndex[kNplane];//  Tracklets index in the tracker list
  Double32_t   fPID[AliPID::kSPECIES];//  PID probabilities
  Double32_t   fBudget[3];            //  Integrated material budget
  Double32_t   fDE;                   //  Integrated delta energy
  const AliTRDReconstructor *fReconstructor;//! reconstructor link 
  AliTRDseedV1 *fTracklet[kNplane];   //  Tracklets array defining the track
  AliTRDtrackV1 *fBackupTrack;        // Backup track
  AliExternalTrackParam *fTrackLow;   // parameters of the track which enter TRD from below (TPC) 
  AliExternalTrackParam *fTrackHigh;  // parameters of the track which enter TRD from above (HMPID, PHOS) 

  ClassDef(AliTRDtrackV1, 4)          // new TRD track
};

//____________________________________________________
inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
{
  return plane >=0 && plane < kNplane && fTrackletIndex[plane] != 0xff ? fTracklet[plane]->GetMomentum() : -1.;
}

//____________________________________________________
inline Int_t AliTRDtrackV1::GetNCross()
{
  Int_t ncross = 0;
  for(Int_t ip=0; ip<kNplane; ip++){
    if(!fTracklet[ip]) continue;
    ncross += fTracklet[ip]->IsRowCross();
  }
  return ncross;
}

//____________________________________________________
inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const
{
  Int_t n = 0;
  for(Int_t ip=0; ip<kNplane; ip++){
    if(!fTracklet[ip]) continue;
    n++;
  }
  return n;
}


//____________________________________________________
inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec)
{
  for(Int_t ip=0; ip<kNplane; ip++){
    if(!fTracklet[ip]) continue;
    fTracklet[ip]->SetReconstructor(rec);
  }
  fReconstructor = rec;
}


//____________________________________________________________________________
inline Float_t AliTRDtrackV1::StatusForTOF()
{
  //
  // Defines the status of the TOF extrapolation
  //

  if(!fTracklet[5]) return 0.;

  // Definition of res ????
  Float_t res = /*(0.2 + 0.8 * (fN / (fNExpected + 5.0))) **/ (0.4 + 0.6 * fTracklet[5]->GetN() / 20.0);
  res *= (0.25 + 0.8 * 40.0 / (40.0 + fBudget[2]));
  return res;
}

#endif
Commit	Line	Data
d9950a5a	1	#ifndef ALITRDTRACKV1_H
	2	#define ALITRDTRACKV1_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
0906e73e	8	///////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// Represents a reconstructed TRD track //
	11	// //
	12	///////////////////////////////////////////////////////////////////////////////
d9950a5a	13
3b57a3f7	14	#ifndef ALIKALMANTRACK_H
3b57a3f7	15	#include "AliKalmanTrack.h"
d9950a5a	16	#endif
d9950a5a	17
3b57a3f7	18	#ifndef ALIESDTRACK_H
	19	#include "AliESDtrack.h"
	20	#endif
	21
	22	#ifndef ALITRDSEEDV1_H
	23	#include "AliTRDseedV1.h"
	24	#endif
d9950a5a	25
3b57a3f7	26	class AliTRDcluster;
3b57a3f7	27	class AliESDtrack;
181c7f7e	28	class AliTRDReconstructor;
3b57a3f7	29	class AliTRDtrackV1 : public AliKalmanTrack
d9950a5a	30	{
3b57a3f7	31	public:
e3cf3d02	32	enum ETRDtrackV1Size {
	33	kNdet = AliTRDgeometry::kNdet
	34	,kNstacks = AliTRDgeometry::kNstack*AliTRDgeometry::kNsector
	35	,kNplane = AliTRDgeometry::kNlayer
	36	,kNcham = AliTRDgeometry::kNstack
	37	,kNsect = AliTRDgeometry::kNsector
	38	,kNslice = 3
	39	,kNMLPslice = 8
	40	,kMAXCLUSTERSPERTRACK = 210
	41	};
3b57a3f7	42
2e2915e7	43	// bits from 0-13 are reserved by ROOT (see TObject.h)
e3cf3d02	44	enum ETRDtrackV1Status {
	45	kOwner = BIT(14)
	46	,kStopped = BIT(15)
e44586fb	47	};
0906e73e	48
3b57a3f7	49	AliTRDtrackV1();
	50	AliTRDtrackV1(AliTRDseedV1 *trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
	51	AliTRDtrackV1(const AliESDtrack &ref);
	52	AliTRDtrackV1(const AliTRDtrackV1 &ref);
	53	virtual ~AliTRDtrackV1();
	54	AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref) { (new(this) AliTRDtrackV1(ref)); return this; }
	55
	56	Bool_t CookPID();
	57	Bool_t CookLabel(Float_t wrong);
	58	AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
	59	Double_t GetBudget(Int_t i) const { return fBudget[i];}
0349cc67	60	AliTRDcluster* GetCluster(Int_t id);
3b57a3f7	61	Int_t GetClusterIndex(Int_t id) const;
	62	Float_t GetEdep() const {return fDE;}
	63	inline Float_t GetMomentum(Int_t plane) const;
	64	inline Int_t GetNCross();
82180ec1	65	inline Int_t GetNumberOfTracklets() const;
1128e139	66	Double_t GetPIDsignal() const { return 0.;}
3b57a3f7	67	Double_t GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;}
1128e139	68	UChar_t GetPIDquality() const { return fPIDquality;}
3b57a3f7	69	Double_t GetPredictedChi2(const AliTRDseedV1 *tracklet) const;
	70	Double_t GetPredictedChi2(const AliCluster* /c/) const { return 0.0; }
	71	Int_t GetProlongation(Double_t xk, Double_t &y, Double_t &z);
	72	AliTRDseedV1* GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : 0x0;}
e3cf3d02	73	Int_t GetTrackletIndex(Int_t plane) const { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;}
	74	AliExternalTrackParam*
	75	GetTrackLow() const { return fTrackLow;}
	76	AliExternalTrackParam*
	77	GetTrackHigh() const { return fTrackHigh;}
	78	UShort_t* GetTrackletIndexes() { return &fTrackletIndex[0];}
3b57a3f7	79
203967fc	80	Bool_t IsEqual(const TObject *inTrack) const;
e44586fb	81	Bool_t IsOwner() const {return TestBit(kOwner);};
e44586fb	82	Bool_t IsStopped() const {return TestBit(kStopped);};
22a4ab0c	83	Bool_t IsElectron() const;
3b57a3f7	84
87a7fa94	85	void MakeBackupTrack();
203967fc	86	void Print(Option_t *o="") const;
203967fc	87
3b57a3f7	88	Bool_t PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
	89	Int_t PropagateToR(Double_t xr, Double_t step);
	90	Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE);
	91	void SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
bee2b41e	92	void SetEdep(Double32_t inDE){fDE = inDE;};
3b57a3f7	93	void SetNumberOfClusters();
3b57a3f7	94	void SetOwner();
bee2b41e	95	void SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
bee2b41e	96	void SetPIDquality(UChar_t inPIDquality){fPIDquality = inPIDquality;};
e44586fb	97	void SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
3b57a3f7	98	void SetTracklet(AliTRDseedV1 *trklt, Int_t index);
e3cf3d02	99	void SetTrackLow();
e3cf3d02	100	void SetTrackHigh(const AliExternalTrackParam *op=0x0);
181c7f7e	101	inline void SetReconstructor(const AliTRDReconstructor *rec);
3b57a3f7	102	inline Float_t StatusForTOF();
181d2c97	103	void UnsetTracklet(Int_t plane);
3b57a3f7	104	Bool_t Update(AliTRDseedV1 *tracklet, Double_t chi2);
	105	//Bool_t Update(const AliTRDcluster *c, Double_t chi2, Int_t index, Double_t h01){ return AliTRDtrack::Update(c,chi2,index,h01); };
	106	Bool_t Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
	107	void UpdateESDtrack(AliESDtrack *t);
3b57a3f7	108
	109	private:
	110	UChar_t fPIDquality; // No of planes used for PID calculation
	111	UShort_t fTrackletIndex[kNplane];// Tracklets index in the tracker list
	112	Double32_t fPID[AliPID::kSPECIES];// PID probabilities
	113	Double32_t fBudget[3]; // Integrated material budget
	114	Double32_t fDE; // Integrated delta energy
215f7116	115	const AliTRDReconstructor *fReconstructor;//! reconstructor link
3b57a3f7	116	AliTRDseedV1 *fTracklet[kNplane]; // Tracklets array defining the track
3b57a3f7	117	AliTRDtrackV1 *fBackupTrack; // Backup track
e3cf3d02	118	AliExternalTrackParam *fTrackLow; // parameters of the track which enter TRD from below (TPC)
e3cf3d02	119	AliExternalTrackParam *fTrackHigh; // parameters of the track which enter TRD from above (HMPID, PHOS)
d9950a5a	120
e3cf3d02	121	ClassDef(AliTRDtrackV1, 4) // new TRD track
d9950a5a	122	};
d9950a5a	123
3b57a3f7	124	//____________________________________________________
	125	inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
	126	{
	127	return plane >=0 && plane < kNplane && fTrackletIndex[plane] != 0xff ? fTracklet[plane]->GetMomentum() : -1.;
	128	}
	129
	130	//____________________________________________________
	131	inline Int_t AliTRDtrackV1::GetNCross()
	132	{
	133	Int_t ncross = 0;
	134	for(Int_t ip=0; ip<kNplane; ip++){
	135	if(!fTracklet[ip]) continue;
	136	ncross += fTracklet[ip]->IsRowCross();
	137	}
	138	return ncross;
	139	}
	140
82180ec1	141	//____________________________________________________
	142	inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const
	143	{
	144	Int_t n = 0;
	145	for(Int_t ip=0; ip<kNplane; ip++){
	146	if(!fTracklet[ip]) continue;
	147	n++;
	148	}
	149	return n;
	150	}
	151
181c7f7e	152
	153	//____________________________________________________
	154	inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec)
	155	{
	156	for(Int_t ip=0; ip<kNplane; ip++){
	157	if(!fTracklet[ip]) continue;
	158	fTracklet[ip]->SetReconstructor(rec);
	159	}
215f7116	160	fReconstructor = rec;
181c7f7e	161	}
	162
	163
3b57a3f7	164	//____________________________________________________________________________
	165	inline Float_t AliTRDtrackV1::StatusForTOF()
	166	{
	167	//
	168	// Defines the status of the TOF extrapolation
	169	//
	170
	171	if(!fTracklet[5]) return 0.;
	172
	173	// Definition of res ????
	174	Float_t res = /(0.2 + 0.8 (fN / (fNExpected + 5.0))) */ (0.4 + 0.6 fTracklet[5]->GetN() / 20.0);
	175	res = (0.25 + 0.8 40.0 / (40.0 + fBudget[2]));
	176	return res;
	177	}
d9950a5a	178
	179	#endif
	180
	181
6e49cfdb	182