如何通过iOS Swift从USDZ文件RoomPlan 3D模型获取带尺寸的2D平面图

核心步骤 #

1. 加载USDZ文件并解析其中的RoomAnchor数据
2. 提取房间的墙壁、门窗等结构元素及其尺寸信息
3. 将3D坐标转换为2D俯视图坐标
4. 绘制带尺寸标注的2D平面图

1. 加载并解析RoomPlan USDZ #

RoomPlan导出的USDZ包含RoomAnchor对象，可通过RoomAnchor.Loader直接加载解析：

import RoomPlan

func loadRoomAnchor(from usdzURL: URL) async throws -> RoomAnchor {
    let loader = try RoomAnchor.Loader()
    return try await loader.load(from: usdzURL)
}

2. 提取房间结构元素 #

从RoomAnchor中提取墙壁、门窗等元素，这些对象自带长度、高度等尺寸属性：

func extractRoomComponents(from anchor: RoomAnchor) -> (walls: [RoomWall], doors: [RoomDoor], windows: [RoomWindow]) {
    return (anchor.walls, anchor.doors, anchor.windows)
}

3. 3D坐标转2D俯视图 #

ARKit坐标系中，X轴向右、Z轴向前、Y轴向上，俯视图可忽略Y轴，将X/Z坐标映射为2D平面的X/Y：

import CoreGraphics

// 缩放比例：1米 = 100pt，可根据需求调整
private let scale: CGFloat = 100

func convertTo2D(_ point: SIMD3<Float>) -> CGPoint {
    return CGPoint(x: CGFloat(point.x) * scale, y: CGFloat(point.z) * scale)
}

// 获取墙壁的2D矩形和尺寸（单位：pt）
func wall2DData(_ wall: RoomWall) -> (rect: CGRect, length: CGFloat, height: CGFloat) {
    let start = convertTo2D(wall.startPoint)
    let end = convertTo2D(wall.endPoint)
    
    let rect = CGRect(
        x: min(start.x, end.x),
        y: min(start.y, end.y),
        width: abs(start.x - end.x),
        height: abs(start.y - end.y)
    )
    let length = CGFloat(wall.length) * scale
    let height = CGFloat(wall.height) * scale
    
    return (rect, length, height)
}

4. 绘制带尺寸的平面图 #

使用CoreGraphics绘制墙壁并添加尺寸标注：

import UIKit

class FloorPlanCanvas: UIView {
    private var wallData: [(rect: CGRect, length: CGFloat)] = []
    
    func update(with walls: [RoomWall]) {
        wallData = walls.map(wall2DData)
        setNeedsDisplay()
    }
    
    override func draw(_ rect: CGRect) {
        super.draw(rect)
        guard let ctx = UIGraphicsGetCurrentContext() else { return }
        
        // 绘制墙壁
        ctx.setStrokeColor(UIColor.black.cgColor)
        ctx.setLineWidth(3)
        wallData.forEach { ctx.stroke($0.rect) }
        
        // 绘制尺寸标注
        let textAttrs: [NSAttributedString.Key: Any] = [
            .font: UIFont.systemFont(ofSize: 14, weight: .medium),
            .foregroundColor: UIColor.systemRed
        ]
        wallData.forEach { data in
            let text = String(format: "%.1fm", data.length / scale)
            let textRect = CGRect(
                x: data.rect.midX - 30,
                y: data.rect.midY - 18,
                width: 60,
                height: 36
            )
            text.draw(in: textRect, withAttributes: textAttrs)
        }
    }
}

// 使用示例
func setupFloorPlan(roomAnchor: RoomAnchor, containerView: UIView) {
    let canvas = FloorPlanCanvas(frame: containerView.bounds)
    canvas.autoresizingMask = [.flexibleWidth, .flexibleHeight]
    containerView.addSubview(canvas)
    
    let (walls, _, _) = extractRoomComponents(from: roomAnchor)
    canvas.update(with: walls)
}

常见问题排查 #

• 若无法解析RoomAnchor：确认USDZ是通过RoomPlan框架导出的，而非普通3D模型
• 坐标偏移/缩放异常：调整scale参数，或检查ARKit坐标系与视图坐标系的映射关系
• 复杂结构显示错误：斜墙、弧形墙需额外处理路径绘制，上述代码仅适配直线墙

内容的提问来源于stack exchange，提问作者Deepika Udayakumar