Skip to content
Snippets Groups Projects
Commit 9a7361d2 authored by Gröne, Tjark Leon Raphael's avatar Gröne, Tjark Leon Raphael
Browse files

Update file maxwell_integrate_to_h5.py

parent 2e23919c
No related branches found
No related tags found
No related merge requests found
...@@ -125,24 +125,32 @@ def integrate_ims_in_dir(path_im, path_int, dtype_im=".tif", dtype_int=".dat"): ...@@ -125,24 +125,32 @@ def integrate_ims_in_dir(path_im, path_int, dtype_im=".tif", dtype_int=".dat"):
# Prepare data for HDF5 file using silx # Prepare data for HDF5 file using silx
hdf5_data = { hdf5_data = {
"entry": { subdir_name: {
"definition": "NXdata", "@NX_class": "NXroot",
"title": subdir_name,
"data": {
"@axes": "q",
"@signal": "I",
"@errors": "dI",
}
} }
} }
for result in results_data: # Sort results_data by filename
filename_key = os.path.basename(result["filename"]) results_data = sorted(results_data, key=lambda x: x["filename"].lower())
hdf5_data["entry"]["data"][filename_key] = {
for idx, result in enumerate(results_data, start=1):
group_name = f"{idx}.1"
hdf5_data[subdir_name][group_name] = {
"@NX_class": "NXentry",
"description": os.path.basename(result["filename"]),
"measurements": {
"@NX_class": "NXcollection",
"q": result["q"].tolist(), "q": result["q"].tolist(),
"I": result["I"].tolist(), "I": result["I"].tolist(),
"dI": result["dI"].tolist() "dI": result["dI"].tolist()
} }
}
# Save to HDF5 file using silx
hdf5_file_path = os.path.join(subdir_path_int, f"{subdir_name}.h5")
dicttoh5(hdf5_data, hdf5_file_path, mode="w")
print(f"Results for subdirectory {subdir_name} saved to HDF5 file using silx.")
# Save to HDF5 file using silx # Save to HDF5 file using silx
hdf5_file_path = os.path.join(subdir_path_int, f"{subdir_name}.h5") hdf5_file_path = os.path.join(subdir_path_int, f"{subdir_name}.h5")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment