Skip to content

Commit

Permalink
Replaced 4F by Olq3
Browse files Browse the repository at this point in the history
  • Loading branch information
@ElieHammou
ElieHammou committed Oct 23, 2024
1 parent 60fd649 commit 0d3e636
Showing 1 changed file with 93 additions and 89 deletions.
182 changes: 93 additions & 89 deletions src/yadism/coefficient_functions/coupling_constants.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,20 +49,22 @@ def __init__(self, theory_config, obs_config):

# BSM couplings -------------------------------------------------------
# Temporary couplings to Olq3
self.BSM_coupling_vectorial = {21: 0}
self.BSM_coupling_axial = {21: 0}
self.Olq3_coupling_vectorial = {21: 0}
self.Olq3_coupling_axial = {21: 0}
for q in range(1, 7):
self.BSM_coupling_vectorial[q] = -1 if q % 2 == 0 else 1 # u if stmt else d
self.BSM_coupling_axial[q] = -1 if q % 2 == 0 else 1 # u if stmt else d
self.Olq3_coupling_vectorial[q] = (
-1 if q % 2 == 0 else 1
) # u if stmt else d
self.Olq3_coupling_axial[q] = -1 if q % 2 == 0 else 1 # u if stmt else d
# leptons: 11 = e-(!)
for pid in [11, 13, 15]:
self.BSM_coupling_vectorial[pid] = 1
self.BSM_coupling_axial[pid] = 1
self.Olq3_coupling_vectorial[pid] = 1
self.Olq3_coupling_axial[pid] = 1
# neutrinos
# CC not yet implemented
for pid in [12, 14, 16]:
self.BSM_coupling_vectorial[pid] = 0
self.BSM_coupling_axial[pid] = 0
self.Olq3_coupling_vectorial[pid] = 0
self.Olq3_coupling_axial[pid] = 0

self.log()

Expand Down Expand Up @@ -108,18 +110,18 @@ def leptonic_coupling(self, mode, quark_coupling_type):
):
pol *= -1
# load Z coupling
projectile_v = 0.0
projectile_a = 0.0
projectile_BSM_v = 0.0
projectile_BSM_a = 0.0
if mode in ["phZ", "ZZ", "Z4F"]:
projectile_v = self.vectorial_coupling(abs(projectile_pid))
projectile_a = self.weak_isospin_3[abs(projectile_pid)]
projectile_BSM_v = self.BSM_coupling_vectorial[abs(projectile_pid)]
projectile_BSM_a = self.BSM_coupling_axial[abs(projectile_pid)]
if mode in ["ph4F"]:
projectile_BSM_v = self.BSM_coupling_vectorial[abs(projectile_pid)]
projectile_BSM_a = self.BSM_coupling_axial[abs(projectile_pid)]
projectile_Z_v = 0.0
projectile_Z_a = 0.0
projectile_Olq3_v = 0.0
projectile_Olq3_a = 0.0
if mode in ["phZ", "ZZ", "Z0lq3"]:
projectile_Z_v = self.vectorial_coupling(abs(projectile_pid))
projectile_Z_a = self.weak_isospin_3[abs(projectile_pid)]
projectile_Olq3_v = self.Olq3_coupling_vectorial[abs(projectile_pid)]
projectile_Olq3_a = self.Olq3_coupling_axial[abs(projectile_pid)]
if mode in ["ph0lq3"]:
projectile_Olq3_v = self.Olq3_coupling_vectorial[abs(projectile_pid)]
projectile_Olq3_a = self.Olq3_coupling_axial[abs(projectile_pid)]
# switch mode
if mode == "phph":
if quark_coupling_type in ["VV", "AA"]:
Expand All @@ -129,51 +131,51 @@ def leptonic_coupling(self, mode, quark_coupling_type):
elif mode == "phZ":
if quark_coupling_type in ["VV", "AA"]:
return self.electric_charge[abs(projectile_pid)] * (
projectile_v + pol * projectile_a
projectile_Z_v + pol * projectile_Z_a
)
else:
return self.electric_charge[abs(projectile_pid)] * (
projectile_a + pol * projectile_v
projectile_Z_a + pol * projectile_Z_v
)
elif mode == "ZZ":
if quark_coupling_type in ["VV", "AA"]:
return (
projectile_v**2
+ projectile_a**2
+ 2.0 * pol * projectile_v * projectile_a
projectile_Z_v**2
+ projectile_Z_a**2
+ 2.0 * pol * projectile_Z_v * projectile_Z_a
)
else:
return 2.0 * projectile_v * projectile_a + pol * (
projectile_v**2 + projectile_a**2
return 2.0 * projectile_Z_v * projectile_Z_a + pol * (
projectile_Z_v**2 + projectile_Z_a**2
)
elif mode == "ph4F":
elif mode == "ph0lq3":
if quark_coupling_type in ["VV", "AA"]:
return self.electric_charge[abs(projectile_pid)] * (
projectile_BSM_v + pol * projectile_BSM_a
projectile_Olq3_v + pol * projectile_Olq3_a
)
else:
return self.electric_charge[abs(projectile_pid)] * (
projectile_BSM_a + pol * projectile_BSM_v
projectile_Olq3_a + pol * projectile_Olq3_v
)
elif mode == "Z4F":
elif mode == "Z0lq3":
if quark_coupling_type in ["VV", "AA"]:
return (
projectile_v * projectile_BSM_v
+ projectile_a * projectile_BSM_a
projectile_Z_v * projectile_Olq3_v
+ projectile_Z_a * projectile_Olq3_a
+ pol
* (
projectile_v * projectile_BSM_a
+ projectile_BSM_v * projectile_a
projectile_Z_v * projectile_Olq3_a
+ projectile_Olq3_v * projectile_Z_a
)
)
else:
return (
projectile_v * projectile_BSM_a
+ projectile_BSM_v * projectile_a
projectile_Z_v * projectile_Olq3_a
+ projectile_Olq3_v * projectile_Z_a
+ pol
* (
projectile_v * projectile_BSM_v
+ projectile_a * projectile_BSM_a
projectile_Z_v * projectile_Olq3_v
+ projectile_Z_a * projectile_Olq3_a
)
)
raise ValueError(f"Unknown mode: {mode}")
Expand All @@ -186,11 +188,11 @@ def nc_partonic_coupling(self, pid):
}
qZ = {"V": self.vectorial_coupling(pid), "A": self.weak_isospin_3[pid]}

