ios - 将 MTLTexture 转换为 CVPixelBuffer

Question

我目前正在使用 Metal 开发实时过滤器。在定义我的 CIImage 之后，我将图像渲染到 MTLTexture。

下面是我的渲染代码。 context 是由 Metal 支持的 CIContext；targetTexture 是附加到我的 MTKView 实例的 currentDrawable 属性的纹理的别名:

context?.render(drawImage, to: targetTexture, commandBuffer: commandBuffer, bounds: targetRect, colorSpace: colorSpace)


它呈现正确，因为我可以看到图像显示在 Metal View 上。

问题是在渲染图像(并显示它)之后，我想提取 CVPixelBuffer 并使用类 AVAssetWriter 将其保存到磁盘。

另一种选择是有两个渲染步骤，一个渲染到纹理，另一个渲染到 CVPixelBuffer。 (但不清楚如何创建这样的缓冲区，或者两个渲染步骤对帧率的影响)

任何帮助将不胜感激，谢谢!

Answer 1

您可以尝试像这样从 MTLTexture 复制原始数据:

var outPixelbuffer: CVPixelBuffer?
if let datas = targetTexture.texture.buffer?.contents() {
    CVPixelBufferCreateWithBytes(kCFAllocatorDefault, targetTexture.width, 
    targetTexture.height, kCVPixelFormatType_64RGBAHalf, datas, 
    targetTexture.texture.bufferBytesPerRow, nil, nil, nil, &outPixelbuffer);
}