updates for train

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

diff --git a/EVGEN/AliGenFunction.cxx b/EVGEN/AliGenFunction.cxx
index 52d87bb3a58858720c656172b368f0c1e73954c8..659df3861d7840fd098209da612149c6c2307426 100644 (file)
--- a/EVGEN/AliGenFunction.cxx
+++ b/EVGEN/AliGenFunction.cxx
@@ -58,9 +58,11 @@
 #include <TDatabasePDG.h>
 
 #include "AliRun.h"
+#include "AliLog.h"
 #include "AliESDtrack.h"
 #include "AliESDVertex.h"
 #include "AliGenFunction.h"
+#include "AliGenEventHeader.h"
 
 ClassImp(AliGenFunction)
 
@@ -160,7 +162,8 @@ void AliGenFunction::Generate()
     naccepted++;
   }
 
-
+  AliGenEventHeader* header = new AliGenEventHeader("THn");
+  gAlice->SetGenEventHeader(header);
   return;
 }
 //-----------------------------------------------------------------------------
@@ -189,7 +192,7 @@ void AliGenFunction::Init()
 
 void AliGenFunction::SetFunctions(TF1 * momentum, TF1 *fphi, TF1 *ftheta,TF3 * position, TF1* pdg){
   //
-  //
+  // Set the function
   //
   fFMomentum = momentum;
   fFPhi = fphi;
@@ -200,7 +203,7 @@ void AliGenFunction::SetFunctions(TF1 * momentum, TF1 *fphi, TF1 *ftheta,TF3 * p
 
 void AliGenFunction::SetCylinder(Double_t refR, Double_t zmin, Double_t zmax){
   //
-  //
+  // Set the cylinder geometry
   //
   fRefRadius = refR;          // reference radius to be crossed
   fZmin = zmin;               // minimal z at reference radius