完整的代码如下,image是QImage,RGB32
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);
保存图片时,因为yuv420,是4个y对应一个u和一个v,data[0]对应y,data[1]对应u,data[2]对应v.
要验证yuv转换是否正确可以用ffplay打开看看是否正确,下载ffmpeg即可