FFmpeg(2016版)——RGB 32フォーマット変換YUV 420
2391 ワード
完全なコードは以下の通りです.imageはQImage、RGB 32です.
画像を保存する場合、yuv 420は、4つのyが1つのuと1つのvに対応するため、data[0]がyに対応する、data[1]がuに対応し、data[2]がvに対応する.
yuv変換が正しいことを検証するにはffplayで開いて正しいかどうかを確認し、ffmpegをダウンロードすればいい.
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をダウンロードすればいい.