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

Adapt counts to new h5 file format

Browse Source
This commit is contained in:
usov_i 2022-05-06 16:54:16 +02:00
parent 6f83292d69
commit f044e739a0
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pyzebra/h5.py
View File

@ -70,9 +70,12 @@ def read_detector_data(filepath, cami_meta=None):
with h5py.File(filepath, "r") as h5f: with h5py.File(filepath, "r") as h5f:
counts = h5f["/entry1/area_detector2/data"][:].astype(np.float64) counts = h5f["/entry1/area_detector2/data"][:].astype(np.float64)
# reshape images (counts) to a correct shape (2006 issue)
n, cols, rows = counts.shape n, cols, rows = counts.shape
counts = counts.reshape(n, rows, cols) if "/entry1/experiment_identifier" in h5f: # old format
# reshape images (counts) to a correct shape (2006 issue)
counts = counts.reshape(n, rows, cols)
else:
counts = counts.swapaxes(1, 2)
scan = {"counts": counts, "counts_err": np.sqrt(np.maximum(counts, 1))} scan = {"counts": counts, "counts_err": np.sqrt(np.maximum(counts, 1))}
scan["original_filename"] = filepath scan["original_filename"] = filepath