New more general analysis implemention for particle identification and correlation...

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

diff --git a/STEER/AliExternalTrackParam.h b/STEER/AliExternalTrackParam.h
index eb7650deaa044444039b78f1a88e5c8944739450..896b877e3cac20805c3409b6341cac29d27ee3fd 100644 (file)
--- a/STEER/AliExternalTrackParam.h
+++ b/STEER/AliExternalTrackParam.h
@@ -18,25 +18,23 @@
  * system rotated by angle alpha with respect to the global coord.system.    *
  *        Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch                     *
  *****************************************************************************/
-#include "TObject.h"
 #include "TMath.h"
 
-const Double_t kAlmost1=0.999;
-const Double_t kAlmost0=1e-33;
-const Double_t kVeryBig=1./kAlmost0;
+#include "AliVParticle.h"
 
-const Double_t kB2C=0.299792458e-3;
-const Double_t kAlmost0Field=1.e-13;
+const Double_t kVeryBig=1./kAlmost0;
 const Double_t kMostProbablePt=0.35;
 
-class AliESDVertex;
-
 Double_t ApproximateBetheBloch(Double_t);
 
-class AliExternalTrackParam: public TObject {
+class AliESDVertex;
+class TPolyMarker3D; 
+
+class AliExternalTrackParam: public AliVParticle {
  public:
   AliExternalTrackParam();
   AliExternalTrackParam(const AliExternalTrackParam &);
+  AliExternalTrackParam& operator=(const AliExternalTrackParam & trkPar);
   AliExternalTrackParam(Double_t x, Double_t alpha, 
                        const Double_t param[5], const Double_t covar[15]);
   virtual ~AliExternalTrackParam(){}
@@ -59,7 +57,7 @@ class AliExternalTrackParam: public TObject {
   Double_t GetZ()    const {return fP[1];}
   Double_t GetSnp()  const {return fP[2];}
   Double_t GetTgl()  const {return fP[3];}
-  Double_t Get1Pt()  const {return fP[4];}
+  Double_t GetSigned1Pt()  const {return fP[4];}
 
   Double_t GetSigmaY2() const {return fC[0];}
   Double_t GetSigmaZY() const {return fC[1];}
@@ -77,9 +75,32 @@ class AliExternalTrackParam: public TObject {
   Double_t GetSigma1PtTgl() const {return fC[13];}
   Double_t GetSigma1Pt2() const {return fC[14];}
 
+  // additional functions for AliVParticle
+  Double_t Px() const;
+  Double_t Py() const;
+  Double_t Pz() const;
+  Double_t Pt() const { return TMath::Abs(GetSignedPt()); }
+  Double_t P() const { return GetP(); }
+  Bool_t   PxPyPz(Double_t p[3]) const { return GetPxPyPz(p); }
+  
+  Double_t Xv() const;
+  Double_t Yv() const;
+  Double_t Zv() const;
+  Bool_t   XvYvZv(Double_t x[3]) const { return GetXYZ(x); }
+
+  Double_t OneOverPt() const { return 1./Pt(); }
+  Double_t Phi() const;
+  Double_t Theta() const;
+  virtual Double_t E() const;
+  virtual Double_t M() const;
+  Double_t Eta() const;
+  virtual Double_t Y() const;
+  Short_t  Charge() const { return (Short_t)GetSign(); }
+  const Double_t *PID() const { return 0x0; }
+
   Double_t GetSign() const {return (fP[4]>0) ? 1 : -1;}
   Double_t GetP() const;
-  Double_t GetPt() const {
+  Double_t GetSignedPt() const {
     return (TMath::Abs(fP[4])>kAlmost0) ? 1./fP[4]:TMath::Sign(kVeryBig,fP[4]);
   }
   Double_t Get1P() const;
@@ -88,7 +109,8 @@ class AliExternalTrackParam: public TObject {
   Double_t GetD(Double_t xv, Double_t yv, Double_t b) const; 
   Double_t GetLinearD(Double_t xv, Double_t yv) const; 
   Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho, 
-        Double_t mass, Double_t (*f)(Double_t)=ApproximateBetheBloch);
+        Double_t mass,  Bool_t anglecorr=kFALSE,
+       Double_t (*f)(Double_t)=ApproximateBetheBloch);
   Double_t GetPredictedChi2(Double_t p[2],Double_t cov[3]) const;
 
   Double_t 
@@ -96,6 +118,7 @@ class AliExternalTrackParam: public TObject {
   Bool_t 
     PropagateTo(Double_t p[3],Double_t covyz[3],Double_t covxyz[3],Double_t b);
 
+  Double_t *GetResiduals(Double_t *p,Double_t *cov,Bool_t updated=kTRUE) const;
   Bool_t Update(Double_t p[2],Double_t cov[3]);
   Bool_t Rotate(Double_t alpha);
   Bool_t PropagateTo(Double_t x, Double_t b);
@@ -111,8 +134,9 @@ class AliExternalTrackParam: public TObject {
   Double_t GetDCA(const AliExternalTrackParam *p, Double_t b,
     Double_t &xthis,Double_t &xp) const;
   Double_t PropagateToDCA(AliExternalTrackParam *p, Double_t b);
-  Bool_t PropagateToDCA(const AliESDVertex *vtx, Double_t b, Double_t maxd);
-
+  Bool_t PropagateToDCA(const AliESDVertex *vtx, Double_t b, Double_t maxd,
+                        Double_t dz[2]=0, Double_t cov[3]=0);
+  
   void GetDirection(Double_t d[3]) const;
   Bool_t GetPxPyPz(Double_t *p) const;
   Bool_t GetXYZ(Double_t *p) const;
@@ -128,11 +152,18 @@ class AliExternalTrackParam: public TObject {
   Bool_t CorrectForMaterial(Double_t d, Double_t x0, Double_t mass,
                            Double_t (*f)(Double_t)=ApproximateBetheBloch);
 
-protected:
+  Bool_t GetDistance(AliExternalTrackParam *param2, Double_t x, Double_t dist[3], Double_t b);
+  Int_t GetIndex(Int_t i, Int_t j) const {return i+(j+1)*j/2;}
+  //
+  // visualization (M. Ivanov)
+  //
+  virtual void FillPolymarker(TPolyMarker3D *pol, Float_t magf, Float_t minR, Float_t maxR, Float_t stepR);
+  virtual void DrawTrack(Float_t magF, Float_t minR, Float_t maxR, Float_t stepR);
+ protected:
   Double_t &Par(Int_t i) {return fP[i];}
   Double_t &Cov(Int_t i) {return fC[i];}
-
-private:
+  
+ private:
   Double32_t           fX;     // X coordinate for the point of parametrisation
   Double32_t           fAlpha; // Local <-->global coor.system rotation angle
   Double32_t           fP[5];  // The track parameters
@@ -140,7 +171,7 @@ private:
 
   static Double32_t    fgMostProbablePt; // "Most probable" pt
                                          // (to be used if Bz=0)
-  ClassDef(AliExternalTrackParam, 6)
+  ClassDef(AliExternalTrackParam, 7)
 };
 
 inline void AliExternalTrackParam::ResetCovariance(Double_t s2) {
@@ -154,4 +185,7 @@ inline void AliExternalTrackParam::ResetCovariance(Double_t s2) {
     fC[10]= 0.;  fC[11]= 0.;  fC[12]= 0.;  fC[13]= 0.;  fC[14]*=s2;
 }
 
+
+
+
 #endif