git.sesse.net Git - ffmpeg/blob - libavcodec/tmv.c

   1 /*
   2  * 8088flex TMV video decoder
   3  * Copyright (c) 2009 Daniel Verkamp <daniel at drv.nu>
   4  *
   5  * This file is part of Libav.
   6  *
   7  * Libav is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * Libav is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with Libav; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 /**
  23  * @file
  24  * 8088flex TMV video decoder
  25  * @author Daniel Verkamp
  26  * @see http://www.oldskool.org/pc/8088_Corruption
  27  */
  28
  29 #include <string.h>
  30
  31 #include "avcodec.h"
  32 #include "internal.h"
  33 #include "libavutil/internal.h"
  34
  35 #include "cga_data.h"
  36
  37 typedef struct TMVContext {
  38     AVFrame pic;
  39 } TMVContext;
  40
  41 static int tmv_decode_frame(AVCodecContext *avctx, void *data,
  42                             int *got_frame, AVPacket *avpkt)
  43 {
  44     TMVContext *tmv    = avctx->priv_data;
  45     const uint8_t *src = avpkt->data;
  46     uint8_t *dst;
  47     unsigned char_cols = avctx->width >> 3;
  48     unsigned char_rows = avctx->height >> 3;
  49     unsigned x, y, fg, bg, c;
  50     int ret;
  51
  52     if (tmv->pic.data[0])
  53         avctx->release_buffer(avctx, &tmv->pic);
  54
  55     if ((ret = ff_get_buffer(avctx, &tmv->pic)) < 0) {
  56         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
  57         return ret;
  58     }
  59
  60     if (avpkt->size < 2*char_rows*char_cols) {
  61         av_log(avctx, AV_LOG_ERROR,
  62                "Input buffer too small, truncated sample?\n");
  63         *got_frame = 0;
  64         return AVERROR_INVALIDDATA;
  65     }
  66
  67     tmv->pic.pict_type = AV_PICTURE_TYPE_I;
  68     tmv->pic.key_frame = 1;
  69     dst                = tmv->pic.data[0];
  70
  71     tmv->pic.palette_has_changed = 1;
  72     memcpy(tmv->pic.data[1], ff_cga_palette, 16 * 4);
  73
  74     for (y = 0; y < char_rows; y++) {
  75         for (x = 0; x < char_cols; x++) {
  76             c  = *src++;
  77             bg = *src  >> 4;
  78             fg = *src++ & 0xF;
  79             ff_draw_pc_font(dst + x * 8, tmv->pic.linesize[0],
  80                             ff_cga_font, 8, c, fg, bg);
  81         }
  82         dst += tmv->pic.linesize[0] * 8;
  83     }
  84
  85     *got_frame = 1;
  86     *(AVFrame *)data = tmv->pic;
  87     return avpkt->size;
  88 }
  89
  90 static av_cold int tmv_decode_init(AVCodecContext *avctx)
  91 {
  92     avctx->pix_fmt = AV_PIX_FMT_PAL8;
  93     return 0;
  94 }
  95
  96 static av_cold int tmv_decode_close(AVCodecContext *avctx)
  97 {
  98     TMVContext *tmv = avctx->priv_data;
  99
 100     if (tmv->pic.data[0])
 101         avctx->release_buffer(avctx, &tmv->pic);
 102
 103     return 0;
 104 }
 105
 106 AVCodec ff_tmv_decoder = {
 107     .name           = "tmv",
 108     .type           = AVMEDIA_TYPE_VIDEO,
 109     .id             = AV_CODEC_ID_TMV,
 110     .priv_data_size = sizeof(TMVContext),
 111     .init           = tmv_decode_init,
 112     .close          = tmv_decode_close,
 113     .decode         = tmv_decode_frame,
 114     .capabilities   = CODEC_CAP_DR1,
 115     .long_name      = NULL_IF_CONFIG_SMALL("8088flex TMV"),
 116 };