Including h-h correlations analysis with hybrid tracks selection

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

diff --git a/EPOS/AliGenEpos.cxx b/EPOS/AliGenEpos.cxx
index ed994f12329629bb8d61e48dbd07adfb51cc58d5..e56fbfa545711712a4018a8075aa880f4af9ebfb 100644 (file)
--- a/EPOS/AliGenEpos.cxx
+++ b/EPOS/AliGenEpos.cxx
@@ -1,11 +1,11 @@
-/*
- * AliGenEpos.cxx
- *
- *  ALICE event generator based on EPOS model from Klaus Werner
- *
- *  Created on: Feb 28, 2009
- *      Author: Piotr Ostrowski, postrow@if.pw.edu.pl
- */
+//
+// AliGenEpos.cxx
+//
+//  ALICE event generator based on EPOS model from Klaus Werner
+//
+//  Created on: Feb 28, 2009
+//      Author: Piotr Ostrowski, postrow@if.pw.edu.pl
+//
 
 #include "AliGenEpos.h"
 #include "TEpos.h"
@@ -36,6 +36,7 @@ AliGenEpos::AliGenEpos(Int_t npart) : AliGenMC(npart),
 }
 
 void AliGenEpos::Init() {
+  // Sets up TEpos
        AliGenMC::Init();
        TEpos *epos = GetTEpos();
        epos->SetLaproj(this->fZProjectile);
@@ -52,19 +53,23 @@ void AliGenEpos::Init() {
 }
 
 void AliGenEpos::Generate() {
-         Float_t polar[3]   =   {0,0,0};
+  // Does actual generation and output conversion
+         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();
@@ -72,6 +77,7 @@ void AliGenEpos::Generate() {
 
          Int_t *idsOnStack = NULL;
          idsOnStack = new Int_t[np];
+         for (int i = 0; i < np; i++) idsOnStack[i] = 0;
          TParticle *iparticle;
 
          for (int i = 0; i < np; i++) {
@@ -88,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++;
@@ -106,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++;
@@ -121,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);
@@ -148,6 +155,7 @@ void AliGenEpos::Generate() {
 
          header->SetNProduced(fNprimaries);
          header->SetPrimaryVertex(eventVertex);
+         header->SetInteractionTime(time0);
 
          header->SetImpactParameter(GetTEpos()->GetBimevt());
          header->SetReactionPlaneAngle(GetTEpos()->GetPhievt());