Adding DY_Z_Y to process_options.py

validphys2/src/validphys/process_options.py

@@ -227,6 +227,17 @@ def _dptboson_xq2map(kin_dict):
     return x, q2
 
 
+def _dptboson_xq2map(kin_dict):
+    """Compute x and q2 mapping for DY Z -> ll + jet process.
+    Here pT refers to the transverse momentum of the Z boson."""
+    pT = kin_dict[_Vars.pT]
+    m_Z2 = kin_dict[_Vars.m_Z2]
+    sqrts = kin_dict[_Vars.sqrts]
+    x = np.sqrt(m_Z2 + pT * pT) / sqrts
+    q2 = m_Z2 + pT * pT
+    return x, q2
+
+
 DIS = _Process(
     "DIS",
     "Deep Inelastic Scattering",
@@ -277,6 +288,13 @@ def _dptboson_xq2map(kin_dict):
     xq2map_function=_displusjet_xq2map,
 )
 
+DY_ZJ_PT = _Process(
+    "DY_ZJ_PT",
+    "DY Z -> ll + j Z boson transverse momentum",
+    accepted_variables=(_Vars.pT, _Vars.m_Z2, _Vars.sqrts),
+    xq2map_function=_dptboson_xq2map,
+)
+
 
 DY_W_ETA = _Process(
     "DY_W_ETA",
@@ -306,7 +324,11 @@ def _dptboson_xq2map(kin_dict):
     "HQP_PTQ": HQP_PTQ,
     "HERAJET": HERAJET,
     "HERADIJET": dataclasses.replace(HERAJET, name="HERADIJET", description="DIS + jj production"),
+<<<<<<< HEAD
     "DY_W_ETA": DY_W_ETA,
+=======
+    "DY_Z_Y": dataclasses.replace(DY_W_ETA, name="DY_Z_Y", description="DY Z -> ll rapidity"),
+>>>>>>> 7c41d4d4ec06df373abe8fa604ad2475d85220f6
     "DY_ZJ_PT": DY_ZJ_PT
 }