[FFMPEG-MACOS] AVFrame에서 NSImage로 전환

3119 단어
AVFrame -> CGImageRef -> NSImage
  • 코드
  • - (NSImage *)currentImage {
        AVFrame *now_frame = vs_obj->v_frame;
        AVCodecContext *nowCodCtx = vs_obj->rtspVideoCodecContext;
        if (!now_frame->data[0]) {
            return nil;
        }
        //[self convertFrameToRGB];
        
        sws_scale(img_convert_context, now_frame->data, now_frame->linesize, 0, nowCodCtx->height, picture.data, picture.linesize);
        
        //return [self imageFromAVPicture:picture width:_outputWidth height:_outputHeight];
        
        CGBitmapInfo bitmapInfo = kCGBitmapByteOrderDefault;
        CFDataRef data = CFDataCreateWithBytesNoCopy(kCFAllocatorDefault,picture.data[0],picture.linesize[0] * nowCodCtx->height,kCFAllocatorNull);
        
        CGDataProviderRef provider = CGDataProviderCreateWithCFData(data);
        CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
        
        //          
        CGImageRef cgImage = CGImageCreate(nowCodCtx->width, nowCodCtx->height, 8, 24, picture.linesize[0], colorSpace, bitmapInfo, provider, NULL, NO, kCGRenderingIntentDefault);
        
        CGColorSpaceRelease(colorSpace);
    //    NSImage *image = [NSImage imageWithCGImage:cgImage];
        NSImage *image = [image_lib imageFromCGImageRef:cgImage];
        
        CGImageRelease(cgImage);
        CGDataProviderRelease(provider);
        CFRelease(data);
        return image;
    }
    
    -(void)setupScaler {
        AVFrame *now_frame = vs_obj->v_frame;
        AVCodecContext *nowCodCtx = vs_obj->rtspVideoCodecContext;
        
        // Release old picture and scaler
        avpicture_free(&picture);
        sws_freeContext(img_convert_context);
        
        // Allocate RGB picture
        avpicture_alloc(&picture, AV_PIX_FMT_RGB24, nowCodCtx->width, nowCodCtx->height);
        
        // Setup scaler
        static int sws_flags =  SWS_FAST_BILINEAR;
        img_convert_context = sws_getContext(nowCodCtx->width,
                                             nowCodCtx->height,
                                             nowCodCtx->pix_fmt,
                                             nowCodCtx->width,
                                             nowCodCtx->height,
                                             AV_PIX_FMT_RGB24,
                                             sws_flags, NULL, NULL, NULL);
    }
    
    //  CGImageRef   NSImage *
    + (NSImage*) imageFromCGImageRef:(CGImageRef)image {
    //    NSRect imageRect = NSMakeRect(0.0, 0.0, 0.0, 0.0);
        
        CGContextRef imageContext = nil;
        
        // Get the image dimensions.
    //    imageRect.size.height = CGImageGetHeight(image);
    //    imageRect.size.width = CGImageGetWidth(image);
    //
        NSSize size = NSMakeSize(CGImageGetWidth(image), CGImageGetHeight(image));
        
        // Create a new image to receive the Quartz image data.
        NSImage* newImage = [[NSImage alloc] initWithCGImage:image size:size];
        
    //    [newImage lockFocus];
    //
    //    // Get the Quartz context and draw.
    //    imageContext = (CGContextRef)[[NSGraphicsContext currentContext] graphicsPort];
    //
    //    CGContextDrawImage(imageContext, *(CGRect*)&imageRect, image);
    //
    //    [newImage unlockFocus];
    //
        return newImage;
        
    }
    

    좋은 웹페이지 즐겨찾기