X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FAliKalmanTrack.h;h=99f407013dfb78a887add3c0efb92e16db79f5ce;hb=dbf5aa232b1e3031d520b32f768ba61e3b970745;hp=f58cf1225eb024bc6812a1f82a1f436da10cdf70;hpb=8de978942f115630e5e289b29aedb2edf7c551c5;p=u%2Fmrichter%2FAliRoot.git diff --git a/STEER/AliKalmanTrack.h b/STEER/AliKalmanTrack.h index f58cf1225eb..99f407013df 100644 --- a/STEER/AliKalmanTrack.h +++ b/STEER/AliKalmanTrack.h @@ -8,11 +8,14 @@ //------------------------------------------------------------------------- // Class AliKalmanTrack -// -// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch +// fixed the interface for the derived reconstructed track classes +// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch //------------------------------------------------------------------------- #include +#include "AliLog.h" +#include "AliPID.h" +#include "AliMagF.h" class AliCluster; @@ -23,77 +26,64 @@ public: virtual ~AliKalmanTrack(){}; void SetLabel(Int_t lab) {fLab=lab;} + void SetFakeRatio(Float_t ratio) {fFakeRatio=ratio;} Bool_t IsSortable() const {return kTRUE;} Int_t GetLabel() const {return fLab;} + Float_t GetFakeRatio() const {return fFakeRatio;} Double_t GetChi2() const {return fChi2;} Double_t GetMass() const {return fMass;} Int_t GetNumberOfClusters() const {return fN;} virtual Int_t GetClusterIndex(Int_t) const { //reserved for AliTracker - Warning("GetClusterIndex(Int_t)","Method must be overloaded !\n"); + AliWarning("Method must be overloaded !\n"); return 0; } virtual Double_t GetPIDsignal() const { - Warning("GetPIDsignal()","Method must be overloaded !\n"); + AliWarning("Method must be overloaded !\n"); return 0.; } - virtual Double_t GetDCA(const AliKalmanTrack *,Double_t &,Double_t &) const; + virtual Double_t GetDCA(const AliKalmanTrack *p,Double_t &xthis,Double_t &xp) const; virtual - Double_t PropagateToDCA(AliKalmanTrack *, Double_t d=0., Double_t x0=0.); + Double_t PropagateToDCA(AliKalmanTrack *p, Double_t d=0., Double_t x0=0.); virtual Double_t GetAlpha() const { - Warning("GetAlpha()","Method must be overloaded !\n"); + AliWarning("Method must be overloaded !\n"); return 0.; } virtual Double_t GetSigmaY2() const { - Warning("GetSigmaY2()","Method must be overloaded !\n"); + AliWarning("Method must be overloaded !\n"); return 0.; } virtual Double_t GetSigmaZ2() const { - Warning("GetSigmaZ2()","Method must be overloaded !\n"); + AliWarning("Method must be overloaded !\n"); return 0.; } virtual Int_t Compare(const TObject *) const {return 0;} - virtual void GetExternalParameters(Double_t &/*xr*/, Double_t /*x*/[5]) const {} - virtual void GetExternalCovariance(Double_t /*cov*/[15]) const {} - - virtual Double_t GetX() const; - virtual Double_t GetdEdx() const; - - virtual Double_t GetY() const; - virtual Double_t GetZ() const; - virtual Double_t GetSnp() const; - virtual Double_t GetTgl() const; - virtual Double_t Get1Pt() const; - - virtual Double_t Phi() const; - virtual Double_t SigmaPhi() const; - virtual Double_t Theta() const; - virtual Double_t SigmaTheta() const; - virtual Double_t Eta() const; - virtual Double_t Px() const; - virtual Double_t Py() const; - virtual Double_t Pz() const; - virtual Double_t Pt() const; - virtual Double_t SigmaPt() const; - virtual Double_t P() const; - - virtual Double_t GetPredictedChi2(const AliCluster *) const {return 0.;} - virtual - Int_t PropagateTo(Double_t /*xr*/, Double_t /*x0*/, Double_t /*rho*/) {return 0;} - virtual Int_t Update(const AliCluster*, Double_t /*chi2*/, UInt_t) {return 0;} + virtual void GetExternalParameters(Double_t&/*xr*/,Double_t/*x*/[5]) const=0; + virtual void GetExternalCovariance(Double_t /*cov*/[15]) const = 0; + + virtual Double_t GetPredictedChi2(const AliCluster *) const = 0; + virtual Int_t PropagateTo(Double_t/*xr*/,Double_t/*x0*/,Double_t/*rho*/) = 0; + //virtual Int_t PropagateToVertex(Double_t /*d*/=0., Double_t /*x0*/=0.) = 0; + virtual Int_t Update(const AliCluster*, Double_t /*chi2*/, UInt_t) = 0; - static void SetConvConst(Double_t cc) {fgConvConst=cc;} - static void SetConvConst(); - Double_t GetConvConst() const {return fgConvConst;} + static void SetFieldMap(const AliMagF *map) { fgkFieldMap=map; } + static const AliMagF *GetFieldMap() { return fgkFieldMap; } - static void SetMagneticField(Double_t f) {// f - Magnetic field in T - fgConvConst=100/0.299792458/f; + static void SetUniformFieldTracking() { + if (fgkFieldMap==0) { + printf("AliKalmanTrack: Field map has not been set !\n"); + exit(1); + } + fgConvConst=1000/0.299792458/(fgkFieldMap->SolenoidField()+1e-13); } - Double_t GetMagneticField() const {return 100/0.299792458/fgConvConst;} + static void SetNonuniformFieldTracking() { fgConvConst=0.; } + static Double_t GetConvConst(); + static Double_t MeanMaterialBudget(Double_t *start, Double_t *end, Double_t *mparam); + // Time integration (S.Radomski@gsi.de) void StartTimeIntegral(); void SetIntegratedLength(Double_t l) {fIntegratedLength=l;} @@ -105,30 +95,63 @@ public: Double_t GetIntegratedTime(Int_t pdg) const; Double_t GetIntegratedLength() const {return fIntegratedLength;} void PrintTime() const; - protected: + virtual void GetXYZ(Float_t r[3]) const = 0; + void SaveLocalConvConst(); + Double_t GetLocalConvConst() const; + + void External2Helix(Double_t helix[6]) const; + void SetChi2(Double_t chi2) {fChi2=chi2;} void SetMass(Double_t mass) {fMass=mass;} void SetNumberOfClusters(Int_t n) {fN=n;} - private: Int_t fLab; // track label + Float_t fFakeRatio; // fake ratio Double_t fChi2; // total chi2 value for this track Double_t fMass; // mass hypothesis Int_t fN; // number of associated clusters - static Double_t fgConvConst; //conversion constant cm -> GeV/c +private: + static const AliMagF *fgkFieldMap;//pointer to the magnetic field map + static Double_t fgConvConst; //conversion "curvature(1/cm) -> pt(GeV/c)" + Double_t fLocalConvConst; //local conversion "curvature(1/cm) -> pt(GeV/c)" // variables for time integration (S.Radomski@gsi.de) - static const Int_t fgkTypes = 5; // Number of track types (e,mu,pi,k,p) Bool_t fStartTimeIntegral; // indicator wether integrate time - Double_t fIntegratedTime[5]; // integrated time + Double_t fIntegratedTime[AliPID::kSPECIES]; // integrated time Double_t fIntegratedLength; // integrated length - ClassDef(AliKalmanTrack,2) // Reconstructed track + ClassDef(AliKalmanTrack,4) // Reconstructed track }; +inline Double_t AliKalmanTrack::GetConvConst() { +// +// For backward compatibility only ! +// + if (fgConvConst > 0 || fgConvConst < 0) return fgConvConst; + return 1000/0.299792458/(fgkFieldMap->SolenoidField()+1e-13); +} + +inline void AliKalmanTrack::SaveLocalConvConst() { + //--------------------------------------------------------------------- + // Saves local conversion constant "curvature (1/cm) -> pt (GeV/c)" + //--------------------------------------------------------------------- + if (fgConvConst > 0 || fgConvConst < 0) return; //uniform field tracking + Float_t r[3]={0.,0.,0.}; GetXYZ(r); + Float_t b[3]; fgkFieldMap->Field(r,b); + fLocalConvConst=1000/0.299792458/(1e-13 - b[2]); +} + +inline Double_t AliKalmanTrack::GetLocalConvConst() const { + //--------------------------------------------------------------------- + // Returns conversion constant "curvature (1/cm) -> pt (GeV/c)" + //--------------------------------------------------------------------- + if (fgConvConst > 0 || fgConvConst < 0) return fgConvConst; //uniform field tracking + return fLocalConvConst; +} + #endif