Merge commit 'a7a07cc98ac548297b5b0628cb81280e11952e3f'
[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 ((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