[deb_ffmpeg.git] / ffmpeg / libavcodec / bintext.c

/*
 * Binary text decoder
 * eXtended BINary text (XBIN) decoder
 * iCEDraw File decoder
 * Copyright (c) 2010 Peter Ross (pross@xvid.org)
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/**
 * @file
 * Binary text decoder
 * eXtended BINary text (XBIN) decoder
 * iCEDraw File decoder
 */

#include "libavutil/intreadwrite.h"
#include "libavutil/xga_font_data.h"
#include "avcodec.h"
#include "cga_data.h"
#include "bintext.h"
#include "internal.h"

typedef struct XbinContext {
    AVFrame *frame;
    int palette[16];
    int flags;
    int font_height;
    const uint8_t *font;
    int x, y;
} XbinContext;

static av_cold int decode_init(AVCodecContext *avctx)
{
    XbinContext *s = avctx->priv_data;
    uint8_t *p;
    int i;

    avctx->pix_fmt = AV_PIX_FMT_PAL8;
    p = avctx->extradata;
    if (p) {
        s->font_height = p[0];
        s->flags = p[1];
        p += 2;
        if(avctx->extradata_size < 2 + (!!(s->flags & BINTEXT_PALETTE))*3*16
                                     + (!!(s->flags & BINTEXT_FONT))*s->font_height*256) {
            av_log(avctx, AV_LOG_ERROR, "not enough extradata\n");
            return AVERROR_INVALIDDATA;
        }
    } else {
        s->font_height = 8;
        s->flags = 0;
    }

    if ((s->flags & BINTEXT_PALETTE)) {
        for (i = 0; i < 16; i++) {
            s->palette[i] = 0xFF000000 | (AV_RB24(p) << 2) | ((AV_RB24(p) >> 4) & 0x30303);
            p += 3;
        }
    } else {
        for (i = 0; i < 16; i++)
            s->palette[i] = 0xFF000000 | ff_cga_palette[i];
    }

    if ((s->flags & BINTEXT_FONT)) {
        s->font = p;
    } else {
        switch(s->font_height) {
        default:
            av_log(avctx, AV_LOG_WARNING, "font height %i not supported\n", s->font_height);
            s->font_height = 8;
        case 8:
            s->font = avpriv_cga_font;
            break;
        case 16:
            s->font = avpriv_vga16_font;
            break;
        }
    }

    s->frame = av_frame_alloc();
    if (!s->frame)
        return AVERROR(ENOMEM);

    return 0;
}

#define DEFAULT_BG_COLOR 0
av_unused static void hscroll(AVCodecContext *avctx)
{
    XbinContext *s = avctx->priv_data;
    if (s->y < avctx->height - s->font_height) {
        s->y += s->font_height;
    } else {
        memmove(s->frame->data[0], s->frame->data[0] + s->font_height*s->frame->linesize[0],
            (avctx->height - s->font_height)*s->frame->linesize[0]);
        memset(s->frame->data[0] + (avctx->height - s->font_height)*s->frame->linesize[0],
            DEFAULT_BG_COLOR, s->font_height * s->frame->linesize[0]);
    }
}

#define FONT_WIDTH 8

/**
 * Draw character to screen
 */
static void draw_char(AVCodecContext *avctx, int c, int a)
{
    XbinContext *s = avctx->priv_data;
    if (s->y > avctx->height - s->font_height)
        return;
    ff_draw_pc_font(s->frame->data[0] + s->y * s->frame->linesize[0] + s->x,
                    s->frame->linesize[0], s->font, s->font_height, c,
                    a & 0x0F, a >> 4);
    s->x += FONT_WIDTH;
    if (s->x > avctx->width - FONT_WIDTH) {
        s->x = 0;
        s->y += s->font_height;
    }
}

