12722bd354
In the latest discussions about curves/hair mesh attachement information (T95776), it was decided to use UV coordinates to store where on the mesh each root is. For that, we have to specify which of the UV map attributes to use for UV lookups. This property isn't used yet, but it will be shortly when refactoring the attachement information in the add brush and the to particle system conversion. Differential Revision: https://developer.blender.org/D15115
150 lines
4.4 KiB
Python
150 lines
4.4 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_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_WORKBENCH'}
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
ob = context.object
|
|
|
|
layout.use_property_split = True
|
|
|
|
layout.prop(ob.data, "surface")
|
|
layout.prop(ob.data, "surface_uv_map", text="UV Map")
|
|
|
|
|
|
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 index, item in enumerate(attributes):
|
|
flags.append(self.bitflag_filter_item if item.is_internal else 0)
|
|
|
|
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_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_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)
|