Как мне преобразовать CGImage в CMSampleBufferRef? - PullRequest
Новая
       63

Как мне преобразовать CGImage в CMSampleBufferRef?

14 голосов
/ 20 сентября 2010

Я хотел бы преобразовать CGImage в CMSampleBufferRef и добавить его к AVAssetWriterInput, используя метод appendSampleBuffer:.Мне удалось получить CMSampleBufferRef, используя следующий код, но appendSampleBuffer: просто возвращает NO, когда я предоставляю полученный CMSampleBufferRef.Что я делаю не так? 

- (void) appendCGImage: (CGImageRef) frame
{
    const int width = CGImageGetWidth(frame);
    const int height = CGImageGetHeight(frame);

    // Create a dummy pixel buffer to try the encoding
    // on something simple.
    CVPixelBufferRef pixelBuffer = NULL;
    CVReturn status = CVPixelBufferCreate(kCFAllocatorDefault, width, height,
        kCVPixelFormatType_32BGRA, NULL, &pixelBuffer);
    NSParameterAssert(status == kCVReturnSuccess && pixelBuffer != NULL);

    // Sample timing info.
    CMTime frameTime = CMTimeMake(1, 30);
    CMTime currentTime = CMTimeAdd(lastSampleTime, frameTime);
    CMSampleTimingInfo timing = {frameTime, currentTime, kCMTimeInvalid};

    OSStatus result = 0;

    // Sample format.
    CMVideoFormatDescriptionRef videoInfo = NULL;
    result = CMVideoFormatDescriptionCreateForImageBuffer(NULL,
         pixelBuffer, &videoInfo);
    NSParameterAssert(result == 0 && videoInfo != NULL);

    // Create sample buffer.
    CMSampleBufferRef sampleBuffer = NULL;
    result = CMSampleBufferCreateForImageBuffer(kCFAllocatorDefault,
        pixelBuffer, true, NULL, NULL, videoInfo, &timing, &sampleBuffer);
    NSParameterAssert(result == 0 && sampleBuffer != NULL);

    // Ship out the frame.
    NSParameterAssert(CMSampleBufferDataIsReady(sampleBuffer));
    NSParameterAssert([writerInput isReadyForMoreMediaData]);
    BOOL success = [writerInput appendSampleBuffer:frame];
    NSParameterAssert(success); // no go :(
}

PS Я знаю, что в этом коде есть утечки памяти, я упустил часть кода для простоты.

1 Ответ

6 голосов
/ 21 сентября 2010

Ага, я полностью пропустил класс AVAssetWriterInputPixelBufferAdaptor, созданный специально для передачи пиксельных буферов на вход устройства записи. Теперь код работает, даже без грязных вещей CMSampleBuffer.

...