STEER/AliExternalTrackParam.h

   1 #ifndef ALIEXTERNALTRACKPARAM_H
   2 #define ALIEXTERNALTRACKPARAM_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 /*****************************************************************************
   9  *              "External" track parametrisation class                       *
  10  *                                                                           *
  11  *      external param0:   local Y-coordinate of a track (cm)                *
  12  *      external param1:   local Z-coordinate of a track (cm)                *
  13  *      external param2:   local sine of the track momentum azimuthal angle  *
  14  *      external param3:   tangent of the track momentum dip angle           *
  15  *      external param4:   1/pt (1/(GeV/c))                                  *
  16  *                                                                           *
  17  * The parameters are estimated at an exact position x in a local coord.     *
  18  * system rotated by angle alpha with respect to the global coord.system.    *
  19  *        Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch                     *
  20  *****************************************************************************/
  21 #include "TMath.h"
  22
  23 #include "AliVParticle.h"
  24
  25 const Double_t kVeryBig=1./kAlmost0;
  26 const Double_t kMostProbablePt=0.35;
  27
  28 Double_t ApproximateBetheBloch(Double_t);
  29
  30 class AliESDVertex;
  31 class TPolyMarker3D;
  32
  33 class AliExternalTrackParam: public AliVParticle {
  34  public:
  35   AliExternalTrackParam();
  36   AliExternalTrackParam(const AliExternalTrackParam &);
  37   AliExternalTrackParam& operator=(const AliExternalTrackParam & trkPar);
  38   AliExternalTrackParam(Double_t x, Double_t alpha,
  39                         const Double_t param[5], const Double_t covar[15]);
  40   virtual ~AliExternalTrackParam(){}
  41
  42   void Set(Double_t x,Double_t alpha,
  43                         const Double_t param[5], const Double_t covar[15]);
  44
  45   static void SetMostProbablePt(Double_t pt) { fgMostProbablePt=pt; }
  46   static Double_t GetMostProbablePt() { return fgMostProbablePt; }
  47
  48   void Reset();
  49   void ResetCovariance(Double_t s2);
  50   void AddCovariance(const Double_t cov[15]);
  51
  52   const Double_t *GetParameter() const {return fP;}
  53   const Double_t *GetCovariance() const {return fC;}
  54
  55   Double_t GetAlpha() const {return fAlpha;}
  56   Double_t GetX() const {return fX;}
  57   Double_t GetY()    const {return fP[0];}
  58   Double_t GetZ()    const {return fP[1];}
  59   Double_t GetSnp()  const {return fP[2];}
  60   Double_t GetTgl()  const {return fP[3];}
  61   Double_t GetSigned1Pt()  const {return fP[4];}
  62
  63   Double_t GetSigmaY2() const {return fC[0];}
  64   Double_t GetSigmaZY() const {return fC[1];}
  65   Double_t GetSigmaZ2() const {return fC[2];}
  66   Double_t GetSigmaSnpY() const {return fC[3];}
  67   Double_t GetSigmaSnpZ() const {return fC[4];}
  68   Double_t GetSigmaSnp2() const {return fC[5];}
  69   Double_t GetSigmaTglY() const {return fC[6];}
  70   Double_t GetSigmaTglZ() const {return fC[7];}
  71   Double_t GetSigmaTglSnp() const {return fC[8];}
  72   Double_t GetSigmaTgl2() const {return fC[9];}
  73   Double_t GetSigma1PtY() const {return fC[10];}
  74   Double_t GetSigma1PtZ() const {return fC[11];}
  75   Double_t GetSigma1PtSnp() const {return fC[12];}
  76   Double_t GetSigma1PtTgl() const {return fC[13];}
  77   Double_t GetSigma1Pt2() const {return fC[14];}
  78
  79   // additional functions for AliVParticle
  80   Double_t Px() const;
  81   Double_t Py() const;
  82   Double_t Pz() const;
  83   Double_t Pt() const { return TMath::Abs(GetSignedPt()); }
  84   Double_t P() const { return GetP(); }
  85   Bool_t   PxPyPz(Double_t p[3]) const { return GetPxPyPz(p); }
  86
  87   Double_t Xv() const;
  88   Double_t Yv() const;
  89   Double_t Zv() const;
  90   Bool_t   XvYvZv(Double_t x[3]) const { return GetXYZ(x); }
  91
  92   Double_t OneOverPt() const { return 1./Pt(); }
  93   Double_t Phi() const;
  94   Double_t Theta() const;
  95   virtual Double_t E() const;
  96   virtual Double_t M() const;
  97   Double_t Eta() const;
  98   virtual Double_t Y() const;
  99   Short_t  Charge() const { return (Short_t)GetSign(); }
 100   const Double_t *PID() const { return 0x0; }
 101
 102   Double_t GetSign() const {return (fP[4]>0) ? 1 : -1;}
 103   Double_t GetP() const;
 104   Double_t GetSignedPt() const {
 105     return (TMath::Abs(fP[4])>kAlmost0) ? 1./fP[4]:TMath::Sign(kVeryBig,fP[4]);
 106   }
 107   Double_t Get1P() const;
 108   Double_t GetC(Double_t b) const {return fP[4]*b*kB2C;}
 109   void GetDZ(Double_t x,Double_t y,Double_t z,Double_t b,Float_t dz[2]) const;
 110   Double_t GetD(Double_t xv, Double_t yv, Double_t b) const;
 111   Double_t GetLinearD(Double_t xv, Double_t yv) const;
 112   Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho,
 113         Double_t mass,  Bool_t anglecorr=kFALSE,
 114         Double_t (*f)(Double_t)=ApproximateBetheBloch);
 115   Double_t GetPredictedChi2(Double_t p[2],Double_t cov[3]) const;
 116
 117   Double_t
 118     GetPredictedChi2(Double_t p[3],Double_t covyz[3],Double_t covxyz[3]) const;
 119   Bool_t
 120     PropagateTo(Double_t p[3],Double_t covyz[3],Double_t covxyz[3],Double_t b);
 121
 122   Double_t *GetResiduals(Double_t *p,Double_t *cov,Bool_t updated=kTRUE) const;
 123   Bool_t Update(Double_t p[2],Double_t cov[3]);
 124   Bool_t Rotate(Double_t alpha);
 125   Bool_t PropagateTo(Double_t x, Double_t b);
 126   Bool_t Propagate(Double_t alpha, Double_t x, Double_t b) {
 127     if (Rotate(alpha))
 128       if (PropagateTo(x,b)) return kTRUE;
 129     return kFALSE;
 130   }
 131   void   Propagate(Double_t len,Double_t x[3],Double_t p[3],Double_t bz) const;
 132   Bool_t Intersect(Double_t pnt[3], Double_t norm[3], Double_t bz) const;
 133
 134   void GetHelixParameters(Double_t h[6], Double_t b) const;
 135   Double_t GetDCA(const AliExternalTrackParam *p, Double_t b,
 136     Double_t &xthis,Double_t &xp) const;
 137   Double_t PropagateToDCA(AliExternalTrackParam *p, Double_t b);
 138   Bool_t PropagateToDCA(const AliESDVertex *vtx, Double_t b, Double_t maxd,
 139                         Double_t dz[2]=0, Double_t cov[3]=0);
 140
 141   void GetDirection(Double_t d[3]) const;
 142   Bool_t GetPxPyPz(Double_t *p) const;
 143   Bool_t GetXYZ(Double_t *p) const;
 144   Bool_t GetCovarianceXYZPxPyPz(Double_t cv[21]) const;
 145   Bool_t GetPxPyPzAt(Double_t x, Double_t b, Double_t p[3]) const;
 146   Bool_t GetXYZAt(Double_t x, Double_t b, Double_t r[3]) const;
 147   Bool_t GetYAt(Double_t x,  Double_t b,  Double_t &y) const;
 148   Bool_t GetZAt(Double_t x,  Double_t b,  Double_t &z) const;
 149   void Print(Option_t* option = "") const;
 150   Double_t GetSnpAt(Double_t x,Double_t b) const;
 151
 152   //Deprecated
 153   Bool_t CorrectForMaterial(Double_t d, Double_t x0, Double_t mass,
 154                             Double_t (*f)(Double_t)=ApproximateBetheBloch);
 155
 156   Bool_t GetDistance(AliExternalTrackParam *param2, Double_t x, Double_t dist[3], Double_t b);
 157   Int_t GetIndex(Int_t i, Int_t j) const {return i+(j+1)*j/2;}
 158   Int_t GetLabel() const {return -1;}
 159   //
 160   // visualization (M. Ivanov)
 161   //
 162   virtual void FillPolymarker(TPolyMarker3D *pol, Float_t magf, Float_t minR, Float_t maxR, Float_t stepR);
 163   virtual void DrawTrack(Float_t magF, Float_t minR, Float_t maxR, Float_t stepR);
 164  protected:
 165   Double_t &Par(Int_t i) {return fP[i];}
 166   Double_t &Cov(Int_t i) {return fC[i];}
 167
 168  private:
 169   Double32_t           fX;     // X coordinate for the point of parametrisation
 170   Double32_t           fAlpha; // Local <-->global coor.system rotation angle
 171   Double32_t           fP[5];  // The track parameters
 172   Double32_t           fC[15]; // The track parameter covariance matrix
 173
 174   static Double32_t    fgMostProbablePt; // "Most probable" pt
 175                                          // (to be used if Bz=0)
 176   ClassDef(AliExternalTrackParam, 7)
 177 };
 178
 179 inline void AliExternalTrackParam::ResetCovariance(Double_t s2) {
 180   //
 181   // Reset the covarince matrix to "something big"
 182   //
 183     fC[0]*= s2;
 184     fC[1] = 0.;  fC[2]*= s2;
 185     fC[3] = 0.;  fC[4] = 0.;  fC[5]*= s2;
 186     fC[6] = 0.;  fC[7] = 0.;  fC[8] = 0.;  fC[9]*= s2;
 187     fC[10]= 0.;  fC[11]= 0.;  fC[12]= 0.;  fC[13]= 0.;  fC[14]*=s2;
 188 }
 189
 190
 191
 192
 193 #endif