lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-05-14 19:43:41 +02:00
Code Issues Projects Releases Packages Wiki Activity

first steps to output in the right format

Browse source
This commit is contained in:
jkuhl-uni 2022-12-20 15:08:34 +01:00
parent ae62548661
commit 134fe7d3f7
1 changed files with 23 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

31
pyerrors/input/openQCD.py
View file

@ -8,6 +8,9 @@ import matplotlib.pyplot as plt
from matplotlib import gridspec from matplotlib import gridspec
from ..obs import Obs from ..obs import Obs
from ..fits import fit_lin from ..fits import fit_lin
from ..obs import CObs
from ..correlators import Corr
def read_rwms(path, prefix, version='2.0', names=None, **kwargs): def read_rwms(path, prefix, version='2.0', names=None, **kwargs):
@ -989,7 +992,8 @@ def read_qtop_sector(path, prefix, c, target=0, **kwargs):
def read_ms5_xsf(path, prefix, qc, corr): def read_ms5_xsf(path, prefix, qc, corr):
""" """
Read and process data from files with a specific prefix and containing the given quark combination in the file name. Read and process data from files produced by the ms5_xsf method
with a specific prefix and for a given quark combination.
Parameters Parameters
---------- ----------
@ -1051,25 +1055,36 @@ def read_ms5_xsf(path, prefix, qc, corr):
chunksize = 4 +( 8 *2*tmax*10)+( 8 *2*2) chunksize = 4 +( 8 *2*tmax*10)+( 8 *2*2)
packstr ='=i'+('d'*2*tmax*10)+('d'*2*2) packstr ='=i'+('d'*2*tmax*10)+('d'*2*2)
cnfgs = []
realsamples = []
imagsamples = []
for t in range(tmax):
realsamples.append([])
imagsamples.append([])
while True: while True:
cnfgt = fp.read(chunksize) cnfgt = fp.read(chunksize)
if not cnfgt: if not cnfgt:
break break
asascii=struct.unpack(packstr, cnfgt) asascii=struct.unpack(packstr, cnfgt)
cnfg = asascii[0] cnfg = asascii[0]
#print(cnfg) cnfgs.append(cnfg)
if not corr in placesBB: if not corr in placesBB:
tmpcorr = asascii[1+2*tmax*placesBI.index(corr):1+2*tmax*placesBI.index(corr)+2*tmax] tmpcorr = asascii[1+2*tmax*placesBI.index(corr):1+2*tmax*placesBI.index(corr)+2*tmax]
corrres = [[],[]] corrres = [[],[]]
for i in range(len(tmpcorr)): for i in range(len(tmpcorr)): corrres[i%2].append(tmpcorr[i])
corrres[i%2].append(tmpcorr[i]) for t in range(tmax): realsamples[t].append(corrres[0][t])
for t in range(tmax): imagsamples[t].append(corrres[1][t])
else: else:
tmpcorr = asascii[1+2*tmax*len(placesBI):1+2*tmax*len(placesBI)+2] tmpcorr = asascii[1+2*tmax*len(placesBI):1+2*tmax*len(placesBI)+2]
corrres = [[],[]] corrres = [[],[]]
for i in range(len(tmpcorr)): for i in range(len(tmpcorr)):
corrres[i%2].append(tmpcorr[i]) corrres[i%2].append(tmpcorr[i])
realObs = []
#print(corrres) imagObs = []
compObs = []
for t in range(tmax): realObs.append(Obs(realsamples[t], names = prefix, idl = cnfgs))
for t in range(tmax): imagObs.append(Obs(imagsamples[t], names = prefix, idl = cnfgs))
for t in range(tmax): compObs.append(CObs(realObs[t], imagObs[t]))
return Corr(compObs)