diff --git a/leenkx/blender/lnx/nodes_logic.py b/leenkx/blender/lnx/nodes_logic.py index dc125bd..1babb94 100644 --- a/leenkx/blender/lnx/nodes_logic.py +++ b/leenkx/blender/lnx/nodes_logic.py @@ -77,6 +77,25 @@ class LNX_MT_NodeAddOverride(bpy.types.Menu): layout.separator() layout.menu(f'LNX_MT_{INTERNAL_GROUPS_MENU_ID}_menu', text=internal_groups_menu_class.bl_label, icon='OUTLINER_OB_GROUP_INSTANCE') + elif context.space_data.tree_type == 'ShaderNodeTree': + # TO DO - Recursively gather nodes and draw them to menu + + LNX_MT_NodeAddOverride.overridden_draw(self, context) + + layout = self.layout + layout.separator() + layout.separator() + col = layout.column() + col.label(text="Custom") + + shader_data_op = col.operator("node.add_node", text="Shader Data") + shader_data_op.type = "LnxShaderDataNode" + shader_data_op.use_transform = True + + particle_op = col.operator("node.add_node", text="Custom Particle") + particle_op.type = "LnxCustomParticleNode" + particle_op.use_transform = True + else: LNX_MT_NodeAddOverride.overridden_draw(self, context)