updated

[u/mrichter/AliRoot.git] / PYTHIA8 / AliDecayerPythia8.cxx

diff --git a/PYTHIA8/AliDecayerPythia8.cxx b/PYTHIA8/AliDecayerPythia8.cxx
index 0be6f407984a155c2eaf348a128c1dec5c2a63df..7fb2265a4bcdd29304058e0643ef10c5bd522d42 100644 (file)
--- a/PYTHIA8/AliDecayerPythia8.cxx
+++ b/PYTHIA8/AliDecayerPythia8.cxx
@@ -20,6 +20,8 @@
 #include <TMath.h>
 #include <TPDGCode.h>
 #include <TLorentzVector.h>
+#include <TClonesArray.h>
+#include <TParticle.h>
 #include "AliTPythia8.h"
 #include "AliDecayerPythia8.h"
 #include "ParticleData.h"
@@ -55,8 +57,17 @@ void AliDecayerPythia8::Decay(Int_t pdg, TLorentzVector* p)
 //___________________________________________________________________________
 Int_t AliDecayerPythia8::ImportParticles(TClonesArray *particles)
 {
-   //import the decay products into particles array
-   return (fPythia8->ImportParticles(particles, "All"));
+  //import the decay products into particles array
+  const Float_t kconvT=0.001/2.999792458e8; // mm/c to seconds conversion
+  const Float_t kconvL=1./10; // mm to cm conversion
+  int np = fPythia8->ImportParticles(particles, "All");
+  // pythia assigns decay time in mm/c, convert to seconds
+  for (int ip=1;ip<np;ip++) {
+    TParticle* prod = (TParticle*)particles->At(ip);
+    if (!prod) continue;
+    prod->SetProductionVertex(prod->Vx()*kconvL,prod->Vy()*kconvL,prod->Vz()*kconvL,kconvT*prod->T());
+  }
+  return np;
 }