#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ SUWood 常量定义 翻译自: SUWConstants.rb """ import os import logging from pathlib import Path from typing import Optional, Dict, Any # 设置日志 logger = logging.getLogger(__name__) # 检查Blender可用性 try: import bpy BLENDER_AVAILABLE = True except ImportError: BLENDER_AVAILABLE = False # 辅助函数：获取选择状态信息 def _get_selection_info(): """获取选择状态信息 - 替代SUWImpl的选择状态""" try: from .suw_core import get_selection_manager selection_manager = get_selection_manager() if selection_manager: return { 'selected_uid': selection_manager.selected_uid(), 'selected_obj': selection_manager.selected_obj(), 'selected_zone': selection_manager.selected_zone(), 'selected_part': selection_manager.selected_part() } except ImportError: pass # 如果无法获取，返回默认值 return { 'selected_uid': None, 'selected_obj': None, 'selected_zone': None, 'selected_part': None } def _get_server_path(): """获取服务器路径 - 替代SUWImpl.server_path""" try: from .suw_core import get_data_manager data_manager = get_data_manager() if data_manager and hasattr(data_manager, 'server_path'): return data_manager.server_path except ImportError: pass # 如果无法获取，返回默认路径 return os.path.dirname(__file__) class SUWood: """SUWood 主要常量和功能类""" # 场景操作常量 SUSceneNew = 1 # 清除之前的订单 SUSceneOpen = 2 # 清除之前的订单 SUSceneSave = 3 SUScenePrice = 4 # 单元操作常量 SUUnitPoint = 11 SUUnitFace = 12 SUUnitDelete = 13 SUUnitContour = 14 # 区域操作常量 SUZoneFront = 20 SUZoneDiv1 = 21 SUZoneResize = 22 SUZoneCombine = 23 SUZoneReplace = 24 SUZoneMaterial = 25 SUZoneHandle = 26 SUZoneCloth = 27 SUZoneLight = 28 # 空间位置常量 VSSpatialPos_F = 1 # 前 VSSpatialPos_K = 2 # 后 VSSpatialPos_L = 3 # 左 VSSpatialPos_R = 4 # 右 VSSpatialPos_B = 5 # 底 VSSpatialPos_T = 6 # 顶 # 单元轮廓常量 VSUnitCont_Zone = 1 # 区域轮廓 VSUnitCont_Part = 2 # 部件轮廓 VSUnitCont_Work = 3 # 挖洞轮廓 # 版本常量 V_Dealer = 1000 V_Machining = 1100 V_Division = 1200 V_PartCategory = 1300 V_Contour = 1400 V_Color = 1500 V_Profile = 1600 V_Surf = 1700 V_StretchPart = 1800 V_Material = 1900 V_Connection = 2000 V_HardwareSchema = 2050 V_HardwareSet = 2100 V_Hardware = 2200 V_Groove = 2300 V_DesignParam = 2400 V_ProfileSchema = 2500 V_StructPart = 2600 V_CraftPart = 2700 V_SeriesPart = 2800 V_Drawer = 2900 V_DesignTemplate = 3000 V_PriceTemplate = 3100 V_MachineCut = 3200 V_MachineCNC = 3300 V_CorpLabel = 3400 V_CorpCAM = 3500 V_PackLabel = 3600 V_Unit = 5000 # 路径常量 PATH = os.path.dirname(__file__) def __init__(self): """初始化SUWood实例""" pass @classmethod def icon_path(cls, icon_name, ext='png'): """获取图标路径""" return f"{cls.PATH}/icon/{icon_name}.{ext}" @classmethod def unit_path(cls): """获取单元路径""" try: server_path = _get_server_path() return f"{server_path}/drawings/Unit" except ImportError: return f"{cls.PATH}/drawings/Unit" @classmethod def suwood_path(cls, ref_v): """根据版本值获取SUWood路径""" try: server_path = _get_server_path() except ImportError: server_path = cls.PATH path_mapping = { cls.V_Material: f"{server_path}/images/texture", cls.V_StretchPart: f"{server_path}/drawings/StretchPart", cls.V_StructPart: f"{server_path}/drawings/StructPart", cls.V_Unit: f"{server_path}/drawings/Unit", cls.V_Connection: f"{server_path}/drawings/Connection", cls.V_HardwareSet: f"{server_path}/drawings/HardwareSet", cls.V_Hardware: f"{server_path}/drawings/Hardware", } return path_mapping.get(ref_v, server_path) @classmethod def suwood_pull_size(cls, pos): """根据位置获取拉手尺寸类型""" size_mapping = { 1: "HW", # 右上 2: "W", # 右中 3: "HW", # 右下 4: "H", # 中上 6: "H", # 中下 11: "HW", # 右上-竖 12: "W", # 右中-竖 13: "HW", # 右下-竖 14: "H", # 中上-竖 16: "H", # 中下-竖 21: "HW", # 右上-横 22: "W", # 右中-横 23: "HW", # 右下-横 24: "H", # 中上-横 26: "H", # 中下-横 } return size_mapping.get(pos) @classmethod def scene_save(cls): """保存场景""" try: import bpy # Blender Python API scene = bpy.context.scene order_id = scene.get("order_id") if order_id is None: return data = { "method": cls.SUSceneSave, "order_id": order_id } cls.set_cmd("r00", data) if not bpy.data.filepath: server_path = _get_server_path() scene_path = Path(f"{server_path}/blender") scene_path.mkdir(exist_ok=True) order_code = scene.get("order_code", "untitled") filepath = scene_path / f"{order_code}.blend" bpy.ops.wm.save_as_mainfile(filepath=str(filepath)) else: bpy.ops.wm.save_mainfile() except ImportError: print("Blender API not available - scene_save not implemented") @classmethod def scene_price(cls): """场景价格计算""" try: import bpy scene = bpy.context.scene order_id = scene.get("order_id") if order_id is None: return params = { "method": cls.SUScenePrice, "order_id": order_id } cls.set_cmd("r00", params) except ImportError: print("Blender API not available - scene_price not implemented") @classmethod def import_unit(cls, uid, values, mold): """点击创体(产品UID)""" # 原本激活SketchUp工具，这里需要适配到Blender try: from .suw_unit_point_tool import SUWUnitPointTool # 创建单元点工具 width = values.get("width", 0) * 0.001 # 转换为米 depth = values.get("depth", 0) * 0.001 height = values.get("height", 0) * 0.001 tool = SUWUnitPointTool(width, depth, height, uid, mold) # 在Blender中激活工具的逻辑需要根据具体实现 print(f"激活单元点工具: {uid}, 尺寸: {width}x{depth}x{height}") except ImportError: print("SUWUnitPointTool not available") @classmethod def import_face(cls, uid, values, mold): """选面创体(产品UID)""" try: from .suw_unit_face_tool import SUWUnitFaceTool tool = SUWUnitFaceTool(cls.VSSpatialPos_F, uid, mold) print(f"激活单元面工具: {uid}") except ImportError: print("SUWUnitFaceTool not available") @classmethod def front_view(cls): """前视图""" try: selection_info = _get_selection_info() uid = selection_info['selected_uid'] obj = selection_info['selected_obj'] if uid is None or obj is None: print("请先选择正视于的基准面!") return params = { "method": cls.SUZoneFront, "uid": uid, "oid": obj } cls.set_cmd("r00", params) except ImportError: print("无法获取选择信息") @classmethod def delete_unit(cls): """删除单元""" try: import bpy selection_info = _get_selection_info() scene = bpy.context.scene order_id = scene.get("order_id") uid = selection_info['selected_uid'] obj = selection_info['selected_obj'] if uid is None: print("请先选择待删除的柜体!") return elif order_id is None: print("当前柜体不是场景方案的柜体!") return # 在实际应用中，这里应该有确认对话框 # 现在简化为直接执行 params = { "method": cls.SUUnitDelete, "order_id": order_id, "uid": uid } if obj: params["oid"] = obj cls.set_cmd("r00", params) except ImportError: print("Blender API or SUWImpl not available") @classmethod def combine_unit(cls, uid, values, mold): """模块拼接""" try: selection_info = _get_selection_info() selected_zone = selection_info['selected_zone'] if selected_zone is None: print("请先选择待拼接的空区域!") return params = { "method": cls.SUZoneCombine, "uid": selected_zone.get("uid"), "zid": selected_zone.get("zid"), "source": uid } if mold: params["module"] = mold cls.set_cmd("r00", params) except ImportError: print("无法获取选择信息") @classmethod def replace_unit(cls, uid, values, mold): """模块/产品替换""" try: selection_info = _get_selection_info() if selection_info['selected_zone'] is None and (mold == 1 or mold == 2): print("请先选择待替换的区域!") return elif selection_info['selected_obj'] is None and (mold == 3): print("请先选择待替换的部件!") return params = { "method": cls.SUZoneReplace, "source": uid, "module": mold } cls.set_cmd("r00", params) except ImportError: print("无法获取选择信息") @classmethod def replace_mat(cls, uid, values, mat_type): """材料替换""" try: selection_info = _get_selection_info() selected_zone = selection_info['selected_zone'] if selected_zone is None: print("请先选择待替换材料的区域!") return params = { "method": cls.SUZoneMaterial, "mat_id": uid, "type": mat_type } cls.set_cmd("r00", params) except ImportError: print("无法获取选择信息") @classmethod def replace_handle(cls, width, height, set_id, conn_id): """替换拉手""" try: selection_info = _get_selection_info() selected_zone = selection_info['selected_zone'] if selected_zone is None: print("请先选择待替换拉手的区域!") return params = { "method": cls.SUZoneHandle, "uid": selected_zone.get("uid"), "zid": selected_zone.get("zid"), "conn_id": conn_id, "set_id": set_id } if width is not None and width != "": params["width"] = int(width) if height is not None and height != "": params["height"] = int(height) cls.set_cmd("r00", params) except ImportError: print("无法获取选择信息") @classmethod def clear_current(cls, ref_v): """清除当前选择""" try: selection_info = _get_selection_info() if (ref_v == 2102 or ref_v == 2103) and selection_info['selected_zone']: params = { "uid": selection_info['selected_uid'] } cls.set_cmd("r01", params) # 清除选择 from .suw_core import get_selection_manager selection_manager = get_selection_manager() if selection_manager: selection_manager.sel_clear() except ImportError: print("无法获取选择信息") @classmethod def replace_clothes(cls, front, back, set_id, conn_id): """挂衣杆替换""" try: selection_info = _get_selection_info() selected_zone = selection_info['selected_zone'] if selected_zone is None: print("请先选择待替换衣杆的区域!") return params = { "method": cls.SUZoneCloth, "uid": selected_zone.get("uid"), "zid": selected_zone.get("zid"), "conn_id": conn_id, "set_id": set_id } if front != 0: params["front"] = front if back != 0: params["back"] = back cls.set_cmd("r00", params) except ImportError: print("无法获取选择信息") @classmethod def replace_lights(cls, front, back, set_id, conn_id): """灯带替换""" try: selection_info = _get_selection_info() selected_zone = selection_info['selected_zone'] if selected_zone is None: print("请先选择待替换灯带的区域!") return # 处理连接ID（可能是数组） if isinstance(conn_id, list): conns = ",".join(map(str, conn_id)) else: conns = str(conn_id) params = { "method": cls.SUZoneLight, "uid": selected_zone.get("uid"), "zid": selected_zone.get("zid"), "conn_id": conns, "set_id": set_id } if front != 0: params["front"] = front if back != 0: params["back"] = back cls.set_cmd("r00", params) except ImportError: print("无法获取选择信息") @classmethod def set_cmd(cls, cmd_type, params): """设置命令""" try: from .suw_client import set_cmd set_cmd(cmd_type, params) except ImportError: print(f"Command: {cmd_type}, Params: {params}") # 创建全局实例 suwood = SUWood() # 导出所有常量到模块级别，便于其他模块使用 # 场景操作常量 SUSceneNew = SUWood.SUSceneNew SUSceneOpen = SUWood.SUSceneOpen SUSceneSave = SUWood.SUSceneSave SUScenePrice = SUWood.SUScenePrice # 单元操作常量 SUUnitPoint = SUWood.SUUnitPoint SUUnitFace = SUWood.SUUnitFace SUUnitDelete = SUWood.SUUnitDelete SUUnitContour = SUWood.SUUnitContour # 区域操作常量 SUZoneFront = SUWood.SUZoneFront SUZoneDiv1 = SUWood.SUZoneDiv1 SUZoneResize = SUWood.SUZoneResize SUZoneCombine = SUWood.SUZoneCombine SUZoneReplace = SUWood.SUZoneReplace SUZoneMaterial = SUWood.SUZoneMaterial SUZoneHandle = SUWood.SUZoneHandle SUZoneCloth = SUWood.SUZoneCloth SUZoneLight = SUWood.SUZoneLight # 空间位置常量 VSSpatialPos_F = SUWood.VSSpatialPos_F VSSpatialPos_K = SUWood.VSSpatialPos_K VSSpatialPos_L = SUWood.VSSpatialPos_L VSSpatialPos_R = SUWood.VSSpatialPos_R VSSpatialPos_B = SUWood.VSSpatialPos_B VSSpatialPos_T = SUWood.VSSpatialPos_T # 单元轮廓常量 VSUnitCont_Zone = SUWood.VSUnitCont_Zone VSUnitCont_Part = SUWood.VSUnitCont_Part VSUnitCont_Work = SUWood.VSUnitCont_Work # 版本常量 V_Dealer = SUWood.V_Dealer V_Machining = SUWood.V_Machining V_Division = SUWood.V_Division V_PartCategory = SUWood.V_PartCategory V_Contour = SUWood.V_Contour V_Color = SUWood.V_Color V_Profile = SUWood.V_Profile V_Surf = SUWood.V_Surf V_StretchPart = SUWood.V_StretchPart V_Material = SUWood.V_Material V_Connection = SUWood.V_Connection V_HardwareSchema = SUWood.V_HardwareSchema V_HardwareSet = SUWood.V_HardwareSet V_Hardware = SUWood.V_Hardware V_Groove = SUWood.V_Groove V_DesignParam = SUWood.V_DesignParam V_ProfileSchema = SUWood.V_ProfileSchema V_StructPart = SUWood.V_StructPart V_CraftPart = SUWood.V_CraftPart V_SeriesPart = SUWood.V_SeriesPart V_Drawer = SUWood.V_Drawer V_DesignTemplate = SUWood.V_DesignTemplate V_PriceTemplate = SUWood.V_PriceTemplate V_MachineCut = SUWood.V_MachineCut V_MachineCNC = SUWood.V_MachineCNC V_CorpLabel = SUWood.V_CorpLabel V_CorpCAM = SUWood.V_CorpCAM V_PackLabel = SUWood.V_PackLabel V_Unit = SUWood.V_Unit # 路径常量 PATH = SUWood.PATH # 导出所有类方法为模块级别函数 icon_path = SUWood.icon_path unit_path = SUWood.unit_path suwood_path = SUWood.suwood_path suwood_pull_size = SUWood.suwood_pull_size scene_save = SUWood.scene_save scene_price = SUWood.scene_price import_unit = SUWood.import_unit import_face = SUWood.import_face front_view = SUWood.front_view delete_unit = SUWood.delete_unit combine_unit = SUWood.combine_unit replace_unit = SUWood.replace_unit replace_mat = SUWood.replace_mat replace_handle = SUWood.replace_handle clear_current = SUWood.clear_current replace_clothes = SUWood.replace_clothes replace_lights = SUWood.replace_lights set_cmd = SUWood.set_cmd