bl_info={
	"name":"Ballance Blender Plugin",
	"description":"Ballance mapping tools for Blender",
	"author":"yyc12345",
	"version":(3,0),
	"blender":(3,3,0),
	"category":"Object",
	"support":"TESTING",
    "warning": "Please read document before using this plugin.",
    "wiki_url": "https://github.com/yyc12345/BallanceBlenderHelper",
    "tracker_url": "https://github.com/yyc12345/BallanceBlenderHelper/issues"
}

# ============================================= 
# import system
import bpy

# import my code (with reload)
if "bpy" in locals():
    import importlib
    if "UTILS_constants" in locals():
        importlib.reload(UTILS_constants)
    if "UTILS_functions" in locals():
        importlib.reload(UTILS_functions)
    if "UTILS_preferences" in locals():
        importlib.reload(UTILS_preferences)
    if "UTILS_file_io" in locals():
        importlib.reload(UTILS_file_io)
    if "UTILS_zip_helper" in locals():
        importlib.reload(UTILS_zip_helper)
    if "UTILS_virtools_prop" in locals():
        importlib.reload(UTILS_virtools_prop)
    if "UTILS_safe_eval" in locals():
        importlib.reload(UTILS_safe_eval)
    if "UTILS_icons_manager" in locals():
        importlib.reload(UTILS_icons_manager)

    if "BMFILE_export" in locals():
        importlib.reload(BMFILE_export)
    if "BMFILE_import" in locals():
        importlib.reload(BMFILE_import)

    if "MODS_rail_uv" in locals():
        importlib.reload(MODS_rail_uv)
    if "MODS_3dsmax_align" in locals():
        importlib.reload(MODS_3dsmax_align)
    if "MODS_flatten_uv" in locals():
        importlib.reload(MODS_flatten_uv)

    if "OBJS_add_components" in locals():
        importlib.reload(OBJS_add_components)
    if "OBJS_add_floors" in locals():
        importlib.reload(OBJS_add_floors)
    if "OBJS_add_rails" in locals():
        importlib.reload(OBJS_add_rails)
    if "OBJS_group_opers" in locals():
        importlib.reload(OBJS_group_opers)

    if "NAMES_rename_system" in locals():
        importlib.reload(NAMES_rename_system)

    if "PROPS_virtools_group" in locals():
        importlib.reload(PROPS_virtools_group)
    if "PROPS_virtools_material" in locals():
        importlib.reload(PROPS_virtools_material)

from . import UTILS_constants, UTILS_functions, UTILS_preferences, UTILS_virtools_prop, UTILS_safe_eval, UTILS_icons_manager
from . import BMFILE_export, BMFILE_import
from . import MODS_3dsmax_align, MODS_flatten_uv, MODS_rail_uv
from . import OBJS_add_components, OBJS_add_floors, OBJS_add_rails, OBJS_group_opers
from . import NAMES_rename_system
from . import PROPS_virtools_group, PROPS_virtools_material

# ============================================= 
# menu system

class BALLANCE_MT_ThreeDViewerMenu(bpy.types.Menu):
    """Ballance 3D operators"""
    bl_idname = "BALLANCE_MT_ThreeDViewerMenu"
    bl_label = "Ballance"

    def draw(self, context):
        layout = self.layout

        layout.operator(MODS_3dsmax_align.BALLANCE_OT_super_align.bl_idname)
        layout.operator(MODS_rail_uv.BALLANCE_OT_rail_uv.bl_idname)
        layout.operator(MODS_flatten_uv.BALLANCE_OT_flatten_uv.bl_idname)

class BALLANCE_MT_AddFloorMenu(bpy.types.Menu):
    """Add Ballance floor"""
    bl_idname = "BALLANCE_MT_AddFloorMenu"
    bl_label = "Floors"

    def draw(self, context):
        layout = self.layout

        layout.label(text="Basic floor")
        for item in UTILS_constants.floor_basicBlockList:
            cop = layout.operator(
                OBJS_add_floors.BALLANCE_OT_add_floors.bl_idname, 
                text=item, icon_value = UTILS_icons_manager.get_floor_icon(item))
            cop.floor_type = item

        layout.separator()
        layout.label(text="Derived floor")
        for item in UTILS_constants.floor_derivedBlockList:
            cop = layout.operator(
                OBJS_add_floors.BALLANCE_OT_add_floors.bl_idname, 
                text=item, icon_value = UTILS_icons_manager.get_floor_icon(item))
            cop.floor_type = item

