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

#ifndef ALITRDTRACK_H
#define ALITRDTRACK_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                                     //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include "AliKalmanTrack.h"

#include "AliTRDtracklet.h"

class AliESDtrack;
class AliTrackReference;
class AliTRDcluster;

const unsigned kMAXCLUSTERSPERTRACK = 210; 

class AliTRDtrack : public AliKalmanTrack {

 public:
   
  enum { kNdet      = 540
       , kNstacks   =  90
       , kNplane    =   6
       , kNcham     =   5
       , kNsect     =  18
       , kNslice    =   3
       , kMidTimeBin=  14};

   AliTRDtrack();
   AliTRDtrack(const AliTRDcluster *c, Int_t index, const Double_t xx[5]
              ,const Double_t cc[15], Double_t xr, Double_t alpha);  
   AliTRDtrack(const AliTRDtrack &t);    
   AliTRDtrack(const AliESDtrack &t);
   ~AliTRDtrack();
   AliTRDtrack(const AliKalmanTrack &t, Double_t alpha); 

           void     ResetClusters()                                         { SetChi2(0.0); 
                                                                              SetNumberOfClusters(0);              }
           Int_t    Compare(const TObject *o) const;
           void     AddNWrong()                                             { fNWrong++;                           }  
           void     IncCross()                                              { fNCross++; 
                                                                              if (fBackupTrack) fBackupTrack->IncCross();  }

           Int_t    GetSector() const;
           Float_t  GetClusterdQdl(Int_t i) const                           { return fdQdl[i];                     }    
           Double_t GetdEdx() const                                         { return fdEdx;                        }
           Int_t    GetNdedx() const                                        { return fNdedx;                       }
           Double_t GetPIDsignal() const                                    { return GetdEdx();                    }
           Int_t    GetClusterIndex(Int_t i) const                          { return fIndex[i];                    }
           Double_t GetC() const                                            { return AliExternalTrackParam::GetC(GetBz()); }
           Double_t GetPredictedChi2(const AliTRDcluster *c, Double_t h01) const;
           Float_t  GetPIDsignals(Int_t iPlane, Int_t iSlice) const         { return fdEdxPlane[iPlane][iSlice];   }
           Int_t    GetPIDTimBin(Int_t i) const                             { return fTimBinPlane[i];              }
           Double_t GetLikelihoodElectron() const                           { return fLhElectron;                  }
           Int_t    GetSeedLabel() const                                    { return fSeedLab;                     }
           Int_t   *GetBackupIndexes()                                      { return fIndexBackup;                 }
           Int_t   *GetIndexes()                                            { return fIndex;                       }
           Int_t    GetProlongation(Double_t xk, Double_t &y, Double_t &z);
           Bool_t   GetStop() const                                         { return fStopped;                     }
           Int_t    GetNRotate() const                                      { return fNRotate;                     }
           Int_t    GetNWrong() const                                       { return fNWrong;                      }
           Int_t    GetNCross() const                                       { return fNCross;                      }
           Int_t    GetNExpected() const                                    { return fNExpected;                   }
           Int_t    GetNLast() const                                        { return fNLast;                       }
           Int_t    GetNExpectedLast() const                                { return fNExpectedLast;               }
  AliTRDtracklet    GetTracklets(Int_t i) const                             { return fTracklets[i];                }
           Float_t  GetBudget(Int_t i) const                                { return fBudget[i];                   }
           Float_t  GetChi2Last() const                                     { return fChi2Last;                    }
  AliTRDtrack      *GetBackupTrack()                                        { return fBackupTrack;                 }

