ce88d773db
Adding panel toggles in nodegroups have somewhat of a UX antipattern. When running the operator, it checks for conditions that indicate it should not run, and if those are hit, it cancels execution and mentions the invalid condition in the footer bar. This is not ideal, the user should not have to call the operator to find out whether it can be called. Why it got implemented like this is likely a consequence of all interface items being the same "New Item" operator. Poll functions cannot use operator properties, so variants of the same operator cannot check for different conditions for execution. This is a problem for panel toggles, as they have more restrictions to when they can be added that don't apply to other interface items. This patch creates a separate operator for adding panel toggles. This allows the condition checks to be implemented in the poll function, which enables greying out the operator buttons and showing on tooltips what condition is invalid. Pull Request: https://projects.blender.org/blender/blender/pulls/146379
1221 lines
40 KiB
Python
1221 lines
40 KiB
Python
# SPDX-FileCopyrightText: 2009-2023 Blender Authors
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
import bpy
|
|
import rna_prop_ui
|
|
|
|
from bpy.types import (
|
|
Header,
|
|
Menu,
|
|
Panel,
|
|
)
|
|
from bpy.app.translations import (
|
|
pgettext_iface as iface_,
|
|
contexts as i18n_contexts,
|
|
)
|
|
from bl_ui import anim, node_add_menu
|
|
from bl_ui.utils import PresetPanel
|
|
from bl_ui.properties_grease_pencil_common import (
|
|
AnnotationDataPanel,
|
|
)
|
|
from bl_ui.space_toolsystem_common import (
|
|
ToolActivePanelHelper,
|
|
)
|
|
from bl_ui.properties_material import (
|
|
EEVEE_MATERIAL_PT_settings,
|
|
EEVEE_MATERIAL_PT_settings_surface,
|
|
EEVEE_MATERIAL_PT_settings_volume,
|
|
MATERIAL_PT_viewport,
|
|
)
|
|
from bl_ui.properties_world import (
|
|
WORLD_PT_viewport_display
|
|
)
|
|
from bl_ui.properties_data_light import (
|
|
DATA_PT_light,
|
|
DATA_PT_EEVEE_light,
|
|
)
|
|
|
|
|
|
class NODE_HT_header(Header):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
scene = context.scene
|
|
snode = context.space_data
|
|
overlay = snode.overlay
|
|
snode_id = snode.id
|
|
id_from = snode.id_from
|
|
tool_settings = context.tool_settings
|
|
is_compositor = snode.tree_type == 'CompositorNodeTree'
|
|
|
|
layout.template_header()
|
|
|
|
# Now expanded via the `ui_type`.
|
|
# layout.prop(snode, "tree_type", text="")
|
|
|
|
display_pin = True
|
|
if snode.tree_type == 'ShaderNodeTree':
|
|
layout.prop(snode, "shader_type", text="")
|
|
|
|
ob = context.object
|
|
if snode.shader_type == 'OBJECT' and ob:
|
|
ob_type = ob.type
|
|
|
|
NODE_MT_editor_menus.draw_collapsible(context, layout)
|
|
types_that_support_material = {
|
|
'MESH', 'CURVE', 'SURFACE', 'FONT', 'META', 'GPENCIL', 'VOLUME', 'CURVES', 'POINTCLOUD',
|
|
}
|
|
|
|
if snode_id:
|
|
row = layout.row()
|
|
if ob_type not in types_that_support_material:
|
|
row.prop(snode_id, "use_nodes")
|
|
|
|
layout.separator_spacer()
|
|
|
|
# disable material slot buttons when pinned, cannot find correct slot within id_from (#36589)
|
|
# disable also when the selected object does not support materials
|
|
has_material_slots = not snode.pin and ob_type in types_that_support_material
|
|
|
|
if ob_type != 'LIGHT':
|
|
row = layout.row()
|
|
row.enabled = has_material_slots
|
|
row.ui_units_x = 4
|
|
row.popover(panel="NODE_PT_material_slots")
|
|
|
|
row = layout.row()
|
|
row.enabled = has_material_slots
|
|
|
|
# Show material.new when no active ID/slot exists
|
|
if not id_from and ob_type in types_that_support_material:
|
|
row.template_ID(ob, "active_material", new="material.new")
|
|
# Material ID, but not for Lights
|
|
if id_from and ob_type != 'LIGHT':
|
|
row.template_ID(id_from, "active_material", new="material.new")
|
|
|
|
if snode.shader_type == 'WORLD':
|
|
NODE_MT_editor_menus.draw_collapsible(context, layout)
|
|
world = scene.world
|
|
|
|
if snode_id:
|
|
row = layout.row()
|
|
if world and world.use_eevee_finite_volume:
|
|
row.operator("world.convert_volume_to_mesh", emboss=False, icon='WORLD', text="Convert Volume")
|
|
|
|
layout.separator_spacer()
|
|
|
|
row = layout.row()
|
|
row.enabled = not snode.pin
|
|
row.template_ID(scene, "world", new="world.new")
|
|
|
|
if snode.shader_type == 'LINESTYLE':
|
|
view_layer = context.view_layer
|
|
lineset = view_layer.freestyle_settings.linesets.active
|
|
|
|
if lineset is not None:
|
|
NODE_MT_editor_menus.draw_collapsible(context, layout)
|
|
|
|
if snode_id:
|
|
row = layout.row()
|
|
row.prop(snode_id, "use_nodes")
|
|
|
|
layout.separator_spacer()
|
|
|
|
row = layout.row()
|
|
row.enabled = not snode.pin
|
|
row.template_ID(lineset, "linestyle", new="scene.freestyle_linestyle_new")
|
|
|
|
elif snode.tree_type == 'TextureNodeTree':
|
|
layout.prop(snode, "texture_type", text="")
|
|
|
|
NODE_MT_editor_menus.draw_collapsible(context, layout)
|
|
|
|
if snode_id:
|
|
layout.prop(snode_id, "use_nodes")
|
|
|
|
layout.separator_spacer()
|
|
|
|
if id_from:
|
|
if snode.texture_type == 'BRUSH':
|
|
layout.template_ID(id_from, "texture", new="texture.new")
|
|
else:
|
|
layout.template_ID(id_from, "active_texture", new="texture.new")
|
|
|
|
elif snode.tree_type == 'CompositorNodeTree':
|
|
layout.prop(snode, "node_tree_sub_type", text="")
|
|
NODE_MT_editor_menus.draw_collapsible(context, layout)
|
|
layout.separator_spacer()
|
|
|
|
if snode.node_tree_sub_type == 'SCENE':
|
|
row = layout.row()
|
|
row.enabled = not snode.pin
|
|
if scene.compositing_node_group:
|
|
row.template_ID(scene, "compositing_node_group", new="node.duplicate_compositing_node_group")
|
|
else:
|
|
row.template_ID(scene, "compositing_node_group", new="node.new_compositing_node_group")
|
|
elif snode.node_tree_sub_type == 'SEQUENCER':
|
|
row = layout.row()
|
|
sequencer_scene = context.workspace.sequencer_scene
|
|
sequencer_editor = sequencer_scene.sequence_editor if sequencer_scene else None
|
|
active_strip = sequencer_editor.active_strip if sequencer_editor else None
|
|
active_modifier = active_strip.modifiers.active if active_strip else None
|
|
is_compositor_modifier_active = active_modifier and active_modifier.type == 'COMPOSITOR'
|
|
if is_compositor_modifier_active and not snode.pin:
|
|
row.template_ID(
|
|
active_modifier,
|
|
"node_group",
|
|
new="node.new_compositor_sequencer_node_group")
|
|
else:
|
|
row.enabled = False
|
|
row.template_ID(snode, "node_tree", new="node.new_compositor_sequencer_node_group")
|
|
|
|
elif snode.tree_type == 'GeometryNodeTree':
|
|
layout.prop(snode, "node_tree_sub_type", text="")
|
|
NODE_MT_editor_menus.draw_collapsible(context, layout)
|
|
layout.separator_spacer()
|
|
|
|
if snode.node_tree_sub_type == 'MODIFIER':
|
|
ob = context.object
|
|
|
|
row = layout.row()
|
|
if snode.pin:
|
|
row.enabled = False
|
|
row.template_ID(snode, "node_tree", new="node.new_geometry_node_group_assign")
|
|
elif ob:
|
|
active_modifier = ob.modifiers.active
|
|
if active_modifier and active_modifier.type == 'NODES':
|
|
if active_modifier.node_group:
|
|
row.template_ID(active_modifier, "node_group", new="object.geometry_node_tree_copy_assign")
|
|
else:
|
|
row.template_ID(active_modifier, "node_group", new="node.new_geometry_node_group_assign")
|
|
else:
|
|
row.template_ID(snode, "node_tree", new="node.new_geometry_nodes_modifier")
|
|
else:
|
|
layout.template_ID(snode, "selected_node_group", new="node.new_geometry_node_group_tool")
|
|
if snode.node_tree:
|
|
layout.popover(panel="NODE_PT_geometry_node_tool_object_types", text="Types")
|
|
layout.popover(panel="NODE_PT_geometry_node_tool_mode", text="Modes")
|
|
layout.popover(panel="NODE_PT_geometry_node_tool_options", text="Options")
|
|
display_pin = False
|
|
else:
|
|
# Custom node tree is edited as independent ID block
|
|
NODE_MT_editor_menus.draw_collapsible(context, layout)
|
|
|
|
layout.separator_spacer()
|
|
|
|
layout.template_ID(snode, "node_tree", new="node.new_node_tree")
|
|
|
|
# Put pin next to ID block
|
|
if display_pin:
|
|
layout.prop(snode, "pin", text="", emboss=False)
|
|
|
|
layout.separator_spacer()
|
|
|
|
if len(snode.path) > 1:
|
|
op = layout.operator("node.tree_path_parent", text="", icon='FILE_PARENT')
|
|
op.parent_tree_index = len(snode.path) - 2
|
|
|
|
# Backdrop
|
|
if is_compositor and snode.node_tree_sub_type == 'SCENE':
|
|
row = layout.row(align=True)
|
|
row.prop(snode, "show_backdrop", toggle=True)
|
|
row.active = snode.node_tree is not None
|
|
sub = row.row(align=True)
|
|
sub.active = snode.show_backdrop
|
|
sub.prop(snode, "backdrop_channels", icon_only=True, text="")
|
|
|
|
# Gizmo toggle and popover.
|
|
row = layout.row(align=True)
|
|
row.prop(snode, "show_gizmo", icon='GIZMO', text="")
|
|
row.active = snode.node_tree is not None
|
|
sub = row.row(align=True)
|
|
sub.active = snode.show_gizmo and row.active
|
|
sub.popover(panel="NODE_PT_gizmo_display", text="")
|
|
|
|
# Snap
|
|
row = layout.row(align=True)
|
|
row.prop(tool_settings, "use_snap_node", text="")
|
|
row.active = snode.node_tree is not None
|
|
|
|
# Overlay toggle & popover
|
|
row = layout.row(align=True)
|
|
row.prop(overlay, "show_overlays", icon='OVERLAY', text="")
|
|
sub = row.row(align=True)
|
|
row.active = snode.node_tree is not None
|
|
sub.active = overlay.show_overlays and row.active
|
|
sub.popover(panel="NODE_PT_overlay", text="")
|
|
|
|
|
|
class NODE_PT_gizmo_display(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'HEADER'
|
|
bl_label = "Gizmos"
|
|
bl_ui_units_x = 8
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
snode = context.space_data
|
|
is_compositor = snode.tree_type == 'CompositorNodeTree'
|
|
|
|
if not is_compositor:
|
|
return
|
|
|
|
col = layout.column()
|
|
col.label(text="Viewport Gizmos")
|
|
col.separator()
|
|
|
|
col.active = snode.show_gizmo
|
|
colsub = col.column()
|
|
colsub.active = snode.node_tree is not None and col.active
|
|
colsub.prop(snode, "show_gizmo_active_node", text="Active Node")
|
|
|
|
|
|
class NODE_MT_editor_menus(Menu):
|
|
bl_idname = "NODE_MT_editor_menus"
|
|
bl_label = ""
|
|
|
|
def draw(self, _context):
|
|
layout = self.layout
|
|
layout.menu("NODE_MT_view")
|
|
layout.menu("NODE_MT_select")
|
|
layout.menu("NODE_MT_add")
|
|
layout.menu("NODE_MT_node")
|
|
|
|
|
|
class NODE_MT_add(node_add_menu.AddNodeMenu):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_label = "Add"
|
|
bl_translation_context = i18n_contexts.operator_default
|
|
bl_options = {'SEARCH_ON_KEY_PRESS'}
|
|
|
|
def draw(self, context):
|
|
import nodeitems_utils
|
|
|
|
layout = self.layout
|
|
|
|
if layout.operator_context == 'EXEC_REGION_WIN':
|
|
layout.operator_context = 'INVOKE_REGION_WIN'
|
|
layout.operator("WM_OT_search_single_menu", text="Search...", icon='VIEWZOOM').menu_idname = "NODE_MT_add"
|
|
layout.separator()
|
|
|
|
layout.operator_context = 'INVOKE_REGION_WIN'
|
|
|
|
snode = context.space_data
|
|
if snode.tree_type == 'GeometryNodeTree':
|
|
layout.menu_contents("NODE_MT_geometry_node_add_all")
|
|
elif snode.tree_type == 'CompositorNodeTree':
|
|
layout.menu_contents("NODE_MT_compositor_node_add_all")
|
|
elif snode.tree_type == 'ShaderNodeTree':
|
|
layout.menu_contents("NODE_MT_shader_node_add_all")
|
|
elif snode.tree_type == 'TextureNodeTree':
|
|
layout.menu_contents("NODE_MT_texture_node_add_all")
|
|
elif nodeitems_utils.has_node_categories(context):
|
|
# Actual node sub-menus are defined by draw functions from node categories.
|
|
nodeitems_utils.draw_node_categories_menu(self, context)
|
|
|
|
|
|
class NODE_MT_swap(node_add_menu.SwapNodeMenu):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_label = "Swap"
|
|
bl_translation_context = i18n_contexts.operator_default
|
|
bl_options = {'SEARCH_ON_KEY_PRESS'}
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
if layout.operator_context == 'EXEC_REGION_WIN':
|
|
layout.operator_context = 'INVOKE_REGION_WIN'
|
|
layout.operator("WM_OT_search_single_menu", text="Search...", icon='VIEWZOOM').menu_idname = "NODE_MT_swap"
|
|
layout.separator()
|
|
|
|
layout.operator_context = 'INVOKE_REGION_WIN'
|
|
|
|
snode = context.space_data
|
|
if snode.tree_type == 'GeometryNodeTree':
|
|
layout.menu_contents("NODE_MT_geometry_node_swap_all")
|
|
elif snode.tree_type == 'CompositorNodeTree':
|
|
layout.menu_contents("NODE_MT_compositor_node_swap_all")
|
|
elif snode.tree_type == 'ShaderNodeTree':
|
|
layout.menu_contents("NODE_MT_shader_node_swap_all")
|
|
elif snode.tree_type == 'TextureNodeTree':
|
|
layout.menu_contents("NODE_MT_texture_node_swap_all")
|
|
|
|
|
|
class NODE_MT_view(Menu):
|
|
bl_label = "View"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
snode = context.space_data
|
|
is_compositor = snode.tree_type == 'CompositorNodeTree'
|
|
|
|
layout.prop(snode, "show_region_toolbar")
|
|
layout.prop(snode, "show_region_ui")
|
|
|
|
if is_compositor:
|
|
layout.prop(snode, "show_region_asset_shelf")
|
|
|
|
layout.separator()
|
|
|
|
sub = layout.column()
|
|
sub.operator_context = 'EXEC_REGION_WIN'
|
|
sub.operator("view2d.zoom_in")
|
|
sub.operator("view2d.zoom_out")
|
|
|
|
layout.separator()
|
|
|
|
layout.operator_context = 'INVOKE_REGION_WIN'
|
|
layout.operator("node.view_selected")
|
|
layout.operator("node.view_all")
|
|
|
|
if context.space_data.show_backdrop:
|
|
layout.separator()
|
|
|
|
layout.operator("node.backimage_move", text="Backdrop Move")
|
|
layout.operator("node.backimage_zoom", text="Backdrop Zoom In").factor = 1.2
|
|
layout.operator("node.backimage_zoom", text="Backdrop Zoom Out").factor = 1.0 / 1.2
|
|
layout.operator("node.backimage_fit", text="Fit Backdrop to Available Space")
|
|
|
|
layout.separator()
|
|
|
|
layout.menu("INFO_MT_area")
|
|
|
|
|
|
class NODE_MT_select(Menu):
|
|
bl_label = "Select"
|
|
|
|
def draw(self, _context):
|
|
layout = self.layout
|
|
|
|
layout.operator("node.select_all", text="All").action = 'SELECT'
|
|
layout.operator("node.select_all", text="None").action = 'DESELECT'
|
|
layout.operator("node.select_all", text="Invert").action = 'INVERT'
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.select_box").tweak = False
|
|
layout.operator("node.select_circle")
|
|
layout.operator_menu_enum("node.select_lasso", "mode")
|
|
|
|
layout.separator()
|
|
layout.operator("node.select_linked_from", text="Linked from")
|
|
layout.operator("node.select_linked_to", text="Linked to")
|
|
|
|
layout.separator()
|
|
|
|
layout.operator_menu_enum("node.select_grouped", "type", text="Select Grouped")
|
|
layout.operator("node.select_same_type_step", text="Activate Same Type Previous").prev = True
|
|
layout.operator("node.select_same_type_step", text="Activate Same Type Next").prev = False
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.find_node", text="Find Node...")
|
|
|
|
|
|
class NODE_MT_node(Menu):
|
|
bl_label = "Node"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
snode = context.space_data
|
|
group = snode.edit_tree
|
|
is_compositor = snode.tree_type == 'CompositorNodeTree'
|
|
|
|
layout.operator("transform.translate").view2d_edge_pan = True
|
|
layout.operator("transform.rotate")
|
|
layout.operator("transform.resize")
|
|
|
|
layout.separator()
|
|
layout.operator("node.clipboard_copy", text="Copy", icon='COPYDOWN')
|
|
layout.operator_context = 'EXEC_DEFAULT'
|
|
layout.operator("node.clipboard_paste", text="Paste", icon='PASTEDOWN')
|
|
layout.operator_context = 'INVOKE_REGION_WIN'
|
|
props = layout.operator("node.duplicate_move", icon='DUPLICATE')
|
|
props.NODE_OT_translate_attach.TRANSFORM_OT_translate.view2d_edge_pan = True
|
|
props = layout.operator("node.duplicate_move_linked")
|
|
props.NODE_OT_translate_attach.TRANSFORM_OT_translate.view2d_edge_pan = True
|
|
|
|
layout.separator()
|
|
layout.operator("node.delete", icon='X')
|
|
layout.operator("node.delete_reconnect")
|
|
|
|
layout.separator()
|
|
layout.operator("node.join", text="Join in New Frame")
|
|
layout.operator("node.detach", text="Remove from Frame")
|
|
layout.operator("node.join_nodes", text="Join Group Inputs")
|
|
layout.operator("node.join_named")
|
|
|
|
layout.separator()
|
|
props = layout.operator("wm.call_panel", text="Rename...")
|
|
props.name = "TOPBAR_PT_name"
|
|
props.keep_open = False
|
|
|
|
layout.separator()
|
|
layout.operator("node.link_make").replace = False
|
|
layout.operator("node.link_make", text="Make and Replace Links").replace = True
|
|
layout.operator("node.links_cut")
|
|
layout.operator("node.links_detach")
|
|
layout.operator("node.links_mute")
|
|
|
|
if not group or group.bl_use_group_interface:
|
|
layout.separator()
|
|
layout.operator("node.group_make", icon='NODETREE')
|
|
layout.operator("node.group_insert", text="Insert Into Group")
|
|
layout.operator("node.group_edit").exit = False
|
|
layout.operator("node.group_ungroup")
|
|
|
|
layout.separator()
|
|
layout.menu("NODE_MT_swap")
|
|
layout.menu("NODE_MT_context_menu_show_hide_menu")
|
|
|
|
if is_compositor:
|
|
layout.separator()
|
|
layout.operator("node.read_viewlayers", icon='RENDERLAYERS')
|
|
|
|
|
|
class NODE_MT_view_pie(Menu):
|
|
bl_label = "View"
|
|
|
|
def draw(self, _context):
|
|
layout = self.layout
|
|
|
|
pie = layout.menu_pie()
|
|
pie.operator("node.view_all")
|
|
pie.operator("node.view_selected", icon='ZOOM_SELECTED')
|
|
|
|
|
|
class NODE_PT_active_tool(ToolActivePanelHelper, Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Tool"
|
|
|
|
|
|
class NODE_PT_material_slots(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'HEADER'
|
|
bl_label = "Slot"
|
|
bl_ui_units_x = 12
|
|
|
|
def draw_header(self, context):
|
|
ob = context.object
|
|
self.bl_label = (
|
|
iface_("Slot {:d}").format(ob.active_material_index + 1) if ob.material_slots else
|
|
iface_("Slot")
|
|
)
|
|
|
|
# Duplicate part of `EEVEE_MATERIAL_PT_context_material`.
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
row = layout.row()
|
|
col = row.column()
|
|
|
|
ob = context.object
|
|
col.template_list("MATERIAL_UL_matslots", "", ob, "material_slots", ob, "active_material_index")
|
|
|
|
col = row.column(align=True)
|
|
col.operator("object.material_slot_add", icon='ADD', text="")
|
|
col.operator("object.material_slot_remove", icon='REMOVE', text="")
|
|
|
|
col.separator()
|
|
|
|
col.menu("MATERIAL_MT_context_menu", icon='DOWNARROW_HLT', text="")
|
|
|
|
if len(ob.material_slots) > 1:
|
|
col.separator()
|
|
|
|
col.operator("object.material_slot_move", icon='TRIA_UP', text="").direction = 'UP'
|
|
col.operator("object.material_slot_move", icon='TRIA_DOWN', text="").direction = 'DOWN'
|
|
|
|
if ob.mode == 'EDIT':
|
|
row = layout.row(align=True)
|
|
row.operator("object.material_slot_assign", text="Assign")
|
|
row.operator("object.material_slot_select", text="Select")
|
|
row.operator("object.material_slot_deselect", text="Deselect")
|
|
|
|
|
|
class NODE_PT_geometry_node_tool_object_types(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'HEADER'
|
|
bl_label = "Object Types"
|
|
bl_ui_units_x = 8
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
snode = context.space_data
|
|
group = snode.node_tree
|
|
|
|
types = [
|
|
("is_type_mesh", "Mesh", 'MESH_DATA'),
|
|
("is_type_curve", "Hair Curves", 'CURVES_DATA'),
|
|
("is_type_grease_pencil", "Grease Pencil", 'OUTLINER_OB_GREASEPENCIL'),
|
|
("is_type_pointcloud", "Point Cloud", 'POINTCLOUD_DATA'),
|
|
]
|
|
|
|
col = layout.column()
|
|
col.active = group.is_tool
|
|
for prop, name, icon in types:
|
|
row = col.row(align=True)
|
|
row.label(text=name, icon=icon)
|
|
row.prop(group, prop, text="")
|
|
|
|
|
|
class NODE_PT_geometry_node_tool_mode(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'HEADER'
|
|
bl_label = "Modes"
|
|
bl_ui_units_x = 8
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
snode = context.space_data
|
|
group = snode.node_tree
|
|
|
|
modes = (
|
|
("is_mode_object", "Object Mode", 'OBJECT_DATAMODE'),
|
|
("is_mode_edit", "Edit Mode", 'EDITMODE_HLT'),
|
|
("is_mode_sculpt", "Sculpt Mode", 'SCULPTMODE_HLT'),
|
|
)
|
|
|
|
col = layout.column()
|
|
col.active = group.is_tool
|
|
for prop, name, icon in modes:
|
|
row = col.row(align=True)
|
|
row.label(text=name, icon=icon)
|
|
row.prop(group, prop, text="")
|
|
|
|
if group.is_type_grease_pencil:
|
|
row = col.row(align=True)
|
|
row.label(text="Draw Mode", icon='GREASEPENCIL')
|
|
row.prop(group, "is_mode_paint", text="")
|
|
|
|
|
|
class NODE_PT_geometry_node_tool_options(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'HEADER'
|
|
bl_label = "Options"
|
|
bl_ui_units_x = 8
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
snode = context.space_data
|
|
group = snode.node_tree
|
|
|
|
layout.prop(group, "use_wait_for_click")
|
|
|
|
|
|
class NODE_PT_node_color_presets(PresetPanel, Panel):
|
|
"""Predefined node color"""
|
|
bl_label = "Color Presets"
|
|
preset_subdir = "node_color"
|
|
preset_operator = "script.execute_preset"
|
|
preset_add_operator = "node.node_color_preset_add"
|
|
|
|
|
|
class NODE_MT_node_color_context_menu(Menu):
|
|
bl_label = "Node Color Specials"
|
|
|
|
def draw(self, _context):
|
|
layout = self.layout
|
|
|
|
layout.operator("node.node_copy_color", icon='COPY_ID')
|
|
|
|
|
|
class NODE_MT_context_menu_show_hide_menu(Menu):
|
|
bl_label = "Show/Hide"
|
|
|
|
def draw(self, context):
|
|
snode = context.space_data
|
|
is_compositor = snode.tree_type == 'CompositorNodeTree'
|
|
|
|
layout = self.layout
|
|
|
|
layout.operator("node.mute_toggle", text="Mute")
|
|
|
|
# Node previews are only available in the Compositor.
|
|
if is_compositor:
|
|
layout.operator("node.preview_toggle", text="Node Preview")
|
|
|
|
layout.operator("node.options_toggle", text="Node Options")
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.hide_socket_toggle", text="Unconnected Sockets")
|
|
layout.operator("node.hide_toggle", text="Collapse")
|
|
layout.operator("node.collapse_hide_unused_toggle")
|
|
|
|
|
|
class NODE_MT_context_menu_select_menu(Menu):
|
|
bl_label = "Select"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
layout.operator("node.select_grouped", text="Select Grouped...").extend = False
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.select_linked_from")
|
|
layout.operator("node.select_linked_to")
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.select_same_type_step", text="Activate Same Type Previous").prev = True
|
|
layout.operator("node.select_same_type_step", text="Activate Same Type Next").prev = False
|
|
|
|
|
|
class NODE_MT_context_menu(Menu):
|
|
bl_label = "Node"
|
|
|
|
def draw(self, context):
|
|
snode = context.space_data
|
|
is_nested = (len(snode.path) > 1)
|
|
is_geometrynodes = snode.tree_type == 'GeometryNodeTree'
|
|
group = snode.edit_tree
|
|
|
|
selected_nodes_len = len(context.selected_nodes)
|
|
active_node = context.active_node
|
|
|
|
layout = self.layout
|
|
|
|
# If no nodes are selected.
|
|
if selected_nodes_len == 0:
|
|
layout.operator_context = 'INVOKE_DEFAULT'
|
|
layout.menu("NODE_MT_add", icon='ADD')
|
|
layout.operator("node.clipboard_paste", text="Paste", icon='PASTEDOWN')
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.find_node", text="Find...", icon='VIEWZOOM')
|
|
|
|
layout.separator()
|
|
|
|
if is_geometrynodes:
|
|
layout.operator_context = 'INVOKE_DEFAULT'
|
|
layout.operator("node.select", text="Clear Viewer", icon='HIDE_ON').clear_viewer = True
|
|
|
|
layout.operator("node.links_cut")
|
|
layout.operator("node.links_mute")
|
|
|
|
if is_nested:
|
|
layout.separator()
|
|
|
|
layout.operator("node.tree_path_parent", text="Exit Group", icon='FILE_PARENT')
|
|
|
|
return
|
|
|
|
if is_geometrynodes:
|
|
layout.operator_context = 'INVOKE_DEFAULT'
|
|
layout.operator("node.link_viewer", text="Link to Viewer", icon='HIDE_OFF')
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.clipboard_copy", text="Copy", icon='COPYDOWN')
|
|
layout.operator("node.clipboard_paste", text="Paste", icon='PASTEDOWN')
|
|
|
|
layout.operator_context = 'INVOKE_DEFAULT'
|
|
layout.operator("node.duplicate_move", icon='DUPLICATE')
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.delete", icon='X')
|
|
layout.operator_context = 'EXEC_REGION_WIN'
|
|
layout.operator("node.delete_reconnect", text="Dissolve")
|
|
|
|
if selected_nodes_len > 1:
|
|
layout.separator()
|
|
|
|
layout.operator("node.link_make").replace = False
|
|
layout.operator("node.link_make", text="Make and Replace Links").replace = True
|
|
layout.operator("node.links_detach")
|
|
|
|
layout.separator()
|
|
|
|
if group and group.bl_use_group_interface:
|
|
layout.operator("node.group_make", text="Make Group", icon='NODETREE')
|
|
layout.operator("node.group_insert", text="Insert Into Group")
|
|
|
|
if active_node and active_node.type == 'GROUP':
|
|
layout.operator("node.group_edit").exit = False
|
|
layout.operator("node.group_ungroup", text="Ungroup")
|
|
|
|
if is_nested:
|
|
layout.operator("node.tree_path_parent", text="Exit Group", icon='FILE_PARENT')
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("node.join", text="Join in New Frame")
|
|
layout.operator("node.detach", text="Remove from Frame")
|
|
|
|
layout.separator()
|
|
|
|
props = layout.operator("wm.call_panel", text="Rename...")
|
|
props.name = "TOPBAR_PT_name"
|
|
props.keep_open = False
|
|
|
|
layout.separator()
|
|
|
|
layout.menu("NODE_MT_context_menu_select_menu")
|
|
layout.menu("NODE_MT_context_menu_show_hide_menu")
|
|
|
|
if active_node:
|
|
layout.separator()
|
|
props = layout.operator("wm.doc_view_manual", text="Online Manual", icon='URL')
|
|
props.doc_id = active_node.bl_idname
|
|
|
|
|
|
class NODE_PT_active_node_generic(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Node"
|
|
bl_label = "Node"
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return context.active_node is not None
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
node = context.active_node
|
|
tree = node.id_data
|
|
|
|
layout.use_property_split = True
|
|
layout.use_property_decorate = False
|
|
|
|
layout.prop(node, "name", icon='NODE')
|
|
layout.prop(node, "label", icon='NODE')
|
|
|
|
if tree.type == 'GEOMETRY':
|
|
layout.prop(node, "warning_propagation")
|
|
|
|
|
|
class NODE_PT_active_node_color(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Node"
|
|
bl_label = "Color"
|
|
bl_options = {'DEFAULT_CLOSED'}
|
|
bl_parent_id = "NODE_PT_active_node_generic"
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
node = context.active_node
|
|
if node is None:
|
|
return False
|
|
if node.bl_idname == "NodeReroute":
|
|
return False
|
|
return True
|
|
|
|
def draw_header(self, context):
|
|
node = context.active_node
|
|
self.layout.prop(node, "use_custom_color", text="")
|
|
|
|
def draw_header_preset(self, _context):
|
|
NODE_PT_node_color_presets.draw_panel_header(self.layout)
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
node = context.active_node
|
|
|
|
layout.enabled = node.use_custom_color
|
|
|
|
row = layout.row()
|
|
row.prop(node, "color", text="")
|
|
row.menu("NODE_MT_node_color_context_menu", text="", icon='DOWNARROW_HLT')
|
|
|
|
|
|
class NODE_PT_active_node_properties(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Node"
|
|
bl_label = "Properties"
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return context.active_node is not None
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
node = context.active_node
|
|
layout.template_node_inputs(node)
|
|
|
|
|
|
class NODE_PT_active_node_custom_properties(rna_prop_ui.PropertyPanel, Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Node"
|
|
|
|
_context_path = "active_node"
|
|
_property_type = bpy.types.Node
|
|
|
|
|
|
class NODE_PT_texture_mapping(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Node"
|
|
bl_label = "Texture Mapping"
|
|
bl_options = {'DEFAULT_CLOSED'}
|
|
COMPAT_ENGINES = {
|
|
'BLENDER_RENDER',
|
|
'BLENDER_WORKBENCH',
|
|
}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
node = context.active_node
|
|
return node and hasattr(node, "texture_mapping") and (context.engine in cls.COMPAT_ENGINES)
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.use_property_split = True
|
|
layout.use_property_decorate = False # No animation.
|
|
|
|
node = context.active_node
|
|
mapping = node.texture_mapping
|
|
|
|
layout.prop(mapping, "vector_type")
|
|
|
|
layout.separator()
|
|
|
|
col = layout.column(align=True)
|
|
col.prop(mapping, "mapping_x", text="Projection X")
|
|
col.prop(mapping, "mapping_y", text="Y")
|
|
col.prop(mapping, "mapping_z", text="Z")
|
|
|
|
layout.separator()
|
|
|
|
layout.prop(mapping, "translation")
|
|
layout.prop(mapping, "rotation")
|
|
layout.prop(mapping, "scale")
|
|
|
|
|
|
# Node Backdrop options
|
|
class NODE_PT_backdrop(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "View"
|
|
bl_label = "Backdrop"
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
snode = context.space_data
|
|
return snode.tree_type == 'CompositorNodeTree'
|
|
|
|
def draw_header(self, context):
|
|
snode = context.space_data
|
|
self.layout.prop(snode, "show_backdrop", text="")
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.use_property_split = True
|
|
layout.use_property_decorate = False
|
|
|
|
snode = context.space_data
|
|
layout.active = snode.show_backdrop
|
|
|
|
col = layout.column()
|
|
|
|
col.prop(snode, "backdrop_channels", text="Channels")
|
|
col.prop(snode, "backdrop_zoom", text="Zoom")
|
|
|
|
col.prop(snode, "backdrop_offset", text="Offset")
|
|
|
|
col.separator()
|
|
|
|
col.operator("node.backimage_move", text="Move")
|
|
col.operator("node.backimage_fit", text="Fit")
|
|
|
|
|
|
class NODE_PT_quality(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Options"
|
|
bl_label = "Performance"
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
snode = context.space_data
|
|
return snode.tree_type == 'CompositorNodeTree' and snode.node_tree is not None
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.use_property_split = True
|
|
layout.use_property_decorate = False
|
|
|
|
scene = context.scene
|
|
rd = scene.render
|
|
|
|
snode = context.space_data
|
|
tree = snode.node_tree
|
|
|
|
col = layout.column()
|
|
col.prop(rd, "compositor_device", text="Device")
|
|
if rd.compositor_device == 'GPU':
|
|
col.prop(rd, "compositor_precision", text="Precision")
|
|
|
|
col = layout.column()
|
|
col.prop(tree, "use_viewer_border")
|
|
|
|
|
|
class NODE_PT_overlay(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'HEADER'
|
|
bl_label = "Overlays"
|
|
bl_ui_units_x = 7
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.label(text="Node Editor Overlays")
|
|
|
|
snode = context.space_data
|
|
overlay = snode.overlay
|
|
|
|
layout.active = overlay.show_overlays
|
|
|
|
col = layout.column()
|
|
col.prop(overlay, "show_wire_color", text="Wire Colors")
|
|
col.prop(overlay, "show_reroute_auto_labels", text="Reroute Auto Labels")
|
|
|
|
col.separator()
|
|
|
|
col.prop(overlay, "show_context_path", text="Context Path")
|
|
col.prop(snode, "show_annotation", text="Annotations")
|
|
|
|
if snode.supports_previews:
|
|
col.separator()
|
|
col.prop(overlay, "show_previews", text="Previews")
|
|
if snode.tree_type == 'ShaderNodeTree':
|
|
row = col.row()
|
|
row.prop(overlay, "preview_shape", expand=True)
|
|
row.active = overlay.show_previews
|
|
|
|
if snode.tree_type == 'GeometryNodeTree':
|
|
col.separator()
|
|
col.prop(overlay, "show_timing", text="Timings")
|
|
col.prop(overlay, "show_named_attributes", text="Named Attributes")
|
|
|
|
if snode.tree_type == 'CompositorNodeTree':
|
|
col.prop(overlay, "show_timing", text="Timings")
|
|
|
|
|
|
class NODE_MT_node_tree_interface_context_menu(Menu):
|
|
bl_label = "Node Tree Interface Specials"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
snode = context.space_data
|
|
tree = snode.edit_tree
|
|
active_item = tree.interface.active
|
|
|
|
layout.operator("node.interface_item_duplicate", icon='DUPLICATE')
|
|
layout.separator()
|
|
if active_item.item_type == 'SOCKET':
|
|
layout.operator("node.interface_item_make_panel_toggle")
|
|
elif active_item.item_type == 'PANEL':
|
|
layout.operator("node.interface_item_unlink_panel_toggle")
|
|
|
|
|
|
class NODE_MT_node_tree_interface_new_item(Menu):
|
|
bl_label = "New Item"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.operator_enum("node.interface_item_new", "item_type")
|
|
|
|
active_item = context.space_data.edit_tree.interface.active
|
|
|
|
if active_item.item_type == 'PANEL':
|
|
layout.operator("node.interface_item_new_panel_toggle", text="Panel Toggle")
|
|
|
|
|
|
class NODE_PT_node_tree_properties(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Group"
|
|
bl_label = "Group"
|
|
bl_order = 0
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
snode = context.space_data
|
|
if snode is None:
|
|
return False
|
|
group = snode.edit_tree
|
|
if group is None:
|
|
return False
|
|
if group.is_embedded_data:
|
|
return False
|
|
return True
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
snode = context.space_data
|
|
group = snode.edit_tree
|
|
layout.use_property_split = True
|
|
layout.use_property_decorate = False
|
|
|
|
layout.prop(group, "name", text="Name")
|
|
|
|
if group.asset_data:
|
|
layout.prop(group.asset_data, "description", text="Description")
|
|
else:
|
|
layout.prop(group, "description", text="Description")
|
|
|
|
if not group.bl_use_group_interface:
|
|
return
|
|
|
|
layout.prop(group, "color_tag")
|
|
row = layout.row(align=True)
|
|
row.prop(group, "default_group_node_width", text="Node Width")
|
|
row.operator("node.default_group_width_set", text="", icon='NODE')
|
|
|
|
if group.bl_idname == "GeometryNodeTree":
|
|
row = layout.row()
|
|
row.active = group.is_modifier
|
|
row.prop(group, "show_modifier_manage_panel")
|
|
|
|
header, body = layout.panel("group_usage")
|
|
header.label(text="Usage")
|
|
if body:
|
|
col = body.column(align=True)
|
|
col.prop(group, "is_modifier")
|
|
col.prop(group, "is_tool")
|
|
|
|
|
|
class NODE_PT_node_tree_animation(Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Group"
|
|
bl_label = "Animation"
|
|
bl_options = {'DEFAULT_CLOSED'}
|
|
bl_order = 20
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
snode = context.space_data
|
|
if snode is None:
|
|
return False
|
|
group = snode.edit_tree
|
|
if group is None:
|
|
return False
|
|
if group.is_embedded_data:
|
|
return False
|
|
return True
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.use_property_split = True
|
|
layout.use_property_decorate = False
|
|
|
|
snode = context.space_data
|
|
group = snode.edit_tree
|
|
|
|
col = layout.column(align=True)
|
|
anim.draw_action_and_slot_selector_for_id(col, group)
|
|
|
|
|
|
# Grease Pencil properties
|
|
class NODE_PT_annotation(AnnotationDataPanel, Panel):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_category = "View"
|
|
bl_options = {'DEFAULT_CLOSED'}
|
|
|
|
# NOTE: this is just a wrapper around the generic GP Panel
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
snode = context.space_data
|
|
return snode is not None and snode.node_tree is not None
|
|
|
|
|
|
# Adapt properties editor panel to display in node editor. We have to
|
|
# copy the class rather than inherit due to the way bpy registration works.
|
|
def node_panel(cls):
|
|
node_cls_dict = cls.__dict__.copy()
|
|
|
|
# Needed for re-registration.
|
|
node_cls_dict.pop("bl_rna", None)
|
|
|
|
node_cls = type('NODE_' + cls.__name__, cls.__bases__, node_cls_dict)
|
|
|
|
node_cls.bl_space_type = 'NODE_EDITOR'
|
|
node_cls.bl_region_type = 'UI'
|
|
node_cls.bl_category = "Options"
|
|
if hasattr(node_cls, "bl_parent_id"):
|
|
node_cls.bl_parent_id = "NODE_" + node_cls.bl_parent_id
|
|
|
|
return node_cls
|
|
|
|
|
|
class NODE_AST_compositor(bpy.types.AssetShelf):
|
|
bl_space_type = 'NODE_EDITOR'
|
|
bl_region_type = 'UI'
|
|
bl_options = {'DEFAULT_VISIBLE'}
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return context.space_data.tree_type == 'CompositorNodeTree'
|
|
|
|
@classmethod
|
|
def asset_poll(cls, asset):
|
|
compositing_type = bpy.types.NodeTree.bl_rna.properties["type"].enum_items["COMPOSITING"]
|
|
return asset.id_type == 'NODETREE' and asset.metadata.get("type") == compositing_type.value
|
|
|
|
|
|
classes = (
|
|
NODE_HT_header,
|
|
NODE_MT_editor_menus,
|
|
NODE_MT_add,
|
|
NODE_MT_swap,
|
|
NODE_MT_select,
|
|
NODE_MT_node,
|
|
NODE_MT_node_color_context_menu,
|
|
NODE_MT_context_menu_show_hide_menu,
|
|
NODE_MT_context_menu_select_menu,
|
|
NODE_MT_context_menu,
|
|
NODE_MT_view,
|
|
NODE_MT_view_pie,
|
|
NODE_PT_material_slots,
|
|
NODE_PT_geometry_node_tool_object_types,
|
|
NODE_PT_geometry_node_tool_mode,
|
|
NODE_PT_geometry_node_tool_options,
|
|
NODE_PT_node_color_presets,
|
|
NODE_PT_node_tree_properties,
|
|
NODE_MT_node_tree_interface_new_item,
|
|
NODE_MT_node_tree_interface_context_menu,
|
|
NODE_PT_node_tree_animation,
|
|
NODE_PT_active_node_generic,
|
|
NODE_PT_active_node_color,
|
|
NODE_PT_texture_mapping,
|
|
NODE_PT_active_tool,
|
|
NODE_PT_backdrop,
|
|
NODE_PT_quality,
|
|
NODE_PT_annotation,
|
|
NODE_PT_overlay,
|
|
NODE_PT_active_node_properties,
|
|
NODE_PT_active_node_custom_properties,
|
|
NODE_PT_gizmo_display,
|
|
NODE_AST_compositor,
|
|
|
|
node_panel(EEVEE_MATERIAL_PT_settings),
|
|
node_panel(EEVEE_MATERIAL_PT_settings_surface),
|
|
node_panel(EEVEE_MATERIAL_PT_settings_volume),
|
|
node_panel(MATERIAL_PT_viewport),
|
|
node_panel(WORLD_PT_viewport_display),
|
|
node_panel(DATA_PT_light),
|
|
node_panel(DATA_PT_EEVEE_light),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__": # only for live edit.
|
|
from bpy.utils import register_class
|
|
for cls in classes:
|
|
register_class(cls)
|