correcting baryon mass subtraction for visible energy in MC

diff --git a/PWG4/totEt/AliAnalysisEtMonteCarlo.cxx b/PWG4/totEt/AliAnalysisEtMonteCarlo.cxx
index 4e148fc295832efa101bb6e352510989b889d336..69bd6f6b2b28c3093a80c4189f9503abc26595ea 100644 (file)
--- a/PWG4/totEt/AliAnalysisEtMonteCarlo.cxx
+++ b/PWG4/totEt/AliAnalysisEtMonteCarlo.cxx
@@ -14,6 +14,7 @@
 #include "AliMCEvent.h"
 #include "TH2F.h"
 #include "TParticle.h"
+#include "TDatabasePDG.h"
 #include "AliGenHijingEventHeader.h"
 #include "AliGenPythiaEventHeader.h"
 
@@ -43,6 +44,8 @@ Int_t AliAnalysisEtMonteCarlo::AnalyseEvent(AliVEvent* ev)
     // Get us an mc event
     AliMCEvent *event = dynamic_cast<AliMCEvent*>(ev);
 
+    Double_t protonMass = fPdgDB->GetParticle("proton")->Mass(); // should maybe use average of proton and neutron, but they are pretty close
+
     // Hijing header
     AliGenEventHeader* genHeader = event->GenEventHeader();
     AliGenHijingEventHeader* hijingGenHeader = dynamic_cast<AliGenHijingEventHeader*>(genHeader);
@@ -113,8 +116,8 @@ Int_t AliAnalysisEtMonteCarlo::AnalyseEvent(AliVEvent* ev)
                 TMath::Abs(pdg->PdgCode()) == fOmegaCode
               )
             {
-             if (pdg->PdgCode() > 0) { particleMassPart = - pdg->Mass();}
-             if (pdg->PdgCode() < 0) { particleMassPart = pdg->Mass();}
+             if (pdg->PdgCode() > 0) { particleMassPart = - protonMass;}
+             if (pdg->PdgCode() < 0) { particleMassPart = protonMass;}
            }
            Double_t et = part->Energy() * TMath::Sin(part->Theta()) + particleMassPart;