SetVertexSmear: Delegate call to all generators. (T. Kuhr)

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

diff --git a/EVGEN/AliGenCocktail.cxx b/EVGEN/AliGenCocktail.cxx
index 36cc83769d62026bbe88d81feae9dc559a163a69..0cd0a24f45c25312ebf6c88386e3e879656b11ff 100644 (file)
--- 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