565 lines
19 KiB
Python
565 lines
19 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
SUW Core - Dimension Manager Module
|
|||
|
|
拆分自: suw_impl.py (Line 5591-5750, 6249-6362)
|
|||
|
|
用途: Blender尺寸标注管理、文本标签、轮廓创建
|
|||
|
|
版本: 1.0.0
|
|||
|
|
作者: SUWood Team
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from .geometry_utils import Point3d, Vector3d
|
|||
|
|
from .memory_manager import memory_manager
|
|||
|
|
from .data_manager import data_manager, get_data_manager
|
|||
|
|
import math
|
|||
|
|
import logging
|
|||
|
|
from typing import Dict, Any, Optional, List
|
|||
|
|
|
|||
|
|
# 设置日志
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
# 检查Blender可用性
|
|||
|
|
try:
|
|||
|
|
import bpy
|
|||
|
|
BLENDER_AVAILABLE = True
|
|||
|
|
except ImportError:
|
|||
|
|
BLENDER_AVAILABLE = False
|
|||
|
|
|
|||
|
|
# 导入依赖模块
|
|||
|
|
|
|||
|
|
# ==================== 尺寸标注管理器类 ====================
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DimensionManager:
|
|||
|
|
"""尺寸标注管理器 - 负责所有尺寸标注相关操作"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
"""
|
|||
|
|
初始化尺寸标注管理器 - 完全独立，不依赖suw_impl
|
|||
|
|
"""
|
|||
|
|
# 使用全局数据管理器
|
|||
|
|
self.data_manager = get_data_manager()
|
|||
|
|
|
|||
|
|
self.dimensions = {}
|
|||
|
|
self.labels = None
|
|||
|
|
self.door_labels = None
|
|||
|
|
self.door_layer = None
|
|||
|
|
self.unit_trans = {}
|
|||
|
|
|
|||
|
|
logger.info("DimensionManager 初始化完成")
|
|||
|
|
|
|||
|
|
# ==================== 核心命令方法 ====================
|
|||
|
|
|
|||
|
|
def c07(self, data: Dict[str, Any]):
|
|||
|
|
"""add_dim - 添加尺寸标注 - 线程安全版本"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
uid = data.get("uid")
|
|||
|
|
|
|||
|
|
def create_dimensions():
|
|||
|
|
try:
|
|||
|
|
dims = data.get("dims", [])
|
|||
|
|
dimensions = []
|
|||
|
|
|
|||
|
|
for dim_data in dims:
|
|||
|
|
p1 = Point3d.parse(dim_data.get("p1", "(0,0,0)"))
|
|||
|
|
p2 = Point3d.parse(dim_data.get("p2", "(0,0,0)"))
|
|||
|
|
direction = Vector3d.parse(
|
|||
|
|
dim_data.get("dir", "(0,0,1)"))
|
|||
|
|
text = dim_data.get("text", "")
|
|||
|
|
|
|||
|
|
dimension = self.create_dimension(
|
|||
|
|
p1, p2, direction, text)
|
|||
|
|
if dimension:
|
|||
|
|
dimensions.append(dimension)
|
|||
|
|
|
|||
|
|
# 存储尺寸标注
|
|||
|
|
if uid not in self.dimensions:
|
|||
|
|
self.dimensions[uid] = []
|
|||
|
|
self.dimensions[uid].extend(dimensions)
|
|||
|
|
|
|||
|
|
for dimension in dimensions:
|
|||
|
|
memory_manager.register_object(dimension)
|
|||
|
|
|
|||
|
|
return len(dimensions)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建尺寸标注失败: {e}")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
# 在主线程中执行尺寸标注创建
|
|||
|
|
count = create_dimensions()
|
|||
|
|
|
|||
|
|
if count > 0:
|
|||
|
|
logger.info(f"✅ 成功创建尺寸标注: uid={uid}, count={count}")
|
|||
|
|
else:
|
|||
|
|
logger.error(f"❌ 尺寸标注创建失败: uid={uid}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ 添加尺寸标注失败: {e}")
|
|||
|
|
|
|||
|
|
def c0c(self, data: Dict[str, Any]):
|
|||
|
|
"""delete_dimensions - 删除尺寸标注"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
def delete_dimensions():
|
|||
|
|
try:
|
|||
|
|
uid = data.get("uid")
|
|||
|
|
|
|||
|
|
if uid in self.dimensions:
|
|||
|
|
dimensions_to_delete = self.dimensions[uid]
|
|||
|
|
deleted_count = 0
|
|||
|
|
|
|||
|
|
for dimension in dimensions_to_delete:
|
|||
|
|
if self.delete_object_safe(dimension):
|
|||
|
|
deleted_count += 1
|
|||
|
|
|
|||
|
|
# 清理维度记录
|
|||
|
|
del self.dimensions[uid]
|
|||
|
|
|
|||
|
|
return deleted_count
|
|||
|
|
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"删除尺寸标注失败: {e}")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
count = delete_dimensions()
|
|||
|
|
logger.info(f"删除尺寸标注完成: count={count}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"删除尺寸标注失败: {e}")
|
|||
|
|
|
|||
|
|
def c12(self, data: Dict[str, Any]):
|
|||
|
|
"""add_contour - 添加轮廓 - 线程安全版本"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
def create_contour():
|
|||
|
|
try:
|
|||
|
|
# 设置添加轮廓标志
|
|||
|
|
self.data_manager.added_contour = True
|
|||
|
|
|
|||
|
|
surf = data.get("surf", {})
|
|||
|
|
|
|||
|
|
contour = self.create_contour_from_surf(surf)
|
|||
|
|
if contour:
|
|||
|
|
memory_manager.register_object(contour)
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建轮廓失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 在主线程中执行轮廓创建
|
|||
|
|
success = create_contour()
|
|||
|
|
|
|||
|
|
if success:
|
|||
|
|
logger.info("✅ 轮廓创建成功")
|
|||
|
|
else:
|
|||
|
|
logger.error("❌ 轮廓创建失败")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"❌ 添加轮廓失败: {e}")
|
|||
|
|
|
|||
|
|
# ==================== 尺寸标注创建 ====================
|
|||
|
|
|
|||
|
|
def create_dimension(self, p1, p2, direction, text):
|
|||
|
|
"""创建尺寸标注"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# 创建尺寸标注几何体
|
|||
|
|
mesh = bpy.data.meshes.new("Dimension")
|
|||
|
|
|
|||
|
|
# 计算标注线的端点
|
|||
|
|
start = (p1.x * 0.001, p1.y * 0.001, p1.z * 0.001)
|
|||
|
|
end = (p2.x * 0.001, p2.y * 0.001, p2.z * 0.001)
|
|||
|
|
|
|||
|
|
# 计算标注偏移
|
|||
|
|
offset_distance = 0.05 # 5cm偏移
|
|||
|
|
offset = (
|
|||
|
|
direction.x * offset_distance,
|
|||
|
|
direction.y * offset_distance,
|
|||
|
|
direction.z * offset_distance
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 创建标注线顶点
|
|||
|
|
vertices = [
|
|||
|
|
start,
|
|||
|
|
end,
|
|||
|
|
(start[0] + offset[0], start[1] +
|
|||
|
|
offset[1], start[2] + offset[2]),
|
|||
|
|
(end[0] + offset[0], end[1] + offset[1], end[2] + offset[2])
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# 创建边
|
|||
|
|
edges = [(0, 2), (1, 3), (2, 3)]
|
|||
|
|
|
|||
|
|
mesh.from_pydata(vertices, edges, [])
|
|||
|
|
mesh.update()
|
|||
|
|
|
|||
|
|
# 创建对象
|
|||
|
|
dim_obj = bpy.data.objects.new("Dimension", mesh)
|
|||
|
|
bpy.context.scene.collection.objects.link(dim_obj)
|
|||
|
|
|
|||
|
|
# 创建文本标签
|
|||
|
|
if text:
|
|||
|
|
label_pos = (
|
|||
|
|
(start[0] + end[0]) / 2 + offset[0],
|
|||
|
|
(start[1] + end[1]) / 2 + offset[1],
|
|||
|
|
(start[2] + end[2]) / 2 + offset[2]
|
|||
|
|
)
|
|||
|
|
text_obj = self.create_text_label(text, label_pos, direction)
|
|||
|
|
if text_obj:
|
|||
|
|
text_obj.parent = dim_obj
|
|||
|
|
|
|||
|
|
return dim_obj
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建尺寸标注失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def create_text_label(self, text, location, direction):
|
|||
|
|
"""创建文本标签"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# 创建文本对象
|
|||
|
|
font_curve = bpy.data.curves.new(type="FONT", name="TextLabel")
|
|||
|
|
font_curve.body = text
|
|||
|
|
font_obj = bpy.data.objects.new("TextLabel", font_curve)
|
|||
|
|
|
|||
|
|
# 设置位置和方向
|
|||
|
|
font_obj.location = location
|
|||
|
|
if isinstance(direction, (list, tuple)) and len(direction) >= 3:
|
|||
|
|
# 简化的方向设置
|
|||
|
|
font_obj.location = (
|
|||
|
|
location[0] + direction.x * 0.1 if hasattr(
|
|||
|
|
direction, 'x') else location[0] + direction[0] * 0.1,
|
|||
|
|
location[1] + direction.y * 0.1 if hasattr(
|
|||
|
|
direction, 'y') else location[1] + direction[1] * 0.1,
|
|||
|
|
location[2] + direction.z * 0.1 if hasattr(
|
|||
|
|
direction, 'z') else location[2] + direction[2] * 0.1
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
bpy.context.scene.collection.objects.link(font_obj)
|
|||
|
|
memory_manager.register_object(font_obj)
|
|||
|
|
|
|||
|
|
return font_obj
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建文本标签失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# ==================== 轮廓创建 ====================
|
|||
|
|
|
|||
|
|
def create_contour_from_surf(self, surf):
|
|||
|
|
"""从表面创建轮廓"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
xaxis = Vector3d.parse(surf.get("vx", "(1,0,0)"))
|
|||
|
|
zaxis = Vector3d.parse(surf.get("vz", "(0,0,1)"))
|
|||
|
|
segs = surf.get("segs", [])
|
|||
|
|
|
|||
|
|
edges = []
|
|||
|
|
for seg in segs:
|
|||
|
|
if "c" in seg:
|
|||
|
|
# 弧形段
|
|||
|
|
c = Point3d.parse(seg["c"])
|
|||
|
|
r = seg.get("r", 1.0) * 0.001
|
|||
|
|
a1 = seg.get("a1", 0.0)
|
|||
|
|
a2 = seg.get("a2", math.pi * 2)
|
|||
|
|
n = seg.get("n", 12)
|
|||
|
|
|
|||
|
|
# 创建弧形边
|
|||
|
|
arc_edges = self.create_arc_edges(
|
|||
|
|
c, xaxis, zaxis, r, a1, a2, n)
|
|||
|
|
edges.extend(arc_edges)
|
|||
|
|
else:
|
|||
|
|
# 直线段
|
|||
|
|
s = Point3d.parse(seg.get("s", "(0,0,0)"))
|
|||
|
|
e = Point3d.parse(seg.get("e", "(0,0,0)"))
|
|||
|
|
edge = self.create_line_edge_simple(
|
|||
|
|
(s.x * 0.001, s.y * 0.001, s.z * 0.001),
|
|||
|
|
(e.x * 0.001, e.y * 0.001, e.z * 0.001))
|
|||
|
|
if edge:
|
|||
|
|
edges.append(edge)
|
|||
|
|
|
|||
|
|
# 尝试创建面
|
|||
|
|
try:
|
|||
|
|
if edges:
|
|||
|
|
return self.create_face_from_edges(edges)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"创建轮廓面失败: {e}")
|
|||
|
|
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建轮廓失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def create_arc_edges(self, center, xaxis, zaxis, radius, start_angle, end_angle, segments):
|
|||
|
|
"""创建弧形边"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE:
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
edges = []
|
|||
|
|
angle_step = (end_angle - start_angle) / segments
|
|||
|
|
|
|||
|
|
for i in range(segments):
|
|||
|
|
angle1 = start_angle + i * angle_step
|
|||
|
|
angle2 = start_angle + (i + 1) * angle_step
|
|||
|
|
|
|||
|
|
# 计算点
|
|||
|
|
x1 = center.x * 0.001 + radius * math.cos(angle1)
|
|||
|
|
y1 = center.y * 0.001 + radius * math.sin(angle1)
|
|||
|
|
z1 = center.z * 0.001
|
|||
|
|
|
|||
|
|
x2 = center.x * 0.001 + radius * math.cos(angle2)
|
|||
|
|
y2 = center.y * 0.001 + radius * math.sin(angle2)
|
|||
|
|
z2 = center.z * 0.001
|
|||
|
|
|
|||
|
|
edge = self.create_line_edge_simple(
|
|||
|
|
(x1, y1, z1), (x2, y2, z2))
|
|||
|
|
if edge:
|
|||
|
|
edges.append(edge)
|
|||
|
|
|
|||
|
|
return edges
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建弧形边失败: {e}")
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
def create_line_edge_simple(self, start, end):
|
|||
|
|
"""创建简单线边"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# 创建线段网格
|
|||
|
|
mesh = bpy.data.meshes.new("Line_Edge")
|
|||
|
|
vertices = [start, end]
|
|||
|
|
edges = [(0, 1)]
|
|||
|
|
|
|||
|
|
mesh.from_pydata(vertices, edges, [])
|
|||
|
|
mesh.update()
|
|||
|
|
|
|||
|
|
# 创建对象
|
|||
|
|
obj = bpy.data.objects.new("Line_Edge_Obj", mesh)
|
|||
|
|
bpy.context.scene.collection.objects.link(obj)
|
|||
|
|
|
|||
|
|
return obj
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建线边失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def create_face_from_edges(self, edges):
|
|||
|
|
"""从边创建面"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE or not edges:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# 收集所有顶点
|
|||
|
|
all_vertices = []
|
|||
|
|
for edge in edges:
|
|||
|
|
if hasattr(edge, 'data') and hasattr(edge.data, 'vertices'):
|
|||
|
|
for vertex in edge.data.vertices:
|
|||
|
|
all_vertices.append(vertex.co)
|
|||
|
|
|
|||
|
|
if len(all_vertices) < 3:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# 创建面网格
|
|||
|
|
mesh = bpy.data.meshes.new("Contour_Face")
|
|||
|
|
faces = [list(range(len(all_vertices)))]
|
|||
|
|
|
|||
|
|
mesh.from_pydata(all_vertices, [], faces)
|
|||
|
|
mesh.update()
|
|||
|
|
|
|||
|
|
# 创建对象
|
|||
|
|
obj = bpy.data.objects.new("Contour_Face_Obj", mesh)
|
|||
|
|
bpy.context.scene.collection.objects.link(obj)
|
|||
|
|
|
|||
|
|
return obj
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"从边创建面失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# ==================== 标签管理 ====================
|
|||
|
|
|
|||
|
|
def add_part_labels(self, uid, parts):
|
|||
|
|
"""添加零件标签"""
|
|||
|
|
try:
|
|||
|
|
for root, part in parts.items():
|
|||
|
|
center = self.get_object_center(part)
|
|||
|
|
pos = part.get("sw_pos", 1)
|
|||
|
|
|
|||
|
|
# 确定标签方向
|
|||
|
|
if pos == 1:
|
|||
|
|
vector = (0, -1, 0) # F
|
|||
|
|
elif pos == 2:
|
|||
|
|
vector = (0, 1, 0) # K
|
|||
|
|
elif pos == 3:
|
|||
|
|
vector = (-1, 0, 0) # L
|
|||
|
|
elif pos == 4:
|
|||
|
|
vector = (1, 0, 0) # R
|
|||
|
|
elif pos == 5:
|
|||
|
|
vector = (0, 0, -1) # B
|
|||
|
|
else:
|
|||
|
|
vector = (0, 0, 1) # T
|
|||
|
|
|
|||
|
|
# 应用单位变换
|
|||
|
|
if uid in self.unit_trans:
|
|||
|
|
vector = self.transform_vector(
|
|||
|
|
vector, self.unit_trans[uid])
|
|||
|
|
|
|||
|
|
# 创建文本标签
|
|||
|
|
ord_seq = part.get("sw_seq", 0)
|
|||
|
|
text_obj = self.create_text_label(
|
|||
|
|
str(ord_seq), center, vector)
|
|||
|
|
|
|||
|
|
if text_obj:
|
|||
|
|
# 根据图层选择父对象
|
|||
|
|
if self.is_in_door_layer(part):
|
|||
|
|
text_obj.parent = self.door_labels
|
|||
|
|
else:
|
|||
|
|
text_obj.parent = self.labels
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"添加零件标签失败: {e}")
|
|||
|
|
|
|||
|
|
def clear_labels(self, label_obj):
|
|||
|
|
"""清理标签"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE or not label_obj:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 删除所有子对象
|
|||
|
|
children = label_obj.children[:]
|
|||
|
|
for child in children:
|
|||
|
|
self.delete_object_safe(child)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"清理标签失败: {e}")
|
|||
|
|
|
|||
|
|
# ==================== 工具方法 ====================
|
|||
|
|
|
|||
|
|
def get_object_center(self, obj):
|
|||
|
|
"""获取对象中心"""
|
|||
|
|
try:
|
|||
|
|
if BLENDER_AVAILABLE and obj and hasattr(obj, 'location'):
|
|||
|
|
return obj.location
|
|||
|
|
return (0, 0, 0)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"获取对象中心失败: {e}")
|
|||
|
|
return (0, 0, 0)
|
|||
|
|
|
|||
|
|
def is_in_door_layer(self, part):
|
|||
|
|
"""检查是否在门图层"""
|
|||
|
|
try:
|
|||
|
|
if not part or not self.door_layer:
|
|||
|
|
return False
|
|||
|
|
return part in self.door_layer.objects
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"检查门图层失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def delete_object_safe(self, obj) -> bool:
|
|||
|
|
"""安全删除对象"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE or not obj:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 检查对象是否仍然有效
|
|||
|
|
if obj.name not in bpy.data.objects:
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
# 从场景中移除
|
|||
|
|
if obj.name in bpy.context.scene.collection.objects:
|
|||
|
|
bpy.context.scene.collection.objects.unlink(obj)
|
|||
|
|
|
|||
|
|
# 删除对象数据
|
|||
|
|
if hasattr(obj, 'data') and obj.data:
|
|||
|
|
if obj.data.name in bpy.data.meshes:
|
|||
|
|
bpy.data.meshes.remove(obj.data)
|
|||
|
|
elif obj.data.name in bpy.data.curves:
|
|||
|
|
bpy.data.curves.remove(obj.data)
|
|||
|
|
|
|||
|
|
# 删除对象
|
|||
|
|
if obj.name in bpy.data.objects:
|
|||
|
|
bpy.data.objects.remove(obj)
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"删除对象失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def transform_vector(self, vector, transform):
|
|||
|
|
"""变换向量"""
|
|||
|
|
try:
|
|||
|
|
if not BLENDER_AVAILABLE or not transform:
|
|||
|
|
return vector
|
|||
|
|
|
|||
|
|
if isinstance(vector, (list, tuple)) and len(vector) >= 3:
|
|||
|
|
import mathutils
|
|||
|
|
vec = mathutils.Vector(vector)
|
|||
|
|
transformed = transform @ vec
|
|||
|
|
return (transformed.x, transformed.y, transformed.z)
|
|||
|
|
|
|||
|
|
return vector
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"变换向量失败: {e}")
|
|||
|
|
return vector
|
|||
|
|
|
|||
|
|
# ==================== 管理器统计 ====================
|
|||
|
|
|
|||
|
|
def get_dimension_stats(self) -> Dict[str, Any]:
|
|||
|
|
"""获取尺寸标注管理器统计信息"""
|
|||
|
|
try:
|
|||
|
|
total_dimensions = sum(len(dims)
|
|||
|
|
for dims in self.dimensions.values())
|
|||
|
|
|
|||
|
|
stats = {
|
|||
|
|
"manager_type": "DimensionManager",
|
|||
|
|
"total_dimensions": total_dimensions,
|
|||
|
|
"units_with_dimensions": len(self.dimensions),
|
|||
|
|
"has_labels": self.labels is not None,
|
|||
|
|
"has_door_labels": self.door_labels is not None,
|
|||
|
|
"blender_available": BLENDER_AVAILABLE
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return stats
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"获取尺寸标注统计失败: {e}")
|
|||
|
|
return {"error": str(e)}
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ==================== 模块实例 ====================
|
|||
|
|
|
|||
|
|
# 全局实例，将由SUWImpl初始化时设置
|
|||
|
|
dimension_manager = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def init_dimension_manager():
|
|||
|
|
"""初始化尺寸标注管理器 - 不再需要suw_impl参数"""
|
|||
|
|
global dimension_manager
|
|||
|
|
dimension_manager = DimensionManager()
|
|||
|
|
return dimension_manager
|