STEER/AliKFParticle.cxx

   1 //----------------------------------------------------------------------------
   2 // Implementation of the AliKFParticle class
   3 // .
   4 // @author  S.Gorbunov, I.Kisel
   5 // @version 1.0
   6 // @since   13.05.07
   7 //
   8 // Class to reconstruct and store the decayed particle parameters.
   9 // The method is described in CBM-SOFT note 2007-003,
  10 // ``Reconstruction of decayed particles based on the Kalman filter'',
  11 // http://www.gsi.de/documents/DOC-2007-May-14-1.pdf
  12 //
  13 // This class is ALICE interface to general mathematics in AliKFParticleCore
  14 //
  15 //  -= Copyright &copy ALICE HLT Group =-
  16 //____________________________________________________________________________
  17
  18
  19 #include "AliKFParticle.h"
  20 #include "TDatabasePDG.h"
  21 #include "TParticlePDG.h"
  22 #include "AliExternalTrackParam.h"
  23 //#include "TMath.h"
  24
  25 ClassImp(AliKFParticle)
  26
  27 Double_t AliKFParticle::fgBz = -5.;  //* Bz compoment of the magnetic field
  28
  29 AliKFParticle::AliKFParticle( const AliExternalTrackParam &track, Int_t PID )
  30 {
  31   // Constructor from ALICE track, PID hypothesis should be provided
  32
  33   TParticlePDG* particlePDG = TDatabasePDG::Instance()->GetParticle(PID);
  34   Double_t mass = (particlePDG) ? particlePDG->Mass() :0.13957;
  35
  36   track.GetXYZ(fP);
  37   track.GetPxPyPz(fP+3);
  38   Double_t energy = TMath::Sqrt( mass*mass + fP[3]*fP[3] + fP[4]*fP[4] + fP[5]*fP[5]);
  39   fP[6] = energy;
  40   fP[7] = 0;
  41   fQ = (track.GetSigned1Pt() >0 ) ?1 :-1; // fQ = track->GetSign() would do the same thing
  42   fNDF = 0;
  43   fChi2 = 0;
  44   fAtProductionVertex = 0;
  45   fIsLinearized = 0;
  46   fSFromDecay = 0;
  47
  48   Double_t energyInv = 1./energy;
  49   Double_t
  50     h0 = fP[3]*energyInv,
  51     h1 = fP[4]*energyInv,
  52     h2 = fP[5]*energyInv;
  53
  54   track.GetCovarianceXYZPxPyPz( fC );
  55
  56   fC[21] = h0*fC[ 6] + h1*fC[10] + h2*fC[15];
  57   fC[22] = h0*fC[ 7] + h1*fC[11] + h2*fC[16];
  58   fC[23] = h0*fC[ 8] + h1*fC[12] + h2*fC[17];
  59   fC[24] = h0*fC[ 9] + h1*fC[13] + h2*fC[18];
  60   fC[25] = h0*fC[13] + h1*fC[14] + h2*fC[19];
  61   fC[26] = h0*fC[18] + h1*fC[19] + h2*fC[20];
  62   fC[27] = h0*h0*fC[ 9] + h1*h1*fC[14] + h2*h2*fC[20]
  63     + 2*(h0*h1*fC[13] + h0*h2*fC[18] + h1*h2*fC[19] );
  64   for( int i=28; i<36; i++ ) fC[i] = 0;
  65   fC[35] = 1.;
  66 }
  67
  68 AliKFParticle::AliKFParticle( const AliESDVertex &vertex )
  69 {
  70   // Constructor from ALICE vertex
  71
  72   vertex.GetXYZ( fP );
  73   vertex.GetCovMatrix( fC );
  74   fChi2 = vertex.GetChi2();
  75   fNDF = 2*vertex.GetNContributors() - 3;
  76   fQ = 0;
  77   fAtProductionVertex = 0;
  78   fIsLinearized = 0;
  79   fSFromDecay = 0;
  80 }
  81
  82
  83 void AliKFParticle::GetExternalTrackParam( const AliKFParticleBase &p, Double_t &X, Double_t &Alpha, Double_t P[5] )
  84 {
  85   Double_t cosA = p.GetPx(), sinA = p.GetPy();
  86   Double_t pt = TMath::Sqrt(cosA*cosA + sinA*sinA);
  87   Double_t pti = 0;
  88   if( pt<1.e-4 ){
  89     cosA = 1;
  90     sinA = 0;
  91   } else {
  92     pti = 1./pt;
  93     cosA*=pti;
  94     sinA*=pti;
  95   }
  96   Alpha = TMath::ATan2(sinA,cosA);
  97   X   = p.GetX()*cosA + p.GetY()*sinA;
  98   P[0]= p.GetY()*cosA - p.GetX()*sinA;
  99   P[1]= p.GetZ();
 100   P[2]= 0;
 101   P[3]= p.GetPz()*pti;
 102   P[4]= p.GetQ()*pti;
 103 }
 104
 105
 106 void AliKFParticle::GetDStoParticleALICE( const AliKFParticleBase &p,
 107                                           Double_t &DS, Double_t &DS1 )
 108   const
 109 {
 110   DS = DS1 = 0;
 111   Double_t x1, a1, x2, a2;
 112   Double_t par1[5], par2[5], cov[15];
 113   for(int i=0; i<15; i++) cov[i] = 0;
 114   cov[0] = cov[2] = cov[5] = cov[9] = cov[14] = .001;
 115
 116   GetExternalTrackParam( *this, x1, a1, par1 );
 117   GetExternalTrackParam( p, x2, a2, par2 );
 118
 119   AliExternalTrackParam t1(x1,a1, par1, cov);
 120   AliExternalTrackParam t2(x2,a2, par2, cov);
 121
 122   Double_t xe1=0, xe2=0;
 123   t1.GetDCA( &t2, -GetFieldAlice(), xe1, xe2 );
 124   t1.PropagateTo( xe1, -GetFieldAlice() );
 125   t2.PropagateTo( xe2, -GetFieldAlice() );
 126
 127   Double_t xyz1[3], xyz2[3];
 128   t1.GetXYZ( xyz1 );
 129   t2.GetXYZ( xyz2 );
 130
 131   DS = GetDStoPoint( xyz1 );
 132   DS1 = p.GetDStoPoint( xyz2 );
 133
 134   return;
 135 }