ffmpeg4.1.3 rgb转yuv

本文详细介绍了如何使用ffmpeg4.1.3将RGB图像转换为YUV格式,包括转换过程和YUV的存储格式,是进行视频处理的实用教程。

 https://blog.csdn.net/asahinokawa/article/details/80596655 YUV存储格式

// QImage 的构造:QImage image(rgb_buffer, 1920, 1080, QImage::Format_RGB32);
void writeRGB32BufferToYuv420P(QImage& image)
{
    // image的尺寸
    int imageWidth = image.width();
    int imageHeight = image.height();

    // 计算AV_PIX_FMT_RGB32格式化 imageWidth*imageHeight大小的image,所占内存大小,并分配缓冲区rgbBuffer保存image.bits()
    int numbytes = av_image_get_buffer_size(AV_PIX_FMT_RGB32, imageWidth, imageHeight, 1);//1920*1080*4
    uint8_t *rgbBuffer = (uint8_t*)av_malloc(numbytes*sizeof(uint8_t));

    // 分配一帧内存,并关联到rgbBuffer,pRgbFrame->data[0]保存image.bits(),pRgbFrame->linesize[0]保存image的一行内存大小
    uint8_t* rgbDataBuffer[4];
    int rgbDataLinesize[4];
    int ret = av_image_fill_arrays(rgbDataBuffer, rgbDataLinesize, rgbBuffer, AV_PIX_FMT_RGB32,imageWidth,imageHeight,1);
    rgbDataBuffer[0]=image.bits();// 内存拷贝
    qDebug()<<"rgb linesize[0]:"<<rgbDataLinesize[0]<<endl; // 1920*4
    qDebug()<<"rgb cmp:"<<(!memcmp(rgbDataBuffer[0],image.bits(), numbytes));

    // 分配一帧yuv420内存
    uint8_t* yuvDataBuffer[4];
    int yuvDataLinesize[4];
    int  numbytes_yuv = av_image_get_buffer_size(AV_PIX_FMT_YUV420P, imageWidth, imageHeight, 1);
    uint8_t *buffer_yuv = (uint8_t*)av_malloc(numbytes_yuv*sizeof(uint8_t));
    int ret_yuv = av_image_fill_arrays(yuvDataBuffer, yuvDataLinesize, buffer_yuv, AV_PIX_FMT_YUV420P,imageWidth,imageHeight,1);
    qDebug()<<"yuv number:"<<numbytes_yuv<<endl; // 1920*1080*3/2
    qDebug()<<"yuv-Y linesize[0]:"<<yuvDataLinesize[0]<<endl; // 1920
    qDebug()<<"yuv-U linesize[1]:"<<yuvDataLinesize[1]<<endl; // 960
    qDebug()<<"yuv-V linesize[1]:"<<yuvDataLinesize[2]<<endl; // 960

    // 颜色编码转换器
    SwsContext *rgb_to_yuv_ctx = sws_getContext(imageWidth, imageHeight, AV_PIX_FMT_RGB32,
                                                imageWidth, imageHeight, AV_PIX_FMT_YUV420P,
                                                SWS_BICUBIC, NULL,NULL,NULL);

    // 颜色编码开始转换
    ret = sws_scale(rgb_to_yuv_ctx, rgbDataBuffer, rgbDataLinesize, 0, imageHeight, yuvDataBuffer, yuvDataLinesize);

    if (ret < 0)
    {
        qDebug() << "error4";
        sws_freeContext(rgb_to_yuv_ctx);
        av_free(rgbBuffer);
        av_free(buffer_yuv);

        return;
    }

    FILE *output=fopen("H:\\common_src\\FFmpeg_QT_rtsp\\h264\\me_0.yuv","ab+");

    fwrite(yuvDataBuffer[0],imageWidth*imageHeight,1,output);
    fwrite(yuvDataBuffer[1],imageWidth*imageHeight/4,1,output);
    fwrite(yuvDataBuffer[2],imageWidth*imageHeight/4,1,output);

    fclose(output);


    sws_freeContext(rgb_to_yuv_ctx);
    av_free(rgbBuffer);
    av_free(buffer_yuv);

}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值