x

src/moldinsight/core/aluminum_foam_mold.py

@@ -29,6 +29,7 @@ from shared.models.schemas import create_mold_cavity_data, create_mold_key_info
 from shared.utils.logger import get_logger
 from moldinsight.core.base_mold_generator import BaseMoldGenerator
 from moldinsight.core.side_action_designer import SideActionDesigner
+from moldinsight.services.material_service import MaterialService
 
 logger = get_logger(__name__)
 
@@ -54,48 +55,6 @@ class AluminumFoamMoldGenerator(BaseMoldGenerator):
 
         self.foam_material = foam_material
 
-        self.foam_materials = {
-            "AlSi10Mg": {
-                "density": 0.45,
-                "expansion_ratio": 2.5,
-                "shrinkage_rate": 0.015,
-                "molding_temp": 380,
-                "description": "常用铝硅泡沫"
-            },
-            "AlSi12": {
-                "density": 0.50,
-                "expansion_ratio": 2.2,
-                "shrinkage_rate": 0.012,
-                "molding_temp": 360,
-                "description": "高强度铝泡沫"
-            },
-            "Pure Al Foam": {
-                "density": 0.35,
-                "expansion_ratio": 3.0,
-                "shrinkage_rate": 0.020,
-                "molding_temp": 400,
-                "description": "纯铝泡沫"
-            },
-            "AlSi7Mg": {
-                "density": 0.40,
-                "expansion_ratio": 2.8,
-                "shrinkage_rate": 0.018,
-                "molding_temp": 390,
-                "description": "轻质铝镁泡沫"
-            }
-        }
-
-        self.plastic_materials = {
-            "ABS": {"density": 1.05, "shrinkage": 0.005},
-            "PP": {"density": 0.90, "shrinkage": 0.016},
-            "PC": {"density": 1.20, "shrinkage": 0.005},
-            "PE": {"density": 0.95, "shrinkage": 0.025},
-            "PS": {"density": 1.05, "shrinkage": 0.004},
-            "PA": {"density": 1.14, "shrinkage": 0.015},
-            "POM": {"density": 1.42, "shrinkage": 0.020},
-            "PMMA": {"density": 1.18, "shrinkage": 0.004}
-        }
-
         self.parting_line_tolerance = 0.1
         self.max_draft_angle = 5.0
         self.min_draft_angle = 1.0
@@ -107,21 +66,21 @@ class AluminumFoamMoldGenerator(BaseMoldGenerator):
 
     def set_foam_material(self, material: str):
         """设置铝泡沫材料"""
-        if material in self.foam_materials:
-            props = self.foam_materials[material]
+        if MaterialService.is_foam_material(material):
+            props = MaterialService.get_material(material)
             self.foam_material = material
             self.material_density = props["density"]
-            self.shrinkage_rate = props["shrinkage_rate"]
+            self.shrinkage_rate = props["shrinkage"]
             logger.info(f"铝泡沫材料设置为 {material}, 密度: {props['density']} g/cm³")
         else:
             logger.warning(f"未知材料 {material}, 使用当前设置")
 
     def set_material(self, material: str):
         """设置材料（自动识别类型）"""
-        if material in self.foam_materials:
+        if MaterialService.is_foam_material(material):
             self.set_foam_material(material)
-        elif material in self.plastic_materials:
-            props = self.plastic_materials[material]
+        elif MaterialService.has_material(material):
+            props = MaterialService.get_material(material)
             self.material_density = props["density"]
             self.shrinkage_rate = props["shrinkage"]
             logger.info(f"塑料材料设置为 {material}, 密度: {props['density']} g/cm³")
@@ -204,7 +163,7 @@ class AluminumFoamMoldGenerator(BaseMoldGenerator):
 
         parting_geometry = self._extract_parting_surface_geometry(parting_surface)
 
-        material_info = self.foam_materials.get(self.foam_material, {})
+        material_info = MaterialService.get_material(self.foam_material)
 
         detailed_json = {
             "metadata": {
@@ -251,7 +210,7 @@ class AluminumFoamMoldGenerator(BaseMoldGenerator):
     def generate_cavity_key_info(self, cavity_data: Dict) -> Dict[str, Any]:
         """生成模具型腔的关键信息"""
         analysis = cavity_data["analysis"]
-        material_info = self.foam_materials.get(self.foam_material, {})
+        material_info = MaterialService.get_material(self.foam_material)
 
         key_info = {
             "mold_parameters": {