x

src/core/aluminum_foam_mold.py

@@ -298,6 +298,36 @@ class AluminumFoamMoldGenerator(BaseMoldGenerator):
         result["normal_statistics"] = self._analyze_parting_direction(shape)
         return result
 
+    def _analyze_parting_direction(self, shape: Any) -> Dict[str, float]:
+        """分析产品法向量分布，按面积加权统计各轴方向强度"""
+        stats = {"X": 0.0, "Y": 0.0, "Z": 0.0}
+        explorer = TopExp_Explorer(shape, TopAbs_FACE)
+
+        while explorer.More():
+            face = TopoDS_Face(explorer.Current())
+            explorer.Next()
+            try:
+                normal = self._get_face_normal(face)
+                if normal is None:
+                    continue
+                props = GProp_GProps()
+                brepgprop.SurfaceProperties(face, props)
+                area = max(float(props.Mass()), 1.0)
+                stats["X"] += abs(float(normal.X())) * area
+                stats["Y"] += abs(float(normal.Y())) * area
+                stats["Z"] += abs(float(normal.Z())) * area
+            except Exception:
+                continue
+
+        total = stats["X"] + stats["Y"] + stats["Z"]
+        if total <= 0:
+            return {"X": 33.3, "Y": 33.3, "Z": 33.4}
+
+        return {
+            axis: round(value / total * 100, 2)
+            for axis, value in stats.items()
+        }
+
     def _split_cavity_core(self, shape: Any, parting_surface: Any) -> Tuple[Any, Any]:
         """分离型腔和型芯（铝泡沫使用更大余量）"""
         return super()._split_cavity_core(shape, parting_surface, margin=25)