adjust output filename for LEGO

[u/mrichter/AliRoot.git] / THydjet / AliGenHydjet.cxx

diff --git a/THydjet/AliGenHydjet.cxx b/THydjet/AliGenHydjet.cxx
index 6f10a334048e5a53e508780de602c5dd2a677b87..915c1745b929dcf7cf0e9100a5125b8d491a2261 100755 (executable)
--- a/THydjet/AliGenHydjet.cxx
+++ b/THydjet/AliGenHydjet.cxx
@@ -27,6 +27,7 @@
 #include <TClonesArray.h>
 
 #include "AliGenHydjet.h"
+#include "AliLog.h"
 #include "AliGenHydjetEventHeader.h"
 #include "AliRun.h"
 #include "AliPythiaRndm.h"
@@ -111,6 +112,7 @@ void AliGenHydjet::Generate()
   Float_t polar[3]    =   {0,0,0};
   Float_t origin[3]   =   {0,0,0};
   Float_t origin0[3]  =   {0,0,0};
+  Float_t time0 = 0.;
   Float_t p[3];
   Float_t tof;
 
@@ -122,9 +124,11 @@ void AliGenHydjet::Generate()
   kf = 0;
 
   for (j = 0;j < 3; j++) origin0[j] = fOrigin[j];
+  time0 = fTimeOrigin;
   if(fVertexSmear == kPerEvent) {
       Vertex();
       for (j=0; j < 3; j++) origin0[j] = fVertex[j];
+      time0 = fTime;
   }
 
   while(1)
@@ -134,7 +138,6 @@ void AliGenHydjet::Generate()
       fHydjet->ImportParticles(&fParticles,"All");
 
       Int_t np = fParticles.GetEntriesFast();
-      printf("\n **************************************************%d\n",np);
       Int_t nc = 0;
       if (np == 0 ) continue;
       Int_t i;
@@ -148,6 +151,7 @@ void AliGenHydjet::Generate()
       fVertex[0] = origin0[0];
       fVertex[1] = origin0[1];
       fVertex[2] = origin0[2];
+      fTime = time0;
 
 //
 // Now select the final state particles
@@ -185,7 +189,7 @@ void AliGenHydjet::Generate()
            origin[0] = fVertex[0]+iparticle->Vx()/10;
            origin[1] = fVertex[1]+iparticle->Vy()/10;
            origin[2] = fVertex[2]+iparticle->Vz()/10;
-           tof   = kconv*iparticle->T();
+           tof   = fTime + kconv*iparticle->T();
 
            imo = -1;
            //imo = iparticle->GetFirstMother();
@@ -197,7 +201,7 @@ void AliGenHydjet::Generate()
       } // particle loop
       delete[] pSelected;
 
-      printf("\n I've put %i particles on the stack \n",nc);
+      AliInfo(Form("\n I've put %i particles on the stack \n",nc));
       if (nc > 0) break;
   } // event loop
   MakeHeader();
@@ -263,6 +267,7 @@ void AliGenHydjet::MakeHeader()
 
 // Event Vertex
     header->SetPrimaryVertex(fVertex);
+    header->SetInteractionTime(fTime);
     AddHeader(header);
     fCollisionGeometry = (AliGenHydjetEventHeader*)  header;
 }