class BALLANCE_MT_AddRailMenu(bpy.types.Menu):
    """Add Ballance rail"""
    bl_idname = "BALLANCE_MT_AddRailMenu"
    bl_label = "Rails"

    def draw(self, context):
        layout = self.layout
        layout.operator(OBJS_add_rails.BALLANCE_OT_add_rails.bl_idname, text="Rail Section")
        layout.operator(OBJS_add_rails.BALLANCE_OT_add_tunnels.bl_idname, text="Tunnel Section")


# ============================================= 
# blender call system

classes = (
    UTILS_preferences.BallanceBlenderPluginPreferences,
    UTILS_preferences.MyPropertyGroup,
    
    BMFILE_import.BALLANCE_OT_import_bm,
    BMFILE_export.BALLANCE_OT_export_bm,

    MODS_rail_uv.BALLANCE_OT_rail_uv,
    MODS_3dsmax_align.BALLANCE_OT_super_align,
    MODS_flatten_uv.BALLANCE_OT_flatten_uv,
    BALLANCE_MT_ThreeDViewerMenu,

    OBJS_add_components.BALLANCE_OT_add_components,
    OBJS_add_rails.BALLANCE_OT_add_rails,
    OBJS_add_rails.BALLANCE_OT_add_tunnels,
    OBJS_add_floors.BALLANCE_OT_add_floors,
    BALLANCE_MT_AddFloorMenu,
    BALLANCE_MT_AddRailMenu,

    NAMES_rename_system.BALLANCE_OT_rename_by_group,
    NAMES_rename_system.BALLANCE_OT_convert_name,
    NAMES_rename_system.BALLANCE_OT_auto_grouping,

    UTILS_virtools_prop.BALLANCE_PG_virtools_material,
    UTILS_virtools_prop.BALLANCE_PG_virtools_group,
    PROPS_virtools_group.BALLANCE_OT_add_virtools_group,
    PROPS_virtools_group.BALLANCE_OT_rm_virtools_group,
    PROPS_virtools_group.BALLANCE_OT_clear_virtools_group,
    PROPS_virtools_group.BALLANCE_UL_virtools_group,
    PROPS_virtools_group.BALLANCE_PT_virtools_group,
    PROPS_virtools_material.BALLANCE_OT_apply_virtools_material,
    PROPS_virtools_material.BALLANCE_OT_parse_virtools_material,
    PROPS_virtools_material.BALLANCE_OT_preset_virtools_material,
    PROPS_virtools_material.BALLANCE_PT_virtools_material,

    OBJS_group_opers.BALLANCE_OT_select_virtools_group,
    OBJS_group_opers.BALLANCE_OT_filter_virtools_group,
    OBJS_group_opers.BALLANCE_OT_ctx_set_group,
    OBJS_group_opers.BALLANCE_OT_ctx_unset_group,
    OBJS_group_opers.BALLANCE_OT_ctx_clear_group,
    
)

def menu_func_bm_import(self, context):
    self.layout.operator(BMFILE_import.BALLANCE_OT_import_bm.bl_idname, text="Ballance Map (.bmx)")
def menu_func_bm_export(self, context):
    self.layout.operator(BMFILE_export.BALLANCE_OT_export_bm.bl_idname, text="Ballance Map (.bmx)")
def menu_func_ballance_3d(self, context):
    layout = self.layout
    layout.menu(BALLANCE_MT_ThreeDViewerMenu.bl_idname)
def menu_func_ballance_add(self, context):
    layout = self.layout
    layout.separator()
    layout.label(text="Ballance")
    layout.menu(BALLANCE_MT_AddFloorMenu.bl_idname, icon='MESH_CUBE')
    layout.menu(BALLANCE_MT_AddRailMenu.bl_idname, icon='MESH_CIRCLE')
    layout.operator_menu_enum(
        OBJS_add_components.BALLANCE_OT_add_components.bl_idname, 
        "elements_type", icon='MESH_ICOSPHERE', text="Elements")
def menu_func_ballance_rename(self, context):
    layout = self.layout
    layout.separator()
    col = layout.column()
    col.operator_context = 'INVOKE_DEFAULT'
    col.label(text="Ballance")
    col.operator(NAMES_rename_system.BALLANCE_OT_rename_by_group.bl_idname, icon='GREASEPENCIL')
    col.operator(NAMES_rename_system.BALLANCE_OT_convert_name.bl_idname, icon='ARROW_LEFTRIGHT')
    col.operator(NAMES_rename_system.BALLANCE_OT_auto_grouping.bl_idname, icon='GROUP')
def menu_func_ballance_select(self, context):
    layout = self.layout
    layout.separator()
    layout.label(text="Ballance")
    layout.operator(OBJS_group_opers.BALLANCE_OT_select_virtools_group.bl_idname, icon='SELECT_SET')
    layout.operator(OBJS_group_opers.BALLANCE_OT_filter_virtools_group.bl_idname, icon='FILTER')
def menu_func_ballance_grouping(self, context):
    layout = self.layout
    layout.separator()
    col = layout.column()
    col.operator_context = 'INVOKE_DEFAULT'
    col.label(text="Ballance")
    col.operator(OBJS_group_opers.BALLANCE_OT_ctx_set_group.bl_idname, icon='ADD', text="Group into...")
    col.operator(OBJS_group_opers.BALLANCE_OT_ctx_unset_group.bl_idname, icon='REMOVE', text="Ungroup from...")
    col.operator(OBJS_group_opers.BALLANCE_OT_ctx_clear_group.bl_idname, icon='TRASH', text="Clear Grouping")


def register():
    # we need init all icon first
    UTILS_icons_manager.register_icons()

    for cls in classes:
        bpy.utils.register_class(cls)

    bpy.types.Scene.BallanceBlenderPluginProperty = bpy.props.PointerProperty(type=UTILS_preferences.MyPropertyGroup)
    UTILS_virtools_prop.register_props()

    bpy.types.TOPBAR_MT_file_import.append(menu_func_bm_import)
    bpy.types.TOPBAR_MT_file_export.append(menu_func_bm_export)

    bpy.types.VIEW3D_MT_editor_menus.prepend(menu_func_ballance_3d)
    bpy.types.VIEW3D_MT_add.append(menu_func_ballance_add)
    bpy.types.OUTLINER_MT_collection.append(menu_func_ballance_rename)

    bpy.types.VIEW3D_MT_select_object.append(menu_func_ballance_select)

    bpy.types.VIEW3D_MT_object_context_menu.append(menu_func_ballance_grouping)
    bpy.types.OUTLINER_MT_object.append(menu_func_ballance_grouping)    # share the same menu
    

def unregister():
    bpy.types.TOPBAR_MT_file_import.remove(menu_func_bm_import)
    bpy.types.TOPBAR_MT_file_export.remove(menu_func_bm_export)
    
    bpy.types.VIEW3D_MT_editor_menus.remove(menu_func_ballance_3d)
    bpy.types.VIEW3D_MT_add.remove(menu_func_ballance_add)
    bpy.types.OUTLINER_MT_collection.remove(menu_func_ballance_rename)

    bpy.types.VIEW3D_MT_select_object.remove(menu_func_ballance_select)

    bpy.types.VIEW3D_MT_object_context_menu.remove(menu_func_ballance_grouping)
    bpy.types.OUTLINER_MT_object.append(menu_func_ballance_grouping)

    UTILS_virtools_prop.unregister_props()
    del bpy.types.Scene.BallanceBlenderPluginProperty

    for cls in classes:
        bpy.utils.unregister_class(cls)

    # we need uninstall all icon after all classes unregister
    UTILS_icons_manager.unregister_icons()
    
if __name__=="__main__":
	register()