[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 ALITRDSEEDV1_H
#include "AliTRDseedV1.h"
//#endif

class AliTRDcluster;
class AliESDtrack;
class AliTRDReconstructor;
class AliTRDtrackV1 : public AliKalmanTrack
{
  friend class AliHLTTRDTrack;
public:
  enum ETRDtrackSize { 
    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 ETRDtrackStatus {
    kOwner     = BIT(14)
   ,kStopped   = BIT(15) 
   ,kKink      = BIT(16) 
  };

  // propagation/update error codes (up to 4 bits)
  enum ETRDtrackError {
    kProlongation = 1
   ,kPropagation
   ,kAdjustSector
   ,kSnp
   ,kTrackletInit
   ,kUpdate
   ,kUnknown      = 0xff
  };

  // data/clusters/tracklet error codes (up to 4 bits/layer)
  enum ETRDlayerError {
    kGeometry = 1
   ,kBoundary
   ,kNoClusters
   ,kNoAttach
   ,kNoClustersTracklet
   ,kNoFit
   ,kChi2
  };

  AliTRDtrackV1();
  AliTRDtrackV1(AliTRDseedV1 * const 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;}
  Int_t          GetESDid() const {return fESDid;}
  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        GetNumberOfTrackletsPID() const;
  Double_t       GetPredictedChi2(const AliTRDseedV1 *tracklet, Double_t *cov) const;
  Double_t       GetPredictedChi2(const AliCluster* /*c*/) const                   { return 0.0; }
  Int_t          GetProlongation(Double_t xk, Double_t &y, Double_t &z);
  inline UChar_t GetStatusTRD(Int_t ly=-1) const;
  Int_t          GetSector() const;
  AliTRDseedV1*  GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : NULL;}
  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;} 
  const Int_t* GetTrackletIndexes() const { return &fTrackletIndex[0];}
  
  Bool_t         IsEqual(const TObject *inTrack) const;
  Bool_t         IsKink() const    { return TestBit(kKink);}
  Bool_t         IsOwner() const   { return TestBit(kOwner);};
  Bool_t         IsStopped() const { return TestBit(kStopped);};
  Bool_t         IsElectron() const;
  inline static Bool_t IsTrackError(ETRDtrackError error, UInt_t status);
  inline static Bool_t IsLayerError(ETRDlayerError error, Int_t layer, UInt_t status);

  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           SetESDid(Int_t id) {fESDid = id;}
  void           SetKink(Bool_t k)        { SetBit(kKink, k);}
  void           SetNumberOfClusters();
  UChar_t        SetNumberOfTrackletsPID(Bool_t recalc);
  void           SetOwner();
  void           SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
  void           SetPIDquality(UChar_t /*inPIDquality*/) const {/*fPIDquality = inPIDquality*/;};
  inline void    SetStatus(UChar_t stat, Int_t ly=-1);
  void           SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
  void           SetTracklet(AliTRDseedV1 *const trklt,  Int_t index);
  void           SetTrackLow();
  void           SetTrackHigh(const AliExternalTrackParam *op=NULL);
  inline void    SetReconstructor(const AliTRDReconstructor *rec);
  inline Float_t StatusForTOF();
  void           UnsetTracklet(Int_t plane);
  Bool_t         Update(Double_t *p, Double_t *cov, Double_t chi2, Bool_t useTRD=kTRUE);
  Bool_t         Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
  void           UpdateESDtrack(AliESDtrack *t);

private:
  UInt_t       fStatus;                //  Bit map for the status of propagation
  Int_t        fTrackletIndex[kNplane];//  Tracklets index in the tracker list
  Int_t        fESDid;                 //  ESD track id 
  Double32_t   fPID[AliPID::kSPECIES]; //  PID probabilities
  Double32_t   fBudget[3];             //  Integrated material budget
  Double32_t   fDE;                    //  Integrated delta energy
  const AliTRDReconstructor *fkReconstructor;//! reconstructor link 
  AliTRDtrackV1 *fBackupTrack;         //! Backup track
  AliTRDseedV1  *fTracklet[kNplane];   //  Tracklets array defining the 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, 7)          // TRD track - tracklet based
};

