add fastmtt producers

config.py

@@ -725,6 +725,7 @@ def build_config(
             pairselection.LVMu1Uncorrected,
             pairselection.LVTau2Uncorrected,
             pairquantities.MTDiTauPairQuantities,
+            # pairquantities.FastMTTQuantities,
             genparticles.MTGenDiTauPairQuantities,
             #  scalefactors.MuonIDIso_SF,
             scalefactors.Tau_2_VsJetTauID_lt_SF,
@@ -1142,6 +1143,10 @@ def build_config(
             q.jtag_value_2,
             q.mjj,
             q.m_vis,
+            # q.m_fastmtt,
+            # q.pt_fastmtt,
+            # q.eta_fastmtt,
+            # q.phi_fastmtt,
             q.deltaR_ditaupair,
             q.pt_vis,
             q.nbtag,

producers/pairquantities.py

@@ -571,6 +571,7 @@
     scopes=["em"],
     subproducers=[UnrollElLV1, UnrollMuLV2, m_vis, pt_vis, deltaR_ditaupair],
 )
+
 ## advanced event quantities (can be caluculated when ditau pair and met and all jets are determined)
 ## leptons: q.p4_1, q.p4_2
 ## met: met_p4_recoilcorrected
@@ -716,3 +717,135 @@
         jet_hemisphere,
     ],
 )
+
+p4_fastmtt_mt = Producer(
+    name="p4_fastmtt_mt",
+    call='quantities::p4_fastmtt({df}, {output}, {input}, "mt")',
+    input=[
+        q.pt_1,
+        q.pt_2,
+        q.eta_1,
+        q.eta_2,
+        q.phi_1,
+        q.phi_2,
+        q.mass_1,
+        q.mass_2,
+        q.met,
+        q.metphi,
+        q.metcov00,
+        q.metcov01,
+        q.metcov11,
+        q.decaymode_2,
+        q.decaymode_2,
+    ],
+    output=[q.p4_fastmtt],
+    scopes=["mt"],
+)
+p4_fastmtt_et = Producer(
+    name="p4_fastmtt_et",
+    call='quantities::p4_fastmtt({df}, {output}, {input}, "et")',
+    input=[
+        q.pt_1,
+        q.pt_2,
+        q.eta_1,
+        q.eta_2,
+        q.phi_1,
+        q.phi_2,
+        q.mass_1,
+        q.mass_2,
+        q.met,
+        q.metphi,
+        q.metcov00,
+        q.metcov01,
+        q.metcov11,
+        q.decaymode_2,
+        q.decaymode_2,
+    ],
+    output=[q.p4_fastmtt],
+    scopes=["et"],
+)
+p4_fastmtt_tt = Producer(
+    name="p4_fastmtt_tt",
+    call='quantities::p4_fastmtt({df}, {output}, {input}, "tt")',
+    input=[
+        q.pt_1,
+        q.pt_2,
+        q.eta_1,
+        q.eta_2,
+        q.phi_1,
+        q.phi_2,
+        q.mass_1,
+        q.mass_2,
+        q.met,
+        q.metphi,
+        q.metcov00,
+        q.metcov01,
+        q.metcov11,
+        q.decaymode_2,
+        q.decaymode_2,
+    ],
+    output=[q.p4_fastmtt],
+    scopes=["tt"],
+)
+p4_fastmtt_em = Producer(
+    name="p4_fastmtt_em",
+    call='quantities::p4_fastmtt({df}, {output}, {input}, "em")',
+    input=[
+        q.pt_1,
+        q.pt_2,
+        q.eta_1,
+        q.eta_2,
+        q.phi_1,
+        q.phi_2,
+        q.mass_1,
+        q.mass_2,
+        q.met,
+        q.metphi,
+        q.metcov00,
+        q.metcov01,
+        q.metcov11,
+    ],
+    output=[q.p4_fastmtt],
+    scopes=["em"],
+)
+pt_fastmtt = Producer(
+    name="pt_fastmtt",
+    call="quantities::pt({df}, {output}, {input})",
+    input=[q.p4_fastmtt],
+    output=[q.pt_fastmtt],
+    scopes=["mt", "et", "tt", "em"],
+)
+eta_fastmtt = Producer(
+    name="eta_fastmtt",
+    call="quantities::eta({df}, {output}, {input})",
+    input=[q.p4_fastmtt],
+    output=[q.eta_fastmtt],
+    scopes=["mt", "et", "tt", "em"],
+)
+phi_fastmtt = Producer(
+    name="phi_fastmtt",
+    call="quantities::phi({df}, {output}, {input})",
+    input=[q.p4_fastmtt],
+    output=[q.phi_fastmtt],
+    scopes=["mt", "et", "tt", "em"],
+)
+m_fastmtt = Producer(
+    name="m_fastmtt",
+    call="quantities::mass({df}, {output}, {input})",
+    input=[q.p4_fastmtt],
+    output=[q.m_fastmtt],
+    scopes=["mt", "et", "tt", "em"],
+)
+FastMTTQuantities = ProducerGroup(
+    name="FastMTTQuantities",
+    call=None,
+    input=None,
+    output=None,
+    scopes=["mt", "et", "tt", "em"],
+    subproducers={
+        "mt": [p4_fastmtt_mt, pt_fastmtt, eta_fastmtt, phi_fastmtt, m_fastmtt],
+        "et": [p4_fastmtt_et, pt_fastmtt, eta_fastmtt, phi_fastmtt, m_fastmtt],
+        "tt": [p4_fastmtt_tt, pt_fastmtt, eta_fastmtt, phi_fastmtt, m_fastmtt],
+        "em": [p4_fastmtt_em, pt_fastmtt, eta_fastmtt, phi_fastmtt, m_fastmtt],
+    },
+)

quantities/output.py

@@ -87,6 +87,11 @@
 
 # Combined event quantities
 m_vis = Quantity("m_vis")
+p4_fastmtt = Quantity("p4_fastmtt")
+m_fastmtt = Quantity("m_fastmtt")
+pt_fastmtt = Quantity("pt_fastmtt")
+eta_fastmtt = Quantity("eta_fastmtt")
+phi_fastmtt = Quantity("phi_fastmtt")
 pt_vis = Quantity("pt_vis")
 pzetamissvis = Quantity("pzetamissvis")
 mTdileptonMET = Quantity("mTdileptonMET")

tau_embedding_settings.py

@@ -83,23 +83,23 @@ def setup_embedding(configuration: Configuration, scopes: List[str]):
             "truegen_mother_pdgid": 23,
             "truegen_daughter_1_pdgid": 15,
             "truegen_daugher_2_pdgid": 15,
-        }
+        },
     )
     configuration.add_config_parameters(
         ["mm"],
         {
             "truegen_mother_pdgid": 23,
             "truegen_daughter_1_pdgid": 13,
             "truegen_daugher_2_pdgid": 13,
-        }
+        },
     )
     configuration.add_config_parameters(
         ["ee"],
         {
             "truegen_mother_pdgid": 23,
             "truegen_daughter_1_pdgid": 11,
             "truegen_daugher_2_pdgid": 11,
-        }
+        },
     )
 
     # add embedding selection scalefactors