]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliKalmanTrack.h
Correct overloading of virtual functions in the derived classes (icc)
[u/mrichter/AliRoot.git] / STEER / AliKalmanTrack.h
index 1c637d5b2c749bda648276f583f7b426dacaf2aa..99f407013dfb78a887add3c0efb92e16db79f5ce 100644 (file)
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
+/* $Id$ */
+
 //-------------------------------------------------------------------------
 //                          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 <TObject.h>
+#include "AliLog.h"
+#include "AliPID.h"
+#include "AliMagF.h"
 
 class AliCluster;
 
 class AliKalmanTrack : public TObject {
 public:
-  AliKalmanTrack() {fN=0; fChi2=0; fLab=-3141593;}
-  AliKalmanTrack(const AliKalmanTrack& t);
-  virtual ~AliKalmanTrack() {}
-  Int_t Compare(const TObject *o) const;
-  void SetLabel(Int_t lab) {fLab=lab;} 
-
-  Double_t GetPredictedChi2(const AliCluster *cluster) const;
-  Bool_t IsSortable() const {return kTRUE;}
-  Int_t GetLabel() const {return fLab;}
-  void GetCovariance(Double_t cov[15]) const;
-  Double_t GetChi2() const {return fChi2;}
-  Int_t GetNumberOfClusters() const {return fN;}
-
- virtual Double_t GetPt() const=0;
- virtual Double_t GetP()  const=0;
- virtual void GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const=0;
- virtual Int_t PropagateTo(Double_t xr,Double_t x0,Double_t rho,Double_t pm)=0;
- virtual void Update(const AliCluster* c, Double_t chi2, UInt_t i)=0;
-
-protected: 
-  Int_t fLab;             // track label
+  AliKalmanTrack();
+  AliKalmanTrack(const AliKalmanTrack &t);
+
+  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
+    AliWarning("Method must be overloaded !\n");
+    return 0;
+  } 
+  virtual Double_t GetPIDsignal() const {
+    AliWarning("Method must be overloaded !\n");
+    return 0.;
+  }
+
+  virtual Double_t GetDCA(const AliKalmanTrack *p,Double_t &xthis,Double_t &xp) const; 
+  virtual 
+  Double_t PropagateToDCA(AliKalmanTrack *p, Double_t d=0., Double_t x0=0.); 
+  virtual Double_t GetAlpha() const {
+    AliWarning("Method must be overloaded !\n");
+    return 0.;
+  }
+  virtual Double_t GetSigmaY2() const {
+    AliWarning("Method must be overloaded !\n");
+    return 0.;
+  }
+  virtual Double_t GetSigmaZ2() const {
+    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=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 SetFieldMap(const AliMagF *map) { fgkFieldMap=map; }
+  static const AliMagF *GetFieldMap() { return fgkFieldMap; }
+
+  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);
+  }
+  static void SetNonuniformFieldTracking() { fgConvConst=0.; }
 
-  Double_t fP0;           // track parameter
-  Double_t fP1;           // track parameter
-  Double_t fP2;           // track parameter
-  Double_t fP3;           // track parameter
-  Double_t fP4;           // track parameter
+  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;}
+  void SetIntegratedTimes(const Double_t *times);
 
-  Double_t fC00;                         // covariance
-  Double_t fC10, fC11;                   // matrix
-  Double_t fC20, fC21, fC22;             // of the
-  Double_t fC30, fC31, fC32, fC33;       // track
-  Double_t fC40, fC41, fC42, fC43, fC44; // parameters
+  Bool_t IsStartedTimeIntegral() const {return fStartTimeIntegral;}
+  void     AddTimeStep(Double_t length);
+  void GetIntegratedTimes(Double_t *times) const;
+  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;} 
+
+  Int_t fLab;             // track label
+  Float_t fFakeRatio;     // fake ratio
   Double_t fChi2;         // total chi2 value for this track
-  Short_t fN;             // number of clusters 
+  Double_t fMass;         // mass hypothesis
+  Int_t fN;               // number of associated clusters
+
+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)"
 
-  ClassDef(AliKalmanTrack,1)    // Reconstructed track
+  // variables for time integration (S.Radomski@gsi.de)
+  Bool_t  fStartTimeIntegral;       // indicator wether integrate time
+  Double_t fIntegratedTime[AliPID::kSPECIES];       // integrated time
+  Double_t fIntegratedLength;        // integrated length
+  
+  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