//____________________________________________________
inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
{
  return plane >=0 && plane < kNplane && fTrackletIndex[plane] != -1 ? 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 UChar_t AliTRDtrackV1::GetStatusTRD(Int_t ly) const
{
  if(ly>=-1 && ly<kNplane) return (fStatus>>((ly+1)*4))&0xf;
  return kUnknown;
}

//____________________________________________________
inline Bool_t AliTRDtrackV1::IsTrackError(ETRDtrackError error, UInt_t status)
{
  return (status&0xf)==UChar_t(error);
}

//____________________________________________________
inline Bool_t AliTRDtrackV1::IsLayerError(ETRDlayerError error, Int_t ly, UInt_t status)
{
  if(ly>=kNplane || ly<0) return kFALSE;
  return ((status>>((ly+1)*4))&0xf) == UChar_t(error);
}

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

//____________________________________________________
inline void AliTRDtrackV1::SetStatus(UChar_t status, Int_t ly)
{
  if(ly<kNplane) fStatus|=((status&0xf)<<((ly+1)*4));
  return;
}


//____________________________________________________________________________
inline Float_t AliTRDtrackV1::StatusForTOF()
{
  // OBSOLETE
  // 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
4d6aee34	14	//#ifndef ALIKALMANTRACK_H
3b57a3f7	15	#include "AliKalmanTrack.h"
4d6aee34	16	//#endif
3b57a3f7	17
4d6aee34	18	//#ifndef ALITRDSEEDV1_H
3b57a3f7	19	#include "AliTRDseedV1.h"
4d6aee34	20	//#endif
d9950a5a	21
3b57a3f7	22	class AliTRDcluster;
3b57a3f7	23	class AliESDtrack;
181c7f7e	24	class AliTRDReconstructor;
3b57a3f7	25	class AliTRDtrackV1 : public AliKalmanTrack
d9950a5a	26	{
3044dfe5	27	friend class AliHLTTRDTrack;
3b57a3f7	28	public:
eb2b4f91	29	enum ETRDtrackSize {
e3cf3d02	30	kNdet = AliTRDgeometry::kNdet
	31	,kNstacks = AliTRDgeometry::kNstack*AliTRDgeometry::kNsector
	32	,kNplane = AliTRDgeometry::kNlayer
	33	,kNcham = AliTRDgeometry::kNstack
	34	,kNsect = AliTRDgeometry::kNsector
	35	,kNslice = 3
	36	,kNMLPslice = 8
	37	,kMAXCLUSTERSPERTRACK = 210
	38	};
3b57a3f7	39
2e2915e7	40	// bits from 0-13 are reserved by ROOT (see TObject.h)
eb2b4f91	41	enum ETRDtrackStatus {
	42	kOwner = BIT(14)
	43	,kStopped = BIT(15)
	44	,kKink = BIT(16)
	45	};
	46
	47	// propagation/update error codes (up to 4 bits)
	48	enum ETRDtrackError {
	49	kProlongation = 1
	50	,kPropagation
	51	,kAdjustSector
	52	,kSnp
	53	,kTrackletInit
	54	,kUpdate
af4e990c	55	,kUnknown = 0xff
eb2b4f91	56	};
	57
	58	// data/clusters/tracklet error codes (up to 4 bits/layer)
	59	enum ETRDlayerError {
	60	kGeometry = 1
	61	,kBoundary
	62	,kNoClusters
	63	,kNoAttach
	64	,kNoClustersTracklet
	65	,kNoFit
	66	,kChi2
e44586fb	67	};
0906e73e	68
3b57a3f7	69	AliTRDtrackV1();
e17f4785	70	AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], const Double_t cov[15], Double_t x, Double_t alpha);
3b57a3f7	71	AliTRDtrackV1(const AliESDtrack &ref);
	72	AliTRDtrackV1(const AliTRDtrackV1 &ref);
	73	virtual ~AliTRDtrackV1();
	74	AliTRDtrackV1 &operator=(const AliTRDtrackV1 &ref) { (new(this) AliTRDtrackV1(ref)); return this; }
	75
	76	Bool_t CookPID();
	77	Bool_t CookLabel(Float_t wrong);
	78	AliTRDtrackV1* GetBackupTrack() const {return fBackupTrack;}
	79	Double_t GetBudget(Int_t i) const { return fBudget[i];}
0349cc67	80	AliTRDcluster* GetCluster(Int_t id);
3b57a3f7	81	Int_t GetClusterIndex(Int_t id) const;
3b57a3f7	82	Float_t GetEdep() const {return fDE;}
352cef8f	83	Int_t GetESDid() const {return fESDid;}
3b57a3f7	84	inline Float_t GetMomentum(Int_t plane) const;
3b57a3f7	85	inline Int_t GetNCross();
82180ec1	86	inline Int_t GetNumberOfTracklets() const;
1128e139	87	Double_t GetPIDsignal() const { return 0.;}
3b57a3f7	88	Double_t GetPID(Int_t is) const { return (is >=0 && is < AliPID::kSPECIES) ? fPID[is] : -1.;}
eb2b4f91	89	UChar_t GetNumberOfTrackletsPID() const;
b72f4eaf	90	Double_t GetPredictedChi2(const AliTRDseedV1 tracklet, Double_t cov) const;
3b57a3f7	91	Double_t GetPredictedChi2(const AliCluster* /c/) const { return 0.0; }
3b57a3f7	92	Int_t GetProlongation(Double_t xk, Double_t &y, Double_t &z);
eb2b4f91	93	inline UChar_t GetStatusTRD(Int_t ly=-1) const;
eb2b4f91	94	Int_t GetSector() const;
4d6aee34	95	AliTRDseedV1* GetTracklet(Int_t plane) const {return plane >=0 && plane <kNplane ? fTracklet[plane] : NULL;}
e3cf3d02	96	Int_t GetTrackletIndex(Int_t plane) const { return (plane>=0 && plane<kNplane) ? fTrackletIndex[plane] : -1;}
	97	AliExternalTrackParam*
	98	GetTrackLow() const { return fTrackLow;}
	99	AliExternalTrackParam*
	100	GetTrackHigh() const { return fTrackHigh;}
17896e82	101	const Int_t* GetTrackletIndexes() const { return &fTrackletIndex[0];}
3b57a3f7	102
203967fc	103	Bool_t IsEqual(const TObject *inTrack) const;
f29f13a6	104	Bool_t IsKink() const { return TestBit(kKink);}
	105	Bool_t IsOwner() const { return TestBit(kOwner);};
	106	Bool_t IsStopped() const { return TestBit(kStopped);};
22a4ab0c	107	Bool_t IsElectron() const;
eb2b4f91	108	inline static Bool_t IsTrackError(ETRDtrackError error, UInt_t status);
	109	inline static Bool_t IsLayerError(ETRDlayerError error, Int_t layer, UInt_t status);
	110
87a7fa94	111	void MakeBackupTrack();
203967fc	112	void Print(Option_t *o="") const;
203967fc	113
3b57a3f7	114	Bool_t PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
	115	Int_t PropagateToR(Double_t xr, Double_t step);
	116	Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE);
	117	void SetBudget(Int_t i, Double_t b) {if(i>=0 && i<3) fBudget[i] = b;}
