274 lines
13 KiB
Python
274 lines
13 KiB
Python
import json
|
|
import bpy
|
|
|
|
from ..registry.operators import COMPONENTS_OT_REFRESH_CUSTOM_PROPERTIES_CURRENT
|
|
from .metadata import do_object_custom_properties_have_missing_metadata, get_bevy_components
|
|
from .operators import AddComponentOperator, CopyComponentOperator, Fix_Component_Operator, RemoveComponentOperator, GenerateComponent_From_custom_property_Operator, PasteComponentOperator, Toggle_ComponentVisibility
|
|
|
|
def draw_propertyGroup( propertyGroup, layout, nesting =[], rootName=None):
|
|
is_enum = getattr(propertyGroup, "with_enum")
|
|
is_list = getattr(propertyGroup, "with_list")
|
|
is_map = getattr(propertyGroup, "with_map")
|
|
# item in our components hierarchy can get the correct propertyGroup by STRINGS because of course, we cannot pass objects to operators...sigh
|
|
|
|
# if it is an enum, the first field name is always the list of enum variants, the others are the variants
|
|
field_names = propertyGroup.field_names
|
|
#print("")
|
|
#print("drawing", propertyGroup, nesting, "component_name", rootName)
|
|
if is_enum:
|
|
subrow = layout.row()
|
|
display_name = field_names[0] if propertyGroup.tupple_or_struct == "struct" else ""
|
|
subrow.prop(propertyGroup, field_names[0], text=display_name)
|
|
subrow.separator()
|
|
selection = getattr(propertyGroup, "selection")
|
|
|
|
for fname in field_names[1:]:
|
|
if fname == "variant_" + selection:
|
|
subrow = layout.row()
|
|
display_name = fname if propertyGroup.tupple_or_struct == "struct" else ""
|
|
|
|
nestedPropertyGroup = getattr(propertyGroup, fname)
|
|
nested = getattr(nestedPropertyGroup, "nested", False)
|
|
#print("nestedPropertyGroup", nestedPropertyGroup, fname, nested)
|
|
if nested:
|
|
draw_propertyGroup(nestedPropertyGroup, subrow.column(), nesting + [fname], rootName )
|
|
# if an enum variant is not a propertyGroup
|
|
break
|
|
elif is_list:
|
|
item_list = getattr(propertyGroup, "list")
|
|
list_index = getattr(propertyGroup, "list_index")
|
|
box = layout.box()
|
|
split = box.split(factor=0.9)
|
|
list_column, buttons_column = (split.column(),split.column())
|
|
|
|
list_column = list_column.box()
|
|
for index, item in enumerate(item_list):
|
|
row = list_column.row()
|
|
draw_propertyGroup(item, row, nesting, rootName)
|
|
icon = 'CHECKBOX_HLT' if list_index == index else 'CHECKBOX_DEHLT'
|
|
op = row.operator('generic_list.select_item', icon=icon, text="")
|
|
op.component_name = rootName
|
|
op.property_group_path = json.dumps(nesting)
|
|
op.selection_index = index
|
|
|
|
#various control buttons
|
|
buttons_column.separator()
|
|
row = buttons_column.row()
|
|
op = row.operator('generic_list.list_action', icon='ADD', text="")
|
|
op.action = 'ADD'
|
|
op.component_name = rootName
|
|
op.property_group_path = json.dumps(nesting)
|
|
|
|
row = buttons_column.row()
|
|
op = row.operator('generic_list.list_action', icon='REMOVE', text="")
|
|
op.action = 'REMOVE'
|
|
op.component_name = rootName
|
|
op.property_group_path = json.dumps(nesting)
|
|
|
|
buttons_column.separator()
|
|
row = buttons_column.row()
|
|
op = row.operator('generic_list.list_action', icon='TRIA_UP', text="")
|
|
op.action = 'UP'
|
|
op.component_name = rootName
|
|
op.property_group_path = json.dumps(nesting)
|
|
|
|
row = buttons_column.row()
|
|
op = row.operator('generic_list.list_action', icon='TRIA_DOWN', text="")
|
|
op.action = 'DOWN'
|
|
op.component_name = rootName
|
|
op.property_group_path = json.dumps(nesting)
|
|
|
|
elif is_map:
|
|
root = layout.row().column()
|
|
if hasattr(propertyGroup, "list"): # TODO: improve handling of non drawable UI
|
|
keys_list = getattr(propertyGroup, "list")
|
|
values_list = getattr(propertyGroup, "values_list")
|
|
box = root.box()
|
|
row = box.row()
|
|
row.label(text="Add entry:")
|
|
keys_setter = getattr(propertyGroup, "keys_setter")
|
|
draw_propertyGroup(keys_setter, row, nesting, rootName)
|
|
|
|
values_setter = getattr(propertyGroup, "values_setter")
|
|
draw_propertyGroup(values_setter, row, nesting, rootName)
|
|
|
|
op = row.operator('generic_map.map_action', icon='ADD', text="")
|
|
op.action = 'ADD'
|
|
op.component_name = rootName
|
|
op.property_group_path = json.dumps(nesting)
|
|
|
|
box = root.box()
|
|
split = box.split(factor=0.9)
|
|
list_column, buttons_column = (split.column(),split.column())
|
|
list_column = list_column.box()
|
|
|
|
for index, item in enumerate(keys_list):
|
|
row = list_column.row()
|
|
draw_propertyGroup(item, row, nesting, rootName)
|
|
|
|
value = values_list[index]
|
|
draw_propertyGroup(value, row, nesting, rootName)
|
|
|
|
op = row.operator('generic_map.map_action', icon='REMOVE', text="")
|
|
op.action = 'REMOVE'
|
|
op.component_name = rootName
|
|
op.property_group_path = json.dumps(nesting)
|
|
op.target_index = index
|
|
|
|
|
|
#various control buttons
|
|
buttons_column.separator()
|
|
row = buttons_column.row()
|
|
|
|
|
|
else:
|
|
for fname in field_names:
|
|
#subrow = layout.row()
|
|
nestedPropertyGroup = getattr(propertyGroup, fname)
|
|
nested = getattr(nestedPropertyGroup, "nested", False)
|
|
display_name = fname if propertyGroup.tupple_or_struct == "struct" else ""
|
|
|
|
if nested:
|
|
layout.separator()
|
|
layout.separator()
|
|
|
|
layout.label(text=display_name) # this is the name of the field/sub field
|
|
layout.separator()
|
|
subrow = layout.row()
|
|
draw_propertyGroup(nestedPropertyGroup, subrow, nesting + [fname], rootName )
|
|
else:
|
|
subrow = layout.row()
|
|
subrow.prop(propertyGroup, fname, text=display_name)
|
|
subrow.separator()
|
|
|
|
|
|
class BEVY_COMPONENTS_PT_ComponentsPanel(bpy.types.Panel):
|
|
bl_idname = "BEVY_COMPONENTS_PT_ComponentsPanel"
|
|
bl_label = "Components"
|
|
bl_space_type = 'VIEW_3D'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Bevy Components"
|
|
bl_context = "objectmode"
|
|
bl_parent_id = "BEVY_COMPONENTS_PT_MainPanel"
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return (context.object is not None)
|
|
|
|
def draw(self, context):
|
|
object = context.object
|
|
layout = self.layout
|
|
|
|
# we get & load our component registry
|
|
registry = bpy.context.window_manager.components_registry
|
|
available_components = bpy.context.window_manager.components_list
|
|
registry_has_type_infos = registry.has_type_infos()
|
|
|
|
if object is not None:
|
|
row = layout.row(align=True)
|
|
row.prop(available_components, "list", text="Component")
|
|
row.prop(available_components, "filter",text="Filter")
|
|
|
|
# add components
|
|
row = layout.row(align=True)
|
|
op = row.operator(AddComponentOperator.bl_idname, text="Add", icon="ADD")
|
|
op.component_type = available_components.list
|
|
row.enabled = available_components.list != ''
|
|
|
|
layout.separator()
|
|
|
|
# paste components
|
|
row = layout.row(align=True)
|
|
row.operator(PasteComponentOperator.bl_idname, text="Paste component ("+bpy.context.window_manager.copied_source_component_name+")", icon="PASTEDOWN")
|
|
row.enabled = registry_has_type_infos and context.window_manager.copied_source_object != ''
|
|
|
|
layout.separator()
|
|
|
|
# upgrate custom props to components
|
|
upgradeable_customProperties = registry.has_type_infos() and do_object_custom_properties_have_missing_metadata(context.object)
|
|
if upgradeable_customProperties:
|
|
row = layout.row(align=True)
|
|
op = row.operator(GenerateComponent_From_custom_property_Operator.bl_idname, text="generate components from custom properties" , icon="LOOP_FORWARDS")
|
|
layout.separator()
|
|
|
|
|
|
components_in_object = object.components_meta.components
|
|
#print("components_names", dict(components_bla).keys())
|
|
|
|
for component_name in sorted(get_bevy_components(object)) : # sorted by component name, practical
|
|
#print("component_name", component_name)
|
|
if component_name == "components_meta":
|
|
continue
|
|
# anything withouth metadata gets skipped, we only want to see real components, not all custom props
|
|
component_meta = next(filter(lambda component: component["long_name"] == component_name, components_in_object), None)
|
|
if component_meta == None:
|
|
continue
|
|
|
|
component_invalid = getattr(component_meta, "invalid")
|
|
invalid_details = getattr(component_meta, "invalid_details")
|
|
component_visible = getattr(component_meta, "visible")
|
|
single_field = False
|
|
|
|
# our whole row
|
|
box = layout.box()
|
|
row = box.row(align=True)
|
|
# "header"
|
|
row.alert = component_invalid
|
|
row.prop(component_meta, "enabled", text="")
|
|
row.label(text=component_name)
|
|
|
|
# we fetch the matching ui property group
|
|
root_propertyGroup_name = registry.get_propertyGroupName_from_longName(component_name)
|
|
"""print("root_propertyGroup_name", root_propertyGroup_name)"""
|
|
print("component_meta", component_meta, component_invalid)
|
|
|
|
if root_propertyGroup_name:
|
|
propertyGroup = getattr(component_meta, root_propertyGroup_name, None)
|
|
"""print("propertyGroup", propertyGroup)"""
|
|
if propertyGroup:
|
|
# if the component has only 0 or 1 field names, display inline, otherwise change layout
|
|
single_field = len(propertyGroup.field_names) < 2
|
|
prop_group_location = box.row(align=True).column()
|
|
"""if single_field:
|
|
prop_group_location = row.column(align=True)#.split(factor=0.9)#layout.row(align=False)"""
|
|
|
|
if component_visible:
|
|
if component_invalid:
|
|
error_message = invalid_details if component_invalid else "Missing component UI data, please reload registry !"
|
|
prop_group_location.label(text=error_message)
|
|
draw_propertyGroup(propertyGroup, prop_group_location, [root_propertyGroup_name], component_name)
|
|
else :
|
|
row.label(text="details hidden, click on toggle to display")
|
|
else:
|
|
error_message = invalid_details if component_invalid else "Missing component UI data, please reload registry !"
|
|
row.label(text=error_message)
|
|
|
|
# "footer" with additional controls
|
|
if component_invalid:
|
|
if root_propertyGroup_name:
|
|
propertyGroup = getattr(component_meta, root_propertyGroup_name, None)
|
|
if propertyGroup:
|
|
unit_struct = len(propertyGroup.field_names) == 0
|
|
if unit_struct:
|
|
op = row.operator(Fix_Component_Operator.bl_idname, text="", icon="SHADERFX")
|
|
op.component_name = component_name
|
|
row.separator()
|
|
|
|
op = row.operator(RemoveComponentOperator.bl_idname, text="", icon="X")
|
|
op.component_name = component_name
|
|
row.separator()
|
|
|
|
op = row.operator(CopyComponentOperator.bl_idname, text="", icon="COPYDOWN")
|
|
op.source_component_name = component_name
|
|
op.source_object_name = object.name
|
|
row.separator()
|
|
|
|
#if not single_field:
|
|
toggle_icon = "TRIA_DOWN" if component_visible else "TRIA_RIGHT"
|
|
op = row.operator(Toggle_ComponentVisibility.bl_idname, text="", icon=toggle_icon)
|
|
op.component_name = component_name
|
|
#row.separator()
|
|
|
|
else:
|
|
layout.label(text ="Select an object to edit its components")
|