           void     SetdEdx(Double_t dedx)                                  { fdEdx                      = dedx;   }
           void     SetStop(Bool_t stop)                                    { fStopped                   = stop;   }
           void     SetPIDsignals(Float_t dedx, Int_t iPlane, Int_t iSlice) { fdEdxPlane[iPlane][iSlice] = dedx;   }
           void     SetPIDTimBin(Int_t timbin, Int_t i)                     { fTimBinPlane[i]            = timbin; }
           void     SetLikelihoodElectron(Float_t l)                        { fLhElectron                = l;      }
           void     SetSampledEdx(Float_t q, Int_t i);
           void     SetSampledEdx(Float_t q);
           void     SetSeedLabel(Int_t lab)                                 { fSeedLab                   = lab;    }
           void     SetNWrong(Int_t nwrong)                                 { fNWrong                    = nwrong; }
           void     SetNCross(Int_t ncross)                                 { fNCross                    = ncross; }
           void     SetNExpected(Int_t nexp)                                { fNExpected                 = nexp;   }
           void     SetNLast(Int_t nlast)                                   { fNLast                     = nlast;  }
           void     SetNExpectedLast(Int_t nexp)                            { fNExpectedLast             = nexp;   }
           void     SetChi2Last(Float_t chi2)                               { fChi2Last                  = chi2;   }
           void     SetTracklets(Int_t i, AliTRDtracklet t)                 { fTracklets[i]              = t;      }
           void     SetBudget(Int_t i, Float_t budget)                      { fBudget[i]                 = budget; }

           Int_t    PropagateToX(Double_t xr, Double_t step);
           Int_t    PropagateToR(Double_t xr, Double_t step);
           Int_t    UpdateMI(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01, Int_t plane);
           void     MakeBackupTrack();
           Bool_t   PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
           Bool_t   Update(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01);
           Bool_t   Rotate(Double_t angle, Bool_t absolute = kFALSE);
           void     CookdEdx(Double_t low = 0.05, Double_t up = 0.7);   
           Float_t  StatusForTOF();
 
 protected:

  AliTRDtrack      &operator=(const AliTRDtrack &t);

           Double_t GetBz() const;
           Bool_t   Update(const AliCluster */*c*/, Double_t /*chi2*/, Int_t /*idx*/) { return 0;   }
           Double_t GetPredictedChi2(const AliCluster* /*c*/) const                   { return 0.0; }

           Int_t    fSeedLab;                           //  Track label taken from seeding  
           Float_t  fdEdx;                              //  dE/dx (truncated mean)
           Float_t  fDE;                                //  Integrated delta energy
           Float_t  fdEdxPlane[kNplane][kNslice];       //  dE/dx from all 6 planes in 3 slices each
           Int_t    fTimBinPlane[kNplane];              //  Time bin of Max cluster from all 6 planes

           Bool_t   fStopped;                           //  Track stop indication
           Int_t    fIndex[kMAXCLUSTERSPERTRACK];       //  Global indexes of clusters  
           Int_t    fIndexBackup[kMAXCLUSTERSPERTRACK]; //  Backup indexes of clusters - used in iterations
           Float_t  fdQdl[kMAXCLUSTERSPERTRACK];        //  Cluster amplitudes corrected for track angles    
                           
           Float_t  fLhElectron;                        //  Likelihood to be an electron    
           Int_t    fNWrong;                            //  Number of wrong clusters
           Int_t    fNRotate;                           //  Number of rotation
           Int_t    fNCross;                            //  Number of the cross materials
           Int_t    fNExpected;                         //  Expected number of cluster
           Int_t    fNLast;                             //  Number of clusters in last 2 layers
           Int_t    fNExpectedLast;                     //  Number of expected clusters on last 2 layers
           Int_t    fNdedx;                             //  Number of clusters for dEdx measurment
           Float_t  fChi2Last;                          //  Chi2 in the  last 2 layers
   AliTRDtracklet   fTracklets[6];                      //  Tracklets
           Float_t  fBudget[3];                         //  Integrated material budget
   AliTRDtrack     *fBackupTrack;                       //! Backup track

   ClassDef(AliTRDtrack,8)                              //  TRD reconstructed tracks

};                     

#endif
Commit	Line	Data
46d29e70	1	#ifndef ALITRDTRACK_H
b3a5a838	2	#define ALITRDTRACK_H
46d29e70	3
46d29e70	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
b3a5a838	5	* See cxx source for full Copyright notice */
46d29e70	6
15ed8ba1	7	/* $Id$ */
46d29e70	8
53c17fbf	9	///////////////////////////////////////////////////////////////////////////////
	10	// //
	11	// Represents a reconstructed TRD track //
	12	// //
	13	///////////////////////////////////////////////////////////////////////////////
