Update Files

leenkx/blender/lnx/logicnode/miscellaneous/LN_group_output.py

@@ -0,0 +1,217 @@
+import bpy
+
+import lnx.utils
+import lnx.node_utils
+from lnx.logicnode.lnx_nodes import *
+import lnx.logicnode.miscellaneous.LN_call_group as LN_call_group
+
+
+class GroupOutputsNode(LnxLogicTreeNode):
+    """Output for a node group."""
+    bl_idname = 'LNGroupOutputsNode'
+    bl_label = 'Group Output Node'
+    lnx_section = 'group'
+    lnx_version = 3
+
+    def __init__(self):
+        self.register_id()
+
+    # Active socket selected
+    active_input: IntProperty(name='active_input', description='', default=0)
+
+    # Flag to store invalid links
+    invalid_link: BoolProperty(name='invalid_link', description='', default=False)
+
+    # Override copy prevention in certain situations such as copying entire group
+    copy_override: BoolProperty(name='copy override', description='', default=False)
+
+    def init(self, context):
+        tree = bpy.context.space_data.edit_tree
+        node_count = 0
+        for node in tree.nodes:
+            if node.bl_idname == 'LNGroupOutputsNode':
+                node_count += 1
+        if node_count > 1:
+            lnx.log.warn("Only one group output node per node tree is allowed")
+            self.mute = True
+        else:
+            super().init(context)
+
+    # Prevent copying of group node
+    def copy(self, node):
+        if not self.copy_override:
+            self.mute = True
+            self.inputs.clear()
+        self.copy_override = False
+
+    def lnx_init(self, context):
+        if not self.mute:
+            self.add_socket()
+
+    # Called when link is created
+    def insert_link(self, link):
+        to_socket = link.to_socket
+        from_node = link.from_node
+        from_socket = None
+        # Recursively search for other socket in case of reroutes
+        if from_node.type == 'REROUTE':
+            _, from_socket = lnx.node_utils.input_get_connected_node(to_socket)
+        else:
+            from_socket = link.from_socket
+        if from_socket is not None:
+            index = self.get_socket_index(to_socket)
+            # If socket connected to LnxAnySocket, link is invalid
+            if from_socket.bl_idname == 'LnxAnySocket':
+                self.invalid_link = True
+            else:
+                call_group_nodes = self.get_call_group_nodes()
+                for node in call_group_nodes:
+                    # Change socket type according to the new link
+                    node.change_output_socket(from_socket.bl_idname, index, link.to_socket.display_label)
+
+    # Use update method to remove invalid links
+    def update(self):
+        super().update()
+        if self.invalid_link:
+            self.remove_invalid_links()
+
+    # Called when name of the socket is changed
+    def socket_name_update(self, socket):
+        index = self.get_socket_index(socket)
+        # Update socket names of the related call group nodes
+        call_node_groups = self.get_call_group_nodes()
+        for node in call_node_groups:
+            out_socket = node.outputs[index]
+            if out_socket.bl_idname == 'LnxAnySocket':
+                out_socket.display_label = socket.display_label
+            else:
+                out_socket.name = socket.display_label
+
+    # Recursively search and remove invalid links
+    def remove_invalid_links(self):
+        for input in self.inputs:
+            for link in input.links:
+                if link.from_socket.bl_idname == 'LnxAnySocket':
+                    tree = self.get_tree()
+                    tree.links.remove(link)
+                    break
+        self.invalid_link = False
+
+    # Function to move socket up and handle the same in related call group nodes
+    def move_socket_up(self):
+        if self.active_input > 0:
+            self.inputs.move(self.active_input, self.active_input - 1)
+            call_node_groups = self.get_call_group_nodes()
+            for nodes in call_node_groups:
+                nodes.outputs.move(self.active_input, self.active_input - 1)
+            self.active_input = self.active_input - 1
+
+    # Function to move socket down and handle the same in related call group nodes
+    def move_socket_down(self):
+        if self.active_input < len(self.inputs) - 1:
+            self.inputs.move(self.active_input, self.active_input + 1)
+            call_node_groups = self.get_call_group_nodes()
+            for nodes in call_node_groups:
+                nodes.outputs.move(self.active_input, self.active_input + 1)
+            self.active_input = self.active_input + 1
+
+    # Function to recursively get related call group nodes
+    def get_call_group_nodes(self):
+        call_group_nodes = []
+        # Return empty list if node is muted
+        if self.mute:
+            return call_group_nodes
+        for tree in bpy.data.node_groups:
+            if tree.bl_idname == "LnxLogicTreeType" or tree.bl_idname == "LnxGroupTree":
+                for node in tree.nodes:
+                    if node.bl_idname == 'LNCallGroupNode':
+                        if node.group_tree == self.get_tree():
+                            call_group_nodes.append(node)
+        return call_group_nodes
+
+    # Function to add a socket and handle the same in the related call group nodes
+    def add_socket(self):
+        self.add_input('LnxAnySocket','',)
+        call_group_nodes = self.get_call_group_nodes()
+        for node in call_group_nodes:
+            node.add_output('LnxAnySocket','')
+
+    # Function to remove a socket and handle the same in the related call group nodes
+    def remove_socket(self):
+        self.inputs.remove(self.inputs[-1])
+        call_group_nodes = self.get_call_group_nodes()
+        for node in call_group_nodes:
+            node.outputs.remove(node.outputs[-1])
+        if self.active_input > len(self.inputs) - 1:
+            self.active_input = self.active_input - 1
+
+    # Function to add a socket at certain index and 
+    # handle the same in the related call group nodes
+    def add_socket_ext(self):
+        index = self.active_input + 1
+        self.insert_input('LnxAnySocket', index, '')
+        call_group_nodes = self.get_call_group_nodes()
+        for node in call_group_nodes:
+            node.insert_output('LnxAnySocket', index, '')
+
+    # Function to remove a socket at certain index and 
+    # handle the same in the related call group nodes
+    def remove_socket_ext(self):
+        self.inputs.remove(self.inputs[self.active_input])
+        call_group_nodes = self.get_call_group_nodes()
+        for node in call_group_nodes:
+            node.outputs.remove(node.outputs[self.active_input])
+        if self.active_input > len(self.inputs) - 1:
+            self.active_input = len(self.inputs) - 1
+
+    # Handle deletion of group input node
+    def free(self):
+        call_group_nodes = self.get_call_group_nodes()
+        for node in call_group_nodes:
+            node.outputs.clear()
+
+    # Draw node UI
+    def draw_buttons(self, context, layout):
+        if self.mute:
+            layout.enabled = False
+        row = layout.row(align=True)
+        op = row.operator('lnx.node_call_func', text='New', icon='PLUS', emboss=True)
+        op.node_index = self.get_id_str()
+        op.callback_name = 'add_socket'
+        if len(self.inputs) > 1:
+            op2 = row.operator('lnx.node_call_func', text='', icon='X', emboss=True)
+            op2.node_index = self.get_id_str()
+            op2.callback_name = 'remove_socket'
+
+    # Draw side panel UI
+    def draw_buttons_ext(self, context, layout):
+        if self.mute:
+            layout.enabled = False
+        node = context.active_node
+        split = layout.row()
+        split.template_list('LNX_UL_InterfaceSockets', 'IN', node, 'inputs', node, 'active_input')
+        ops_col = split.column()
+        add_remove_col = ops_col.column(align=True)
+        props = add_remove_col.operator('lnx.node_call_func', icon='ADD', text="")
+        props.node_index = self.get_id_str()
+        props.callback_name = 'add_socket_ext'
+        if len(self.inputs) > 1:
+            props = add_remove_col.operator('lnx.node_call_func', icon='REMOVE', text="")
+            props.node_index = self.get_id_str()
+            props.callback_name = 'remove_socket_ext'
+
+        ops_col.separator()
+
+        up_down_col = ops_col.column(align=True)
+        props = up_down_col.operator('lnx.node_call_func', icon='TRIA_UP', text="")
+        props.node_index = self.get_id_str()
+        props.callback_name = 'move_socket_up'
+        props = up_down_col.operator('lnx.node_call_func', icon='TRIA_DOWN', text="")
+        props.node_index = self.get_id_str()
+        props.callback_name = 'move_socket_down'
+
+    def get_replacement_node(self, node_tree: bpy.types.NodeTree):
+        if self.lnx_version not in (0, 1, 2):
+            raise LookupError()
+
+        return node_tree.nodes.new('LNGroupOutputsNode')