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

read_qtop now also hands over idl of the result Obs

Browse source
This commit is contained in:
jkuhl-uni 2022-01-03 14:40:12 +01:00
parent 01ada964b2
commit 76b7f159aa
1 changed files with 50 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

92
pyerrors/input/openQCD.py
View file

@ -363,10 +363,11 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
(optional) parameter that specifies the number of trajectories between two configs. (optional) parameter that specifies the number of trajectories between two configs.
if it is not set, the distance between two measurements in the file is assumed to be 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
(optional) (maybe only necessary for openQCD2.0)
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
steps: int
step size of measurements
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
@ -380,7 +381,6 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
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
""" """
#dtr_cnfg = 4# was ist das denn hier?
#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"]
@ -390,8 +390,7 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
target = 0 target = 0
if "steps" in kwargs: if "steps" in kwargs:
steps = kwargs.get("steps") steps = kwargs.get("steps")
else:
steps = 1
if 'target' in kwargs: if 'target' in kwargs:
target = kwargs.get('target') target = kwargs.get('target')
if version == "sfqcd": if version == "sfqcd":
@ -429,6 +428,7 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
rep_names = [] rep_names = []
deltas = [] deltas = []
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:
@ -457,30 +457,34 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
print('eps:', eps) print('eps:', eps)
Q = [] Q = []
ncs = []
i = r_meas_start*steps
while 0 < 1: while 0 < 1:
t = fp.read(4) #int nt t = fp.read(4) #int nt
if(len(t) < 4): if(len(t) < 4):
break break
nc = struct.unpack('i',t)[0] ncs.append(struct.unpack('i',t)[0])
if(nc != i): t = fp.read(8 * tmax * (nn + 1))#Wsl
print(nc) t = fp.read(8 * tmax * (nn + 1))#Ysl
raise Exception('Config ' + str(i) + ' missing?') t = fp.read(8 * tmax * (nn + 1))#Qsl, which is asked for in this method
else: #unpack the array of Qtops, on each timeslice t=0,...,tmax-1 and the
t = fp.read(8 * tmax * (nn + 1))#Wsl #measurement number in = 0...nn (see README.qcd1)
t = fp.read(8 * tmax * (nn + 1))#Ysl tmpd = struct.unpack('d' * tmax * (nn + 1), t)
t = fp.read(8 * tmax * (nn + 1))#Qsl, which is asked for in this method Q.append(tmpd)
#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)
Q.append(tmpd)
i += 1*steps
#set step by reading all entries, then set stepsize, then check if everything is there #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... #make a dtr_config param, which is checked against difference...
#difference != step #difference != step
#len(set(difference)) == 1
#!!!also implement the idl stuff for everything... if not len(set([ncs[i]-ncs[i-1] for i in range(1,len(ncs))])):
raise Exception("Irregularities in stepsize found")
else:
if 'steps' in kwargs:
if steps != ncs[1]-ncs[0]:
raise Exception("steps and the found stepsize are not the same")
else:
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)
@ -499,36 +503,40 @@ def read_qtop(path, prefix,c, dtr_cnfg = 1, version = "1.2",**kwargs):
Q_round = [] Q_round = []
for i in range(len(Q) // dtr_cnfg): for i in range(len(Q) // dtr_cnfg):
Q_round.append(round(Q_sum[dtr_cnfg * i][index_aim])) Q_round.append(round(Q_sum[dtr_cnfg * i][index_aim]))
if len(Q_round) != len(ncs)//dtr_cnfg:
replica = len(files) raise Exception("qtops and ncs dont have the same length")
truncated_file = file[:-7] #as seen in previous examples, this could lead to some weird behaviour... maybe -7 fixes this. #replica = len(files)
print(truncated_file)
try: truncated_file = file[:-7]
idx = truncated_file.index('r') print(truncated_file)
except: idl_start = 1
if not "names" in kwargs:
raise Exception("Automatic recognition of replicum failed, please enter the key word 'names'.")
# this might be a quite fishy way to find out which replicum we are actually talking about...
if "r_start" in kwargs: if "r_start" in kwargs:
Q_round = Q_round[r_start[int(truncated_file[idx+1:])-1]:] Q_round = Q_round[r_start[rep]:]
idl_start = r_start[rep]
if "r_stop" in kwargs: if "r_stop" in kwargs:
Q_round = Q_round[:r_stop[int(truncated_file[idx+1:])-1]] Q_round = Q_round[:r_stop[rep]]
if "ens_name" in kwargs: idl_stop = idl_start+len(Q_round)
ens_name = kwargs.get("ens_name")
else:
ens_name = truncated_file[:idx]
#keyword "names" prevails over "ens_name" #keyword "names" prevails over "ens_name"
if not "names" in kwargs: if not "names" in kwargs:
try:
idx = truncated_file.index('r')
except:
if not "names" in kwargs:
raise Exception("Automatic recognition of replicum failed, please enter the key word 'names'.")
if "ens_name" in kwargs:
ens_name = kwargs.get("ens_name")
else:
ens_name = truncated_file[:idx]
rep_names.append(ens_name + '|' + truncated_file[idx:]) rep_names.append(ens_name + '|' + truncated_file[idx:])
else: else:
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))
#print(idl)
result = Obs(deltas, rep_names) 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):