能否一次性初始化Metal的array texture？私有2D数组纹理创建填充方法

问题背景 #

Metal中的array texture（请勿与texture array混淆）可用于同时向GPU传递数量非编译时常量、尺寸相同的纹理。目前已知的创建方式是自定义MTLTextureDescriptor后手动复制数据，当前通过循环逐个复制切片：

let descriptor = MTLTextureDescriptor()
descriptor.width = 32
descriptor.height = 32
descriptor.mipmapLevelCount = 5
descriptor.storageMode = .private
descriptor.textureType = .type2DArray
descriptor.pixelFormat = .rgba8Unorm
descriptor.arrayLength = NUM_TEXTURES
if let texture = device.makeTexture(descriptor: descriptor) {
    for i in 0..<NUM_TEXTURES {
        commandEncoder.copy(from: sharedBufferWithTextureData, sourceOffset: i<<4096, sourceBytesPerRow: 128, sourceBytesPerImage: 4096, to: texture, destinationSlice: i, destinationLevel: 0, destinationOrigin: MTLOrigin())
    }
}
commandEncoder.generateMipmaps(for: texture)

核心疑问：

• 有没有办法一次性复制所有切片？OpenGL支持类似操作，Metal中如何实现？
• 创建textureType为type2DArray且storageMode为private的MTLTexture对象并填充数据的最佳方式是什么？

注：选择private作为storageMode是因为该类型纹理不支持shared模式，而managed模式会产生数据副本，造成不必要的内存占用。

一次性批量复制切片的实现方式 #

Metal没有直接的API把buffer数据一次性复制到2D数组纹理的所有切片，但可以通过临时staging纹理中转实现批量传输，避免循环调用copy(from:to:)：

1. 创建与目标private纹理参数完全一致的临时staging纹理（用managed模式兼容更多设备，若设备支持也可用shared模式）；
2. 批量写入所有切片数据到临时纹理；
3. 用单次copy命令把整个临时纹理的base level复制到目标private纹理。

示例代码：

// 1. 创建临时staging纹理（仅需base level，后续生成mipmap）
let stagingDescriptor = MTLTextureDescriptor()
stagingDescriptor.width = 32
stagingDescriptor.height = 32
stagingDescriptor.mipmapLevelCount = 1
stagingDescriptor.storageMode = .managed
stagingDescriptor.textureType = .type2DArray
stagingDescriptor.pixelFormat = .rgba8Unorm
stagingDescriptor.arrayLength = NUM_TEXTURES

guard let stagingTexture = device.makeTexture(descriptor: stagingDescriptor) else {
    fatalError("Failed to create staging texture")
}

// 2. 批量写入所有切片到临时纹理
stagingTexture.replace(
    region: MTLRegionMake2D(0, 0, 32, 32),
    mipmapLevel: 0,
    slice: 0,
    withBytes: sharedBufferWithTextureData.contents(),
    bytesPerRow: 128,
    bytesPerImage: 4096
)
// 若用shared模式，可直接memcpy到映射后的内存，无需调用replace

// 3. 创建目标private纹理
let targetDescriptor = MTLTextureDescriptor()
targetDescriptor.width = 32
targetDescriptor.height = 32
targetDescriptor.mipmapLevelCount = 5
targetDescriptor.storageMode = .private
targetDescriptor.textureType = .type2DArray
targetDescriptor.pixelFormat = .rgba8Unorm
targetDescriptor.arrayLength = NUM_TEXTURES

guard let targetTexture = device.makeTexture(descriptor: targetDescriptor) else {
    fatalError("Failed to create target private texture")
}

// 4. 一次性复制整个base level的所有切片
commandEncoder.copy(
    from: stagingTexture,
    sourceSlice: 0,
    sourceLevel: 0,
    sourceOrigin: MTLOrigin(),
    sourceSize: MTLSizeMake(32, 32, NUM_TEXTURES),
    to: targetTexture,
    destinationSlice: 0,
    destinationLevel: 0,
    destinationOrigin: MTLOrigin()
)

// 5. 生成mipmap
commandEncoder.generateMipmaps(for: targetTexture)

private模式2D数组纹理的最佳填充方案 #

结合性能与内存开销，推荐两种场景化方案：

方案一：临时staging纹理中转（优先推荐） #

适合切片数量较多的场景，通过staging纹理批量传输，减少命令编码器调用次数，降低GPU命令队列压力，同时避免managed模式的双份内存占用。

• 注意：若使用managed模式的staging纹理，需在复制前调用stagingTexture.didModify(region:)通知GPU同步数据。

方案二：循环复制优化（小数量切片场景） #

如果切片数量在几十个以内，原循环方案的性能开销可忽略，只需做简单优化：

• 提前计算固定参数（如单切片字节数），避免循环内重复计算；
• 统一提交命令缓冲区，不在循环内频繁提交。

示例优化代码：

guard let texture = device.makeTexture(descriptor: descriptor) else { return }
let bytesPerImage = 4096
for i in 0..<NUM_TEXTURES {
    let sourceOffset = i * bytesPerImage
    commandEncoder.copy(
        from: sharedBufferWithTextureData,
        sourceOffset: sourceOffset,
        sourceBytesPerRow: 128,
        sourceBytesPerImage: bytesPerImage,
        to: texture,
        destinationSlice: i,
        destinationLevel: 0,
        destinationOrigin: MTLOrigin()
    )
}
// 统一提交命令缓冲区
commandBuffer.commit()
commandBuffer.waitUntilCompleted()

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