Decay options or W added. (Z. Conesa)

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

diff --git a/EVGEN/AliGenMC.cxx b/EVGEN/AliGenMC.cxx
index aeb67fe7a78cdb6c6b30fd02505f883a492eca31..bdbb1d224d5d444cf2b0d176af84993c0beb4f19 100644 (file)
--- a/EVGEN/AliGenMC.cxx
+++ b/EVGEN/AliGenMC.cxx
@@ -21,11 +21,13 @@
 // decay products and particle selection.
 // andreas.morsch@cern.ch
 
+#include <TClonesArray.h>
 #include <TMath.h>
 #include <TPDGCode.h>
 #include <TParticle.h>
 
 #include "AliGenMC.h"
+#include "AliGeometry.h"
 
 ClassImp(AliGenMC)
 
@@ -47,6 +49,7 @@ AliGenMC::AliGenMC()
     SetProjectile();
     fParentSelect.Set(8);
     fChildSelect.Set(8);
+    fForceDecay = kAll;
 }
 
 AliGenMC::AliGenMC(Int_t npart)
@@ -69,6 +72,7 @@ AliGenMC::AliGenMC(Int_t npart)
     SetNumberOfAcceptedParticles();
     SetTarget();
     SetProjectile();
+    fForceDecay = kAll;
 }
 
 AliGenMC::AliGenMC(const AliGenMC & mc):
@@ -101,14 +105,22 @@ void AliGenMC::Init()
     case kBPsiPrimeDiMuon:
     case kPiToMu:
     case kKaToMu:
+    case kWToMuon:
+    case kWToCharmToMuon:
        fChildSelect[0]=kMuonMinus;
        break;
+    case kWToCharm:
+       break;
     case kHadronicD:
        fChildSelect[0]=kPiPlus;
        fChildSelect[1]=kKPlus;
        break;
     case kPhiKK:
        fChildSelect[0]=kKPlus;
+       break;
+    case kBJpsi:
+       fChildSelect[0]=443;
+       break;
     case kOmega:       
     case kAll:
     case kNoDecay: