ROOT map generation + RAW module

[u/mrichter/AliRoot.git] / EVGEN / AliGenCocktail.cxx

diff --git a/EVGEN/AliGenCocktail.cxx b/EVGEN/AliGenCocktail.cxx
index 2ac18c385f6ae409641bf7dbbf1db06edeef2129..32fd1bb97f636482bc2b369e235853033abd2926 100644 (file)
--- a/EVGEN/AliGenCocktail.cxx
+++ b/EVGEN/AliGenCocktail.cxx
@@ -259,10 +259,15 @@ AddGenerator(AliGenerator *Generator, const char* Name, Float_t RateExp, TFormul
     
     next.Reset();
 
-// Event Vertex
+    // Event Vertex
     fHeader->SetPrimaryVertex(eventVertex);
     fHeader->CalcNProduced();
-    gAlice->SetGenEventHeader(fHeader); 
+    if (fContainer) {
+      fHeader->SetName(fName);
+      fContainer->AddHeader(fHeader);
+    } else {
+      gAlice->SetGenEventHeader(fHeader);      
+    }
 }
 
 void AliGenCocktail::SetVertexSmear(VertexSmear_t smear)