267d449a50
Similar to curves, the ".selection" attribute is used to store selection.
Selection isn't visualized in the viewport yet, but this should be added
at some point anywya. See f89b32382a.
126 lines
4.4 KiB
Python
126 lines
4.4 KiB
Python
# SPDX-FileCopyrightText: 2009-2023 Blender Foundation
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
import bpy
|
|
|
|
|
|
class SPREADSHEET_HT_header(bpy.types.Header):
|
|
bl_space_type = 'SPREADSHEET'
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
space = context.space_data
|
|
|
|
layout.template_header()
|
|
viewer_path = space.viewer_path.path
|
|
|
|
if len(viewer_path) == 0:
|
|
self.draw_without_viewer_path(layout)
|
|
return
|
|
root_context = viewer_path[0]
|
|
if root_context.type != 'ID':
|
|
self.draw_without_viewer_path(layout)
|
|
return
|
|
if not isinstance(root_context.id, bpy.types.Object):
|
|
self.draw_without_viewer_path(layout)
|
|
return
|
|
obj = root_context.id
|
|
if obj is None:
|
|
self.draw_without_viewer_path(layout)
|
|
return
|
|
|
|
layout.prop(space, "object_eval_state", text="")
|
|
|
|
if space.object_eval_state == 'ORIGINAL':
|
|
# Only show first context.
|
|
viewer_path = viewer_path[:1]
|
|
if space.display_viewer_path_collapsed:
|
|
self.draw_collapsed_viewer_path(context, layout, viewer_path)
|
|
else:
|
|
self.draw_full_viewer_path(context, layout, viewer_path)
|
|
|
|
pin_icon = 'PINNED' if space.is_pinned else 'UNPINNED'
|
|
layout.operator("spreadsheet.toggle_pin", text="", icon=pin_icon, emboss=False)
|
|
|
|
if space.object_eval_state == 'VIEWER_NODE' and len(viewer_path) < 3:
|
|
layout.label(text="No active viewer node", icon='INFO')
|
|
|
|
layout.separator_spacer()
|
|
|
|
row = layout.row(align=True)
|
|
sub = row.row(align=True)
|
|
sub.active = self.selection_filter_available(space)
|
|
sub.prop(space, "show_only_selected", text="")
|
|
row.prop(space, "use_filter", toggle=True, icon='FILTER', icon_only=True)
|
|
|
|
def draw_without_viewer_path(self, layout):
|
|
layout.label(text="No active context")
|
|
|
|
def draw_full_viewer_path(self, context, layout, viewer_path):
|
|
space = context.space_data
|
|
row = layout.row()
|
|
for ctx in viewer_path[:-1]:
|
|
subrow = row.row(align=True)
|
|
self.draw_spreadsheet_context(subrow, ctx)
|
|
self.draw_spreadsheet_viewer_path_icon(subrow, space)
|
|
|
|
self.draw_spreadsheet_context(row, viewer_path[-1])
|
|
|
|
def draw_collapsed_viewer_path(self, context, layout, viewer_path):
|
|
space = context.space_data
|
|
row = layout.row(align=True)
|
|
self.draw_spreadsheet_context(row, viewer_path[0])
|
|
if len(viewer_path) == 1:
|
|
return
|
|
self.draw_spreadsheet_viewer_path_icon(row, space)
|
|
if len(viewer_path) > 2:
|
|
self.draw_spreadsheet_viewer_path_icon(row, space, icon='DOT')
|
|
self.draw_spreadsheet_viewer_path_icon(row, space)
|
|
self.draw_spreadsheet_context(row, viewer_path[-1])
|
|
|
|
def draw_spreadsheet_context(self, layout, ctx):
|
|
if ctx.type == 'ID':
|
|
if ctx.id is not None and isinstance(ctx.id, bpy.types.Object):
|
|
layout.label(text=ctx.id.name, icon='OBJECT_DATA')
|
|
else:
|
|
layout.label(text="Invalid id")
|
|
elif ctx.type == 'MODIFIER':
|
|
layout.label(text=ctx.modifier_name, icon='MODIFIER')
|
|
elif ctx.type == 'GROUP_NODE':
|
|
layout.label(text=ctx.ui_name, icon='NODE')
|
|
elif ctx.type == 'SIMULATION_ZONE':
|
|
layout.label(text="Simulation Zone")
|
|
elif ctx.type == 'VIEWER_NODE':
|
|
layout.label(text=ctx.ui_name)
|
|
|
|
def draw_spreadsheet_viewer_path_icon(self, layout, space, icon='RIGHTARROW_THIN'):
|
|
layout.prop(space, "display_viewer_path_collapsed", icon_only=True, emboss=False, icon=icon)
|
|
|
|
def selection_filter_available(self, space):
|
|
root_context = space.viewer_path.path[0]
|
|
if root_context.type != 'ID':
|
|
return False
|
|
if not isinstance(root_context.id, bpy.types.Object):
|
|
return False
|
|
obj = root_context.id
|
|
if obj is None:
|
|
return False
|
|
if obj.type == 'MESH':
|
|
return obj.mode == 'EDIT'
|
|
if obj.type == 'CURVES':
|
|
return obj.mode in {'SCULPT_CURVES', 'EDIT'}
|
|
if obj.type == 'POINTCLOUD':
|
|
return obj.mode == 'EDIT'
|
|
return False
|
|
|
|
|
|
classes = (
|
|
SPREADSHEET_HT_header,
|
|
)
|
|
|
|
if __name__ == "__main__": # Only for live edit.
|
|
from bpy.utils import register_class
|
|
for cls in classes:
|
|
register_class(cls)
|