yading@10: /*
yading@10:  * Interplay MVE Video Decoder
yading@10:  * Copyright (C) 2003 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:  * Interplay MVE Video Decoder by Mike Melanson (melanson@pcisys.net)
yading@10:  * For more information about the Interplay MVE format, visit:
yading@10:  *   http://www.pcisys.net/~melanson/codecs/interplay-mve.txt
yading@10:  * This code is written in such a way that the identifiers match up
yading@10:  * with the encoding descriptions in the document.
yading@10:  *
yading@10:  * This decoder presently only supports a PAL8 output colorspace.
yading@10:  *
yading@10:  * An Interplay video frame consists of 2 parts: The decoding map and
yading@10:  * the video data. A demuxer must load these 2 parts together in a single
yading@10:  * buffer before sending it through the stream to this decoder.
yading@10:  */
yading@10: 
yading@10: #include <stdio.h>
yading@10: #include <stdlib.h>
yading@10: #include <string.h>
yading@10: 
yading@10: #include "avcodec.h"
yading@10: #include "bytestream.h"
yading@10: #include "hpeldsp.h"
yading@10: #define BITSTREAM_READER_LE
yading@10: #include "get_bits.h"
yading@10: #include "internal.h"
yading@10: 
yading@10: #define PALETTE_COUNT 256
yading@10: 
yading@10: typedef struct IpvideoContext {
yading@10: 
yading@10:     AVCodecContext *avctx;
yading@10:     HpelDSPContext hdsp;
yading@10:     AVFrame *second_last_frame;
yading@10:     AVFrame *last_frame;
yading@10:     const unsigned char *decoding_map;
yading@10:     int decoding_map_size;
yading@10: 
yading@10:     int is_16bpp;
yading@10:     GetByteContext stream_ptr, mv_ptr;
yading@10:     unsigned char *pixel_ptr;
yading@10:     int line_inc;
yading@10:     int stride;
yading@10:     int upper_motion_limit_offset;
yading@10: 
yading@10:     uint32_t pal[256];
yading@10: } IpvideoContext;
yading@10: 
yading@10: static int copy_from(IpvideoContext *s, AVFrame *src, AVFrame *dst, int delta_x, int delta_y)
yading@10: {
yading@10:     int current_offset = s->pixel_ptr - dst->data[0];
yading@10:     int motion_offset = current_offset + delta_y * dst->linesize[0]
yading@10:                        + delta_x * (1 + s->is_16bpp);
yading@10:     if (motion_offset < 0) {
yading@10:         av_log(s->avctx, AV_LOG_ERROR, "motion offset < 0 (%d)\n", motion_offset);
yading@10:         return AVERROR_INVALIDDATA;
yading@10:     } else if (motion_offset > s->upper_motion_limit_offset) {
yading@10:         av_log(s->avctx, AV_LOG_ERROR, "motion offset above limit (%d >= %d)\n",
yading@10:             motion_offset, s->upper_motion_limit_offset);
yading@10:         return AVERROR_INVALIDDATA;
yading@10:     }
yading@10:     if (src->data[0] == NULL) {
yading@10:         av_log(s->avctx, AV_LOG_ERROR, "Invalid decode type, corrupted header?\n");
yading@10:         return AVERROR(EINVAL);
yading@10:     }
yading@10:     s->hdsp.put_pixels_tab[!s->is_16bpp][0](s->pixel_ptr, src->data[0] + motion_offset,
yading@10:                                             dst->linesize[0], 8);
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x0(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     return copy_from(s, s->last_frame, frame, 0, 0);
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x1(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     return copy_from(s, s->second_last_frame, frame, 0, 0);
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x2(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     unsigned char B;
yading@10:     int x, y;
yading@10: 
yading@10:     /* copy block from 2 frames ago using a motion vector; need 1 more byte */
yading@10:     if (!s->is_16bpp) {
yading@10:         B = bytestream2_get_byte(&s->stream_ptr);
yading@10:     } else {
yading@10:         B = bytestream2_get_byte(&s->mv_ptr);
yading@10:     }
yading@10: 
yading@10:     if (B < 56) {
yading@10:         x = 8 + (B % 7);
yading@10:         y = B / 7;
yading@10:     } else {
yading@10:         x = -14 + ((B - 56) % 29);
yading@10:         y =   8 + ((B - 56) / 29);
yading@10:     }
yading@10: 
yading@10:     av_dlog(s->avctx, "motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
yading@10:     return copy_from(s, s->second_last_frame, frame, x, y);
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x3(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     unsigned char B;
yading@10:     int x, y;
yading@10: 
yading@10:     /* copy 8x8 block from current frame from an up/left block */
yading@10: 
yading@10:     /* need 1 more byte for motion */
yading@10:     if (!s->is_16bpp) {
yading@10:         B = bytestream2_get_byte(&s->stream_ptr);
yading@10:     } else {
yading@10:         B = bytestream2_get_byte(&s->mv_ptr);
yading@10:     }
yading@10: 
yading@10:     if (B < 56) {
yading@10:         x = -(8 + (B % 7));
yading@10:         y = -(B / 7);
yading@10:     } else {
yading@10:         x = -(-14 + ((B - 56) % 29));
yading@10:         y = -(  8 + ((B - 56) / 29));
yading@10:     }
yading@10: 
yading@10:     av_dlog(s->avctx, "motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
yading@10:     return copy_from(s, frame, frame, x, y);
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x4(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     unsigned char B, BL, BH;
yading@10: 
yading@10:     /* copy a block from the previous frame; need 1 more byte */
yading@10:     if (!s->is_16bpp) {
yading@10:         B = bytestream2_get_byte(&s->stream_ptr);
yading@10:     } else {
yading@10:         B = bytestream2_get_byte(&s->mv_ptr);
yading@10:     }
yading@10: 
yading@10:     BL = B & 0x0F;
yading@10:     BH = (B >> 4) & 0x0F;
yading@10:     x = -8 + BL;
yading@10:     y = -8 + BH;
yading@10: 
yading@10:     av_dlog(s->avctx, "motion byte = %d, (x, y) = (%d, %d)\n", B, x, y);
yading@10:     return copy_from(s, s->last_frame, frame, x, y);
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x5(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     signed char x, y;
yading@10: 
yading@10:     /* copy a block from the previous frame using an expanded range;
yading@10:      * need 2 more bytes */
yading@10:     x = bytestream2_get_byte(&s->stream_ptr);
yading@10:     y = bytestream2_get_byte(&s->stream_ptr);
yading@10: 
yading@10:     av_dlog(s->avctx, "motion bytes = %d, %d\n", x, y);
yading@10:     return copy_from(s, s->last_frame, frame, x, y);
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x6(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     /* mystery opcode? skip multiple blocks? */
yading@10:     av_log(s->avctx, AV_LOG_ERROR, "Help! Mystery opcode 0x6 seen\n");
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x7(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     unsigned char P[2];
yading@10:     unsigned int flags;
yading@10: 
yading@10:     /* 2-color encoding */
yading@10:     P[0] = bytestream2_get_byte(&s->stream_ptr);
yading@10:     P[1] = bytestream2_get_byte(&s->stream_ptr);
yading@10: 
yading@10:     if (P[0] <= P[1]) {
yading@10: 
yading@10:         /* need 8 more bytes from the stream */
yading@10:         for (y = 0; y < 8; y++) {
yading@10:             flags = bytestream2_get_byte(&s->stream_ptr) | 0x100;
yading@10:             for (; flags != 1; flags >>= 1)
yading@10:                 *s->pixel_ptr++ = P[flags & 1];
yading@10:             s->pixel_ptr += s->line_inc;
yading@10:         }
yading@10: 
yading@10:     } else {
yading@10: 
yading@10:         /* need 2 more bytes from the stream */
yading@10:         flags = bytestream2_get_le16(&s->stream_ptr);
yading@10:         for (y = 0; y < 8; y += 2) {
yading@10:             for (x = 0; x < 8; x += 2, flags >>= 1) {
yading@10:                 s->pixel_ptr[x                ] =
yading@10:                 s->pixel_ptr[x + 1            ] =
yading@10:                 s->pixel_ptr[x +     s->stride] =
yading@10:                 s->pixel_ptr[x + 1 + s->stride] = P[flags & 1];
yading@10:             }
yading@10:             s->pixel_ptr += s->stride * 2;
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x8(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     unsigned char P[4];
yading@10:     unsigned int flags = 0;
yading@10: 
yading@10:     /* 2-color encoding for each 4x4 quadrant, or 2-color encoding on
yading@10:      * either top and bottom or left and right halves */
yading@10:     P[0] = bytestream2_get_byte(&s->stream_ptr);
yading@10:     P[1] = bytestream2_get_byte(&s->stream_ptr);
yading@10: 
yading@10:     if (P[0] <= P[1]) {
yading@10:         for (y = 0; y < 16; y++) {
yading@10:             // new values for each 4x4 block
yading@10:             if (!(y & 3)) {
yading@10:                 if (y) {
yading@10:                     P[0]  = bytestream2_get_byte(&s->stream_ptr);
yading@10:                     P[1]  = bytestream2_get_byte(&s->stream_ptr);
yading@10:                 }
yading@10:                 flags = bytestream2_get_le16(&s->stream_ptr);
yading@10:             }
yading@10: 
yading@10:             for (x = 0; x < 4; x++, flags >>= 1)
yading@10:                 *s->pixel_ptr++ = P[flags & 1];
yading@10:             s->pixel_ptr += s->stride - 4;
yading@10:             // switch to right half
yading@10:             if (y == 7) s->pixel_ptr -= 8 * s->stride - 4;
yading@10:         }
yading@10: 
yading@10:     } else {
yading@10:         flags = bytestream2_get_le32(&s->stream_ptr);
yading@10:         P[2] = bytestream2_get_byte(&s->stream_ptr);
yading@10:         P[3] = bytestream2_get_byte(&s->stream_ptr);
yading@10: 
yading@10:         if (P[2] <= P[3]) {
yading@10: 
yading@10:             /* vertical split; left & right halves are 2-color encoded */
yading@10: 
yading@10:             for (y = 0; y < 16; y++) {
yading@10:                 for (x = 0; x < 4; x++, flags >>= 1)
yading@10:                     *s->pixel_ptr++ = P[flags & 1];
yading@10:                 s->pixel_ptr += s->stride - 4;
yading@10:                 // switch to right half
yading@10:                 if (y == 7) {
yading@10:                     s->pixel_ptr -= 8 * s->stride - 4;
yading@10:                     P[0]  = P[2];
yading@10:                     P[1]  = P[3];
yading@10:                     flags = bytestream2_get_le32(&s->stream_ptr);
yading@10:                 }
yading@10:             }
yading@10: 
yading@10:         } else {
yading@10: 
yading@10:             /* horizontal split; top & bottom halves are 2-color encoded */
yading@10: 
yading@10:             for (y = 0; y < 8; y++) {
yading@10:                 if (y == 4) {
yading@10:                     P[0]  = P[2];
yading@10:                     P[1]  = P[3];
yading@10:                     flags = bytestream2_get_le32(&s->stream_ptr);
yading@10:                 }
yading@10: 
yading@10:                 for (x = 0; x < 8; x++, flags >>= 1)
yading@10:                     *s->pixel_ptr++ = P[flags & 1];
yading@10:                 s->pixel_ptr += s->line_inc;
yading@10:             }
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x9(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     unsigned char P[4];
yading@10: 
yading@10:     /* 4-color encoding */
yading@10:     bytestream2_get_buffer(&s->stream_ptr, P, 4);
yading@10: 
yading@10:     if (P[0] <= P[1]) {
yading@10:         if (P[2] <= P[3]) {
yading@10: 
yading@10:             /* 1 of 4 colors for each pixel, need 16 more bytes */
yading@10:             for (y = 0; y < 8; y++) {
yading@10:                 /* get the next set of 8 2-bit flags */
yading@10:                 int flags = bytestream2_get_le16(&s->stream_ptr);
yading@10:                 for (x = 0; x < 8; x++, flags >>= 2)
yading@10:                     *s->pixel_ptr++ = P[flags & 0x03];
yading@10:                 s->pixel_ptr += s->line_inc;
yading@10:             }
yading@10: 
yading@10:         } else {
yading@10:             uint32_t flags;
yading@10: 
yading@10:             /* 1 of 4 colors for each 2x2 block, need 4 more bytes */
yading@10:             flags = bytestream2_get_le32(&s->stream_ptr);
yading@10: 
yading@10:             for (y = 0; y < 8; y += 2) {
yading@10:                 for (x = 0; x < 8; x += 2, flags >>= 2) {
yading@10:                     s->pixel_ptr[x                ] =
yading@10:                     s->pixel_ptr[x + 1            ] =
yading@10:                     s->pixel_ptr[x +     s->stride] =
yading@10:                     s->pixel_ptr[x + 1 + s->stride] = P[flags & 0x03];
yading@10:                 }
yading@10:                 s->pixel_ptr += s->stride * 2;
yading@10:             }
yading@10: 
yading@10:         }
yading@10:     } else {
yading@10:         uint64_t flags;
yading@10: 
yading@10:         /* 1 of 4 colors for each 2x1 or 1x2 block, need 8 more bytes */
yading@10:         flags = bytestream2_get_le64(&s->stream_ptr);
yading@10:         if (P[2] <= P[3]) {
yading@10:             for (y = 0; y < 8; y++) {
yading@10:                 for (x = 0; x < 8; x += 2, flags >>= 2) {
yading@10:                     s->pixel_ptr[x    ] =
yading@10:                     s->pixel_ptr[x + 1] = P[flags & 0x03];
yading@10:                 }
yading@10:                 s->pixel_ptr += s->stride;
yading@10:             }
yading@10:         } else {
yading@10:             for (y = 0; y < 8; y += 2) {
yading@10:                 for (x = 0; x < 8; x++, flags >>= 2) {
yading@10:                     s->pixel_ptr[x            ] =
yading@10:                     s->pixel_ptr[x + s->stride] = P[flags & 0x03];
yading@10:                 }
yading@10:                 s->pixel_ptr += s->stride * 2;
yading@10:             }
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xA(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     unsigned char P[8];
yading@10:     int flags = 0;
yading@10: 
yading@10:     bytestream2_get_buffer(&s->stream_ptr, P, 4);
yading@10: 
yading@10:     /* 4-color encoding for each 4x4 quadrant, or 4-color encoding on
yading@10:      * either top and bottom or left and right halves */
yading@10:     if (P[0] <= P[1]) {
yading@10: 
yading@10:         /* 4-color encoding for each quadrant; need 32 bytes */
yading@10:         for (y = 0; y < 16; y++) {
yading@10:             // new values for each 4x4 block
yading@10:             if (!(y & 3)) {
yading@10:                 if (y) bytestream2_get_buffer(&s->stream_ptr, P, 4);
yading@10:                 flags = bytestream2_get_le32(&s->stream_ptr);
yading@10:             }
yading@10: 
yading@10:             for (x = 0; x < 4; x++, flags >>= 2)
yading@10:                 *s->pixel_ptr++ = P[flags & 0x03];
yading@10: 
yading@10:             s->pixel_ptr += s->stride - 4;
yading@10:             // switch to right half
yading@10:             if (y == 7) s->pixel_ptr -= 8 * s->stride - 4;
yading@10:         }
yading@10: 
yading@10:     } else {
yading@10:         // vertical split?
yading@10:         int vert;
yading@10:         uint64_t flags = bytestream2_get_le64(&s->stream_ptr);
yading@10: 
yading@10:         bytestream2_get_buffer(&s->stream_ptr, P + 4, 4);
yading@10:         vert = P[4] <= P[5];
yading@10: 
yading@10:         /* 4-color encoding for either left and right or top and bottom
yading@10:          * halves */
yading@10: 
yading@10:         for (y = 0; y < 16; y++) {
yading@10:             for (x = 0; x < 4; x++, flags >>= 2)
yading@10:                 *s->pixel_ptr++ = P[flags & 0x03];
yading@10: 
yading@10:             if (vert) {
yading@10:                 s->pixel_ptr += s->stride - 4;
yading@10:                 // switch to right half
yading@10:                 if (y == 7) s->pixel_ptr -= 8 * s->stride - 4;
yading@10:             } else if (y & 1) s->pixel_ptr += s->line_inc;
yading@10: 
yading@10:             // load values for second half
yading@10:             if (y == 7) {
yading@10:                 memcpy(P, P + 4, 4);
yading@10:                 flags = bytestream2_get_le64(&s->stream_ptr);
yading@10:             }
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xB(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int y;
yading@10: 
yading@10:     /* 64-color encoding (each pixel in block is a different color) */
yading@10:     for (y = 0; y < 8; y++) {
yading@10:         bytestream2_get_buffer(&s->stream_ptr, s->pixel_ptr, 8);
yading@10:         s->pixel_ptr  += s->stride;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xC(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10: 
yading@10:     /* 16-color block encoding: each 2x2 block is a different color */
yading@10:     for (y = 0; y < 8; y += 2) {
yading@10:         for (x = 0; x < 8; x += 2) {
yading@10:             s->pixel_ptr[x                ] =
yading@10:             s->pixel_ptr[x + 1            ] =
yading@10:             s->pixel_ptr[x +     s->stride] =
yading@10:             s->pixel_ptr[x + 1 + s->stride] = bytestream2_get_byte(&s->stream_ptr);
yading@10:         }
yading@10:         s->pixel_ptr += s->stride * 2;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xD(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int y;
yading@10:     unsigned char P[2];
yading@10: 
yading@10:     /* 4-color block encoding: each 4x4 block is a different color */
yading@10:     for (y = 0; y < 8; y++) {
yading@10:         if (!(y & 3)) {
yading@10:             P[0] = bytestream2_get_byte(&s->stream_ptr);
yading@10:             P[1] = bytestream2_get_byte(&s->stream_ptr);
yading@10:         }
yading@10:         memset(s->pixel_ptr,     P[0], 4);
yading@10:         memset(s->pixel_ptr + 4, P[1], 4);
yading@10:         s->pixel_ptr += s->stride;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xE(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int y;
yading@10:     unsigned char pix;
yading@10: 
yading@10:     /* 1-color encoding: the whole block is 1 solid color */
yading@10:     pix = bytestream2_get_byte(&s->stream_ptr);
yading@10: 
yading@10:     for (y = 0; y < 8; y++) {
yading@10:         memset(s->pixel_ptr, pix, 8);
yading@10:         s->pixel_ptr += s->stride;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xF(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     unsigned char sample[2];
yading@10: 
yading@10:     /* dithered encoding */
yading@10:     sample[0] = bytestream2_get_byte(&s->stream_ptr);
yading@10:     sample[1] = bytestream2_get_byte(&s->stream_ptr);
yading@10: 
yading@10:     for (y = 0; y < 8; y++) {
yading@10:         for (x = 0; x < 8; x += 2) {
yading@10:             *s->pixel_ptr++ = sample[  y & 1 ];
yading@10:             *s->pixel_ptr++ = sample[!(y & 1)];
yading@10:         }
yading@10:         s->pixel_ptr += s->line_inc;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x6_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     signed char x, y;
yading@10: 
yading@10:     /* copy a block from the second last frame using an expanded range */
yading@10:     x = bytestream2_get_byte(&s->stream_ptr);
yading@10:     y = bytestream2_get_byte(&s->stream_ptr);
yading@10: 
yading@10:     av_dlog(s->avctx, "motion bytes = %d, %d\n", x, y);
yading@10:     return copy_from(s, s->second_last_frame, frame, x, y);
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x7_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     uint16_t P[2];
yading@10:     unsigned int flags;
yading@10:     uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;
yading@10: 
yading@10:     /* 2-color encoding */
yading@10:     P[0] = bytestream2_get_le16(&s->stream_ptr);
yading@10:     P[1] = bytestream2_get_le16(&s->stream_ptr);
yading@10: 
yading@10:     if (!(P[0] & 0x8000)) {
yading@10: 
yading@10:         for (y = 0; y < 8; y++) {
yading@10:             flags = bytestream2_get_byte(&s->stream_ptr) | 0x100;
yading@10:             for (; flags != 1; flags >>= 1)
yading@10:                 *pixel_ptr++ = P[flags & 1];
yading@10:             pixel_ptr += s->line_inc;
yading@10:         }
yading@10: 
yading@10:     } else {
yading@10: 
yading@10:         flags = bytestream2_get_le16(&s->stream_ptr);
yading@10:         for (y = 0; y < 8; y += 2) {
yading@10:             for (x = 0; x < 8; x += 2, flags >>= 1) {
yading@10:                 pixel_ptr[x                ] =
yading@10:                 pixel_ptr[x + 1            ] =
yading@10:                 pixel_ptr[x +     s->stride] =
yading@10:                 pixel_ptr[x + 1 + s->stride] = P[flags & 1];
yading@10:             }
yading@10:             pixel_ptr += s->stride * 2;
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x8_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     uint16_t P[4];
yading@10:     unsigned int flags = 0;
yading@10:     uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;
yading@10: 
yading@10:     /* 2-color encoding for each 4x4 quadrant, or 2-color encoding on
yading@10:      * either top and bottom or left and right halves */
yading@10:     P[0] = bytestream2_get_le16(&s->stream_ptr);
yading@10:     P[1] = bytestream2_get_le16(&s->stream_ptr);
yading@10: 
yading@10:     if (!(P[0] & 0x8000)) {
yading@10: 
yading@10:         for (y = 0; y < 16; y++) {
yading@10:             // new values for each 4x4 block
yading@10:             if (!(y & 3)) {
yading@10:                 if (y) {
yading@10:                     P[0] = bytestream2_get_le16(&s->stream_ptr);
yading@10:                     P[1] = bytestream2_get_le16(&s->stream_ptr);
yading@10:                 }
yading@10:                 flags = bytestream2_get_le16(&s->stream_ptr);
yading@10:             }
yading@10: 
yading@10:             for (x = 0; x < 4; x++, flags >>= 1)
yading@10:                 *pixel_ptr++ = P[flags & 1];
yading@10:             pixel_ptr += s->stride - 4;
yading@10:             // switch to right half
yading@10:             if (y == 7) pixel_ptr -= 8 * s->stride - 4;
yading@10:         }
yading@10: 
yading@10:     } else {
yading@10: 
yading@10:         flags = bytestream2_get_le32(&s->stream_ptr);
yading@10:         P[2]  = bytestream2_get_le16(&s->stream_ptr);
yading@10:         P[3]  = bytestream2_get_le16(&s->stream_ptr);
yading@10: 
yading@10:         if (!(P[2] & 0x8000)) {
yading@10: 
yading@10:             /* vertical split; left & right halves are 2-color encoded */
yading@10: 
yading@10:             for (y = 0; y < 16; y++) {
yading@10:                 for (x = 0; x < 4; x++, flags >>= 1)
yading@10:                     *pixel_ptr++ = P[flags & 1];
yading@10:                 pixel_ptr += s->stride - 4;
yading@10:                 // switch to right half
yading@10:                 if (y == 7) {
yading@10:                     pixel_ptr -= 8 * s->stride - 4;
yading@10:                     P[0]  = P[2];
yading@10:                     P[1]  = P[3];
yading@10:                     flags = bytestream2_get_le32(&s->stream_ptr);
yading@10:                 }
yading@10:             }
yading@10: 
yading@10:         } else {
yading@10: 
yading@10:             /* horizontal split; top & bottom halves are 2-color encoded */
yading@10: 
yading@10:             for (y = 0; y < 8; y++) {
yading@10:                 if (y == 4) {
yading@10:                     P[0]  = P[2];
yading@10:                     P[1]  = P[3];
yading@10:                     flags = bytestream2_get_le32(&s->stream_ptr);
yading@10:                 }
yading@10: 
yading@10:                 for (x = 0; x < 8; x++, flags >>= 1)
yading@10:                     *pixel_ptr++ = P[flags & 1];
yading@10:                 pixel_ptr += s->line_inc;
yading@10:             }
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0x9_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     uint16_t P[4];
yading@10:     uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;
yading@10: 
yading@10:     /* 4-color encoding */
yading@10:     for (x = 0; x < 4; x++)
yading@10:         P[x] = bytestream2_get_le16(&s->stream_ptr);
yading@10: 
yading@10:     if (!(P[0] & 0x8000)) {
yading@10:         if (!(P[2] & 0x8000)) {
yading@10: 
yading@10:             /* 1 of 4 colors for each pixel */
yading@10:             for (y = 0; y < 8; y++) {
yading@10:                 /* get the next set of 8 2-bit flags */
yading@10:                 int flags = bytestream2_get_le16(&s->stream_ptr);
yading@10:                 for (x = 0; x < 8; x++, flags >>= 2)
yading@10:                     *pixel_ptr++ = P[flags & 0x03];
yading@10:                 pixel_ptr += s->line_inc;
yading@10:             }
yading@10: 
yading@10:         } else {
yading@10:             uint32_t flags;
yading@10: 
yading@10:             /* 1 of 4 colors for each 2x2 block */
yading@10:             flags = bytestream2_get_le32(&s->stream_ptr);
yading@10: 
yading@10:             for (y = 0; y < 8; y += 2) {
yading@10:                 for (x = 0; x < 8; x += 2, flags >>= 2) {
yading@10:                     pixel_ptr[x                ] =
yading@10:                     pixel_ptr[x + 1            ] =
yading@10:                     pixel_ptr[x +     s->stride] =
yading@10:                     pixel_ptr[x + 1 + s->stride] = P[flags & 0x03];
yading@10:                 }
yading@10:                 pixel_ptr += s->stride * 2;
yading@10:             }
yading@10: 
yading@10:         }
yading@10:     } else {
yading@10:         uint64_t flags;
yading@10: 
yading@10:         /* 1 of 4 colors for each 2x1 or 1x2 block */
yading@10:         flags = bytestream2_get_le64(&s->stream_ptr);
yading@10:         if (!(P[2] & 0x8000)) {
yading@10:             for (y = 0; y < 8; y++) {
yading@10:                 for (x = 0; x < 8; x += 2, flags >>= 2) {
yading@10:                     pixel_ptr[x    ] =
yading@10:                     pixel_ptr[x + 1] = P[flags & 0x03];
yading@10:                 }
yading@10:                 pixel_ptr += s->stride;
yading@10:             }
yading@10:         } else {
yading@10:             for (y = 0; y < 8; y += 2) {
yading@10:                 for (x = 0; x < 8; x++, flags >>= 2) {
yading@10:                     pixel_ptr[x            ] =
yading@10:                     pixel_ptr[x + s->stride] = P[flags & 0x03];
yading@10:                 }
yading@10:                 pixel_ptr += s->stride * 2;
yading@10:             }
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xA_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     uint16_t P[8];
yading@10:     int flags = 0;
yading@10:     uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;
yading@10: 
yading@10:     for (x = 0; x < 4; x++)
yading@10:         P[x] = bytestream2_get_le16(&s->stream_ptr);
yading@10: 
yading@10:     /* 4-color encoding for each 4x4 quadrant, or 4-color encoding on
yading@10:      * either top and bottom or left and right halves */
yading@10:     if (!(P[0] & 0x8000)) {
yading@10: 
yading@10:         /* 4-color encoding for each quadrant */
yading@10:         for (y = 0; y < 16; y++) {
yading@10:             // new values for each 4x4 block
yading@10:             if (!(y & 3)) {
yading@10:                 if (y)
yading@10:                     for (x = 0; x < 4; x++)
yading@10:                         P[x] = bytestream2_get_le16(&s->stream_ptr);
yading@10:                 flags = bytestream2_get_le32(&s->stream_ptr);
yading@10:             }
yading@10: 
yading@10:             for (x = 0; x < 4; x++, flags >>= 2)
yading@10:                 *pixel_ptr++ = P[flags & 0x03];
yading@10: 
yading@10:             pixel_ptr += s->stride - 4;
yading@10:             // switch to right half
yading@10:             if (y == 7) pixel_ptr -= 8 * s->stride - 4;
yading@10:         }
yading@10: 
yading@10:     } else {
yading@10:         // vertical split?
yading@10:         int vert;
yading@10:         uint64_t flags = bytestream2_get_le64(&s->stream_ptr);
yading@10: 
yading@10:         for (x = 4; x < 8; x++)
yading@10:             P[x] = bytestream2_get_le16(&s->stream_ptr);
yading@10:         vert = !(P[4] & 0x8000);
yading@10: 
yading@10:         /* 4-color encoding for either left and right or top and bottom
yading@10:          * halves */
yading@10: 
yading@10:         for (y = 0; y < 16; y++) {
yading@10:             for (x = 0; x < 4; x++, flags >>= 2)
yading@10:                 *pixel_ptr++ = P[flags & 0x03];
yading@10: 
yading@10:             if (vert) {
yading@10:                 pixel_ptr += s->stride - 4;
yading@10:                 // switch to right half
yading@10:                 if (y == 7) pixel_ptr -= 8 * s->stride - 4;
yading@10:             } else if (y & 1) pixel_ptr += s->line_inc;
yading@10: 
yading@10:             // load values for second half
yading@10:             if (y == 7) {
yading@10:                 memcpy(P, P + 4, 8);
yading@10:                 flags = bytestream2_get_le64(&s->stream_ptr);
yading@10:             }
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xB_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;
yading@10: 
yading@10:     /* 64-color encoding (each pixel in block is a different color) */
yading@10:     for (y = 0; y < 8; y++) {
yading@10:         for (x = 0; x < 8; x++)
yading@10:             pixel_ptr[x] = bytestream2_get_le16(&s->stream_ptr);
yading@10:         pixel_ptr  += s->stride;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xC_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;
yading@10: 
yading@10:     /* 16-color block encoding: each 2x2 block is a different color */
yading@10:     for (y = 0; y < 8; y += 2) {
yading@10:         for (x = 0; x < 8; x += 2) {
yading@10:             pixel_ptr[x                ] =
yading@10:             pixel_ptr[x + 1            ] =
yading@10:             pixel_ptr[x +     s->stride] =
yading@10:             pixel_ptr[x + 1 + s->stride] = bytestream2_get_le16(&s->stream_ptr);
yading@10:         }
yading@10:         pixel_ptr += s->stride * 2;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xD_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     uint16_t P[2];
yading@10:     uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;
yading@10: 
yading@10:     /* 4-color block encoding: each 4x4 block is a different color */
yading@10:     for (y = 0; y < 8; y++) {
yading@10:         if (!(y & 3)) {
yading@10:             P[0] = bytestream2_get_le16(&s->stream_ptr);
yading@10:             P[1] = bytestream2_get_le16(&s->stream_ptr);
yading@10:         }
yading@10:         for (x = 0; x < 8; x++)
yading@10:             pixel_ptr[x] = P[x >> 2];
yading@10:         pixel_ptr += s->stride;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_decode_block_opcode_0xE_16(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     uint16_t pix;
yading@10:     uint16_t *pixel_ptr = (uint16_t*)s->pixel_ptr;
yading@10: 
yading@10:     /* 1-color encoding: the whole block is 1 solid color */
yading@10:     pix = bytestream2_get_le16(&s->stream_ptr);
yading@10: 
yading@10:     for (y = 0; y < 8; y++) {
yading@10:         for (x = 0; x < 8; x++)
yading@10:             pixel_ptr[x] = pix;
yading@10:         pixel_ptr += s->stride;
yading@10:     }
yading@10: 
yading@10:     /* report success */
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int (* const ipvideo_decode_block[])(IpvideoContext *s, AVFrame *frame) = {
yading@10:     ipvideo_decode_block_opcode_0x0, ipvideo_decode_block_opcode_0x1,
yading@10:     ipvideo_decode_block_opcode_0x2, ipvideo_decode_block_opcode_0x3,
yading@10:     ipvideo_decode_block_opcode_0x4, ipvideo_decode_block_opcode_0x5,
yading@10:     ipvideo_decode_block_opcode_0x6, ipvideo_decode_block_opcode_0x7,
yading@10:     ipvideo_decode_block_opcode_0x8, ipvideo_decode_block_opcode_0x9,
yading@10:     ipvideo_decode_block_opcode_0xA, ipvideo_decode_block_opcode_0xB,
yading@10:     ipvideo_decode_block_opcode_0xC, ipvideo_decode_block_opcode_0xD,
yading@10:     ipvideo_decode_block_opcode_0xE, ipvideo_decode_block_opcode_0xF,
yading@10: };
yading@10: 
yading@10: static int (* const ipvideo_decode_block16[])(IpvideoContext *s, AVFrame *frame) = {
yading@10:     ipvideo_decode_block_opcode_0x0,    ipvideo_decode_block_opcode_0x1,
yading@10:     ipvideo_decode_block_opcode_0x2,    ipvideo_decode_block_opcode_0x3,
yading@10:     ipvideo_decode_block_opcode_0x4,    ipvideo_decode_block_opcode_0x5,
yading@10:     ipvideo_decode_block_opcode_0x6_16, ipvideo_decode_block_opcode_0x7_16,
yading@10:     ipvideo_decode_block_opcode_0x8_16, ipvideo_decode_block_opcode_0x9_16,
yading@10:     ipvideo_decode_block_opcode_0xA_16, ipvideo_decode_block_opcode_0xB_16,
yading@10:     ipvideo_decode_block_opcode_0xC_16, ipvideo_decode_block_opcode_0xD_16,
yading@10:     ipvideo_decode_block_opcode_0xE_16, ipvideo_decode_block_opcode_0x1,
yading@10: };
yading@10: 
yading@10: static void ipvideo_decode_opcodes(IpvideoContext *s, AVFrame *frame)
yading@10: {
yading@10:     int x, y;
yading@10:     unsigned char opcode;
yading@10:     int ret;
yading@10:     GetBitContext gb;
yading@10: 
yading@10:     bytestream2_skip(&s->stream_ptr, 14); /* data starts 14 bytes in */
yading@10:     if (!s->is_16bpp) {
yading@10:         /* this is PAL8, so make the palette available */
yading@10:         memcpy(frame->data[1], s->pal, AVPALETTE_SIZE);
yading@10: 
yading@10:         s->stride = frame->linesize[0];
yading@10:     } else {
yading@10:         s->stride = frame->linesize[0] >> 1;
yading@10:         s->mv_ptr = s->stream_ptr;
yading@10:         bytestream2_skip(&s->mv_ptr, bytestream2_get_le16(&s->stream_ptr));
yading@10:     }
yading@10:     s->line_inc = s->stride - 8;
yading@10:     s->upper_motion_limit_offset = (s->avctx->height - 8) * frame->linesize[0]
yading@10:                                   + (s->avctx->width - 8) * (1 + s->is_16bpp);
yading@10: 
yading@10:     init_get_bits(&gb, s->decoding_map, s->decoding_map_size * 8);
yading@10:     for (y = 0; y < s->avctx->height; y += 8) {
yading@10:         for (x = 0; x < s->avctx->width; x += 8) {
yading@10:             opcode = get_bits(&gb, 4);
yading@10: 
yading@10:             av_dlog(s->avctx,
yading@10:                     "  block @ (%3d, %3d): encoding 0x%X, data ptr offset %d\n",
yading@10:                     x, y, opcode, bytestream2_tell(&s->stream_ptr));
yading@10: 
yading@10:             if (!s->is_16bpp) {
yading@10:                 s->pixel_ptr = frame->data[0] + x
yading@10:                               + y*frame->linesize[0];
yading@10:                 ret = ipvideo_decode_block[opcode](s, frame);
yading@10:             } else {
yading@10:                 s->pixel_ptr = frame->data[0] + x*2
yading@10:                               + y*frame->linesize[0];
yading@10:                 ret = ipvideo_decode_block16[opcode](s, frame);
yading@10:             }
yading@10:             if (ret != 0) {
yading@10:                 av_log(s->avctx, AV_LOG_ERROR, "decode problem on frame %d, @ block (%d, %d)\n",
yading@10:                        s->avctx->frame_number, x, y);
yading@10:                 return;
yading@10:             }
yading@10:         }
yading@10:     }
yading@10:     if (bytestream2_get_bytes_left(&s->stream_ptr) > 1) {
yading@10:         av_log(s->avctx, AV_LOG_ERROR,
yading@10:                "decode finished with %d bytes left over\n",
yading@10:                bytestream2_get_bytes_left(&s->stream_ptr));
yading@10:     }
yading@10: }
yading@10: 
yading@10: static av_cold int ipvideo_decode_init(AVCodecContext *avctx)
yading@10: {
yading@10:     IpvideoContext *s = avctx->priv_data;
yading@10: 
yading@10:     s->avctx = avctx;
yading@10: 
yading@10:     s->is_16bpp = avctx->bits_per_coded_sample == 16;
yading@10:     avctx->pix_fmt = s->is_16bpp ? AV_PIX_FMT_RGB555 : AV_PIX_FMT_PAL8;
yading@10: 
yading@10:     ff_hpeldsp_init(&s->hdsp, avctx->flags);
yading@10: 
yading@10:     s->last_frame        = av_frame_alloc();
yading@10:     s->second_last_frame = av_frame_alloc();
yading@10:     if (!s->last_frame || !s->second_last_frame) {
yading@10:         av_frame_free(&s->last_frame);
yading@10:         av_frame_free(&s->second_last_frame);
yading@10:         return AVERROR(ENOMEM);
yading@10:     }
yading@10: 
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int ipvideo_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:     IpvideoContext *s = avctx->priv_data;
yading@10:     AVFrame *frame = data;
yading@10:     int ret;
yading@10: 
yading@10:     /* decoding map contains 4 bits of information per 8x8 block */
yading@10:     s->decoding_map_size = avctx->width * avctx->height / (8 * 8 * 2);
yading@10: 
yading@10:     /* compressed buffer needs to be large enough to at least hold an entire
yading@10:      * decoding map */
yading@10:     if (buf_size < s->decoding_map_size)
yading@10:         return buf_size;
yading@10: 
yading@10:     if (av_packet_get_side_data(avpkt, AV_PKT_DATA_PARAM_CHANGE, NULL)) {
yading@10:         av_frame_unref(s->last_frame);
yading@10:         av_frame_unref(s->second_last_frame);
yading@10:     }
yading@10: 
yading@10:     s->decoding_map = buf;
yading@10:     bytestream2_init(&s->stream_ptr, buf + s->decoding_map_size,
yading@10:                      buf_size - s->decoding_map_size);
yading@10: 
yading@10:     if ((ret = ff_get_buffer(avctx, frame, AV_GET_BUFFER_FLAG_REF)) < 0)
yading@10:         return ret;
yading@10: 
yading@10:     if (!s->is_16bpp) {
yading@10:         const uint8_t *pal = av_packet_get_side_data(avpkt, AV_PKT_DATA_PALETTE, NULL);
yading@10:         if (pal) {
yading@10:             frame->palette_has_changed = 1;
yading@10:             memcpy(s->pal, pal, AVPALETTE_SIZE);
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     ipvideo_decode_opcodes(s, frame);
yading@10: 
yading@10:     *got_frame = 1;
yading@10: 
yading@10:     /* shuffle frames */
yading@10:     av_frame_unref(s->second_last_frame);
yading@10:     FFSWAP(AVFrame*, s->second_last_frame, s->last_frame);
yading@10:     if ((ret = av_frame_ref(s->last_frame, frame)) < 0)
yading@10:         return ret;
yading@10: 
yading@10:     /* report that the buffer was completely consumed */
yading@10:     return buf_size;
yading@10: }
yading@10: 
yading@10: static av_cold int ipvideo_decode_end(AVCodecContext *avctx)
yading@10: {
yading@10:     IpvideoContext *s = avctx->priv_data;
yading@10: 
yading@10:     av_frame_free(&s->last_frame);
yading@10:     av_frame_free(&s->second_last_frame);
yading@10: 
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: AVCodec ff_interplay_video_decoder = {
yading@10:     .name           = "interplayvideo",
yading@10:     .type           = AVMEDIA_TYPE_VIDEO,
yading@10:     .id             = AV_CODEC_ID_INTERPLAY_VIDEO,
yading@10:     .priv_data_size = sizeof(IpvideoContext),
yading@10:     .init           = ipvideo_decode_init,
yading@10:     .close          = ipvideo_decode_end,
yading@10:     .decode         = ipvideo_decode_frame,
yading@10:     .capabilities   = CODEC_CAP_DR1 | CODEC_CAP_PARAM_CHANGE,
yading@10:     .long_name      = NULL_IF_CONFIG_SMALL("Interplay MVE video"),
yading@10: };