nuv: use the AVFrame API properly.

This commit is contained in:
Anton Khirnov 2013-11-09 10:14:46 +01:00
parent d56a114adc
commit 5290e02a0a

View File

@ -31,7 +31,7 @@
#include "rtjpeg.h" #include "rtjpeg.h"
typedef struct { typedef struct {
AVFrame pic; AVFrame *pic;
int codec_frameheader; int codec_frameheader;
int quality; int quality;
int width, height; int width, height;
@ -137,7 +137,7 @@ static int codec_reinit(AVCodecContext *avctx, int width, int height,
c->decomp_buf = ptr; c->decomp_buf = ptr;
ff_rtjpeg_decode_init(&c->rtj, &c->dsp, c->width, c->height, ff_rtjpeg_decode_init(&c->rtj, &c->dsp, c->width, c->height,
c->lq, c->cq); c->lq, c->cq);
av_frame_unref(&c->pic); av_frame_unref(c->pic);
} else if (quality != c->quality) } else if (quality != c->quality)
ff_rtjpeg_decode_init(&c->rtj, &c->dsp, c->width, c->height, ff_rtjpeg_decode_init(&c->rtj, &c->dsp, c->width, c->height,
c->lq, c->cq); c->lq, c->cq);
@ -232,23 +232,23 @@ static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
} }
if (keyframe) { if (keyframe) {
av_frame_unref(&c->pic); av_frame_unref(c->pic);
init_frame = 1; init_frame = 1;
} }
result = ff_reget_buffer(avctx, &c->pic); result = ff_reget_buffer(avctx, c->pic);
if (result < 0) { if (result < 0) {
av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
return result; return result;
} }
if (init_frame) { if (init_frame) {
memset(c->pic.data[0], 0, avctx->height * c->pic.linesize[0]); memset(c->pic->data[0], 0, avctx->height * c->pic->linesize[0]);
memset(c->pic.data[1], 0x80, avctx->height * c->pic.linesize[1] / 2); memset(c->pic->data[1], 0x80, avctx->height * c->pic->linesize[1] / 2);
memset(c->pic.data[2], 0x80, avctx->height * c->pic.linesize[2] / 2); memset(c->pic->data[2], 0x80, avctx->height * c->pic->linesize[2] / 2);
} }
c->pic.pict_type = keyframe ? AV_PICTURE_TYPE_I : AV_PICTURE_TYPE_P; c->pic->pict_type = keyframe ? AV_PICTURE_TYPE_I : AV_PICTURE_TYPE_P;
c->pic.key_frame = keyframe; c->pic->key_frame = keyframe;
// decompress/copy/whatever data // decompress/copy/whatever data
switch (comptype) { switch (comptype) {
case NUV_LZO: case NUV_LZO:
@ -258,19 +258,19 @@ static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
av_log(avctx, AV_LOG_ERROR, "uncompressed frame too short\n"); av_log(avctx, AV_LOG_ERROR, "uncompressed frame too short\n");
height = buf_size / c->width / 3 * 2; height = buf_size / c->width / 3 * 2;
} }
copy_frame(&c->pic, buf, c->width, height); copy_frame(c->pic, buf, c->width, height);
break; break;
} }
case NUV_RTJPEG_IN_LZO: case NUV_RTJPEG_IN_LZO:
case NUV_RTJPEG: case NUV_RTJPEG:
ret = ff_rtjpeg_decode_frame_yuv420(&c->rtj, &c->pic, buf, buf_size); ret = ff_rtjpeg_decode_frame_yuv420(&c->rtj, c->pic, buf, buf_size);
if (ret < 0) if (ret < 0)
return ret; return ret;
break; break;
case NUV_BLACK: case NUV_BLACK:
memset(c->pic.data[0], 0, c->width * c->height); memset(c->pic->data[0], 0, c->width * c->height);
memset(c->pic.data[1], 128, c->width * c->height / 4); memset(c->pic->data[1], 128, c->width * c->height / 4);
memset(c->pic.data[2], 128, c->width * c->height / 4); memset(c->pic->data[2], 128, c->width * c->height / 4);
break; break;
case NUV_COPY_LAST: case NUV_COPY_LAST:
/* nothing more to do here */ /* nothing more to do here */
@ -280,7 +280,7 @@ static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
} }
if ((result = av_frame_ref(picture, &c->pic)) < 0) if ((result = av_frame_ref(picture, c->pic)) < 0)
return result; return result;
*got_frame = 1; *got_frame = 1;
@ -292,8 +292,11 @@ static av_cold int decode_init(AVCodecContext *avctx)
NuvContext *c = avctx->priv_data; NuvContext *c = avctx->priv_data;
int ret; int ret;
c->pic = av_frame_alloc();
if (!c->pic)
return AVERROR(ENOMEM);
avctx->pix_fmt = AV_PIX_FMT_YUV420P; avctx->pix_fmt = AV_PIX_FMT_YUV420P;
c->pic.data[0] = NULL;
c->decomp_buf = NULL; c->decomp_buf = NULL;
c->quality = -1; c->quality = -1;
c->width = 0; c->width = 0;
@ -317,7 +320,7 @@ static av_cold int decode_end(AVCodecContext *avctx)
NuvContext *c = avctx->priv_data; NuvContext *c = avctx->priv_data;
av_freep(&c->decomp_buf); av_freep(&c->decomp_buf);
av_frame_unref(&c->pic); av_frame_free(&c->pic);
return 0; return 0;
} }