annotate ffmpeg/libavcodec/flicvideo.c @ 13:844d341cf643 tip

Back up before ISMIR
author Yading Song <yading.song@eecs.qmul.ac.uk>
date Thu, 31 Oct 2013 13:17:06 +0000
parents 6840f77b83aa
children
rev   line source
yading@10 1 /*
yading@10 2 * FLI/FLC Animation Video Decoder
yading@10 3 * Copyright (C) 2003, 2004 the ffmpeg project
yading@10 4 *
yading@10 5 * This file is part of FFmpeg.
yading@10 6 *
yading@10 7 * FFmpeg is free software; you can redistribute it and/or
yading@10 8 * modify it under the terms of the GNU Lesser General Public
yading@10 9 * License as published by the Free Software Foundation; either
yading@10 10 * version 2.1 of the License, or (at your option) any later version.
yading@10 11 *
yading@10 12 * FFmpeg is distributed in the hope that it will be useful,
yading@10 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
yading@10 15 * Lesser General Public License for more details.
yading@10 16 *
yading@10 17 * You should have received a copy of the GNU Lesser General Public
yading@10 18 * License along with FFmpeg; if not, write to the Free Software
yading@10 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10 20 */
yading@10 21
yading@10 22 /**
yading@10 23 * @file
yading@10 24 * Autodesk Animator FLI/FLC Video Decoder
yading@10 25 * by Mike Melanson (melanson@pcisys.net)
yading@10 26 * for more information on the .fli/.flc file format and all of its many
yading@10 27 * variations, visit:
yading@10 28 * http://www.compuphase.com/flic.htm
yading@10 29 *
yading@10 30 * This decoder outputs PAL8/RGB555/RGB565 and maybe one day RGB24
yading@10 31 * colorspace data, depending on the FLC. To use this decoder, be
yading@10 32 * sure that your demuxer sends the FLI file header to the decoder via
yading@10 33 * the extradata chunk in AVCodecContext. The chunk should be 128 bytes
yading@10 34 * large. The only exception is for FLI files from the game "Magic Carpet",
yading@10 35 * in which the header is only 12 bytes.
yading@10 36 */
yading@10 37
yading@10 38 #include <stdio.h>
yading@10 39 #include <stdlib.h>
yading@10 40 #include <string.h>
yading@10 41
yading@10 42 #include "libavutil/intreadwrite.h"
yading@10 43 #include "avcodec.h"
yading@10 44 #include "bytestream.h"
yading@10 45 #include "internal.h"
yading@10 46 #include "mathops.h"
yading@10 47
yading@10 48 #define FLI_256_COLOR 4
yading@10 49 #define FLI_DELTA 7
yading@10 50 #define FLI_COLOR 11
yading@10 51 #define FLI_LC 12
yading@10 52 #define FLI_BLACK 13
yading@10 53 #define FLI_BRUN 15
yading@10 54 #define FLI_COPY 16
yading@10 55 #define FLI_MINI 18
yading@10 56 #define FLI_DTA_BRUN 25
yading@10 57 #define FLI_DTA_COPY 26
yading@10 58 #define FLI_DTA_LC 27
yading@10 59
yading@10 60 #define FLI_TYPE_CODE (0xAF11)
yading@10 61 #define FLC_FLX_TYPE_CODE (0xAF12)
yading@10 62 #define FLC_DTA_TYPE_CODE (0xAF44) /* Marks an "Extended FLC" comes from Dave's Targa Animator (DTA) */
yading@10 63 #define FLC_MAGIC_CARPET_SYNTHETIC_TYPE_CODE (0xAF13)
yading@10 64
yading@10 65 #define CHECK_PIXEL_PTR(n) \
yading@10 66 if (pixel_ptr + n > pixel_limit) { \
yading@10 67 av_log (s->avctx, AV_LOG_ERROR, "Invalid pixel_ptr = %d > pixel_limit = %d\n", \
yading@10 68 pixel_ptr + n, pixel_limit); \
yading@10 69 return AVERROR_INVALIDDATA; \
yading@10 70 } \
yading@10 71
yading@10 72 typedef struct FlicDecodeContext {
yading@10 73 AVCodecContext *avctx;
yading@10 74 AVFrame frame;
yading@10 75
yading@10 76 unsigned int palette[256];
yading@10 77 int new_palette;
yading@10 78 int fli_type; /* either 0xAF11 or 0xAF12, affects palette resolution */
yading@10 79 } FlicDecodeContext;
yading@10 80
yading@10 81 static av_cold int flic_decode_init(AVCodecContext *avctx)
yading@10 82 {
yading@10 83 FlicDecodeContext *s = avctx->priv_data;
yading@10 84 unsigned char *fli_header = (unsigned char *)avctx->extradata;
yading@10 85 int depth;
yading@10 86
yading@10 87 if (avctx->extradata_size != 0 &&
yading@10 88 avctx->extradata_size != 12 &&
yading@10 89 avctx->extradata_size != 128 &&
yading@10 90 avctx->extradata_size != 256 &&
yading@10 91 avctx->extradata_size != 904 &&
yading@10 92 avctx->extradata_size != 1024) {
yading@10 93 av_log(avctx, AV_LOG_ERROR, "Unexpected extradata size %d\n", avctx->extradata_size);
yading@10 94 return AVERROR_INVALIDDATA;
yading@10 95 }
yading@10 96
yading@10 97 s->avctx = avctx;
yading@10 98
yading@10 99 if (s->avctx->extradata_size == 12) {
yading@10 100 /* special case for magic carpet FLIs */
yading@10 101 s->fli_type = FLC_MAGIC_CARPET_SYNTHETIC_TYPE_CODE;
yading@10 102 depth = 8;
yading@10 103 } else if (avctx->extradata_size == 1024) {
yading@10 104 uint8_t *ptr = avctx->extradata;
yading@10 105 int i;
yading@10 106
yading@10 107 for (i = 0; i < 256; i++) {
yading@10 108 s->palette[i] = AV_RL32(ptr);
yading@10 109 ptr += 4;
yading@10 110 }
yading@10 111 depth = 8;
yading@10 112 /* FLI in MOV, see e.g. FFmpeg trac issue #626 */
yading@10 113 } else if (avctx->extradata_size == 0 ||
yading@10 114 avctx->extradata_size == 256 ||
yading@10 115 /* see FFmpeg ticket #1234 */
yading@10 116 avctx->extradata_size == 904) {
yading@10 117 s->fli_type = FLI_TYPE_CODE;
yading@10 118 depth = 8;
yading@10 119 } else {
yading@10 120 s->fli_type = AV_RL16(&fli_header[4]);
yading@10 121 depth = AV_RL16(&fli_header[12]);
yading@10 122 }
yading@10 123
yading@10 124 if (depth == 0) {
yading@10 125 depth = 8; /* Some FLC generators set depth to zero, when they mean 8Bpp. Fix up here */
yading@10 126 }
yading@10 127
yading@10 128 if ((s->fli_type == FLC_FLX_TYPE_CODE) && (depth == 16)) {
yading@10 129 depth = 15; /* Original Autodesk FLX's say the depth is 16Bpp when it is really 15Bpp */
yading@10 130 }
yading@10 131
yading@10 132 switch (depth) {
yading@10 133 case 8 : avctx->pix_fmt = AV_PIX_FMT_PAL8; break;
yading@10 134 case 15 : avctx->pix_fmt = AV_PIX_FMT_RGB555; break;
yading@10 135 case 16 : avctx->pix_fmt = AV_PIX_FMT_RGB565; break;
yading@10 136 case 24 : avctx->pix_fmt = AV_PIX_FMT_BGR24; /* Supposedly BGR, but havent any files to test with */
yading@10 137 av_log(avctx, AV_LOG_ERROR, "24Bpp FLC/FLX is unsupported due to no test files.\n");
yading@10 138 return AVERROR_PATCHWELCOME;
yading@10 139 default :
yading@10 140 av_log(avctx, AV_LOG_ERROR, "Unknown FLC/FLX depth of %d Bpp is unsupported.\n",depth);
yading@10 141 return AVERROR_INVALIDDATA;
yading@10 142 }
yading@10 143
yading@10 144 avcodec_get_frame_defaults(&s->frame);
yading@10 145 s->new_palette = 0;
yading@10 146
yading@10 147 return 0;
yading@10 148 }
yading@10 149
yading@10 150 static int flic_decode_frame_8BPP(AVCodecContext *avctx,
yading@10 151 void *data, int *got_frame,
yading@10 152 const uint8_t *buf, int buf_size)
yading@10 153 {
yading@10 154 FlicDecodeContext *s = avctx->priv_data;
yading@10 155
yading@10 156 GetByteContext g2;
yading@10 157 int pixel_ptr;
yading@10 158 int palette_ptr;
yading@10 159 unsigned char palette_idx1;
yading@10 160 unsigned char palette_idx2;
yading@10 161
yading@10 162 unsigned int frame_size;
yading@10 163 int num_chunks;
yading@10 164
yading@10 165 unsigned int chunk_size;
yading@10 166 int chunk_type;
yading@10 167
yading@10 168 int i, j, ret;
yading@10 169
yading@10 170 int color_packets;
yading@10 171 int color_changes;
yading@10 172 int color_shift;
yading@10 173 unsigned char r, g, b;
yading@10 174
yading@10 175 int lines;
yading@10 176 int compressed_lines;
yading@10 177 int starting_line;
yading@10 178 signed short line_packets;
yading@10 179 int y_ptr;
yading@10 180 int byte_run;
yading@10 181 int pixel_skip;
yading@10 182 int pixel_countdown;
yading@10 183 unsigned char *pixels;
yading@10 184 unsigned int pixel_limit;
yading@10 185
yading@10 186 bytestream2_init(&g2, buf, buf_size);
yading@10 187
yading@10 188 if ((ret = ff_reget_buffer(avctx, &s->frame)) < 0)
yading@10 189 return ret;
yading@10 190
yading@10 191 pixels = s->frame.data[0];
yading@10 192 pixel_limit = s->avctx->height * s->frame.linesize[0];
yading@10 193 if (buf_size < 16 || buf_size > INT_MAX - (3 * 256 + FF_INPUT_BUFFER_PADDING_SIZE))
yading@10 194 return AVERROR_INVALIDDATA;
yading@10 195 frame_size = bytestream2_get_le32(&g2);
yading@10 196 if (frame_size > buf_size)
yading@10 197 frame_size = buf_size;
yading@10 198 bytestream2_skip(&g2, 2); /* skip the magic number */
yading@10 199 num_chunks = bytestream2_get_le16(&g2);
yading@10 200 bytestream2_skip(&g2, 8); /* skip padding */
yading@10 201
yading@10 202 frame_size -= 16;
yading@10 203
yading@10 204 /* iterate through the chunks */
yading@10 205 while ((frame_size >= 6) && (num_chunks > 0)) {
yading@10 206 int stream_ptr_after_chunk;
yading@10 207 chunk_size = bytestream2_get_le32(&g2);
yading@10 208 if (chunk_size > frame_size) {
yading@10 209 av_log(avctx, AV_LOG_WARNING,
yading@10 210 "Invalid chunk_size = %u > frame_size = %u\n", chunk_size, frame_size);
yading@10 211 chunk_size = frame_size;
yading@10 212 }
yading@10 213 stream_ptr_after_chunk = bytestream2_tell(&g2) - 4 + chunk_size;
yading@10 214
yading@10 215 chunk_type = bytestream2_get_le16(&g2);
yading@10 216
yading@10 217 switch (chunk_type) {
yading@10 218 case FLI_256_COLOR:
yading@10 219 case FLI_COLOR:
yading@10 220 /* check special case: If this file is from the Magic Carpet
yading@10 221 * game and uses 6-bit colors even though it reports 256-color
yading@10 222 * chunks in a 0xAF12-type file (fli_type is set to 0xAF13 during
yading@10 223 * initialization) */
yading@10 224 if ((chunk_type == FLI_256_COLOR) && (s->fli_type != FLC_MAGIC_CARPET_SYNTHETIC_TYPE_CODE))
yading@10 225 color_shift = 0;
yading@10 226 else
yading@10 227 color_shift = 2;
yading@10 228 /* set up the palette */
yading@10 229 color_packets = bytestream2_get_le16(&g2);
yading@10 230 palette_ptr = 0;
yading@10 231 for (i = 0; i < color_packets; i++) {
yading@10 232 /* first byte is how many colors to skip */
yading@10 233 palette_ptr += bytestream2_get_byte(&g2);
yading@10 234
yading@10 235 /* next byte indicates how many entries to change */
yading@10 236 color_changes = bytestream2_get_byte(&g2);
yading@10 237
yading@10 238 /* if there are 0 color changes, there are actually 256 */
yading@10 239 if (color_changes == 0)
yading@10 240 color_changes = 256;
yading@10 241
yading@10 242 if (bytestream2_tell(&g2) + color_changes * 3 > stream_ptr_after_chunk)
yading@10 243 break;
yading@10 244
yading@10 245 for (j = 0; j < color_changes; j++) {
yading@10 246 unsigned int entry;
yading@10 247
yading@10 248 /* wrap around, for good measure */
yading@10 249 if ((unsigned)palette_ptr >= 256)
yading@10 250 palette_ptr = 0;
yading@10 251
yading@10 252 r = bytestream2_get_byte(&g2) << color_shift;
yading@10 253 g = bytestream2_get_byte(&g2) << color_shift;
yading@10 254 b = bytestream2_get_byte(&g2) << color_shift;
yading@10 255 entry = 0xFFU << 24 | r << 16 | g << 8 | b;
yading@10 256 if (color_shift == 2)
yading@10 257 entry |= entry >> 6 & 0x30303;
yading@10 258 if (s->palette[palette_ptr] != entry)
yading@10 259 s->new_palette = 1;
yading@10 260 s->palette[palette_ptr++] = entry;
yading@10 261 }
yading@10 262 }
yading@10 263 break;
yading@10 264
yading@10 265 case FLI_DELTA:
yading@10 266 y_ptr = 0;
yading@10 267 compressed_lines = bytestream2_get_le16(&g2);
yading@10 268 while (compressed_lines > 0) {
yading@10 269 if (bytestream2_tell(&g2) + 2 > stream_ptr_after_chunk)
yading@10 270 break;
yading@10 271 line_packets = bytestream2_get_le16(&g2);
yading@10 272 if ((line_packets & 0xC000) == 0xC000) {
yading@10 273 // line skip opcode
yading@10 274 line_packets = -line_packets;
yading@10 275 y_ptr += line_packets * s->frame.linesize[0];
yading@10 276 } else if ((line_packets & 0xC000) == 0x4000) {
yading@10 277 av_log(avctx, AV_LOG_ERROR, "Undefined opcode (%x) in DELTA_FLI\n", line_packets);
yading@10 278 } else if ((line_packets & 0xC000) == 0x8000) {
yading@10 279 // "last byte" opcode
yading@10 280 pixel_ptr= y_ptr + s->frame.linesize[0] - 1;
yading@10 281 CHECK_PIXEL_PTR(0);
yading@10 282 pixels[pixel_ptr] = line_packets & 0xff;
yading@10 283 } else {
yading@10 284 compressed_lines--;
yading@10 285 pixel_ptr = y_ptr;
yading@10 286 CHECK_PIXEL_PTR(0);
yading@10 287 pixel_countdown = s->avctx->width;
yading@10 288 for (i = 0; i < line_packets; i++) {
yading@10 289 if (bytestream2_tell(&g2) + 2 > stream_ptr_after_chunk)
yading@10 290 break;
yading@10 291 /* account for the skip bytes */
yading@10 292 pixel_skip = bytestream2_get_byte(&g2);
yading@10 293 pixel_ptr += pixel_skip;
yading@10 294 pixel_countdown -= pixel_skip;
yading@10 295 byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10 296 if (byte_run < 0) {
yading@10 297 byte_run = -byte_run;
yading@10 298 palette_idx1 = bytestream2_get_byte(&g2);
yading@10 299 palette_idx2 = bytestream2_get_byte(&g2);
yading@10 300 CHECK_PIXEL_PTR(byte_run * 2);
yading@10 301 for (j = 0; j < byte_run; j++, pixel_countdown -= 2) {
yading@10 302 pixels[pixel_ptr++] = palette_idx1;
yading@10 303 pixels[pixel_ptr++] = palette_idx2;
yading@10 304 }
yading@10 305 } else {
yading@10 306 CHECK_PIXEL_PTR(byte_run * 2);
yading@10 307 if (bytestream2_tell(&g2) + byte_run * 2 > stream_ptr_after_chunk)
yading@10 308 break;
yading@10 309 for (j = 0; j < byte_run * 2; j++, pixel_countdown--) {
yading@10 310 pixels[pixel_ptr++] = bytestream2_get_byte(&g2);
yading@10 311 }
yading@10 312 }
yading@10 313 }
yading@10 314
yading@10 315 y_ptr += s->frame.linesize[0];
yading@10 316 }
yading@10 317 }
yading@10 318 break;
yading@10 319
yading@10 320 case FLI_LC:
yading@10 321 /* line compressed */
yading@10 322 starting_line = bytestream2_get_le16(&g2);
yading@10 323 y_ptr = 0;
yading@10 324 y_ptr += starting_line * s->frame.linesize[0];
yading@10 325
yading@10 326 compressed_lines = bytestream2_get_le16(&g2);
yading@10 327 while (compressed_lines > 0) {
yading@10 328 pixel_ptr = y_ptr;
yading@10 329 CHECK_PIXEL_PTR(0);
yading@10 330 pixel_countdown = s->avctx->width;
yading@10 331 if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10 332 break;
yading@10 333 line_packets = bytestream2_get_byte(&g2);
yading@10 334 if (line_packets > 0) {
yading@10 335 for (i = 0; i < line_packets; i++) {
yading@10 336 /* account for the skip bytes */
yading@10 337 if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10 338 break;
yading@10 339 pixel_skip = bytestream2_get_byte(&g2);
yading@10 340 pixel_ptr += pixel_skip;
yading@10 341 pixel_countdown -= pixel_skip;
yading@10 342 byte_run = sign_extend(bytestream2_get_byte(&g2),8);
yading@10 343 if (byte_run > 0) {
yading@10 344 CHECK_PIXEL_PTR(byte_run);
yading@10 345 if (bytestream2_tell(&g2) + byte_run > stream_ptr_after_chunk)
yading@10 346 break;
yading@10 347 for (j = 0; j < byte_run; j++, pixel_countdown--) {
yading@10 348 pixels[pixel_ptr++] = bytestream2_get_byte(&g2);
yading@10 349 }
yading@10 350 } else if (byte_run < 0) {
yading@10 351 byte_run = -byte_run;
yading@10 352 palette_idx1 = bytestream2_get_byte(&g2);
yading@10 353 CHECK_PIXEL_PTR(byte_run);
yading@10 354 for (j = 0; j < byte_run; j++, pixel_countdown--) {
yading@10 355 pixels[pixel_ptr++] = palette_idx1;
yading@10 356 }
yading@10 357 }
yading@10 358 }
yading@10 359 }
yading@10 360
yading@10 361 y_ptr += s->frame.linesize[0];
yading@10 362 compressed_lines--;
yading@10 363 }
yading@10 364 break;
yading@10 365
yading@10 366 case FLI_BLACK:
yading@10 367 /* set the whole frame to color 0 (which is usually black) */
yading@10 368 memset(pixels, 0,
yading@10 369 s->frame.linesize[0] * s->avctx->height);
yading@10 370 break;
yading@10 371
yading@10 372 case FLI_BRUN:
yading@10 373 /* Byte run compression: This chunk type only occurs in the first
yading@10 374 * FLI frame and it will update the entire frame. */
yading@10 375 y_ptr = 0;
yading@10 376 for (lines = 0; lines < s->avctx->height; lines++) {
yading@10 377 pixel_ptr = y_ptr;
yading@10 378 /* disregard the line packets; instead, iterate through all
yading@10 379 * pixels on a row */
yading@10 380 bytestream2_skip(&g2, 1);
yading@10 381 pixel_countdown = s->avctx->width;
yading@10 382 while (pixel_countdown > 0) {
yading@10 383 if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10 384 break;
yading@10 385 byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10 386 if (!byte_run) {
yading@10 387 av_log(avctx, AV_LOG_ERROR, "Invalid byte run value.\n");
yading@10 388 return AVERROR_INVALIDDATA;
yading@10 389 }
yading@10 390
yading@10 391 if (byte_run > 0) {
yading@10 392 palette_idx1 = bytestream2_get_byte(&g2);
yading@10 393 CHECK_PIXEL_PTR(byte_run);
yading@10 394 for (j = 0; j < byte_run; j++) {
yading@10 395 pixels[pixel_ptr++] = palette_idx1;
yading@10 396 pixel_countdown--;
yading@10 397 if (pixel_countdown < 0)
yading@10 398 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d) at line %d\n",
yading@10 399 pixel_countdown, lines);
yading@10 400 }
yading@10 401 } else { /* copy bytes if byte_run < 0 */
yading@10 402 byte_run = -byte_run;
yading@10 403 CHECK_PIXEL_PTR(byte_run);
yading@10 404 if (bytestream2_tell(&g2) + byte_run > stream_ptr_after_chunk)
yading@10 405 break;
yading@10 406 for (j = 0; j < byte_run; j++) {
yading@10 407 pixels[pixel_ptr++] = bytestream2_get_byte(&g2);
yading@10 408 pixel_countdown--;
yading@10 409 if (pixel_countdown < 0)
yading@10 410 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d) at line %d\n",
yading@10 411 pixel_countdown, lines);
yading@10 412 }
yading@10 413 }
yading@10 414 }
yading@10 415
yading@10 416 y_ptr += s->frame.linesize[0];
yading@10 417 }
yading@10 418 break;
yading@10 419
yading@10 420 case FLI_COPY:
yading@10 421 /* copy the chunk (uncompressed frame) */
yading@10 422 if (chunk_size - 6 != s->avctx->width * s->avctx->height) {
yading@10 423 av_log(avctx, AV_LOG_ERROR, "In chunk FLI_COPY : source data (%d bytes) " \
yading@10 424 "has incorrect size, skipping chunk\n", chunk_size - 6);
yading@10 425 bytestream2_skip(&g2, chunk_size - 6);
yading@10 426 } else {
yading@10 427 for (y_ptr = 0; y_ptr < s->frame.linesize[0] * s->avctx->height;
yading@10 428 y_ptr += s->frame.linesize[0]) {
yading@10 429 bytestream2_get_buffer(&g2, &pixels[y_ptr],
yading@10 430 s->avctx->width);
yading@10 431 }
yading@10 432 }
yading@10 433 break;
yading@10 434
yading@10 435 case FLI_MINI:
yading@10 436 /* some sort of a thumbnail? disregard this chunk... */
yading@10 437 break;
yading@10 438
yading@10 439 default:
yading@10 440 av_log(avctx, AV_LOG_ERROR, "Unrecognized chunk type: %d\n", chunk_type);
yading@10 441 break;
yading@10 442 }
yading@10 443
yading@10 444 if (stream_ptr_after_chunk - bytestream2_tell(&g2) > 0)
yading@10 445 bytestream2_skip(&g2, stream_ptr_after_chunk - bytestream2_tell(&g2));
yading@10 446
yading@10 447 frame_size -= chunk_size;
yading@10 448 num_chunks--;
yading@10 449 }
yading@10 450
yading@10 451 /* by the end of the chunk, the stream ptr should equal the frame
yading@10 452 * size (minus 1 or 2, possibly); if it doesn't, issue a warning */
yading@10 453 if (bytestream2_get_bytes_left(&g2) > 2)
yading@10 454 av_log(avctx, AV_LOG_ERROR, "Processed FLI chunk where chunk size = %d " \
yading@10 455 "and final chunk ptr = %d\n", buf_size,
yading@10 456 buf_size - bytestream2_get_bytes_left(&g2));
yading@10 457
yading@10 458 /* make the palette available on the way out */
yading@10 459 memcpy(s->frame.data[1], s->palette, AVPALETTE_SIZE);
yading@10 460 if (s->new_palette) {
yading@10 461 s->frame.palette_has_changed = 1;
yading@10 462 s->new_palette = 0;
yading@10 463 }
yading@10 464
yading@10 465 if ((ret = av_frame_ref(data, &s->frame)) < 0)
yading@10 466 return ret;
yading@10 467
yading@10 468 *got_frame = 1;
yading@10 469
yading@10 470 return buf_size;
yading@10 471 }
yading@10 472
yading@10 473 static int flic_decode_frame_15_16BPP(AVCodecContext *avctx,
yading@10 474 void *data, int *got_frame,
yading@10 475 const uint8_t *buf, int buf_size)
yading@10 476 {
yading@10 477 /* Note, the only difference between the 15Bpp and 16Bpp */
yading@10 478 /* Format is the pixel format, the packets are processed the same. */
yading@10 479 FlicDecodeContext *s = avctx->priv_data;
yading@10 480
yading@10 481 GetByteContext g2;
yading@10 482 int pixel_ptr;
yading@10 483 unsigned char palette_idx1;
yading@10 484
yading@10 485 unsigned int frame_size;
yading@10 486 int num_chunks;
yading@10 487
yading@10 488 unsigned int chunk_size;
yading@10 489 int chunk_type;
yading@10 490
yading@10 491 int i, j, ret;
yading@10 492
yading@10 493 int lines;
yading@10 494 int compressed_lines;
yading@10 495 signed short line_packets;
yading@10 496 int y_ptr;
yading@10 497 int byte_run;
yading@10 498 int pixel_skip;
yading@10 499 int pixel_countdown;
yading@10 500 unsigned char *pixels;
yading@10 501 int pixel;
yading@10 502 unsigned int pixel_limit;
yading@10 503
yading@10 504 bytestream2_init(&g2, buf, buf_size);
yading@10 505
yading@10 506 if ((ret = ff_reget_buffer(avctx, &s->frame)) < 0)
yading@10 507 return ret;
yading@10 508
yading@10 509 pixels = s->frame.data[0];
yading@10 510 pixel_limit = s->avctx->height * s->frame.linesize[0];
yading@10 511
yading@10 512 frame_size = bytestream2_get_le32(&g2);
yading@10 513 bytestream2_skip(&g2, 2); /* skip the magic number */
yading@10 514 num_chunks = bytestream2_get_le16(&g2);
yading@10 515 bytestream2_skip(&g2, 8); /* skip padding */
yading@10 516 if (frame_size > buf_size)
yading@10 517 frame_size = buf_size;
yading@10 518
yading@10 519 frame_size -= 16;
yading@10 520
yading@10 521 /* iterate through the chunks */
yading@10 522 while ((frame_size > 0) && (num_chunks > 0)) {
yading@10 523 int stream_ptr_after_chunk;
yading@10 524 chunk_size = bytestream2_get_le32(&g2);
yading@10 525 if (chunk_size > frame_size) {
yading@10 526 av_log(avctx, AV_LOG_WARNING,
yading@10 527 "Invalid chunk_size = %u > frame_size = %u\n", chunk_size, frame_size);
yading@10 528 chunk_size = frame_size;
yading@10 529 }
yading@10 530 stream_ptr_after_chunk = bytestream2_tell(&g2) - 4 + chunk_size;
yading@10 531
yading@10 532 chunk_type = bytestream2_get_le16(&g2);
yading@10 533
yading@10 534
yading@10 535 switch (chunk_type) {
yading@10 536 case FLI_256_COLOR:
yading@10 537 case FLI_COLOR:
yading@10 538 /* For some reason, it seems that non-palettized flics do
yading@10 539 * include one of these chunks in their first frame.
yading@10 540 * Why I do not know, it seems rather extraneous. */
yading@10 541 av_dlog(avctx,
yading@10 542 "Unexpected Palette chunk %d in non-palettized FLC\n",
yading@10 543 chunk_type);
yading@10 544 bytestream2_skip(&g2, chunk_size - 6);
yading@10 545 break;
yading@10 546
yading@10 547 case FLI_DELTA:
yading@10 548 case FLI_DTA_LC:
yading@10 549 y_ptr = 0;
yading@10 550 compressed_lines = bytestream2_get_le16(&g2);
yading@10 551 while (compressed_lines > 0) {
yading@10 552 if (bytestream2_tell(&g2) + 2 > stream_ptr_after_chunk)
yading@10 553 break;
yading@10 554 line_packets = bytestream2_get_le16(&g2);
yading@10 555 if (line_packets < 0) {
yading@10 556 line_packets = -line_packets;
yading@10 557 y_ptr += line_packets * s->frame.linesize[0];
yading@10 558 } else {
yading@10 559 compressed_lines--;
yading@10 560 pixel_ptr = y_ptr;
yading@10 561 CHECK_PIXEL_PTR(0);
yading@10 562 pixel_countdown = s->avctx->width;
yading@10 563 for (i = 0; i < line_packets; i++) {
yading@10 564 /* account for the skip bytes */
yading@10 565 if (bytestream2_tell(&g2) + 2 > stream_ptr_after_chunk)
yading@10 566 break;
yading@10 567 pixel_skip = bytestream2_get_byte(&g2);
yading@10 568 pixel_ptr += (pixel_skip*2); /* Pixel is 2 bytes wide */
yading@10 569 pixel_countdown -= pixel_skip;
yading@10 570 byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10 571 if (byte_run < 0) {
yading@10 572 byte_run = -byte_run;
yading@10 573 pixel = bytestream2_get_le16(&g2);
yading@10 574 CHECK_PIXEL_PTR(2 * byte_run);
yading@10 575 for (j = 0; j < byte_run; j++, pixel_countdown -= 2) {
yading@10 576 *((signed short*)(&pixels[pixel_ptr])) = pixel;
yading@10 577 pixel_ptr += 2;
yading@10 578 }
yading@10 579 } else {
yading@10 580 if (bytestream2_tell(&g2) + 2*byte_run > stream_ptr_after_chunk)
yading@10 581 break;
yading@10 582 CHECK_PIXEL_PTR(2 * byte_run);
yading@10 583 for (j = 0; j < byte_run; j++, pixel_countdown--) {
yading@10 584 *((signed short*)(&pixels[pixel_ptr])) = bytestream2_get_le16(&g2);
yading@10 585 pixel_ptr += 2;
yading@10 586 }
yading@10 587 }
yading@10 588 }
yading@10 589
yading@10 590 y_ptr += s->frame.linesize[0];
yading@10 591 }
yading@10 592 }
yading@10 593 break;
yading@10 594
yading@10 595 case FLI_LC:
yading@10 596 av_log(avctx, AV_LOG_ERROR, "Unexpected FLI_LC chunk in non-paletised FLC\n");
yading@10 597 bytestream2_skip(&g2, chunk_size - 6);
yading@10 598 break;
yading@10 599
yading@10 600 case FLI_BLACK:
yading@10 601 /* set the whole frame to 0x0000 which is black in both 15Bpp and 16Bpp modes. */
yading@10 602 memset(pixels, 0x0000,
yading@10 603 s->frame.linesize[0] * s->avctx->height);
yading@10 604 break;
yading@10 605
yading@10 606 case FLI_BRUN:
yading@10 607 y_ptr = 0;
yading@10 608 for (lines = 0; lines < s->avctx->height; lines++) {
yading@10 609 pixel_ptr = y_ptr;
yading@10 610 /* disregard the line packets; instead, iterate through all
yading@10 611 * pixels on a row */
yading@10 612 bytestream2_skip(&g2, 1);
yading@10 613 pixel_countdown = (s->avctx->width * 2);
yading@10 614
yading@10 615 while (pixel_countdown > 0) {
yading@10 616 if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10 617 break;
yading@10 618 byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10 619 if (byte_run > 0) {
yading@10 620 palette_idx1 = bytestream2_get_byte(&g2);
yading@10 621 CHECK_PIXEL_PTR(byte_run);
yading@10 622 for (j = 0; j < byte_run; j++) {
yading@10 623 pixels[pixel_ptr++] = palette_idx1;
yading@10 624 pixel_countdown--;
yading@10 625 if (pixel_countdown < 0)
yading@10 626 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d) (linea%d)\n",
yading@10 627 pixel_countdown, lines);
yading@10 628 }
yading@10 629 } else { /* copy bytes if byte_run < 0 */
yading@10 630 byte_run = -byte_run;
yading@10 631 if (bytestream2_tell(&g2) + byte_run > stream_ptr_after_chunk)
yading@10 632 break;
yading@10 633 CHECK_PIXEL_PTR(byte_run);
yading@10 634 for (j = 0; j < byte_run; j++) {
yading@10 635 palette_idx1 = bytestream2_get_byte(&g2);
yading@10 636 pixels[pixel_ptr++] = palette_idx1;
yading@10 637 pixel_countdown--;
yading@10 638 if (pixel_countdown < 0)
yading@10 639 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d) at line %d\n",
yading@10 640 pixel_countdown, lines);
yading@10 641 }
yading@10 642 }
yading@10 643 }
yading@10 644
yading@10 645 /* Now FLX is strange, in that it is "byte" as opposed to "pixel" run length compressed.
yading@10 646 * This does not give us any good opportunity to perform word endian conversion
yading@10 647 * during decompression. So if it is required (i.e., this is not a LE target, we do
yading@10 648 * a second pass over the line here, swapping the bytes.
yading@10 649 */
yading@10 650 #if HAVE_BIGENDIAN
yading@10 651 pixel_ptr = y_ptr;
yading@10 652 pixel_countdown = s->avctx->width;
yading@10 653 while (pixel_countdown > 0) {
yading@10 654 *((signed short*)(&pixels[pixel_ptr])) = AV_RL16(&buf[pixel_ptr]);
yading@10 655 pixel_ptr += 2;
yading@10 656 }
yading@10 657 #endif
yading@10 658 y_ptr += s->frame.linesize[0];
yading@10 659 }
yading@10 660 break;
yading@10 661
yading@10 662 case FLI_DTA_BRUN:
yading@10 663 y_ptr = 0;
yading@10 664 for (lines = 0; lines < s->avctx->height; lines++) {
yading@10 665 pixel_ptr = y_ptr;
yading@10 666 /* disregard the line packets; instead, iterate through all
yading@10 667 * pixels on a row */
yading@10 668 bytestream2_skip(&g2, 1);
yading@10 669 pixel_countdown = s->avctx->width; /* Width is in pixels, not bytes */
yading@10 670
yading@10 671 while (pixel_countdown > 0) {
yading@10 672 if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10 673 break;
yading@10 674 byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10 675 if (byte_run > 0) {
yading@10 676 pixel = bytestream2_get_le16(&g2);
yading@10 677 CHECK_PIXEL_PTR(2 * byte_run);
yading@10 678 for (j = 0; j < byte_run; j++) {
yading@10 679 *((signed short*)(&pixels[pixel_ptr])) = pixel;
yading@10 680 pixel_ptr += 2;
yading@10 681 pixel_countdown--;
yading@10 682 if (pixel_countdown < 0)
yading@10 683 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d)\n",
yading@10 684 pixel_countdown);
yading@10 685 }
yading@10 686 } else { /* copy pixels if byte_run < 0 */
yading@10 687 byte_run = -byte_run;
yading@10 688 if (bytestream2_tell(&g2) + 2 * byte_run > stream_ptr_after_chunk)
yading@10 689 break;
yading@10 690 CHECK_PIXEL_PTR(2 * byte_run);
yading@10 691 for (j = 0; j < byte_run; j++) {
yading@10 692 *((signed short*)(&pixels[pixel_ptr])) = bytestream2_get_le16(&g2);
yading@10 693 pixel_ptr += 2;
yading@10 694 pixel_countdown--;
yading@10 695 if (pixel_countdown < 0)
yading@10 696 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d)\n",
yading@10 697 pixel_countdown);
yading@10 698 }
yading@10 699 }
yading@10 700 }
yading@10 701
yading@10 702 y_ptr += s->frame.linesize[0];
yading@10 703 }
yading@10 704 break;
yading@10 705
yading@10 706 case FLI_COPY:
yading@10 707 case FLI_DTA_COPY:
yading@10 708 /* copy the chunk (uncompressed frame) */
yading@10 709 if (chunk_size - 6 > (unsigned int)(s->avctx->width * s->avctx->height)*2) {
yading@10 710 av_log(avctx, AV_LOG_ERROR, "In chunk FLI_COPY : source data (%d bytes) " \
yading@10 711 "bigger than image, skipping chunk\n", chunk_size - 6);
yading@10 712 bytestream2_skip(&g2, chunk_size - 6);
yading@10 713 } else {
yading@10 714
yading@10 715 for (y_ptr = 0; y_ptr < s->frame.linesize[0] * s->avctx->height;
yading@10 716 y_ptr += s->frame.linesize[0]) {
yading@10 717
yading@10 718 pixel_countdown = s->avctx->width;
yading@10 719 pixel_ptr = 0;
yading@10 720 while (pixel_countdown > 0) {
yading@10 721 *((signed short*)(&pixels[y_ptr + pixel_ptr])) = bytestream2_get_le16(&g2);
yading@10 722 pixel_ptr += 2;
yading@10 723 pixel_countdown--;
yading@10 724 }
yading@10 725 }
yading@10 726 }
yading@10 727 break;
yading@10 728
yading@10 729 case FLI_MINI:
yading@10 730 /* some sort of a thumbnail? disregard this chunk... */
yading@10 731 bytestream2_skip(&g2, chunk_size - 6);
yading@10 732 break;
yading@10 733
yading@10 734 default:
yading@10 735 av_log(avctx, AV_LOG_ERROR, "Unrecognized chunk type: %d\n", chunk_type);
yading@10 736 break;
yading@10 737 }
yading@10 738
yading@10 739 frame_size -= chunk_size;
yading@10 740 num_chunks--;
yading@10 741 }
yading@10 742
yading@10 743 /* by the end of the chunk, the stream ptr should equal the frame
yading@10 744 * size (minus 1, possibly); if it doesn't, issue a warning */
yading@10 745 if ((bytestream2_get_bytes_left(&g2) != 0) && (bytestream2_get_bytes_left(&g2) != 1))
yading@10 746 av_log(avctx, AV_LOG_ERROR, "Processed FLI chunk where chunk size = %d " \
yading@10 747 "and final chunk ptr = %d\n", buf_size, bytestream2_tell(&g2));
yading@10 748
yading@10 749 if ((ret = av_frame_ref(data, &s->frame)) < 0)
yading@10 750 return ret;
yading@10 751
yading@10 752 *got_frame = 1;
yading@10 753
yading@10 754 return buf_size;
yading@10 755 }
yading@10 756
yading@10 757 static int flic_decode_frame_24BPP(AVCodecContext *avctx,
yading@10 758 void *data, int *got_frame,
yading@10 759 const uint8_t *buf, int buf_size)
yading@10 760 {
yading@10 761 av_log(avctx, AV_LOG_ERROR, "24Bpp FLC Unsupported due to lack of test files.\n");
yading@10 762 return AVERROR_PATCHWELCOME;
yading@10 763 }
yading@10 764
yading@10 765 static int flic_decode_frame(AVCodecContext *avctx,
yading@10 766 void *data, int *got_frame,
yading@10 767 AVPacket *avpkt)
yading@10 768 {
yading@10 769 const uint8_t *buf = avpkt->data;
yading@10 770 int buf_size = avpkt->size;
yading@10 771 if (avctx->pix_fmt == AV_PIX_FMT_PAL8) {
yading@10 772 return flic_decode_frame_8BPP(avctx, data, got_frame,
yading@10 773 buf, buf_size);
yading@10 774 }
yading@10 775 else if ((avctx->pix_fmt == AV_PIX_FMT_RGB555) ||
yading@10 776 (avctx->pix_fmt == AV_PIX_FMT_RGB565)) {
yading@10 777 return flic_decode_frame_15_16BPP(avctx, data, got_frame,
yading@10 778 buf, buf_size);
yading@10 779 }
yading@10 780 else if (avctx->pix_fmt == AV_PIX_FMT_BGR24) {
yading@10 781 return flic_decode_frame_24BPP(avctx, data, got_frame,
yading@10 782 buf, buf_size);
yading@10 783 }
yading@10 784
yading@10 785 /* Should not get here, ever as the pix_fmt is processed */
yading@10 786 /* in flic_decode_init and the above if should deal with */
yading@10 787 /* the finite set of possibilites allowable by here. */
yading@10 788 /* But in case we do, just error out. */
yading@10 789 av_log(avctx, AV_LOG_ERROR, "Unknown FLC format, my science cannot explain how this happened.\n");
yading@10 790 return AVERROR_BUG;
yading@10 791 }
yading@10 792
yading@10 793
yading@10 794 static av_cold int flic_decode_end(AVCodecContext *avctx)
yading@10 795 {
yading@10 796 FlicDecodeContext *s = avctx->priv_data;
yading@10 797
yading@10 798 av_frame_unref(&s->frame);
yading@10 799
yading@10 800 return 0;
yading@10 801 }
yading@10 802
yading@10 803 AVCodec ff_flic_decoder = {
yading@10 804 .name = "flic",
yading@10 805 .type = AVMEDIA_TYPE_VIDEO,
yading@10 806 .id = AV_CODEC_ID_FLIC,
yading@10 807 .priv_data_size = sizeof(FlicDecodeContext),
yading@10 808 .init = flic_decode_init,
yading@10 809 .close = flic_decode_end,
yading@10 810 .decode = flic_decode_frame,
yading@10 811 .capabilities = CODEC_CAP_DR1,
yading@10 812 .long_name = NULL_IF_CONFIG_SMALL("Autodesk Animator Flic video"),
yading@10 813 };