DiamondLightSource /
Savu
| @@ 600-623 (lines=24) @@ | ||
| 597 | else: |
|
| 598 | entry[name][...] = data |
|
| 599 | ||
| 600 | def __output_axis_labels(self, data, entry): |
|
| 601 | axis_labels = data.data_info.get("axis_labels") |
|
| 602 | ddict = data.meta_data.get_dictionary() |
|
| 603 | ||
| 604 | axes = [] |
|
| 605 | count = 0 |
|
| 606 | for labels in axis_labels: |
|
| 607 | name = list(labels.keys())[0] |
|
| 608 | axes.append(name) |
|
| 609 | entry.attrs[name + '_indices'] = count |
|
| 610 | ||
| 611 | mData = ddict[name] if name in list(ddict.keys()) \ |
|
| 612 | else np.arange(data.get_shape()[count]) |
|
| 613 | if isinstance(mData, list): |
|
| 614 | mData = np.array(mData) |
|
| 615 | ||
| 616 | if 'U' in str(mData.dtype): |
|
| 617 | mData = mData.astype(np.string_) |
|
| 618 | ||
| 619 | axis_entry = entry.require_dataset(name, mData.shape, mData.dtype) |
|
| 620 | axis_entry[...] = mData[...] |
|
| 621 | axis_entry.attrs['units'] = list(labels.values())[0] |
|
| 622 | count += 1 |
|
| 623 | entry.attrs['axes'] = axes |
|
| 624 | ||
| 625 | def __output_data_patterns(self, data, entry): |
|
| 626 | data_patterns = data.data_info.get("data_patterns") |
|
| @@ 272-295 (lines=24) @@ | ||
| 269 | plugin_entry.attrs['NX_class'] = 'NXdata' |
|
| 270 | ||
| 271 | ||
| 272 | def __output_axis_labels(self, data, entry): |
|
| 273 | axis_labels = data.data_info.get("axis_labels") |
|
| 274 | ddict = data.meta_data.get_dictionary() |
|
| 275 | ||
| 276 | axes = [] |
|
| 277 | count = 0 |
|
| 278 | for labels in axis_labels: |
|
| 279 | name = list(labels.keys())[0] |
|
| 280 | axes.append(name) |
|
| 281 | entry.attrs[name + '_indices'] = count |
|
| 282 | ||
| 283 | mData = ddict[name] if name in list(ddict.keys()) \ |
|
| 284 | else np.arange(self.parameters['proj_data_dims'][count]) |
|
| 285 | if isinstance(mData, list): |
|
| 286 | mData = np.array(mData) |
|
| 287 | ||
| 288 | if 'U' in str(mData.dtype): |
|
| 289 | mData = mData.astype(np.string_) |
|
| 290 | if name not in list(entry.keys()): |
|
| 291 | axis_entry = entry.require_dataset(name, mData.shape, mData.dtype) |
|
| 292 | axis_entry[...] = mData[...] |
|
| 293 | axis_entry.attrs['units'] = list(labels.values())[0] |
|
| 294 | count += 1 |
|
| 295 | entry.attrs['axes'] = axes |
|
| 296 | ||
| 297 | def __output_data_patterns(self, data, entry): |
|
| 298 | data_patterns = data.data_info.get("data_patterns") |
|
