virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00
Code Issues Releases Activity

gifdec: port to bytestream2 API

Browse Source 
While here remove ff_lzw_cur_ptr() as it become obsolete.

Signed-off-by: Paul B Mahol <onemda@gmail.com>
This commit is contained in:
Paul B Mahol 2012-12-05 16:06:54 +00:00
parent e0a553d526
commit 3fd60d8049
3 changed files with 46 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

98
libavcodec/gifdec.c
View File

@ -65,9 +65,8 @@ typedef struct GifState {
int stored_img_size; int stored_img_size;
int stored_bg_color; int stored_bg_color;
GetByteContext gb;
/* LZW compatible decoder */ /* LZW compatible decoder */
const uint8_t *bytestream;
const uint8_t *bytestream_end;
LZWState *lzw; LZWState *lzw;
/* aux buffers */ /* aux buffers */
@ -79,12 +78,12 @@ typedef struct GifState {
int trans_color; /**< color value that is used instead of transparent color */ int trans_color; /**< color value that is used instead of transparent color */
} GifState; } GifState;
static void gif_read_palette(const uint8_t **buf, uint32_t *pal, int nb) static void gif_read_palette(GifState *s, uint32_t *pal, int nb)
{ {
const uint8_t *pal_end = *buf + nb * 3; int i;
for (; *buf < pal_end; *buf += 3, pal++) for (i = 0; i < nb; i++, pal++)
*pal = (0xffu << 24) | AV_RB24(*buf); *pal = (0xffu << 24) | bytestream2_get_be24u(&s->gb);
} }
static void gif_fill(AVFrame *picture, uint32_t color) static void gif_fill(AVFrame *picture, uint32_t color)
@ -141,14 +140,14 @@ static int gif_read_image(GifState *s)
uint8_t *idx; uint8_t *idx;
/* At least 9 bytes of Image Descriptor. */ /* At least 9 bytes of Image Descriptor. */
if (s->bytestream_end < s->bytestream + 9) if (bytestream2_get_bytes_left(&s->gb) < 9)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
left = bytestream_get_le16(&s->bytestream); left = bytestream2_get_le16u(&s->gb);
top = bytestream_get_le16(&s->bytestream); top = bytestream2_get_le16u(&s->gb);
width = bytestream_get_le16(&s->bytestream); width = bytestream2_get_le16u(&s->gb);
height = bytestream_get_le16(&s->bytestream); height = bytestream2_get_le16u(&s->gb);
flags = bytestream_get_byte(&s->bytestream); flags = bytestream2_get_byteu(&s->gb);
is_interleaved = flags & 0x40; is_interleaved = flags & 0x40;
has_local_palette = flags & 0x80; has_local_palette = flags & 0x80;
bits_per_pixel = (flags & 0x07) + 1; bits_per_pixel = (flags & 0x07) + 1;
@ -158,10 +157,10 @@ static int gif_read_image(GifState *s)
if (has_local_palette) { if (has_local_palette) {
pal_size = 1 << bits_per_pixel; pal_size = 1 << bits_per_pixel;
if (s->bytestream_end < s->bytestream + pal_size * 3) if (bytestream2_get_bytes_left(&s->gb) < pal_size * 3)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
gif_read_palette(&s->bytestream, s->local_palette, pal_size); gif_read_palette(s, s->local_palette, pal_size);
pal = s->local_palette; pal = s->local_palette;
} else { } else {
if (!s->has_global_palette) { if (!s->has_global_palette) {
@ -218,13 +217,13 @@ static int gif_read_image(GifState *s)
} }
/* Expect at least 2 bytes: 1 for lzw code size and 1 for block size. */ /* Expect at least 2 bytes: 1 for lzw code size and 1 for block size. */
if (s->bytestream_end < s->bytestream + 2) if (bytestream2_get_bytes_left(&s->gb) < 2)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
/* now get the image data */ /* now get the image data */
code_size = bytestream_get_byte(&s->bytestream); code_size = bytestream2_get_byteu(&s->gb);
if ((ret = ff_lzw_decode_init(s->lzw, code_size, s->bytestream, if ((ret = ff_lzw_decode_init(s->lzw, code_size, s->gb.buffer,
s->bytestream_end - s->bytestream, FF_LZW_GIF)) < 0) { bytestream2_get_bytes_left(&s->gb), FF_LZW_GIF)) < 0) {
av_log(s->avctx, AV_LOG_ERROR, "LZW init failed\n"); av_log(s->avctx, AV_LOG_ERROR, "LZW init failed\n");
return ret; return ret;
} }
@ -281,7 +280,6 @@ static int gif_read_image(GifState *s)
decode_tail: decode_tail:
/* read the garbage data until end marker is found */ /* read the garbage data until end marker is found */
ff_lzw_decode_tail(s->lzw); ff_lzw_decode_tail(s->lzw);
s->bytestream = ff_lzw_cur_ptr(s->lzw);
/* Graphic Control Extension's scope is single frame. /* Graphic Control Extension's scope is single frame.
* Remove its influence. */ * Remove its influence. */
@ -293,15 +291,15 @@ static int gif_read_image(GifState *s)
static int gif_read_extension(GifState *s) static int gif_read_extension(GifState *s)
{ {
int ext_code, ext_len, i, gce_flags, gce_transparent_index; int ext_code, ext_len, gce_flags, gce_transparent_index;
/* There must be at least 2 bytes: /* There must be at least 2 bytes:
* 1 for extension label and 1 for extension length. */ * 1 for extension label and 1 for extension length. */
if (s->bytestream_end < s->bytestream + 2) if (bytestream2_get_bytes_left(&s->gb) < 2)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
ext_code = bytestream_get_byte(&s->bytestream); ext_code = bytestream2_get_byteu(&s->gb);
ext_len = bytestream_get_byte(&s->bytestream); ext_len = bytestream2_get_byteu(&s->gb);
av_dlog(s->avctx, "ext_code=0x%x len=%d\n", ext_code, ext_len); av_dlog(s->avctx, "ext_code=0x%x len=%d\n", ext_code, ext_len);
@ -312,13 +310,13 @@ static int gif_read_extension(GifState *s)
/* We need at least 5 bytes more: 4 is for extension body /* We need at least 5 bytes more: 4 is for extension body
* and 1 for next block size. */ * and 1 for next block size. */
if (s->bytestream_end < s->bytestream + 5) if (bytestream2_get_bytes_left(&s->gb) < 5)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
s->transparent_color_index = -1; s->transparent_color_index = -1;
gce_flags = bytestream_get_byte(&s->bytestream); gce_flags = bytestream2_get_byteu(&s->gb);
bytestream_get_le16(&s->bytestream); // delay during which the frame is shown bytestream2_skipu(&s->gb, 2); // delay during which the frame is shown
gce_transparent_index = bytestream_get_byte(&s->bytestream); gce_transparent_index = bytestream2_get_byteu(&s->gb);
if (gce_flags & 0x01) if (gce_flags & 0x01)
s->transparent_color_index = gce_transparent_index; s->transparent_color_index = gce_transparent_index;
else else
@ -334,7 +332,7 @@ static int gif_read_extension(GifState *s)
av_dlog(s->avctx, "invalid value in gce_disposal (%d). Using default value of 0.\n", ext_len); av_dlog(s->avctx, "invalid value in gce_disposal (%d). Using default value of 0.\n", ext_len);
} }
ext_len = bytestream_get_byte(&s->bytestream); ext_len = bytestream2_get_byteu(&s->gb);
break; break;
} }
@ -342,12 +340,11 @@ static int gif_read_extension(GifState *s)
discard_ext: discard_ext:
while (ext_len != 0) { while (ext_len != 0) {
/* There must be at least ext_len bytes and 1 for next block size byte. */ /* There must be at least ext_len bytes and 1 for next block size byte. */
if (s->bytestream_end < s->bytestream + ext_len + 1) if (bytestream2_get_bytes_left(&s->gb) < ext_len + 1)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
for (i = 0; i < ext_len; i++) bytestream2_skipu(&s->gb, ext_len);
bytestream_get_byte(&s->bytestream); ext_len = bytestream2_get_byteu(&s->gb);
ext_len = bytestream_get_byte(&s->bytestream);
av_dlog(s->avctx, "ext_len1=%d\n", ext_len); av_dlog(s->avctx, "ext_len1=%d\n", ext_len);
} }
@ -360,19 +357,19 @@ static int gif_read_header1(GifState *s)
int v, n; int v, n;
int background_color_index; int background_color_index;
if (s->bytestream_end < s->bytestream + 13) if (bytestream2_get_bytes_left(&s->gb) < 13)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
/* read gif signature */ /* read gif signature */
bytestream_get_buffer(&s->bytestream, sig, 6); bytestream2_get_bufferu(&s->gb, sig, 6);
if (memcmp(sig, gif87a_sig, 6) != 0 && if (memcmp(sig, gif87a_sig, 6) != 0 &&
memcmp(sig, gif89a_sig, 6) != 0) memcmp(sig, gif89a_sig, 6) != 0)
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
/* read screen header */ /* read screen header */
s->transparent_color_index = -1; s->transparent_color_index = -1;
s->screen_width = bytestream_get_le16(&s->bytestream); s->screen_width = bytestream2_get_le16u(&s->gb);
s->screen_height = bytestream_get_le16(&s->bytestream); s->screen_height = bytestream2_get_le16u(&s->gb);
if( (unsigned)s->screen_width > 32767 if( (unsigned)s->screen_width > 32767
|| (unsigned)s->screen_height > 32767){ || (unsigned)s->screen_height > 32767){
av_log(s->avctx, AV_LOG_ERROR, "picture size too large\n"); av_log(s->avctx, AV_LOG_ERROR, "picture size too large\n");
@ -383,12 +380,12 @@ static int gif_read_header1(GifState *s)
if (!s->idx_line) if (!s->idx_line)
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
v = bytestream_get_byte(&s->bytestream); v = bytestream2_get_byteu(&s->gb);
s->color_resolution = ((v & 0x70) >> 4) + 1; s->color_resolution = ((v & 0x70) >> 4) + 1;
s->has_global_palette = (v & 0x80); s->has_global_palette = (v & 0x80);
s->bits_per_pixel = (v & 0x07) + 1; s->bits_per_pixel = (v & 0x07) + 1;
background_color_index = bytestream_get_byte(&s->bytestream); background_color_index = bytestream2_get_byteu(&s->gb);
n = bytestream_get_byte(&s->bytestream); n = bytestream2_get_byteu(&s->gb);
if (n) { if (n) {
s->avctx->sample_aspect_ratio.num = n + 15; s->avctx->sample_aspect_ratio.num = n + 15;
s->avctx->sample_aspect_ratio.den = 64; s->avctx->sample_aspect_ratio.den = 64;
@ -401,10 +398,10 @@ static int gif_read_header1(GifState *s)
if (s->has_global_palette) { if (s->has_global_palette) {
s->background_color_index = background_color_index; s->background_color_index = background_color_index;
n = 1 << s->bits_per_pixel; 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; return AVERROR_INVALIDDATA;
gif_read_palette(&s->bytestream, s->global_palette, n); gif_read_palette(s, s->global_palette, n);
s->bg_color = s->global_palette[s->background_color_index]; s->bg_color = s->global_palette[s->background_color_index];
} else } else
s->background_color_index = -1; s->background_color_index = -1;
@ -416,8 +413,8 @@ static int gif_parse_next_image(GifState *s, int *got_picture)
{ {
int ret; int ret;
*got_picture = 1; *got_picture = 1;
while (s->bytestream < s->bytestream_end) { while (bytestream2_get_bytes_left(&s->gb)) {
int code = bytestream_get_byte(&s->bytestream); int code = bytestream2_get_byte(&s->gb);
av_dlog(s->avctx, "code=%02x '%c'\n", code, code); av_dlog(s->avctx, "code=%02x '%c'\n", code, code);
@ -456,23 +453,20 @@ static av_cold int gif_decode_init(AVCodecContext *avctx)
static int gif_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt) static int gif_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
{ {
const uint8_t *buf = avpkt->data;
int buf_size = avpkt->size;
GifState *s = avctx->priv_data; GifState *s = avctx->priv_data;
AVFrame *picture = data; AVFrame *picture = data;
int ret; int ret;
bytestream2_init(&s->gb, avpkt->data, avpkt->size);
s->picture.pts = avpkt->pts; s->picture.pts = avpkt->pts;
s->picture.pkt_pts = avpkt->pts; s->picture.pkt_pts = avpkt->pts;
s->picture.pkt_dts = avpkt->dts; s->picture.pkt_dts = avpkt->dts;
s->picture.pkt_duration = avpkt->duration; s->picture.pkt_duration = avpkt->duration;
s->bytestream = buf; if (avpkt->size >= 6) {
s->bytestream_end = buf + buf_size; s->keyframe = memcmp(avpkt->data, gif87a_sig, 6) == 0 ||
memcmp(avpkt->data, gif89a_sig, 6) == 0;
if (buf_size >= 6) {
s->keyframe = memcmp(s->bytestream, gif87a_sig, 6) == 0 ||
memcmp(s->bytestream, gif89a_sig, 6) == 0;
} else { } else {
s->keyframe = 0; s->keyframe = 0;
} }
@ -511,7 +505,7 @@ static int gif_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, A
else if (*got_frame) else if (*got_frame)
*picture = s->picture; *picture = s->picture;
return s->bytestream - buf; return avpkt->size;
} }
static av_cold int gif_decode_close(AVCodecContext *avctx) static av_cold int gif_decode_close(AVCodecContext *avctx)

5
libavcodec/lzw.c
View File

@ -92,11 +92,6 @@ static int lzw_get_code(struct LZWState * s)
return c & s->curmask; return c & s->curmask;
} }
const uint8_t* ff_lzw_cur_ptr(LZWState *p)
{
return ((struct LZWState*)p)->pbuf;
}
void ff_lzw_decode_tail(LZWState *p) void ff_lzw_decode_tail(LZWState *p)
{ {
struct LZWState *s = (struct LZWState *)p; struct LZWState *s = (struct LZWState *)p;

1
libavcodec/lzw.h
View File

@ -47,7 +47,6 @@ void ff_lzw_decode_open(LZWState **p);
void ff_lzw_decode_close(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_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); int ff_lzw_decode(LZWState *s, uint8_t *buf, int len);
const uint8_t* ff_lzw_cur_ptr(LZWState *lzw);
void ff_lzw_decode_tail(LZWState *lzw); void ff_lzw_decode_tail(LZWState *lzw);
/** LZW encode state */ /** LZW encode state */