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

feat: read_DistillationContraction_hd5 can now deal with missing files.

Browse source
This commit is contained in:
Fabian Joswig 2023-01-18 17:51:34 +00:00
parent 88fd37b241
commit 4e71da0b59
1 changed files with 42 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

81
pyerrors/input/hadrons.py
View file

@ -163,54 +163,57 @@ def read_DistillationContraction_hd5(path, ens_id, diagrams=["direct"], idl=None
for diagram in diagrams: for diagram in diagrams:
corr_data[diagram] = [] corr_data[diagram] = []
for n_file, (hd5_file, n_traj) in enumerate(zip(file_list, list(idx))): try:
h5file = h5py.File(hd5_file) for n_file, (hd5_file, n_traj) in enumerate(zip(file_list, list(idx))):
h5file = h5py.File(hd5_file)
if n_file == 0: if n_file == 0:
if h5file["DistillationContraction/Metadata"].attrs.get("TimeSources")[0].decode() != "0...": if h5file["DistillationContraction/Metadata"].attrs.get("TimeSources")[0].decode() != "0...":
raise Exception("Routine is only implemented for files containing inversions on all timeslices.") raise Exception("Routine is only implemented for files containing inversions on all timeslices.")
Nt = h5file["DistillationContraction/Metadata"].attrs.get("Nt")[0] Nt = h5file["DistillationContraction/Metadata"].attrs.get("Nt")[0]
identifier = [] identifier = []
for in_file in range(len(h5file["DistillationContraction/Metadata/DmfInputFiles"].attrs.keys()) - 1): for in_file in range(len(h5file["DistillationContraction/Metadata/DmfInputFiles"].attrs.keys()) - 1):
encoded_info = h5file["DistillationContraction/Metadata/DmfInputFiles"].attrs.get("DmfInputFiles_" + str(in_file)) encoded_info = h5file["DistillationContraction/Metadata/DmfInputFiles"].attrs.get("DmfInputFiles_" + str(in_file))
full_info = encoded_info[0].decode().split("/")[-1].replace(".h5", "").split("_") full_info = encoded_info[0].decode().split("/")[-1].replace(".h5", "").split("_")
my_tuple = (full_info[0], full_info[1][1:], full_info[2], full_info[3]) my_tuple = (full_info[0], full_info[1][1:], full_info[2], full_info[3])
identifier.append(my_tuple) identifier.append(my_tuple)
identifier = tuple(identifier) identifier = tuple(identifier)
# "DistillationContraction/Metadata/DmfSuffix" contains info about different quarks, irrelevant in the SU(3) case. # "DistillationContraction/Metadata/DmfSuffix" contains info about different quarks, irrelevant in the SU(3) case.
for diagram in diagrams:
if diagram == "triangle" and "Identity" not in str(identifier):
part = "im"
else:
part = "re"
real_data = np.zeros(Nt)
for x0 in range(Nt):
raw_data = h5file["DistillationContraction/Correlators/" + diagram + "/" + str(x0)][:][part].astype(np.double)
real_data += np.roll(raw_data, -x0)
real_data /= Nt
corr_data[diagram].append(real_data)
h5file.close()
res_dict[str(identifier)] = {}
for diagram in diagrams: for diagram in diagrams:
if diagram == "triangle" and "Identity" not in str(identifier): tmp_data = np.array(corr_data[diagram])
part = "im"
else:
part = "re"
real_data = np.zeros(Nt) l_obs = []
for x0 in range(Nt): for c in tmp_data.T:
raw_data = h5file["DistillationContraction/Correlators/" + diagram + "/" + str(x0)][:][part].astype(np.double) l_obs.append(Obs([c], [ens_id], idl=[idx]))
real_data += np.roll(raw_data, -x0)
real_data /= Nt
corr_data[diagram].append(real_data) corr = Corr(l_obs)
h5file.close() corr.tag = str(identifier)
res_dict[str(identifier)] = {} res_dict[str(identifier)][diagram] = corr
except FileNotFoundError:
for diagram in diagrams: print("Skip", stem)
tmp_data = np.array(corr_data[diagram])
l_obs = []
for c in tmp_data.T:
l_obs.append(Obs([c], [ens_id], idl=[idx]))
corr = Corr(l_obs)
corr.tag = str(identifier)
res_dict[str(identifier)][diagram] = corr
return res_dict return res_dict