Merge pull request #96 from preghenella/rdev-dndneta

Prototype for hybrid full/fast simulation (missing parts)

Author: preghenella

src/TrackUtils.cc

@@ -2,6 +2,8 @@
 /// @email: [email protected]
 
 #include "TrackUtils.hh"
+#include "TParticle.h"
+#include "TParticlePDG.h"
 
 namespace o2
 {
@@ -140,6 +142,19 @@ TrackUtils::convertGenParticleToO2Track(const GenParticle& particle, O2Track& o2
 
 /*****************************************************************/
 
+void
+TrackUtils::convertTParticleToO2Track(const TParticle& particle, O2Track& o2track)
+{
+
+  std::array<float, 3> xyz = {static_cast<float>(particle.Vx()), static_cast<float>(particle.Vy()), static_cast<float>(particle.Vz())};
+  std::array<float, 3> ptetaphi = {static_cast<float>(particle.Pt()), static_cast<float>(particle.Eta()), static_cast<float>(particle.Phi())};
+  int charge = particle.GetPDG() ? particle.GetPDG()->Charge() / 3. : 0;
+  makeO2Track(o2track, xyz, ptetaphi, charge);
+  
+}
+  
+/*****************************************************************/
+
 bool
 TrackUtils::propagateToDCA(O2Track &o2track, std::array<float, 3> xyz, float Bz)
 {

src/TrackUtils.hh

@@ -8,7 +8,9 @@
 #include "classes/DelphesClasses.h"
 
 using O2Track = o2::track::TrackParCov;
-  
+
+class TParticle;
+
 namespace o2
 {
 namespace delphes
@@ -23,6 +25,7 @@ public:
 
   static void convertTrackToO2Track(const Track &track, O2Track &o2track, bool atDCA);
   static void convertGenParticleToO2Track(const GenParticle &particle, O2Track &o2track);
+  static void convertTParticleToO2Track(const TParticle &particle, O2Track &o2track);
 
   static bool propagateToDCA(O2Track &o2track, std::array<float, 3> xyz, float Bz);