New files needed by the HBT processor (P.Skowronski)
[u/mrichter/AliRoot.git] / EVGEN / AliGenCocktailAfterBurner.h
index 3a0e78d..229a16d 100644 (file)
@@ -11,7 +11,7 @@
 //
 #include "AliGenCocktailAfterBurner.h"
 #include "AliGenCocktail.h"
-
+#include "AliRun.h"
 
 class AliGenCocktailEntry;
 class AliStack;
@@ -38,18 +38,19 @@ class AliGenCocktailAfterBurner : public  AliGenCocktail
     AliGenerator* GetCurrentGenerator();
     virtual void  SetActiveEventNumber(Int_t actev);
     Int_t GetActiveEventNumber() {return fActiveEvent;}
-    virtual void SetNumberOfEvents(Int_t n)   {fNumberOfEvents=n;}
-    virtual Int_t GetNumberOfEvents() {return fNumberOfEvents;}
+    virtual Int_t GetNumberOfEvents() {return gAlice->GetEventsPerRun();}
 
     static AliMCProcess IntToMCProcess(Int_t no);
  protected:
     Int_t fNAfterBurners;       // Number of afterburners  
     TList  *fAfterBurnerEntries;// List of afterburners
-    Bool_t fGenerationDone;
+    Bool_t fGenerationDone;     // flag if generation is already done 
+                                //   during first call of Generate method
+                                //   if true just return event to gAlice
+                                //   
     TObjArray *fInternalStacks; //! List of internal stacks
     Int_t fCurrentEvent;        //  Number of current event/stack
     
-    Int_t fNumberOfEvents;      // Number of events to process
 
     AliStack* fActiveStack;   //! pointer to the current stack
     Int_t fActiveEvent;       //HBT Processor needs more then one event to do correlations