zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Support new UB matrix format in ccl files

Browse Source
This commit is contained in:
usov_i 2022-05-06 11:19:25 +02:00
parent 2810cf3d2b
commit 4ebb343afb
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pyzebra/ccl_io.py
View File

@ -56,7 +56,7 @@ META_VARS_FLOAT = (
"s2hl",
)
META_UB_MATRIX = ("ub1j", "ub2j", "ub3j")
META_UB_MATRIX = ("ub1j", "ub2j", "ub3j", "ub")
CCL_FIRST_LINE = (("idx", int), ("h", float), ("k", float), ("l", float))
@ -119,10 +119,13 @@ def parse_1D(fileobj, data_type):
metadata[variable] = float(value)
elif variable in META_UB_MATRIX:
if "ub" not in metadata:
metadata["ub"] = np.zeros((3, 3))
row = int(variable[-2]) - 1
metadata["ub"][row, :] = list(map(float, value.split()))
if variable == "ub":
metadata["ub"] = np.array(value)
else:
if "ub" not in metadata:
metadata["ub"] = np.zeros((3, 3))
row = int(variable[-2]) - 1
metadata["ub"][row, :] = list(map(float, value.split()))
if "#data" in line:
# this is the end of metadata and the start of data section