yading@10: /*
yading@10:  * FLI/FLC Animation Video Decoder
yading@10:  * Copyright (C) 2003, 2004 the ffmpeg project
yading@10:  *
yading@10:  * This file is part of FFmpeg.
yading@10:  *
yading@10:  * FFmpeg is free software; you can redistribute it and/or
yading@10:  * modify it under the terms of the GNU Lesser General Public
yading@10:  * License as published by the Free Software Foundation; either
yading@10:  * version 2.1 of the License, or (at your option) any later version.
yading@10:  *
yading@10:  * FFmpeg is distributed in the hope that it will be useful,
yading@10:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
yading@10:  * Lesser General Public License for more details.
yading@10:  *
yading@10:  * You should have received a copy of the GNU Lesser General Public
yading@10:  * License along with FFmpeg; if not, write to the Free Software
yading@10:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10:  */
yading@10: 
yading@10: /**
yading@10:  * @file
yading@10:  * Autodesk Animator FLI/FLC Video Decoder
yading@10:  * by Mike Melanson (melanson@pcisys.net)
yading@10:  * for more information on the .fli/.flc file format and all of its many
yading@10:  * variations, visit:
yading@10:  *   http://www.compuphase.com/flic.htm
yading@10:  *
yading@10:  * This decoder outputs PAL8/RGB555/RGB565 and maybe one day RGB24
yading@10:  * colorspace data, depending on the FLC. To use this decoder, be
yading@10:  * sure that your demuxer sends the FLI file header to the decoder via
yading@10:  * the extradata chunk in AVCodecContext. The chunk should be 128 bytes
yading@10:  * large. The only exception is for FLI files from the game "Magic Carpet",
yading@10:  * in which the header is only 12 bytes.
yading@10:  */
yading@10: 
yading@10: #include <stdio.h>
yading@10: #include <stdlib.h>
yading@10: #include <string.h>
yading@10: 
yading@10: #include "libavutil/intreadwrite.h"
yading@10: #include "avcodec.h"
yading@10: #include "bytestream.h"
yading@10: #include "internal.h"
yading@10: #include "mathops.h"
yading@10: 
yading@10: #define FLI_256_COLOR 4
yading@10: #define FLI_DELTA     7
yading@10: #define FLI_COLOR     11
yading@10: #define FLI_LC        12
yading@10: #define FLI_BLACK     13
yading@10: #define FLI_BRUN      15
yading@10: #define FLI_COPY      16
yading@10: #define FLI_MINI      18
yading@10: #define FLI_DTA_BRUN  25
yading@10: #define FLI_DTA_COPY  26
yading@10: #define FLI_DTA_LC    27
yading@10: 
yading@10: #define FLI_TYPE_CODE     (0xAF11)
yading@10: #define FLC_FLX_TYPE_CODE (0xAF12)
yading@10: #define FLC_DTA_TYPE_CODE (0xAF44) /* Marks an "Extended FLC" comes from Dave's Targa Animator (DTA) */
yading@10: #define FLC_MAGIC_CARPET_SYNTHETIC_TYPE_CODE (0xAF13)
yading@10: 
yading@10: #define CHECK_PIXEL_PTR(n) \
yading@10:     if (pixel_ptr + n > pixel_limit) { \
yading@10:         av_log (s->avctx, AV_LOG_ERROR, "Invalid pixel_ptr = %d > pixel_limit = %d\n", \
yading@10:         pixel_ptr + n, pixel_limit); \
yading@10:         return AVERROR_INVALIDDATA; \
yading@10:     } \
yading@10: 
yading@10: typedef struct FlicDecodeContext {
yading@10:     AVCodecContext *avctx;
yading@10:     AVFrame frame;
yading@10: 
yading@10:     unsigned int palette[256];
yading@10:     int new_palette;
yading@10:     int fli_type;  /* either 0xAF11 or 0xAF12, affects palette resolution */
yading@10: } FlicDecodeContext;
yading@10: 
yading@10: static av_cold int flic_decode_init(AVCodecContext *avctx)
yading@10: {
yading@10:     FlicDecodeContext *s = avctx->priv_data;
yading@10:     unsigned char *fli_header = (unsigned char *)avctx->extradata;
yading@10:     int depth;
yading@10: 
yading@10:     if (avctx->extradata_size != 0 &&
yading@10:         avctx->extradata_size != 12 &&
yading@10:         avctx->extradata_size != 128 &&
yading@10:         avctx->extradata_size != 256 &&
yading@10:         avctx->extradata_size != 904 &&
yading@10:         avctx->extradata_size != 1024) {
yading@10:         av_log(avctx, AV_LOG_ERROR, "Unexpected extradata size %d\n", avctx->extradata_size);
yading@10:         return AVERROR_INVALIDDATA;
yading@10:     }
yading@10: 
yading@10:     s->avctx = avctx;
yading@10: 
yading@10:     if (s->avctx->extradata_size == 12) {
yading@10:         /* special case for magic carpet FLIs */
yading@10:         s->fli_type = FLC_MAGIC_CARPET_SYNTHETIC_TYPE_CODE;
yading@10:         depth = 8;
yading@10:     } else if (avctx->extradata_size == 1024) {
yading@10:         uint8_t *ptr = avctx->extradata;
yading@10:         int i;
yading@10: 
yading@10:         for (i = 0; i < 256; i++) {
yading@10:             s->palette[i] = AV_RL32(ptr);
yading@10:             ptr += 4;
yading@10:         }
yading@10:         depth = 8;
yading@10:         /* FLI in MOV, see e.g. FFmpeg trac issue #626 */
yading@10:     } else if (avctx->extradata_size == 0 ||
yading@10:                avctx->extradata_size == 256 ||
yading@10:         /* see FFmpeg ticket #1234 */
yading@10:                avctx->extradata_size == 904) {
yading@10:         s->fli_type = FLI_TYPE_CODE;
yading@10:         depth = 8;
yading@10:     } else {
yading@10:         s->fli_type = AV_RL16(&fli_header[4]);
yading@10:         depth = AV_RL16(&fli_header[12]);
yading@10:     }
yading@10: 
yading@10:     if (depth == 0) {
yading@10:         depth = 8; /* Some FLC generators set depth to zero, when they mean 8Bpp. Fix up here */
yading@10:     }
yading@10: 
yading@10:     if ((s->fli_type == FLC_FLX_TYPE_CODE) && (depth == 16)) {
yading@10:         depth = 15; /* Original Autodesk FLX's say the depth is 16Bpp when it is really 15Bpp */
yading@10:     }
yading@10: 
yading@10:     switch (depth) {
yading@10:         case 8  : avctx->pix_fmt = AV_PIX_FMT_PAL8; break;
yading@10:         case 15 : avctx->pix_fmt = AV_PIX_FMT_RGB555; break;
yading@10:         case 16 : avctx->pix_fmt = AV_PIX_FMT_RGB565; break;
yading@10:         case 24 : avctx->pix_fmt = AV_PIX_FMT_BGR24; /* Supposedly BGR, but havent any files to test with */
yading@10:                   av_log(avctx, AV_LOG_ERROR, "24Bpp FLC/FLX is unsupported due to no test files.\n");
yading@10:                   return AVERROR_PATCHWELCOME;
yading@10:         default :
yading@10:                   av_log(avctx, AV_LOG_ERROR, "Unknown FLC/FLX depth of %d Bpp is unsupported.\n",depth);
yading@10:                   return AVERROR_INVALIDDATA;
yading@10:     }
yading@10: 
yading@10:     avcodec_get_frame_defaults(&s->frame);
yading@10:     s->new_palette = 0;
yading@10: 
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int flic_decode_frame_8BPP(AVCodecContext *avctx,
yading@10:                                   void *data, int *got_frame,
yading@10:                                   const uint8_t *buf, int buf_size)
yading@10: {
yading@10:     FlicDecodeContext *s = avctx->priv_data;
yading@10: 
yading@10:     GetByteContext g2;
yading@10:     int pixel_ptr;
yading@10:     int palette_ptr;
yading@10:     unsigned char palette_idx1;
yading@10:     unsigned char palette_idx2;
yading@10: 
yading@10:     unsigned int frame_size;
yading@10:     int num_chunks;
yading@10: 
yading@10:     unsigned int chunk_size;
yading@10:     int chunk_type;
yading@10: 
yading@10:     int i, j, ret;
yading@10: 
yading@10:     int color_packets;
yading@10:     int color_changes;
yading@10:     int color_shift;
yading@10:     unsigned char r, g, b;
yading@10: 
yading@10:     int lines;
yading@10:     int compressed_lines;
yading@10:     int starting_line;
yading@10:     signed short line_packets;
yading@10:     int y_ptr;
yading@10:     int byte_run;
yading@10:     int pixel_skip;
yading@10:     int pixel_countdown;
yading@10:     unsigned char *pixels;
yading@10:     unsigned int pixel_limit;
yading@10: 
yading@10:     bytestream2_init(&g2, buf, buf_size);
yading@10: 
yading@10:     if ((ret = ff_reget_buffer(avctx, &s->frame)) < 0)
yading@10:         return ret;
yading@10: 
yading@10:     pixels = s->frame.data[0];
yading@10:     pixel_limit = s->avctx->height * s->frame.linesize[0];
yading@10:     if (buf_size < 16 || buf_size > INT_MAX - (3 * 256 + FF_INPUT_BUFFER_PADDING_SIZE))
yading@10:         return AVERROR_INVALIDDATA;
yading@10:     frame_size = bytestream2_get_le32(&g2);
yading@10:     if (frame_size > buf_size)
yading@10:         frame_size = buf_size;
yading@10:     bytestream2_skip(&g2, 2); /* skip the magic number */
yading@10:     num_chunks = bytestream2_get_le16(&g2);
yading@10:     bytestream2_skip(&g2, 8);  /* skip padding */
yading@10: 
yading@10:     frame_size -= 16;
yading@10: 
yading@10:     /* iterate through the chunks */
yading@10:     while ((frame_size >= 6) && (num_chunks > 0)) {
yading@10:         int stream_ptr_after_chunk;
yading@10:         chunk_size = bytestream2_get_le32(&g2);
yading@10:         if (chunk_size > frame_size) {
yading@10:             av_log(avctx, AV_LOG_WARNING,
yading@10:                    "Invalid chunk_size = %u > frame_size = %u\n", chunk_size, frame_size);
yading@10:             chunk_size = frame_size;
yading@10:         }
yading@10:         stream_ptr_after_chunk = bytestream2_tell(&g2) - 4 + chunk_size;
yading@10: 
yading@10:         chunk_type = bytestream2_get_le16(&g2);
yading@10: 
yading@10:         switch (chunk_type) {
yading@10:         case FLI_256_COLOR:
yading@10:         case FLI_COLOR:
yading@10:             /* check special case: If this file is from the Magic Carpet
yading@10:              * game and uses 6-bit colors even though it reports 256-color
yading@10:              * chunks in a 0xAF12-type file (fli_type is set to 0xAF13 during
yading@10:              * initialization) */
yading@10:             if ((chunk_type == FLI_256_COLOR) && (s->fli_type != FLC_MAGIC_CARPET_SYNTHETIC_TYPE_CODE))
yading@10:                 color_shift = 0;
yading@10:             else
yading@10:                 color_shift = 2;
yading@10:             /* set up the palette */
yading@10:             color_packets = bytestream2_get_le16(&g2);
yading@10:             palette_ptr = 0;
yading@10:             for (i = 0; i < color_packets; i++) {
yading@10:                 /* first byte is how many colors to skip */
yading@10:                 palette_ptr += bytestream2_get_byte(&g2);
yading@10: 
yading@10:                 /* next byte indicates how many entries to change */
yading@10:                 color_changes = bytestream2_get_byte(&g2);
yading@10: 
yading@10:                 /* if there are 0 color changes, there are actually 256 */
yading@10:                 if (color_changes == 0)
yading@10:                     color_changes = 256;
yading@10: 
yading@10:                 if (bytestream2_tell(&g2) + color_changes * 3 > stream_ptr_after_chunk)
yading@10:                     break;
yading@10: 
yading@10:                 for (j = 0; j < color_changes; j++) {
yading@10:                     unsigned int entry;
yading@10: 
yading@10:                     /* wrap around, for good measure */
yading@10:                     if ((unsigned)palette_ptr >= 256)
yading@10:                         palette_ptr = 0;
yading@10: 
yading@10:                     r = bytestream2_get_byte(&g2) << color_shift;
yading@10:                     g = bytestream2_get_byte(&g2) << color_shift;
yading@10:                     b = bytestream2_get_byte(&g2) << color_shift;
yading@10:                     entry = 0xFFU << 24 | r << 16 | g << 8 | b;
yading@10:                     if (color_shift == 2)
yading@10:                         entry |= entry >> 6 & 0x30303;
yading@10:                     if (s->palette[palette_ptr] != entry)
yading@10:                         s->new_palette = 1;
yading@10:                     s->palette[palette_ptr++] = entry;
yading@10:                 }
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_DELTA:
yading@10:             y_ptr = 0;
yading@10:             compressed_lines = bytestream2_get_le16(&g2);
yading@10:             while (compressed_lines > 0) {
yading@10:                 if (bytestream2_tell(&g2) + 2 > stream_ptr_after_chunk)
yading@10:                     break;
yading@10:                 line_packets = bytestream2_get_le16(&g2);
yading@10:                 if ((line_packets & 0xC000) == 0xC000) {
yading@10:                     // line skip opcode
yading@10:                     line_packets = -line_packets;
yading@10:                     y_ptr += line_packets * s->frame.linesize[0];
yading@10:                 } else if ((line_packets & 0xC000) == 0x4000) {
yading@10:                     av_log(avctx, AV_LOG_ERROR, "Undefined opcode (%x) in DELTA_FLI\n", line_packets);
yading@10:                 } else if ((line_packets & 0xC000) == 0x8000) {
yading@10:                     // "last byte" opcode
yading@10:                     pixel_ptr= y_ptr + s->frame.linesize[0] - 1;
yading@10:                     CHECK_PIXEL_PTR(0);
yading@10:                     pixels[pixel_ptr] = line_packets & 0xff;
yading@10:                 } else {
yading@10:                     compressed_lines--;
yading@10:                     pixel_ptr = y_ptr;
yading@10:                     CHECK_PIXEL_PTR(0);
yading@10:                     pixel_countdown = s->avctx->width;
yading@10:                     for (i = 0; i < line_packets; i++) {
yading@10:                         if (bytestream2_tell(&g2) + 2 > stream_ptr_after_chunk)
yading@10:                             break;
yading@10:                         /* account for the skip bytes */
yading@10:                         pixel_skip = bytestream2_get_byte(&g2);
yading@10:                         pixel_ptr += pixel_skip;
yading@10:                         pixel_countdown -= pixel_skip;
yading@10:                         byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10:                         if (byte_run < 0) {
yading@10:                             byte_run = -byte_run;
yading@10:                             palette_idx1 = bytestream2_get_byte(&g2);
yading@10:                             palette_idx2 = bytestream2_get_byte(&g2);
yading@10:                             CHECK_PIXEL_PTR(byte_run * 2);
yading@10:                             for (j = 0; j < byte_run; j++, pixel_countdown -= 2) {
yading@10:                                 pixels[pixel_ptr++] = palette_idx1;
yading@10:                                 pixels[pixel_ptr++] = palette_idx2;
yading@10:                             }
yading@10:                         } else {
yading@10:                             CHECK_PIXEL_PTR(byte_run * 2);
yading@10:                             if (bytestream2_tell(&g2) + byte_run * 2 > stream_ptr_after_chunk)
yading@10:                                 break;
yading@10:                             for (j = 0; j < byte_run * 2; j++, pixel_countdown--) {
yading@10:                                 pixels[pixel_ptr++] = bytestream2_get_byte(&g2);
yading@10:                             }
yading@10:                         }
yading@10:                     }
yading@10: 
yading@10:                     y_ptr += s->frame.linesize[0];
yading@10:                 }
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_LC:
yading@10:             /* line compressed */
yading@10:             starting_line = bytestream2_get_le16(&g2);
yading@10:             y_ptr = 0;
yading@10:             y_ptr += starting_line * s->frame.linesize[0];
yading@10: 
yading@10:             compressed_lines = bytestream2_get_le16(&g2);
yading@10:             while (compressed_lines > 0) {
yading@10:                 pixel_ptr = y_ptr;
yading@10:                 CHECK_PIXEL_PTR(0);
yading@10:                 pixel_countdown = s->avctx->width;
yading@10:                 if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10:                     break;
yading@10:                 line_packets = bytestream2_get_byte(&g2);
yading@10:                 if (line_packets > 0) {
yading@10:                     for (i = 0; i < line_packets; i++) {
yading@10:                         /* account for the skip bytes */
yading@10:                         if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10:                             break;
yading@10:                         pixel_skip = bytestream2_get_byte(&g2);
yading@10:                         pixel_ptr += pixel_skip;
yading@10:                         pixel_countdown -= pixel_skip;
yading@10:                         byte_run = sign_extend(bytestream2_get_byte(&g2),8);
yading@10:                         if (byte_run > 0) {
yading@10:                             CHECK_PIXEL_PTR(byte_run);
yading@10:                             if (bytestream2_tell(&g2) + byte_run > stream_ptr_after_chunk)
yading@10:                                 break;
yading@10:                             for (j = 0; j < byte_run; j++, pixel_countdown--) {
yading@10:                                 pixels[pixel_ptr++] = bytestream2_get_byte(&g2);
yading@10:                             }
yading@10:                         } else if (byte_run < 0) {
yading@10:                             byte_run = -byte_run;
yading@10:                             palette_idx1 = bytestream2_get_byte(&g2);
yading@10:                             CHECK_PIXEL_PTR(byte_run);
yading@10:                             for (j = 0; j < byte_run; j++, pixel_countdown--) {
yading@10:                                 pixels[pixel_ptr++] = palette_idx1;
yading@10:                             }
yading@10:                         }
yading@10:                     }
yading@10:                 }
yading@10: 
yading@10:                 y_ptr += s->frame.linesize[0];
yading@10:                 compressed_lines--;
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_BLACK:
yading@10:             /* set the whole frame to color 0 (which is usually black) */
yading@10:             memset(pixels, 0,
yading@10:                 s->frame.linesize[0] * s->avctx->height);
yading@10:             break;
yading@10: 
yading@10:         case FLI_BRUN:
yading@10:             /* Byte run compression: This chunk type only occurs in the first
yading@10:              * FLI frame and it will update the entire frame. */
yading@10:             y_ptr = 0;
yading@10:             for (lines = 0; lines < s->avctx->height; lines++) {
yading@10:                 pixel_ptr = y_ptr;
yading@10:                 /* disregard the line packets; instead, iterate through all
yading@10:                  * pixels on a row */
yading@10:                  bytestream2_skip(&g2, 1);
yading@10:                 pixel_countdown = s->avctx->width;
yading@10:                 while (pixel_countdown > 0) {
yading@10:                     if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10:                         break;
yading@10:                     byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10:                     if (!byte_run) {
yading@10:                         av_log(avctx, AV_LOG_ERROR, "Invalid byte run value.\n");
yading@10:                         return AVERROR_INVALIDDATA;
yading@10:                     }
yading@10: 
yading@10:                     if (byte_run > 0) {
yading@10:                         palette_idx1 = bytestream2_get_byte(&g2);
yading@10:                         CHECK_PIXEL_PTR(byte_run);
yading@10:                         for (j = 0; j < byte_run; j++) {
yading@10:                             pixels[pixel_ptr++] = palette_idx1;
yading@10:                             pixel_countdown--;
yading@10:                             if (pixel_countdown < 0)
yading@10:                                 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d) at line %d\n",
yading@10:                                        pixel_countdown, lines);
yading@10:                         }
yading@10:                     } else {  /* copy bytes if byte_run < 0 */
yading@10:                         byte_run = -byte_run;
yading@10:                         CHECK_PIXEL_PTR(byte_run);
yading@10:                         if (bytestream2_tell(&g2) + byte_run > stream_ptr_after_chunk)
yading@10:                             break;
yading@10:                         for (j = 0; j < byte_run; j++) {
yading@10:                             pixels[pixel_ptr++] = bytestream2_get_byte(&g2);
yading@10:                             pixel_countdown--;
yading@10:                             if (pixel_countdown < 0)
yading@10:                                 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d) at line %d\n",
yading@10:                                        pixel_countdown, lines);
yading@10:                         }
yading@10:                     }
yading@10:                 }
yading@10: 
yading@10:                 y_ptr += s->frame.linesize[0];
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_COPY:
yading@10:             /* copy the chunk (uncompressed frame) */
yading@10:             if (chunk_size - 6 != s->avctx->width * s->avctx->height) {
yading@10:                 av_log(avctx, AV_LOG_ERROR, "In chunk FLI_COPY : source data (%d bytes) " \
yading@10:                        "has incorrect size, skipping chunk\n", chunk_size - 6);
yading@10:                 bytestream2_skip(&g2, chunk_size - 6);
yading@10:             } else {
yading@10:                 for (y_ptr = 0; y_ptr < s->frame.linesize[0] * s->avctx->height;
yading@10:                      y_ptr += s->frame.linesize[0]) {
yading@10:                     bytestream2_get_buffer(&g2, &pixels[y_ptr],
yading@10:                                            s->avctx->width);
yading@10:                 }
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_MINI:
yading@10:             /* some sort of a thumbnail? disregard this chunk... */
yading@10:             break;
yading@10: 
yading@10:         default:
yading@10:             av_log(avctx, AV_LOG_ERROR, "Unrecognized chunk type: %d\n", chunk_type);
yading@10:             break;
yading@10:         }
yading@10: 
yading@10:         if (stream_ptr_after_chunk - bytestream2_tell(&g2) > 0)
yading@10:             bytestream2_skip(&g2, stream_ptr_after_chunk - bytestream2_tell(&g2));
yading@10: 
yading@10:         frame_size -= chunk_size;
yading@10:         num_chunks--;
yading@10:     }
yading@10: 
yading@10:     /* by the end of the chunk, the stream ptr should equal the frame
yading@10:      * size (minus 1 or 2, possibly); if it doesn't, issue a warning */
yading@10:     if (bytestream2_get_bytes_left(&g2) > 2)
yading@10:         av_log(avctx, AV_LOG_ERROR, "Processed FLI chunk where chunk size = %d " \
yading@10:                "and final chunk ptr = %d\n", buf_size,
yading@10:                buf_size - bytestream2_get_bytes_left(&g2));
yading@10: 
yading@10:     /* make the palette available on the way out */
yading@10:     memcpy(s->frame.data[1], s->palette, AVPALETTE_SIZE);
yading@10:     if (s->new_palette) {
yading@10:         s->frame.palette_has_changed = 1;
yading@10:         s->new_palette = 0;
yading@10:     }
yading@10: 
yading@10:     if ((ret = av_frame_ref(data, &s->frame)) < 0)
yading@10:         return ret;
yading@10: 
yading@10:     *got_frame = 1;
yading@10: 
yading@10:     return buf_size;
yading@10: }
yading@10: 
yading@10: static int flic_decode_frame_15_16BPP(AVCodecContext *avctx,
yading@10:                                       void *data, int *got_frame,
yading@10:                                       const uint8_t *buf, int buf_size)
yading@10: {
yading@10:     /* Note, the only difference between the 15Bpp and 16Bpp */
yading@10:     /* Format is the pixel format, the packets are processed the same. */
yading@10:     FlicDecodeContext *s = avctx->priv_data;
yading@10: 
yading@10:     GetByteContext g2;
yading@10:     int pixel_ptr;
yading@10:     unsigned char palette_idx1;
yading@10: 
yading@10:     unsigned int frame_size;
yading@10:     int num_chunks;
yading@10: 
yading@10:     unsigned int chunk_size;
yading@10:     int chunk_type;
yading@10: 
yading@10:     int i, j, ret;
yading@10: 
yading@10:     int lines;
yading@10:     int compressed_lines;
yading@10:     signed short line_packets;
yading@10:     int y_ptr;
yading@10:     int byte_run;
yading@10:     int pixel_skip;
yading@10:     int pixel_countdown;
yading@10:     unsigned char *pixels;
yading@10:     int pixel;
yading@10:     unsigned int pixel_limit;
yading@10: 
yading@10:     bytestream2_init(&g2, buf, buf_size);
yading@10: 
yading@10:     if ((ret = ff_reget_buffer(avctx, &s->frame)) < 0)
yading@10:         return ret;
yading@10: 
yading@10:     pixels = s->frame.data[0];
yading@10:     pixel_limit = s->avctx->height * s->frame.linesize[0];
yading@10: 
yading@10:     frame_size = bytestream2_get_le32(&g2);
yading@10:     bytestream2_skip(&g2, 2);  /* skip the magic number */
yading@10:     num_chunks = bytestream2_get_le16(&g2);
yading@10:     bytestream2_skip(&g2, 8);  /* skip padding */
yading@10:     if (frame_size > buf_size)
yading@10:         frame_size = buf_size;
yading@10: 
yading@10:     frame_size -= 16;
yading@10: 
yading@10:     /* iterate through the chunks */
yading@10:     while ((frame_size > 0) && (num_chunks > 0)) {
yading@10:         int stream_ptr_after_chunk;
yading@10:         chunk_size = bytestream2_get_le32(&g2);
yading@10:         if (chunk_size > frame_size) {
yading@10:             av_log(avctx, AV_LOG_WARNING,
yading@10:                    "Invalid chunk_size = %u > frame_size = %u\n", chunk_size, frame_size);
yading@10:             chunk_size = frame_size;
yading@10:         }
yading@10:         stream_ptr_after_chunk = bytestream2_tell(&g2) - 4 + chunk_size;
yading@10: 
yading@10:         chunk_type = bytestream2_get_le16(&g2);
yading@10: 
yading@10: 
yading@10:         switch (chunk_type) {
yading@10:         case FLI_256_COLOR:
yading@10:         case FLI_COLOR:
yading@10:             /* For some reason, it seems that non-palettized flics do
yading@10:              * include one of these chunks in their first frame.
yading@10:              * Why I do not know, it seems rather extraneous. */
yading@10:             av_dlog(avctx,
yading@10:                     "Unexpected Palette chunk %d in non-palettized FLC\n",
yading@10:                     chunk_type);
yading@10:             bytestream2_skip(&g2, chunk_size - 6);
yading@10:             break;
yading@10: 
yading@10:         case FLI_DELTA:
yading@10:         case FLI_DTA_LC:
yading@10:             y_ptr = 0;
yading@10:             compressed_lines = bytestream2_get_le16(&g2);
yading@10:             while (compressed_lines > 0) {
yading@10:                 if (bytestream2_tell(&g2) + 2 > stream_ptr_after_chunk)
yading@10:                     break;
yading@10:                 line_packets = bytestream2_get_le16(&g2);
yading@10:                 if (line_packets < 0) {
yading@10:                     line_packets = -line_packets;
yading@10:                     y_ptr += line_packets * s->frame.linesize[0];
yading@10:                 } else {
yading@10:                     compressed_lines--;
yading@10:                     pixel_ptr = y_ptr;
yading@10:                     CHECK_PIXEL_PTR(0);
yading@10:                     pixel_countdown = s->avctx->width;
yading@10:                     for (i = 0; i < line_packets; i++) {
yading@10:                         /* account for the skip bytes */
yading@10:                         if (bytestream2_tell(&g2) + 2 > stream_ptr_after_chunk)
yading@10:                             break;
yading@10:                         pixel_skip = bytestream2_get_byte(&g2);
yading@10:                         pixel_ptr += (pixel_skip*2); /* Pixel is 2 bytes wide */
yading@10:                         pixel_countdown -= pixel_skip;
yading@10:                         byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10:                         if (byte_run < 0) {
yading@10:                             byte_run = -byte_run;
yading@10:                             pixel    = bytestream2_get_le16(&g2);
yading@10:                             CHECK_PIXEL_PTR(2 * byte_run);
yading@10:                             for (j = 0; j < byte_run; j++, pixel_countdown -= 2) {
yading@10:                                 *((signed short*)(&pixels[pixel_ptr])) = pixel;
yading@10:                                 pixel_ptr += 2;
yading@10:                             }
yading@10:                         } else {
yading@10:                             if (bytestream2_tell(&g2) + 2*byte_run > stream_ptr_after_chunk)
yading@10:                                 break;
yading@10:                             CHECK_PIXEL_PTR(2 * byte_run);
yading@10:                             for (j = 0; j < byte_run; j++, pixel_countdown--) {
yading@10:                                 *((signed short*)(&pixels[pixel_ptr])) = bytestream2_get_le16(&g2);
yading@10:                                 pixel_ptr += 2;
yading@10:                             }
yading@10:                         }
yading@10:                     }
yading@10: 
yading@10:                     y_ptr += s->frame.linesize[0];
yading@10:                 }
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_LC:
yading@10:             av_log(avctx, AV_LOG_ERROR, "Unexpected FLI_LC chunk in non-paletised FLC\n");
yading@10:             bytestream2_skip(&g2, chunk_size - 6);
yading@10:             break;
yading@10: 
yading@10:         case FLI_BLACK:
yading@10:             /* set the whole frame to 0x0000 which is black in both 15Bpp and 16Bpp modes. */
yading@10:             memset(pixels, 0x0000,
yading@10:                    s->frame.linesize[0] * s->avctx->height);
yading@10:             break;
yading@10: 
yading@10:         case FLI_BRUN:
yading@10:             y_ptr = 0;
yading@10:             for (lines = 0; lines < s->avctx->height; lines++) {
yading@10:                 pixel_ptr = y_ptr;
yading@10:                 /* disregard the line packets; instead, iterate through all
yading@10:                  * pixels on a row */
yading@10:                 bytestream2_skip(&g2, 1);
yading@10:                 pixel_countdown = (s->avctx->width * 2);
yading@10: 
yading@10:                 while (pixel_countdown > 0) {
yading@10:                     if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10:                         break;
yading@10:                     byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10:                     if (byte_run > 0) {
yading@10:                         palette_idx1 = bytestream2_get_byte(&g2);
yading@10:                         CHECK_PIXEL_PTR(byte_run);
yading@10:                         for (j = 0; j < byte_run; j++) {
yading@10:                             pixels[pixel_ptr++] = palette_idx1;
yading@10:                             pixel_countdown--;
yading@10:                             if (pixel_countdown < 0)
yading@10:                                 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d) (linea%d)\n",
yading@10:                                        pixel_countdown, lines);
yading@10:                         }
yading@10:                     } else {  /* copy bytes if byte_run < 0 */
yading@10:                         byte_run = -byte_run;
yading@10:                         if (bytestream2_tell(&g2) + byte_run > stream_ptr_after_chunk)
yading@10:                             break;
yading@10:                         CHECK_PIXEL_PTR(byte_run);
yading@10:                         for (j = 0; j < byte_run; j++) {
yading@10:                             palette_idx1 = bytestream2_get_byte(&g2);
yading@10:                             pixels[pixel_ptr++] = palette_idx1;
yading@10:                             pixel_countdown--;
yading@10:                             if (pixel_countdown < 0)
yading@10:                                 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d) at line %d\n",
yading@10:                                        pixel_countdown, lines);
yading@10:                         }
yading@10:                     }
yading@10:                 }
yading@10: 
yading@10:                 /* Now FLX is strange, in that it is "byte" as opposed to "pixel" run length compressed.
yading@10:                  * This does not give us any good opportunity to perform word endian conversion
yading@10:                  * during decompression. So if it is required (i.e., this is not a LE target, we do
yading@10:                  * a second pass over the line here, swapping the bytes.
yading@10:                  */
yading@10: #if HAVE_BIGENDIAN
yading@10:                 pixel_ptr = y_ptr;
yading@10:                 pixel_countdown = s->avctx->width;
yading@10:                 while (pixel_countdown > 0) {
yading@10:                     *((signed short*)(&pixels[pixel_ptr])) = AV_RL16(&buf[pixel_ptr]);
yading@10:                     pixel_ptr += 2;
yading@10:                 }
yading@10: #endif
yading@10:                 y_ptr += s->frame.linesize[0];
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_DTA_BRUN:
yading@10:             y_ptr = 0;
yading@10:             for (lines = 0; lines < s->avctx->height; lines++) {
yading@10:                 pixel_ptr = y_ptr;
yading@10:                 /* disregard the line packets; instead, iterate through all
yading@10:                  * pixels on a row */
yading@10:                 bytestream2_skip(&g2, 1);
yading@10:                 pixel_countdown = s->avctx->width; /* Width is in pixels, not bytes */
yading@10: 
yading@10:                 while (pixel_countdown > 0) {
yading@10:                     if (bytestream2_tell(&g2) + 1 > stream_ptr_after_chunk)
yading@10:                         break;
yading@10:                     byte_run = sign_extend(bytestream2_get_byte(&g2), 8);
yading@10:                     if (byte_run > 0) {
yading@10:                         pixel    = bytestream2_get_le16(&g2);
yading@10:                         CHECK_PIXEL_PTR(2 * byte_run);
yading@10:                         for (j = 0; j < byte_run; j++) {
yading@10:                             *((signed short*)(&pixels[pixel_ptr])) = pixel;
yading@10:                             pixel_ptr += 2;
yading@10:                             pixel_countdown--;
yading@10:                             if (pixel_countdown < 0)
yading@10:                                 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d)\n",
yading@10:                                        pixel_countdown);
yading@10:                         }
yading@10:                     } else {  /* copy pixels if byte_run < 0 */
yading@10:                         byte_run = -byte_run;
yading@10:                         if (bytestream2_tell(&g2) + 2 * byte_run > stream_ptr_after_chunk)
yading@10:                             break;
yading@10:                         CHECK_PIXEL_PTR(2 * byte_run);
yading@10:                         for (j = 0; j < byte_run; j++) {
yading@10:                             *((signed short*)(&pixels[pixel_ptr])) = bytestream2_get_le16(&g2);
yading@10:                             pixel_ptr  += 2;
yading@10:                             pixel_countdown--;
yading@10:                             if (pixel_countdown < 0)
yading@10:                                 av_log(avctx, AV_LOG_ERROR, "pixel_countdown < 0 (%d)\n",
yading@10:                                        pixel_countdown);
yading@10:                         }
yading@10:                     }
yading@10:                 }
yading@10: 
yading@10:                 y_ptr += s->frame.linesize[0];
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_COPY:
yading@10:         case FLI_DTA_COPY:
yading@10:             /* copy the chunk (uncompressed frame) */
yading@10:             if (chunk_size - 6 > (unsigned int)(s->avctx->width * s->avctx->height)*2) {
yading@10:                 av_log(avctx, AV_LOG_ERROR, "In chunk FLI_COPY : source data (%d bytes) " \
yading@10:                        "bigger than image, skipping chunk\n", chunk_size - 6);
yading@10:                 bytestream2_skip(&g2, chunk_size - 6);
yading@10:             } else {
yading@10: 
yading@10:                 for (y_ptr = 0; y_ptr < s->frame.linesize[0] * s->avctx->height;
yading@10:                      y_ptr += s->frame.linesize[0]) {
yading@10: 
yading@10:                     pixel_countdown = s->avctx->width;
yading@10:                     pixel_ptr = 0;
yading@10:                     while (pixel_countdown > 0) {
yading@10:                       *((signed short*)(&pixels[y_ptr + pixel_ptr])) = bytestream2_get_le16(&g2);
yading@10:                       pixel_ptr += 2;
yading@10:                       pixel_countdown--;
yading@10:                     }
yading@10:                 }
yading@10:             }
yading@10:             break;
yading@10: 
yading@10:         case FLI_MINI:
yading@10:             /* some sort of a thumbnail? disregard this chunk... */
yading@10:             bytestream2_skip(&g2, chunk_size - 6);
yading@10:             break;
yading@10: 
yading@10:         default:
yading@10:             av_log(avctx, AV_LOG_ERROR, "Unrecognized chunk type: %d\n", chunk_type);
yading@10:             break;
yading@10:         }
yading@10: 
yading@10:         frame_size -= chunk_size;
yading@10:         num_chunks--;
yading@10:     }
yading@10: 
yading@10:     /* by the end of the chunk, the stream ptr should equal the frame
yading@10:      * size (minus 1, possibly); if it doesn't, issue a warning */
yading@10:     if ((bytestream2_get_bytes_left(&g2) != 0) && (bytestream2_get_bytes_left(&g2) != 1))
yading@10:         av_log(avctx, AV_LOG_ERROR, "Processed FLI chunk where chunk size = %d " \
yading@10:                "and final chunk ptr = %d\n", buf_size, bytestream2_tell(&g2));
yading@10: 
yading@10:     if ((ret = av_frame_ref(data, &s->frame)) < 0)
yading@10:         return ret;
yading@10: 
yading@10:     *got_frame = 1;
yading@10: 
yading@10:     return buf_size;
yading@10: }
yading@10: 
yading@10: static int flic_decode_frame_24BPP(AVCodecContext *avctx,
yading@10:                                    void *data, int *got_frame,
yading@10:                                    const uint8_t *buf, int buf_size)
yading@10: {
yading@10:   av_log(avctx, AV_LOG_ERROR, "24Bpp FLC Unsupported due to lack of test files.\n");
yading@10:   return AVERROR_PATCHWELCOME;
yading@10: }
yading@10: 
yading@10: static int flic_decode_frame(AVCodecContext *avctx,
yading@10:                              void *data, int *got_frame,
yading@10:                              AVPacket *avpkt)
yading@10: {
yading@10:     const uint8_t *buf = avpkt->data;
yading@10:     int buf_size = avpkt->size;
yading@10:     if (avctx->pix_fmt == AV_PIX_FMT_PAL8) {
yading@10:       return flic_decode_frame_8BPP(avctx, data, got_frame,
yading@10:                                     buf, buf_size);
yading@10:     }
yading@10:     else if ((avctx->pix_fmt == AV_PIX_FMT_RGB555) ||
yading@10:              (avctx->pix_fmt == AV_PIX_FMT_RGB565)) {
yading@10:       return flic_decode_frame_15_16BPP(avctx, data, got_frame,
yading@10:                                         buf, buf_size);
yading@10:     }
yading@10:     else if (avctx->pix_fmt == AV_PIX_FMT_BGR24) {
yading@10:       return flic_decode_frame_24BPP(avctx, data, got_frame,
yading@10:                                      buf, buf_size);
yading@10:     }
yading@10: 
yading@10:     /* Should not get  here, ever as the pix_fmt is processed */
yading@10:     /* in flic_decode_init and the above if should deal with */
yading@10:     /* the finite set of possibilites allowable by here. */
yading@10:     /* But in case we do, just error out. */
yading@10:     av_log(avctx, AV_LOG_ERROR, "Unknown FLC format, my science cannot explain how this happened.\n");
yading@10:     return AVERROR_BUG;
yading@10: }
yading@10: 
yading@10: 
yading@10: static av_cold int flic_decode_end(AVCodecContext *avctx)
yading@10: {
yading@10:     FlicDecodeContext *s = avctx->priv_data;
yading@10: 
yading@10:     av_frame_unref(&s->frame);
yading@10: 
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: AVCodec ff_flic_decoder = {
yading@10:     .name           = "flic",
yading@10:     .type           = AVMEDIA_TYPE_VIDEO,
yading@10:     .id             = AV_CODEC_ID_FLIC,
yading@10:     .priv_data_size = sizeof(FlicDecodeContext),
yading@10:     .init           = flic_decode_init,
yading@10:     .close          = flic_decode_end,
yading@10:     .decode         = flic_decode_frame,
yading@10:     .capabilities   = CODEC_CAP_DR1,
yading@10:     .long_name      = NULL_IF_CONFIG_SMALL("Autodesk Animator Flic video"),
yading@10: };