On 01/02/2012 11:50 AM, Reinhard Tartler wrote:
> From: Eli Friedman <[email protected]>
>
> Imported from FFmpeg, based on 7d6439 and all futher revisions on the
> file until bef8df.
> ---
> Changelog | 1 +
> libavcodec/Makefile | 1 +
> libavcodec/allcodecs.c | 1 +
> libavcodec/avcodec.h | 1 +
> libavcodec/escape130.c | 319
> ++++++++++++++++++++++++++++++++++++++++++++++++
> libavformat/rpl.c | 2 -
> 6 files changed, 323 insertions(+), 2 deletions(-)
> create mode 100644 libavcodec/escape130.c
also needs a minor bump in libavcodec/version.h
> diff --git a/Changelog b/Changelog
> index 3730208..78b7e00 100644
> --- a/Changelog
> +++ b/Changelog
> @@ -5,6 +5,7 @@ releases are sorted from youngest to oldest.
> version <next>:
>
> - Automatic thread count based on detection number of (available) CPU cores
> +- Escape 130 decoder
>
>
> version 0.8_beta1:
> diff --git a/libavcodec/Makefile b/libavcodec/Makefile
> index 741df91..06b828f 100644
> --- a/libavcodec/Makefile
> +++ b/libavcodec/Makefile
> @@ -142,6 +142,7 @@ OBJS-$(CONFIG_EIGHTBPS_DECODER) += 8bps.o
> OBJS-$(CONFIG_EIGHTSVX_EXP_DECODER) += 8svx.o
> OBJS-$(CONFIG_EIGHTSVX_FIB_DECODER) += 8svx.o
> OBJS-$(CONFIG_ESCAPE124_DECODER) += escape124.o
> +OBJS-$(CONFIG_ESCAPE130_DECODER) += escape130.o
> OBJS-$(CONFIG_FFV1_DECODER) += ffv1.o rangecoder.o
> OBJS-$(CONFIG_FFV1_ENCODER) += ffv1.o rangecoder.o
> OBJS-$(CONFIG_FFVHUFF_DECODER) += huffyuv.o
> diff --git a/libavcodec/allcodecs.c b/libavcodec/allcodecs.c
> index 3980982..f12c075 100644
> --- a/libavcodec/allcodecs.c
> +++ b/libavcodec/allcodecs.c
> @@ -106,6 +106,7 @@ void avcodec_register_all(void)
> REGISTER_DECODER (EIGHTSVX_EXP, eightsvx_exp);
> REGISTER_DECODER (EIGHTSVX_FIB, eightsvx_fib);
> REGISTER_DECODER (ESCAPE124, escape124);
> + REGISTER_DECODER (ESCAPE130, escape130);
> REGISTER_ENCDEC (FFV1, ffv1);
> REGISTER_ENCDEC (FFVHUFF, ffvhuff);
> REGISTER_ENCDEC (FLASHSV, flashsv);
> diff --git a/libavcodec/avcodec.h b/libavcodec/avcodec.h
> index b139a8a..01af3c2 100644
> --- a/libavcodec/avcodec.h
> +++ b/libavcodec/avcodec.h
> @@ -208,6 +208,7 @@ enum CodecID {
> CODEC_ID_8SVX_FIB,
> #endif
> CODEC_ID_ESCAPE124,
> + CODEC_ID_ESCAPE130,
> CODEC_ID_DIRAC,
> CODEC_ID_BFI,
> CODEC_ID_CMV,
breaks abi
> diff --git a/libavcodec/escape130.c b/libavcodec/escape130.c
> new file mode 100644
> index 0000000..c9f4d77
> --- /dev/null
> +++ b/libavcodec/escape130.c
> @@ -0,0 +1,319 @@
> +/*
> + * Escape 130 Video Decoder
> + * Copyright (C) 2008 Eli Friedman (eli.friedman <at> gmail.com)
> + *
> + * This file is part of FFmpeg.
> + *
> + * FFmpeg is free software; you can redistribute it and/or
> + * modify it under the terms of the GNU Lesser General Public
> + * License as published by the Free Software Foundation; either
> + * version 2.1 of the License, or (at your option) any later version.
> + *
> + * FFmpeg is distributed in the hope that it will be useful,
> + * but WITHOUT ANY WARRANTY; without even the implied warranty of
> + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
> + * Lesser General Public License for more details.
> + *
> + * You should have received a copy of the GNU Lesser General Public
> + * License along with FFmpeg; if not, write to the Free Software
> + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
> USA
> + */
> +
> +#include "avcodec.h"
> +
> +#define BITSTREAM_READER_LE
> +#include "get_bits.h"
> +
> +typedef struct Escape130Context {
> + AVFrame frame;
> + uint8_t *bases;
> +} Escape130Context;
> +
> +/**
> + * Initialize the decoder
> + * @param avctx decoder context
> + * @return 0 success, negative on error
> + */
documentation for AVCodec functions is not needed
> +static av_cold int escape130_decode_init(AVCodecContext *avctx)
> +{
> + Escape130Context *s = avctx->priv_data;
> + avctx->pix_fmt = PIX_FMT_YUV420P;
> +
> + if((avctx->width&1) || (avctx->height&1)){
fix spacing
> + av_log(avctx, AV_LOG_ERROR, "Dimensions are not a multiple of the
> block size\n");
> + return AVERROR(EINVAL);
> + }
> +
> + s->bases= av_malloc(avctx->width * avctx->height /4);
fix spacing
> +
> + return 0;
> +}
> +
> +static av_cold int escape130_decode_close(AVCodecContext *avctx)
> +{
> + Escape130Context *s = avctx->priv_data;
> +
> + if (s->frame.data[0])
> + avctx->release_buffer(avctx, &s->frame);
> +
> + av_freep(&s->bases);
> +
> + return 0;
> +}
> +
> +static unsigned decode_skip_count(GetBitContext* gb) {
> + unsigned value;
> + // This function reads a maximum of 27 bits,
> + // which is within the padding space
> + if (get_bits_left(gb) < 1+3)
> + return -1;
is this check really needed then?
also, it looks like it can read just 1 bit.
> +
> + value = get_bits1(gb);
> + if (value)
> + return 0;
> +
> + value = get_bits(gb, 3);
> + if (value)
> + return value;
> +
> + value = get_bits(gb, 8);
> + if (value)
> + return value + 7;
> +
> + value = get_bits(gb, 15);
> + if (value)
> + return value + 262;
> +
> + return -1;
> +}
> +
> +/**
> + * Decode a single frame
> + * @param avctx decoder context
> + * @param data decoded frame
> + * @param data_size size of the decoded frame
> + * @param buf input buffer
> + * @param buf_size input buffer size
> + * @return 0 success, -1 on error
> + */
documentation for AVCodec functions is not needed
> +static int escape130_decode_frame(AVCodecContext *avctx,
> + void *data, int *data_size,
> + AVPacket *avpkt)
> +{
> + const uint8_t *buf = avpkt->data;
> + int buf_size = avpkt->size;
> + Escape130Context *s = avctx->priv_data;
vertical alignment
> +
> + GetBitContext gb;
> + unsigned i;
> +
> + uint8_t *old_y, *old_cb, *old_cr,
> + *new_y, *new_cb, *new_cr;
> + unsigned old_y_stride, old_cb_stride, old_cr_stride,
> + new_y_stride, new_cb_stride, new_cr_stride;
> + unsigned total_blocks = avctx->width * avctx->height / 4,
> + block_index, row_index = 0;
> + unsigned y[4] = {0}, cb = 16, cr = 16;
> + unsigned skip = -1;
> + unsigned y_base = 0;
> + uint8_t *yb= s->bases;
> +
> + AVFrame new_frame = { { 0 } };
wouldn't avcodec_get_frame_defaults() be more appropriate here?
also, why not have both AVFrames in the private context instead of
having one on the stack?
> +
> + init_get_bits(&gb, buf, buf_size * 8);
> +
> + if (get_bits_left(&gb) < 128)
> + return -1;
print an error message
return AVERROR(INVALIDDATA)
> +
> + // Header; no useful information in here
> + skip_bits_long(&gb, 128);
> +
> + new_frame.reference = 3;
> + if (avctx->get_buffer(avctx, &new_frame)) {
> + av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
> + return -1;
> + }
get the get_buffer() return value and return it on error instead of -1
> +
> + new_y = new_frame.data[0];
> + new_cb = new_frame.data[1];
> + new_cr = new_frame.data[2];
> + new_y_stride = new_frame.linesize[0];
> + new_cb_stride = new_frame.linesize[1];
> + new_cr_stride = new_frame.linesize[2];
> + old_y = s->frame.data[0];
> + old_cb = s->frame.data[1];
> + old_cr = s->frame.data[2];
> + old_y_stride = s->frame.linesize[0];
> + old_cb_stride = s->frame.linesize[1];
> + old_cr_stride = s->frame.linesize[2];
vertical alignment
> +
> + av_log(avctx, AV_LOG_DEBUG,
> + "Strides: %i, %i\n",
> + new_y_stride, new_cb_stride);
probably should be av_dlog()
> +
> + for (block_index = 0; block_index < total_blocks; block_index++) {
> + // Note that this call will make us skip the rest of the blocks
> + // if the frame prematurely ends
> + if (skip == -1)
> + skip = decode_skip_count(&gb);
> +
> + if (skip) {
> + if (old_y) {
> + y[0] = old_y[0] / 4;
> + y[1] = old_y[1] / 4;
> + y[2] = old_y[old_y_stride] / 4;
> + y[3] = old_y[old_y_stride+1] / 4;
> + y_base= yb[0];
vertical alignment and spacing
> + cb = old_cb[0] / 8;
> + cr = old_cr[0] / 8;
> + } else {
> + y_base=y[0] = y[1] = y[2] = y[3] = 0;
> + cb = cr = 16;
> + }
> + } else {
> + if (get_bits1(&gb)) {
> + static const uint8_t offset_table[] = {2, 4, 10, 20};
> + static const int8_t sign_table[64][4] =
> + { {0, 0, 0, 0},
> + {-1, 1, 0, 0},
> + {1, -1, 0, 0},
> + {-1, 0, 1, 0},
> + {-1, 1, 1, 0},
> + {0, -1, 1, 0},
> + {1, -1, 1, 0},
> + {-1, -1, 1, 0},
> + {1, 0, -1, 0},
> + {0, 1, -1, 0},
> + {1, 1, -1, 0},
> + {-1, 1, -1, 0},
> + {1, -1, -1, 0},
> + {-1, 0, 0, 1},
> + {-1, 1, 0, 1},
> + {0, -1, 0, 1},
> +
> + {0, 0, 0, 0},
> + {1, -1, 0, 1},
> + {-1, -1, 0, 1},
> + {-1, 0, 1, 1},
> + {-1, 1, 1, 1},
> + {0, -1, 1, 1},
> + {1, -1, 1, 1},
> + {-1, -1, 1, 1},
> + {0, 0, -1, 1},
> + {1, 0, -1, 1},
> + {-1, 0, -1, 1},
> + {0, 1, -1, 1},
> + {1, 1, -1, 1},
> + {-1, 1, -1, 1},
> + {0, -1, -1, 1},
> + {1, -1, -1, 1},
> +
> + {0, 0, 0, 0},
> + {-1, -1, -1, 1},
> + {1, 0, 0, -1},
> + {0, 1, 0, -1},
> + {1, 1, 0, -1},
> + {-1, 1, 0, -1},
> + {1, -1, 0, -1},
> + {0, 0, 1, -1},
> + {1, 0, 1, -1},
> + {-1, 0, 1, -1},
> + {0, 1, 1, -1},
> + {1, 1, 1, -1},
> + {-1, 1, 1, -1},
> + {0, -1, 1, -1},
> + {1, -1, 1, -1},
> + {-1, -1, 1, -1},
> +
> + {0, 0, 0, 0},
> + {1, 0, -1, -1},
> + {0, 1, -1, -1},
> + {1, 1, -1, -1},
> + {-1, 1, -1, -1},
> + {1, -1, -1, -1} };
at least the large table should probably be separate.
and also formatted better.
> + unsigned sign_selector = get_bits(&gb, 6);
> + unsigned difference_selector = get_bits(&gb, 2);
> + y_base = 2 * get_bits(&gb, 5);
vertical alignment
> + for (i = 0; i < 4; i++) {
> + y[i] = av_clip((int)y_base +
> offset_table[difference_selector] *
> + sign_table[sign_selector][i], 0,
> 63);
vertical alignment
> + }
> + } else if (get_bits1(&gb)) {
> + if (get_bits1(&gb)) {
> + y_base = get_bits(&gb, 6);
> + } else {
> + unsigned adjust_index = get_bits(&gb, 3);
> + static const int8_t adjust[] = {-4, -3, -2, -1, 1, 2, 3,
> 4};
spacing
> + y_base = (y_base + adjust[adjust_index]) & 63;
> + }
> + for (i = 0; i < 4; i++)
> + y[i] = y_base;
> + }
> +
> + if (get_bits1(&gb)) {
> + if (get_bits1(&gb)) {
> + cb = get_bits(&gb, 5);
> + cr = get_bits(&gb, 5);
> + } else {
> + unsigned adjust_index = get_bits(&gb, 3);
> + static const int8_t adjust[2][8] =
> + { { 1, 1, 0, -1, -1, -1, 0, 1 },
> + { 0, 1, 1, 1, 0, -1, -1, -1 } };
> + cb = (cb + adjust[0][adjust_index]) & 31;
> + cr = (cr + adjust[1][adjust_index]) & 31;
> + }
> + }
> + }
> + *yb++= y_base;
> +
> + new_y[0] = y[0] * 4;
> + new_y[1] = y[1] * 4;
> + new_y[new_y_stride] = y[2] * 4;
> + new_y[new_y_stride + 1] = y[3] * 4;
vertical alignment
> + *new_cb = cb * 8;
> + *new_cr = cr * 8;
> +
> + if (old_y)
> + old_y += 2, old_cb++, old_cr++;
> + new_y += 2, new_cb++, new_cr++;
> + row_index++;
> + if (avctx->width / 2 == row_index) {
> + row_index = 0;
> + if (old_y) {
> + old_y += old_y_stride * 2 - avctx->width;
> + old_cb += old_cb_stride - avctx->width / 2;
> + old_cr += old_cr_stride - avctx->width / 2;
> + }
> + new_y += new_y_stride * 2 - avctx->width;
> + new_cb += new_cb_stride - avctx->width / 2;
> + new_cr += new_cr_stride - avctx->width / 2;
more vertical alignment
> + }
> +
> + skip--;
> + }
> +
> + av_log(avctx, AV_LOG_DEBUG,
> + "Escape sizes: %i, %i\n",
> + buf_size, get_bits_count(&gb) / 8);
av_dlog()
> +
> + if (s->frame.data[0])
> + avctx->release_buffer(avctx, &s->frame);
> +
> + *(AVFrame*)data = s->frame = new_frame;
> + *data_size = sizeof(AVFrame);
> +
> + return buf_size;
> +}
> +
> +
> +AVCodec ff_escape130_decoder = {
> + .name = "escape130",
> + .type = AVMEDIA_TYPE_VIDEO,
> + .id = CODEC_ID_ESCAPE130,
> + .priv_data_size = sizeof(Escape130Context),
> + .init = escape130_decode_init,
> + .close = escape130_decode_close,
> + .decode = escape130_decode_frame,
> + .capabilities = CODEC_CAP_DR1,
> + .long_name = NULL_IF_CONFIG_SMALL("Escape 130"),
> +};
> diff --git a/libavformat/rpl.c b/libavformat/rpl.c
> index 0fa00f3..bf5fe0a 100644
> --- a/libavformat/rpl.c
> +++ b/libavformat/rpl.c
> @@ -164,11 +164,9 @@ static int rpl_read_header(AVFormatContext *s,
> AVFormatParameters *ap)
> // The header is wrong here, at least sometimes
> vst->codec->bits_per_coded_sample = 16;
> break;
> -#if 0
> case 130:
> vst->codec->codec_id = CODEC_ID_ESCAPE130;
> break;
> -#endif
> default:
> av_log(s, AV_LOG_WARNING,
> "RPL video format %i not supported yet!\n",
_______________________________________________
libav-devel mailing list
[email protected]
https://lists.libav.org/mailman/listinfo/libav-devel
