yyc12345
e7376a3e9c
- add tunnel section creation and re-organise rail creation menu. - allow duplicated elements (Nong xxx) creation for some special, such as Fan and Extra Point.
87 lines
3.1 KiB
Python
87 lines
3.1 KiB
Python
import bpy, mathutils
|
|
from . import UTILS_constants, UTILS_functions
|
|
|
|
# ================================================= actual add
|
|
|
|
class BALLANCE_OT_add_components(bpy.types.Operator):
|
|
"""Add sector related elements"""
|
|
bl_idname = "ballance.add_components"
|
|
bl_label = "Add elements"
|
|
bl_options = {'UNDO'}
|
|
|
|
elements_type: bpy.props.EnumProperty(
|
|
name="Type",
|
|
description="This element type",
|
|
items=tuple(map(lambda x: (x, x, ""), UTILS_constants.bmfile_componentList)),
|
|
)
|
|
|
|
attentionElements = ("PC_TwoFlames", "PR_Resetpoint")
|
|
uniqueElements = ("PS_FourFlames", "PE_Balloon")
|
|
canDuplicatedElements = ('P_Extra_Point', 'P_Modul_18', 'P_Modul_26')
|
|
|
|
elements_sector: bpy.props.IntProperty(
|
|
name="Sector",
|
|
description="Define which sector the object will be grouped in",
|
|
min=1, max=8,
|
|
default=1,
|
|
)
|
|
|
|
elements_duplicated: bpy.props.BoolProperty(
|
|
name="Duplicated",
|
|
description="Whether duplicate elements (Nong xxx / 脓xxx)",
|
|
default=False,
|
|
)
|
|
elements_dup_times: bpy.props.IntProperty(
|
|
name="Duplication Times",
|
|
description="How many this element should be duplicated.",
|
|
min=1, max=64,
|
|
soft_min=1, soft_max=32,
|
|
default=1,
|
|
)
|
|
|
|
def execute(self, context):
|
|
# get name
|
|
if self.elements_type in self.uniqueElements:
|
|
finalObjectName = self.elements_type + "_01"
|
|
elif self.elements_type in self.attentionElements:
|
|
finalObjectName = self.elements_type + "_0" + str(self.elements_sector)
|
|
else:
|
|
finalObjectName = self.elements_type + "_0" + str(self.elements_sector) + "_"
|
|
|
|
# create object
|
|
loadedMesh = UTILS_functions.load_component(
|
|
UTILS_constants.bmfile_componentList.index(self.elements_type))
|
|
obj = bpy.data.objects.new(finalObjectName, loadedMesh)
|
|
UTILS_functions.add_into_scene_and_move_to_cursor(obj)
|
|
|
|
# extra duplication
|
|
if self.elements_type in self.canDuplicatedElements:
|
|
for i in range(self.elements_dup_times - 1):
|
|
obj = bpy.data.objects.new(finalObjectName, loadedMesh)
|
|
UTILS_functions.add_into_scene_and_move_to_cursor(obj)
|
|
|
|
|
|
return {'FINISHED'}
|
|
|
|
def invoke(self, context, event):
|
|
wm = context.window_manager
|
|
return wm.invoke_props_dialog(self)
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
# attension notice
|
|
if self.elements_type in self.attentionElements:
|
|
layout.label(text="Please note that sector is suffix.")
|
|
if self.elements_type in self.canDuplicatedElements:
|
|
layout.label(text="This element can use duplication feature.")
|
|
|
|
# cfg
|
|
layout.prop(self, "elements_type")
|
|
if self.elements_type not in self.uniqueElements:
|
|
layout.prop(self, "elements_sector")
|
|
|
|
if self.elements_type in self.canDuplicatedElements:
|
|
layout.separator()
|
|
layout.prop(self, "elements_duplicated")
|
|
layout.prop(self, "elements_dup_times")
|