Allow access all primary particles for selective transport

Author: preghenella

DataFormats/simulation/include/SimulationDataFormat/Stack.h

@@ -225,7 +225,7 @@ class Stack : public FairGenericStack
   /// update values in the current event header
   void updateEventStats();
 
-  typedef std::function<bool(const TParticle& p)> TransportFcn;
+  typedef std::function<bool(const TParticle& p, const std::vector<TParticle>& particles)> TransportFcn;
 
  private:
   /// STL stack (FILO) used to handle the TParticles for tracking
@@ -283,7 +283,7 @@ class Stack : public FairGenericStack
 
   bool mIsExternalMode = false; // is stack an external factory or directly used inside simulation?
 
-  TransportFcn mTransportPrimary = [](const TParticle& p) { return false; }; //! a function to inhibit the tracking of a particle
+  TransportFcn mTransportPrimary = [](const TParticle& p, const std::vector<TParticle>& particles) { return false; }; //! a function to inhibit the tracking of a particle
 
   // storage for track references
   std::vector<o2::TrackReference>* mTrackRefs = nullptr; //!

DataFormats/simulation/src/Stack.cxx

@@ -85,15 +85,15 @@ Stack::Stack(Int_t size)
   LOG(INFO) << param;
   TransportFcn transportPrimary;
   if (param.transportPrimary.compare("none") == 0) {
-    transportPrimary = [](const TParticle& p) {
+    transportPrimary = [](const TParticle& p, const std::vector<TParticle>& particles) {
       return false;
     };
   } else if (param.transportPrimary.compare("all") == 0) {
-    transportPrimary = [](const TParticle& p) {
+    transportPrimary = [](const TParticle& p, const std::vector<TParticle>& particles) {
       return true;
     };
   } else if (param.transportPrimary.compare("barrel") == 0) {
-    transportPrimary = [](const TParticle& p) {
+    transportPrimary = [](const TParticle& p, const std::vector<TParticle>& particles) {
       return (std::fabs(p.Eta()) < 2.0);
     };
   } else if (param.transportPrimary.compare("external") == 0) {
@@ -108,7 +108,7 @@ Stack::Stack(Int_t size)
   }
 
   if (param.transportPrimaryInvert) {
-    mTransportPrimary = [transportPrimary](const TParticle& p) { return !transportPrimary; };
+    mTransportPrimary = [transportPrimary](const TParticle& p, const std::vector<TParticle>& particles) { return !transportPrimary; };
   } else {
     mTransportPrimary = transportPrimary;
   }
@@ -230,7 +230,7 @@ void Stack::PushTrack(Int_t toBeDone, Int_t parentId, Int_t pdgCode, Double_t px
     p.SetBit(ParticleStatus::kKeep);
     p.SetBit(ParticleStatus::kPrimary);
     if (toBeDone == 1) {
-      if (mTransportPrimary(p)) {
+      if (mTransportPrimary(p, mPrimaryParticles)) {
         p.SetBit(ParticleStatus::kToBeDone, 1);
         mNumberOfPrimariesforTracking++;
       } else {

run/SimExamples/Selective_Tranport/run.sh …n/SimExamples/Selective_Transport/run.shrun/SimExamples/Selective_Tranport/run.sh renamed to run/SimExamples/Selective_Transport/run.sh run/SimExamples/Selective_Tranport/run.sh renamed to run/SimExamples/Selective_Transport/run.sh

@@ -1,7 +1,7 @@
 o2::data::Stack::TransportFcn
 transportPDG(int pdg = 321)
 {
-  return [pdg](const TParticle& p) -> bool {
+  return [pdg](const TParticle& p, const std::vector<TParticle>& particles) -> bool {
 	   return p.GetPdgCode() == pdg;
 	 };
 }