jkuhl/corrlib
1
0
Fork 0
Code Issues Pull requests Releases Packages Activity

more search filters added to sfcf_filter

Browse source
This commit is contained in:
Justus Kuhlmann 2025-05-15 16:18:28 +00:00
parent 4f7d9d71b3
commit 11a5c07277
1 changed files with 28 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

30
corrlib/find.py
View file

@ -5,7 +5,7 @@ import json
import pandas as pd import pandas as pd
import numpy as np import numpy as np
from .input.implementations import codes from .input.implementations import codes
from .tools import m2k, k2m from .tools import k2m
# this will implement the search functionality # this will implement the search functionality
@ -71,6 +71,33 @@ def sfcf_filter(results, **kwargs):
if (not np.isclose(kappas[0], param['quarks'][0]['mass']) or not np.isclose(kappas[1], param['quarks'][1]['mass'])): if (not np.isclose(kappas[0], param['quarks'][0]['mass']) or not np.isclose(kappas[1], param['quarks'][1]['mass'])):
drops.append(ind) drops.append(ind)
continue continue
if 'quark_masses' in kwargs:
masses = kwargs['quark_masses']
if (not np.isclose(masses[0], k2m(param['quarks'][0]['mass'])) or not np.isclose(masses[1], k2m(param['quarks'][1]['mass']))):
drops.append(ind)
continue
if 'qk1' in kwargs:
quark_kappa1 = kwargs['qk1']
if not isinstance(quark_kappa1, list):
if (not np.isclose(quark_kappa1, param['quarks'][0]['mass'])):
drops.append(ind)
continue
else:
if len(quark_kappa1) == 2:
if (quark_kappa1[0] > param['quarks'][0]['mass']) or (quark_kappa1[1] < param['quarks'][0]['mass']):
drops.append(ind)
continue
if 'qk2' in kwargs:
quark_kappa2 = kwargs['qk2']
if not isinstance(quark_kappa2, list):
if (not np.isclose(quark_kappa2, param['quarks'][1]['mass'])):
drops.append(ind)
continue
else:
if len(quark_kappa2) == 2:
if (quark_kappa2[0] > param['quarks'][1]['mass']) or (quark_kappa2[1] < param['quarks'][1]['mass']):
drops.append(ind)
continue
if 'qm1' in kwargs: if 'qm1' in kwargs:
quark_mass1 = kwargs['qm1'] quark_mass1 = kwargs['qm1']
if not isinstance(quark_mass1, list): if not isinstance(quark_mass1, list):
@ -109,7 +136,6 @@ def sfcf_filter(results, **kwargs):
if not (np.isclose(wf2[0][0], param['wf2'][0][0], 1e-8) and np.isclose(wf2[0][1][0], param['wf2'][0][1][0], 1e-8) and np.isclose(wf2[0][1][1], param['wf2'][0][1][1], 1e-8)): if not (np.isclose(wf2[0][0], param['wf2'][0][0], 1e-8) and np.isclose(wf2[0][1][0], param['wf2'][0][1][0], 1e-8) and np.isclose(wf2[0][1][1], param['wf2'][0][1][1], 1e-8)):
drops.append(ind) drops.append(ind)
continue continue
return results.drop(drops) return results.drop(drops)