Binary text decoder
[ffmpeg.git] / 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 libavcodec/xbin.c
26  * Binary text decoder
27  * eXtended BINary text (XBIN) decoder
28  * iCEDraw File decoder
29  */
30
31 #include "libavutil/intreadwrite.h"
32 #include "avcodec.h"
33 #include "cga_data.h"
34 #include "bintext.h"
35
36 typedef struct XbinContext {
37     AVFrame frame;
38     int palette[16];
39     int flags;
40     int font_height;
41     const uint8_t *font;
42     int x, y;
43 } XbinContext;
44
45 static av_cold int decode_init(AVCodecContext *avctx)
46 {
47     XbinContext *s = avctx->priv_data;
48     uint8_t *p;
49     int i;
50
51     avctx->pix_fmt = PIX_FMT_PAL8;
52     s->frame.reference = 1;
53     if (avctx->get_buffer(avctx, &s->frame)) {
54         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
55         return -1;
56     }
57
58     p = avctx->extradata;
59     if (p) {
60         s->font_height = p[0];
61         s->flags = p[1];
62         p += 2;
63     } else {
64         s->font_height = 8;
65         s->flags = 0;
66     }
67
68     if ((s->flags & BINTEXT_PALETTE)) {
69         for (i = 0; i < 16; i++) {
70             s->palette[i] = 0xFF000000 | (AV_RB24(p) << 2);
71             p += 3;
72         }
73     } else {
74         for (i = 0; i < 16; i++)
75             s->palette[i] = 0xFF000000 | ff_cga_palette[i];
76     }
77
78     if ((s->flags & BINTEXT_FONT)) {
79         s->font = p;
80     } else {
81         switch(s->font_height) {
82         default:
83             av_log(avctx, AV_LOG_WARNING, "font height %i not support\n", s->font_height);
84             s->font_height = 8;
85         case 8:
86             s->font = ff_cga_font;
87             break;
88         case 16:
89             s->font = ff_vga16_font;
90             break;
91         }
92     }
93
94     return 0;
95 }
96
97 #define DEFAULT_BG_COLOR 0
98 static void hscroll(AVCodecContext *avctx)
99 {
100     XbinContext *s = avctx->priv_data;
101     if (s->y < avctx->height - s->font_height) {
102         s->y += s->font_height;
103     } else {
104         memmove(s->frame.data[0], s->frame.data[0] + s->font_height*s->frame.linesize[0],
105             (avctx->height - s->font_height)*s->frame.linesize[0]);
106         memset(s->frame.data[0] + (avctx->height - s->font_height)*s->frame.linesize[0],
107             DEFAULT_BG_COLOR, s->font_height * s->frame.linesize[0]);
108     }
109 }
110
111 #define FONT_WIDTH 8
112
113 /**
114  * Draw character to screen
115  */
116 static void draw_char(AVCodecContext *avctx, int c, int a)
117 {
118     XbinContext *s = avctx->priv_data;
119     if (s->y > avctx->height - s->font_height)
120         return;
121     ff_draw_pc_font(s->frame.data[0] + s->y * s->frame.linesize[0] + s->x,
122                     s->frame.linesize[0], s->font, s->font_height, c,
123                     a & 0x0F, a >> 4);
124     s->x += FONT_WIDTH;
125     if (s->x >= avctx->width) {
126         s->x = 0;
127         s->y += s->font_height;
128     }
129 }
130
131 static int decode_frame(AVCodecContext *avctx,
132                             void *data, int *data_size,
133                             AVPacket *avpkt)
134 {
135     XbinContext *s = avctx->priv_data;
136     const uint8_t *buf = avpkt->data;
137     int buf_size = avpkt->size;
138     const uint8_t *buf_end = buf+buf_size;
139
140     if (avctx->reget_buffer(avctx, &s->frame)) {
141         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
142         return -1;
143     }
144     s->frame.pict_type           = FF_I_TYPE;
145     s->frame.palette_has_changed = 1;
146     memcpy(s->frame.data[1], s->palette, 16 * 4);
147
148     if (avctx->codec_id == CODEC_ID_XBIN) {
149         while (buf + 2 < buf_end) {
150             int i,c,a;
151             int type  = *buf >> 6;
152             int count = (*buf & 0x3F) + 1;
153             buf++;
154             switch (type) {
155             case 0: //no compression
156                 for (i = 0; i < count && buf + 1 < buf_end; i++) {
157                     draw_char(avctx, buf[0], buf[1]);
158                     buf += 2;
159                 }
160                 break;
161             case 1: //character compression
162                 c = *buf++;
163                 for (i = 0; i < count && buf < buf_end; i++)
164                     draw_char(avctx, c, *buf++);
165                 break;
166             case 2: //attribute compression
167                 a = *buf++;
168                 for (i = 0; i < count && buf < buf_end; i++)
169                     draw_char(avctx, *buf++, a);
170                 break;
171             case 3: //character/attribute compression
172                 c = *buf++;
173                 a = *buf++;
174                 for (i = 0; i < count && buf < buf_end; i++)
175                     draw_char(avctx, c, a);
176                 break;
177             }
178         }
179     } else if (avctx->codec_id == CODEC_ID_IDF) {
180         while (buf + 2 < buf_end) {
181             if (AV_RL16(buf) == 1) {
182                int i;
183                if (buf + 6 > buf_end)
184                    break;
185                for (i = 0; i < buf[2]; i++)
186                    draw_char(avctx, buf[4], buf[5]);
187                buf += 6;
188             } else {
189                draw_char(avctx, buf[0], buf[1]);
190                buf += 2;
191             }
192         }
193     } else {
194         while (buf + 1 < buf_end) {
195             draw_char(avctx, buf[0], buf[1]);
196             buf += 2;
197         }
198     }
199
200     *data_size = sizeof(AVFrame);
201     *(AVFrame*)data = s->frame;
202     return buf_size;
203 }
204
205 static av_cold int decode_end(AVCodecContext *avctx)
206 {
207     XbinContext *s = avctx->priv_data;
208
209     if (s->frame.data[0])
210         avctx->release_buffer(avctx, &s->frame);
211
212     return 0;
213 }
214
215 AVCodec ff_bintext_decoder = {
216     "xbin",
217     AVMEDIA_TYPE_VIDEO,
218     CODEC_ID_BINTEXT,
219     sizeof(XbinContext),
220     decode_init,
221     NULL,
222     decode_end,
223     decode_frame,
224     CODEC_CAP_DR1,
225     .long_name = NULL_IF_CONFIG_SMALL("Binary text"),
226 };
227
228 AVCodec ff_xbin_decoder = {
229     "xbin",
230     AVMEDIA_TYPE_VIDEO,
231     CODEC_ID_XBIN,
232     sizeof(XbinContext),
233     decode_init,
234     NULL,
235     decode_end,
236     decode_frame,
237     CODEC_CAP_DR1,
238     .long_name = NULL_IF_CONFIG_SMALL("eXtended BINary text"),
239 };
240
241 AVCodec ff_idf_decoder = {
242     "idf",
243     AVMEDIA_TYPE_VIDEO,
244     CODEC_ID_IDF,
245     sizeof(XbinContext),
246     decode_init,
247     NULL,
248     decode_end,
249     decode_frame,
250     CODEC_CAP_DR1,
251     .long_name = NULL_IF_CONFIG_SMALL("iCEDraw text"),
252 };