updates for train

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

diff --git a/EVGEN/AliGenCocktailEntry.h b/EVGEN/AliGenCocktailEntry.h
index b896f22ace44076d39f86d2ec52b1e73646bcd56..17b3e0988784f6c9961fa588d11635584233d832 100644 (file)
--- a/EVGEN/AliGenCocktailEntry.h
+++ b/EVGEN/AliGenCocktailEntry.h
@@ -12,7 +12,7 @@
 #include <TNamed.h>
 
 class AliGenerator;
-
+class TFormula;
 
 class AliGenCocktailEntry : public TNamed
 {
@@ -20,16 +20,19 @@ class AliGenCocktailEntry : public TNamed
     AliGenCocktailEntry();
     AliGenCocktailEntry(const AliGenCocktailEntry &entry);
     AliGenCocktailEntry
-       (AliGenerator* pGenerator, const char* Name, Float_t RateExp);
+      (AliGenerator* pGenerator, const char* Name, Float_t RateExp);
     ~AliGenCocktailEntry(){;}
     AliGenerator* Generator() {return fGenerator;}
     void SetGenerator(AliGenerator* generator){fGenerator=generator;}
-    void SetFirst(Int_t first){fFirst=first;}
-    void SetLast (Int_t last ){fLast =last;}
+    void SetFormula(TFormula* formula) {fFormula = formula;}
+    void SetFirst(Int_t first) {fFirst=first;}
+    void SetLast (Int_t last ) {fLast =last;}
     Int_t GetFirst() const {return fFirst;}
     Int_t GetLast () const {return fLast;}
     Float_t Rate()   const {return fRate;}
+    Int_t   Bin()    const {return Int_t(fRate);}
     void  PrintInfo() const;
+    TFormula* Formula() const {return fFormula;}
     AliGenCocktailEntry & operator =(const AliGenCocktailEntry & rhs);
  protected:
     AliGenerator *fGenerator;   // Pointer to generator
@@ -39,9 +42,10 @@ class AliGenCocktailEntry : public TNamed
     Float_t fRate;              // Rate per event
     Float_t fKineBias;          // Bias due to kinematic selecion
     Float_t fBias;              // Bias
+    TFormula* fFormula;         // Formula to calculate number of signals per event    
     void Copy(TObject&) const;
  private:
-    ClassDef(AliGenCocktailEntry,1) // Generator entry of AliGenCocktail
+    ClassDef(AliGenCocktailEntry, 2) // Generator entry of AliGenCocktail
 };
 #endif