99dc90accc
`is_internal` is supposed to mean that the attribute shouldn't be visible in lists or the spreadsheet by default, and that it can't be accessed in geometry nodes. But the value was reversed, which just happened to work because the list filtering was swapped. Differential Revision: https://developer.blender.org/D16680
156 lines
4.7 KiB
Python
156 lines
4.7 KiB
Python
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
import bpy
|
|
from bpy.types import Menu, Panel, UIList
|
|
from rna_prop_ui import PropertyPanel
|
|
|
|
|
|
class DataButtonsPanel:
|
|
bl_space_type = 'PROPERTIES'
|
|
bl_region_type = 'WINDOW'
|
|
bl_context = "data"
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
engine = context.scene.render.engine
|
|
return hasattr(context, 'curves') and context.curves and (engine in cls.COMPAT_ENGINES)
|
|
|
|
|
|
class DATA_PT_context_curves(DataButtonsPanel, Panel):
|
|
bl_label = ""
|
|
bl_options = {'HIDE_HEADER'}
|
|
COMPAT_ENGINES = {'BLENDER_RENDER', 'BLENDER_EEVEE', 'BLENDER_EEVEE_NEXT', 'BLENDER_WORKBENCH'}
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
|
|
ob = context.object
|
|
curves = context.curves
|
|
space = context.space_data
|
|
|
|
if ob:
|
|
layout.template_ID(ob, "data")
|
|
elif curves:
|
|
layout.template_ID(space, "pin_id")
|
|
|
|
|
|
class DATA_PT_curves_surface(DataButtonsPanel, Panel):
|
|
bl_label = "Surface"
|
|
COMPAT_ENGINES = {'BLENDER_RENDER', 'BLENDER_EEVEE', 'BLENDER_EEVEE_NEXT', 'BLENDER_WORKBENCH'}
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
ob = context.object
|
|
|
|
layout.use_property_split = True
|
|
|
|
layout.prop(ob.data, "surface")
|
|
has_surface = ob.data.surface is not None
|
|
if has_surface:
|
|
layout.prop_search(ob.data, "surface_uv_map", ob.data.surface.data, "uv_layers", text="UV Map")
|
|
else:
|
|
row = layout.row()
|
|
row.prop(ob.data, "surface_uv_map", text="UV Map")
|
|
row.enabled = has_surface
|
|
|
|
|
|
class CURVES_MT_add_attribute(Menu):
|
|
bl_label = "Add Attribute"
|
|
|
|
@staticmethod
|
|
def add_standard_attribute(layout, curves, name, data_type, domain):
|
|
exists = curves.attributes.get(name) is not None
|
|
|
|
col = layout.column()
|
|
col.enabled = not exists
|
|
col.operator_context = 'EXEC_DEFAULT'
|
|
|
|
props = col.operator("geometry.attribute_add", text=name)
|
|
props.name = name
|
|
props.data_type = data_type
|
|
props.domain = domain
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
curves = context.curves
|
|
|
|
self.add_standard_attribute(layout, curves, 'radius', 'FLOAT', 'POINT')
|
|
self.add_standard_attribute(layout, curves, 'color', 'FLOAT_COLOR', 'POINT')
|
|
|
|
layout.separator()
|
|
|
|
layout.operator_context = 'INVOKE_DEFAULT'
|
|
layout.operator("geometry.attribute_add", text="Custom...")
|
|
|
|
|
|
class CURVES_UL_attributes(UIList):
|
|
def filter_items(self, _context, data, property):
|
|
attributes = getattr(data, property)
|
|
flags = []
|
|
indices = [i for i in range(len(attributes))]
|
|
|
|
for item in attributes:
|
|
flags.append(0 if item.is_internal else self.bitflag_filter_item)
|
|
|
|
return flags, indices
|
|
|
|
def draw_item(self, _context, layout, _data, attribute, _icon, _active_data, _active_propname, _index):
|
|
data_type = attribute.bl_rna.properties['data_type'].enum_items[attribute.data_type]
|
|
domain = attribute.bl_rna.properties['domain'].enum_items[attribute.domain]
|
|
|
|
split = layout.split(factor=0.5)
|
|
split.emboss = 'NONE'
|
|
row = split.row()
|
|
row.prop(attribute, "name", text="")
|
|
sub = split.split()
|
|
sub.alignment = 'RIGHT'
|
|
sub.active = False
|
|
sub.label(text=domain.name)
|
|
sub.label(text=data_type.name)
|
|
|
|
|
|
class DATA_PT_CURVES_attributes(DataButtonsPanel, Panel):
|
|
bl_label = "Attributes"
|
|
COMPAT_ENGINES = {'BLENDER_RENDER', 'BLENDER_EEVEE', 'BLENDER_EEVEE_NEXT', 'BLENDER_WORKBENCH'}
|
|
|
|
def draw(self, context):
|
|
curves = context.curves
|
|
|
|
layout = self.layout
|
|
row = layout.row()
|
|
|
|
col = row.column()
|
|
col.template_list(
|
|
"CURVES_UL_attributes",
|
|
"attributes",
|
|
curves,
|
|
"attributes",
|
|
curves.attributes,
|
|
"active_index",
|
|
rows=3,
|
|
)
|
|
|
|
col = row.column(align=True)
|
|
col.menu("CURVES_MT_add_attribute", icon='ADD', text="")
|
|
col.operator("geometry.attribute_remove", icon='REMOVE', text="")
|
|
|
|
|
|
class DATA_PT_custom_props_curves(DataButtonsPanel, PropertyPanel, Panel):
|
|
COMPAT_ENGINES = {'BLENDER_RENDER', 'BLENDER_EEVEE', 'BLENDER_EEVEE_NEXT', 'BLENDER_WORKBENCH'}
|
|
_context_path = "object.data"
|
|
_property_type = bpy.types.Curves if hasattr(bpy.types, "Curves") else None
|
|
|
|
|
|
classes = (
|
|
DATA_PT_context_curves,
|
|
DATA_PT_CURVES_attributes,
|
|
DATA_PT_curves_surface,
|
|
DATA_PT_custom_props_curves,
|
|
CURVES_MT_add_attribute,
|
|
CURVES_UL_attributes,
|
|
)
|
|
|
|
if __name__ == "__main__": # only for live edit.
|
|
from bpy.utils import register_class
|
|
for cls in classes:
|
|
register_class(cls)
|