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