macOS环境下在后台线程中使用Metal API的安全实践

核心原则 #

Metal的MTLCommandQueue本身是线程安全的，可在任意线程调用commandBuffer()获取命令缓冲区，但命令编码器（如MTLRenderCommandEncoder、MPSImageGaussianBlur的编码操作）必须在创建它的线程上使用，禁止跨线程传递。结合你的分块隔离场景，按以下步骤实现即可保证安全。

具体实现步骤 #

• 主线程提前初始化共享资源：创建并初始化MTLDevice、MTLCommandQueue、MPSImageGaussianBlur对象，这些对象均线程安全，可在任意线程复用。
• 单线程内的完整Metal流程：
  1. 线程计算完顶点数据后，从共享commandQueue获取新的MTLCommandBuffer。
  2. 创建当前分块专属的临时MTLTexture，设置匹配的像素格式与存储模式（建议用MTLStorageModeShared方便CPU与GPU数据交互）。
  3. 通过MTLBlitCommandEncoder将线程内的顶点数据写入临时输入纹理。
  4. 在当前线程调用MPSImageGaussianBlur的encodeToCommandBuffer:方法，将模糊操作编码到当前命令缓冲区。
  5. 再次使用MTLBlitCommandEncoder将模糊后的纹理数据拷贝到CPU可访问的MTLBuffer中。
  6. 提交命令缓冲区并调用waitUntilCompleted()同步等待GPU完成（确保线程内完成渲染流程）。
  7. 从MTLBuffer读取像素数据，转换为CGImage后绘制到当前线程专属的CGContext。
• 线程安全细节：
  • 每个线程必须独立创建自己的命令编码器、临时纹理和缓冲区，绝不跨线程共享这些对象。
  • 确保各线程的CGContext是独立实例，禁止跨线程操作同一上下文。

关键代码示例 #

// 主线程初始化的共享资源
@property (nonatomic, strong) id<MTLDevice> metalDevice;
@property (nonatomic, strong) id<MTLCommandQueue> commandQueue;
@property (nonatomic, strong) MPSImageGaussianBlur *gaussianBlurFilter;

// 并发线程内的分块处理函数
- (void)processTile:(TileInfo *)tile {
    // 1. 获取线程专属的命令缓冲区
    id<MTLCommandBuffer> commandBuffer = [self.commandQueue commandBuffer];
    
    // 2. 创建临时输入纹理
    MTLTextureDescriptor *inputDesc = [MTLTextureDescriptor texture2DDescriptorWithPixelFormat:MTLPixelFormatRGBA8Unorm
                                                                                        width:tile.width
                                                                                       height:tile.height
                                                                                    mipmapped:NO];
    inputDesc.storageMode = MTLStorageModeShared;
    id<MTLTexture> inputTexture = [self.metalDevice newTextureWithDescriptor:inputDesc];
    
    // 3. 将线程计算的顶点数据写入纹理
    id<MTLBlitCommandEncoder> blitEncoder = [commandBuffer blitCommandEncoder];
    [blitEncoder copyFromBuffer:tile.vertexBuffer
                   sourceOffset:0
              sourceBytesPerRow:tile.width * 4
                    sourceSize:(MTLSize){tile.width, tile.height, 1}
                    toTexture:inputTexture
            destinationSlice:0
            destinationLevel:0
           destinationOrigin:(MTLOrigin){0, 0, 0}];
    [blitEncoder endEncoding];
    
    // 4. 执行高斯模糊
    [self.gaussianBlurFilter encodeToCommandBuffer:commandBuffer sourceTexture:inputTexture destinationTexture:inputTexture];
    
    // 5. 拷贝模糊后的数据到CPU缓冲区
    id<MTLBuffer> outputBuffer = [self.metalDevice newBufferWithLength:tile.width * tile.height * 4 options:MTLResourceStorageModeShared];
    blitEncoder = [commandBuffer blitCommandEncoder];
    [blitEncoder copyFromTexture:inputTexture
                   sourceSlice:0
                   sourceLevel:0
                   sourceOrigin:(MTLOrigin){0, 0, 0}
                   sourceSize:(MTLSize){tile.width, tile.height, 1}
                   toBuffer:outputBuffer
           destinationOffset:0
      destinationBytesPerRow:tile.width * 4];
    [blitEncoder endEncoding];
    
    // 6. 提交并等待GPU完成
    [commandBuffer commit];
    [commandBuffer waitUntilCompleted];
    
    // 7. 转换为CGImage并绘制到当前线程的CGContext
    void *pixelData = outputBuffer.contents;
    CGDataProviderRef provider = CGDataProviderCreateWithData(NULL, pixelData, outputBuffer.length, NULL);
    CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
    CGImageRef cgImage = CGImageCreate(tile.width, tile.height, 8, 32, tile.width * 4, colorSpace, kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big, provider, NULL, NO, kCGRenderingIntentDefault);
    
    CGContextDrawImage(tile.context, tile.drawRect, cgImage);
    
    // 释放资源
    CGImageRelease(cgImage);
    CGColorSpaceRelease(colorSpace);
    CGDataProviderRelease(provider);
}

优化建议 #

• 可在主线程创建资源池，存放预分配的临时纹理和缓冲区，线程从池中获取资源、使用后归还，减少频繁创建销毁的开销。
• 若无需同步等待，可使用commandBuffer的addCompletedHandler:异步回调处理像素绘制，但需注意回调可能在任意线程执行，此时要切换回原线程操作CGContext。

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