Extra header added to the list

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

diff --git a/EVGEN/AliGenCocktailEntry.h b/EVGEN/AliGenCocktailEntry.h
index f9423ab2584bc3530a9fe6f61544645ab0279f5c..9a241262bff21f26afdcf0b1a26560f27eecda51 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,29 +20,35 @@ class AliGenCocktailEntry : public TNamed
     AliGenCocktailEntry();
     AliGenCocktailEntry(const AliGenCocktailEntry &entry);
     AliGenCocktailEntry
-       (AliGenerator* Generator, 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;}
-    Int_t GetFirst(){return fFirst;}
-    Int_t GetLast (){return fLast;}
-    Float_t Rate(){return fRate;}
-    void  PrintInfo();
+    void SetFormula(TFormula* formula) {fFormula = formula;}
+    void SetNTimes(Int_t ntimes) {fNTimes = ntimes;}
+    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 NTimes()   const {return fNTimes;}
+    Int_t   Bin()    const {return Int_t(fRate);}
+    void  PrintInfo() const;
+    TFormula* Formula() const {return fFormula;}
     AliGenCocktailEntry & operator =(const AliGenCocktailEntry & rhs);
- private:
-    void Copy(AliGenCocktailEntry&) const;
  protected:
     AliGenerator *fGenerator;   // Pointer to generator
     Int_t fNGenerated;          // Number of primaries generated
     Int_t fFirst;               // First index in list of primaries
     Int_t fLast;                // Last index in list of primaries
     Float_t fRate;              // Rate per event
+    Int_t   fNTimes;            // Generate is called N times
     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, 3) // Generator entry of AliGenCocktail
 };
 #endif