From 7ccc97134dbf4291ea924f3e9e16c60c7a83b60a Mon Sep 17 00:00:00 2001
From: SZCJW <792430652@qq.com>
Date: Sat, 7 Mar 2026 01:09:13 +0800
Subject: [PATCH] x

---
 src/utils/html_generator.py | 118 ++++++++++++++++++++++++++++++++----
 1 file changed, 105 insertions(+), 13 deletions(-)

diff --git a/src/utils/html_generator.py b/src/utils/html_generator.py
index 98737a4..c1f1ed0 100644
--- a/src/utils/html_generator.py
+++ b/src/utils/html_generator.py
@@ -175,10 +175,19 @@ class HTMLGenerator:
         <div id="controls">
             <button onclick="resetView()">重置视图</button>
             <button onclick="toggleWireframe()">切换线框</button>
+            <button onclick="toggleProduct()">显示/隐藏产品</button>
+            <button onclick="toggleMold()">显示/隐藏模具</button>
+            <button onclick="toggleParting()">显示/隐藏分型面</button>
         </div>
     </div>
 
     <script>
+        // 全局变量
+        let productMesh, cavityMesh, coreMesh, partingMesh;
+        let productVisible = true;
+        let moldVisible = true;
+        let partingVisible = true;
+        
         // 初始化Three.js场景
         const scene = new THREE.Scene();
         const camera = new THREE.PerspectiveCamera(75, window.innerWidth / window.innerHeight, 0.1, 1000);
@@ -201,6 +210,7 @@ class HTMLGenerator:
         
         // 创建几何体（模拟模具形状）
         const geometryData = {json.dumps(geometry_data, indent=2)};
+        const cavityData = {json.dumps(cavity_data, indent=2) if cavity_data else 'null'};
         
         // 根据边界框创建模拟几何体
         const bbox = geometryData.bounding_box;
@@ -209,25 +219,85 @@ class HTMLGenerator:
             const height = bbox.dimensions ? bbox.dimensions[1] : 100;
             const depth = bbox.dimensions ? bbox.dimensions[2] : 100;
             
-            // 创建基础几何体
-            const geometry = new THREE.BoxGeometry(width, height, depth);
-            const material = new THREE.MeshPhongMaterial({{
+            // 创建产品几何体（半透明绿色）
+            const productGeometry = new THREE.BoxGeometry(width * 0.9, height * 0.9, depth * 0.9);
+            const productMaterial = new THREE.MeshPhongMaterial({{
                 color: 0x4CAF50,
                 transparent: true,
-                opacity: 0.8,
+                opacity: 0.6,
                 wireframe: false
             }});
             
-            const mesh = new THREE.Mesh(geometry, material);
-            scene.add(mesh);
+            const productMesh = new THREE.Mesh(productGeometry, productMaterial);
+            productMesh.position.set(0, 0, 0);
+            scene.add(productMesh);
             
-            // 添加线框
-            const wireframe = new THREE.WireframeGeometry(geometry);
-            const line = new THREE.LineSegments(wireframe);
-            line.material.depthTest = false;
-            line.material.opacity = 0.25;
-            line.material.transparent = true;
-            scene.add(line);
+            // 创建型腔（上半模 - 蓝色）
+            const cavityGeometry = new THREE.BoxGeometry(width * 1.1, height * 0.3, depth * 1.1);
+            const cavityMaterial = new THREE.MeshPhongMaterial({{
+                color: 0x2196F3,
+                transparent: true,
+                opacity: 0.4,
+                wireframe: false
+            }});
+            
+            const cavityMesh = new THREE.Mesh(cavityGeometry, cavityMaterial);
+            cavityMesh.position.set(0, height * 0.6, 0);
+            scene.add(cavityMesh);
+            
+            // 创建型芯（下半模 - 橙色）
+            const coreGeometry = new THREE.BoxGeometry(width * 1.1, height * 0.3, depth * 1.1);
+            const coreMaterial = new THREE.MeshPhongMaterial({{
+                color: 0xFF9800,
+                transparent: true,
+                opacity: 0.4,
+                wireframe: false
+            }});
+            
+            const coreMesh = new THREE.Mesh(coreGeometry, coreMaterial);
+            coreMesh.position.set(0, -height * 0.6, 0);
+            scene.add(coreMesh);
+            
+            // 创建分型面（红色平面）
+            const partingGeometry = new THREE.PlaneGeometry(width * 1.2, depth * 1.2);
+            const partingMaterial = new THREE.MeshBasicMaterial({{
+                color: 0xF44336,
+                transparent: true,
+                opacity: 0.3,
+                side: THREE.DoubleSide
+            }});
+            
+            const partingMesh = new THREE.Mesh(partingGeometry, partingMaterial);
+            partingMesh.rotation.x = Math.PI / 2;
+            partingMesh.position.set(0, 0, 0);
+            scene.add(partingMesh);
+            
+            // 添加产品线框
+            const productWireframe = new THREE.WireframeGeometry(productGeometry);
+            const productLine = new THREE.LineSegments(productWireframe);
+            productLine.material.depthTest = false;
+            productLine.material.opacity = 0.5;
+            productLine.material.transparent = true;
+            productLine.material.color = new THREE.Color(0x2E7D32);
+            productMesh.add(productLine);
+            
+            // 添加型腔线框
+            const cavityWireframe = new THREE.WireframeGeometry(cavityGeometry);
+            const cavityLine = new THREE.LineSegments(cavityWireframe);
+            cavityLine.material.depthTest = false;
+            cavityLine.material.opacity = 0.6;
+            cavityLine.material.transparent = true;
+            cavityLine.material.color = new THREE.Color(0x1565C0);
+            cavityMesh.add(cavityLine);
+            
+            // 添加型芯线框
+            const coreWireframe = new THREE.WireframeGeometry(coreGeometry);
+            const coreLine = new THREE.LineSegments(coreWireframe);
+            coreLine.material.depthTest = false;
+            coreLine.material.opacity = 0.6;
+            coreLine.material.transparent = true;
+            coreLine.material.color = new THREE.Color(0xE65100);
+            coreMesh.add(coreLine);
         }}
         
         // 设置相机位置
@@ -267,6 +337,28 @@ class HTMLGenerator:
                 }}
             }});
         }}
+        
+        function toggleProduct() {{
+            if (productMesh) {{
+                productVisible = !productVisible;
+                productMesh.visible = productVisible;
+            }}
+        }}
+        
+        function toggleMold() {{
+            if (cavityMesh && coreMesh) {{
+                moldVisible = !moldVisible;
+                cavityMesh.visible = moldVisible;
+                coreMesh.visible = moldVisible;
+            }}
+        }}
+        
+        function toggleParting() {{
+            if (partingMesh) {{
+                partingVisible = !partingVisible;
+                partingMesh.visible = partingVisible;
+            }}
+        }}
     </script>
 </body>
 </html>