Changes, needed to use correctly the new AliESDCaloCluster. (Christian)

[u/mrichter/AliRoot.git] / EVGEN / AliGenMUONCocktailpp.h

diff --git a/EVGEN/AliGenMUONCocktailpp.h b/EVGEN/AliGenMUONCocktailpp.h
index 02c269a34516d3d1c6e642187b951e94318593f4..524d42bd4c0392aea08e0384ddd339ea8008bc97 100644 (file)
--- a/EVGEN/AliGenMUONCocktailpp.h
+++ b/EVGEN/AliGenMUONCocktailpp.h
@@ -4,6 +4,7 @@
  * See cxx source for full Copyright notice                               */
 
 #include "AliGenCocktail.h"
+#include "AliDecayer.h"
 
 class AliGenCocktailEntry;
 
@@ -19,31 +20,44 @@ class AliGenMUONCocktailpp : public AliGenCocktail
     Int_t   GetNGenerated()        const {return fNGenerated;}
     Int_t   GetMuonMultiplicity()  const {return fMuonMultiplicity;}
     Float_t GetMuonPtCut()         const {return fMuonPtCut;}
+    Float_t GetMuonPCut()          const {return fMuonPCut;}    
     Float_t GetMuonThetaMin()      const {return fMuonThetaMinCut;}
     Float_t GetMuonThetaMax()      const {return fMuonThetaMaxCut;}        
+    Float_t GetMuonOriginCut()     const {return fMuonOriginCut;}          
+    Float_t GetDecayModeResonance()const {return fDecayModeResonance;}
+    Float_t GetDecayModePythia()   const {return fDecayModePythia;}
     
-    void    SetMuonMultiplicity(Int_t MuonMultiplicity) { fMuonMultiplicity= MuonMultiplicity;}
+    void    SetMuonMultiplicity(Int_t MuonMultiplicity) { fMuonMultiplicity = MuonMultiplicity;}
     void    SetMuonPtCut(Float_t PtCut) { fMuonPtCut = PtCut;}
+    void    SetMuonPCut(Float_t PCut) { fMuonPCut = PCut;}    
+    void    SetMuonOriginCut(Float_t originCut) { fMuonOriginCut = originCut;}
     void    SetMuonThetaRange(Float_t ThetaMin, Float_t ThetaMax){
        fMuonThetaMinCut=ThetaMin;
        fMuonThetaMaxCut=ThetaMax; }    
-
+    void    SetDecayer(AliDecayer* decayer){fDecayer = decayer;}
+    void    SetDecayModeResonance(Decay_t decay){ fDecayModeResonance = decay;}
+    void    SetDecayModePythia(Decay_t decay){ fDecayModePythia = decay;}
  protected:
 
     //
  private:
     AliGenMUONCocktailpp(const AliGenMUONCocktailpp &cocktail); 
     AliGenMUONCocktailpp & operator=(const AliGenMUONCocktailpp &cocktail); 
-
+    AliDecayer* fDecayer;
+    Decay_t fDecayModeResonance; //decay mode in which resonances are forced to decay, default: kAll
+    Decay_t fDecayModePythia; //decay mode in which particles in Pythia are forced to decay, default: kAll
     Float_t fTotalRate;// Total rate of the full cocktail processes
     Int_t   fMuonMultiplicity; // Muon multiplicity for the primordial trigger
     Float_t fMuonPtCut;// Transverse momentum cut for muons
+    Float_t fMuonPCut;// Momentum cut for muons    
     Float_t fMuonThetaMinCut;// Minimum theta cut for muons
     Float_t fMuonThetaMaxCut; // Maximum theta cut for muons
+    Float_t fMuonOriginCut; //use only muons whose "part->Vz()" value is larger than fMuonOrigin
     Int_t   fNSucceded;// Number of Succes in the (di)-muon generation in the acceptance
     Int_t   fNGenerated;// Number of generated cocktails
+    
 
-    ClassDef(AliGenMUONCocktailpp,1)  //  cocktail for physics in the Alice
+    ClassDef(AliGenMUONCocktailpp,2)  //  cocktail for physics in the Alice
 };
 
 #endif