]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliExternalTrackParam.h
Record changes.
[u/mrichter/AliRoot.git] / STEER / AliExternalTrackParam.h
index 08e2211d6022735928e209b732a1c82cd2b7e2d7..fbf32af269c2196a557d1de8e177e693577118aa 100644 (file)
 
 /* $Id$ */
 
-
+/*****************************************************************************
+ *              "External" track parametrisation class                       *
+ *                                                                           *
+ *      external param0:   local Y-coordinate of a track (cm)                *
+ *      external param1:   local Z-coordinate of a track (cm)                *
+ *      external param2:   local sine of the track momentum azimuthal angle  *
+ *      external param3:   tangent of the track momentum dip angle           *
+ *      external param4:   1/pt (1/(GeV/c))                                  *
+ *                                                                           *
+ * The parameters are estimated at an exact position x in a local coord.     *
+ * system rotated by angle alpha with respect to the global coord.system.    *
+ *        Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch                     *
+ *****************************************************************************/
 #include "TObject.h"
-#include "AliMagF.h"
-#include "TVector3.h"
+#include "TMath.h"
+
+const Double_t kAlmost1=0.999;
+const Double_t kAlmost0=1e-33;
+const Double_t kVeryBig=1./kAlmost0;
+
+const Double_t kB2C=0.299792458e-3;
+const Double_t kAlmost0Field=1.e-13;
+const Double_t kMostProbablePt=0.35;
 
-class AliKalmanTrack;
-class AliTrackReference;
+class AliESDVertex;
+
+Double_t ApproximateBetheBloch(Double_t);
 
 class AliExternalTrackParam: public TObject {
  public:
   AliExternalTrackParam();
+  AliExternalTrackParam(const AliExternalTrackParam &);
   AliExternalTrackParam(Double_t x, Double_t alpha, 
                        const Double_t param[5], const Double_t covar[15]);
-  AliExternalTrackParam(const AliKalmanTrack& track);
-  static AliExternalTrackParam * MakeTrack(const AliTrackReference *ref, Double_t mass);
-
-  virtual const Double_t* GetParameter() const;
-  virtual const Double_t* GetCovariance() const;
-  virtual Double_t     X() const {return fX;};
-  virtual Double_t     Alpha() const {return fAlpha;};
-  virtual AliExternalTrackParam* CreateExternalParam() const;
-  virtual void         ResetCovariance(Double_t factor = 10.,
-                                      Bool_t clearOffDiagonal = kTRUE);
-  virtual Double_t     Y() const {return fParam[0];};
-  virtual Double_t     Z() const {return fParam[1];};
-  virtual void         GetXYZ(Float_t r[3]) const;
-  virtual void         GetGlobalXYZ(Double_t &x, Double_t &y, Double_t &z ) const;
-  virtual Bool_t       PropagateTo(Double_t x,  Double_t x0, Double_t rho);
-  virtual Bool_t       PropagateToDCA(Double_t x, Double_t y,  Double_t x0, Double_t rho);
-  virtual Bool_t       RotateTo(Double_t alpha);
-  virtual Bool_t       CorrectForMaterial(Double_t d, Double_t x0, Double_t rho);
-  virtual Bool_t       GetProlongationAt(Double_t x, Double_t& y, 
-                                        Double_t& z) const;
-  virtual Double_t     GetXAtVertex(Double_t x = 0, Double_t y = 0) const;
-
-  //  virtual Double_t     GetPredictedChi2(const AliCluster* cluster);
-  //  virtual Bool_t       Update(const AliCluster* cluster);
-
-  virtual Double_t     SigmaPhi() const;
-  virtual Double_t     SigmaTheta() const;
-  virtual Double_t     SigmaPt() const;
-  virtual TVector3     Momentum() const;
-  virtual TVector3     Position() const;
-
-  virtual void         Print(Option_t* option = "") const;
-  // local magnetic field manipulation 
-  void     SaveLocalConvConst();
-  Double_t GetLocalConvConst() const;
-  
-  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.; }
-
- private:
-  Double_t             fMass;       // mass associated to the particle
-  Double_t             fX;          // x coordinate for the parametrisation
-  Double_t             fAlpha;      // azimuthal angle for the parametrisation
-  Double_t             fParam[5];   // track parameter (y, z, sin(azimuthal angel), tan(dip angle), 1/pt)
-  Double_t             fCovar[15];  // track parameter covariance
-  //
-  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)"
+  virtual ~AliExternalTrackParam(){}
 
+  void Set(Double_t x,Double_t alpha,
+                       const Double_t param[5], const Double_t covar[15]);
+  void Reset();
+  void ResetCovariance(Double_t s2) {
+    fC[0]*= s2;
+    fC[1] = 0.;  fC[2]*= s2;
+    fC[3] = 0.;  fC[4] = 0.;  fC[5]*= s2;
+    fC[6] = 0.;  fC[7] = 0.;  fC[8] = 0.;  fC[9]*= s2;
+    fC[10]= 0.;  fC[11]= 0.;  fC[12]= 0.;  fC[13]= 0.;  fC[14]*=s2;
+  }
 
-  ClassDef(AliExternalTrackParam, 3)
+  const Double_t *GetParameter() const {return fP;}
+  const Double_t *GetCovariance() const {return fC;}
+
+  Double_t GetAlpha() const {return fAlpha;}
+  Double_t GetX() const {return fX;}
+  Double_t GetY()    const {return fP[0];}
+  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 GetSigmaY2() const {return fC[0];}
+  Double_t GetSigmaZY() const {return fC[1];}
+  Double_t GetSigmaZ2() const {return fC[2];}
+  Double_t GetSigmaSnpY() const {return fC[3];}
+  Double_t GetSigmaSnpZ() const {return fC[4];}
+  Double_t GetSigmaSnp2() const {return fC[5];}
+  Double_t GetSigmaTglY() const {return fC[6];}
+  Double_t GetSigmaTglZ() const {return fC[7];}
+  Double_t GetSigmaTglSnp() const {return fC[8];}
+  Double_t GetSigmaTgl2() const {return fC[9];}
+  Double_t GetSigma1PtY() const {return fC[10];}
+  Double_t GetSigma1PtZ() const {return fC[11];}
+  Double_t GetSigma1PtSnp() const {return fC[12];}
+  Double_t GetSigma1PtTgl() const {return fC[13];}
+  Double_t GetSigma1Pt2() const {return fC[14];}
+
+  Double_t GetSign() const {return (fP[4]>0) ? 1 : -1;}
+  Double_t GetP() const;
+  Double_t GetPt() const {
+    return (TMath::Abs(fP[4])>kAlmost0) ? 1./fP[4]:TMath::Sign(kVeryBig,fP[4]);
+  }
+  Double_t Get1P() const;
+  Double_t GetC(Double_t b) const {return fP[4]*b*kB2C;}
+  void GetDZ(Double_t x,Double_t y,Double_t z,Double_t b,Float_t dz[2]) const; 
+  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 CorrectForMaterial(Double_t d, Double_t x0, Double_t mass,
+                           Double_t (*f)(Double_t)=ApproximateBetheBloch);
+  Double_t GetPredictedChi2(Double_t p[2],Double_t cov[3]) 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);
+  Bool_t Propagate(Double_t alpha, Double_t x, Double_t b) {
+    if (Rotate(alpha))
+      if (PropagateTo(x,b)) return kTRUE;
+    return kFALSE;
+  }
+  void   Propagate(Double_t len,Double_t x[3],Double_t p[3],Double_t bz) const;
+  Bool_t Intersect(Double_t pnt[3], Double_t norm[3], Double_t bz) const;
+
+  void GetHelixParameters(Double_t h[6], Double_t b) const;
+  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);
+
+  void GetDirection(Double_t d[3]) const;
+  Bool_t GetPxPyPz(Double_t *p) const;
+  Bool_t GetXYZ(Double_t *p) const;
+  Bool_t GetCovarianceXYZPxPyPz(Double_t cv[21]) const;
+  Bool_t GetPxPyPzAt(Double_t x, Double_t b, Double_t p[3]) const;
+  Bool_t GetXYZAt(Double_t x, Double_t b, Double_t r[3]) const;
+  Bool_t GetYAt(Double_t x,  Double_t b,  Double_t &y) const;
+  Bool_t GetZAt(Double_t x,  Double_t b,  Double_t &z) const;
+  void Print(Option_t* option = "") const;
+  Double_t GetSnpAt(Double_t x,Double_t b) const;
+
+protected:
+  Double_t &Par(Int_t i) {return fP[i];}
+  Double_t &Cov(Int_t i) {return fC[i];}
+
+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
+  Double32_t           fC[15]; // The track parameter covariance matrix
+
+  ClassDef(AliExternalTrackParam, 5)
 };
 
-
-inline void AliExternalTrackParam::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 AliExternalTrackParam::GetLocalConvConst() const {
-  //---------------------------------------------------------------------
-  // Returns conversion constant "curvature (1/cm) -> pt (GeV/c)" 
-  //---------------------------------------------------------------------
-     if (fgConvConst > 0 || fgConvConst < 0) return fgConvConst; //uniform field tracking
-     return fLocalConvConst;
-} 
-
-
-inline void AliExternalTrackParam::GetXYZ(Float_t r[3]) const {
-  //---------------------------------------------------------------------
-  // Returns the position of the track in the global coord. system 
-  //---------------------------------------------------------------------
-  Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha);
-  r[0]=fX*cs - fParam[0]*sn; r[1]=fX*sn + fParam[0]*cs; r[2]=fParam[1];
-}
-
-inline void AliExternalTrackParam::GetGlobalXYZ(Double_t &x, Double_t &y, Double_t &z) const {
-  //---------------------------------------------------------------------
-  // Returns the position of the track in the global coord. system 
-  //---------------------------------------------------------------------
-  Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha);
-  x=fX*cs - fParam[0]*sn; y=fX*sn + fParam[0]*cs; z=fParam[1];
-}
-
-
-
-
 #endif