46d29e70	14
15ed8ba1	15	#include "AliKalmanTrack.h"
fe33d239	16
15ed8ba1	17	#include "AliTRDtracklet.h"
	18
	19	class AliESDtrack;
	20	class AliTrackReference;
dab811d0	21	class AliTRDcluster;
15ed8ba1	22
	23	const unsigned kMAXCLUSTERSPERTRACK = 210;
	24
7ad19338	25	class AliTRDtrack : public AliKalmanTrack {
27eaf44b	26
27eaf44b	27	public:
53c17fbf	28
6d45eaef	29	enum { kNdet = 540
15ed8ba1	30	, kNstacks = 90
	31	, kNplane = 6
	32	, kNcham = 5
	33	, kNsect = 18
dab811d0	34	, kNslice = 3
dab811d0	35	, kMidTimeBin= 14};
7ad19338	36
6c94f330	37	AliTRDtrack();
fe33d239	38	AliTRDtrack(const AliTRDcluster *c, Int_t index, const Double_t xx[5]
	39	,const Double_t cc[15], Double_t xr, Double_t alpha);
	40	AliTRDtrack(const AliTRDtrack &t);
	41	AliTRDtrack(const AliESDtrack &t);
6c94f330	42	~AliTRDtrack();
fe33d239	43	AliTRDtrack(const AliKalmanTrack &t, Double_t alpha);
	44
	45	void ResetClusters() { SetChi2(0.0);
	46	SetNumberOfClusters(0); }
	47	Int_t Compare(const TObject *o) const;
	48	void AddNWrong() { fNWrong++; }
	49	void IncCross() { fNCross++;
	50	if (fBackupTrack) fBackupTrack->IncCross(); }
	51
	52	Int_t GetSector() const;
	53	Float_t GetClusterdQdl(Int_t i) const { return fdQdl[i]; }
	54	Double_t GetdEdx() const { return fdEdx; }
77566f2a	55	Int_t GetNdedx() const { return fNdedx; }
fe33d239	56	Double_t GetPIDsignal() const { return GetdEdx(); }
	57	Int_t GetClusterIndex(Int_t i) const { return fIndex[i]; }
	58	Double_t GetC() const { return AliExternalTrackParam::GetC(GetBz()); }
	59	Double_t GetPredictedChi2(const AliTRDcluster *c, Double_t h01) const;
	60	Float_t GetPIDsignals(Int_t iPlane, Int_t iSlice) const { return fdEdxPlane[iPlane][iSlice]; }
	61	Int_t GetPIDTimBin(Int_t i) const { return fTimBinPlane[i]; }
	62	Double_t GetLikelihoodElectron() const { return fLhElectron; }
	63	Int_t GetSeedLabel() const { return fSeedLab; }
	64	Int_t *GetBackupIndexes() { return fIndexBackup; }
	65	Int_t *GetIndexes() { return fIndex; }
	66	Int_t GetProlongation(Double_t xk, Double_t &y, Double_t &z);
	67	Bool_t GetStop() const { return fStopped; }
fe33d239	68	Int_t GetNRotate() const { return fNRotate; }
27eaf44b	69	Int_t GetNWrong() const { return fNWrong; }
fe33d239	70	Int_t GetNCross() const { return fNCross; }
27eaf44b	71	Int_t GetNExpected() const { return fNExpected; }
	72	Int_t GetNLast() const { return fNLast; }
	73	Int_t GetNExpectedLast() const { return fNExpectedLast; }
	74	AliTRDtracklet GetTracklets(Int_t i) const { return fTracklets[i]; }
	75	Float_t GetBudget(Int_t i) const { return fBudget[i]; }
	76	Float_t GetChi2Last() const { return fChi2Last; }
fe33d239	77	AliTRDtrack *GetBackupTrack() { return fBackupTrack; }
	78
	79	void SetdEdx(Double_t dedx) { fdEdx = dedx; }
	80	void SetStop(Bool_t stop) { fStopped = stop; }
	81	void SetPIDsignals(Float_t dedx, Int_t iPlane, Int_t iSlice) { fdEdxPlane[iPlane][iSlice] = dedx; }
	82	void SetPIDTimBin(Int_t timbin, Int_t i) { fTimBinPlane[i] = timbin; }
	83	void SetLikelihoodElectron(Float_t l) { fLhElectron = l; }
	84	void SetSampledEdx(Float_t q, Int_t i);
	85	void SetSampledEdx(Float_t q);
	86	void SetSeedLabel(Int_t lab) { fSeedLab = lab; }
27eaf44b	87	void SetNWrong(Int_t nwrong) { fNWrong = nwrong; }
	88	void SetNCross(Int_t ncross) { fNCross = ncross; }
	89	void SetNExpected(Int_t nexp) { fNExpected = nexp; }
	90	void SetNLast(Int_t nlast) { fNLast = nlast; }
	91	void SetNExpectedLast(Int_t nexp) { fNExpectedLast = nexp; }
	92	void SetChi2Last(Float_t chi2) { fChi2Last = chi2; }
	93	void SetTracklets(Int_t i, AliTRDtracklet t) { fTracklets[i] = t; }
	94	void SetBudget(Int_t i, Float_t budget) { fBudget[i] = budget; }
fe33d239	95
	96	Int_t PropagateToX(Double_t xr, Double_t step);
	97	Int_t PropagateToR(Double_t xr, Double_t step);
	98	Int_t UpdateMI(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01, Int_t plane);
	99	void MakeBackupTrack();
	100	Bool_t PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
	101	Bool_t Update(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01);
	102	Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE);
	103	void CookdEdx(Double_t low = 0.05, Double_t up = 0.7);
	104	Float_t StatusForTOF();
