include check for default lorentzvectors in LV combination

KIT-CMS · Jul 26, 2024 · 51a34c2 · 51a34c2
1 parent f55d433
commit 51a34c2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/include/lorentzvectors.hxx b/include/lorentzvectors.hxx
@@ -5,6 +5,7 @@
 #include "ROOT/RDataFrame.hxx"
 #include "ROOT/RVec.hxx"
 #include "utility/utility.hxx"
+#include "defaults.hxx"
 #include <Math/Vector4D.h>
 
 namespace lorentzvectors {
@@ -39,6 +40,12 @@ ROOT::RDF::RNode CombineP4s(ROOT::RDF::RNode df, const std::string &outputflag,
         outputflag,
         utility::PassAsVec<np4s, ROOT::Math::PtEtaPhiMVector>(
             [](const ROOT::RVec<ROOT::Math::PtEtaPhiMVector> &p4s) {
+                for (int i = 0; i < p4s.size(); i++) {
+                    auto p4 = p4s.at(i);
+                    if (p4.pt() < 0.) {
+                        return default_lorentzvector;
+                    }
+                }
                 return Sum(p4s, ROOT::Math::PtEtaPhiMVector());
             }),
         P4List);
@@ -57,4 +64,4 @@ ROOT::RDF::RNode build(ROOT::RDF::RNode df, const std::string &pairname,
 } // namespace mutau
 
 } // namespace lorentzvectors
-#endif /* GUARDLVECS_H */
+#endif /* GUARDLVECS_H */