q4F = {
"V": self.BSM_coupling_vectorial[pid],
"A": self.BSM_coupling_axial[pid],
qOlq3 = {
"V": self.Olq3_coupling_vectorial[pid],
"A": self.Olq3_coupling_axial[pid],
}
return qph, qZ, q4F
return qph, qZ, qOlq3

def partonic_coupling(self, mode, pid, quark_coupling_type, cc_mask=None):
"""Compute the coupling of the boson to the parton.
Expand All @@ -217,7 +219,7 @@ def partonic_coupling(self, mode, pid, quark_coupling_type, cc_mask=None):
if mode == "WW":
return np.sum(self.theory_config["CKM"].masked(cc_mask)(pid))
# load couplings
qph, qZ, q4F = self.nc_partonic_coupling(pid)
qph, qZ, qOlq3 = self.nc_partonic_coupling(pid)

first = quark_coupling_type[0]
second = quark_coupling_type[1]
Expand All @@ -228,10 +230,10 @@ def partonic_coupling(self, mode, pid, quark_coupling_type, cc_mask=None):
return qph[first] * qZ[second]
if mode == "ZZ":
return qZ[first] * qZ[second]
if mode == "ph4F":
return qph[first] * q4F[second]
if mode == "Z4F":
return qZ[first] * q4F[second]
if mode == "ph0lq3":
return qph[first] * qOlq3[second]
if mode == "Z0lq3":
return qZ[first] * qOlq3[second]
raise ValueError(f"Unknown mode: {mode}")

def partonic_coupling_fl11(self, mode, pid, nf, quark_coupling_type):
Expand Down Expand Up @@ -273,13 +275,13 @@ def partonic_coupling_fl11(self, mode, pid, nf, quark_coupling_type):
second = quark_coupling_type[1]

def switch_mode(quark, coupling_type):
qph, qZ, q4F = self.nc_partonic_coupling(quark)
if mode in ["phph", "Zph", "4Fph"]:
qph, qZ, qOlq3 = self.nc_partonic_coupling(quark)
if mode in ["phph", "Zph", "Olq3ph"]:
return qph[coupling_type]
if mode in ["phZ", "ZZ", "4FZ"]:
if mode in ["phZ", "ZZ", "Olq3Z"]:
return qZ[coupling_type]
if mode in ["ph4F", "Z4F"]:
return q4F[coupling_type]
if mode in ["ph0lq3", "Z0lq3"]:
return qOlq3[coupling_type]

# first coupling contribute with a mean
pids = range(1, nf + 1)
Expand Down Expand Up @@ -314,20 +316,20 @@ def propagator_factor(self, mode, Q2):
eta_phZ /= 1 - self.obs_config["propagatorCorrection"]

# Need to specify Wilson coefficient?
C_4F = 1 / 4
C_Olq3 = 1 / 4
# Modify with more precise value
alpha = 1 / 137
# Should get it from param card
BSM_scale = 1000
eta_ph4F = ((C_4F) / (4 * np.pi * alpha)) * (Q2 / BSM_scale**2)
eta_phOlq3 = ((C_Olq3) / (4 * np.pi * alpha)) * (Q2 / BSM_scale**2)
if mode == "phZ":
return eta_phZ
if mode == "ZZ":
return eta_phZ**2
if mode == "ph4F":
return eta_ph4F
if mode == "Z4F":
return eta_phZ * eta_ph4F
if mode == "ph0lq3":
return eta_phOlq3
if mode == "Z0lq3":
return eta_phZ * eta_phOlq3
if mode == "WW":
eta_W = (
(eta_phZ / 2)
Expand Down Expand Up @@ -404,21 +406,21 @@ def get_weight(self, pid, Q2, quark_coupling_type, cc_mask=None):
* self.propagator_factor("ZZ", Q2)
* self.partonic_coupling("ZZ", pid, quark_coupling_type)
)
# photon-4F interference
w_ph4F = (
# photon-Olq3 interference
w_ph0lq3 = (
2
* self.leptonic_coupling("ph4F", quark_coupling_type)
* self.propagator_factor("ph4F", Q2)
* self.partonic_coupling("ph4F", pid, quark_coupling_type)
* self.leptonic_coupling("ph0lq3", quark_coupling_type)
* self.propagator_factor("ph0lq3", Q2)
* self.partonic_coupling("ph0lq3", pid, quark_coupling_type)
)
# Z-4F interference
w_Z4F = (
# Z-Olq3 interference
w_Z0lq3 = (
2
* self.leptonic_coupling("Z4F", quark_coupling_type)
* self.propagator_factor("Z4F", Q2)
* self.partonic_coupling("Z4F", pid, quark_coupling_type)
* self.leptonic_coupling("Z0lq3", quark_coupling_type)
* self.propagator_factor("Z0lq3", Q2)
* self.partonic_coupling("Z0lq3", pid, quark_coupling_type)
)
return w_phph + w_phZ + w_ZZ + w_ph4F + w_Z4F
return w_phph + w_phZ + w_ZZ + w_ph0lq3 + w_Z0lq3
raise ValueError(f"Unknown process: {self.obs_config['process']}")

def get_fl11_weight(self, pid, Q2, nf, quark_coupling_type):
Expand Down Expand Up @@ -484,29 +486,31 @@ def get_fl11_weight(self, pid, Q2, nf, quark_coupling_type):
* self.propagator_factor("ZZ", Q2)
* self.partonic_coupling_fl11("ZZ", pid, nf, quark_coupling_type)
)
# photon-4F interference
w_ph4F = (
self.leptonic_coupling("ph4F", quark_coupling_type)
* self.propagator_factor("ph4F", Q2)
* self.partonic_coupling("ph4F", pid, quark_coupling_type)
# photon-Olq3 interference
w_ph0lq3 = (
self.leptonic_coupling("ph0lq3", quark_coupling_type)
* self.propagator_factor("ph0lq3", Q2)
* self.partonic_coupling("ph0lq3", pid, quark_coupling_type)
)
w_4Fph = (
self.leptonic_coupling("ph4F", quark_coupling_type)
* self.propagator_factor("ph4F", Q2)
* self.partonic_coupling("ph4F", pid, quark_coupling_type)
w_Olq3ph = (
self.leptonic_coupling("ph0lq3", quark_coupling_type)
* self.propagator_factor("ph0lq3", Q2)
* self.partonic_coupling("ph0lq3", pid, quark_coupling_type)
)
# Z-4F interference
w_Z4F = (
self.leptonic_coupling("Z4F", quark_coupling_type)
* self.propagator_factor("Z4F", Q2)
* self.partonic_coupling("Z4F", pid, quark_coupling_type)
# Z-Olq3 interference
w_Z0lq3 = (
self.leptonic_coupling("Z0lq3", quark_coupling_type)
* self.propagator_factor("Z0lq3", Q2)
* self.partonic_coupling("Z0lq3", pid, quark_coupling_type)
)
w_4FZ = (
self.leptonic_coupling("Z4F", quark_coupling_type)
* self.propagator_factor("Z4F", Q2)
* self.partonic_coupling("Z4F", pid, quark_coupling_type)
w_Olq3Z = (
self.leptonic_coupling("Z0lq3", quark_coupling_type)
* self.propagator_factor("Z0lq3", Q2)
* self.partonic_coupling("Z0lq3", pid, quark_coupling_type)
)
return (
w_phph + w_phZ + w_Zph + w_ZZ + w_ph0lq3 + w_Olq3ph + w_Z0lq3 + w_Olq3Z
)
return w_phph + w_phZ + w_Zph + w_ZZ + w_ph4F + w_4Fph + w_Z4F + w_4FZ
raise ValueError(f"Unknown process: {self.obs_config['process']}")

@classmethod
Expand Down

0 comments on commit 0d3e636

Please sign in to comment.