def _build_nodes(self, tree, factories, imported_structs): """Build nodes of the main tree, other dependencies should be already initialized""" with tree.init_tree(): # first all nodes should be created without applying their inner data # because some nodes can have `parent` property which points into another node node_structs = [] for node_name, raw_structure in self._struct["main_tree"]["nodes"].items(): with self.logger.add_fail("Init node (main tree)", f"Name: {node_name}"): node_struct = factories.node(node_name, self.logger, raw_structure) # register optional node classes if old_nodes.is_old(node_struct.read_bl_type()): old_nodes.register_old(node_struct.read_bl_type()) if dummy_nodes.is_dependent(node_struct.read_bl_type()): dummy_nodes.register_dummy(node_struct.read_bl_type()) # add node an save its new name node = tree.nodes.new(node_struct.read_bl_type()) node.name = node_name imported_structs[(StrTypes.NODE, tree.name, node_name)] = node.name node_structs.append(node_struct) for node_struct in node_structs: with self.logger.add_fail("Build node (main tree)", f"Name {node_struct.name}"): new_name = imported_structs[(StrTypes.NODE, tree.name, node_struct.name)] node = tree.nodes[new_name] node_struct.build(node, factories, imported_structs) for raw_struct in self._struct["main_tree"]["links"]: with self.logger.add_fail("Build link (main tree)", f"Struct: {raw_struct}"): factories.link(None, self.logger, raw_struct).build(tree, factories, imported_structs)
def build(self, tree, factories: StructFactory, imported_structs: OldNewNames): """Reads and generates nodes, links, dependent data blocks""" with tree.init_tree(): # first all nodes should be created without applying their inner data # because some nodes can have `parent` property which points into another node node_structs = [] for node_name, raw_structure in self._struct["nodes"].items(): with self.logger.add_fail("Init node", f"Tree: {tree.name}, Node: {node_name}"): node_struct = factories.node(node_name, self.logger, raw_structure) # register optional node classes if old_nodes.is_old(node_struct.read_bl_type()): old_nodes.register_old(node_struct.read_bl_type()) if dummy_nodes.is_dependent(node_struct.read_bl_type()): dummy_nodes.register_dummy(node_struct.read_bl_type()) # add node an save its new name node = tree.nodes.new(node_struct.read_bl_type()) node.name = node_name imported_structs[(StrTypes.NODE, tree.name, node_name)] = node.name node_structs.append(node_struct) for node_struct in node_structs: with self.logger.add_fail("Build node", f"Tree: {tree.name}, Node: {node_struct.name}"): new_name = imported_structs[(StrTypes.NODE, tree.name, node_struct.name)] node = tree.nodes[new_name] node_struct.build(node, factories, imported_structs) for raw_struct in self._struct["links"]: with self.logger.add_fail("Build link", f"Tree: {tree.name}, Struct: {raw_struct}"): factories.link(None, self.logger, raw_struct).build(tree, factories, imported_structs) for prop_name, prop_value in self._struct.get("properties", dict()).items(): with self.logger.add_fail("Setting tree property", f'Tree: {node.id_data.name}, prop: {prop_name}'): factories.prop(prop_name, self.logger, prop_value).build(tree, factories, imported_structs)
def add_node(self, bl_type: str, node_name: str) -> Union[SverchCustomTreeNode, None]: """ Trying to add node with given bl_idname into given tree Also it can register dummy and old nodes and register fails """ with self._fails_log.add_fail("Creating node", f'Tree: {self._tree_name}, Node: {node_name}'): if old_nodes.is_old(bl_type): # old node classes are registered only by request old_nodes.register_old(bl_type) if dummy_nodes.is_dependent(bl_type): # some node types are not registered if dependencies are not installed # in this case such nodes are registered as dummies dummy_nodes.register_dummy(bl_type) node = self._tree.nodes.new(bl_type) node.name = node_name return node
def add_node_to_tree(nodes, n, nodes_to_import, name_remap, create_texts): node_ref = nodes_to_import[n] bl_idname = node_ref['bl_idname'] try: if old_nodes.is_old(bl_idname): old_nodes.register_old(bl_idname) if bl_idname == 'SvMonadGenericNode': node = unpack_monad(nodes, node_ref) if not node: raise Exception( "It seems no valid node was created for this Monad {0}". format(node_ref)) else: if dummy_nodes.is_dependent(bl_idname): try: node = nodes.new(bl_idname) except RuntimeError: dummy_nodes.register_dummy(bl_idname) node = nodes.new(bl_idname) else: node = nodes.new(bl_idname) except Exception as err: exception(err) error('%s not currently registered, skipping', bl_idname) return # fix_enum_identifier_spaces_if_needed(node, node_ref) if create_texts: add_texts(node, node_ref) if hasattr(node, 'storage_set_data'): node.storage_set_data(node_ref) if bl_idname == 'SvObjectsNodeMK3': for named_object in node_ref.get('object_names', []): node.object_names.add().name = named_object gather_remapped_names(node, n, name_remap) import_node_settings(node, node_ref)