static int decode_frame(AVCodecContext *avctx,
                            void *data, int *got_frame,
                            AVPacket *avpkt)
{
    XbinContext *s = avctx->priv_data;
    const uint8_t *buf = avpkt->data;
    int buf_size = avpkt->size;
    const uint8_t *buf_end = buf+buf_size;
    int ret;

    s->x = s->y = 0;
    if ((ret = ff_reget_buffer(avctx, s->frame)) < 0)
        return ret;
    s->frame->pict_type           = AV_PICTURE_TYPE_I;
    s->frame->palette_has_changed = 1;
    memcpy(s->frame->data[1], s->palette, 16 * 4);

    if (avctx->codec_id == AV_CODEC_ID_XBIN) {
        while (buf + 2 < buf_end) {
            int i,c,a;
            int type  = *buf >> 6;
            int count = (*buf & 0x3F) + 1;
            buf++;
            switch (type) {
            case 0: //no compression
                for (i = 0; i < count && buf + 1 < buf_end; i++) {
                    draw_char(avctx, buf[0], buf[1]);
                    buf += 2;
                }
                break;
            case 1: //character compression
                c = *buf++;
                for (i = 0; i < count && buf < buf_end; i++)
                    draw_char(avctx, c, *buf++);
                break;
            case 2: //attribute compression
                a = *buf++;
                for (i = 0; i < count && buf < buf_end; i++)
                    draw_char(avctx, *buf++, a);
                break;
            case 3: //character/attribute compression
                c = *buf++;
                a = *buf++;
                for (i = 0; i < count && buf < buf_end; i++)
                    draw_char(avctx, c, a);
                break;
            }
        }
    } else if (avctx->codec_id == AV_CODEC_ID_IDF) {
        while (buf + 2 < buf_end) {
            if (AV_RL16(buf) == 1) {
               int i;
               if (buf + 6 > buf_end)
                   break;
               for (i = 0; i < buf[2]; i++)
                   draw_char(avctx, buf[4], buf[5]);
               buf += 6;
            } else {
               draw_char(avctx, buf[0], buf[1]);
               buf += 2;
            }
        }
    } else {
        while (buf + 1 < buf_end) {
            draw_char(avctx, buf[0], buf[1]);
            buf += 2;
        }
    }

    if ((ret = av_frame_ref(data, s->frame)) < 0)
        return ret;
    *got_frame      = 1;
    return buf_size;
}

static av_cold int decode_end(AVCodecContext *avctx)
{
    XbinContext *s = avctx->priv_data;

    av_frame_free(&s->frame);

    return 0;
}

#if CONFIG_BINTEXT_DECODER
AVCodec ff_bintext_decoder = {
    .name           = "bintext",
    .long_name      = NULL_IF_CONFIG_SMALL("Binary text"),
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_BINTEXT,
    .priv_data_size = sizeof(XbinContext),
    .init           = decode_init,
    .close          = decode_end,
    .decode         = decode_frame,
    .capabilities   = CODEC_CAP_DR1,
};
#endif
#if CONFIG_XBIN_DECODER
AVCodec ff_xbin_decoder = {
    .name           = "xbin",
    .long_name      = NULL_IF_CONFIG_SMALL("eXtended BINary text"),
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_XBIN,
    .priv_data_size = sizeof(XbinContext),
    .init           = decode_init,
    .close          = decode_end,
    .decode         = decode_frame,
    .capabilities   = CODEC_CAP_DR1,
};
#endif
#if CONFIG_IDF_DECODER
AVCodec ff_idf_decoder = {
    .name           = "idf",
    .long_name      = NULL_IF_CONFIG_SMALL("iCEDraw text"),
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_IDF,
    .priv_data_size = sizeof(XbinContext),
    .init           = decode_init,
    .close          = decode_end,
    .decode         = decode_frame,
    .capabilities   = CODEC_CAP_DR1,
};
#endif
Commit	Line	Data
2ba45a60 DM	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))316
	60	+ (!!(s->flags & BINTEXT_FONT))s->font_height256) {
	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