gifdec: convert to bytestream2
authorAnton Khirnov <anton@khirnov.net>
Fri, 15 Nov 2013 09:23:04 +0000 (10:23 +0100)
committerReinhard Tartler <siretart@tauware.de>
Sun, 5 Jan 2014 22:00:51 +0000 (17:00 -0500)
(cherry picked from commit 1f3e56b6dcc163a705704e98569d4850a31d651c)
Signed-off-by: Reinhard Tartler <siretart@tauware.de>
libavcodec/gifdec.c
libavcodec/lzw.c
libavcodec/lzw.h

index 8636780..99c24d5 100644 (file)
@@ -49,8 +49,7 @@ typedef struct GifState {
     int gce_delay;
 
     /* LZW compatible decoder */
-    const uint8_t *bytestream;
-    const uint8_t *bytestream_end;
+    GetByteContext gb;
     LZWState *lzw;
 
     /* aux buffers */
@@ -69,11 +68,11 @@ static int gif_read_image(GifState *s)
     int is_interleaved, has_local_palette, y, pass, y1, linesize, n, i;
     uint8_t *ptr, *spal, *palette, *ptr1;
 
-    left = bytestream_get_le16(&s->bytestream);
-    top = bytestream_get_le16(&s->bytestream);
-    width = bytestream_get_le16(&s->bytestream);
-    height = bytestream_get_le16(&s->bytestream);
-    flags = bytestream_get_byte(&s->bytestream);
+    left   = bytestream2_get_le16(&s->gb);
+    top    = bytestream2_get_le16(&s->gb);
+    width  = bytestream2_get_le16(&s->gb);
+    height = bytestream2_get_le16(&s->gb);
+    flags  = bytestream2_get_byte(&s->gb);
     is_interleaved = flags & 0x40;
     has_local_palette = flags & 0x80;
     bits_per_pixel = (flags & 0x07) + 1;
@@ -81,7 +80,7 @@ static int gif_read_image(GifState *s)
     av_dlog(s->avctx, "gif: image x=%d y=%d w=%d h=%d\n", left, top, width, height);
 
     if (has_local_palette) {
-        bytestream_get_buffer(&s->bytestream, s->local_palette, 3 * (1 << bits_per_pixel));
+        bytestream2_get_buffer(&s->gb, s->local_palette, 3 * (1 << bits_per_pixel));
         palette = s->local_palette;
     } else {
         palette = s->global_palette;
@@ -110,9 +109,9 @@ static int gif_read_image(GifState *s)
         s->image_palette[s->transparent_color_index] = 0;
 
     /* now get the image data */
-    code_size = bytestream_get_byte(&s->bytestream);
-    ff_lzw_decode_init(s->lzw, code_size, s->bytestream,
-                       s->bytestream_end - s->bytestream, FF_LZW_GIF);
+    code_size = bytestream2_get_byte(&s->gb);
+    ff_lzw_decode_init(s->lzw, code_size, s->gb.buffer,
+                       bytestream2_get_bytes_left(&s->gb), FF_LZW_GIF);
 
     /* read all the image */
     linesize = s->picture.linesize[0];
@@ -155,7 +154,8 @@ static int gif_read_image(GifState *s)
     }
     /* read the garbage data until end marker is found */
     ff_lzw_decode_tail(s->lzw);
-    s->bytestream = ff_lzw_cur_ptr(s->lzw);
+
+    bytestream2_skip(&s->gb, ff_lzw_size_read(s->lzw));
     return 0;
 }
 
@@ -164,8 +164,8 @@ static int gif_read_extension(GifState *s)
     int ext_code, ext_len, i, gce_flags, gce_transparent_index;
 
     /* extension */
-    ext_code = bytestream_get_byte(&s->bytestream);
-    ext_len = bytestream_get_byte(&s->bytestream);
+    ext_code = bytestream2_get_byte(&s->gb);
+    ext_len  = bytestream2_get_byte(&s->gb);
 
     av_dlog(s->avctx, "gif: ext_code=0x%x len=%d\n", ext_code, ext_len);
 
@@ -174,9 +174,9 @@ static int gif_read_extension(GifState *s)
         if (ext_len != 4)
             goto discard_ext;
         s->transparent_color_index = -1;
-        gce_flags = bytestream_get_byte(&s->bytestream);
-        s->gce_delay = bytestream_get_le16(&s->bytestream);
-        gce_transparent_index = bytestream_get_byte(&s->bytestream);
+        gce_flags    = bytestream2_get_byte(&s->gb);
+        s->gce_delay = bytestream2_get_le16(&s->gb);
+        gce_transparent_index = bytestream2_get_byte(&s->gb);
         if (gce_flags & 0x01)
             s->transparent_color_index = gce_transparent_index;
         else
@@ -187,7 +187,7 @@ static int gif_read_extension(GifState *s)
                gce_flags, s->gce_delay,
                s->transparent_color_index, s->gce_disposal);
 
-        ext_len = bytestream_get_byte(&s->bytestream);
+        ext_len = bytestream2_get_byte(&s->gb);
         break;
     }
 
