Kalman filter vertex in Psi2s task

[u/mrichter/AliRoot.git] / THijing / AliGenBeamGasNew.cxx

diff --git a/THijing/AliGenBeamGasNew.cxx b/THijing/AliGenBeamGasNew.cxx
index 3c8560ab1d739a6b346a95757d61bcb497635c30..60d235c0c1eaa85a9b47b7e254aeaf5d2d3ea846 100644 (file)
--- a/THijing/AliGenBeamGasNew.cxx
+++ b/THijing/AliGenBeamGasNew.cxx
@@ -35,7 +35,7 @@
 #include "AliGenCocktail.h" 
 #include "AliGenCocktailEntry.h"
 #include "AliGenCocktailEventHeader.h"
-#include "../THijing/AliGenHijing.h" 
+#include "AliGenHijing.h" 
 #include "AliCollisionGeometry.h"
 #include "AliRun.h"
 #include "AliMC.h"
@@ -66,6 +66,7 @@ AliGenBeamGasNew::~AliGenBeamGasNew()
 void AliGenBeamGasNew::SetTimeWindow(Float_t twindow) { fTwindow = twindow; }
 
 bool AliGenBeamGasNew::SetRate(Float_t rate) {
+// Set the Rate
   if (rate >= 0) {  
     fRate = rate; 
     return true;
@@ -83,7 +84,7 @@ void AliGenBeamGasNew::Init()
 
   fVertexSmear = kPerEvent;
   fVertexSource = kInternal;
-  fRandom = kTRUE;
+  fSRandom = kTRUE;
 
   // Adding default underlying event in case none was specified
   // p-O-collision at 7 TeV (fixed target)
@@ -120,7 +121,7 @@ void AliGenBeamGasNew::Init()
   
   next.Reset();
   
-  if (fRandom) {
+  if (fSRandom) {
     fProb.Set(fNGenerators);
     next.Reset();
     Float_t sum = 0.;
@@ -229,8 +230,8 @@ void AliGenBeamGasNew::Generate()
       entry = (AliGenCocktailEntry*) fEntries->At(i);
       entry->SetFirst(lastpart);
       gen = entry->Generator();
-      entry->Generator()->SetVertex(fVertex.At(0), fVertex.At(1), fVertex.At(2));
-      entry->Generator()->Generate();
+      gen->SetVertex(fVertex.At(0), fVertex.At(1), fVertex.At(2));
+      gen->Generate();
       entry->SetLast(partArray->GetEntriesFast());
     }