allow selection of signed PDG

Author: alcaliva

MC/config/PWGLF/pythia8/generator_pythia8_coalescence.C

@@ -25,13 +25,19 @@ class GeneratorPythia8Coalescence : public o2::eventgen::GeneratorPythia8
 {
 public:
   /// Constructor
-  GeneratorPythia8Coalescence(std::vector<unsigned int> pdgList, int input_trigger_ratio = 1, double coal_momentum = 0.4)
+  GeneratorPythia8Coalescence(std::vector<int> pdgList, int input_trigger_ratio = 1, double coal_momentum = 0.4)
       : o2::eventgen::GeneratorPythia8()
   {
     fmt::printf(">> Coalescence generator %d\n", input_trigger_ratio);
     mInverseTriggerRatio = input_trigger_ratio;
     mCoalMomentum = coal_momentum;
-    mPdgList = pdgList;
+    mRequestedPdgList = pdgList;
+    for (auto pdg : pdgList) {
+      unsigned int absPdg = static_cast<unsigned int>(std::abs(pdg));
+      if (std::find(mPdgList.begin(), mPdgList.end(), absPdg) == mPdgList.end()) {
+        mPdgList.push_back(absPdg);
+      }
+    }
   }
   /// Destructor
   ~GeneratorPythia8Coalescence() = default;
@@ -57,7 +63,14 @@ protected:
       {
         if (GeneratorPythia8::generateEvent())
         {
-          genOk = CoalescencePythia8(mPythia.event, mPdgList, false, mCoalMomentum);
+          if (CoalescencePythia8(mPythia.event, mPdgList, false, mCoalMomentum)) {
+            for (int i = 0; i < mPythia.event.size(); i++) {
+              if (mPythia.event[i].isFinal() && std::find(mRequestedPdgList.begin(), mRequestedPdgList.end(), mPythia.event[i].id()) != mRequestedPdgList.end()) {
+                genOk = true;
+                break;
+              }
+            }
+          }
         }
         localCounter++;
       }
@@ -82,14 +95,15 @@ protected:
 
 
 private:
+  std::vector<int> mRequestedPdgList; /// signed pdg codes to be generated
   std::vector<unsigned int> mPdgList; /// list of pdg codes to be generated
   float mCoalMomentum = 0.4; /// coalescence momentum
   uint64_t mGeneratedEvents = 0; /// number of events generated so far
   int mInverseTriggerRatio = 1;  /// injection gap
 };
 
 ///___________________________________________________________
-FairGenerator *generateCoalescence(std::vector<unsigned int> pdgList, int input_trigger_ratio, double coal_momentum = 0.4)
+FairGenerator *generateCoalescence(std::vector<int> pdgList, int input_trigger_ratio, double coal_momentum = 0.4)
 {
   auto myGen = new GeneratorPythia8Coalescence(pdgList, input_trigger_ratio, coal_momentum);
   auto seed = (gRandom->TRandom::GetSeed() % 900000000);