X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDtrack.h;h=486757a2e781540d6b54937ebdaea37cb29648b8;hb=2ed19889577edad1ad764a222d7f351ef6014034;hp=540bf8f6e48704f1e980a7c062226169b046228c;hpb=a2b90f83f274c4e3f873d61e00fa82a1a7715150;p=u%2Fmrichter%2FAliRoot.git diff --git a/TRD/AliTRDtrack.h b/TRD/AliTRDtrack.h index 540bf8f6e48..486757a2e78 100644 --- a/TRD/AliTRDtrack.h +++ b/TRD/AliTRDtrack.h @@ -1,99 +1,160 @@ #ifndef ALITRDTRACK_H -#define ALITRDTRACK_H +#define ALITRDTRACK_H /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * - * See cxx source for full Copyright notice */ + * See cxx source for full Copyright notice */ -#include +/* $Id$ */ +/////////////////////////////////////////////////////////////////////////////// +// // +// Represents a reconstructed TRD track // +// // +/////////////////////////////////////////////////////////////////////////////// + +#include "AliKalmanTrack.h" + +#include "AliTRDtracklet.h" + +class AliESDtrack; +class AliTrackReference; class AliTRDcluster; -class AliTRDtrack : public TObject { - -// Represents reconstructed TRD track - -public: - - AliTRDtrack() { fN=0;} - AliTRDtrack(UInt_t index, const Double_t xx[5], - const Double_t cc[15], Double_t xr, Double_t alpha); - AliTRDtrack(const AliTRDtrack& t); - - Int_t Compare(const TObject *o) const; - - Double_t GetAlpha() const {return fAlpha;} - Double_t GetC() const {return fC;} - Int_t GetClusterIndex(Int_t i) const {return fIndex[i];} - Double_t GetChi2() const {return fChi2;} - void GetCovariance(Double_t cov[15]) const; - Double_t GetdEdX() const {return fdEdx;} - Double_t GetEta() const {return fE;} - Int_t GetLabel() const {return fLab;} - Int_t GetNclusters() const {return fN;} - Double_t GetP() const { - return TMath::Abs(GetPt())*sqrt(1.+GetTgl()*GetTgl()); - } - Double_t GetPredictedChi2(const AliTRDcluster*) const ; - //Double_t GetPt() const {return 0.3*0.2/GetC()/100;} - Double_t GetPt() const {return 0.3*0.4/GetC()/100;} - void GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const ; - Double_t GetSigmaC2() const {return fCcc;} - Double_t GetSigmaTgl2() const {return fCtt;} - Double_t GetSigmaY2() const {return fCyy;} - Double_t GetSigmaZ2() const {return fCzz;} - Double_t GetTgl() const {return fT;} - Double_t GetX() const {return fX;} - Double_t GetY() const {return fY;} // returns running Y - Double_t GetZ() const {return fZ;} - - Float_t GetLikelihoodPion() const { return fLhPion; }; - Float_t GetLikelihoodElectron() const { return fLhElectron; }; - - Bool_t IsSortable() const {return kTRUE;} - - Int_t PropagateTo(Double_t xr, - Double_t x0=8.72,Double_t rho=5.86e-3,Double_t pm=0.139); - void PropagateToVertex( - Double_t x0=36.66,Double_t rho=1.2e-3,Double_t pm=0.139); - Int_t Rotate(Double_t angle); - - void SetLabel(Int_t lab=0) {fLab=lab;} - void SetdEdx(Float_t dedx) {fdEdx=dedx;} - - void Update(const AliTRDcluster* c, Double_t chi2, UInt_t i); - - void SetLikelihoodPion(Float_t l) { fLhPion = l; }; - void SetLikelihoodElectron(Float_t l) { fLhElectron = l; }; - -protected: - - Int_t fLab; // track label - Double_t fChi2; // total chi2 value for the track - Float_t fdEdx; // dE/dx - - Double_t fAlpha; // rotation angle - Double_t fX; // running local X-coordinate of the track (time bin) - - Double_t fY; // Y-coordinate of the track - Double_t fZ; // Z-coordinate of the track - Double_t fC; // track curvature - Double_t fE; // C*x0 - Double_t fT; // tangent of the track dip angle - - Double_t fCyy; // covariance - Double_t fCzy, fCzz; // matrix - Double_t fCcy, fCcz, fCcc; // of the - Double_t fCey, fCez, fCec, fCee; // track - Double_t fCty, fCtz, fCtc, fCte, fCtt; // parameters - - Short_t fN; // number of clusters associated with the track - UInt_t fIndex[200]; // global indexes of these clusters - - Float_t fLhElectron; // Likelihood to be an electron - Float_t fLhPion; // Likelihood to be a pion - - ClassDef(AliTRDtrack,2) // TRD reconstructed tracks -}; +const unsigned kMAXCLUSTERSPERTRACK = 210; + +class AliTRDtrack : public AliKalmanTrack { + + public: + + enum { kNdet = 540 + , kNstacks = 90 + , kNplane = 6 + , kNcham = 5 + , kNsect = 18 + , kNslice = 3}; + + 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/*, const Bool_t owner = kTRUE*/); + 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; } + +// A. Bercuci new functions + void SetTrackSegmentDirMom(const Int_t plane); + void CookdEdx(Double_t low = 0.05, Double_t up = 0.7); + void CookdEdxTimBin(); + Int_t CookPID(AliESDtrack *p); + void SetCluster(AliTRDcluster* cl, Int_t index = -1) {fClusters[(index == -1) ? GetNumberOfClusters()-1 : index] = cl;} + inline AliTRDcluster* GetCluster(Int_t layer){ return (layer >= 0 && layer < GetNumberOfClusters()) ? fClusters[layer] : 0x0;} + inline Float_t GetMomentumPlane(Int_t plane) const {return (plane >= 0 && plane < kNplane) ? fMom[plane] : 0.;} + inline Float_t GetSnpPlane(Int_t plane) const {return (plane >= 0 && plane < kNplane) ? fSnp[plane] : 0.;} + inline Float_t GetTglPlane(Int_t plane) const {return (plane >= 0 && plane < kNplane) ? fTgl[plane] : 0.;} + Float_t GetTrackLengthPlane(Int_t plane) const; +//end A.Bercuci + + void MakeBackupTrack(); + 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); + Int_t PropagateToX(Double_t xr, Double_t step); + Bool_t Rotate(Double_t angle, Bool_t absolute = kFALSE); + Float_t StatusForTOF(); + Bool_t Update(const AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01); + Int_t UpdateMI(/*const */AliTRDcluster *c, Double_t chi2, Int_t i, Double_t h01, Int_t plane); + + 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 + + // A.Bercuci + Float_t fMom[kNplane]; // Track momentum at chamber entrance + Float_t fSnp[kNplane]; // track direction + Float_t fTgl[kNplane]; // track direction + AliTRDcluster* fClusters[kMAXCLUSTERSPERTRACK]; + Bool_t fClusterOwner; // indicates the track is owner of cluster + // end A.Bercuci + + 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