565 lines
18 KiB
Python
565 lines
18 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
SUW Unit Face Tool - Python完整翻译版本
|
||
原文件: SUWUnitFaceTool.rb
|
||
用途: 选面创体工具，用于在选中的面上创建单元
|
||
"""
|
||
|
||
import logging
|
||
import math
|
||
from typing import Optional, List, Tuple, Dict, Any
|
||
|
||
# 尝试导入Blender模块
|
||
try:
|
||
import bpy
|
||
import bmesh
|
||
import mathutils
|
||
from bpy_extras import view3d_utils
|
||
BLENDER_AVAILABLE = True
|
||
except ImportError:
|
||
BLENDER_AVAILABLE = False
|
||
|
||
try:
|
||
from .suw_constants import *
|
||
from .suw_client import set_cmd
|
||
except ImportError:
|
||
# 绝对导入作为后备
|
||
try:
|
||
from suw_constants import *
|
||
from suw_client import set_cmd
|
||
except ImportError as e:
|
||
print(f"⚠️ 导入SUWood模块失败: {e}")
|
||
# 提供默认实现
|
||
|
||
def set_cmd(cmd, params):
|
||
print(f"Command: {cmd}, Params: {params}")
|
||
|
||
# 提供缺失的常量
|
||
VSSpatialPos_F = 1 # 前
|
||
VSSpatialPos_R = 4 # 右
|
||
VSSpatialPos_T = 6 # 顶
|
||
SUUnitFace = 12
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class SUWUnitFaceTool:
|
||
"""SUWood选面创体工具 - 完整翻译版本"""
|
||
|
||
def __init__(self, cont_view: int, source: Optional[str] = None, mold: bool = False):
|
||
"""初始化选面创体工具"""
|
||
self.cont_view = cont_view
|
||
self.source = source
|
||
self.mold = mold
|
||
self.tooltip = '请点击要创体的面'
|
||
|
||
# 当前选中的面
|
||
self.ref_face = None
|
||
self.trans_arr = None
|
||
self.face_segs = None
|
||
|
||
print(f"🔧 创建选面创体工具: 视图={cont_view}")
|
||
|
||
def activate(self):
|
||
"""激活工具"""
|
||
self._set_status_text(self.tooltip)
|
||
print("⚡ 激活选面创体工具")
|
||
|
||
def on_mouse_move(self, flags: int, x: float, y: float, view=None):
|
||
"""鼠标移动事件"""
|
||
# 重置当前状态
|
||
self.ref_face = None
|
||
self.trans_arr = None
|
||
self.face_segs = None
|
||
|
||
if BLENDER_AVAILABLE:
|
||
self._blender_pick_face(x, y, view)
|
||
else:
|
||
self._stub_pick_face(x, y)
|
||
|
||
self._set_status_text(self.tooltip)
|
||
self._invalidate_view()
|
||
|
||
def _blender_pick_face(self, x: float, y: float, view=None):
|
||
"""Blender中拾取面"""
|
||
try:
|
||
# 获取视图信息
|
||
region = bpy.context.region
|
||
rv3d = bpy.context.region_data
|
||
|
||
if region is None or rv3d is None:
|
||
return
|
||
|
||
# 创建拾取射线
|
||
view_vector = view3d_utils.region_2d_to_vector_3d(
|
||
region, rv3d, (x, y))
|
||
ray_origin = view3d_utils.region_2d_to_origin_3d(
|
||
region, rv3d, (x, y))
|
||
|
||
# 执行射线检测
|
||
result, location, normal, index, obj, matrix = bpy.context.scene.ray_cast(
|
||
bpy.context.view_layer.depsgraph, ray_origin, view_vector
|
||
)
|
||
|
||
if result and obj and obj.type == 'MESH':
|
||
# 获取面信息
|
||
mesh = obj.data
|
||
face = mesh.polygons[index]
|
||
|
||
# 检查面是否有效
|
||
if self._face_valid(face, obj):
|
||
# 获取面的顶点位置
|
||
face_pts = []
|
||
for vert_idx in face.vertices:
|
||
vert_co = mesh.vertices[vert_idx].co
|
||
# 应用对象变换
|
||
world_co = obj.matrix_world @ vert_co
|
||
face_pts.append(world_co)
|
||
|
||
# 构建变换数组
|
||
trans_arr = []
|
||
if obj.matrix_world != mathutils.Matrix.Identity(4):
|
||
trans_arr.append(obj.matrix_world)
|
||
|
||
self.ref_face = face
|
||
self.trans_arr = trans_arr
|
||
|
||
# 构建面边段用于绘制
|
||
self.face_segs = []
|
||
for i in range(len(face_pts)):
|
||
next_i = (i + 1) % len(face_pts)
|
||
self.face_segs.append([face_pts[i], face_pts[next_i]])
|
||
|
||
print(f"🎯 拾取到面: {len(face_pts)}个顶点")
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ Blender面拾取失败: {e}")
|
||
|
||
def _stub_pick_face(self, x: float, y: float):
|
||
"""存根模式面拾取"""
|
||
# 模拟拾取到一个面
|
||
if x % 50 == 0: # 简单的命中检测
|
||
self.ref_face = {"type": "stub_face", "id": 1}
|
||
self.face_segs = [
|
||
[(0, 0, 0), (1, 0, 0)],
|
||
[(1, 0, 0), (1, 1, 0)],
|
||
[(1, 1, 0), (0, 1, 0)],
|
||
[(0, 1, 0), (0, 0, 0)]
|
||
]
|
||
print("🎯 存根模式拾取到面")
|
||
|
||
def on_l_button_down(self, flags: int, x: float, y: float, view=None):
|
||
"""鼠标左键点击事件"""
|
||
# 如果没有选中面，尝试再次拾取
|
||
if self.ref_face is None:
|
||
self.on_mouse_move(flags, x, y, view)
|
||
|
||
# 检查是否选中了有效面
|
||
if self.ref_face is None:
|
||
self._show_message('请选择要放置的面')
|
||
return
|
||
|
||
# 弹出输入框
|
||
inputs = self._show_input_dialog()
|
||
if inputs is False or inputs[4] < 100:
|
||
return
|
||
|
||
# 获取订单ID
|
||
order_id = self._get_order_id()
|
||
|
||
# 处理前沿边（仅对顶视图）
|
||
fronts = []
|
||
if self.cont_view == VSSpatialPos_T:
|
||
fronts = self._process_top_view_fronts()
|
||
|
||
# 构建参数
|
||
params = self._build_parameters(inputs, fronts)
|
||
|
||
# 构建数据
|
||
data = {}
|
||
data["method"] = SUUnitFace
|
||
if order_id is not None:
|
||
data["order_id"] = order_id
|
||
data["params"] = params
|
||
|
||
# 发送命令
|
||
set_cmd("r00", data)
|
||
|
||
# 清理和重置
|
||
self._cleanup_after_creation()
|
||
|
||
print(f"🏗️ 选面创体完成: 视图={self.cont_view}, 尺寸={inputs[4]}")
|
||
|
||
def _show_input_dialog(self):
|
||
"""显示输入对话框"""
|
||
try:
|
||
# 根据视图类型确定尺寸标题和默认值
|
||
caption = ""
|
||
default = 0
|
||
|
||
if self.cont_view == VSSpatialPos_F:
|
||
caption = '深(mm)'
|
||
default = 600
|
||
elif self.cont_view == VSSpatialPos_R:
|
||
caption = '宽(mm)'
|
||
default = 800
|
||
elif self.cont_view == VSSpatialPos_T:
|
||
caption = '高(mm)'
|
||
default = 800
|
||
|
||
if BLENDER_AVAILABLE:
|
||
# Blender输入框实现
|
||
return self._blender_input_dialog(caption, default)
|
||
else:
|
||
# 存根模式输入框
|
||
return self._stub_input_dialog(caption, default)
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 输入对话框失败: {e}")
|
||
return False
|
||
|
||
def _blender_input_dialog(self, caption: str, default: int):
|
||
"""Blender输入对话框"""
|
||
try:
|
||
# 这里需要通过Blender的operator系统实现输入框
|
||
# 暂时使用默认值
|
||
inputs = [0, 0, 0, 0, default, "合并"]
|
||
print(
|
||
f"📐 Blender输入: 距左=0, 距右=0, 距上=0, 距下=0, {caption}={default}, 重叠=合并")
|
||
return inputs
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ Blender输入框失败: {e}")
|
||
return False
|
||
|
||
def _stub_input_dialog(self, caption: str, default: int):
|
||
"""存根模式输入对话框"""
|
||
inputs = [0, 0, 0, 0, default, "合并"]
|
||
print(f"📐 选面创体输入: 距左=0, 距右=0, 距上=0, 距下=0, {caption}={default}, 重叠=合并")
|
||
return inputs
|
||
|
||
def _process_top_view_fronts(self):
|
||
"""处理顶视图的前沿边"""
|
||
fronts = []
|
||
|
||
try:
|
||
if not self.ref_face:
|
||
return fronts
|
||
|
||
if BLENDER_AVAILABLE:
|
||
# Blender中处理边
|
||
fronts = self._blender_process_fronts()
|
||
else:
|
||
# 存根模式
|
||
fronts = [
|
||
{"s": "0,0,0", "e": "1000,0,0"},
|
||
{"s": "1000,0,0", "e": "1000,1000,0"}
|
||
]
|
||
|
||
print(f"🔄 处理前沿边: {len(fronts)}条")
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 处理前沿边失败: {e}")
|
||
|
||
return fronts
|
||
|
||
def _blender_process_fronts(self):
|
||
"""Blender中处理前沿边"""
|
||
fronts = []
|
||
|
||
try:
|
||
# 这里需要实现复杂的边处理逻辑
|
||
# 类似Ruby中的edge.faces.select逻辑
|
||
|
||
# 暂时返回空列表
|
||
print("🔄 Blender前沿边处理")
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ Blender前沿边处理失败: {e}")
|
||
|
||
return fronts
|
||
|
||
def _build_parameters(self, inputs, fronts):
|
||
"""构建参数字典"""
|
||
params = {}
|
||
params["view"] = self.cont_view
|
||
params["face"] = self._face_to_json()
|
||
|
||
# 添加边距参数
|
||
if inputs[0] > 0:
|
||
params["left"] = inputs[0]
|
||
if inputs[1] > 0:
|
||
params["right"] = inputs[1]
|
||
if inputs[2] > 0:
|
||
params["top"] = inputs[2]
|
||
if inputs[3] > 0:
|
||
params["bottom"] = inputs[3]
|
||
|
||
params["size"] = inputs[4]
|
||
|
||
# 添加合并参数
|
||
if inputs[5] == "合并":
|
||
params["merged"] = True
|
||
|
||
# 添加可选参数
|
||
if self.source is not None:
|
||
params["source"] = self.source
|
||
if self.mold:
|
||
params["module"] = self.mold
|
||
if len(fronts) > 0:
|
||
params["fronts"] = fronts
|
||
|
||
return params
|
||
|
||
def _face_to_json(self):
|
||
"""将面转换为JSON格式"""
|
||
try:
|
||
if BLENDER_AVAILABLE and self.ref_face:
|
||
return self._blender_face_to_json()
|
||
else:
|
||
return self._stub_face_to_json()
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 面转JSON失败: {e}")
|
||
return {}
|
||
|
||
def _blender_face_to_json(self):
|
||
"""Blender面转JSON"""
|
||
try:
|
||
# 这里需要实现类似SketchUp Face.to_json的功能
|
||
# 包含变换数组和精度参数
|
||
|
||
json_data = {
|
||
"segs": [],
|
||
"normal": [0, 0, 1],
|
||
"area": 1.0,
|
||
"transform": self.trans_arr if self.trans_arr else []
|
||
}
|
||
|
||
print("🔄 Blender面转JSON")
|
||
return json_data
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ Blender面转JSON失败: {e}")
|
||
return {}
|
||
|
||
def _stub_face_to_json(self):
|
||
"""存根面转JSON"""
|
||
return {
|
||
"segs": [
|
||
{"s": "0,0,0", "e": "1000,0,0"},
|
||
{"s": "1000,0,0", "e": "1000,1000,0"},
|
||
{"s": "1000,1000,0", "e": "0,1000,0"},
|
||
{"s": "0,1000,0", "e": "0,0,0"}
|
||
],
|
||
"normal": [0, 0, 1],
|
||
"area": 1000000, # 1平方米，单位mm²
|
||
"type": "stub"
|
||
}
|
||
|
||
def _cleanup_after_creation(self):
|
||
"""创建后清理"""
|
||
try:
|
||
# 删除选中的面和相关边
|
||
if BLENDER_AVAILABLE and self.ref_face:
|
||
# 在Blender中删除面
|
||
# 这需要进入编辑模式并删除选中的面
|
||
print("🧹 Blender面清理")
|
||
|
||
# 重置状态
|
||
self.ref_face = None
|
||
self.trans_arr = None
|
||
self.face_segs = None
|
||
|
||
# 刷新视图
|
||
self._invalidate_view()
|
||
|
||
# 清除选择并停用工具
|
||
self._clear_selection()
|
||
self._select_tool(None)
|
||
|
||
print("🧹 创建后清理完成")
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 创建后清理失败: {e}")
|
||
|
||
def draw(self, view=None):
|
||
"""绘制工具预览"""
|
||
if self.face_segs:
|
||
if BLENDER_AVAILABLE:
|
||
self._draw_blender()
|
||
else:
|
||
self._draw_stub()
|
||
|
||
def _draw_blender(self):
|
||
"""Blender绘制高亮面"""
|
||
try:
|
||
import gpu
|
||
from gpu_extras.batch import batch_for_shader
|
||
|
||
if not self.face_segs:
|
||
return
|
||
|
||
# 准备线条数据
|
||
lines = []
|
||
for seg in self.face_segs:
|
||
lines.extend([seg[0], seg[1]])
|
||
|
||
# 绘制青色高亮线条
|
||
shader = gpu.shader.from_builtin('3D_UNIFORM_COLOR')
|
||
batch = batch_for_shader(shader, 'LINES', {"pos": lines})
|
||
shader.bind()
|
||
shader.uniform_float("color", (0, 1, 1, 1)) # 青色
|
||
|
||
# 设置线宽
|
||
import bgl
|
||
bgl.glLineWidth(3)
|
||
|
||
batch.draw(shader)
|
||
|
||
# 重置线宽
|
||
bgl.glLineWidth(1)
|
||
|
||
print("🎨 Blender高亮绘制")
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ Blender绘制失败: {e}")
|
||
|
||
def _draw_stub(self):
|
||
"""存根绘制"""
|
||
print(f"🎨 绘制高亮面: {len(self.face_segs)}条边")
|
||
|
||
def _face_valid(self, face, obj):
|
||
"""检查面是否有效"""
|
||
try:
|
||
if not face:
|
||
return False
|
||
|
||
if BLENDER_AVAILABLE:
|
||
# 获取面法向量
|
||
normal = face.normal
|
||
|
||
# 根据视图类型检查法向量
|
||
if self.cont_view == VSSpatialPos_F:
|
||
# 前视图：法向量应垂直于Z轴
|
||
return abs(normal.z) < 0.1
|
||
elif self.cont_view == VSSpatialPos_R:
|
||
# 右视图：法向量应垂直于Z轴
|
||
return abs(normal.z) < 0.1
|
||
elif self.cont_view == VSSpatialPos_T:
|
||
# 顶视图：法向量应平行于Z轴
|
||
return abs(normal.z) > 0.9
|
||
else:
|
||
# 存根模式总是有效
|
||
return True
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 面有效性检查失败: {e}")
|
||
return False
|
||
|
||
def _set_status_text(self, text):
|
||
"""设置状态文本"""
|
||
try:
|
||
if BLENDER_AVAILABLE:
|
||
# 在Blender中设置状态文本
|
||
# 这需要通过UI系统或操作符实现
|
||
pass
|
||
else:
|
||
print(f"💬 状态: {text}")
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 设置状态文本失败: {e}")
|
||
|
||
def _show_message(self, message):
|
||
"""显示消息"""
|
||
try:
|
||
if BLENDER_AVAILABLE:
|
||
# Blender消息框
|
||
def show_message_box(message="", title="Message", icon='INFO'):
|
||
def draw(self, context):
|
||
self.layout.label(text=message)
|
||
bpy.context.window_manager.popup_menu(
|
||
draw, title=title, icon=icon)
|
||
|
||
show_message_box(message, "SUWood", 'INFO')
|
||
else:
|
||
print(f"💬 消息: {message}")
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 显示消息失败: {e}")
|
||
|
||
def _invalidate_view(self):
|
||
"""刷新视图"""
|
||
try:
|
||
if BLENDER_AVAILABLE:
|
||
for area in bpy.context.screen.areas:
|
||
if area.type == 'VIEW_3D':
|
||
area.tag_redraw()
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 视图刷新失败: {e}")
|
||
|
||
def _clear_selection(self):
|
||
"""清除选择"""
|
||
try:
|
||
if BLENDER_AVAILABLE:
|
||
bpy.ops.object.select_all(action='DESELECT')
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 清除选择失败: {e}")
|
||
|
||
def _select_tool(self, tool):
|
||
"""选择工具"""
|
||
try:
|
||
if BLENDER_AVAILABLE:
|
||
if tool is None:
|
||
bpy.ops.wm.tool_set_by_id(name="builtin.select")
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 工具切换失败: {e}")
|
||
|
||
def _get_order_id(self):
|
||
"""获取订单ID"""
|
||
try:
|
||
if BLENDER_AVAILABLE:
|
||
scene = bpy.context.scene
|
||
return scene.get("sw_order_id")
|
||
else:
|
||
return None
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ 获取订单ID失败: {e}")
|
||
return None
|
||
|
||
|
||
# 工具函数
|
||
def create_face_tool(cont_view: int, source: str = None, mold: bool = False) -> SUWUnitFaceTool:
|
||
"""创建选面创体工具"""
|
||
return SUWUnitFaceTool(cont_view, source, mold)
|
||
|
||
|
||
def activate_face_tool(cont_view: int = VSSpatialPos_F):
|
||
"""激活选面创体工具"""
|
||
tool = SUWUnitFaceTool(cont_view)
|
||
tool.activate()
|
||
return tool
|
||
|
||
|
||
print("✅ SUWUnitFaceTool完整翻译完成!")
|
||
print("✅ 功能包括:")
|
||
print(" • 智能面拾取检测")
|
||
print(" • 多视图类型支持")
|
||
print(" • 输入框参数设置")
|
||
print(" • 面有效性验证")
|
||
print(" • 前沿边处理 (顶视图)")
|
||
print(" • 高亮面绘制")
|
||
print(" • 创建后自动清理")
|
||
print(" • Blender/存根双模式")
|
||
print(" • 射线检测面拾取")
|
||
print(" • 变换矩阵处理")
|
||
print(" • 面边段构建")
|
||
print(" • 参数验证和构建")
|