bee2b41e	118	void SetEdep(Double32_t inDE){fDE = inDE;};
352cef8f	119	void SetESDid(Int_t id) {fESDid = id;}
f29f13a6	120	void SetKink(Bool_t k) { SetBit(kKink, k);}
3b57a3f7	121	void SetNumberOfClusters();
eb2b4f91	122	UChar_t SetNumberOfTrackletsPID(Bool_t recalc);
3b57a3f7	123	void SetOwner();
bee2b41e	124	void SetPID(Short_t is, Double_t inPID){if (is >=0 && is < AliPID::kSPECIES) fPID[is]=inPID;};
4d6aee34	125	void SetPIDquality(UChar_t /inPIDquality/) const {/fPIDquality = inPIDquality/;};
eb2b4f91	126	inline void SetStatus(UChar_t stat, Int_t ly=-1);
e44586fb	127	void SetStopped(Bool_t stop) {SetBit(kStopped, stop);}
4d6aee34	128	void SetTracklet(AliTRDseedV1 *const trklt, Int_t index);
e3cf3d02	129	void SetTrackLow();
4d6aee34	130	void SetTrackHigh(const AliExternalTrackParam *op=NULL);
181c7f7e	131	inline void SetReconstructor(const AliTRDReconstructor *rec);
3b57a3f7	132	inline Float_t StatusForTOF();
181d2c97	133	void UnsetTracklet(Int_t plane);
b06a50a5	134	Bool_t Update(Double_t p, Double_t cov, Double_t chi2, Bool_t useTRD=kTRUE);
4d6aee34	135	Bool_t Update(const AliCluster *, Double_t, Int_t) { return kFALSE; };
3b57a3f7	136	void UpdateESDtrack(AliESDtrack *t);
3b57a3f7	137
3b57a3f7	138	private:
eb2b4f91	139	UInt_t fStatus; // Bit map for the status of propagation
17896e82	140	Int_t fTrackletIndex[kNplane];// Tracklets index in the tracker list
352cef8f	141	Int_t fESDid; // ESD track id
eb2b4f91	142	Double32_t fPID[AliPID::kSPECIES]; // PID probabilities
	143	Double32_t fBudget[3]; // Integrated material budget
	144	Double32_t fDE; // Integrated delta energy
