From c44d1c08de6c32d4e2a5295ca23e06087aa76846 Mon Sep 17 00:00:00 2001
From: morsch <morsch@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Wed, 4 Aug 2004 14:44:50 +0000
Subject: [PATCH] SetVertexSmear: Delegate call to all generators. (T. Kuhr)

---
 EVGEN/AliGenCocktail.cxx | 11 +++++++++++
 EVGEN/AliGenCocktail.h   |  1 +
 2 files changed, 12 insertions(+)

diff --git a/EVGEN/AliGenCocktail.cxx b/EVGEN/AliGenCocktail.cxx
index 36cc83769d6..0cd0a24f45c 100644
--- a/EVGEN/AliGenCocktail.cxx
+++ b/EVGEN/AliGenCocktail.cxx
@@ -186,6 +186,17 @@ AddGenerator(AliGenerator *Generator, const char* Name, Float_t RateExp)
     gAlice->SetGenEventHeader(header); 
 }
 
+void AliGenCocktail::SetVertexSmear(VertexSmear_t smear)
+{
+// Set vertex smearing and propagate it to the generators
+
+  AliGenerator::SetVertexSmear(smear);
+  TIter next(fEntries);
+  while (AliGenCocktailEntry* entry = (AliGenCocktailEntry*)next()) {
+    entry->Generator()->SetVertexSmear(smear);
+  }
+}
+
 AliGenCocktailEntry *  AliGenCocktail::FirstGenerator()
 {
 // Iterator over generators: Initialisation
diff --git a/EVGEN/AliGenCocktail.h b/EVGEN/AliGenCocktail.h
index 20afc493b1b..8883b6a0826 100644
--- a/EVGEN/AliGenCocktail.h
+++ b/EVGEN/AliGenCocktail.h
@@ -25,6 +25,7 @@ class AliGenCocktail : public AliGenerator
     virtual void Init();
     virtual void FinishRun();
     virtual void Generate();
+    virtual void SetVertexSmear(VertexSmear_t smear);
     //
     // Add a new generator to the list
     virtual void AddGenerator
-- 
2.43.0