yading@10: /*
yading@10:  * Silicon Graphics Motion Video Compressor 1 & 2 decoder
yading@10:  * Copyright (c) 2012 Peter Ross
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:  * Silicon Graphics Motion Video Compressor 1 & 2 decoder
yading@10:  */
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: 
yading@10: typedef struct MvcContext {
yading@10:     AVFrame *frame;
yading@10:     int vflip;
yading@10: } MvcContext;
yading@10: 
yading@10: static av_cold int mvc_decode_init(AVCodecContext *avctx)
yading@10: {
yading@10:     MvcContext *s = avctx->priv_data;
yading@10:     int width  = avctx->width;
yading@10:     int height = avctx->height;
yading@10: 
yading@10:     if (avctx->codec_id == AV_CODEC_ID_MVC1) {
yading@10:         width  += 3;
yading@10:         height += 3;
yading@10:     }
yading@10:     width  &= ~3;
yading@10:     height &= ~3;
yading@10:     if (width != avctx->width || height != avctx->height)
yading@10:         avcodec_set_dimensions(avctx, width, height);
yading@10: 
yading@10:     avctx->pix_fmt = (avctx->codec_id == AV_CODEC_ID_MVC1) ? AV_PIX_FMT_RGB555 : AV_PIX_FMT_BGRA;
yading@10:     s->frame = av_frame_alloc();
yading@10:     if (!s->frame)
yading@10:         return AVERROR(ENOMEM);
yading@10: 
yading@10:     s->vflip = avctx->extradata_size >= 9 && !memcmp(avctx->extradata + avctx->extradata_size - 9, "BottomUp", 9);
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int decode_mvc1(AVCodecContext *avctx, GetByteContext *gb, uint8_t *dst_start, int width, int height, int linesize)
yading@10: {
yading@10:     uint8_t *dst;
yading@10:     uint16_t v[8];
yading@10:     int mask, x, y, i;
yading@10: 
yading@10:     x = y= 0;
yading@10:     while (bytestream2_get_bytes_left(gb) >= 6) {
yading@10:         mask = bytestream2_get_be16u(gb);
yading@10:         v[0] = bytestream2_get_be16u(gb);
yading@10:         v[1] = bytestream2_get_be16u(gb);
yading@10:         if ((v[0] & 0x8000)) {
yading@10:             if (bytestream2_get_bytes_left(gb) < 12) {
yading@10:                 av_log(avctx, AV_LOG_WARNING, "buffer overflow\n");
yading@10:                 return AVERROR_INVALIDDATA;
yading@10:             }
yading@10:             for (i = 2; i < 8; i++)
yading@10:                 v[i] = bytestream2_get_be16u(gb);
yading@10:         } else {
yading@10:             v[2] = v[4] = v[6] = v[0];
yading@10:             v[3] = v[5] = v[7] = v[1];
yading@10:         }
yading@10: 
yading@10: #define PIX16(target, true, false) \
yading@10:         i = (mask & target) ? true : false; \
yading@10:         AV_WN16A(dst, (v[i] & 0x7C00) | (v[i] & 0x3E0) | (v[i] & 0x1F)); \
yading@10:         dst += 2;
yading@10: 
yading@10: #define ROW16(row, a1, a0, b1, b0) \
yading@10:         dst = dst_start + (y + row) * linesize + x * 2; \
yading@10:         PIX16(1 << (row * 4),     a1, a0) \
yading@10:         PIX16(1 << (row * 4 + 1), a1, a0) \
yading@10:         PIX16(1 << (row * 4 + 2), b1, b0) \
yading@10:         PIX16(1 << (row * 4 + 3), b1, b0)
yading@10: 
yading@10:         ROW16(0, 0, 1, 2, 3);
yading@10:         ROW16(1, 0, 1, 2, 3);
yading@10:         ROW16(2, 4, 5, 6, 7);
yading@10:         ROW16(3, 4, 5, 6, 7);
yading@10: 
yading@10:         x += 4;
yading@10:         if (x >= width) {
yading@10:             y += 4;
yading@10:             if (y >= height) {
yading@10:                 break;
yading@10:             }
yading@10:             x = 0;
yading@10:         }
yading@10:     }
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static void set_4x4_block(uint8_t *dst, int linesize, uint32_t pixel)
yading@10: {
yading@10:     int i, j;
yading@10:     for (j = 0; j < 4; j++)
yading@10:         for (i = 0; i < 4; i++)
yading@10:             AV_WN32A(dst + j * linesize + i * 4, pixel);
yading@10: }
yading@10: 
yading@10: #define PIX32(target, true, false) \
yading@10:     AV_WN32A(dst, (mask & target) ? v[true] : v[false]); \
yading@10:     dst += 4;
yading@10: 
yading@10: #define ROW32(row, a1, a0, b1, b0) \
yading@10:     dst = dst_start + (y + row) * linesize + x * 4; \
yading@10:     PIX32(1 << (row * 4),     a1, a0) \
yading@10:     PIX32(1 << (row * 4 + 1), a1, a0) \
yading@10:     PIX32(1 << (row * 4 + 2), b1, b0) \
yading@10:     PIX32(1 << (row * 4 + 3), b1, b0)
yading@10: 
yading@10: #define MVC2_BLOCK \
yading@10:     ROW32(0, 1, 0, 3, 2); \
yading@10:     ROW32(1, 1, 0, 3, 2); \
yading@10:     ROW32(2, 5, 4, 7, 6); \
yading@10:     ROW32(3, 5, 4, 7, 6);
yading@10: 
yading@10: static int decode_mvc2(AVCodecContext *avctx, GetByteContext *gb, uint8_t *dst_start, int width, int height, int linesize, int vflip)
yading@10: {
yading@10:     uint8_t *dst;
yading@10:     uint32_t color[128], v[8];
yading@10:     int w, h, nb_colors, i, x, y, p0, p1, mask;
yading@10: 
yading@10:     if (bytestream2_get_bytes_left(gb) < 6)
yading@10:         return AVERROR_INVALIDDATA;
yading@10: 
yading@10:     w = bytestream2_get_be16u(gb);
yading@10:     h = bytestream2_get_be16u(gb);
yading@10:     if ((w & ~3) != width || (h & ~3) != height)
yading@10:         av_log(avctx, AV_LOG_WARNING, "dimension mismatch\n");
yading@10: 
yading@10:     if (bytestream2_get_byteu(gb)) {
yading@10:         avpriv_request_sample(avctx, "bitmap feature");
yading@10:         return AVERROR_PATCHWELCOME;
yading@10:     }
yading@10: 
yading@10:     nb_colors = bytestream2_get_byteu(gb);
yading@10:     if (bytestream2_get_bytes_left(gb) < nb_colors * 3)
yading@10:         return AVERROR_INVALIDDATA;
yading@10:     for (i = 0; i < FFMIN(nb_colors, 128); i++)
yading@10:         color[i] = 0xFF000000 | bytestream2_get_be24u(gb);
yading@10:     if (nb_colors > 128)
yading@10:         bytestream2_skip(gb, (nb_colors - 128) * 3);
yading@10: 
yading@10:     if (vflip) {
yading@10:         dst_start += (height - 1) * linesize;
yading@10:         linesize = -linesize;
yading@10:     }
yading@10:     x = y = 0;
yading@10:     while (bytestream2_get_bytes_left(gb) >= 1) {
yading@10:         p0 = bytestream2_get_byteu(gb);
yading@10:         if ((p0 & 0x80)) {
yading@10:             if ((p0 & 0x40)) {
yading@10:                 p0 &= 0x3F;
yading@10:                 p0 = (p0 << 2) | (p0 >> 4);
yading@10:                 set_4x4_block(dst_start + y * linesize + x * 4, linesize, 0xFF000000 | (p0 << 16) | (p0 << 8) | p0);
yading@10:             } else {
yading@10:                 int g, r;
yading@10:                 p0 &= 0x3F;
yading@10:                 p0 = (p0 << 2) | (p0 >> 4);
yading@10:                 if (bytestream2_get_bytes_left(gb) < 2)
yading@10:                     return AVERROR_INVALIDDATA;
yading@10:                 g = bytestream2_get_byteu(gb);
yading@10:                 r = bytestream2_get_byteu(gb);
yading@10:                 set_4x4_block(dst_start + y * linesize + x * 4, linesize, 0xFF000000 | (r << 16) | (g << 8) | p0);
yading@10:             }
yading@10:         } else {
yading@10:             if (bytestream2_get_bytes_left(gb) < 1)
yading@10:                 return AVERROR_INVALIDDATA;
yading@10:             p1 = bytestream2_get_byteu(gb);
yading@10:             if ((p1 & 0x80)) {
yading@10:                 if ((p0 & 0x7F) == (p1 & 0x7F)) {
yading@10:                     set_4x4_block(dst_start + y * linesize + x * 4, linesize, color[p0 & 0x7F]);
yading@10:                 } else {
yading@10:                     if (bytestream2_get_bytes_left(gb) < 2)
yading@10:                         return AVERROR_INVALIDDATA;
yading@10:                     v[0] = v[2] = v[4] = v[6] = color[p0 & 0x7F];
yading@10:                     v[1] = v[3] = v[5] = v[7] = color[p1 & 0x7F];
yading@10:                     mask = bytestream2_get_le16u(gb);
yading@10:                     MVC2_BLOCK
yading@10:                 }
yading@10:             } else {
yading@10:                 if (bytestream2_get_bytes_left(gb) < 8)
yading@10:                     return AVERROR_INVALIDDATA;
yading@10:                 v[0] = color[p0 & 0x7F];
yading@10:                 v[1] = color[p1 & 0x7F];
yading@10:                 for (i = 2; i < 8; i++)
yading@10:                     v[i] = color[bytestream2_get_byteu(gb) & 0x7F];
yading@10:                 mask = bytestream2_get_le16u(gb);
yading@10:                 MVC2_BLOCK
yading@10:             }
yading@10:         }
yading@10: 
yading@10:         x += 4;
yading@10:         if (x >= width) {
yading@10:             y += 4;
yading@10:             if (y >= height)
yading@10:                 break;
yading@10:             x = 0;
yading@10:         }
yading@10:     }
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: static int mvc_decode_frame(AVCodecContext *avctx,
yading@10:                             void *data, int *got_frame,
yading@10:                             AVPacket *avpkt)
yading@10: {
yading@10:     MvcContext *s = avctx->priv_data;
yading@10:     GetByteContext gb;
yading@10:     int ret;
yading@10: 
yading@10:     if ((ret = ff_reget_buffer(avctx, s->frame)) < 0)
yading@10:         return ret;
yading@10: 
yading@10:     bytestream2_init(&gb, avpkt->data, avpkt->size);
yading@10:     if (avctx->codec_id == AV_CODEC_ID_MVC1)
yading@10:         ret = decode_mvc1(avctx, &gb, s->frame->data[0], avctx->width, avctx->height, s->frame->linesize[0]);
yading@10:     else
yading@10:         ret = decode_mvc2(avctx, &gb, s->frame->data[0], avctx->width, avctx->height, s->frame->linesize[0], s->vflip);
yading@10:     if (ret < 0)
yading@10:         return ret;
yading@10: 
yading@10:     *got_frame = 1;
yading@10:     if ((ret = av_frame_ref(data, s->frame)) < 0)
yading@10:         return ret;
yading@10: 
yading@10:     return avpkt->size;
yading@10: }
yading@10: 
yading@10: static av_cold int mvc_decode_end(AVCodecContext *avctx)
yading@10: {
yading@10:     MvcContext *s = avctx->priv_data;
yading@10: 
yading@10:     av_frame_free(&s->frame);
yading@10: 
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: #if CONFIG_MVC1_DECODER
yading@10: AVCodec ff_mvc1_decoder = {
yading@10:     .name           = "mvc1",
yading@10:     .type           = AVMEDIA_TYPE_VIDEO,
yading@10:     .id             = AV_CODEC_ID_MVC1,
yading@10:     .priv_data_size = sizeof(MvcContext),
yading@10:     .init           = mvc_decode_init,
yading@10:     .close          = mvc_decode_end,
yading@10:     .decode         = mvc_decode_frame,
yading@10:     .capabilities   = CODEC_CAP_DR1,
yading@10:     .long_name      = NULL_IF_CONFIG_SMALL("Silicon Graphics Motion Video Compressor 1"),
yading@10: };
yading@10: #endif
yading@10: 
yading@10: #if CONFIG_MVC2_DECODER
yading@10: AVCodec ff_mvc2_decoder = {
yading@10:     .name           = "mvc2",
yading@10:     .type           = AVMEDIA_TYPE_VIDEO,
yading@10:     .id             = AV_CODEC_ID_MVC2,
yading@10:     .priv_data_size = sizeof(MvcContext),
yading@10:     .init           = mvc_decode_init,
yading@10:     .close          = mvc_decode_end,
yading@10:     .decode         = mvc_decode_frame,
yading@10:     .capabilities   = CODEC_CAP_DR1,
yading@10:     .long_name      = NULL_IF_CONFIG_SMALL("Silicon Graphics Motion Video Compressor 2"),
yading@10: };
yading@10: #endif