yading@10: /*
yading@10:  * Binary text decoder
yading@10:  * eXtended BINary text (XBIN) decoder
yading@10:  * iCEDraw File decoder
yading@10:  * Copyright (c) 2010 Peter Ross (pross@xvid.org)
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:  * Binary text decoder
yading@10:  * eXtended BINary text (XBIN) decoder
yading@10:  * iCEDraw File decoder
yading@10:  */
yading@10: 
yading@10: #include "libavutil/intreadwrite.h"
yading@10: #include "libavutil/xga_font_data.h"
yading@10: #include "avcodec.h"
yading@10: #include "cga_data.h"
yading@10: #include "bintext.h"
yading@10: #include "internal.h"
yading@10: 
yading@10: typedef struct XbinContext {
yading@10:     AVFrame *frame;
yading@10:     int palette[16];
yading@10:     int flags;
yading@10:     int font_height;
yading@10:     const uint8_t *font;
yading@10:     int x, y;
yading@10: } XbinContext;
yading@10: 
yading@10: static av_cold int decode_init(AVCodecContext *avctx)
yading@10: {
yading@10:     XbinContext *s = avctx->priv_data;
yading@10:     uint8_t *p;
yading@10:     int i;
yading@10: 
yading@10:     avctx->pix_fmt = AV_PIX_FMT_PAL8;
yading@10:     p = avctx->extradata;
yading@10:     if (p) {
yading@10:         s->font_height = p[0];
yading@10:         s->flags = p[1];
yading@10:         p += 2;
yading@10:         if(avctx->extradata_size < 2 + (!!(s->flags & BINTEXT_PALETTE))*3*16
yading@10:                                      + (!!(s->flags & BINTEXT_FONT))*s->font_height*256) {
yading@10:             av_log(avctx, AV_LOG_ERROR, "not enough extradata\n");
yading@10:             return AVERROR_INVALIDDATA;
yading@10:         }
yading@10:     } else {
yading@10:         s->font_height = 8;
yading@10:         s->flags = 0;
yading@10:     }
yading@10: 
yading@10:     if ((s->flags & BINTEXT_PALETTE)) {
yading@10:         for (i = 0; i < 16; i++) {
yading@10:             s->palette[i] = 0xFF000000 | (AV_RB24(p) << 2) | ((AV_RB24(p) >> 4) & 0x30303);
yading@10:             p += 3;
yading@10:         }
yading@10:     } else {
yading@10:         for (i = 0; i < 16; i++)
yading@10:             s->palette[i] = 0xFF000000 | ff_cga_palette[i];
yading@10:     }
yading@10: 
yading@10:     if ((s->flags & BINTEXT_FONT)) {
yading@10:         s->font = p;
yading@10:     } else {
yading@10:         switch(s->font_height) {
yading@10:         default:
yading@10:             av_log(avctx, AV_LOG_WARNING, "font height %i not supported\n", s->font_height);
yading@10:             s->font_height = 8;
yading@10:         case 8:
yading@10:             s->font = avpriv_cga_font;
yading@10:             break;
yading@10:         case 16:
yading@10:             s->font = avpriv_vga16_font;
yading@10:             break;
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     s->frame = av_frame_alloc();
yading@10:     if (!s->frame)
yading@10:         return AVERROR(ENOMEM);
yading@10: 
yading@10:     return 0;
yading@10: }
yading@10: 
yading@10: #define DEFAULT_BG_COLOR 0
yading@10: av_unused static void hscroll(AVCodecContext *avctx)
yading@10: {
yading@10:     XbinContext *s = avctx->priv_data;
yading@10:     if (s->y < avctx->height - s->font_height) {
yading@10:         s->y += s->font_height;
yading@10:     } else {
yading@10:         memmove(s->frame->data[0], s->frame->data[0] + s->font_height*s->frame->linesize[0],
yading@10:             (avctx->height - s->font_height)*s->frame->linesize[0]);
yading@10:         memset(s->frame->data[0] + (avctx->height - s->font_height)*s->frame->linesize[0],
yading@10:             DEFAULT_BG_COLOR, s->font_height * s->frame->linesize[0]);
yading@10:     }
yading@10: }
yading@10: 
yading@10: #define FONT_WIDTH 8
yading@10: 
yading@10: /**
yading@10:  * Draw character to screen
yading@10:  */
yading@10: static void draw_char(AVCodecContext *avctx, int c, int a)
yading@10: {
yading@10:     XbinContext *s = avctx->priv_data;
yading@10:     if (s->y > avctx->height - s->font_height)
yading@10:         return;
yading@10:     ff_draw_pc_font(s->frame->data[0] + s->y * s->frame->linesize[0] + s->x,
yading@10:                     s->frame->linesize[0], s->font, s->font_height, c,
yading@10:                     a & 0x0F, a >> 4);
yading@10:     s->x += FONT_WIDTH;
yading@10:     if (s->x > avctx->width - FONT_WIDTH) {
yading@10:         s->x = 0;
yading@10:         s->y += s->font_height;
yading@10:     }
yading@10: }
yading@10: 
yading@10: static int decode_frame(AVCodecContext *avctx,
yading@10:                             void *data, int *got_frame,
yading@10:                             AVPacket *avpkt)
yading@10: {
yading@10:     XbinContext *s = avctx->priv_data;
yading@10:     const uint8_t *buf = avpkt->data;
yading@10:     int buf_size = avpkt->size;
yading@10:     const uint8_t *buf_end = buf+buf_size;
yading@10:     int ret;
yading@10: 
yading@10:     s->x = s->y = 0;
yading@10:     if ((ret = ff_reget_buffer(avctx, s->frame)) < 0)
yading@10:         return ret;
yading@10:     s->frame->pict_type           = AV_PICTURE_TYPE_I;
yading@10:     s->frame->palette_has_changed = 1;
yading@10:     memcpy(s->frame->data[1], s->palette, 16 * 4);
yading@10: 
yading@10:     if (avctx->codec_id == AV_CODEC_ID_XBIN) {
yading@10:         while (buf + 2 < buf_end) {
yading@10:             int i,c,a;
yading@10:             int type  = *buf >> 6;
yading@10:             int count = (*buf & 0x3F) + 1;
yading@10:             buf++;
yading@10:             switch (type) {
yading@10:             case 0: //no compression
yading@10:                 for (i = 0; i < count && buf + 1 < buf_end; i++) {
yading@10:                     draw_char(avctx, buf[0], buf[1]);
yading@10:                     buf += 2;
yading@10:                 }
yading@10:                 break;
yading@10:             case 1: //character compression
yading@10:                 c = *buf++;
yading@10:                 for (i = 0; i < count && buf < buf_end; i++)
yading@10:                     draw_char(avctx, c, *buf++);
yading@10:                 break;
yading@10:             case 2: //attribute compression
yading@10:                 a = *buf++;
yading@10:                 for (i = 0; i < count && buf < buf_end; i++)
yading@10:                     draw_char(avctx, *buf++, a);
yading@10:                 break;
yading@10:             case 3: //character/attribute compression
yading@10:                 c = *buf++;
yading@10:                 a = *buf++;
yading@10:                 for (i = 0; i < count && buf < buf_end; i++)
yading@10:                     draw_char(avctx, c, a);
yading@10:                 break;
yading@10:             }
yading@10:         }
yading@10:     } else if (avctx->codec_id == AV_CODEC_ID_IDF) {
yading@10:         while (buf + 2 < buf_end) {
yading@10:             if (AV_RL16(buf) == 1) {
yading@10:                int i;
yading@10:                if (buf + 6 > buf_end)
yading@10:                    break;
yading@10:                for (i = 0; i < buf[2]; i++)
yading@10:                    draw_char(avctx, buf[4], buf[5]);
yading@10:                buf += 6;
yading@10:             } else {
yading@10:                draw_char(avctx, buf[0], buf[1]);
yading@10:                buf += 2;
yading@10:             }
yading@10:         }
yading@10:     } else {
yading@10:         while (buf + 1 < buf_end) {
yading@10:             draw_char(avctx, buf[0], buf[1]);
yading@10:             buf += 2;
yading@10:         }
yading@10:     }
yading@10: 
yading@10:     if ((ret = av_frame_ref(data, s->frame)) < 0)
yading@10:         return ret;
yading@10:     *got_frame      = 1;
yading@10:     return buf_size;
yading@10: }
yading@10: 
yading@10: static av_cold int decode_end(AVCodecContext *avctx)
yading@10: {
yading@10:     XbinContext *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_BINTEXT_DECODER
yading@10: AVCodec ff_bintext_decoder = {
yading@10:     .name           = "bintext",
yading@10:     .type           = AVMEDIA_TYPE_VIDEO,
yading@10:     .id             = AV_CODEC_ID_BINTEXT,
yading@10:     .priv_data_size = sizeof(XbinContext),
yading@10:     .init           = decode_init,
yading@10:     .close          = decode_end,
yading@10:     .decode         = decode_frame,
yading@10:     .capabilities   = CODEC_CAP_DR1,
yading@10:     .long_name      = NULL_IF_CONFIG_SMALL("Binary text"),
yading@10: };
yading@10: #endif
yading@10: #if CONFIG_XBIN_DECODER
yading@10: AVCodec ff_xbin_decoder = {
yading@10:     .name           = "xbin",
yading@10:     .type           = AVMEDIA_TYPE_VIDEO,
yading@10:     .id             = AV_CODEC_ID_XBIN,
yading@10:     .priv_data_size = sizeof(XbinContext),
yading@10:     .init           = decode_init,
yading@10:     .close          = decode_end,
yading@10:     .decode         = decode_frame,
yading@10:     .capabilities   = CODEC_CAP_DR1,
yading@10:     .long_name      = NULL_IF_CONFIG_SMALL("eXtended BINary text"),
yading@10: };
yading@10: #endif
yading@10: #if CONFIG_IDF_DECODER
yading@10: AVCodec ff_idf_decoder = {
yading@10:     .name           = "idf",
yading@10:     .type           = AVMEDIA_TYPE_VIDEO,
yading@10:     .id             = AV_CODEC_ID_IDF,
yading@10:     .priv_data_size = sizeof(XbinContext),
yading@10:     .init           = decode_init,
yading@10:     .close          = decode_end,
yading@10:     .decode         = decode_frame,
yading@10:     .capabilities   = CODEC_CAP_DR1,
yading@10:     .long_name      = NULL_IF_CONFIG_SMALL("iCEDraw text"),
yading@10: };
yading@10: #endif