write shit

bbp_ng/OP_ADDS_component.py

@@ -0,0 +1,77 @@
+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)