[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;

const unsigned kMAXCLUSTERSPERTRACK = 210; 

class AliTRDtrack : public AliKalmanTrack {
   
  enum { kNdet      = 540
       , kNstacks   =  90
       , kNplane    =   6
       , kNcham     =   5
       , kNsect     =  18
       , kNslice    =   3
       , kNtimeBins =  22 };

  friend class AliTRDtracker;

 public:

   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;                        }
           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    GetNWrong() const                                       { return fNWrong;                      }
           Int_t    GetNRotate() const                                      { return fNRotate;                     }
           Int_t    GetNCross() const                                       { return fNCross;                      }
  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;    }

           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;
15ed8ba1	21
	22	const unsigned kMAXCLUSTERSPERTRACK = 210;
	23
7ad19338	24	class AliTRDtrack : public AliKalmanTrack {
53c17fbf	25
6d45eaef	26	enum { kNdet = 540
15ed8ba1	27	, kNstacks = 90
	28	, kNplane = 6
	29	, kNcham = 5
	30	, kNsect = 18
	31	, kNslice = 3
	32	, kNtimeBins = 22 };
7ad19338	33
7ad19338	34	friend class AliTRDtracker;
46d29e70	35
53c17fbf	36	public:
53c17fbf	37
6c94f330	38	AliTRDtrack();
fe33d239	39	AliTRDtrack(const AliTRDcluster *c, Int_t index, const Double_t xx[5]
	40	,const Double_t cc[15], Double_t xr, Double_t alpha);
	41	AliTRDtrack(const AliTRDtrack &t);
	42	AliTRDtrack(const AliESDtrack &t);
6c94f330	43	~AliTRDtrack();
fe33d239	44	AliTRDtrack(const AliKalmanTrack &t, Double_t alpha);
	45
	46	void ResetClusters() { SetChi2(0.0);
	47	SetNumberOfClusters(0); }
	48	Int_t Compare(const TObject *o) const;
	49	void AddNWrong() { fNWrong++; }
	50	void IncCross() { fNCross++;
	51	if (fBackupTrack) fBackupTrack->IncCross(); }
	52
	53	Int_t GetSector() const;
	54	Float_t GetClusterdQdl(Int_t i) const { return fdQdl[i]; }
	55	Double_t GetdEdx() const { return fdEdx; }
	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; }
	68	Int_t GetNWrong() const { return fNWrong; }
	69	Int_t GetNRotate() const { return fNRotate; }
	70	Int_t GetNCross() const { return fNCross; }
	71	AliTRDtrack *GetBackupTrack() { return fBackupTrack; }
	72
	73	void SetdEdx(Double_t dedx) { fdEdx = dedx; }
	74	void SetStop(Bool_t stop) { fStopped = stop; }
	75	void SetPIDsignals(Float_t dedx, Int_t iPlane, Int_t iSlice) { fdEdxPlane[iPlane][iSlice] = dedx; }
	76	void SetPIDTimBin(Int_t timbin, Int_t i) { fTimBinPlane[i] = timbin; }
	77	void SetLikelihoodElectron(Float_t l) { fLhElectron = l; }
	78	void SetSampledEdx(Float_t q, Int_t i);
	79	void SetSampledEdx(Float_t q);
	80	void SetSeedLabel(Int_t lab) { fSeedLab = lab; }
	81
	82	Int_t PropagateToX(Double_t xr, Double_t step);
	83	Int_t PropagateToR(Double_t xr, Double_t step);
	84	Int_t UpdateMI(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01, Int_t plane);
	85	void MakeBackupTrack();
	86	Bool_t PropagateTo(Double_t xr, Double_t x0 = 8.72, Double_t rho = 5.86e-3);
	87	Bool_t Update(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01);
	88	Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE);
	89	void CookdEdx(Double_t low = 0.05, Double_t up = 0.7);
	90	Float_t StatusForTOF();
6c94f330	91
fe33d239	92	protected:
6c94f330	93
fe33d239	94	AliTRDtrack &operator=(const AliTRDtrack &t);
6c94f330	95
fe33d239	96	Double_t GetBz() const;
	97	Bool_t Update(const AliCluster /c/, Double_t /chi2/, Int_t /idx*/) { return 0; }
	98	Double_t GetPredictedChi2(const AliCluster* /c/) const { return 0.0; }
6c94f330	99
fe33d239	100	Int_t fSeedLab; // Track label taken from seeding
	101	Float_t fdEdx; // dE/dx (truncated mean)
	102	Float_t fDE; // Integrated delta energy
	103	Float_t fdEdxPlane[kNplane][kNslice]; // dE/dx from all 6 planes in 3 slices each
	104	Int_t fTimBinPlane[kNplane]; // Time bin of Max cluster from all 6 planes
6c94f330	105
fe33d239	106	Bool_t fStopped; // Track stop indication
	107	Int_t fIndex[kMAXCLUSTERSPERTRACK]; // Global indexes of clusters
	108	Int_t fIndexBackup[kMAXCLUSTERSPERTRACK]; // Backup indexes of clusters - used in iterations
	109	Float_t fdQdl[kMAXCLUSTERSPERTRACK]; // Cluster amplitudes corrected for track angles
b8dc2353	110
fe33d239	111	Float_t fLhElectron; // Likelihood to be an electron
	112	Int_t fNWrong; // Number of wrong clusters
	113	Int_t fNRotate; // Number of rotation
	114	Int_t fNCross; // Number of the cross materials
	115	Int_t fNExpected; // Expected number of cluster
	116	Int_t fNLast; // Number of clusters in last 2 layers
	117	Int_t fNExpectedLast; // Number of expected clusters on last 2 layers
	118	Int_t fNdedx; // Number of clusters for dEdx measurment
	119	Float_t fChi2Last; // Chi2 in the last 2 layers
	120	AliTRDtracklet fTracklets[6]; // Tracklets
	121	Float_t fBudget[3]; // Integrated material budget
	122	AliTRDtrack *fBackupTrack; //! Backup track
	123
	124	ClassDef(AliTRDtrack,8) // TRD reconstructed tracks
46d29e70	125
53c17fbf	126	};
46d29e70	127
46d29e70	128	#endif