FFmpeg YUV2RGB
1376 ワード
AVFrame* YUV2RGB( AVCodecContext * avctx, AVFrame * frame )
{
AVFrame* pFrameRGB=NULL;
pFrameRGB=avcodec_alloc_frame();
if(pFrameRGB==NULL)
return NULL;
// Determine required buffer size and allocate buffer
int numBytes=avpicture_get_size(PIX_FMT_BGR24, avctx->width,avctx->height);
uint8_t * buffer=(uint8_t *)av_malloc(numBytes*sizeof(uint8_t));
// Assign appropriate parts of buffer to image planes in pFrameRGB
// Note that pFrameRGB is an AVFrame, but AVFrame is a superset
// of AVPicture
avpicture_fill((AVPicture *)pFrameRGB, buffer, PIX_FMT_BGR24,avctx->width, avctx->height);
struct SwsContext *img_convert_ctx=NULL;
img_convert_ctx=sws_getCachedContext(img_convert_ctx,avctx->width,
avctx->height, avctx->pix_fmt,avctx->width, avctx->height,
PIX_FMT_RGB24, SWS_BICUBIC,NULL, NULL, NULL);
if( !img_convert_ctx )
{
fprintf(stderr, "Cannot initialize sws conversion context
");
}
sws_scale(img_convert_ctx, (const uint8_t* const*)frame->data,
frame->linesize, 0, avctx->height, pFrameRGB->data,
pFrameRGB->linesize);
return pFrameRGB;
}
pFrameRGB free