added two new variables for ML use

Author: Oussama3030

PWGLF/Tasks/Resonances/k892hadronphotonBkg.cxx

@@ -341,6 +341,8 @@ struct k892hadronphotonBkg {
       histos.add("BDT/h2dPhotonQt", "h2dPhotonQt", kTH2D, {axisConfig.mlProb, axisConfig.axisV0APQt});
       histos.add("BDT/h2dPhotonRadius", "h2dPhotonRadius", kTH2D, {axisConfig.mlProb, axisConfig.axisV0Radius});
       histos.add("BDT/h2dOPAngle", "h2dOPAngle", kTH2D, {axisConfig.mlProb, axisConfig.axisOPAngle});
+      histos.add("BDT/h2dAPAlpha", "h2dAPAlpha", kTH2D, {axisConfig.mlProb, axisConfig.axisAPAlpha});
+      histos.add("BDT/h2dAPQt", "h2dAPQt", kTH2D, {axisConfig.mlProb, axisConfig.axisAPQt});
     }
 
     histos.add("hEventCentrality", "hEventCentrality", kTH1D, {axisConfig.axisCentrality});
@@ -894,7 +896,7 @@ struct k892hadronphotonBkg {
   //_______________________________________________
   // Fill BDT performance QA
   template <typename TV0Object>
-  void fillBDTPerformance(TV0Object const& lambda, TV0Object const& photon, float openAngle, float score, float pt, float mass)
+  void fillBDTPerformance(TV0Object const& lambda, TV0Object const& photon, float openAngle, float apAlpha, float apQt, float score, float pt, float mass)
   {
     float bkgScore = 1.0f - score;
 
@@ -924,24 +926,26 @@ struct k892hadronphotonBkg {
     histos.fill(HIST("BDT/h2dPhotonQt"), score, photon.qtarm());
     histos.fill(HIST("BDT/h2dPhotonRadius"), score, photon.v0radius());
     histos.fill(HIST("BDT/h2dOPAngle"), score, openAngle);
+    histos.fill(HIST("BDT/h2dAPAlpha"), score, apAlpha);
+    histos.fill(HIST("BDT/h2dAPQt"), score, apQt);
   }
 
   //_______________________________________________
   // BDT selection of a Lambda + photon pair
   template <typename TV0Object>
   bool selectML(TV0Object const& lambda, TV0Object const& photon,
-                float openAngle, float pt, float mass)
+                float openAngle, float apAlpha, float apQt, float pt, float mass)
   {
     // No model outside the bdt.ptBinEdges range
     if (pt < bdt.ptBinEdges.value.front() || pt >= bdt.ptBinEdges.value.back())
       return false;
 
     // Features in the order of bdt.namesInputFeatures
-    auto inputFeatures = mlResponse.getInputFeatures(lambda, photon, openAngle);
+    auto inputFeatures = mlResponse.getInputFeatures(lambda, photon, openAngle, apAlpha, apQt);
     std::vector<float> outputMl;
     const bool isSelected = mlResponse.isSelectedMl(inputFeatures, pt, outputMl); // model and cut of the pT bin
 
-    fillBDTPerformance(lambda, photon, openAngle, outputMl[1], pt, mass);
+    fillBDTPerformance(lambda, photon, openAngle, apAlpha, apQt, outputMl[1], pt, mass);
 
     return isSelected;
   }
@@ -1103,20 +1107,20 @@ struct k892hadronphotonBkg {
         if (std::abs(rapidity) > maxRap)
           continue;
 
+        // Armenteros-Podolanski of the mixed pair
+        const std::array<float, 3> gammaMom{photon.px(), photon.py(), photon.pz()};
+        const std::array<float, 3> hadronMom{hadron.px(), hadron.py(), hadron.pz()};
+        const float apAlpha = armenterosAlpha(gammaMom, hadronMom);
+        const float apQt = armenterosQt(gammaMom, hadronMom);
+
         // BDT selection (Lambda(1520) only)
         if constexpr (resonance == kResoLambdaStar) {
           if (bdt.enableML) {
-            if (!selectML(hadron, photon, openAngle, pt, mass))
+            if (!selectML(hadron, photon, openAngle, apAlpha, apQt, pt, mass))
               continue;
           }
         }
 
-        // Armenteros-Podolanski of the mixed pair
-        const std::array<float, 3> gammaMom{photon.px(), photon.py(), photon.pz()};
-        const std::array<float, 3> hadronMom{hadron.px(), hadron.py(), hadron.pz()};
-        const float apAlpha = armenterosAlpha(gammaMom, hadronMom);
-        const float apQt = armenterosQt(gammaMom, hadronMom);
-
         if constexpr (resonance == kResoKStar) {
           histos.fill(HIST("KStarBkg/h2dMixedKStarMassVsPt"), mass, pt);
           histos.fill(HIST("KStarBkg/h3dMixedKStarMassVsPt"), centrality, pt, mass);

PWGLF/Tasks/Strangeness/sigmaanalysis.cxx

@@ -398,6 +398,8 @@ struct sigmaanalysis {
       histos.add("BDT/h2dPhotonQt", "h2dPhotonQt", kTH2D, {mlProb, axisAPQt});
       histos.add("BDT/h2dPhotonRadius", "h2dPhotonRadius", kTH2D, {mlProb, axisV0Radius});
       histos.add("BDT/h2dOPAngle", "h2dOPAngle", kTH2D, {mlProb, axisOPAngle});
+      histos.add("BDT/h2dAPAlpha", "h2dAPAlpha", kTH2D, {mlProb, axisAPAlpha});
+      histos.add("BDT/h2dAPQt", "h2dAPQt", kTH2D, {mlProb, axisAPQt});
 
       // MC-truth-based score
       if (doprocessMonteCarlo) {
@@ -1773,6 +1775,8 @@ struct sigmaanalysis {
     histos.fill(HIST("BDT/h2dPhotonQt"), score, cand.photonQt());
     histos.fill(HIST("BDT/h2dPhotonRadius"), score, cand.photonRadius());
     histos.fill(HIST("BDT/h2dOPAngle"), score, cand.opAngle());
+    histos.fill(HIST("BDT/h2dAPAlpha"), score, cand.lStarAlpha());
+    histos.fill(HIST("BDT/h2dAPQt"), score, cand.lStarQtarm());
 
     // MC-truth-based separation (signal = particle + antiparticle)
     if constexpr (requires { cand.isSigma0(); cand.isLambdaStar(); }) {
@@ -1796,7 +1800,7 @@ struct sigmaanalysis {
       return false;
 
     // Features in the order of bdt.namesInputFeatures
-    auto inputFeatures = mlResponse.getInputFeatures(cand, cand, cand.opAngle());
+    auto inputFeatures = mlResponse.getInputFeatures(cand, cand, cand.opAngle(), cand.lStarAlpha(), cand.lStarQtarm());
     std::vector<float> outputMl;                                                  // [background, signal]
     const bool isSelected = mlResponse.isSelectedMl(inputFeatures, pt, outputMl); // model and cut of the pT bin
 

PWGLF/Utils/ResonanceMlResponse.h

@@ -60,7 +60,9 @@ enum class InputFeaturesResonance : uint8_t {
   photonPosTPCNSigmaEl,
   photonNegTPCNSigmaEl,
   // Photon-Lambda pair
-  opAngle
+  opAngle,
+  apAlpha,
+  apQt
 };
 
 template <typename TypeOutputScore = float>
@@ -72,7 +74,7 @@ class ResonanceMlResponse : public MlResponse<TypeOutputScore>
 
   /// Input features of a photon-Lambda pair
   template <typename TDauTracks = o2::soa::Join<o2::aod::DauTrackExtras, o2::aod::DauTrackTPCPIDs>, typename TLambda, typename TPhoton>
-  std::vector<float> getInputFeatures(TLambda const& lambda, TPhoton const& photon, float opAngle)
+  std::vector<float> getInputFeatures(TLambda const& lambda, TPhoton const& photon, float opAngle, float apAlpha, float apQt)
   {
     // sigma0builder candidates carry prefixed columns (lambdaQt(), photonQt(), ...), derived V0s the plain V0 getters
     constexpr bool LambdaFromSigma0 = requires(TLambda const& cand) { cand.lambdaQt(); };
@@ -271,6 +273,14 @@ class ResonanceMlResponse : public MlResponse<TypeOutputScore>
         case InputFeaturesResonance::opAngle:
           inputFeatures.emplace_back(opAngle);
           break;
+
+        case InputFeaturesResonance::apAlpha:
+          inputFeatures.emplace_back(apAlpha);
+          break;
+
+        case InputFeaturesResonance::apQt:
+          inputFeatures.emplace_back(apQt);
+          break;
       }
     }
     return inputFeatures;
@@ -307,7 +317,9 @@ class ResonanceMlResponse : public MlResponse<TypeOutputScore>
       {"photonNegEta", static_cast<uint8_t>(InputFeaturesResonance::photonNegEta)},
       {"photonPosTPCNSigmaEl", static_cast<uint8_t>(InputFeaturesResonance::photonPosTPCNSigmaEl)},
       {"photonNegTPCNSigmaEl", static_cast<uint8_t>(InputFeaturesResonance::photonNegTPCNSigmaEl)},
-      {"opAngle", static_cast<uint8_t>(InputFeaturesResonance::opAngle)}};
+      {"opAngle", static_cast<uint8_t>(InputFeaturesResonance::opAngle)},
+      {"apAlpha", static_cast<uint8_t>(InputFeaturesResonance::apAlpha)},
+      {"apQt", static_cast<uint8_t>(InputFeaturesResonance::apQt)}};
   }
 };