Common primary vertex access for ESD and AOD (B.HIPPOLYTE)

diff --git a/PWG2/SPECTRA/AliAnalysisTaskCheckV0.cxx b/PWG2/SPECTRA/AliAnalysisTaskCheckV0.cxx
index 650c67993e9cafb22062154cb52f6968601690f1..53f6e97866dc7adf565c5d8d49e45fcd11b67f73 100644 (file)
--- a/PWG2/SPECTRA/AliAnalysisTaskCheckV0.cxx
+++ b/PWG2/SPECTRA/AliAnalysisTaskCheckV0.cxx
@@ -262,16 +262,15 @@ void AliAnalysisTaskCheckV0::UserExec(Option_t *)
   Double_t lInvMassK0s = 0, lInvMassLambda = 0, lInvMassAntiLambda = 0;
   Double_t lAlphaV0 = 0, lPtArmV0 = 0;
 
-  if(fAnalysisType == "ESD") {
-
-    const AliESDVertex *primaryVtx = ((AliESDEvent*)lEvent)->GetPrimaryVertex();
-    tPrimaryVtxPosition[0] = primaryVtx->GetXv();
-    tPrimaryVtxPosition[1] = primaryVtx->GetYv();
-    tPrimaryVtxPosition[2] = primaryVtx->GetZv();
+  const AliVVertex *primaryVtx = ((AliESDEvent*)lEvent)->GetPrimaryVertex();
+  tPrimaryVtxPosition[0] = primaryVtx->GetX();
+  tPrimaryVtxPosition[1] = primaryVtx->GetY();
+  tPrimaryVtxPosition[2] = primaryVtx->GetZ();
+  fHistPrimaryVertexPosX->Fill(tPrimaryVtxPosition[0]);
+  fHistPrimaryVertexPosY->Fill(tPrimaryVtxPosition[1]);
+  fHistPrimaryVertexPosZ->Fill(tPrimaryVtxPosition[2]);
 
-    fHistPrimaryVertexPosX->Fill(tPrimaryVtxPosition[0]);
-    fHistPrimaryVertexPosY->Fill(tPrimaryVtxPosition[1]);
-    fHistPrimaryVertexPosZ->Fill(tPrimaryVtxPosition[2]);
+  if(fAnalysisType == "ESD") {
 
     Double_t lMagneticField = ((AliESDEvent*)lEvent)->GetMagneticField();
 
@@ -366,15 +365,6 @@ void AliAnalysisTaskCheckV0::UserExec(Option_t *)
 
   else if(fAnalysisType == "AOD") {
 
-    const AliAODVertex *primaryVtx = ((AliAODEvent*)lEvent)->GetPrimaryVertex();
-    tPrimaryVtxPosition[0] = primaryVtx->GetX();
-    tPrimaryVtxPosition[1] = primaryVtx->GetY();
-    tPrimaryVtxPosition[2] = primaryVtx->GetZ();
-
-    fHistPrimaryVertexPosX->Fill(tPrimaryVtxPosition[0]);
-    fHistPrimaryVertexPosY->Fill(tPrimaryVtxPosition[1]);
-    fHistPrimaryVertexPosZ->Fill(tPrimaryVtxPosition[2]);
-
     for (Int_t iV0 = 0; iV0 < nv0s; iV0++) 
       {// This is the begining of the V0 loop
        AliAODv0 *v0 = ((AliAODEvent*)lEvent)->GetV0(iV0);