Skip to content
Snippets Groups Projects
Commit c5158124 authored by Alexis GAMELIN's avatar Alexis GAMELIN
Browse files

[Fix] merge_files

Fix for BeamSpectrumMonitor and BunchSpectrumMonitor with merge_files
parent c523269b
No related branches found
No related tags found
No related merge requests found
...@@ -39,6 +39,9 @@ def merge_files(files_prefix, files_number, file_name=None): ...@@ -39,6 +39,9 @@ def merge_files(files_prefix, files_number, file_name=None):
for group in list(f0): for group in list(f0):
f.require_group(group) f.require_group(group)
for dataset_name in list(f0[group]): for dataset_name in list(f0[group]):
if dataset_name == "freq":
f0[group].copy(dataset_name, f[group])
continue
shape = f0[group][dataset_name].shape shape = f0[group][dataset_name].shape
dtype = f0[group][dataset_name].dtype dtype = f0[group][dataset_name].dtype
shape_needed = list(shape) shape_needed = list(shape)
...@@ -60,6 +63,8 @@ def merge_files(files_prefix, files_number, file_name=None): ...@@ -60,6 +63,8 @@ def merge_files(files_prefix, files_number, file_name=None):
for n in range(n_slice): for n in range(n_slice):
slice_list.append(slice(None)) slice_list.append(slice(None))
slice_list.append(slice(length*i,length*(i+1))) slice_list.append(slice(length*i,length*(i+1)))
if (dataset_name == "freq"):
continue
if (dataset_name == "time") and (i != 0): if (dataset_name == "time") and (i != 0):
f[group][dataset_name][tuple(slice_list)] = f[group][dataset_name][(length*i) - 1] + fi[group][dataset_name] f[group][dataset_name][tuple(slice_list)] = f[group][dataset_name][(length*i) - 1] + fi[group][dataset_name]
else: else:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment