Fix axis label

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

diff --git a/EVGEN/AliGenEMCocktail.h b/EVGEN/AliGenEMCocktail.h
index 96ed2497244226586eb2947f5de3b1da3db2fe25..62f7476847191dc159d96bc4cac4a6aa9c89e40c 100644 (file)
--- a/EVGEN/AliGenEMCocktail.h
+++ b/EVGEN/AliGenEMCocktail.h
@@ -23,7 +23,8 @@ class AliGenEMCocktail : public AliGenCocktail
 public:
 
     AliGenEMCocktail();
-  enum GeneratorCode { kGenPizero=0, kGenEta, kGenRho, kGenOmega, kGenEtaprime, kGenPhi, kGenJpsi, kGenPromptRealGamma, kGenThermRealGamma, kGenPromptVirtGamma, kGenThermVirtGamma, kGENs };
+  enum GeneratorIndex_t { kPizero=0, kEta, kRho, kOmega, kEtaprime, kPhi, kJpsi, kDirectRealGamma, kDirectVirtGamma, kGENs };
+  enum ParticeGenerator_t { kGenPizero=0x001, kGenEta=0x002, kGenRho=0x004, kGenOmega=0x008, kGenEtaprime=0x010, kGenPhi=0x020, kGenJpsi=0x040, kGenDirectRealGamma=0x100, kGenDirectVirtGamma=0x200, kGenHadrons=0x7f, kGenGammas=0x300 };
 
     virtual ~AliGenEMCocktail();    
     virtual void Init();
@@ -35,11 +36,12 @@ public:
     void    SetDecayMode(Decay_t decay){ fDecayMode = decay;}
     void    SetWeightingMode(Weighting_t weight){ fWeightingMode = weight;}
   void    SetNPart(Int_t npart){ fNPart = npart; }
-  void    SetPtParam(Int_t PtSelect){ fPtSelect = PtSelect; }
-  void    SetCentrality(Int_t cent){ fCentrality = cent; }
+  void    SetPtParam(AliGenEMlib::PtParamSet_t PtSelect){ fPtSelect = PtSelect; }
+  void    SetCentrality(AliGenEMlib::Centrality_t cent){ fCentrality = cent; }
   void    SetV2Systematic(Int_t v2sys){ fV2Systematic = v2sys; }
   void    SetForceGammaConversion(Bool_t force=kTRUE){ fForceConv=force; }
-  void    SetHeaviestParticle(Int_t part){ fHeaviestParticle=part; }
+  void    SetHeaviestHadron(ParticeGenerator_t part);
+  void    SelectMotherParticles(ParticeGenerator_t part){ fSelectedParticles=part; }
     
 private:
     AliGenEMCocktail(const AliGenEMCocktail &cocktail); 
@@ -54,12 +56,12 @@ private:
     Int_t    fNPart;             // multiplicity of each source per event
     Double_t fYieldArray[kGENs]; // array of dN/dy for each source
 
-  Int_t fPtSelect; // selected pT parameterization
-  Int_t fCentrality; // selected centrality
+  AliGenEMlib::PtParamSet_t fPtSelect; // selected pT parameterization
+  AliGenEMlib::Centrality_t fCentrality; // selected centrality
   Int_t fV2Systematic; //selected systematic error for v2 parameters
 
   Bool_t fForceConv; //select whether you want to force all gammas to convert imidediately
-  Int_t fHeaviestParticle; //select up to which particle to simulate
+  Int_t fSelectedParticles; //which particles to simulate
 
     ClassDef(AliGenEMCocktail,1)  //  cocktail for EM physics
 };