def _set_transformation_values(self):
if isinstance(self.values, StreamModule):
nx_log = Group(name=self.name, parent_node=self) # type: ignore
nx_log.nx_class = LOG_CLASS_NAME
self.values.parent_node = nx_log
nx_log.children.append(self.values)
self.values = nx_log
def get_transforms_group(self) -> Group:
if self[TRANSFORMS_GROUP_NAME] is not None:
return self[TRANSFORMS_GROUP_NAME]
self[TRANSFORMS_GROUP_NAME] = Group(TRANSFORMS_GROUP_NAME,
parent_node=self)
self[TRANSFORMS_GROUP_NAME].nx_class = "NXtransformations"
return self[TRANSFORMS_GROUP_NAME]
def test_get_flags_transformation_list():
test_component_tree_model, test_instrument = create_component_tree_model(
[get_component()])
transformation_group = Group(name="transformations")
transformation_group.nx_class = NX_TRANSFORMATIONS
index = test_component_tree_model.createIndex(0, 0, transformation_group)
assert (test_component_tree_model.flags(index) == Qt.ItemIsEnabled
| Qt.ItemIsSelectable)
def _create_user(self, group: Group, user_data: Dict[str, str]) -> Group:
group.name = f"user_{user_data['name'].replace(' ', '')}"
group.nx_class = NX_USER
for name, value in user_data.items():
group.children.append(
Dataset(
name=name, parent_node=group, type=ValueTypes.STRING, values=value
)
)
return group
def _create_datasets_and_add_to_shape_group(self):
group = Group(name=SHAPE_GROUP_NAME)
group.nx_class = SHAPE_NX_CLASS
group.parent_node = self
new_child = create_fw_module_object(
WriterModules.DATASET.value,
self._get_dataset_config(self._size, SIZE),
group,
)
new_child.type = ValueTypes.DOUBLE
attributes = Attributes()
attributes.set_attribute_value(CommonAttrs.UNITS, self._units)
new_child.attributes = attributes
new_child.parent_node = group
group.children.append(new_child)
group[SHAPE_GROUP_NAME] = create_fw_module_object(
WriterModules.DATASET.value,
self._get_dataset_config(NX_BOX, SHAPE_GROUP_NAME),
group,
)
self[SHAPE_GROUP_NAME] = group