@@ -195,8 +195,8 @@ static int gif_read_extension(GifState *s)
  discard_ext:
     while (ext_len != 0) {
         for (i = 0; i < ext_len; i++)
-            bytestream_get_byte(&s->bytestream);
-        ext_len = bytestream_get_byte(&s->bytestream);
+            bytestream2_get_byte(&s->gb);
+        ext_len = bytestream2_get_byte(&s->gb);
 
         av_dlog(s->avctx, "gif: ext_len1=%d\n", ext_len);
     }
@@ -209,31 +209,31 @@ static int gif_read_header1(GifState *s)
     int v, n;
     int has_global_palette;
 
-    if (s->bytestream_end < s->bytestream + 13)
+    if (bytestream2_get_bytes_left(&s->gb) < 13)
         return AVERROR_INVALIDDATA;
 
     /* read gif signature */
-    bytestream_get_buffer(&s->bytestream, sig, 6);
+    bytestream2_get_buffer(&s->gb, sig, 6);
     if (memcmp(sig, gif87a_sig, 6) != 0 &&
         memcmp(sig, gif89a_sig, 6) != 0)
         return AVERROR_INVALIDDATA;
 
     /* read screen header */
     s->transparent_color_index = -1;
-    s->screen_width = bytestream_get_le16(&s->bytestream);
-    s->screen_height = bytestream_get_le16(&s->bytestream);
+    s->screen_width  = bytestream2_get_le16(&s->gb);
+    s->screen_height = bytestream2_get_le16(&s->gb);
     if(   (unsigned)s->screen_width  > 32767
        || (unsigned)s->screen_height > 32767){
         av_log(NULL, AV_LOG_ERROR, "picture size too large\n");
         return AVERROR_INVALIDDATA;
     }
 
-    v = bytestream_get_byte(&s->bytestream);
+    v = bytestream2_get_byte(&s->gb);
     s->color_resolution = ((v & 0x70) >> 4) + 1;
     has_global_palette = (v & 0x80);
     s->bits_per_pixel = (v & 0x07) + 1;
-    s->background_color_index = bytestream_get_byte(&s->bytestream);
-    bytestream_get_byte(&s->bytestream);                /* ignored */
+    s->background_color_index = bytestream2_get_byte(&s->gb);
+    bytestream2_get_byte(&s->gb);                /* ignored */
 
     av_dlog(s->avctx, "gif: screen_w=%d screen_h=%d bpp=%d global_palette=%d\n",
            s->screen_width, s->screen_height, s->bits_per_pixel,
@@ -241,17 +241,17 @@ static int gif_read_header1(GifState *s)
 
     if (has_global_palette) {
         n = 1 << s->bits_per_pixel;
-        if (s->bytestream_end < s->bytestream + n * 3)
+        if (bytestream2_get_bytes_left(&s->gb) < n * 3)
             return AVERROR_INVALIDDATA;
-        bytestream_get_buffer(&s->bytestream, s->global_palette, n * 3);
+        bytestream2_get_buffer(&s->gb, s->global_palette, n * 3);
     }
     return 0;
 }
 
 static int gif_parse_next_image(GifState *s)
 {
-    while (s->bytestream < s->bytestream_end) {
-        int code = bytestream_get_byte(&s->bytestream);
+    while (bytestream2_get_bytes_left(&s->gb) > 0) {
+        int code = bytestream2_get_byte(&s->gb);
         int ret;
 
         av_dlog(s->avctx, "gif: code=%02x '%c'\n", code, code);
@@ -295,8 +295,7 @@ static int gif_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
     AVFrame *picture = data;
     int ret;
 
-    s->bytestream = buf;
-    s->bytestream_end = buf + buf_size;
+    bytestream2_init(&s->gb, buf, buf_size);
     if ((ret = gif_read_header1(s)) < 0)
         return ret;
 
@@ -318,7 +317,7 @@ static int gif_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
 
     *picture = s->picture;
     *got_frame = 1;
-    return s->bytestream - buf;
+    return bytestream2_tell(&s->gb);
 }
 
 static av_cold int gif_decode_close(AVCodecContext *avctx)
index 2c99014..0167140 100644 (file)
@@ -43,7 +43,7 @@ static const uint16_t mask[17] =
 };
 
 struct LZWState {
-    const uint8_t *pbuf, *ebuf;
+    const uint8_t *buf_start, *pbuf, *ebuf;
     int bbits;
     unsigned int bbuf;
 
@@ -92,9 +92,10 @@ static int lzw_get_code(struct LZWState * s)
     return c & s->curmask;
 }
 
-const uint8_t* ff_lzw_cur_ptr(LZWState *p)
+int ff_lzw_size_read(LZWState *p)
 {
-    return ((struct LZWState*)p)->pbuf;
+    struct LZWState *s = p;
+    return s->pbuf - s->buf_start;
 }
 
 void ff_lzw_decode_tail(LZWState *p)
index ab782f5..d925d35 100644 (file)
@@ -47,7 +47,7 @@ void ff_lzw_decode_open(LZWState **p);
 void ff_lzw_decode_close(LZWState **p);
 int ff_lzw_decode_init(LZWState *s, int csize, const uint8_t *buf, int buf_size, int mode);
 int ff_lzw_decode(LZWState *s, uint8_t *buf, int len);
-const uint8_t* ff_lzw_cur_ptr(LZWState *lzw);
+int ff_lzw_size_read(LZWState *lzw);
 void ff_lzw_decode_tail(LZWState *lzw);
 
 /** LZW encode state */