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

   1 /*
   2  * Binary text decoder
   3  * eXtended BINary text (XBIN) decoder
   4  * iCEDraw File decoder
   5  * Copyright (c) 2010 Peter Ross (pross@xvid.org)
   6  *
   7  * This file is part of FFmpeg.
   8  *
   9  * FFmpeg is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU Lesser General Public
  11  * License as published by the Free Software Foundation; either
  12  * version 2.1 of the License, or (at your option) any later version.
  13  *
  14  * FFmpeg is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17  * Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public
  20  * License along with FFmpeg; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  22  */
  23
  24 /**
  25  * @file
  26  * Binary text decoder
  27  * eXtended BINary text (XBIN) decoder
  28  * iCEDraw File decoder
  29  */
  30
  31 #include "libavutil/intreadwrite.h"
  32 #include "libavutil/xga_font_data.h"
  33 #include "avcodec.h"
  34 #include "cga_data.h"
  35 #include "bintext.h"
  36 #include "internal.h"
  37
  38 typedef struct XbinContext {
  39     AVFrame *frame;
  40     int palette[16];
  41     int flags;
  42     int font_height;
  43     const uint8_t *font;
  44     int x, y;
  45 } XbinContext;
  46
  47 static av_cold int decode_init(AVCodecContext *avctx)
  48 {
  49     XbinContext *s = avctx->priv_data;
  50     uint8_t *p;
  51     int i;
  52
  53     avctx->pix_fmt = AV_PIX_FMT_PAL8;
  54     p = avctx->extradata;
  55     if (p) {
  56         s->font_height = p[0];
  57         s->flags = p[1];
  58         p += 2;
  59         if(avctx->extradata_size < 2 + (!!(s->flags & BINTEXT_PALETTE))*3*16
  60                                      + (!!(s->flags & BINTEXT_FONT))*s->font_height*256) {
  61             av_log(avctx, AV_LOG_ERROR, "not enough extradata\n");
  62             return AVERROR_INVALIDDATA;
  63         }
  64     } else {
  65         s->font_height = 8;
  66         s->flags = 0;
  67     }
  68
  69     if ((s->flags & BINTEXT_PALETTE)) {
  70         for (i = 0; i < 16; i++) {
  71             s->palette[i] = 0xFF000000 | (AV_RB24(p) << 2) | ((AV_RB24(p) >> 4) & 0x30303);
  72             p += 3;
  73         }
  74     } else {
  75         for (i = 0; i < 16; i++)
  76             s->palette[i] = 0xFF000000 | ff_cga_palette[i];
  77     }
  78
  79     if ((s->flags & BINTEXT_FONT)) {
  80         s->font = p;
  81     } else {
  82         switch(s->font_height) {
  83         default:
  84             av_log(avctx, AV_LOG_WARNING, "font height %i not supported\n", s->font_height);
  85             s->font_height = 8;
  86         case 8:
  87             s->font = avpriv_cga_font;
  88             break;
  89         case 16:
  90             s->font = avpriv_vga16_font;
  91             break;
  92         }
  93     }
  94
  95     s->frame = av_frame_alloc();
  96     if (!s->frame)
  97         return AVERROR(ENOMEM);
  98
  99     return 0;
 100 }
 101
 102 #define DEFAULT_BG_COLOR 0
 103 av_unused static void hscroll(AVCodecContext *avctx)
 104 {
 105     XbinContext *s = avctx->priv_data;
 106     if (s->y < avctx->height - s->font_height) {
 107         s->y += s->font_height;
 108     } else {
 109         memmove(s->frame->data[0], s->frame->data[0] + s->font_height*s->frame->linesize[0],
 110             (avctx->height - s->font_height)*s->frame->linesize[0]);
 111         memset(s->frame->data[0] + (avctx->height - s->font_height)*s->frame->linesize[0],
 112             DEFAULT_BG_COLOR, s->font_height * s->frame->linesize[0]);
 113     }
 114 }
 115
 116 #define FONT_WIDTH 8
 117
 118 /**
 119  * Draw character to screen
 120  */
 121 static void draw_char(AVCodecContext *avctx, int c, int a)
 122 {
 123     XbinContext *s = avctx->priv_data;
 124     if (s->y > avctx->height - s->font_height)
 125         return;
 126     ff_draw_pc_font(s->frame->data[0] + s->y * s->frame->linesize[0] + s->x,
 127                     s->frame->linesize[0], s->font, s->font_height, c,
 128                     a & 0x0F, a >> 4);
 129     s->x += FONT_WIDTH;
 130     if (s->x > avctx->width - FONT_WIDTH) {
 131         s->x = 0;
 132         s->y += s->font_height;
 133     }
 134 }
 135
 136 static int decode_frame(AVCodecContext *avctx,
 137                             void *data, int *got_frame,
 138                             AVPacket *avpkt)
 139 {
 140     XbinContext *s = avctx->priv_data;
 141     const uint8_t *buf = avpkt->data;
 142     int buf_size = avpkt->size;
 143     const uint8_t *buf_end = buf+buf_size;
 144     int ret;
 145
 146     s->x = s->y = 0;
 147     if ((ret = ff_reget_buffer(avctx, s->frame)) < 0)
 148         return ret;
 149     s->frame->pict_type           = AV_PICTURE_TYPE_I;
 150     s->frame->palette_has_changed = 1;
 151     memcpy(s->frame->data[1], s->palette, 16 * 4);
 152
 153     if (avctx->codec_id == AV_CODEC_ID_XBIN) {
 154         while (buf + 2 < buf_end) {
 155             int i,c,a;
 156             int type  = *buf >> 6;
 157             int count = (*buf & 0x3F) + 1;
 158             buf++;
 159             switch (type) {
 160             case 0: //no compression
 161                 for (i = 0; i < count && buf + 1 < buf_end; i++) {
 162                     draw_char(avctx, buf[0], buf[1]);
 163                     buf += 2;
 164                 }
 165                 break;
 166             case 1: //character compression
 167                 c = *buf++;
 168                 for (i = 0; i < count && buf < buf_end; i++)
 169                     draw_char(avctx, c, *buf++);
 170                 break;
 171             case 2: //attribute compression
 172                 a = *buf++;
 173                 for (i = 0; i < count && buf < buf_end; i++)
 174                     draw_char(avctx, *buf++, a);
 175                 break;
 176             case 3: //character/attribute compression
 177                 c = *buf++;
 178                 a = *buf++;
 179                 for (i = 0; i < count && buf < buf_end; i++)
 180                     draw_char(avctx, c, a);
 181                 break;
 182             }
 183         }
 184     } else if (avctx->codec_id == AV_CODEC_ID_IDF) {
 185         while (buf + 2 < buf_end) {
 186             if (AV_RL16(buf) == 1) {
 187                int i;
 188                if (buf + 6 > buf_end)
 189                    break;
 190                for (i = 0; i < buf[2]; i++)
 191                    draw_char(avctx, buf[4], buf[5]);
 192                buf += 6;
 193             } else {
 194                draw_char(avctx, buf[0], buf[1]);
 195                buf += 2;
 196             }
 197         }
 198     } else {
 199         while (buf + 1 < buf_end) {
 200             draw_char(avctx, buf[0], buf[1]);
 201             buf += 2;
 202         }
 203     }
 204
 205     if ((ret = av_frame_ref(data, s->frame)) < 0)
 206         return ret;
 207     *got_frame      = 1;
 208     return buf_size;
 209 }
 210
 211 static av_cold int decode_end(AVCodecContext *avctx)
 212 {
 213     XbinContext *s = avctx->priv_data;
 214
 215     av_frame_free(&s->frame);
 216
 217     return 0;
 218 }
 219
 220 #if CONFIG_BINTEXT_DECODER
 221 AVCodec ff_bintext_decoder = {
 222     .name           = "bintext",
 223     .long_name      = NULL_IF_CONFIG_SMALL("Binary text"),
 224     .type           = AVMEDIA_TYPE_VIDEO,
 225     .id             = AV_CODEC_ID_BINTEXT,
 226     .priv_data_size = sizeof(XbinContext),
 227     .init           = decode_init,
 228     .close          = decode_end,
 229     .decode         = decode_frame,
 230     .capabilities   = CODEC_CAP_DR1,
 231 };
 232 #endif
 233 #if CONFIG_XBIN_DECODER
 234 AVCodec ff_xbin_decoder = {
 235     .name           = "xbin",
 236     .long_name      = NULL_IF_CONFIG_SMALL("eXtended BINary text"),
 237     .type           = AVMEDIA_TYPE_VIDEO,
 238     .id             = AV_CODEC_ID_XBIN,
 239     .priv_data_size = sizeof(XbinContext),
 240     .init           = decode_init,
 241     .close          = decode_end,
 242     .decode         = decode_frame,
 243     .capabilities   = CODEC_CAP_DR1,
 244 };
 245 #endif
 246 #if CONFIG_IDF_DECODER
 247 AVCodec ff_idf_decoder = {
 248     .name           = "idf",
 249     .long_name      = NULL_IF_CONFIG_SMALL("iCEDraw text"),
 250     .type           = AVMEDIA_TYPE_VIDEO,
 251     .id             = AV_CODEC_ID_IDF,
 252     .priv_data_size = sizeof(XbinContext),
 253     .init           = decode_init,
 254     .close          = decode_end,
 255     .decode         = decode_frame,
 256     .capabilities   = CODEC_CAP_DR1,
 257 };
 258 #endif