78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
|
import bpy
|
||
|
|
from . import UTIL_functions, UTIL_icons_manager
|
||
|
|
from . import PROP_preferences, PROP_ballance_element, PROP_virtools_group
|
||
|
|
|
||
|
|
_g_UniqueElements = {
|
||
|
|
"PS_FourFlames": 'PS_FourFlames_01',
|
||
|
|
"PE_Balloon": 'PE_Balloon_01'
|
||
|
|
}
|
||
|
|
|
||
|
|
def _get_component_name(comp_name: str, comp_sector: int) -> str:
|
||
|
|
return '{}_{:0>2d}_'.format(comp_name, comp_sector)
|
||
|
|
|
||
|
|
class BBP_OT_add_component(bpy.types.Operator):
|
||
|
|
"""Add Element"""
|
||
|
|
bl_idname = "bbp.add_component"
|
||
|
|
bl_label = "Add Element"
|
||
|
|
bl_options = {'UNDO'}
|
||
|
|
|
||
|
|
element_sector: bpy.props.IntProperty(
|
||
|
|
name = "Sector",
|
||
|
|
description = "Define which sector the object will be grouped in",
|
||
|
|
min = 1, max = 999,
|
||
|
|
soft_min = 1, soft_max = 8,
|
||
|
|
default = 1,
|
||
|
|
)
|
||
|
|
|
||
|
|
element_type: bpy.props.EnumProperty(
|
||
|
|
name = "Type",
|
||
|
|
description = "This element type",
|
||
|
|
#items=tuple(map(lambda x: (x, x, ""), UTILS_constants.bmfile_componentList)),
|
||
|
|
items = tuple(
|
||
|
|
# token, display name, descriptions, icon, index
|
||
|
|
(str(item.value), item.name, "", UTIL_icons_manager.get_element_icon(item.name), item.value)
|
||
|
|
for item in PROP_ballance_element.BallanceElementType
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
def invoke(self, context, event):
|
||
|
|
wm = context.window_manager
|
||
|
|
return wm.invoke_props_dialog(self)
|
||
|
|
|
||
|
|
def draw(self, context):
|
||
|
|
layout = self.layout
|
||
|
|
layout.prop(self, "element_type")
|
||
|
|
layout.prop(self, "element_sector")
|
||
|
|
|
||
|
|
# check for unique name and show warning
|
||
|
|
elename: str | None = _g_UniqueElements.get(PROP_ballance_element.BallanceElementType(int(self.element_type)).name, None)
|
||
|
|
if elename is not None and elename in bpy.data.objects:
|
||
|
|
layout.label(f'Warning: {elename} already exist.')
|
||
|
|
|
||
|
|
def execute(self, context):
|
||
|
|
# create by ballance elements
|
||
|
|
eletype: PROP_ballance_element.BallanceElementType = PROP_ballance_element.BallanceElementType(int(self.element_type))
|
||
|
|
with PROP_ballance_element.BallanceElementsHelper(bpy.context.scene) as creator:
|
||
|
|
obj = bpy.data.objects.new(
|
||
|
|
_get_component_name(eletype.name, self.element_sector),
|
||
|
|
creator.get_element(eletype.value)
|
||
|
|
)
|
||
|
|
UTIL_functions.add_into_scene_and_move_to_cursor(obj)
|
||
|
|
|
||
|
|
return {'FINISHED'}
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def draw_blc_menu(self, layout: bpy.types.UILayout):
|
||
|
|
for item in PROP_ballance_element.BallanceElementType:
|
||
|
|
cop = layout.operator(
|
||
|
|
self.bl_idname, text = item.name,
|
||
|
|
icon_value = UTIL_icons_manager.get_element_icon(item.name))
|
||
|
|
cop.element_type = str(item.value)
|
||
|
|
|
||
|
|
def register():
|
||
|
|
# register all classes
|
||
|
|
bpy.utils.register_class(BBP_OT_add_component)
|
||
|
|
|
||
|
|
def unregister():
|
||
|
|
bpy.utils.unregister_class(BBP_OT_add_component)
|