mirror of
https://github.com/fjosw/pyerrors.git
synced 2025-05-15 12:03:42 +02:00
flake8 compliance openQCD.py
This commit is contained in:
parent
302a7ae439
commit
5f156e4821
2 changed files with 144 additions and 107 deletions
|
@ -8,7 +8,6 @@ import struct
|
||||||
import numpy as np # Thinly-wrapped numpy
|
import numpy as np # Thinly-wrapped numpy
|
||||||
from ..obs import Obs
|
from ..obs import Obs
|
||||||
from ..fits import fit_lin
|
from ..fits import fit_lin
|
||||||
from . import utils
|
|
||||||
|
|
||||||
|
|
||||||
def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
||||||
|
@ -67,7 +66,8 @@ def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
||||||
else:
|
else:
|
||||||
r_stop = [None] * replica
|
r_stop = [None] * replica
|
||||||
|
|
||||||
print('Read reweighting factors from', prefix[:-1], ',', replica, 'replica', end='')
|
print('Read reweighting factors from', prefix[:-1], ',',
|
||||||
|
replica, 'replica', end='')
|
||||||
|
|
||||||
# Adjust replica names to new bookmarking system
|
# Adjust replica names to new bookmarking system
|
||||||
if names is None:
|
if names is None:
|
||||||
|
@ -75,7 +75,8 @@ def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
||||||
for entry in ls:
|
for entry in ls:
|
||||||
truncated_entry = entry.split('.')[0]
|
truncated_entry = entry.split('.')[0]
|
||||||
idx = truncated_entry.index('r')
|
idx = truncated_entry.index('r')
|
||||||
rep_names.append(truncated_entry[:idx] + '|' + truncated_entry[idx:])
|
rep_names.append(truncated_entry[:idx] + '|'
|
||||||
|
+ truncated_entry[idx:])
|
||||||
|
|
||||||
print_err = 0
|
print_err = 0
|
||||||
if 'print_err' in kwargs:
|
if 'print_err' in kwargs:
|
||||||
|
@ -97,8 +98,13 @@ def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
||||||
for k in range(nrw):
|
for k in range(nrw):
|
||||||
deltas.append([])
|
deltas.append([])
|
||||||
else:
|
else:
|
||||||
if ((nrw != struct.unpack('i', t)[0] and (not version == '2.0')) or (nrw != struct.unpack('i', t)[0] / 2 and version == '2.0')): # little weird if-clause due to the /2 operation needed.
|
# little weird if-clause due to the /2 operation needed.
|
||||||
raise Exception('Error: different number of reweighting factors for replicum', rep)
|
if ((nrw != struct.unpack('i', t)[0] and
|
||||||
|
(not version == '2.0')) or
|
||||||
|
(nrw != struct.unpack('i', t)[0] / 2 and
|
||||||
|
version == '2.0')):
|
||||||
|
raise Exception('Error: different number of reweighting\
|
||||||
|
factors for replicum', rep)
|
||||||
|
|
||||||
for k in range(nrw):
|
for k in range(nrw):
|
||||||
tmp_array.append([])
|
tmp_array.append([])
|
||||||
|
@ -109,7 +115,8 @@ def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
||||||
for i in range(nrw):
|
for i in range(nrw):
|
||||||
t = fp.read(4)
|
t = fp.read(4)
|
||||||
nfct.append(struct.unpack('i', t)[0])
|
nfct.append(struct.unpack('i', t)[0])
|
||||||
# print('nfct: ', nfct) # Hasenbusch factor, 1 for rat reweighting
|
# print('nfct: ', nfct) # Hasenbusch factor,
|
||||||
|
# 1 for rat reweighting
|
||||||
else:
|
else:
|
||||||
for i in range(nrw):
|
for i in range(nrw):
|
||||||
nfct.append(1)
|
nfct.append(1)
|
||||||
|
@ -138,8 +145,11 @@ def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
||||||
for j in range(tmpd['n'][0]):
|
for j in range(tmpd['n'][0]):
|
||||||
tmp_nfct *= np.mean(np.exp(-np.asarray(tmp_rw[j])))
|
tmp_nfct *= np.mean(np.exp(-np.asarray(tmp_rw[j])))
|
||||||
if print_err:
|
if print_err:
|
||||||
print(config_no, i, j, np.mean(np.exp(-np.asarray(tmp_rw[j]))), np.std(np.exp(-np.asarray(tmp_rw[j]))))
|
print(config_no, i, j,
|
||||||
print('Sources:', np.exp(-np.asarray(tmp_rw[j])))
|
np.mean(np.exp(-np.asarray(tmp_rw[j]))),
|
||||||
|
np.std(np.exp(-np.asarray(tmp_rw[j]))))
|
||||||
|
print('Sources:',
|
||||||
|
np.exp(-np.asarray(tmp_rw[j])))
|
||||||
print('Partial factor:', tmp_nfct)
|
print('Partial factor:', tmp_nfct)
|
||||||
elif version == '1.6' or version == '1.4':
|
elif version == '1.6' or version == '1.4':
|
||||||
tmp_nfct = 1.0
|
tmp_nfct = 1.0
|
||||||
|
@ -149,7 +159,9 @@ def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
||||||
tmp_rw = struct.unpack('d' * nsrc[i], t)
|
tmp_rw = struct.unpack('d' * nsrc[i], t)
|
||||||
tmp_nfct *= np.mean(np.exp(-np.asarray(tmp_rw)))
|
tmp_nfct *= np.mean(np.exp(-np.asarray(tmp_rw)))
|
||||||
if print_err:
|
if print_err:
|
||||||
print(config_no, i, j, np.mean(np.exp(-np.asarray(tmp_rw))), np.std(np.exp(-np.asarray(tmp_rw))))
|
print(config_no, i, j,
|
||||||
|
np.mean(np.exp(-np.asarray(tmp_rw))),
|
||||||
|
np.std(np.exp(-np.asarray(tmp_rw))))
|
||||||
print('Sources:', np.exp(-np.asarray(tmp_rw)))
|
print('Sources:', np.exp(-np.asarray(tmp_rw)))
|
||||||
print('Partial factor:', tmp_nfct)
|
print('Partial factor:', tmp_nfct)
|
||||||
tmp_array[i].append(tmp_nfct)
|
tmp_array[i].append(tmp_nfct)
|
||||||
|
@ -168,11 +180,14 @@ def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|
||||||
def extract_t0(path, prefix, dtr_read, xmin, spatial_extent, fit_range=5, **kwargs):
|
def extract_t0(path, prefix, dtr_read, xmin,
|
||||||
|
spatial_extent, fit_range=5, **kwargs):
|
||||||
"""Extract t0 from given .ms.dat files. Returns t0 as Obs.
|
"""Extract t0 from given .ms.dat files. Returns t0 as Obs.
|
||||||
|
|
||||||
It is assumed that all boundary effects have sufficiently decayed at x0=xmin.
|
It is assumed that all boundary effects have
|
||||||
The data around the zero crossing of t^2<E> - 0.3 is fitted with a linear function
|
sufficiently decayed at x0=xmin.
|
||||||
|
The data around the zero crossing of t^2<E> - 0.3
|
||||||
|
is fitted with a linear function
|
||||||
from which the exact root is extracted.
|
from which the exact root is extracted.
|
||||||
Only works with openQCD v 1.2.
|
Only works with openQCD v 1.2.
|
||||||
|
|
||||||
|
@ -183,14 +198,17 @@ def extract_t0(path, prefix, dtr_read, xmin, spatial_extent, fit_range=5, **kwar
|
||||||
prefix : str
|
prefix : str
|
||||||
Ensemble prefix
|
Ensemble prefix
|
||||||
dtr_read : int
|
dtr_read : int
|
||||||
Determines how many trajectories should be skipped when reading the ms.dat files.
|
Determines how many trajectories should be skipped
|
||||||
|
when reading the ms.dat files.
|
||||||
Corresponds to dtr_cnfg / dtr_ms in the openQCD input file.
|
Corresponds to dtr_cnfg / dtr_ms in the openQCD input file.
|
||||||
xmin : int
|
xmin : int
|
||||||
First timeslice where the boundary effects have sufficiently decayed.
|
First timeslice where the boundary
|
||||||
|
effects have sufficiently decayed.
|
||||||
spatial_extent : int
|
spatial_extent : int
|
||||||
spatial extent of the lattice, required for normalization.
|
spatial extent of the lattice, required for normalization.
|
||||||
fit_range : int
|
fit_range : int
|
||||||
Number of data points left and right of the zero crossing to be included in the linear fit. (Default: 5)
|
Number of data points left and right of the zero
|
||||||
|
crossing to be included in the linear fit. (Default: 5)
|
||||||
r_start : list
|
r_start : list
|
||||||
list which contains the first config to be read for each replicum.
|
list which contains the first config to be read for each replicum.
|
||||||
r_stop: list
|
r_stop: list
|
||||||
|
@ -276,7 +294,9 @@ def extract_t0(path, prefix, dtr_read, xmin, spatial_extent, fit_range=5, **kwar
|
||||||
|
|
||||||
Ysum.append([])
|
Ysum.append([])
|
||||||
for i, item in enumerate(Ysl):
|
for i, item in enumerate(Ysl):
|
||||||
Ysum[-1].append([np.mean(item[current + xmin:current + tmax - xmin]) for current in range(0, len(item), tmax)])
|
Ysum[-1].append([np.mean(item[current + xmin:
|
||||||
|
current + tmax - xmin])
|
||||||
|
for current in range(0, len(item), tmax)])
|
||||||
|
|
||||||
t2E_dict = {}
|
t2E_dict = {}
|
||||||
for n in range(nn + 1):
|
for n in range(nn + 1):
|
||||||
|
@ -287,12 +307,16 @@ def extract_t0(path, prefix, dtr_read, xmin, spatial_extent, fit_range=5, **kwar
|
||||||
samples[-1].append(cnfg[n])
|
samples[-1].append(cnfg[n])
|
||||||
samples[-1] = samples[-1][r_start[nrep]:r_stop[nrep]]
|
samples[-1] = samples[-1][r_start[nrep]:r_stop[nrep]]
|
||||||
new_obs = Obs(samples, [(w.split('.'))[0] for w in ls])
|
new_obs = Obs(samples, [(w.split('.'))[0] for w in ls])
|
||||||
t2E_dict[n * dn * eps] = (n * dn * eps) ** 2 * new_obs / (spatial_extent ** 3) - 0.3
|
t2E_dict[n * dn * eps] = (n * dn * eps) ** 2 * new_obs \
|
||||||
|
/ (spatial_extent ** 3) - 0.3
|
||||||
|
|
||||||
zero_crossing = np.argmax(np.array([o.value for o in t2E_dict.values()]) > 0.0)
|
zero_crossing = np.argmax(np.array(
|
||||||
|
[o.value for o in t2E_dict.values()]) > 0.0)
|
||||||
|
|
||||||
x = list(t2E_dict.keys())[zero_crossing - fit_range: zero_crossing + fit_range]
|
x = list(t2E_dict.keys())[zero_crossing - fit_range:
|
||||||
y = list(t2E_dict.values())[zero_crossing - fit_range: zero_crossing + fit_range]
|
zero_crossing + fit_range]
|
||||||
|
y = list(t2E_dict.values())[zero_crossing - fit_range:
|
||||||
|
zero_crossing + fit_range]
|
||||||
[o.gamma_method() for o in y]
|
[o.gamma_method() for o in y]
|
||||||
|
|
||||||
fit_result = fit_lin(x, y)
|
fit_result = fit_lin(x, y)
|
||||||
|
@ -348,7 +372,7 @@ def _read_array_openQCD2(fp):
|
||||||
return {'d': d, 'n': n, 'size': size, 'arr': arr}
|
return {'d': d, 'n': n, 'size': size, 'arr': arr}
|
||||||
|
|
||||||
|
|
||||||
def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
|
def read_qtop(path, prefix, c, dtr_cnfg=1, version="1.2", **kwargs):
|
||||||
"""Read qtop format from given folder structure.
|
"""Read qtop format from given folder structure.
|
||||||
|
|
||||||
Parameters
|
Parameters
|
||||||
|
@ -360,130 +384,136 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
|
||||||
c: double
|
c: double
|
||||||
Smearing radius in units of the lattice extent, c = sqrt(8 t0) / L
|
Smearing radius in units of the lattice extent, c = sqrt(8 t0) / L
|
||||||
dtr_cnfg: int
|
dtr_cnfg: int
|
||||||
(optional) parameter that specifies the number of trajectories between two configs.
|
(optional) parameter that specifies the number of trajectories
|
||||||
if it is not set, the distance between two measurements in the file is assumed to be
|
between two configs.
|
||||||
|
if it is not set, the distance between two measurements
|
||||||
|
in the file is assumed to be
|
||||||
the distance between two configurations.
|
the distance between two configurations.
|
||||||
steps: int
|
steps: int
|
||||||
(optional) (maybe only necessary for openQCD2.0)
|
(optional) (maybe only necessary for openQCD2.0)
|
||||||
nt step size, guessed if not given
|
nt step size, guessed if not given
|
||||||
version: str
|
version: str
|
||||||
version string of the openQCD (sfqcd) version used to create the ensemble
|
version string of the openQCD (sfqcd) version used to create
|
||||||
|
the ensemble
|
||||||
L: int
|
L: int
|
||||||
spatial length of the lattice in L/a. HAS to be set if version != sfqcd, since openQCD does not provide this in the header
|
spatial length of the lattice in L/a.
|
||||||
|
HAS to be set if version != sfqcd, since openQCD does not provide
|
||||||
|
this in the header
|
||||||
r_start: list
|
r_start: list
|
||||||
offset of the first ensemble, making it easier to match later on with other Obs
|
offset of the first ensemble, making it easier to match
|
||||||
|
later on with other Obs
|
||||||
r_stop: list
|
r_stop: list
|
||||||
last configurations that need to be read (per replicum)
|
last configurations that need to be read (per replicum)
|
||||||
r_meas_start: list
|
|
||||||
offset of the first measured ensemble, if there is any
|
|
||||||
files: list
|
files: list
|
||||||
specify the exact files that need to be read from path, pratical if e.g. only one replicum is needed
|
specify the exact files that need to be read
|
||||||
|
from path, pratical if e.g. only one replicum is needed
|
||||||
names: list
|
names: list
|
||||||
Alternative labeling for replicas/ensembles. Has to have the appropriate length
|
Alternative labeling for replicas/ensembles.
|
||||||
|
Has to have the appropriate length
|
||||||
"""
|
"""
|
||||||
#one could read L from the header in case of sfQCD
|
# one could read L from the header in case of sfQCD
|
||||||
#c = 0.35
|
# c = 0.35
|
||||||
known_versions = ["1.0","1.2","1.4","1.6","2.0", "sfqcd"]
|
known_versions = ["1.0", "1.2", "1.4", "1.6", "2.0", "sfqcd"]
|
||||||
|
|
||||||
if not version in known_versions:
|
if version not in known_versions:
|
||||||
raise Exception("Unknown openQCD version.")
|
raise Exception("Unknown openQCD version.")
|
||||||
target = 0
|
|
||||||
if "steps" in kwargs:
|
if "steps" in kwargs:
|
||||||
steps = kwargs.get("steps")
|
steps = kwargs.get("steps")
|
||||||
|
|
||||||
if 'target' in kwargs:
|
|
||||||
target = kwargs.get('target')
|
|
||||||
if version == "sfqcd":
|
if version == "sfqcd":
|
||||||
if "L" in kwargs:
|
if "L" in kwargs:
|
||||||
supposed_L = kwargs.get("L")
|
supposed_L = kwargs.get("L")
|
||||||
else:
|
else:
|
||||||
if not "L" in kwargs:
|
if "L" not in kwargs:
|
||||||
raise Exception("This version of openQCD needs you to provide the spatial length of the lattice as parameter 'L'.")
|
raise Exception("This version of openQCD needs you \
|
||||||
|
to provide the spatial length of the \
|
||||||
|
lattice as parameter 'L'.")
|
||||||
else:
|
else:
|
||||||
L = kwargs.get("L")
|
L = kwargs.get("L")
|
||||||
r_start = 1
|
r_start = 1
|
||||||
r_meas_start = 1
|
|
||||||
if "r_meas_start" in kwargs:
|
|
||||||
r_meas_start = kwargs.get("r_meas_start")
|
|
||||||
if "r_start" in kwargs:
|
if "r_start" in kwargs:
|
||||||
r_start = kwargs.get("r_start")
|
r_start = kwargs.get("r_start")
|
||||||
if "r_stop" in kwargs:
|
if "r_stop" in kwargs:
|
||||||
r_stop = kwargs.get("r_stop")
|
r_stop = kwargs.get("r_stop")
|
||||||
#if one wants to read specific files with this method...
|
# if one wants to read specific files with this method...
|
||||||
if "files" in kwargs:
|
if "files" in kwargs:
|
||||||
files = kwargs.get("files")
|
files = kwargs.get("files")
|
||||||
else:
|
else:
|
||||||
#find files in path
|
# find files in path
|
||||||
found = []
|
found = []
|
||||||
files = []
|
files = []
|
||||||
for (dirpath, dirnames, filenames) in os.walk(path+"/"):
|
for (dirpath, dirnames, filenames) in os.walk(path+"/"):
|
||||||
#print(filenames)
|
# print(filenames)
|
||||||
found.extend(filenames)
|
found.extend(filenames)
|
||||||
break
|
break
|
||||||
for f in found:
|
for f in found:
|
||||||
if fnmatch.fnmatch(f, prefix+"*"+".ms.dat"):
|
if fnmatch.fnmatch(f, prefix+"*"+".ms.dat"):
|
||||||
files.append(f)
|
files.append(f)
|
||||||
print(files)
|
print(files)
|
||||||
#now that we found our files, we dechiffer them...
|
# now that we found our files, we dechiffer them...
|
||||||
rep_names = []
|
rep_names = []
|
||||||
|
|
||||||
deltas = []
|
deltas = []
|
||||||
idl = []
|
idl = []
|
||||||
for rep,file in enumerate(files):
|
for rep, file in enumerate(files):
|
||||||
|
|
||||||
with open(path+"/"+file, "rb") as fp:
|
with open(path+"/"+file, "rb") as fp:
|
||||||
#this, for now, is for version 1.2,1.4,1.6 and 2.0, but needs to be tested for the last 3, isncethe doc says its the same
|
# header
|
||||||
#header
|
|
||||||
t = fp.read(12)
|
t = fp.read(12)
|
||||||
header = struct.unpack('<iii', t)
|
header = struct.unpack('<iii', t)
|
||||||
dn = header[0] # step size in integration steps "dnms"
|
# step size in integration steps "dnms"
|
||||||
nn = header[1] # number of measurements, so "ntot"/dn
|
dn = header[0]
|
||||||
tmax = header[2]# lattice T/a
|
# number of measurements, so "ntot"/dn
|
||||||
|
nn = header[1]
|
||||||
|
# lattice T/a
|
||||||
|
tmax = header[2]
|
||||||
if version == "sfqcd":
|
if version == "sfqcd":
|
||||||
t = fp.read(12)
|
t = fp.read(12)
|
||||||
Ls = struct.unpack('<iii', t)
|
Ls = struct.unpack('<iii', t)
|
||||||
if(Ls[0] == Ls[1] and Ls[1] == Ls[2]):
|
if(Ls[0] == Ls[1] and Ls[1] == Ls[2]):
|
||||||
L = Ls[0]
|
L = Ls[0]
|
||||||
if not (supposed_L == L):
|
if not (supposed_L == L):
|
||||||
raise Exception("It seems the length given in the header and by you contradict each other")
|
raise Exception("It seems the length given\
|
||||||
|
in the header and by you \
|
||||||
|
contradict each other")
|
||||||
else:
|
else:
|
||||||
raise Exception("Found more than one spatial length in header!")
|
raise Exception("Found more than one \
|
||||||
|
spatial length in header!")
|
||||||
|
|
||||||
print('dnms:', dn)
|
print('dnms:', dn)
|
||||||
print('nn:', nn)
|
print('nn:', nn)
|
||||||
print('tmax:', tmax)
|
print('tmax:', tmax)
|
||||||
t = fp.read(8) #header4 wird mit 16 bytes ausgelesen... cmax fehlt also als Angabe
|
t = fp.read(8)
|
||||||
eps = struct.unpack('d', t)[0]# das hier ist vllt. tol
|
eps = struct.unpack('d', t)[0]
|
||||||
print('eps:', eps)
|
print('eps:', eps)
|
||||||
|
|
||||||
Q = []
|
Q = []
|
||||||
ncs = []
|
ncs = []
|
||||||
while 0 < 1:
|
while 0 < 1:
|
||||||
t = fp.read(4) #int nt
|
# int nt
|
||||||
|
t = fp.read(4)
|
||||||
if(len(t) < 4):
|
if(len(t) < 4):
|
||||||
break
|
break
|
||||||
ncs.append(struct.unpack('i',t)[0])
|
ncs.append(struct.unpack('i', t)[0])
|
||||||
t = fp.read(8 * tmax * (nn + 1))#Wsl
|
# Wsl
|
||||||
t = fp.read(8 * tmax * (nn + 1))#Ysl
|
t = fp.read(8 * tmax * (nn + 1))
|
||||||
t = fp.read(8 * tmax * (nn + 1))#Qsl, which is asked for in this method
|
# Ysl
|
||||||
#unpack the array of Qtops, on each timeslice t=0,...,tmax-1 and the
|
t = fp.read(8 * tmax * (nn + 1))
|
||||||
#measurement number in = 0...nn (see README.qcd1)
|
# Qsl, which is asked for in this method
|
||||||
|
t = fp.read(8 * tmax * (nn + 1))
|
||||||
|
# unpack the array of Qtops,
|
||||||
|
# on each timeslice t=0,...,tmax-1 and the
|
||||||
|
# measurement number in = 0...nn (see README.qcd1)
|
||||||
tmpd = struct.unpack('d' * tmax * (nn + 1), t)
|
tmpd = struct.unpack('d' * tmax * (nn + 1), t)
|
||||||
Q.append(tmpd)
|
Q.append(tmpd)
|
||||||
#set step by reading all entries, then set stepsize, then check if everything is there
|
|
||||||
#make a dtr_config param, which is checked against difference...
|
|
||||||
#difference != step
|
|
||||||
|
|
||||||
if not len(set([ncs[i]-ncs[i-1] for i in range(1,len(ncs))])):
|
if not len(set([ncs[i]-ncs[i-1] for i in range(1, len(ncs))])):
|
||||||
raise Exception("Irregularities in stepsize found")
|
raise Exception("Irregularities in stepsize found")
|
||||||
else:
|
else:
|
||||||
if 'steps' in kwargs:
|
if 'steps' in kwargs:
|
||||||
if steps != ncs[1]-ncs[0]:
|
if steps != ncs[1]-ncs[0]:
|
||||||
raise Exception("steps and the found stepsize are not the same")
|
raise Exception("steps and the found stepsize \
|
||||||
|
are not the same")
|
||||||
else:
|
else:
|
||||||
steps = ncs[1]-ncs[0]
|
steps = ncs[1]-ncs[0]
|
||||||
if ncs[0]//steps == ncs[0]/steps:
|
|
||||||
r_meas_start = ncs[0]//steps
|
|
||||||
|
|
||||||
print(len(Q))
|
print(len(Q))
|
||||||
print('max_t:', dn * (nn) * eps)
|
print('max_t:', dn * (nn) * eps)
|
||||||
|
@ -494,10 +524,10 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
|
||||||
index_aim = round(t_aim / eps / dn)
|
index_aim = round(t_aim / eps / dn)
|
||||||
print('index_aim:', index_aim)
|
print('index_aim:', index_aim)
|
||||||
|
|
||||||
|
|
||||||
Q_sum = []
|
Q_sum = []
|
||||||
for i, item in enumerate(Q):
|
for i, item in enumerate(Q):
|
||||||
Q_sum.append([sum(item[current:current + tmax]) for current in range(0, len(item), tmax)])
|
Q_sum.append([sum(item[current:current + tmax])
|
||||||
|
for current in range(0, len(item), tmax)])
|
||||||
print(len(Q_sum))
|
print(len(Q_sum))
|
||||||
print(len(Q_sum[0]))
|
print(len(Q_sum[0]))
|
||||||
Q_round = []
|
Q_round = []
|
||||||
|
@ -506,7 +536,7 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
|
||||||
if len(Q_round) != len(ncs)//dtr_cnfg:
|
if len(Q_round) != len(ncs)//dtr_cnfg:
|
||||||
raise Exception("qtops and ncs dont have the same length")
|
raise Exception("qtops and ncs dont have the same length")
|
||||||
|
|
||||||
#replica = len(files)
|
# replica = len(files)
|
||||||
|
|
||||||
truncated_file = file[:-7]
|
truncated_file = file[:-7]
|
||||||
print(truncated_file)
|
print(truncated_file)
|
||||||
|
@ -518,13 +548,14 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
|
||||||
if "r_stop" in kwargs:
|
if "r_stop" in kwargs:
|
||||||
Q_round = Q_round[:r_stop[rep]]
|
Q_round = Q_round[:r_stop[rep]]
|
||||||
idl_stop = idl_start+len(Q_round)
|
idl_stop = idl_start+len(Q_round)
|
||||||
#keyword "names" prevails over "ens_name"
|
# keyword "names" prevails over "ens_name"
|
||||||
if not "names" in kwargs:
|
if "names" not in kwargs:
|
||||||
try:
|
try:
|
||||||
idx = truncated_file.index('r')
|
idx = truncated_file.index('r')
|
||||||
except:
|
except Exception:
|
||||||
if not "names" in kwargs:
|
if "names" not in kwargs:
|
||||||
raise Exception("Automatic recognition of replicum failed, please enter the key word 'names'.")
|
raise Exception("Automatic recognition of replicum failed,\
|
||||||
|
please enter the key word 'names'.")
|
||||||
if "ens_name" in kwargs:
|
if "ens_name" in kwargs:
|
||||||
ens_name = kwargs.get("ens_name")
|
ens_name = kwargs.get("ens_name")
|
||||||
else:
|
else:
|
||||||
|
@ -534,12 +565,13 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
|
||||||
names = kwargs.get("names")
|
names = kwargs.get("names")
|
||||||
rep_names = names
|
rep_names = names
|
||||||
deltas.append(np.array(Q_round))
|
deltas.append(np.array(Q_round))
|
||||||
idl.append(range(idl_start,idl_stop))
|
idl.append(range(idl_start, idl_stop))
|
||||||
#print(idl)
|
# print(idl)
|
||||||
result = Obs(deltas, rep_names, idl = idl)
|
result = Obs(deltas, rep_names, idl=idl)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def read_qtop_sector(target = 0, **kwargs):
|
|
||||||
|
def read_qtop_sector(target=0, **kwargs):
|
||||||
"""target: int
|
"""target: int
|
||||||
specifies the topological sector to be reweighted to (default 0)
|
specifies the topological sector to be reweighted to (default 0)
|
||||||
q_top: Obs
|
q_top: Obs
|
||||||
|
@ -552,12 +584,14 @@ def read_qtop_sector(target = 0, **kwargs):
|
||||||
path = kwargs.get("path")
|
path = kwargs.get("path")
|
||||||
del kwargs["path"]
|
del kwargs["path"]
|
||||||
else:
|
else:
|
||||||
raise Exception("If you are not providing q_top, please provide path")
|
raise Exception("If you are not providing q_top,\
|
||||||
|
please provide path")
|
||||||
if "prefix" in kwargs:
|
if "prefix" in kwargs:
|
||||||
prefix = kwargs.get("prefix")
|
prefix = kwargs.get("prefix")
|
||||||
del kwargs["prefix"]
|
del kwargs["prefix"]
|
||||||
else:
|
else:
|
||||||
raise Exception("If you are not providing q_top, please provide prefix")
|
raise Exception("If you are not providing q_top,\
|
||||||
|
please provide prefix")
|
||||||
if "c" in kwargs:
|
if "c" in kwargs:
|
||||||
c = kwargs.get("c")
|
c = kwargs.get("c")
|
||||||
del kwargs["c"]
|
del kwargs["c"]
|
||||||
|
@ -573,14 +607,16 @@ def read_qtop_sector(target = 0, **kwargs):
|
||||||
del kwargs["dtr_cnfg"]
|
del kwargs["dtr_cnfg"]
|
||||||
else:
|
else:
|
||||||
dtr_cnfg = 1
|
dtr_cnfg = 1
|
||||||
qtop = read_qtop(path, prefix,c, dtr_cnfg = dtr_cnfg, version = version,**kwargs)
|
qtop = read_qtop(path, prefix, c, dtr_cnfg=dtr_cnfg,
|
||||||
#unpack to original values, project onto target sector
|
version=version, **kwargs)
|
||||||
|
# unpack to original values, project onto target sector
|
||||||
names = qtop.names
|
names = qtop.names
|
||||||
print(names)
|
print(names)
|
||||||
print(qtop.deltas.keys())
|
print(qtop.deltas.keys())
|
||||||
proj_qtop = []
|
proj_qtop = []
|
||||||
for n in qtop.deltas:
|
for n in qtop.deltas:
|
||||||
proj_qtop.append(np.array([1 if int(qtop.value+q) == target else 0 for q in qtop.deltas[n]]))
|
proj_qtop.append(np.array([1 if int(qtop.value+q) ==
|
||||||
|
target else 0 for q in qtop.deltas[n]]))
|
||||||
|
|
||||||
result = Obs(proj_qtop, qtop.names)
|
result = Obs(proj_qtop, qtop.names)
|
||||||
return result
|
return result
|
||||||
|
|
|
@ -76,8 +76,6 @@ def read_sfcf(path, prefix, name, quarks='.*', noffset=0, wf=0, wf2=0,
|
||||||
single = 0
|
single = 0
|
||||||
if "replica" in kwargs:
|
if "replica" in kwargs:
|
||||||
reps = kwargs.get("replica")
|
reps = kwargs.get("replica")
|
||||||
if "files" in kwargs:
|
|
||||||
files = kwargs.get("files")
|
|
||||||
|
|
||||||
# due to higher usage in current projects,
|
# due to higher usage in current projects,
|
||||||
# compact file format is default
|
# compact file format is default
|
||||||
|
@ -338,6 +336,9 @@ def read_sfcf(path, prefix, name, quarks='.*', noffset=0, wf=0, wf2=0,
|
||||||
deltas[k - start][i][cnfg] = \
|
deltas[k - start][i][cnfg] = \
|
||||||
floats[1 + im - single]
|
floats[1 + im - single]
|
||||||
|
|
||||||
|
else:
|
||||||
|
if "files" in kwargs:
|
||||||
|
ls = kwargs.get("files")
|
||||||
else:
|
else:
|
||||||
for exc in ls:
|
for exc in ls:
|
||||||
if not fnmatch.fnmatch(exc, prefix + '*.'+name):
|
if not fnmatch.fnmatch(exc, prefix + '*.'+name):
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue