lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-11-30 13:06:53 +01:00
Code Issues Projects Releases Packages Wiki Activity

simplify r_start and r_stop in read_pbp

Browse source
This commit is contained in:
Justus Kuhlmann 2025-11-03 16:17:57 +00:00
commit fb7ba0cbc3
1 changed files with 10 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

14
pyerrors/input/misc.py
View file

@ -136,21 +136,15 @@ def read_pbp(path: str, prefix: str, **kwargs):
ls.sort(key=lambda x: int(re.findall(r'\d+', x[len(prefix):])[0])) ls.sort(key=lambda x: int(re.findall(r'\d+', x[len(prefix):])[0]))
replica = len(ls) replica = len(ls)
if 'r_start' in kwargs: r_start = kwargs.get('r_start', [1] * replica)
r_start = kwargs.get('r_start')
if len(r_start) != replica: if len(r_start) != replica:
raise Exception('r_start does not match number of replicas') raise Exception('r_start does not match number of replicas')
# Adjust Configuration numbering to python index # Adjust Configuration numbering to python index
r_start = [o - 1 if o else None for o in r_start] r_start = [o - 1 if o else None for o in r_start]
else:
r_start = [None] * replica
if 'r_stop' in kwargs: r_stop = kwargs.get('r_stop', [-1] * replica)
r_stop = kwargs.get('r_stop')
if len(r_stop) != replica: if len(r_stop) != replica:
raise Exception('r_stop does not match number of replicas') raise Exception('r_stop does not match number of replicas')
else:
r_stop = [None] * replica
print(r'Read <bar{psi}\psi> from', prefix[:-1], ',', replica, 'replica', end='') print(r'Read <bar{psi}\psi> from', prefix[:-1], ',', replica, 'replica', end='')
@ -159,10 +153,10 @@ def read_pbp(path: str, prefix: str, **kwargs):
print_err = 1 print_err = 1
print() print()
deltas = [] deltas: list[list[float]] = []
for rep in range(replica): for rep in range(replica):
tmp_array = [] tmp_array: list[list[float]] = []
with open(path + '/' + ls[rep], 'rb') as fp: with open(path + '/' + ls[rep], 'rb') as fp:
t = fp.read(4) # number of reweighting factors t = fp.read(4) # number of reweighting factors