X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=EVGEN%2FAliGenCocktailAfterBurner.h;h=0930e19167839aa868598ae6134ebab634101fee;hb=08f220c8b0cda791859fb0854c89f2d57ec9034e;hp=fd255051f623dd5610c17f698b30b04e32388acf;hpb=acc86a241efbb2413429532a3006c66e29f2ab8c;p=u%2Fmrichter%2FAliRoot.git

diff --git a/EVGEN/AliGenCocktailAfterBurner.h b/EVGEN/AliGenCocktailAfterBurner.h
index fd255051f62..0930e191678 100644
--- a/EVGEN/AliGenCocktailAfterBurner.h
+++ b/EVGEN/AliGenCocktailAfterBurner.h
@@ -9,11 +9,17 @@
 // (Container is itself an AliGenerator)
 // Author: piotr.skowronski@cern.ch 
 //
+
+#include <TMCProcess.h>
+
 #include "AliGenCocktail.h"
-#include "AliRun.h"
+#include "AliRunLoader.h"
 
 class AliGenCocktailEntry;
 class AliStack;
+class AliCollisionGeometry;
+class AliGenCocktailEventHeader;
+
 
 class AliGenCocktailAfterBurner : public  AliGenCocktail
 {
@@ -23,9 +29,7 @@ class AliGenCocktailAfterBurner : public  AliGenCocktail
 
  public:
     AliGenCocktailAfterBurner();
-    AliGenCocktailAfterBurner(const AliGenCocktailAfterBurner& in);
     virtual ~AliGenCocktailAfterBurner();
-    AliGenCocktailAfterBurner & operator=(const AliGenCocktailAfterBurner & rhs);
     
     virtual void  Init();
     virtual void  Generate();
@@ -37,12 +41,15 @@ class AliGenCocktailAfterBurner : public  AliGenCocktail
     
     AliStack*     GetStack(Int_t n) const;
     AliStack*     GetActiveStack() const{return fActiveStack;}
-    
+
+    AliCollisionGeometry* GetCollisionGeometry(Int_t n) const;
+ 
     AliGenerator* GetCurrentGenerator() const;
     virtual void  SetActiveEventNumber(Int_t actev);
     Int_t         GetActiveEventNumber() const {return fActiveEvent;}
-    virtual Int_t GetNumberOfEvents() const {return gAlice->GetEventsPerRun() + fNBgEvents;}
+    virtual Int_t GetNumberOfEvents() const {return AliRunLoader::Instance()->GetNumberOfEventsPerRun() + fNBgEvents;}
     void          SetNBgEvents(Int_t nbg=0){fNBgEvents = nbg;}
+    AliGenCocktailEventHeader* GetActiveEventHeader() const { return fHeaders[fActiveEvent]; }
 
     static TMCProcess IntToMCProcess(Int_t no);
 
@@ -54,6 +61,10 @@ class AliGenCocktailAfterBurner : public  AliGenCocktail
                                 //   if true just return event to gAlice
                                 //   
     TObjArray *fInternalStacks; //! List of internal stacks
+
+    AliCollisionGeometry**      fCollisionGeometries; //! List of Collision Geometries
+    AliGenCocktailEventHeader** fHeaders;             //! List of Headers
+    
     Int_t fCurrentEvent;        //  Number of current event/stack
     
 
@@ -68,7 +79,10 @@ class AliGenCocktailAfterBurner : public  AliGenCocktail
                                           //(events that are generated only temporarly)
                                           //needed by some afterburners that works better with higher statistics 
                                           //this generates such a artificial one
- private:   
+ private:
+    AliGenCocktailAfterBurner(const AliGenCocktailAfterBurner& in);
+    AliGenCocktailAfterBurner & operator=(const AliGenCocktailAfterBurner & rhs);
+
     ClassDef(AliGenCocktailAfterBurner,2) // Particle cocktail generator a la SHAKER
                                           //background events added
 };