]> git.sesse.net Git - ffmpeg/blobdiff - libavcodec/vble.c
mpeg12: Move finding the best frame rate to common code
[ffmpeg] / libavcodec / vble.c
index 00ac8a3491b4d03b400616f14dcee3d97bc459a5..c3451cb9ed517e56092eacabd2a6654019dada9c 100644 (file)
  * VBLE Decoder
  */
 
-#define BITSTREAM_READER_LE
+#include "libavutil/imgutils.h"
 
+#define BITSTREAM_READER_LE
 #include "avcodec.h"
-#include "get_bits.h"
+#include "bitstream.h"
 #include "huffyuvdsp.h"
 #include "internal.h"
 #include "mathops.h"
@@ -40,18 +41,18 @@ typedef struct VBLEContext {
     uint8_t        *val; /* First holds the lengths of vlc symbols and then their values */
 } VBLEContext;
 
-static uint8_t vble_read_reverse_unary(GetBitContext *gb)
+static uint8_t vble_read_reverse_unary(BitstreamContext *bc)
 {
     /* At most we need to read 9 bits total to get indices up to 8 */
-    uint8_t val = show_bits(gb, 8);
+    uint8_t val = bitstream_peek(bc, 8);
 
     if (val) {
         val = 7 - av_log2_16bit(ff_reverse[val]);
-        skip_bits(gb, val + 1);
+        bitstream_skip(bc, val + 1);
         return val;
     } else {
-        skip_bits(gb, 8);
-        if (get_bits1(gb))
+        bitstream_skip(bc, 8);
+        if (bitstream_read_bit(bc))
             return 8;
     }
 
@@ -59,13 +60,13 @@ static uint8_t vble_read_reverse_unary(GetBitContext *gb)
     return UINT8_MAX;
 }
 
-static int vble_unpack(VBLEContext *ctx, GetBitContext *gb)
+static int vble_unpack(VBLEContext *ctx, BitstreamContext *bc)
 {
     int i;
 
     /* Read all the lengths in first */
     for (i = 0; i < ctx->size; i++) {
-        ctx->val[i] = vble_read_reverse_unary(gb);
+        ctx->val[i] = vble_read_reverse_unary(bc);
 
         if (ctx->val[i] == UINT8_MAX)
             return -1;
@@ -73,12 +74,12 @@ static int vble_unpack(VBLEContext *ctx, GetBitContext *gb)
 
     for (i = 0; i < ctx->size; i++) {
         /* Check we have enough bits left */
-        if (get_bits_left(gb) < ctx->val[i])
+        if (bitstream_bits_left(bc) < ctx->val[i])
             return -1;
 
         /* get_bits can't take a length of 0 */
         if (ctx->val[i])
-            ctx->val[i] = (1 << ctx->val[i]) + get_bits(gb, ctx->val[i]) - 1;
+            ctx->val[i] = (1 << ctx->val[i]) + bitstream_read(bc, ctx->val[i]) - 1;
     }
 
     return 0;
@@ -117,7 +118,7 @@ static int vble_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
 {
     VBLEContext *ctx = avctx->priv_data;
     AVFrame *pic     = data;
-    GetBitContext gb;
+    BitstreamContext bc;
     const uint8_t *src = avpkt->data;
     int version;
     int offset = 0;
@@ -139,10 +140,10 @@ static int vble_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
     if (version != 1)
         av_log(avctx, AV_LOG_WARNING, "Unsupported VBLE Version: %d\n", version);
 
-    init_get_bits(&gb, src + 4, (avpkt->size - 4) * 8);
+    bitstream_init8(&bc, src + 4, avpkt->size - 4);
 
     /* Unpack */
-    if (vble_unpack(ctx, &gb) < 0) {
+    if (vble_unpack(ctx, &bc) < 0) {
         av_log(avctx, AV_LOG_ERROR, "Invalid Code\n");
         return AVERROR_INVALIDDATA;
     }
@@ -183,8 +184,8 @@ static av_cold int vble_decode_init(AVCodecContext *avctx)
     avctx->pix_fmt = AV_PIX_FMT_YUV420P;
     avctx->bits_per_raw_sample = 8;
 
-    ctx->size = avpicture_get_size(avctx->pix_fmt,
-                                   avctx->width, avctx->height);
+    ctx->size = av_image_get_buffer_size(avctx->pix_fmt,
+                                         avctx->width, avctx->height, 1);
 
     ctx->val = av_malloc(ctx->size * sizeof(*ctx->val));
 
@@ -206,5 +207,6 @@ AVCodec ff_vble_decoder = {
     .init           = vble_decode_init,
     .close          = vble_decode_close,
     .decode         = vble_decode_frame,
-    .capabilities   = CODEC_CAP_DR1,
+    .capabilities   = AV_CODEC_CAP_DR1,
+    .caps_internal  = FF_CODEC_CAP_INIT_THREADSAFE,
 };