4d6aee34	145	const AliTRDReconstructor *fkReconstructor;//! reconstructor link
eb2b4f91	146	AliTRDtrackV1 *fBackupTrack; //! Backup track
17896e82	147	AliTRDseedV1 *fTracklet[kNplane]; // Tracklets array defining the track
eb2b4f91	148	AliExternalTrackParam *fTrackLow; // parameters of the track which enter TRD from below (TPC)
17896e82	149	AliExternalTrackParam *fTrackHigh; // parameters of the track which enter TRD from above (HMPID, PHOS)
d9950a5a	150
352cef8f	151	ClassDef(AliTRDtrackV1, 7) // TRD track - tracklet based
d9950a5a	152	};
d9950a5a	153
3b57a3f7	154	//____________________________________________________
	155	inline Float_t AliTRDtrackV1::GetMomentum(Int_t plane) const
	156	{
17896e82	157	return plane >=0 && plane < kNplane && fTrackletIndex[plane] != -1 ? fTracklet[plane]->GetMomentum() : -1.;
3b57a3f7	158	}
	159
	160	//____________________________________________________
	161	inline Int_t AliTRDtrackV1::GetNCross()
	162	{
	163	Int_t ncross = 0;
	164	for(Int_t ip=0; ip<kNplane; ip++){
	165	if(!fTracklet[ip]) continue;
	166	ncross += fTracklet[ip]->IsRowCross();
	167	}
	168	return ncross;
	169	}
	170
82180ec1	171	//____________________________________________________
	172	inline Int_t AliTRDtrackV1::GetNumberOfTracklets() const
	173	{
	174	Int_t n = 0;
	175	for(Int_t ip=0; ip<kNplane; ip++){
	176	if(!fTracklet[ip]) continue;
	177	n++;
	178	}
	179	return n;
	180	}
	181
eb2b4f91	182	//____________________________________________________
	183	inline UChar_t AliTRDtrackV1::GetStatusTRD(Int_t ly) const
	184	{
af4e990c	185	if(ly>=-1 && ly<kNplane) return (fStatus>>((ly+1)*4))&0xf;
af4e990c	186	return kUnknown;
eb2b4f91	187	}
	188
	189	//____________________________________________________
	190	inline Bool_t AliTRDtrackV1::IsTrackError(ETRDtrackError error, UInt_t status)
	191	{
	192	return (status&0xf)==UChar_t(error);
	193	}
	194
	195	//____________________________________________________
	196	inline Bool_t AliTRDtrackV1::IsLayerError(ETRDlayerError error, Int_t ly, UInt_t status)
	197	{
	198	if(ly>=kNplane \|\| ly<0) return kFALSE;
	199	return ((status>>((ly+1)*4))&0xf) == UChar_t(error);
	200	}
181c7f7e	201
	202	//____________________________________________________
	203	inline void AliTRDtrackV1::SetReconstructor(const AliTRDReconstructor *rec)
	204	{
	205	for(Int_t ip=0; ip<kNplane; ip++){
	206	if(!fTracklet[ip]) continue;
	207	fTracklet[ip]->SetReconstructor(rec);
	208	}
4d6aee34	209	fkReconstructor = rec;
181c7f7e	210	}
181c7f7e	211
eb2b4f91	212	//____________________________________________________
	213	inline void AliTRDtrackV1::SetStatus(UChar_t status, Int_t ly)
	214	{
	215	if(ly<kNplane) fStatus\|=((status&0xf)<<((ly+1)*4));
	216	return;
	217	}
	218
181c7f7e	219
3b57a3f7	220	//____________________________________________________________________________
	221	inline Float_t AliTRDtrackV1::StatusForTOF()
	222	{
eb2b4f91	223	// OBSOLETE
3b57a3f7	224	// Defines the status of the TOF extrapolation
	225	//
	226
	227	if(!fTracklet[5]) return 0.;
	228
	229	// Definition of res ????
	230	Float_t res = /(0.2 + 0.8 (fN / (fNExpected + 5.0))) */ (0.4 + 0.6 fTracklet[5]->GetN() / 20.0);
	231	res = (0.25 + 0.8 40.0 / (40.0 + fBudget[2]));
	232	return res;
	233	}
d9950a5a	234
	235	#endif
	236
	237
6e49cfdb	238