general s->d UsV

Author: Jason Aebischer

wetrunner/UsV.py

@@ -92,7 +92,8 @@ def run(self, scale_out, sectors='all'):
                                  Etas, p_i['alpha_s'], p_i['alpha_e'],
                                  p_i['m_b'], p_i['m_c'], p_i['m_tau'],
                                  betas))
-        C_out = {k: v for k, v in C_out.items() if v != 0}
+        C_out = {k: v for k, v in C_out.items()
+                 if v != 0 and k in wcxf.Basis[self.eft, 'Bern'].all_wcs}
         return wcxf.WC(eft=self.eft, basis='Bern',
                        scale=scale_out,
                        values=wcxf.WC.dict2values(C_out))

wetrunner/classes.py

@@ -46,11 +46,21 @@ def listdict():
         sectors[qq]['IV'].append(['{}{}{}'.format(i, p, qq)
                                   for i in range(1, 11, 2)])
 
+
+qqdict = {
+'sb': ['uu', 'dd', 'cc', 'ss', 'bb'],
+'db': ['uu', 'ss', 'cc', 'dd', 'bb'],
+'ds': ['uu', 'bb', 'cc', 'dd', 'ss'],
+}
 # class 5
-for qq in ['sb', 'db']:
+for qq in ['sb', 'db', 'ds']:
+    if qq == 'ds':
+        sname = 'sd'
+    else:
+        sname = qq
     for p in ['', 'p']:
         _C = []
-        for pp in ['uu', 'dd', 'cc', 'ss', 'bb']:
+        for pp in qqdict[qq]:
             if pp[0] in qq:
                 _C += ['{}{}{}{}'.format(i, p, qq, pp)
                        for i in range(1, 11, 2)]  # 1, 3, 5, 7, 9
@@ -63,7 +73,7 @@ def listdict():
             # e.g. 1sbee
             _C += ['{}{}{}{}'.format(i, p, qq, 2 * l)
                    for i in range(1, 11, 2)]  # 1, 3, 5, 7, 9
-        sectors[qq]['V'].append(_C)
+        sectors[sname]['V'].append(_C)
 
 # class 5b
 for qq in ['sb', 'db']:

wetrunner/definitions.py

@@ -2,7 +2,7 @@
 
 import numpy as np
 from math import sqrt, pi, log
-from wetrunner.UsV import get_UsV
+from wetrunner.UsV import get_UsV, get_UsVdeltaS
 from wetrunner.UeV import get_UeV
 from functools import lru_cache
 
@@ -91,6 +91,10 @@ def UeIV(Etas, Alphas, Alphaem, mb, mc, mtau, Betas):
 def UsV(Etas, Alphas, Alphaem, mb, mc, mtau, Betas):
     return get_UsV(Etas, mb, mc, mtau)
 
+@lru_cache(maxsize=32)
+def UsVdeltaS(Etas, Alphas, Alphaem, mb, mc, mtau, Betas):
+    return get_UsVdeltaS(Etas)
+
 @lru_cache(maxsize=32)
 def UeV(Etas, Alphas, Alphaem, mb, mc, mtau, Betas):
     return get_UeV(Etas, Alphaem, Alphas, mb, mc, mtau)

wetrunner/evmat.py

@@ -18,7 +18,10 @@ def run_sector(sector, C_in, Etas, Alphas, Alphaem, mb, mc, mtau, betas):
                 # nothing to do for SM-like WCs or qqnunu operators
                 C_result = C_input
             else:
-                Us = getattr(evmat, 'Us' + classname)(*args)
+                if sector == 'sd':
+                    Us = evmat.UsVdeltaS(*args)
+                else:
+                    Us = getattr(evmat, 'Us' + classname)(*args)
                 Ue = getattr(evmat, 'Ue' + classname)(*args)
                 C_result = (Us + Ue) @ C_input
             for j in range(len(C_result)):

wetrunner/rge.py

@@ -81,6 +81,10 @@ def test_inverse_s(self):
                                       np.linalg.inv(evmat.UsIV(1/0.123, *args)))
         npt.assert_array_almost_equal(evmat.UsV(0.123, *args),
                                       np.linalg.inv(evmat.UsV(1/0.123, *args)))
+        npt.assert_array_almost_equal(evmat.UsVdeltaS(0.123, *args)
+                                      @ evmat.UsVdeltaS(1/0.123, *args),
+                                      np.eye(57),
+                                      decimal=2)  # FIXME not precise enough!
         npt.assert_array_almost_equal(evmat.UsVb(0.123, *args),
                                       np.linalg.inv(evmat.UsVb(1/0.123, *args)))