From e9d4b52036c5cadab706784186b637c7739f22d0 Mon Sep 17 00:00:00 2001
From: nzardosh <nima.zardoshti@cern.ch>
Date: Thu, 18 Jul 2024 15:49:04 +0200
Subject: [PATCH] PWGJE: Adding constituent pair pt cut

---
 PWGJE/Tasks/jetsubstructure.cxx   | 5 ++++-
 PWGJE/Tasks/jetsubstructurehf.cxx | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/PWGJE/Tasks/jetsubstructure.cxx b/PWGJE/Tasks/jetsubstructure.cxx
index c4398703437..bd564e4bc65 100644
--- a/PWGJE/Tasks/jetsubstructure.cxx
+++ b/PWGJE/Tasks/jetsubstructure.cxx
@@ -49,6 +49,7 @@ struct JetSubstructureTask {
 
   Configurable<float> zCut{"zCut", 0.1, "soft drop z cut"};
   Configurable<float> beta{"beta", 0.0, "soft drop beta"};
+  Configurable<float> pairConstituentPtMin{"pairConstituentPtMin", 1.0, "pt cut off for constituents going into pairs"};
 
   Service<o2::framework::O2DatabasePDG> pdg;
   std::vector<fastjet::PseudoJet> jetConstituents;
@@ -154,7 +155,9 @@ struct JetSubstructureTask {
     pairThetaVec.clear();
     std::vector<typename U::iterator> tracksVec;
     for (auto const& constituent : jet.template tracks_as<U>()) {
-      tracksVec.push_back(constituent);
+      if (constituent.pt() >= pairConstituentPtMin) {
+        tracksVec.push_back(constituent);
+      }
     }
     if (tracksVec.size() >= 2) {
       for (typename std::vector<typename U::iterator>::size_type track1Index = 0; track1Index < tracksVec.size() - 1; track1Index++) {
diff --git a/PWGJE/Tasks/jetsubstructurehf.cxx b/PWGJE/Tasks/jetsubstructurehf.cxx
index 9cdb933c20a..07eb5b9120b 100644
--- a/PWGJE/Tasks/jetsubstructurehf.cxx
+++ b/PWGJE/Tasks/jetsubstructurehf.cxx
@@ -53,6 +53,7 @@ struct JetSubstructureHFTask {
   // Jet level configurables
   Configurable<float> zCut{"zCut", 0.1, "soft drop z cut"};
   Configurable<float> beta{"beta", 0.0, "soft drop beta"};
+  Configurable<float> pairConstituentPtMin{"pairConstituentPtMin", 1.0, "pt cut off for constituents going into pairs"};
 
   Service<o2::framework::O2DatabasePDG> pdg;
   float candMass;
@@ -168,7 +169,9 @@ struct JetSubstructureHFTask {
     std::vector<std::decay_t<typename U::iterator>> tracksVec;
     std::vector<std::decay_t<typename V::iterator>> candidatesVec;
     for (auto& constituent : jet.template tracks_as<U>()) {
-      tracksVec.push_back(constituent);
+      if (constituent.pt() >= pairConstituentPtMin) {
+        tracksVec.push_back(constituent);
+      }
     }
     for (auto& candidate : jet.template candidates_as<V>()) {
       candidatesVec.push_back(candidate);