Swift中使用SceneKit计算3D模型表面积的问题求助

问题分析 #

你当前的代码存在几个关键错误，导致表面积计算结果异常：

• 顶点选取逻辑错误：直接按i, i+2, i+3的方式选取三角形顶点完全不符合3D模型的三角形索引规则，会组合出大量无效三角形，大幅高估面积。
• 向量计算错误：计算ac向量的z分量时写反了，应该是vc.z - va.z而非va.z - vc.z。
• 未使用索引缓冲：SCNGeometry的顶点数组通常包含重复顶点，三角形的实际组成需要通过**索引元素（SCNGeometryElement）**获取，而非直接按顺序遍历顶点。

修正后的代码 #

1. 完善顶点与索引的获取扩展

extension SCNGeometry {
    // 获取顶点数组
    func vertices() -> [SCNVector3]? {
        guard let source = sources(for: .vertex).first else { return nil }
        
        let stride = source.dataStride / source.bytesPerComponent
        let offset = source.dataOffset / source.bytesPerComponent
        let vectorCount = source.vectorCount
        
        return source.data.withUnsafeBytes { buffer in
            let floatBuffer = buffer.baseAddress!.assumingMemoryBound(to: Float.self)
            var vertices = [SCNVector3]()
            vertices.reserveCapacity(vectorCount)
            
            for i in 0..<vectorCount {
                let start = i * stride + offset
                let x = floatBuffer[start]
                let y = floatBuffer[start + 1]
                let z = floatBuffer[start + 2]
                vertices.append(SCNVector3(x, y, z))
            }
            return vertices
        }
    }
    
    // 获取三角形索引数组
    func triangleIndices() -> [UInt32]? {
        guard let element = geometryElements.first else { return nil }
        
        let indexCount = element.primitiveCount * 3
        var indices = [UInt32]()
        indices.reserveCapacity(indexCount)
        
        element.data.withUnsafeBytes { buffer in
            switch element.indexType {
            case .uint16:
                let uint16Buffer = buffer.baseAddress!.assumingMemoryBound(to: UInt16.self)
                for i in 0..<indexCount {
                    indices.append(UInt32(uint16Buffer[i]))
                }
            case .uint32:
                let uint32Buffer = buffer.baseAddress!.assumingMemoryBound(to: UInt32.self)
                for i in 0..<indexCount {
                    indices.append(uint32Buffer[i])
                }
            @unknown default:
                break
            }
        }
        return indices
    }
}

2. 正确计算表面积的方法

private func calculateSurfaceAreaOfMesh() -> Float {
    guard let geometry = modelNode.childNodes[0].geometry,
          let vertices = geometry.vertices(),
          let indices = geometry.triangleIndices(),
          indices.count >= 3 else {
        return 0.0
    }
    
    var totalArea = 0.0
    
    // 每3个索引对应一个三角形
    for i in stride(from: 0, to: indices.count, by: 3) {
        let i0 = indices[i]
        let i1 = indices[i+1]
        let i2 = indices[i+2]
        
        guard i0 < UInt32(vertices.count),
              i1 < UInt32(vertices.count),
              i2 < UInt32(vertices.count) else {
            continue
        }
        
        let v0 = vertices[Int(i0)]
        let v1 = vertices[Int(i1)]
        let v2 = vertices[Int(i2)]
        
        // 计算边向量
        let ab = SCNVector3(x: v1.x - v0.x, y: v1.y - v0.y, z: v1.z - v0.z)
        let ac = SCNVector3(x: v2.x - v0.x, y: v2.y - v0.y, z: v2.z - v0.z)
        
        // 计算叉积
        let cross = ab.cross(vector: ac)
        // 三角形面积是叉积模长的一半
        let triangleArea = sqrt(pow(cross.x, 2) + pow(cross.y, 2) + pow(cross.z, 2)) / 2.0
        totalArea += triangleArea
    }
    
    return Float(totalArea)
}

// 补充SCNVector3的叉积扩展
extension SCNVector3 {
    func cross(vector: SCNVector3) -> SCNVector3 {
        return SCNVector3(
            y * vector.z - z * vector.y,
            z * vector.x - x * vector.z,
            x * vector.y - y * vector.x
        )
    }
}

额外说明 #

• 如果需要计算体积，可以参考类似思路，通过索引遍历每个四面体（基于模型的三角面）进行累加，但需要注意模型必须是闭合的流形网格才能得到准确体积。
• 确保模型没有重复的三角面或非流形结构，否则计算结果仍可能存在误差。

内容的提问来源于stack exchange，提问作者Jaswant Singh