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

Add counts_err, idx and scan_motors to hdf data

Browse Source
This commit is contained in:
usov_i 2022-01-28 10:42:31 +01:00
parent bbe7b7d305
commit 5cfa5c176d
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pyzebra/h5.py
View File

@ -74,7 +74,7 @@ def read_detector_data(filepath, cami_meta=None):
n, cols, rows = counts.shape n, cols, rows = counts.shape
counts = counts.reshape(n, rows, cols) counts = counts.reshape(n, rows, cols)
scan = {"counts": counts} scan = {"counts": counts, "counts_err": np.sqrt(np.maximum(counts, 1))}
scan["original_filename"] = filepath scan["original_filename"] = filepath
if "/entry1/zebra_mode" in h5f: if "/entry1/zebra_mode" in h5f:
@ -88,6 +88,7 @@ def read_detector_data(filepath, cami_meta=None):
scan["zebra_mode"] = cami_meta["zebra_mode"][0] scan["zebra_mode"] = cami_meta["zebra_mode"][0]
scan["monitor"] = h5f["/entry1/control/data"][0] scan["monitor"] = h5f["/entry1/control/data"][0]
scan["idx"] = 1
if scan["zebra_mode"] == "nb": if scan["zebra_mode"] == "nb":
scan["omega"] = h5f["/entry1/area_detector2/rotation_angle"][:] scan["omega"] = h5f["/entry1/area_detector2/rotation_angle"][:]
@ -116,6 +117,8 @@ def read_detector_data(filepath, cami_meta=None):
else: else:
raise ValueError("No angles that vary") raise ValueError("No angles that vary")
scan["scan_motors"] = [scan["scan_motor"], ]
# optional parameters # optional parameters
if "/entry1/sample/magnetic_field" in h5f: if "/entry1/sample/magnetic_field" in h5f:
scan["mf"] = h5f["/entry1/sample/magnetic_field"][:] scan["mf"] = h5f["/entry1/sample/magnetic_field"][:]