Plotting macro updates

[u/mrichter/AliRoot.git] / EPOS / AliGenEpos.cxx

diff --git a/EPOS/AliGenEpos.cxx b/EPOS/AliGenEpos.cxx
index 495f9002856f3f76109be32aad603c7daec04eee..e56fbfa545711712a4018a8075aa880f4af9ebfb 100644 (file)
--- a/EPOS/AliGenEpos.cxx
+++ b/EPOS/AliGenEpos.cxx
@@ -57,16 +57,19 @@ void AliGenEpos::Generate() {
          Float_t polar[3]    =   {0,0,0};
          Float_t origin0[3]  =   {0,0,0};
          Float_t origin[3]   =   {0,0,0};
+         Float_t time0 = 0.;
+         Float_t time  = 0.;
          fNprimaries = 0;
          Int_t nt  = 0; //output parameter for PushTrack
 
          Vertex();
          for (int j=0; j < 3; j++) origin0[j] = fVertex[j];
+         time0 = fTime;
 
          // Generate one event
 
          GetTEpos()->GenerateEvent();
-         AliWarning("Generated");
+         //AliWarning("Generated");
          GetTEpos()->ImportParticles(&fParticles);
 
          Int_t np = fParticles.GetEntriesFast();
@@ -91,15 +94,15 @@ void AliGenEpos::Generate() {
                          origin[0] = iparticle->Vx();
                          origin[1] = iparticle->Vy();
                          origin[2] = iparticle->Vz();
+                         time      = iparticle->T();
                          //doubled track with freeze out coordinates for femtoscopy
-                         PushTrack(0,
-                                         imo>=0?idsOnStack[imo]:-1,
-                                         iparticle->GetPdgCode(),
+                         PushTrack(0, imo>=0?idsOnStack[imo]:-1,
+                               iparticle->GetPdgCode(),
                                iparticle->Px(),iparticle->Py(),iparticle->Pz(),iparticle->Energy(),
-                               origin[0], origin[1], origin[2],
-                               iparticle->T(),
+                               origin[0], origin[1], origin[2], time,
                                polar[0],polar[1],polar[2],
-                               hasMother ? kPDecay:kPNoProcess,nt);
+                               hasMother ? kPDecay:kPNoProcess, nt, 1., 
+                               iparticle->GetStatusCode());
 
                      idsOnStack[i] = nt;
                      fNprimaries++;
@@ -109,12 +112,13 @@ void AliGenEpos::Generate() {
                      origin[0] += origin0[0];
                      origin[1] += origin0[1];
                      origin[2] += origin0[2];
+                     time      += time0;
                          PushTrack(1,
                                          nt,   //doubled track as mother
                                          iparticle->GetPdgCode(),
                                iparticle->Px(),iparticle->Py(),iparticle->Pz(),iparticle->Energy(),
                                origin[0], origin[1], origin[2],
-                               iparticle->T(),
+                               time,
                                polar[0],polar[1],polar[2],
                                kPDecay,nt);
                      fNprimaries++;
@@ -124,14 +128,14 @@ void AliGenEpos::Generate() {
                          origin[0] = iparticle->Vx();
                          origin[1] = iparticle->Vy();
                          origin[2] = iparticle->Vz();
-                         PushTrack(0,
-                                         imo>=0?idsOnStack[imo]:-1,
-                                         iparticle->GetPdgCode(),
+                         time      = iparticle->T();
+                         PushTrack(0, imo>=0?idsOnStack[imo]:-1,
+                               iparticle->GetPdgCode(),
                                iparticle->Px(),iparticle->Py(),iparticle->Pz(),iparticle->Energy(),
-                               origin[0], origin[1], origin[2],
-                               iparticle->T(),
+                               origin[0], origin[1], origin[2], time,
                                polar[0],polar[1],polar[2],
-                               hasMother ? kPDecay:kPNoProcess,nt);
+                               hasMother ? kPDecay:kPNoProcess, nt, 1., 
+                               iparticle->GetStatusCode());
                      idsOnStack[i] = nt;
                      fNprimaries++;
                      KeepTrack(nt);
@@ -151,6 +155,7 @@ void AliGenEpos::Generate() {
 
          header->SetNProduced(fNprimaries);
          header->SetPrimaryVertex(eventVertex);
+         header->SetInteractionTime(time0);
 
          header->SetImpactParameter(GetTEpos()->GetBimevt());
          header->SetReactionPlaneAngle(GetTEpos()->GetPhievt());