Added AliGenHepMCEventHeader class and put it in STEERBase library. Updated AliGenRea...

[u/mrichter/AliRoot.git] / EVGEN / AliGenMC.cxx

diff --git a/EVGEN/AliGenMC.cxx b/EVGEN/AliGenMC.cxx
index 1ce5e58f925ed87c80d574f1bd4fd4169ff69fc4..115736974493104cdb40900fb1efe1bb4204cdf8 100644 (file)
--- a/EVGEN/AliGenMC.cxx
+++ b/EVGEN/AliGenMC.cxx
@@ -29,8 +29,10 @@
 #include <TVector3.h>
 
 #include "AliGenMC.h"
+#include "AliGenEventHeader.h"
 #include "AliRun.h"
 #include "AliGeometry.h"
+#include "AliDecayer.h"
 
 ClassImp(AliGenMC)
 
@@ -129,6 +131,7 @@ void AliGenMC::Init()
     case kBSemiMuonic:
     case kSemiMuonic:
     case kDiMuon:
+    case kJpsiDiMuon:
     case kBJpsiDiMuon:
     case kBPsiPrimeDiMuon:
     case kPiToMu:
@@ -150,6 +153,7 @@ void AliGenMC::Init()
        fChildSelect[0]=kKPlus;
        break;
     case kBJpsi:
+    case kBJpsiUndecayed:
        fChildSelect[0]= 443;
        break;
    case kChiToJpsiGammaToMuonMuon:
@@ -178,6 +182,7 @@ void AliGenMC::Init()
     case kNoDecayHeavy:
     case kNoDecayBeauty:
     case kNeutralPion:
+    case kBeautyUpgrade:
        break;
     }
 
@@ -209,7 +214,7 @@ Bool_t AliGenMC::ChildSelected(Int_t ip) const
     return kFALSE;
 }
 
-Bool_t AliGenMC::KinematicSelection(TParticle *particle, Int_t flag) const
+Bool_t AliGenMC::KinematicSelection(const TParticle *particle, Int_t flag) const
 {
 // Perform kinematic selection
     Double_t pz    = particle->Pz();
@@ -447,6 +452,7 @@ void AliGenMC::AddHeader(AliGenEventHeader* header)
 {
     // Passes header either to the container or to gAlice
     if (fContainer) {
+        header->SetName(fName);
        fContainer->AddHeader(header);
     } else {
        gAlice->SetGenEventHeader(header);