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") |
|
| @@ 290-313 (lines=24) @@ | ||
| 287 | plugin_entry.attrs['NX_class'] = 'NXdata' |
|
| 288 | ||
| 289 | ||
| 290 | def __output_axis_labels(self, data, entry): |
|
| 291 | axis_labels = data.data_info.get("axis_labels") |
|
| 292 | ddict = data.meta_data.get_dictionary() |
|
| 293 | ||
| 294 | axes = [] |
|
| 295 | count = 0 |
|
| 296 | for labels in axis_labels: |
|
| 297 | name = list(labels.keys())[0] |
|
| 298 | axes.append(name) |
|
| 299 | entry.attrs[name + '_indices'] = count |
|
| 300 | ||
| 301 | mData = ddict[name] if name in list(ddict.keys()) \ |
|
| 302 | else np.arange(self.parameters['proj_data_dims'][count]) |
|
| 303 | if isinstance(mData, list): |
|
| 304 | mData = np.array(mData) |
|
| 305 | ||
| 306 | if 'U' in str(mData.dtype): |
|
| 307 | mData = mData.astype(np.string_) |
|
| 308 | if name not in list(entry.keys()): |
|
| 309 | axis_entry = entry.require_dataset(name, mData.shape, mData.dtype) |
|
| 310 | axis_entry[...] = mData[...] |
|
| 311 | axis_entry.attrs['units'] = list(labels.values())[0] |
|
| 312 | count += 1 |
|
| 313 | entry.attrs['axes'] = axes |
|
| 314 | ||
| 315 | def __output_data_patterns(self, data, entry): |
|
| 316 | data_patterns = data.data_info.get("data_patterns") |
|
