fRefVolumeId for reference volume identification added.

[u/mrichter/AliRoot.git] / ITS / AliV0vertex.h

diff --git a/ITS/AliV0vertex.h b/ITS/AliV0vertex.h
index 3de575a13314aa47dac3ed0077a5941a207bbda2..b18fca0c62ce3b968dd49cdec40efeeeba237c50 100644 (file)
--- a/ITS/AliV0vertex.h
+++ b/ITS/AliV0vertex.h
@@ -11,7 +11,7 @@
 //-------------------------------------------------------------------------
 
 #include <TObject.h>
-#include "AliPDG.h"
+#include <TPDGCode.h>
 
 class AliITStrackV2;
 
@@ -20,12 +20,14 @@ public:
   AliV0vertex();
   AliV0vertex(const AliITStrackV2 &neg, const AliITStrackV2 &pos);
 
-  void ChangeMassHypothesis(Int_t code=kLambda0); 
+  Double_t ChangeMassHypothesis(Int_t code=kK0Short); 
 
   Int_t GetPdgCode() const {return fPdgCode;}
   Double_t GetEffMass() const {return fEffMass;}
   Double_t GetChi2() const {return fChi2;}
   void GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const;
+  void GetNPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const;
+  void GetPPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const;
   void GetXYZ(Double_t &x, Double_t &y, Double_t &z) const;
   Double_t GetD(Double_t x0=0.,Double_t y0=0.,Double_t z0=0.) const;
   Int_t GetNlabel() const {return fNlab;}
@@ -49,6 +51,16 @@ private:
   ClassDef(AliV0vertex,1)   // reconstructed V0 vertex
 };
 
+inline 
+void AliV0vertex::GetNPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const {
+px=fNmom[0]; py=fNmom[1]; pz=fNmom[2];
+}
+
+inline 
+void AliV0vertex::GetPPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const {
+px=fPmom[0]; py=fPmom[1]; pz=fPmom[2];
+}
+
 #endif