x

src/services/calculation_service.py

@@ -316,7 +316,7 @@ class CalculationService:
                 file_path=file_path,
                 cavity_mesh_data=None,
             )
-            return {
+            result = {
                 "best_scheme_id": "scheme_1",
                 "candidate_schemes": [
                     {
@@ -349,6 +349,8 @@ class CalculationService:
                 "cavity_data": legacy,
                 "key_info": legacy.get("mold_cavities", {}).get("cavity_key_info", {}),
             }
+            cls.attach_injection_system_summaries(result, material["name"])
+            return result
 
         candidate_schemes = plan_result.get("candidate_schemes", [])
         best_scheme = cls.get_best_scheme(plan_result)
@@ -359,8 +361,54 @@ class CalculationService:
             "cavity_data": best_scheme.get("cavity_data", {}) if best_scheme else {},
             "key_info": best_scheme.get("key_info", {}) if best_scheme else {},
         }
+        cls.attach_injection_system_summaries(result, material["name"])
         return result
 
+    @classmethod
+    def attach_injection_system_summaries(
+        cls,
+        plan_result: Dict[str, Any],
+        material_name: str,
+    ) -> Dict[str, Any]:
+        """为每个候选方案补充注塑模冷却/浇注摘要。"""
+        from core.mold_system_designer import MoldSystemDesigner
+
+        designer = MoldSystemDesigner()
+        for scheme in plan_result.get("candidate_schemes", []):
+            cavity_data = scheme.get("cavity_data") or {}
+            product_bbox = cavity_data.get("product_analysis", {}).get("bounding_box", {})
+            mold_size = cavity_data.get("manufacturing_info", {}).get("estimated_mold_size", {})
+            cavity_count = cavity_data.get("mold_cavities", {}).get("cavity_count", 1)
+
+            if not product_bbox or not mold_size:
+                continue
+
+            system_result = designer.design_complete_system(
+                mold_size=mold_size,
+                product_bbox=product_bbox,
+                material=material_name,
+                cavity_count=cavity_count,
+            )
+
+            cavity_data["injection_system"] = system_result
+            cavity_data.setdefault("manufacturing_info", {})
+            cavity_data["manufacturing_info"]["cooling_summary"] = {
+                "cooling_time": system_result.get("cooling", {}).get("cooling_time"),
+                "channel_count": system_result.get("cooling", {}).get("thermal_check", {}).get("channel_count"),
+                "flow_rate_lpm": system_result.get("cooling", {}).get("flow_rate", {}).get("flow_rate_lpm"),
+            }
+            cavity_data["manufacturing_info"]["gating_summary"] = {
+                "gate_type": system_result.get("gating", {}).get("gate_type"),
+                "runner_type": system_result.get("gating", {}).get("runner", {}).get("type"),
+                "estimated_cycle_time": system_result.get("overall_assessment", {}).get("estimated_cycle_time"),
+            }
+
+        best_scheme = cls.get_best_scheme(plan_result)
+        if best_scheme:
+            plan_result["injection_system"] = best_scheme.get("cavity_data", {}).get("injection_system")
+
+        return plan_result
+
     @staticmethod
     def get_best_scheme(plan_result: Optional[Dict[str, Any]]) -> Optional[Dict[str, Any]]:
         if not plan_result: