分模候选方案

2026-04-23 23:37:39 +08:00
parent 0bdfc7f7d5
commit c1f6b6e827
16 changed files with 1302 additions and 497 deletions
@@ -299,3 +299,78 @@ class CalculationService:
        }

        return detailed_cavity_json
+
+    @classmethod
+    def build_plan_result(
+        cls,
+        geometry_data: Dict[str, Any],
+        material: Dict[str, Any],
+        file_path: str,
+        plan_result: Optional[Dict[str, Any]] = None,
+    ) -> Dict[str, Any]:
+        """构建多方案分模结果，并保留单方案兼容字段。"""
+        if not plan_result or not plan_result.get("candidate_schemes"):
+            legacy = cls.build_detailed_cavity_json(
+                geometry_data=geometry_data,
+                material=material,
+                file_path=file_path,
+                cavity_mesh_data=None,
+            )
+            return {
+                "best_scheme_id": "scheme_1",
+                "candidate_schemes": [
+                    {
+                        "scheme_id": "scheme_1",
+                        "rank": 1,
+                        "title": "推荐方案",
+                        "method": "legacy_fallback",
+                        "score": 60.0,
+                        "score_breakdown": {},
+                        "summary": "当前模型未生成多方案，返回兼容单方案结果",
+                        "parting": {
+                            "axis": legacy.get("metadata", {}).get("parting_direction", "Z"),
+                            "direction": None,
+                            "surface": legacy.get("parting_surface", {}),
+                            "line": [],
+                        },
+                        "cavity_data": legacy,
+                        "key_info": legacy.get("mold_cavities", {}).get("cavity_key_info", {}),
+                        "undercut_regions": legacy.get("undercut_regions", []),
+                        "side_actions": legacy.get("side_actions", {}),
+                    }
+                ],
+                "global_summary": {
+                    "scheme_count": 1,
+                    "recommended_reason": "兼容旧版单方案结果",
+                },
+                "cavity_data": legacy,
+                "key_info": legacy.get("mold_cavities", {}).get("cavity_key_info", {}),
+            }
+
+        candidate_schemes = plan_result.get("candidate_schemes", [])
+        best_scheme = cls.get_best_scheme(plan_result)
+        result = {
+            "best_scheme_id": plan_result.get("best_scheme_id"),
+            "candidate_schemes": candidate_schemes,
+            "global_summary": plan_result.get("global_summary", {}),
+            "cavity_data": best_scheme.get("cavity_data", {}) if best_scheme else {},
+            "key_info": best_scheme.get("key_info", {}) if best_scheme else {},
+        }
+        return result
+
+    @staticmethod
+    def get_best_scheme(plan_result: Optional[Dict[str, Any]]) -> Optional[Dict[str, Any]]:
+        if not plan_result:
+            return None
+
+        schemes = plan_result.get("candidate_schemes", [])
+        if not schemes:
+            return None
+
+        best_scheme_id = plan_result.get("best_scheme_id")
+        if best_scheme_id:
+            for scheme in schemes:
+                if scheme.get("scheme_id") == best_scheme_id:
+                    return scheme
+
+        return schemes[0]