suwoodblender/blenderpython/suw_core/__init__.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
SUW Core Module - 核心模块集合
拆分自: suw_impl.py
用途: 将大文件拆分为可维护的模块
版本: 1.0.0 (阶段6完成 - 最终版本)
作者: SUWood Team
"""

# 阶段1: 导入内存管理模块
import suw_core.command_dispatcher as cd_module
import suw_core.dimension_manager as dim_module
import suw_core.door_drawer_manager as ddm_module
import suw_core.hardware_manager as hw_module
import suw_core.machining_manager as mach_module
import suw_core.selection_manager as sm_module
import suw_core.deletion_manager as dm_module
import suw_core.part_creator as pc_module
import suw_core.material_manager as mm_module
import suw_core.data_manager as data_module
import logging
from .memory_manager import (
    BlenderMemoryManager,
    DependencyGraphManager,
    memory_manager,
    dependency_manager,
    init_main_thread,
    execute_in_main_thread_async,
    execute_in_main_thread,
    process_main_thread_tasks,
    safe_blender_operation
)

# 阶段1: 导入几何工具模块
from .geometry_utils import (
    Point3d,
    Vector3d,
    Transformation,
    MAT_TYPE_NORMAL,
    MAT_TYPE_OBVERSE,
    MAT_TYPE_REVERSE,
    MAT_TYPE_THIN,
    MAT_TYPE_NATURE
)

# 阶段0: 导入数据管理模块 (基础数据层)
from .data_manager import (
    DataManager,
    data_manager,
    init_data_manager,
    get_data_manager,
    get_zones,
    get_parts,
    get_hardwares,
    get_texture,
    sel_clear
)

# 阶段2: 导入材质管理模块
from .material_manager import (
    MaterialManager,
    material_manager,
    init_material_manager
)

# 阶段2: 导入部件创建模块
from .part_creator import (
    PartCreator,
    part_creator,
    init_part_creator
)

# 阶段3: 导入加工管理模块
from .machining_manager import (
    MachiningManager,
    machining_manager,
    init_machining_manager,
    get_machining_manager
)

# 阶段3: 导入选择管理模块
from .selection_manager import (
    SelectionManager,
    selection_manager,
    init_selection_manager,
    get_selection_manager
)

# 阶段4: 导入删除管理模块
from .deletion_manager import (
    DeletionManager,
    deletion_manager,
    init_deletion_manager,
    get_deletion_manager
)

# 阶段4: 导入五金管理模块
from .hardware_manager import (
    HardwareManager,
    hardware_manager,
    init_hardware_manager,
    get_hardware_manager
)

# 阶段5: 导出门抽屉管理模块
from .door_drawer_manager import (
    DoorDrawerManager,
    door_drawer_manager,
    init_door_drawer_manager
)

# 阶段5: 导入尺寸标注管理模块
from .dimension_manager import (
    DimensionManager,
    dimension_manager,
    init_dimension_manager
)

# 阶段6: 导入命令分发器模块
from .command_dispatcher import (
    CommandDispatcher,
    command_dispatcher,
    init_command_dispatcher,
    get_dispatcher_stats
)

# ==================== 版本信息 ====================

__version__ = "1.0.0"
__author__ = "SUWood Team"
__description__ = "SUW Core - 模块化SUWImpl核心功能"

# ==================== 拆分进度 ====================

REFACTOR_STATUS = {
    "阶段1": "✅ 完成 - 内存管理器 + 几何工具",
    "阶段2": "✅ 完成 - 材质管理器 + 部件创建器",
    "阶段3": "✅ 完成 - 加工管理器 + 选择管理器",
    "阶段4": "✅ 完成 - 删除管理器 + 五金管理器",
    "阶段5": "✅ 完成 - 门抽屉管理器 + 尺寸标注管理器",
    "阶段6": "✅ 完成 - 命令分发器 + 最终整合",
}

# ==================== 模块初始化函数 ====================


def init_all_managers():
    """
    初始化所有管理器 - 完全独立，不再需要suw_impl

    Returns:
        dict: 包含所有管理器实例的字典
    """
    try:
        logger.info("🚀 开始初始化所有独立管理器...")

        managers = {}

        # 1. 先初始化数据管理器（最基础）
        from .data_manager import init_data_manager
        managers['data_manager'] = init_data_manager()
        logger.info("✅ DataManager 初始化完成")

        # 2. 初始化材质管理器
        from .material_manager import init_material_manager
        managers['material_manager'] = init_material_manager()
        logger.info("✅ MaterialManager 初始化完成")

        # 3. 初始化部件创建器
        from .part_creator import init_part_creator
        managers['part_creator'] = init_part_creator()
        logger.info("✅ PartCreator 初始化完成")

        # 4. 初始化加工管理器
        from .machining_manager import init_machining_manager
        managers['machining_manager'] = init_machining_manager()
        logger.info("✅ MachiningManager 初始化完成")

        # 5. 初始化选择管理器
        from .selection_manager import init_selection_manager
        managers['selection_manager'] = init_selection_manager()
        logger.info("✅ SelectionManager 初始化完成")

        # 6. 初始化删除管理器
        from .deletion_manager import init_deletion_manager
        managers['deletion_manager'] = init_deletion_manager()
        logger.info("✅ DeletionManager 初始化完成")

        # 7. 初始化硬件管理器
        from .hardware_manager import init_hardware_manager
        managers['hardware_manager'] = init_hardware_manager()
        logger.info("✅ HardwareManager 初始化完成")

        # 8. 初始化门抽屉管理器
        from .door_drawer_manager import init_door_drawer_manager
        managers['door_drawer_manager'] = init_door_drawer_manager()
        logger.info("✅ DoorDrawerManager 初始化完成")

        # 9. 初始化尺寸管理器
        from .dimension_manager import init_dimension_manager
        managers['dimension_manager'] = init_dimension_manager()
        logger.info("✅ DimensionManager 初始化完成")

        # 10. 最后初始化命令分发器（依赖其他所有管理器）
        from .command_dispatcher import init_command_dispatcher
        managers['command_dispatcher'] = init_command_dispatcher()
        logger.info("✅ CommandDispatcher 初始化完成")

        logger.info(f"🎉 所有管理器初始化完成！共 {len(managers)} 个管理器")
        return managers

    except Exception as e:
        logger.error(f"❌ 管理器初始化失败: {e}")
        import traceback
        logger.error(traceback.format_exc())
        return {}


def get_all_stats():
    """获取所有管理器的统计信息"""
    try:
        all_stats = {
            "module_version": __version__,
            "memory_manager": memory_manager.get_memory_stats() if memory_manager else None,
            "material_manager": material_manager.get_material_stats() if material_manager else None,
            "part_creator": part_creator.get_part_creator_stats() if part_creator else None,
            "machining_manager": machining_manager.get_machining_stats() if machining_manager else None,
            "selection_manager": selection_manager.get_selection_stats() if selection_manager else None,
            "deletion_manager": deletion_manager.get_deletion_stats() if deletion_manager else None,
            "hardware_manager": hardware_manager.get_hardware_stats() if hardware_manager else None,
            "door_drawer_manager": door_drawer_manager.get_door_drawer_stats() if door_drawer_manager else None,
            "dimension_manager": dimension_manager.get_dimension_stats() if dimension_manager else None,
            "command_dispatcher": get_dispatcher_stats(),
        }
        return all_stats
    except Exception as e:
        print(f"获取统计信息失败: {e}")
        return {"error": str(e)}


# 为向后兼容添加别名
get_all_manager_stats = get_all_stats

# ==================== 模块导出 ====================


__all__ = [
    # 阶段0: 数据管理层 (基础数据层)
    "DataManager", "data_manager", "init_data_manager", "get_data_manager",
    "get_zones", "get_parts", "get_hardwares", "get_texture", "sel_clear",

    # 阶段1: 内存管理和几何工具
    "BlenderMemoryManager", "DependencyGraphManager",
    "memory_manager", "dependency_manager",
    "init_main_thread", "execute_in_main_thread_async",
    "execute_in_main_thread", "process_main_thread_tasks",
    "safe_blender_operation",
    "Point3d", "Vector3d", "Transformation",
    "MAT_TYPE_NORMAL", "MAT_TYPE_OBVERSE", "MAT_TYPE_REVERSE",
    "MAT_TYPE_THIN", "MAT_TYPE_NATURE",

    # 阶段2: 材质管理和部件创建
    "MaterialManager", "material_manager", "init_material_manager",
    "PartCreator", "part_creator", "init_part_creator",

    # 阶段3: 加工管理和选择管理
    "MachiningManager", "machining_manager", "init_machining_manager", "get_machining_manager",
    "SelectionManager", "selection_manager", "init_selection_manager", "get_selection_manager",

    # 阶段4: 删除管理和五金管理
    "DeletionManager", "deletion_manager", "init_deletion_manager", "get_deletion_manager",
    "HardwareManager", "hardware_manager", "init_hardware_manager", "get_hardware_manager",

    # 阶段5: 门抽屉管理和尺寸标注管理
    "DoorDrawerManager", "door_drawer_manager", "init_door_drawer_manager",
    "DimensionManager", "dimension_manager", "init_dimension_manager",

    # 阶段6: 命令分发
    "CommandDispatcher", "command_dispatcher", "init_command_dispatcher",
    "get_dispatcher_stats",

    # 工具函数
    "init_all_managers", "get_all_stats", "get_all_manager_stats",

    # 元信息
    "__version__", "__author__", "__description__", "REFACTOR_STATUS"
]

# 初始化日志
logger = logging.getLogger(__name__)
logger.info(f"SUW Core 模块加载完成 - {__version__} ({__author__})")

# 导出版本和阶段信息
__phase__ = "Phase 2"

# ==================== 【修复】强制初始化所有全局变量 ====================

# 确保所有管理器全局实例正确初始化
if mm_module.material_manager is None:
    print("🔧 强制初始化MaterialManager全局实例...")
    mm_module.material_manager = mm_module.MaterialManager()  # 【修复】移除None参数

if pc_module.part_creator is None:
    print("🔧 强制初始化PartCreator全局实例...")
    pc_module.part_creator = pc_module.PartCreator()  # 【修复】移除None参数

# 导入其他管理器模块

# 初始化其他管理器
if dm_module.deletion_manager is None:
    print("🔧 强制初始化DeletionManager全局实例...")
    dm_module.deletion_manager = dm_module.DeletionManager()

if sm_module.selection_manager is None:
    print("🔧 强制初始化SelectionManager全局实例...")
    sm_module.selection_manager = sm_module.SelectionManager()

if mach_module.machining_manager is None:
    print("🔧 强制初始化MachiningManager全局实例...")
    mach_module.machining_manager = mach_module.MachiningManager()

if hw_module.hardware_manager is None:
    print("🔧 强制初始化HardwareManager全局实例...")
    hw_module.hardware_manager = hw_module.HardwareManager()

if ddm_module.door_drawer_manager is None:
    print("🔧 强制初始化DoorDrawerManager全局实例...")
    ddm_module.door_drawer_manager = ddm_module.DoorDrawerManager()

if dim_module.dimension_manager is None:
    print("🔧 强制初始化DimensionManager全局实例...")
    dim_module.dimension_manager = dim_module.DimensionManager()

if cd_module.command_dispatcher is None:
    print("🔧 强制初始化CommandDispatcher全局实例...")
    cd_module.command_dispatcher = cd_module.CommandDispatcher()

# 强制初始化全局实例，确保能够独立运行
if not globals().get('_managers_initialized', False):
    _managers = init_all_managers()
    globals()['_managers_initialized'] = True

# 在文件末尾添加一个非阻塞的启动函数


def start_suwood_non_blocking():
    """非阻塞启动SUWood客户端（适合在Blender中使用）"""
    print("🚀 非阻塞启动SUWood客户端...")
    client_manager = start_suwood_blender_client()

    if client_manager:
        print("✅ 客户端已在后台运行")
        print("📝 使用以下命令管理客户端:")
        print("   print_system_status(suwood_client)")
        print("   suwood_client.stop_client()")

        # 设为全局变量方便访问
        globals()['suwood_client'] = client_manager
        return client_manager
    return None


# 修改主执行逻辑
if __name__ == "__main__":
    show_usage_guide()
    print("\n🚀 自动启动客户端...")

    # 检查是否在Blender环境中
    try:
        import bpy
        # 在Blender中，使用非阻塞启动
        start_suwood_non_blocking()
    except ImportError:
        # 在命令行中，使用原来的阻塞启动
        client_manager = start_suwood_blender_client()
        if client_manager:
            try:
                while True:
                    time.sleep(30)
                    print_system_status(client_manager)
            except KeyboardInterrupt:
                client_manager.stop_client()