BallanceBlenderHelper/ballance_blender_plugin/OBJS_add_components.py

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) and self.elements_duplicated:
            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")