6c94f330	105
fe33d239	106	protected:
6c94f330	107
fe33d239	108	AliTRDtrack &operator=(const AliTRDtrack &t);
6c94f330	109
fe33d239	110	Double_t GetBz() const;
	111	Bool_t Update(const AliCluster /c/, Double_t /chi2/, Int_t /idx*/) { return 0; }
	112	Double_t GetPredictedChi2(const AliCluster* /c/) const { return 0.0; }
6c94f330	113
fe33d239	114	Int_t fSeedLab; // Track label taken from seeding
	115	Float_t fdEdx; // dE/dx (truncated mean)
	116	Float_t fDE; // Integrated delta energy
	117	Float_t fdEdxPlane[kNplane][kNslice]; // dE/dx from all 6 planes in 3 slices each
	118	Int_t fTimBinPlane[kNplane]; // Time bin of Max cluster from all 6 planes
6c94f330	119
fe33d239	120	Bool_t fStopped; // Track stop indication
	121	Int_t fIndex[kMAXCLUSTERSPERTRACK]; // Global indexes of clusters
	122	Int_t fIndexBackup[kMAXCLUSTERSPERTRACK]; // Backup indexes of clusters - used in iterations
	123	Float_t fdQdl[kMAXCLUSTERSPERTRACK]; // Cluster amplitudes corrected for track angles
b8dc2353	124
fe33d239	125	Float_t fLhElectron; // Likelihood to be an electron
	126	Int_t fNWrong; // Number of wrong clusters
	127	Int_t fNRotate; // Number of rotation
	128	Int_t fNCross; // Number of the cross materials
	129	Int_t fNExpected; // Expected number of cluster
	130	Int_t fNLast; // Number of clusters in last 2 layers
	131	Int_t fNExpectedLast; // Number of expected clusters on last 2 layers
	132	Int_t fNdedx; // Number of clusters for dEdx measurment
	133	Float_t fChi2Last; // Chi2 in the last 2 layers
	134	AliTRDtracklet fTracklets[6]; // Tracklets
	135	Float_t fBudget[3]; // Integrated material budget
	136	AliTRDtrack *fBackupTrack; //! Backup track
	137
	138	ClassDef(AliTRDtrack,8) // TRD reconstructed tracks
46d29e70	139
53c17fbf	140	};
46d29e70	141
46d29e70	142	#endif