From 17c8121456730a6a004437a63c088158a432b36e Mon Sep 17 00:00:00 2001 From: Devin Date: Wed, 12 Aug 2026 21:05:53 +0200 Subject: [PATCH 1/2] Add electron super cluster eta to flaf --- Common/BaselineSelection.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/Common/BaselineSelection.py b/Common/BaselineSelection.py index 0131940b2..21e9a9735 100644 --- a/Common/BaselineSelection.py +++ b/Common/BaselineSelection.py @@ -200,6 +200,12 @@ def CreateRecoP4(df, suffix="nano", nano_version="v12"): f"{obj}_p4_bsConstrainedPt", f"GetP4({obj}_bsConstrainedPt, {obj}_eta, {obj}_phi, {obj}_mass, {obj}_idx)", ) + if obj == "Electron": + if f"{obj}_superclusterEta" not in df.GetColumnNames(): + df = df.Define( + f"{obj}_superclusterEta", + "RVecF ele_SC_eta; for(size_t i = 0 ; i < Electron_eta.size(); i++) {{ele_SC_eta.push_back(Electron_deltaEtaSC[i]+Electron_eta[i]);}} return ele_SC_eta;", + ) return df From 7e988d5a19ad9ae1d6d2c2846af43eef0cc6175e Mon Sep 17 00:00:00 2001 From: Devin Date: Wed, 12 Aug 2026 22:46:50 +0200 Subject: [PATCH 2/2] Drop RVecF loop and just add columns --- Common/BaselineSelection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Common/BaselineSelection.py b/Common/BaselineSelection.py index 21e9a9735..b21bbedac 100644 --- a/Common/BaselineSelection.py +++ b/Common/BaselineSelection.py @@ -204,7 +204,7 @@ def CreateRecoP4(df, suffix="nano", nano_version="v12"): if f"{obj}_superclusterEta" not in df.GetColumnNames(): df = df.Define( f"{obj}_superclusterEta", - "RVecF ele_SC_eta; for(size_t i = 0 ; i < Electron_eta.size(); i++) {{ele_SC_eta.push_back(Electron_deltaEtaSC[i]+Electron_eta[i]);}} return ele_SC_eta;", + f"Electron_eta + Electron_deltaEtaSC", ) return df