x

src/core/multi_scheme_planner.py

@@ -35,9 +35,11 @@ class MultiSchemeMoldPlanner:
         material: Dict[str, Any],
         is_foam_material: bool = False,
         max_schemes: int = 3,
+        process_params: Optional[Dict[str, Any]] = None,
     ) -> Dict[str, Any]:
         generator = self.aluminum_foam_generator if is_foam_material else self.mold_generator
         generator.set_material(material["name"])
+        self._apply_process_params(generator, material, process_params)
 
         analysis = generator._analyze_product_geometry(shape)
         analysis["axis_normal_stats"] = self._collect_axis_normal_stats(generator, shape)
@@ -67,16 +69,20 @@ class MultiSchemeMoldPlanner:
             raise ValueError("未能生成任何可用分模方案")
 
         scored_schemes = self.scheme_scorer.score_schemes(schemes)[:max_schemes]
+        export_shapes = {}
         for idx, scheme in enumerate(scored_schemes, start=1):
             scheme["raw_scheme_id"] = scheme.get("scheme_id")
             scheme["scheme_id"] = f"scheme_{idx}"
             if scheme.get("cavity_data", {}).get("metadata") is not None:
                 scheme["cavity_data"]["metadata"]["scheme_id"] = scheme["scheme_id"]
+                scheme["cavity_data"]["metadata"]["process_parameters"] = dict(process_params or {})
+            export_shapes[scheme["scheme_id"]] = scheme.pop("_export_shapes", {})
         best_scheme = scored_schemes[0]
 
         return {
             "best_scheme_id": best_scheme["scheme_id"],
             "candidate_schemes": scored_schemes,
+            "_export_shapes": export_shapes,
             "global_summary": {
                 "scheme_count": len(scored_schemes),
                 "recommended_reason": best_scheme.get("summary", ""),
@@ -178,8 +184,26 @@ class MultiSchemeMoldPlanner:
             "side_actions": side_action_result,
             "cavity_data": cavity_data,
             "key_info": key_info,
+            "_export_shapes": {
+                "cavity": cavity,
+                "core": core,
+                "parting_surface": parting_surface,
+            },
         }
 
+    @staticmethod
+    def _apply_process_params(generator: Any, material: Dict[str, Any], process_params: Optional[Dict[str, Any]]):
+        params = process_params or {}
+        draft_angle = float(params.get("draft_angle", getattr(generator, "draft_angle", 2.0)))
+        shrinkage_rate = float(params.get("shrinkage_rate", material.get("shrinkage", 0.005) * 100.0)) / 100.0
+        parting_precision = float(params.get("parting_precision", getattr(generator, "parting_line_tolerance", 0.1)))
+        cavity_match = float(params.get("cavity_match", getattr(generator, "cavity_match_rate", 95.0)))
+
+        generator.draft_angle = draft_angle
+        generator.shrinkage_rate = shrinkage_rate
+        generator.parting_line_tolerance = parting_precision
+        generator.cavity_match_rate = cavity_match
+
     def _build_parting_surface(
         self,
         generator: Any,