From 5cfa5c176d76d06a0b4945b41e06e5e312ab64fa Mon Sep 17 00:00:00 2001 From: Ivan Usov Date: Fri, 28 Jan 2022 10:42:31 +0100 Subject: [PATCH] Add counts_err, idx and scan_motors to hdf data --- pyzebra/h5.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pyzebra/h5.py b/pyzebra/h5.py index deb2b07..ad8d1c1 100644 --- a/pyzebra/h5.py +++ b/pyzebra/h5.py @@ -74,7 +74,7 @@ def read_detector_data(filepath, cami_meta=None): n, cols, rows = counts.shape counts = counts.reshape(n, rows, cols) - scan = {"counts": counts} + scan = {"counts": counts, "counts_err": np.sqrt(np.maximum(counts, 1))} scan["original_filename"] = filepath 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["monitor"] = h5f["/entry1/control/data"][0] + scan["idx"] = 1 if scan["zebra_mode"] == "nb": scan["omega"] = h5f["/entry1/area_detector2/rotation_angle"][:] @@ -116,6 +117,8 @@ def read_detector_data(filepath, cami_meta=None): else: raise ValueError("No angles that vary") + scan["scan_motors"] = [scan["scan_motor"], ] + # optional parameters if "/entry1/sample/magnetic_field" in h5f: scan["mf"] = h5f["/entry1/sample/magnetic_field"][:]