Apply encoding to boxgen particles

AliceO2Group · Jan 6, 2025 · c68ddc8 · c68ddc8
1 parent 8b7ba4e
commit c68ddc8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Generators/include/Generators/BoxGenerator.h b/Generators/include/Generators/BoxGenerator.h
@@ -19,6 +19,8 @@
 #include <vector>
 #include <Generators/BoxGunParam.h>
 #include "SimulationDataFormat/MCEventHeader.h"
+#include "SimulationDataFormat/ParticleStatus.h"
+#include <SimulationDataFormat/MCGenProperties.h>
 
 namespace o2::eventgen
 {
@@ -90,6 +92,14 @@ class BoxGenerator : public Generator
   {
     mParticles.clear();
     std::copy(mEvent.begin(), mEvent.end(), std::back_insert_iterator(mParticles));
+    for (auto& particle : mParticles) {
+      auto statusCode = particle.GetStatusCode();
+      if (!mcgenstatus::isEncoded(statusCode)) {
+        particle.SetStatusCode(mcgenstatus::MCGenStatusEncoding(statusCode, 0).fullEncoding);
+      }
+      // Set the transport bit according to the HepMC status code
+      particle.SetBit(ParticleStatus::kToBeDone, mcgenstatus::getHepMCStatusCode(particle.GetStatusCode()) == 1);
+    }
     return true;
   }