DiamondLightSource /
Savu
| @@ 118-132 (lines=15) @@ | ||
| 115 | ||
| 116 | return self.hdf5._open_backing_h5(fname, 'r') |
|
| 117 | ||
| 118 | def __get_start_slice_list(self, slice_dirs, shape, n_frames): |
|
| 119 | n_processes = len(self.exp.get('processes')) |
|
| 120 | rank = self.exp.get('process') |
|
| 121 | frames = np.array_split(np.arange(n_frames), n_processes)[rank] |
|
| 122 | f_range = list(range(0, frames[0])) if len(frames) else [] |
|
| 123 | sl = [slice(0, 1) if i in slice_dirs else slice(None) |
|
| 124 | for i in range(len(shape))] |
|
| 125 | idx = 0 |
|
| 126 | for i in f_range: |
|
| 127 | if sl[slice_dirs[idx]] == shape[slice_dirs[idx]]-1: |
|
| 128 | idx += 1 |
|
| 129 | tmp = sl[slice_dirs[idx]] |
|
| 130 | sl[slice_dirs[idx]] = slice(tmp.start+1, tmp.stop+1) |
|
| 131 | ||
| 132 | return sl, len(frames) |
|
| 133 | ||
| 134 | def __convert_patterns(self, data_obj): |
|
| 135 | pattern_list = self.parameters['patterns'] |
|
| @@ 171-185 (lines=15) @@ | ||
| 168 | ||
| 169 | return self.hdf5._open_backing_h5(fname, 'r') |
|
| 170 | ||
| 171 | def __get_start_slice_list(self, slice_dirs, shape, n_frames): |
|
| 172 | n_processes = len(self.exp.get('processes')) |
|
| 173 | rank = self.exp.get('process') |
|
| 174 | frames = np.array_split(np.arange(n_frames), n_processes)[rank] |
|
| 175 | f_range = list(range(0, frames[0])) if len(frames) else [] |
|
| 176 | sl = [slice(0, 1) if i in slice_dirs else slice(None) |
|
| 177 | for i in range(len(shape))] |
|
| 178 | idx = 0 |
|
| 179 | for i in f_range: |
|
| 180 | if sl[slice_dirs[idx]] == shape[slice_dirs[idx]]-1: |
|
| 181 | idx += 1 |
|
| 182 | tmp = sl[slice_dirs[idx]] |
|
| 183 | sl[slice_dirs[idx]] = slice(tmp.start+1, tmp.stop+1) |
|
| 184 | ||
| 185 | return sl, len(frames) |
|
| 186 | ||
| 187 | def __convert_patterns(self, data_obj, object_type): |
|
| 188 | if (object_type == 'synth_proj_data'): |
|
