]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliKFParticle.h
Change according to Federico's suggestions (Massimo)
[u/mrichter/AliRoot.git] / STEER / AliKFParticle.h
index fa250d1cdcab3124028a7fe3eec6215089e3eb2d..716b8a51d736d3c4d05c9cc6d6293e693910e078 100644 (file)
@@ -23,7 +23,6 @@
 
 class AliExternalTrackParam;
 
-
 class AliKFParticle :public AliKFParticleBase
 {
   
@@ -33,26 +32,42 @@ class AliKFParticle :public AliKFParticleBase
   //*  INITIALIZATION
   //*
 
+  //* Set magnetic field for all particles
+  
+  static void SetField( Double_t Bz );
+
   //* Constructor (empty)
 
-  AliKFParticle():AliKFParticleBase(), fBz(0) {} 
+  AliKFParticle():AliKFParticleBase(){ ; }
 
   //* Destructor (empty)
 
-  ~AliKFParticle(){}             
+  ~AliKFParticle(){ ; }
 
-  //* Initialisation from ALICE track, PID hypothesis can be provided 
+  //* Construction of mother particle by its 2-3-4 daughters
 
-  AliKFParticle( const AliExternalTrackParam &track, Double_t bz, Int_t PID = 211 );
+  AliKFParticle( const AliKFParticle &d1, const AliKFParticle &d2 );
 
-  //* Initialisation from ESD vertex 
+  AliKFParticle( const AliKFParticle &d1, const AliKFParticle &d2, 
+                const AliKFParticle &d3 );
 
-  AliKFParticle( const AliESDVertex &vertex, Double_t bz );
+  AliKFParticle( const AliKFParticle &d1, const AliKFParticle &d2, 
+                const AliKFParticle &d3, const AliKFParticle &d4 );
+ //* Initialisation from "cartesian" coordinates ( X Y Z Px Py Pz )
+ //* Parameters, covariance matrix, charge and PID hypothesis should be provided 
+
+  void Create( const Double_t Param[], const Double_t Cov[], Int_t Charge, Int_t PID );
 
-  void SetBz(Double_t bz) {fBz=bz;}
-  Double_t GetBz() const  {return fBz;}
+ //* Initialisation from ALICE track, PID hypothesis shoould be provided 
 
-  //* Copy vertex part to ESD vertex 
+  AliKFParticle( const AliExternalTrackParam &track, Int_t PID );
+
+  //* Initialisation from ESD vertex 
+
+  AliKFParticle( const AliESDVertex &vertex );
+
+  //* Copy position part to ESD vertex 
 
   void CopyToESDVertex( AliESDVertex &Vtx ) const ;
 
@@ -86,6 +101,28 @@ class AliKFParticle :public AliKFParticleBase
   Double_t GetCovariance( int i ) const ;
   Double_t GetCovariance( int i, int j ) const ;
 
+  //* Accessors with calculations, value returned w/o error flag
+  
+  Double_t GetMomentum    () const;
+  Double_t GetMass        () const;
+  Double_t GetDecayLength () const;
+  Double_t GetLifeTime    () const;
+
+  //* Accessors to estimated errors
+
+  Double_t GetErrX           () const ; //* x of current position
+  Double_t GetErrY           () const ; //* y of current position
+  Double_t GetErrZ           () const ; //* z of current position
+  Double_t GetErrPx          () const ; //* x-compoment of 3-momentum
+  Double_t GetErrPy          () const ; //* y-compoment of 3-momentum
+  Double_t GetErrPz          () const ; //* z-compoment of 3-momentum
+  Double_t GetErrE           () const ; //* energy
+  Double_t GetErrS           () const ; //* decay length / momentum
+  Double_t GetErrMomentum    () const;
+  Double_t GetErrMass        () const;
+  Double_t GetErrDecayLength () const;
+  Double_t GetErrLifeTime    () const;
+
   //* Accessors with calculations( &value, &estimated sigma )
   //* error flag returned (0 means no error during calculations) 
 
@@ -94,6 +131,7 @@ class AliKFParticle :public AliKFParticleBase
   int GetDecayLength ( Double_t &L, Double_t &SigmaL ) const ;
   int GetLifeTime    ( Double_t &T, Double_t &SigmaT ) const ;
 
+
   //*
   //*  MODIFIERS
   //*
@@ -120,24 +158,26 @@ class AliKFParticle :public AliKFParticleBase
   //*
 
 
-  //* Simple way to construct particles ex. D0 = Pion + Kaon; 
+  //* Add daughter to the particle 
 
-  AliKFParticle operator +( const AliKFParticle &Daughter ) const;
-
-  void operator +=( const AliKFParticle &Daughter );  
+  void AddDaughter( const AliKFParticle &Daughter );
 
-  //* Add daughter track to the particle 
+  //* Add daughter via += operator: ex.{ D0; D0+=Pion; D0+= Kaon; }
 
-  void AddDaughter( const AliKFParticle &Daughter );
+  void operator +=( const AliKFParticle &Daughter );  
 
   //* Set production vertex 
 
   void SetProductionVertex( const AliKFParticle &Vtx );
 
-  //* Set hard mass constraint 
+  //* Set mass constraint 
 
-  void SetMassConstraint( Double_t Mass );
+  void SetMassConstraint( Double_t Mass, Double_t SigmaMass = 0  );
   
+  //* Set no decay length for resonances
+
+  void SetNoDecayLength();
+
   //* Everything in one go  
 
   void Construct( const AliKFParticle *vDaughters[], int NDaughters, 
@@ -183,6 +223,11 @@ class AliKFParticle :public AliKFParticleBase
   void GetDStoParticle( const AliKFParticle &p, 
                        Double_t &DS, Double_t &DSp ) const ;
   
+  //* Get dS to other particle p in XY-plane
+
+  void GetDStoParticleXY( const AliKFParticleBase &p, 
+                         Double_t &DS, Double_t &DSp ) const ;
+  
   //* 
   //* OTHER UTILITIES
   //*
@@ -202,13 +247,33 @@ class AliKFParticle :public AliKFParticleBase
   Double_t GetDeviationFromVertex( const AliKFParticle &Vtx ) const ;
   Double_t GetDeviationFromVertex( const AliESDVertex &Vtx ) const ;
   Double_t GetDeviationFromParticle( const AliKFParticle &p ) const ;
+  //* Calculate distance from another object [cm] in XY-plane
+
+  Double_t GetDistanceFromVertexXY( const Double_t vtx[] ) const ;
+  Double_t GetDistanceFromVertexXY( const AliKFParticle &Vtx ) const ;
+  Double_t GetDistanceFromVertexXY( const AliESDVertex &Vtx ) const ;
+  Double_t GetDistanceFromParticleXY( const AliKFParticle &p ) const ;
+
+  //* Calculate sqrt(Chi2/ndf) deviation from another object in XY plane
+  //* ( v = [xyz]-vertex, Cv=[Cxx,Cxy,Cyy,Cxz,Cyz,Czz]-covariance matrix )
+
+  Double_t GetDeviationFromVertexXY( const Double_t v[], const Double_t Cv[]=0 ) const ;
+  Double_t GetDeviationFromVertexXY( const AliKFParticle &Vtx ) const ;
+  Double_t GetDeviationFromVertexXY( const AliESDVertex &Vtx ) const ;
+  Double_t GetDeviationFromParticleXY( const AliKFParticle &p ) const ;
+
+  //* Calculate opennig angle between two particles
+
+  Double_t GetAngle  ( const AliKFParticle &p ) const ;
+  Double_t GetAngleXY( const AliKFParticle &p ) const ;
+  Double_t GetAngleRZ( const AliKFParticle &p ) const ;
 
   //* Subtract the particle from the vertex  
 
   void SubtractFromVertex( AliKFParticle &v ) const ;
   void SubtractFromVertex( AliESDVertex &v ) const ;
 
-
  protected: 
   
   //*
@@ -217,18 +282,20 @@ class AliKFParticle :public AliKFParticleBase
 
   //* Method to access ALICE field 
  
-  Double_t GetFieldAlice() const ;
+  static Double_t GetFieldAlice();
   
   //* Other methods required by the abstract AliKFParticleBase class 
   
   void GetFieldValue( const Double_t xyz[], Double_t B[] ) const ;
   void GetDStoParticle( const AliKFParticleBase &p, Double_t &DS, Double_t &DSp )const ;
   void Transport( Double_t dS, Double_t P[], Double_t C[] ) const ;
+  static void GetExternalTrackParam( const AliKFParticleBase &p, Double_t &X, Double_t &Alpha, Double_t P[5]  ) ;
 
-private:
-  Double_t fBz;  // Bz compoment of the magnetic field
+ private:
 
-  ClassDef( AliKFParticle, 2 );
+  static Double_t fgBz;  //* Bz compoment of the magnetic field
+
+  ClassDef( AliKFParticle, 1 );
 
 };
 
@@ -241,6 +308,45 @@ private:
 //---------------------------------------------------------------------
 
 
+inline void AliKFParticle::SetField( Double_t Bz )
+{ 
+  fgBz = -Bz;//!!!
+}
+
+
+inline AliKFParticle::AliKFParticle( const AliKFParticle &d1, 
+                                    const AliKFParticle &d2 )
+{
+  AliKFParticle mother;
+  mother+= d1;
+  mother+= d2;
+  *this = mother;
+}
+
+inline AliKFParticle::AliKFParticle( const AliKFParticle &d1, 
+                                    const AliKFParticle &d2, 
+                                    const AliKFParticle &d3 )
+{
+  AliKFParticle mother;
+  mother+= d1;
+  mother+= d2;
+  mother+= d3;
+  *this = mother;
+}
+
+inline AliKFParticle::AliKFParticle( const AliKFParticle &d1, 
+                                    const AliKFParticle &d2, 
+                                    const AliKFParticle &d3, 
+                                    const AliKFParticle &d4 )
+{
+  AliKFParticle mother;
+  mother+= d1;
+  mother+= d2;
+  mother+= d3;
+  mother+= d4;
+  *this = mother;
+}
+
 
 inline void AliKFParticle::Initialize()
 { 
@@ -322,6 +428,104 @@ inline Double_t AliKFParticle::GetCovariance( int i, int j ) const
   return AliKFParticleBase::GetCovariance(i,j);
 }
 
+
+inline Double_t AliKFParticle::GetMomentum    () const
+{
+  Double_t par, err;
+  if( AliKFParticleBase::GetMomentum( par, err ) ) return 0;
+  else return par;
+}
+
+inline Double_t AliKFParticle::GetMass        () const
+{
+  Double_t par, err;
+  if( AliKFParticleBase::GetMass( par, err ) ) return 0;
+  else return par;
+}
+
+inline Double_t AliKFParticle::GetDecayLength () const
+{
+  Double_t par, err;
+  if( AliKFParticleBase::GetDecayLength( par, err ) ) return 0;
+  else return par;
+}
+
+inline Double_t AliKFParticle::GetLifeTime    () const
+{
+  Double_t par, err;
+  if( AliKFParticleBase::GetLifeTime( par, err ) ) return 0;
+  else return par;
+}
+
+inline Double_t AliKFParticle::GetErrX           () const 
+{
+  return TMath::Sqrt(TMath::Abs( GetCovariance(0,0) ));
+}
+
+inline Double_t AliKFParticle::GetErrY           () const 
+{
+  return TMath::Sqrt(TMath::Abs( GetCovariance(1,1) ));
+}
+
+inline Double_t AliKFParticle::GetErrZ           () const 
+{
+  return TMath::Sqrt(TMath::Abs( GetCovariance(2,2) ));
+}
+
+inline Double_t AliKFParticle::GetErrPx          () const 
+{
+  return TMath::Sqrt(TMath::Abs( GetCovariance(3,3) ));
+}
+
+inline Double_t AliKFParticle::GetErrPy          () const 
+{
+  return TMath::Sqrt(TMath::Abs( GetCovariance(4,4) ));
+}
+
+inline Double_t AliKFParticle::GetErrPz          () const 
+{
+  return TMath::Sqrt(TMath::Abs( GetCovariance(5,5) ));
+}
+
+inline Double_t AliKFParticle::GetErrE           () const 
+{
+  return TMath::Sqrt(TMath::Abs( GetCovariance(6,6) ));
+}
+
+inline Double_t AliKFParticle::GetErrS           () const 
+{
+  return TMath::Sqrt(TMath::Abs( GetCovariance(7,7) ));
+}
+
+inline Double_t AliKFParticle::GetErrMomentum    () const
+{
+  Double_t par, err;
+  if( AliKFParticleBase::GetMomentum( par, err ) ) return 1.e10;
+  else return err;
+}
+
+inline Double_t AliKFParticle::GetErrMass        () const
+{
+  Double_t par, err;
+  if( AliKFParticleBase::GetMass( par, err ) ) return 1.e10;
+  else return err;
+}
+
+inline Double_t AliKFParticle::GetErrDecayLength () const
+{
+  Double_t par, err;
+  if( AliKFParticleBase::GetDecayLength( par, err ) ) return 1.e10;
+  else return err;
+}
+
+inline Double_t AliKFParticle::GetErrLifeTime    () const
+{
+  Double_t par, err;
+  if( AliKFParticleBase::GetLifeTime( par, err ) ) return 1.e10;
+  else return err;
+}
+
+
 inline int AliKFParticle::GetMomentum( Double_t &P, Double_t &SigmaP ) const 
 {
   return AliKFParticleBase::GetMomentum( P, SigmaP );
@@ -418,13 +622,6 @@ inline void AliKFParticle::operator +=( const AliKFParticle &Daughter )
   AliKFParticleBase::operator +=( Daughter );
 }
   
-inline AliKFParticle AliKFParticle::operator +( const AliKFParticle &Daughter ) const 
-{
-  AliKFParticle tmp;
-  const AliKFParticle *v[2] = {this, &Daughter };
-  tmp.Construct( v,2);
-  return tmp;
-}
 
 inline void AliKFParticle::AddDaughter( const AliKFParticle &Daughter )
 {
@@ -436,11 +633,16 @@ inline void AliKFParticle::SetProductionVertex( const AliKFParticle &Vtx )
   AliKFParticleBase::SetProductionVertex( Vtx );
 }
 
-inline void AliKFParticle::SetMassConstraint( Double_t Mass )
+inline void AliKFParticle::SetMassConstraint( Double_t Mass, Double_t SigmaMass )
 {
-  AliKFParticleBase::SetMassConstraint( Mass );
+  AliKFParticleBase::SetMassConstraint( Mass, SigmaMass );
 }
-  
+    
+inline void AliKFParticle::SetNoDecayLength()
+{
+  AliKFParticleBase::SetNoDecayLength();
+}
+
 inline void AliKFParticle::Construct( const AliKFParticle *vDaughters[], int NDaughters, 
                               const AliKFParticle *ProdVtx,   Double_t Mass  )
 {    
@@ -465,7 +667,7 @@ inline void AliKFParticle::TransportToPoint( const Double_t xyz[] )
 
 inline void AliKFParticle::TransportToVertex( const AliESDVertex &v )
 {       
-  TransportToPoint( AliKFParticle(v,fBz).fP );
+  TransportToPoint( AliKFParticle(v).fP );
 }
 
 inline void AliKFParticle::TransportToParticle( const AliKFParticle &p )
@@ -484,11 +686,12 @@ inline Double_t AliKFParticle::GetDStoPoint( const Double_t xyz[] ) const
 {
   return AliKFParticleBase::GetDStoPointBz( GetFieldAlice(), xyz );
 }
+
   
 inline void AliKFParticle::GetDStoParticle( const AliKFParticle &p, 
-                                    Double_t &DS, Double_t &DSp ) const 
+                                           Double_t &DS, Double_t &DSp ) const 
 {
-  return AliKFParticleBase::GetDStoParticleBz( GetFieldAlice(), p, DS,DSp);
+  GetDStoParticleXY( p, DS, DSp );
 }
 
 
@@ -515,22 +718,22 @@ inline Double_t AliKFParticle::GetDeviationFromVertex( const AliKFParticle &Vtx
 
 inline Double_t AliKFParticle::GetDistanceFromVertex( const AliESDVertex &Vtx ) const
 {
-  return GetDistanceFromVertex( AliKFParticle(Vtx,fBz) );
+  return GetDistanceFromVertex( AliKFParticle(Vtx) );
 }
 
 inline Double_t AliKFParticle::GetDeviationFromVertex( const AliESDVertex &Vtx ) const
 {
-  return GetDeviationFromVertex( AliKFParticle(Vtx,fBz) );
+  return GetDeviationFromVertex( AliKFParticle(Vtx) );
 }
   
 inline Double_t AliKFParticle::GetDistanceFromParticle( const AliKFParticle &p ) const 
 {
-  return AliKFParticleBase::GetDistanceFromParticleBz( GetFieldAlice(), p);
+  return AliKFParticleBase::GetDistanceFromParticle( p );
 }
 
 inline Double_t AliKFParticle::GetDeviationFromParticle( const AliKFParticle &p ) const 
 {
-  return AliKFParticleBase::GetDeviationFromParticleBz( GetFieldAlice(),p);
+  return AliKFParticleBase::GetDeviationFromParticle( p );
 }
 
 inline void AliKFParticle::SubtractFromVertex( AliKFParticle &v ) const 
@@ -540,7 +743,7 @@ inline void AliKFParticle::SubtractFromVertex( AliKFParticle &v ) const
 
 inline void AliKFParticle::SubtractFromVertex( AliESDVertex &v ) const 
 {
-  AliKFParticle vTmp(v,fBz);
+  AliKFParticle vTmp(v);
   SubtractFromVertex( vTmp );
   v = AliESDVertex( vTmp.fP, vTmp.fC, vTmp.fChi2, (vTmp.fNDF +3)/2, v.GetName() );
 }
@@ -551,9 +754,9 @@ inline void AliKFParticle::CopyToESDVertex( AliESDVertex &v ) const
   v = AliESDVertex( vTmp.fP, vTmp.fC, vTmp.fChi2, (vTmp.fNDF +3)/2 );
 }
 
-inline Double_t AliKFParticle::GetFieldAlice() const 
+inline Double_t AliKFParticle::GetFieldAlice()
 { 
-  return fBz; 
+  return fgBz; 
 }
 
 inline void AliKFParticle::GetFieldValue( const Double_t * /*xyz*/, Double_t B[] ) const 
@@ -563,9 +766,15 @@ inline void AliKFParticle::GetFieldValue( const Double_t * /*xyz*/, Double_t B[]
 }
 
 inline void AliKFParticle::GetDStoParticle( const AliKFParticleBase &p, 
-                                    Double_t &DS, Double_t &DSp )const
+                                           Double_t &DS, Double_t &DSp )const
 {
-  return AliKFParticleBase::GetDStoParticleBz( GetFieldAlice(), p, DS,DSp);
+  GetDStoParticleXY( p, DS, DSp );
+}
+
+inline void AliKFParticle::GetDStoParticleXY( const AliKFParticleBase &p, 
+                                      Double_t &DS, Double_t &DSp ) const
+{ 
+  AliKFParticleBase::GetDStoParticleBz( GetFieldAlice(), p, DS, DSp ) ;
 }
 
 inline void AliKFParticle::Transport( Double_t dS, Double_t P[], Double_t C[] ) const