[PWGDQ] Adding MCSignal for POWHEG Drell-Yan muon selection (#10817)

Co-authored-by: Sahil Upadhyaya <upadhyay@cern.ch>

Author: sahilupadhyaya92

PWGDQ/Core/MCProng.cxx

@@ -22,7 +22,8 @@ std::map<TString, int> MCProng::fgSourceNames = {
   {"kProducedInTransport", MCProng::kProducedInTransport},
   {"kProducedByGenerator", MCProng::kProducedByGenerator},
   {"kFromBackgroundEvent", MCProng::kFromBackgroundEvent},
-  {"kHEPMCFinalState", MCProng::kHEPMCFinalState}};
+  {"kHEPMCFinalState", MCProng::kHEPMCFinalState},
+  {"kIsPowhegDYMuon", MCProng::kIsPowhegDYMuon}};
 
 //________________________________________________________________________________________________________________
 MCProng::MCProng() : fNGenerations(0),

PWGDQ/Core/MCProng.h

@@ -73,6 +73,7 @@ class MCProng
     kProducedByGenerator, // Produced by generator (if not, then produced by GEANT)
     kFromBackgroundEvent, // Produced in the underlying event
     kHEPMCFinalState,     // HEPMC code 11
+    kIsPowhegDYMuon,      // POWHEG muons based on Pythia Status Code (=23) -> Drell-Yan signal
     kNSources
   };
 

PWGDQ/Core/MCSignal.h

@@ -253,6 +253,12 @@ bool MCSignal::CheckProng(int i, bool checkSources, const T& track)
             sourcesDecision |= (static_cast<uint64_t>(1) << MCProng::kHEPMCFinalState);
           }
         }
+        // Check kIsPowhegDYMuon
+        if (fProngs[i].fSourceBits[j] & (static_cast<uint64_t>(1) << MCProng::kIsPowhegDYMuon)) {
+          if ((fProngs[i].fExcludeSource[j] & (static_cast<uint64_t>(1) << MCProng::kIsPowhegDYMuon)) != (currentMCParticle.getGenStatusCode() == 23)) {
+            sourcesDecision |= (static_cast<uint64_t>(1) << MCProng::kIsPowhegDYMuon);
+          }
+        }
       } // end if(hasSources)
       // no source bit is fulfilled
       if (hasSources && !sourcesDecision) {

PWGDQ/Core/MCSignalLibrary.cxx

@@ -348,6 +348,12 @@ MCSignal* o2::aod::dqmcsignals::GetMCSignal(const char* name)
     signal = new MCSignal(name, "electron from a photon conversion", {prong}, {-1});
     return signal;
   }
+  if (!nameStr.compare("PowhegDYMuon1")) {
+    MCProng prong(1, {13}, {true}, {false}, {0}, {0}, {false});        // selecting muons
+    prong.SetSourceBit(0, MCProng::kIsPowhegDYMuon);                   // set source to be Muon from POWHEG
+    signal = new MCSignal(name, "POWHEG Muon singles", {prong}, {-1}); // define a signal with 1-prong
+    return signal;
+  }
 
   // 2-prong signals
   if (!nameStr.compare("dielectron")) {
@@ -381,6 +387,12 @@ MCSignal* o2::aod::dqmcsignals::GetMCSignal(const char* name)
     signal = new MCSignal(name, "dielectron from a photon conversion from a pi0", {prong, prong}, {1, 1});
     return signal;
   }
+  if (!nameStr.compare("PowhegDYMuon2")) {
+    MCProng prong(1, {13}, {true}, {false}, {0}, {0}, {false});                // selecting muons
+    prong.SetSourceBit(0, MCProng::kIsPowhegDYMuon);                           // set source to be Muon from POWHEG
+    signal = new MCSignal(name, "POWHEG Muon pair", {prong, prong}, {-1, -1}); // define a signal with 2-prong
+    return signal;
+  }
 
   // LMEE single signals
   // electron signals with mother X: e from mother X