To Activate the Kinematics cuts on the gen MC

Author: navkumar123

PWGLF/Tasks/Resonances/chargedkstaranalysis.cxx

@@ -122,6 +122,9 @@ struct Chargedkstaranalysis {
     // Other cuts on Ks
     Configurable<int> rotationalCut{"rotationalCut", 6, "Cut value (Rotation angle pi - pi/cut and pi + pi/cut)"};
 
+    // for Kinematics checks
+    Configurable<bool> genKinematicsChecks{"genKinematicsChecks", false, "Activate the Kinematics cuts on the gen MC"};
+
     // fixed variables
     float rapidityMotherData = 0.5;
     float beamEnergy = 13600.0;
@@ -1442,24 +1445,46 @@ struct Chargedkstaranalysis {
       const int pionWanted = (part.pdgCode() > 0) ? +kPiPlus : -kPiPlus;
       bool hasRightPion = false;
       bool hasK0sToPipi = false;
+      bool passDauAcceptance = false;
       for (const auto& d1 : part.template daughters_as<aod::McParticles>()) {
         const int pdg1 = d1.pdgCode();
         if (pdg1 == pionWanted) {
           lDecayDaughter_bach = LorentzVectorSetXYZM(d1.px(), d1.py(), d1.pz(), MassPionCharged);
+          if (helicityCfgs.genKinematicsChecks) {
+            if (!((lDecayDaughter_bach.pt() > trackCutCfgs.cMinPtcut) && (std::abs(lDecayDaughter_bach.eta()) < trackCutCfgs.cMaxEtacut))) {
+              continue;
+            }
+          }
+          passDauAcceptance = (lDecayDaughter_bach.pt() > trackCutCfgs.cMinPtcut) && (std::abs(lDecayDaughter_bach.eta()) < trackCutCfgs.cMaxEtacut);
           hasRightPion = true;
         } else if (std::abs(pdg1) == kPDGK0) {
           for (const auto& d2 : d1.template daughters_as<aod::McParticles>()) {
             if (std::abs(d2.pdgCode()) == kPDGK0s) {
               bool seenPip = false, seenPim = false;
               for (const auto& d3 : d2.template daughters_as<aod::McParticles>()) {
                 if (d3.pdgCode() == +kPiPlus) {
+                  if (helicityCfgs.genKinematicsChecks) {
+                    if (!((d3.pt() > trackCutCfgs.cMinPtcut) && (std::abs(d3.eta()) < trackCutCfgs.cMaxEtacut))) {
+                      continue;
+                    }
+                  }
                   seenPip = true;
                 } else if (d3.pdgCode() == -kPiPlus) {
+                  if (helicityCfgs.genKinematicsChecks) {
+                    if (!((d3.pt() > trackCutCfgs.cMinPtcut) && (std::abs(d3.eta()) < trackCutCfgs.cMaxEtacut))) {
+                      continue;
+                    }
+                  }
                   seenPim = true;
                 }
               }
               if (seenPip && seenPim) {
                 lResoSecondary = LorentzVectorSetXYZM(d2.px(), d2.py(), d2.pz(), MassK0Short);
+                if (helicityCfgs.genKinematicsChecks) {
+                  if (!((d2.pt() > secondaryCutsCfgs.cSecondaryPtMin) && (std::abs(d2.eta()) < secondaryCutsCfgs.cSecondaryRapidityMax))) {
+                    continue;
+                  }
+                }
                 hasK0sToPipi = true;
                 break;
               }