stream.h changed to iostream.h

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

diff --git a/EVGEN/AliGenCocktailAfterBurner.h b/EVGEN/AliGenCocktailAfterBurner.h
index ccef04fdda0e7243fa8555d3a7abadcbde25ca62..6817fbb4ee37cb99c9b54337193ddd9a1de17cde 100644 (file)
--- a/EVGEN/AliGenCocktailAfterBurner.h
+++ b/EVGEN/AliGenCocktailAfterBurner.h
@@ -11,7 +11,7 @@
 //
 #include "AliGenCocktailAfterBurner.h"
 #include "AliGenCocktail.h"
-
+#include "AliRun.h"
 
 class AliGenCocktailEntry;
 class AliStack;
@@ -38,21 +38,34 @@ class AliGenCocktailAfterBurner : public  AliGenCocktail
     AliGenerator* GetCurrentGenerator();
     virtual void  SetActiveEventNumber(Int_t actev);
     Int_t GetActiveEventNumber() {return fActiveEvent;}
+    virtual Int_t GetNumberOfEvents() {return gAlice->GetEventsPerRun() + fNBgEvents;}
+
     static AliMCProcess IntToMCProcess(Int_t no);
+    void SetNBgEvents(Int_t nbg=0){fNBgEvents = nbg;}
+
+
  protected:
-    Int_t fNAfterBurners; 
-    TList  *fAfterBurnerEntries;
-    Bool_t fGenerationDone;
-    TObjArray *fInternalStacks; // ! List of internal stacks
-    Int_t fCurrentEvent; //  Number of current event/stack
+    Int_t fNAfterBurners;       // Number of afterburners  
+    TList  *fAfterBurnerEntries;// List of afterburners
+    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
     
+
+    AliStack* fActiveStack;   //! pointer to the current stack
+    Int_t fActiveEvent;       //HBT Processor needs more then one event to do correlations
+                              //Due to complications in fortran, it first calls C routine
+                              //that sets the active event to be read. All alihbtp_gettrack
+                              //are addressed to this event
     
-    AliStack* fActiveStack; // ! pointer to the current stack
-    Int_t fActiveEvent;
-   
-    AliGenerator *fCurrentGenerator;
+    AliGenerator *fCurrentGenerator;      // Current event generator 
+    Int_t fNBgEvents;
     
-    ClassDef(AliGenCocktailAfterBurner,1) // Particle cocktail generator a la SHAKER
+    ClassDef(AliGenCocktailAfterBurner,2) // Particle cocktail generator a la SHAKER
+                                          //background events added
 };
 
 inline  AliGenerator*