If present, pass header to container.
[u/mrichter/AliRoot.git] / PYTHIA6 / AliGenPythia.cxx
index 90f7144..f505b7e 100644 (file)
@@ -872,11 +872,21 @@ void AliGenPythia::MakeHeader()
        }
     
 //
-//  Pass header to RunLoader
+//  Pass header
 //
-    AliRunLoader::GetRunLoader()->GetHeader()->SetGenEventHeader(fHeader);   
+    AddHeader(fHeader);
 }
-       
+
+void AliGenPythia::AddHeader(AliGenEventHeader* header)
+{
+    // Add header to container or runloader
+    if (fContainer) {
+       fContainer->AddHeader(header);
+    } else {
+       AliRunLoader::GetRunLoader()->GetHeader()->SetGenEventHeader(header);   
+    }
+}
+
 
 Bool_t AliGenPythia::CheckTrigger(TParticle* jet1, TParticle* jet2)
 {