0

我正在尝试使用 startCaptureWithHandler 从 replaykit 中获取每一帧。

startCaptureWithHandler 返回一个 CMSampleBufferRef ,我需要将其转换为图像。

我使用这种方法转换为 UIImage 但它总是白色的。

- (UIImage *) imageFromSampleBuffer3:(CMSampleBufferRef) sampleBuffer
{
    CVImageBufferRef imageBuffer = CMSampleBufferGetImageBuffer(sampleBuffer);
    size_t width = CVPixelBufferGetWidth(imageBuffer);
    size_t height = CVPixelBufferGetHeight(imageBuffer);

    NSDictionary *options = [NSDictionary dictionaryWithObjectsAndKeys: [NSNumber numberWithBool:YES], kCVPixelBufferCGImageCompatibilityKey, [NSNumber numberWithBool:YES], kCVPixelBufferCGBitmapContextCompatibilityKey, nil];
    CVPixelBufferRef pxbuffer = NULL;
    CVPixelBufferCreate(kCFAllocatorDefault, width, height, kCVPixelFormatType_32ARGB, (__bridge CFDictionaryRef) options, &pxbuffer);

    CVPixelBufferLockFlags flags = (CVPixelBufferLockFlags)0;
    CVPixelBufferLockBaseAddress(pxbuffer, flags);
    void *pxdata = CVPixelBufferGetBaseAddress(pxbuffer);

    CGColorSpaceRef rgbColorSpace = CGColorSpaceCreateDeviceRGB();
//    CGContextRef context = CGBitmapContextCreate(pxdata, width, height, 8, CVPixelBufferGetBytesPerRow(pxbuffer), rgbColorSpace, kCGImageAlphaPremultipliedFirst);
    CGContextRef context = CGBitmapContextCreate(pxdata, width, height, 8, CVPixelBufferGetBytesPerRow(pxbuffer), rgbColorSpace, kCGImageAlphaPremultipliedFirst);

    CGImageRef quartzImage = CGBitmapContextCreateImage(context);

    CGColorSpaceRelease(rgbColorSpace);
    CGContextRelease(context);
    CVPixelBufferUnlockBaseAddress(pxbuffer, flags);

    UIImage *image = [UIImage imageWithCGImage:quartzImage scale:1.0f orientation:UIImageOrientationRight];

    CGImageRelease(quartzImage);
    return image;
}

谁能告诉我我哪里出错了?

4

1 回答 1

0

sampleBuffer 是 420f 格式,它有 2 个平面。
对于锁定内存,CVPixelBufferLockBaseAddress(imageBuffer, 0 then 1)。
对于获取 Y 平面数据,CVPixelBufferGetBaseAddressOfPlane(imageBuffer, 0)。
对于 UV 平面数据,CVPixelBufferGetBaseAddressOfPlane(imageBuffer, 1)。
不要忘记解锁记忆。我不确定如何将 YUV 转换为 RGB。

在您的代码中,您不会从 imageBuffer 读取图像数据。您只在没有图像数据的 pxbuffer 和 pxdata 上工作。

于 2018-09-25T17:59:34.057 回答