Updates (A. Toia)

[u/mrichter/AliRoot.git] / STEER / AliKalmanTrack.h

diff --git a/STEER/AliKalmanTrack.h b/STEER/AliKalmanTrack.h
index 1c637d5b2c749bda648276f583f7b426dacaf2aa..e76538f9ff7cbaae0b466512107d6bd1f4d0d69f 100644 (file)
--- a/STEER/AliKalmanTrack.h
+++ b/STEER/AliKalmanTrack.h
@@ -4,56 +4,97 @@
 /* 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 "AliExternalTrackParam.h"
+#include "AliLog.h"
+#include "AliPID.h"
 
 class AliCluster;
 
-class AliKalmanTrack : public TObject {
+class AliKalmanTrack : public AliExternalTrackParam {
 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(){};
+  AliKalmanTrack& operator=(const AliKalmanTrack &o);
+  void SetLabel(Int_t lab) {fLab=lab;}
+
+  virtual Double_t GetPredictedChi2(const AliCluster *c) const = 0;
+  virtual Bool_t PropagateTo(Double_t xr, Double_t x0, Double_t rho) = 0;
+  virtual Bool_t Update(const AliCluster* c, Double_t chi2, Int_t index) = 0;
+
+  Bool_t   IsSortable() const {return kTRUE;}
+  Int_t    GetLabel()   const {return fLab;}
+  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 Int_t GetNumberOfTracklets() const {
+    AliWarning("Method must be overloaded !");
+    return 0;
+  }
+  virtual Int_t GetTrackletIndex(Int_t) const { //reserved for AliTracker
+    AliWarning("Method must be overloaded !");
+    return -1;
+  } 
+  virtual Double_t GetPIDsignal() const {
+    AliWarning("Method must be overloaded !\n");
+    return 0.;
+  }
+
+  virtual Int_t Compare(const TObject *) const {return 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
+  void GetExternalParameters(Double_t &xr,Double_t p[5]) const {
+    xr=GetX();
+    for (Int_t i=0; i<5; i++) p[i]=GetParameter()[i];
+  }
+  void GetExternalCovariance(Double_t cov[15]) const {
+    for (Int_t i=0; i<15; i++) cov[i]=GetCovariance()[i];
+  }
 
-  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
+  // Time integration (S.Radomski@gsi.de)
+  void StartTimeIntegral();
+  void SetIntegratedLength(Double_t l) {fIntegratedLength=l;}
+  void SetIntegratedTimes(const Double_t *times);
 
-  Double_t fChi2;         // total chi2 value for this track
-  Short_t fN;             // number of clusters 
+  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 SetNumberOfClusters(Int_t n) {fN=n;} 
+
+  void SetFakeRatio(Float_t ratio) {fFakeRatio=ratio;}
+  Float_t  GetFakeRatio()   const {return fFakeRatio;}
+  void SetMass(Double_t mass) {fMass=mass;}
+
+protected:
+  void SetChi2(Double_t chi2) {fChi2=chi2;} 
+
+  Double32_t fFakeRatio;  // fake ratio
+  Double32_t fChi2;       // total chi2 value for this track
+  Double32_t fMass;       // mass hypothesis
+  Int_t fLab;             // track label
+  Int_t fN;               // number of associated clusters
 
-  ClassDef(AliKalmanTrack,1)    // Reconstructed track
+private:
+  Bool_t  fStartTimeIntegral;       // indicator wether integrate time
+  // variables for time integration (S.Radomski@gsi.de)
+  Double32_t fIntegratedTime[AliPID::kSPECIES];       // integrated time
+  Double32_t fIntegratedLength;        // integrated length
+  
+  ClassDef(AliKalmanTrack,7)    // Reconstructed track
 };
 
 #endif