avcodec: change number of plane pointers from 4 to 8 at next major bump.
[libav.git] / libavcodec / dsicinav.c
CommitLineData
72450e50
BC
1/*
2 * Delphine Software International CIN Audio/Video Decoders
3 * Copyright (c) 2006 Gregory Montoir (cyx@users.sourceforge.net)
4 *
2912e87a 5 * This file is part of Libav.
72450e50 6 *
2912e87a 7 * Libav is free software; you can redistribute it and/or
72450e50
BC
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
2912e87a 12 * Libav is distributed in the hope that it will be useful,
72450e50
BC
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
2912e87a 18 * License along with Libav; if not, write to the Free Software
72450e50
BC
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22/**
ba87f080 23 * @file
72450e50
BC
24 * Delphine Software International CIN audio/video decoders
25 */
26
27#include "avcodec.h"
2c124cb6 28#include "bytestream.h"
7d1b17b8 29#include "mathops.h"
72450e50
BC
30
31
32typedef enum CinVideoBitmapIndex {
33 CIN_CUR_BMP = 0, /* current */
34 CIN_PRE_BMP = 1, /* previous */
35 CIN_INT_BMP = 2 /* intermediate */
36} CinVideoBitmapIndex;
37
38typedef struct CinVideoContext {
39 AVCodecContext *avctx;
40 AVFrame frame;
41 unsigned int bitmap_size;
42 uint32_t palette[256];
43 uint8_t *bitmap_table[3];
44} CinVideoContext;
45
46typedef struct CinAudioContext {
72450e50
BC
47 int initial_decode_frame;
48 int delta;
49} CinAudioContext;
50
51
52/* table defining a geometric sequence with multiplier = 32767 ^ (1 / 128) */
53static const int16_t cinaudio_delta16_table[256] = {
54 0, 0, 0, 0, 0, 0, 0, 0,
55 0, 0, 0, 0, 0, 0, 0, 0,
56 0, 0, 0, -30210, -27853, -25680, -23677, -21829,
57 -20126, -18556, -17108, -15774, -14543, -13408, -12362, -11398,
58 -10508, -9689, -8933, -8236, -7593, -7001, -6455, -5951,
59 -5487, -5059, -4664, -4300, -3964, -3655, -3370, -3107,
60 -2865, -2641, -2435, -2245, -2070, -1908, -1759, -1622,
61 -1495, -1379, -1271, -1172, -1080, -996, -918, -847,
62 -781, -720, -663, -612, -564, -520, -479, -442,
63 -407, -376, -346, -319, -294, -271, -250, -230,
64 -212, -196, -181, -166, -153, -141, -130, -120,
65 -111, -102, -94, -87, -80, -74, -68, -62,
66 -58, -53, -49, -45, -41, -38, -35, -32,
67 -30, -27, -25, -23, -21, -20, -18, -17,
68 -15, -14, -13, -12, -11, -10, -9, -8,
69 -7, -6, -5, -4, -3, -2, -1, 0,
70 0, 1, 2, 3, 4, 5, 6, 7,
71 8, 9, 10, 11, 12, 13, 14, 15,
72 17, 18, 20, 21, 23, 25, 27, 30,
73 32, 35, 38, 41, 45, 49, 53, 58,
74 62, 68, 74, 80, 87, 94, 102, 111,
75 120, 130, 141, 153, 166, 181, 196, 212,
76 230, 250, 271, 294, 319, 346, 376, 407,
77 442, 479, 520, 564, 612, 663, 720, 781,
78 847, 918, 996, 1080, 1172, 1271, 1379, 1495,
79 1622, 1759, 1908, 2070, 2245, 2435, 2641, 2865,
80 3107, 3370, 3655, 3964, 4300, 4664, 5059, 5487,
81 5951, 6455, 7001, 7593, 8236, 8933, 9689, 10508,
82 11398, 12362, 13408, 14543, 15774, 17108, 18556, 20126,
83 21829, 23677, 25680, 27853, 30210, 0, 0, 0,
84 0, 0, 0, 0, 0, 0, 0, 0,
85 0, 0, 0, 0, 0, 0, 0, 0
86};
87
88
98a6fff9 89static av_cold int cinvideo_decode_init(AVCodecContext *avctx)
72450e50 90{
e4141433 91 CinVideoContext *cin = avctx->priv_data;
72450e50
BC
92 unsigned int i;
93
94 cin->avctx = avctx;
95 avctx->pix_fmt = PIX_FMT_PAL8;
72450e50
BC
96
97 cin->frame.data[0] = NULL;
98
99 cin->bitmap_size = avctx->width * avctx->height;
100 for (i = 0; i < 3; ++i) {
101 cin->bitmap_table[i] = av_mallocz(cin->bitmap_size);
102 if (!cin->bitmap_table[i])
103 av_log(avctx, AV_LOG_ERROR, "Can't allocate bitmap buffers.\n");
104 }
105
106 return 0;
107}
108
109static void cin_apply_delta_data(const unsigned char *src, unsigned char *dst, int size)
110{
111 while (size--)
112 *dst++ += *src++;
113}
114
115static int cin_decode_huffman(const unsigned char *src, int src_size, unsigned char *dst, int dst_size)
116{
117 int b, huff_code = 0;
118 unsigned char huff_code_table[15];
119 unsigned char *dst_cur = dst;
120 unsigned char *dst_end = dst + dst_size;
121 const unsigned char *src_end = src + src_size;
122
123 memcpy(huff_code_table, src, 15); src += 15; src_size -= 15;
124
125 while (src < src_end) {
126 huff_code = *src++;
127 if ((huff_code >> 4) == 15) {
128 b = huff_code << 4;
129 huff_code = *src++;
130 *dst_cur++ = b | (huff_code >> 4);
131 } else
132 *dst_cur++ = huff_code_table[huff_code >> 4];
133 if (dst_cur >= dst_end)
134 break;
135
136 huff_code &= 15;
137 if (huff_code == 15) {
138 *dst_cur++ = *src++;
139 } else
140 *dst_cur++ = huff_code_table[huff_code];
141 if (dst_cur >= dst_end)
142 break;
143 }
144
145 return dst_cur - dst;
146}
147
148static void cin_decode_lzss(const unsigned char *src, int src_size, unsigned char *dst, int dst_size)
149{
150 uint16_t cmd;
151 int i, sz, offset, code;
152 unsigned char *dst_end = dst + dst_size;
153 const unsigned char *src_end = src + src_size;
154
155 while (src < src_end && dst < dst_end) {
156 code = *src++;
157 for (i = 0; i < 8 && src < src_end && dst < dst_end; ++i) {
158 if (code & (1 << i)) {
159 *dst++ = *src++;
160 } else {
fead30d4 161 cmd = AV_RL16(src); src += 2;
72450e50
BC
162 offset = cmd >> 4;
163 sz = (cmd & 0xF) + 2;
164 /* don't use memcpy/memmove here as the decoding routine (ab)uses */
165 /* buffer overlappings to repeat bytes in the destination */
166 sz = FFMIN(sz, dst_end - dst);
167 while (sz--) {
168 *dst = *(dst - offset - 1);
169 ++dst;
170 }
171 }
172 }
173 }
174}
175
176static void cin_decode_rle(const unsigned char *src, int src_size, unsigned char *dst, int dst_size)
177{
178 int len, code;
179 unsigned char *dst_end = dst + dst_size;
180 const unsigned char *src_end = src + src_size;
181
182 while (src < src_end && dst < dst_end) {
183 code = *src++;
184 if (code & 0x80) {
185 len = code - 0x7F;
186 memset(dst, *src++, FFMIN(len, dst_end - dst));
187 } else {
188 len = code + 1;
189 memcpy(dst, src, FFMIN(len, dst_end - dst));
190 src += len;
191 }
192 dst += len;
193 }
194}
195
196static int cinvideo_decode_frame(AVCodecContext *avctx,
197 void *data, int *data_size,
7a00bbad 198 AVPacket *avpkt)
72450e50 199{
7a00bbad
TB
200 const uint8_t *buf = avpkt->data;
201 int buf_size = avpkt->size;
e4141433 202 CinVideoContext *cin = avctx->priv_data;
72450e50
BC
203 int i, y, palette_type, palette_colors_count, bitmap_frame_type, bitmap_frame_size;
204
205 cin->frame.buffer_hints = FF_BUFFER_HINTS_VALID | FF_BUFFER_HINTS_PRESERVE | FF_BUFFER_HINTS_REUSABLE;
206 if (avctx->reget_buffer(avctx, &cin->frame)) {
207 av_log(cin->avctx, AV_LOG_ERROR, "delphinecinvideo: reget_buffer() failed to allocate a frame\n");
208 return -1;
209 }
210
211 palette_type = buf[0];
2c124cb6 212 palette_colors_count = AV_RL16(buf+1);
72450e50
BC
213 bitmap_frame_type = buf[3];
214 buf += 4;
215
216 bitmap_frame_size = buf_size - 4;
217
218 /* handle palette */
e3ca9b93
LA
219 if (bitmap_frame_size < palette_colors_count * (3 + (palette_type != 0)))
220 return AVERROR_INVALIDDATA;
72450e50 221 if (palette_type == 0) {
17206032
LA
222 if (palette_colors_count > 256)
223 return AVERROR_INVALIDDATA;
72450e50 224 for (i = 0; i < palette_colors_count; ++i) {
2c124cb6 225 cin->palette[i] = bytestream_get_le24(&buf);
72450e50
BC
226 bitmap_frame_size -= 3;
227 }
228 } else {
229 for (i = 0; i < palette_colors_count; ++i) {
2c124cb6 230 cin->palette[buf[0]] = AV_RL24(buf+1);
72450e50
BC
231 buf += 4;
232 bitmap_frame_size -= 4;
233 }
234 }
235 memcpy(cin->frame.data[1], cin->palette, sizeof(cin->palette));
236 cin->frame.palette_has_changed = 1;
237
238 /* note: the decoding routines below assumes that surface.width = surface.pitch */
239 switch (bitmap_frame_type) {
240 case 9:
241 cin_decode_rle(buf, bitmap_frame_size,
242 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
243 break;
244 case 34:
245 cin_decode_rle(buf, bitmap_frame_size,
246 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
247 cin_apply_delta_data(cin->bitmap_table[CIN_PRE_BMP],
248 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
249 break;
250 case 35:
251 cin_decode_huffman(buf, bitmap_frame_size,
252 cin->bitmap_table[CIN_INT_BMP], cin->bitmap_size);
253 cin_decode_rle(cin->bitmap_table[CIN_INT_BMP], bitmap_frame_size,
254 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
255 break;
256 case 36:
257 bitmap_frame_size = cin_decode_huffman(buf, bitmap_frame_size,
258 cin->bitmap_table[CIN_INT_BMP], cin->bitmap_size);
259 cin_decode_rle(cin->bitmap_table[CIN_INT_BMP], bitmap_frame_size,
260 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
261 cin_apply_delta_data(cin->bitmap_table[CIN_PRE_BMP],
262 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
263 break;
264 case 37:
265 cin_decode_huffman(buf, bitmap_frame_size,
266 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
267 break;
268 case 38:
269 cin_decode_lzss(buf, bitmap_frame_size,
270 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
271 break;
272 case 39:
273 cin_decode_lzss(buf, bitmap_frame_size,
274 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
275 cin_apply_delta_data(cin->bitmap_table[CIN_PRE_BMP],
276 cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_size);
277 break;
278 }
279
280 for (y = 0; y < cin->avctx->height; ++y)
281 memcpy(cin->frame.data[0] + (cin->avctx->height - 1 - y) * cin->frame.linesize[0],
282 cin->bitmap_table[CIN_CUR_BMP] + y * cin->avctx->width,
283 cin->avctx->width);
284
1345f4ed 285 FFSWAP(uint8_t *, cin->bitmap_table[CIN_CUR_BMP], cin->bitmap_table[CIN_PRE_BMP]);
72450e50
BC
286
287 *data_size = sizeof(AVFrame);
288 *(AVFrame *)data = cin->frame;
289
290 return buf_size;
291}
292
98a6fff9 293static av_cold int cinvideo_decode_end(AVCodecContext *avctx)
72450e50 294{
e4141433 295 CinVideoContext *cin = avctx->priv_data;
72450e50
BC
296 int i;
297
298 if (cin->frame.data[0])
299 avctx->release_buffer(avctx, &cin->frame);
300
301 for (i = 0; i < 3; ++i)
302 av_free(cin->bitmap_table[i]);
303
304 return 0;
305}
306
98a6fff9 307static av_cold int cinaudio_decode_init(AVCodecContext *avctx)
72450e50 308{
e4141433 309 CinAudioContext *cin = avctx->priv_data;
72450e50 310
03381c12
JR
311 if (avctx->channels != 1) {
312 av_log_ask_for_sample(avctx, "Number of channels is not supported\n");
313 return AVERROR_PATCHWELCOME;
314 }
315
72450e50
BC
316 cin->initial_decode_frame = 1;
317 cin->delta = 0;
5d6e4c16 318 avctx->sample_fmt = AV_SAMPLE_FMT_S16;
72450e50
BC
319
320 return 0;
321}
322
323static int cinaudio_decode_frame(AVCodecContext *avctx,
324 void *data, int *data_size,
7a00bbad 325 AVPacket *avpkt)
72450e50 326{
7a00bbad 327 const uint8_t *buf = avpkt->data;
e4141433 328 CinAudioContext *cin = avctx->priv_data;
405af431 329 const uint8_t *buf_end = buf + avpkt->size;
64e19ba4 330 int16_t *samples = data;
405af431 331 int delta, out_size;
72450e50 332
405af431
JR
333 out_size = (avpkt->size - cin->initial_decode_frame) *
334 av_get_bytes_per_sample(avctx->sample_fmt);
335 if (*data_size < out_size) {
336 av_log(avctx, AV_LOG_ERROR, "Output buffer is too small\n");
337 return AVERROR(EINVAL);
338 }
675a0583 339
859bdc33 340 delta = cin->delta;
72450e50
BC
341 if (cin->initial_decode_frame) {
342 cin->initial_decode_frame = 0;
7d1b17b8 343 delta = sign_extend(AV_RL16(buf), 16);
405af431 344 buf += 2;
859bdc33 345 *samples++ = delta;
72450e50 346 }
405af431
JR
347 while (buf < buf_end) {
348 delta += cinaudio_delta16_table[*buf++];
859bdc33
JR
349 delta = av_clip_int16(delta);
350 *samples++ = delta;
72450e50 351 }
859bdc33 352 cin->delta = delta;
72450e50 353
405af431 354 *data_size = out_size;
72450e50 355
405af431 356 return avpkt->size;
72450e50
BC
357}
358
359
d36beb3f 360AVCodec ff_dsicinvideo_decoder = {
ec6402b7
AK
361 .name = "dsicinvideo",
362 .type = AVMEDIA_TYPE_VIDEO,
363 .id = CODEC_ID_DSICINVIDEO,
364 .priv_data_size = sizeof(CinVideoContext),
365 .init = cinvideo_decode_init,
366 .close = cinvideo_decode_end,
367 .decode = cinvideo_decode_frame,
368 .capabilities = CODEC_CAP_DR1,
fe4bf374 369 .long_name = NULL_IF_CONFIG_SMALL("Delphine Software International CIN video"),
72450e50
BC
370};
371
d36beb3f 372AVCodec ff_dsicinaudio_decoder = {
ec6402b7
AK
373 .name = "dsicinaudio",
374 .type = AVMEDIA_TYPE_AUDIO,
375 .id = CODEC_ID_DSICINAUDIO,
376 .priv_data_size = sizeof(CinAudioContext),
377 .init = cinaudio_decode_init,
378 .decode = cinaudio_decode_frame,
fe4bf374 379 .long_name = NULL_IF_CONFIG_SMALL("Delphine Software International CIN audio"),
72450e50 380};