x

2026-04-30 16:06:15 +08:00
parent f9acdb767b
commit 838604fddf
3 changed files with 104 additions and 0 deletions
@@ -219,6 +219,52 @@ class MoldCavityGenerator(BaseMoldGenerator):
        logger.info("使用简化方法检测分型面")
        return self._simple_parting_surface(shape, analysis)

+    def _detect_primary_parting(self, shape: Any, analysis: Dict) -> Dict[str, Any]:
+        """检测主分型面（AI优先 → 几何法向量 → 简化回退）"""
+        if self.ai_parting_detector is not None:
+            try:
+                ai_result = self.ai_parting_detector.detect(shape, analysis)
+                if ai_result is not None:
+                    surface, line = self._create_parting_surface_from_ai(ai_result, analysis, shape)
+                    return {
+                        "surface": surface,
+                        "line": line,
+                        "direction": ai_result.get("normal", [0, 0, 1]),
+                        "method": ai_result.get("method", "ai"),
+                        "confidence": ai_result.get("confidence", 0.8),
+                    }
+            except Exception as e:
+                logger.warning(f"AI 分型面检测失败: {e}")
+
+        try:
+            normal_dir = self._analyze_face_normals(shape)
+            parting_plane = self._create_optimal_parting_plane(shape, analysis, normal_dir)
+            dims = analysis.get("bounding_box", {}).get("dimensions", [100, 100, 100])
+            span = max(dims) * 1.5 + 30
+            parting_surface = BRepBuilderAPI_MakeFace(
+                parting_plane, -span, span, -span, span
+            ).Face()
+            parting_surface = self.extend_parting_surface(parting_surface, shape, extension=30.0)
+            parting_line = self._calculate_parting_line(shape, parting_surface)
+            return {
+                "surface": parting_surface,
+                "line": parting_line,
+                "direction": [float(normal_dir.X()), float(normal_dir.Y()), float(normal_dir.Z())],
+                "method": "face_normal_analysis",
+                "confidence": 0.85,
+            }
+        except Exception as e:
+            logger.warning(f"法向量分析失败，使用简化方法：{e}")
+
+        surface, line = self._simple_parting_surface(shape, analysis)
+        return {
+            "surface": surface,
+            "line": line,
+            "direction": [0, 0, 1],
+            "method": "simple",
+            "confidence": 0.6,
+        }
+
    def _build_undercut_regions(self, undercut_analysis: Dict[str, Any]) -> List[Dict[str, Any]]:
        """将侧向机构分析结果转换为兼容旧结构的倒扣区域列表。"""
        undercut_faces = undercut_analysis.get("undercut_faces", [])