FFmpeg(2016版)——RGB 32フォーマット変換YUV 420

2391 ワード

完全なコードは以下の通りです.imageはQImage、RGB 32です.
    static AVFrame *pframe = av_frame_alloc();


    int w = image.width();
    int h = image.height();
    //qDebug() << w;
    //qDebug() << h;

    //int numbytes = avpicture_get_size(AV_PIX_FMT_RGB32,w,h);   // , 
    static int numbytes = av_image_get_buffer_size(AV_PIX_FMT_RGB32, w, h, 1);

    static uint8_t *buffer = (uint8_t*)av_malloc(numbytes*sizeof(uint8_t));

    //avpicture_fill((AVPicture*)pframe, buffer, AV_PIX_FMT_RGB32, w, h); // 
    static int ret = av_image_fill_arrays(pframe->data, pframe->linesize, buffer,
                         AV_PIX_FMT_RGB32,w,h,1);
    if (ret < 0)
    {
        qDebug() << "error1";
    }
    pframe->data[0] = image.bits(); // image 

    static AVFrame *pframe_yuv = av_frame_alloc();
    static int  numbytes_yuv = av_image_get_buffer_size(AV_PIX_FMT_YUV420P, w, h, 1);
    static uint8_t *buffer_yuv = (uint8_t*)av_malloc(numbytes_yuv*sizeof(uint8_t));

    //avpicture_fill((AVPicture*)pframe_yuv, buffer_yuv, AV_PIX_FMT_YUV420P, w, h);

    static int ret_yuv = av_image_fill_arrays(pframe_yuv->data, pframe_yuv->linesize, buffer_yuv,
                            AV_PIX_FMT_YUV420P,w,h,1);
    if (ret_yuv < 0)
    {
        qDebug() << "error2";
    }

    static SwsContext *rgb_to_yuv_ctx = sws_getContext(w, h, AV_PIX_FMT_RGB32,
                                                w, h, AV_PIX_FMT_YUV420P,
                                                SWS_BICUBIC, NULL,NULL,NULL);  // 

    if (!rgb_to_yuv_ctx)
    {
        qDebug() << "error3";
    }

    ret = sws_scale(rgb_to_yuv_ctx, pframe->data, pframe->linesize,
              0, h, pframe_yuv->data, pframe_yuv->linesize);

    if (ret < 0)
    {
        qDebug() << "error4";
    }

    FILE *output=fopen("me_0.yuv","ab+");

    fwrite(pframe_yuv->data[0],w*h,1,output);
    fwrite(pframe_yuv->data[1],w*h/4,1,output);
    fwrite(pframe_yuv->data[2],w*h/4,1,output);

    fclose(output);

    sws_freeContext(rgb_to_yuv_ctx);
    av_free(buffer);
    av_free(buffer_yuv);
    av_free(pframe);
    av_free(pframe_yuv);

画像を保存する場合、yuv 420は、4つのyが1つのuと1つのvに対応するため、data[0]がyに対応する、data[1]がuに対応し、data[2]がvに対応する.
yuv変換が正しいことを検証するにはffplayで開いて正しいかどうかを確認し、ffmpegをダウンロードすればいい.