b6d7d0f84cb06b789aa5791e080a1e70bbe5730f
[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
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 (ff_reget_buffer(avctx, s->frame) < 0) {
148         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
149         return -1;
150     }
151     s->frame->pict_type           = AV_PICTURE_TYPE_I;
152     s->frame->palette_has_changed = 1;
153     memcpy(s->frame->data[1], s->palette, 16 * 4);
154
155     if (avctx->codec_id == AV_CODEC_ID_XBIN) {
156         while (buf + 2 < buf_end) {
157             int i,c,a;
158             int type  = *buf >> 6;
159             int count = (*buf & 0x3F) + 1;
160             buf++;
161             switch (type) {
162             case 0: //no compression
163                 for (i = 0; i < count && buf + 1 < buf_end; i++) {
164                     draw_char(avctx, buf[0], buf[1]);
165                     buf += 2;
166                 }
167                 break;
168             case 1: //character compression
169                 c = *buf++;
170                 for (i = 0; i < count && buf < buf_end; i++)
171                     draw_char(avctx, c, *buf++);
172                 break;
173             case 2: //attribute compression
174                 a = *buf++;
175                 for (i = 0; i < count && buf < buf_end; i++)
176                     draw_char(avctx, *buf++, a);
177                 break;
178             case 3: //character/attribute compression
179                 c = *buf++;
180                 a = *buf++;
181                 for (i = 0; i < count && buf < buf_end; i++)
182                     draw_char(avctx, c, a);
183                 break;
184             }
185         }
186     } else if (avctx->codec_id == AV_CODEC_ID_IDF) {
187         while (buf + 2 < buf_end) {
188             if (AV_RL16(buf) == 1) {
189                int i;
190                if (buf + 6 > buf_end)
191                    break;
192                for (i = 0; i < buf[2]; i++)
193                    draw_char(avctx, buf[4], buf[5]);
194                buf += 6;
195             } else {
196                draw_char(avctx, buf[0], buf[1]);
197                buf += 2;
198             }
199         }
200     } else {
201         while (buf + 1 < buf_end) {
202             draw_char(avctx, buf[0], buf[1]);
203             buf += 2;
204         }
205     }
206
207     if ((ret = av_frame_ref(data, s->frame)) < 0)
208         return ret;
209     *got_frame      = 1;
210     return buf_size;
211 }
212
213 static av_cold int decode_end(AVCodecContext *avctx)
214 {
215     XbinContext *s = avctx->priv_data;
216
217     av_frame_free(&s->frame);
218
219     return 0;
220 }
221
222 #if CONFIG_BINTEXT_DECODER
223 AVCodec ff_bintext_decoder = {
224     .name           = "bintext",
225     .type           = AVMEDIA_TYPE_VIDEO,
226     .id             = AV_CODEC_ID_BINTEXT,
227     .priv_data_size = sizeof(XbinContext),
228     .init           = decode_init,
229     .close          = decode_end,
230     .decode         = decode_frame,
231     .capabilities   = CODEC_CAP_DR1,
232     .long_name      = NULL_IF_CONFIG_SMALL("Binary text"),
233 };
234 #endif
235 #if CONFIG_XBIN_DECODER
236 AVCodec ff_xbin_decoder = {
237     .name           = "xbin",
238     .type           = AVMEDIA_TYPE_VIDEO,
239     .id             = AV_CODEC_ID_XBIN,
240     .priv_data_size = sizeof(XbinContext),
241     .init           = decode_init,
242     .close          = decode_end,
243     .decode         = decode_frame,
244     .capabilities   = CODEC_CAP_DR1,
245     .long_name      = NULL_IF_CONFIG_SMALL("eXtended BINary text"),
246 };
247 #endif
248 #if CONFIG_IDF_DECODER
249 AVCodec ff_idf_decoder = {
250     .name           = "idf",
251     .type           = AVMEDIA_TYPE_VIDEO,
252     .id             = AV_CODEC_ID_IDF,
253     .priv_data_size = sizeof(XbinContext),
254     .init           = decode_init,
255     .close          = decode_end,
256     .decode         = decode_frame,
257     .capabilities   = CODEC_CAP_DR1,
258     .long_name      = NULL_IF_CONFIG_SMALL("iCEDraw text"),
259 };
260 #endif