|
|
@ -34,7 +34,7 @@ |
|
|
|
|
|
|
|
|
|
|
|
typedef struct Ir2Context{ |
|
|
|
typedef struct Ir2Context{ |
|
|
|
AVCodecContext *avctx; |
|
|
|
AVCodecContext *avctx; |
|
|
|
AVFrame picture; |
|
|
|
AVFrame *picture; |
|
|
|
GetBitContext gb; |
|
|
|
GetBitContext gb; |
|
|
|
int decode_delta; |
|
|
|
int decode_delta; |
|
|
|
} Ir2Context; |
|
|
|
} Ir2Context; |
|
|
@ -146,7 +146,7 @@ static int ir2_decode_frame(AVCodecContext *avctx, |
|
|
|
const uint8_t *buf = avpkt->data; |
|
|
|
const uint8_t *buf = avpkt->data; |
|
|
|
int buf_size = avpkt->size; |
|
|
|
int buf_size = avpkt->size; |
|
|
|
AVFrame *picture = data; |
|
|
|
AVFrame *picture = data; |
|
|
|
AVFrame * const p = &s->picture; |
|
|
|
AVFrame * const p = s->picture; |
|
|
|
int start, ret; |
|
|
|
int start, ret; |
|
|
|
|
|
|
|
|
|
|
|
if ((ret = ff_reget_buffer(avctx, p)) < 0) { |
|
|
|
if ((ret = ff_reget_buffer(avctx, p)) < 0) { |
|
|
@ -173,36 +173,36 @@ static int ir2_decode_frame(AVCodecContext *avctx, |
|
|
|
|
|
|
|
|
|
|
|
if (s->decode_delta) { /* intraframe */ |
|
|
|
if (s->decode_delta) { /* intraframe */ |
|
|
|
if ((ret = ir2_decode_plane(s, avctx->width, avctx->height, |
|
|
|
if ((ret = ir2_decode_plane(s, avctx->width, avctx->height, |
|
|
|
s->picture.data[0], s->picture.linesize[0], |
|
|
|
p->data[0], p->linesize[0], |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
return ret; |
|
|
|
return ret; |
|
|
|
|
|
|
|
|
|
|
|
/* swapped U and V */ |
|
|
|
/* swapped U and V */ |
|
|
|
if ((ret = ir2_decode_plane(s, avctx->width >> 2, avctx->height >> 2, |
|
|
|
if ((ret = ir2_decode_plane(s, avctx->width >> 2, avctx->height >> 2, |
|
|
|
s->picture.data[2], s->picture.linesize[2], |
|
|
|
p->data[2], p->linesize[2], |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
return ret; |
|
|
|
return ret; |
|
|
|
if ((ret = ir2_decode_plane(s, avctx->width >> 2, avctx->height >> 2, |
|
|
|
if ((ret = ir2_decode_plane(s, avctx->width >> 2, avctx->height >> 2, |
|
|
|
s->picture.data[1], s->picture.linesize[1], |
|
|
|
p->data[1], p->linesize[1], |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
return ret; |
|
|
|
return ret; |
|
|
|
} else { /* interframe */ |
|
|
|
} else { /* interframe */ |
|
|
|
if ((ret = ir2_decode_plane_inter(s, avctx->width, avctx->height, |
|
|
|
if ((ret = ir2_decode_plane_inter(s, avctx->width, avctx->height, |
|
|
|
s->picture.data[0], s->picture.linesize[0], |
|
|
|
p->data[0], p->linesize[0], |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
return ret; |
|
|
|
return ret; |
|
|
|
/* swapped U and V */ |
|
|
|
/* swapped U and V */ |
|
|
|
if ((ret = ir2_decode_plane_inter(s, avctx->width >> 2, avctx->height >> 2, |
|
|
|
if ((ret = ir2_decode_plane_inter(s, avctx->width >> 2, avctx->height >> 2, |
|
|
|
s->picture.data[2], s->picture.linesize[2], |
|
|
|
p->data[2], p->linesize[2], |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
return ret; |
|
|
|
return ret; |
|
|
|
if ((ret = ir2_decode_plane_inter(s, avctx->width >> 2, avctx->height >> 2, |
|
|
|
if ((ret = ir2_decode_plane_inter(s, avctx->width >> 2, avctx->height >> 2, |
|
|
|
s->picture.data[1], s->picture.linesize[1], |
|
|
|
p->data[1], p->linesize[1], |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
ir2_luma_table)) < 0) |
|
|
|
return ret; |
|
|
|
return ret; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
if ((ret = av_frame_ref(picture, &s->picture)) < 0) |
|
|
|
if ((ret = av_frame_ref(picture, p)) < 0) |
|
|
|
return ret; |
|
|
|
return ret; |
|
|
|
|
|
|
|
|
|
|
|
*got_frame = 1; |
|
|
|
*got_frame = 1; |
|
|
@ -219,7 +219,9 @@ static av_cold int ir2_decode_init(AVCodecContext *avctx) |
|
|
|
|
|
|
|
|
|
|
|
avctx->pix_fmt= AV_PIX_FMT_YUV410P; |
|
|
|
avctx->pix_fmt= AV_PIX_FMT_YUV410P; |
|
|
|
|
|
|
|
|
|
|
|
avcodec_get_frame_defaults(&ic->picture); |
|
|
|
ic->picture = av_frame_alloc(); |
|
|
|
|
|
|
|
if (!ic->picture) |
|
|
|
|
|
|
|
return AVERROR(ENOMEM); |
|
|
|
|
|
|
|
|
|
|
|
ir2_vlc.table = vlc_tables; |
|
|
|
ir2_vlc.table = vlc_tables; |
|
|
|
ir2_vlc.table_allocated = 1 << CODE_VLC_BITS; |
|
|
|
ir2_vlc.table_allocated = 1 << CODE_VLC_BITS; |
|
|
@ -239,9 +241,8 @@ static av_cold int ir2_decode_init(AVCodecContext *avctx) |
|
|
|
static av_cold int ir2_decode_end(AVCodecContext *avctx) |
|
|
|
static av_cold int ir2_decode_end(AVCodecContext *avctx) |
|
|
|
{ |
|
|
|
{ |
|
|
|
Ir2Context * const ic = avctx->priv_data; |
|
|
|
Ir2Context * const ic = avctx->priv_data; |
|
|
|
AVFrame *pic = &ic->picture; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
av_frame_unref(pic); |
|
|
|
av_frame_free(&ic->picture); |
|
|
|
|
|
|
|
|
|
|
|
return 0; |
|
|
|
return 0; |
|
|
|
} |
|
|
|
} |
|
|
|