cosmetics: remove one unneeded space
[libav.git] / libavcodec / dpcm.c
CommitLineData
3ef8be2b
MM
1/*
2 * Assorted DPCM codecs
3 * Copyright (c) 2003 The ffmpeg Project.
4 *
b78e7197
DB
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
3ef8be2b
MM
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
b78e7197 10 * version 2.1 of the License, or (at your option) any later version.
3ef8be2b 11 *
b78e7197 12 * FFmpeg is distributed in the hope that it will be useful,
3ef8be2b
MM
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
b78e7197 18 * License along with FFmpeg; if not, write to the Free Software
5509bffa 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
3ef8be2b
MM
20 */
21
22/**
23 * @file: dpcm.c
24 * Assorted DPCM (differential pulse code modulation) audio codecs
25 * by Mike Melanson (melanson@pcisys.net)
9937e686 26 * Xan DPCM decoder by Mario Brito (mbrito@student.dei.uc.pt)
3ef8be2b
MM
27 * for more information on the specific data formats, visit:
28 * http://www.pcisys.net/~melanson/codecs/simpleaudio.html
d08d7142 29 * SOL DPCMs implemented by Konstantin Shishkov
9937e686
MM
30 *
31 * Note about using the Xan DPCM decoder: Xan DPCM is used in AVI files
32 * found in the Wing Commander IV computer game. These AVI files contain
33 * WAVEFORMAT headers which report the audio format as 0x01: raw PCM.
34 * Clearly incorrect. To detect Xan DPCM, you will probably have to
35 * special-case your AVI demuxer to use Xan DPCM if the file uses 'Xxan'
36 * (Xan video) for its video codec. Alternately, such AVI files also contain
37 * the fourcc 'Axan' in the 'auds' chunk of the AVI header.
3ef8be2b
MM
38 */
39
40#include "avcodec.h"
41
42typedef struct DPCMContext {
43 int channels;
44 short roq_square_array[256];
d08d7142 45 long sample[2];//for SOL_DPCM
62a05b5b 46 const int *sol_table;//for SOL_DPCM
3ef8be2b
MM
47} DPCMContext;
48
3ef8be2b 49#define SE_16BIT(x) if (x & 0x8000) x -= 0x10000;
3ef8be2b 50
cf2baeb3 51static const int interplay_delta_table[] = {
3ef8be2b
MM
52 0, 1, 2, 3, 4, 5, 6, 7,
53 8, 9, 10, 11, 12, 13, 14, 15,
54 16, 17, 18, 19, 20, 21, 22, 23,
55 24, 25, 26, 27, 28, 29, 30, 31,
56 32, 33, 34, 35, 36, 37, 38, 39,
57 40, 41, 42, 43, 47, 51, 56, 61,
58 66, 72, 79, 86, 94, 102, 112, 122,
59 133, 145, 158, 173, 189, 206, 225, 245,
60 267, 292, 318, 348, 379, 414, 452, 493,
61 538, 587, 640, 699, 763, 832, 908, 991,
62 1081, 1180, 1288, 1405, 1534, 1673, 1826, 1993,
63 2175, 2373, 2590, 2826, 3084, 3365, 3672, 4008,
64 4373, 4772, 5208, 5683, 6202, 6767, 7385, 8059,
65 8794, 9597, 10472, 11428, 12471, 13609, 14851, 16206,
66 17685, 19298, 21060, 22981, 25078, 27367, 29864, 32589,
67 -29973, -26728, -23186, -19322, -15105, -10503, -5481, -1,
68 1, 1, 5481, 10503, 15105, 19322, 23186, 26728,
69 29973, -32589, -29864, -27367, -25078, -22981, -21060, -19298,
70 -17685, -16206, -14851, -13609, -12471, -11428, -10472, -9597,
71 -8794, -8059, -7385, -6767, -6202, -5683, -5208, -4772,
72 -4373, -4008, -3672, -3365, -3084, -2826, -2590, -2373,
73 -2175, -1993, -1826, -1673, -1534, -1405, -1288, -1180,
74 -1081, -991, -908, -832, -763, -699, -640, -587,
75 -538, -493, -452, -414, -379, -348, -318, -292,
76 -267, -245, -225, -206, -189, -173, -158, -145,
77 -133, -122, -112, -102, -94, -86, -79, -72,
78 -66, -61, -56, -51, -47, -43, -42, -41,
79 -40, -39, -38, -37, -36, -35, -34, -33,
80 -32, -31, -30, -29, -28, -27, -26, -25,
81 -24, -23, -22, -21, -20, -19, -18, -17,
82 -16, -15, -14, -13, -12, -11, -10, -9,
83 -8, -7, -6, -5, -4, -3, -2, -1
84
85};
86
62a05b5b 87static const int sol_table_old[16] =
d08d7142
MM
88 { 0x0, 0x1, 0x2 , 0x3, 0x6, 0xA, 0xF, 0x15,
89 -0x15, -0xF, -0xA, -0x6, -0x3, -0x2, -0x1, 0x0};
90
62a05b5b 91static const int sol_table_new[16] =
d08d7142
MM
92 { 0x0, 0x1, 0x2, 0x3, 0x6, 0xA, 0xF, 0x15,
93 0x0, -0x1, -0x2, -0x3, -0x6, -0xA, -0xF, -0x15};
115329f1 94
62a05b5b 95static const int sol_table_16[128] = {
d08d7142
MM
96 0x000, 0x008, 0x010, 0x020, 0x030, 0x040, 0x050, 0x060, 0x070, 0x080,
97 0x090, 0x0A0, 0x0B0, 0x0C0, 0x0D0, 0x0E0, 0x0F0, 0x100, 0x110, 0x120,
98 0x130, 0x140, 0x150, 0x160, 0x170, 0x180, 0x190, 0x1A0, 0x1B0, 0x1C0,
99 0x1D0, 0x1E0, 0x1F0, 0x200, 0x208, 0x210, 0x218, 0x220, 0x228, 0x230,
100 0x238, 0x240, 0x248, 0x250, 0x258, 0x260, 0x268, 0x270, 0x278, 0x280,
101 0x288, 0x290, 0x298, 0x2A0, 0x2A8, 0x2B0, 0x2B8, 0x2C0, 0x2C8, 0x2D0,
102 0x2D8, 0x2E0, 0x2E8, 0x2F0, 0x2F8, 0x300, 0x308, 0x310, 0x318, 0x320,
103 0x328, 0x330, 0x338, 0x340, 0x348, 0x350, 0x358, 0x360, 0x368, 0x370,
104 0x378, 0x380, 0x388, 0x390, 0x398, 0x3A0, 0x3A8, 0x3B0, 0x3B8, 0x3C0,
105 0x3C8, 0x3D0, 0x3D8, 0x3E0, 0x3E8, 0x3F0, 0x3F8, 0x400, 0x440, 0x480,
106 0x4C0, 0x500, 0x540, 0x580, 0x5C0, 0x600, 0x640, 0x680, 0x6C0, 0x700,
107 0x740, 0x780, 0x7C0, 0x800, 0x900, 0xA00, 0xB00, 0xC00, 0xD00, 0xE00,
108 0xF00, 0x1000, 0x1400, 0x1800, 0x1C00, 0x2000, 0x3000, 0x4000
109};
110
111
112
98a6fff9 113static av_cold int dpcm_decode_init(AVCodecContext *avctx)
3ef8be2b
MM
114{
115 DPCMContext *s = avctx->priv_data;
116 int i;
117 short square;
118
119 s->channels = avctx->channels;
d08d7142 120 s->sample[0] = s->sample[1] = 0;
3ef8be2b
MM
121
122 switch(avctx->codec->id) {
123
124 case CODEC_ID_ROQ_DPCM:
125 /* initialize square table */
126 for (i = 0; i < 128; i++) {
127 square = i * i;
128 s->roq_square_array[i] = square;
129 s->roq_square_array[i + 128] = -square;
130 }
131 break;
132
115329f1 133
d08d7142
MM
134 case CODEC_ID_SOL_DPCM:
135 switch(avctx->codec_tag){
136 case 1:
137 s->sol_table=sol_table_old;
138 s->sample[0] = s->sample[1] = 0x80;
139 break;
140 case 2:
141 s->sol_table=sol_table_new;
142 s->sample[0] = s->sample[1] = 0x80;
143 break;
144 case 3:
145 s->sol_table=sol_table_16;
146 break;
147 default:
148 av_log(avctx, AV_LOG_ERROR, "Unknown SOL subcodec\n");
149 return -1;
150 }
151 break;
115329f1 152
3ef8be2b
MM
153 default:
154 break;
155 }
156
fd76c37f 157 avctx->sample_fmt = SAMPLE_FMT_S16;
3ef8be2b
MM
158 return 0;
159}
160
161static int dpcm_decode_frame(AVCodecContext *avctx,
162 void *data, int *data_size,
7993df65 163 const uint8_t *buf, int buf_size)
3ef8be2b
MM
164{
165 DPCMContext *s = avctx->priv_data;
166 int in, out = 0;
3ef8be2b
MM
167 int predictor[2];
168 int channel_number = 0;
169 short *output_samples = data;
9937e686
MM
170 int shift[2];
171 unsigned char byte;
172 short diff;
3ef8be2b 173
c89c1d25
MM
174 if (!buf_size)
175 return 0;
176
a2085a7e
KS
177 // almost every DPCM variant expands one byte of data into two
178 if(*data_size/2 < buf_size)
179 return -1;
180
3ef8be2b
MM
181 switch(avctx->codec->id) {
182
183 case CODEC_ID_ROQ_DPCM:
184 if (s->channels == 1)
fead30d4 185 predictor[0] = AV_RL16(&buf[6]);
3ef8be2b
MM
186 else {
187 predictor[0] = buf[7] << 8;
188 predictor[1] = buf[6] << 8;
189 }
190 SE_16BIT(predictor[0]);
191 SE_16BIT(predictor[1]);
192
193 /* decode the samples */
194 for (in = 8, out = 0; in < buf_size; in++, out++) {
195 predictor[channel_number] += s->roq_square_array[buf[in]];
aee481ce 196 predictor[channel_number] = av_clip_int16(predictor[channel_number]);
3ef8be2b
MM
197 output_samples[out] = predictor[channel_number];
198
199 /* toggle channel */
200 channel_number ^= s->channels - 1;
201 }
202 break;
203
204 case CODEC_ID_INTERPLAY_DPCM:
b10529b4 205 in = 6; /* skip over the stream mask and stream length */
fead30d4 206 predictor[0] = AV_RL16(&buf[in]);
b10529b4
MM
207 in += 2;
208 SE_16BIT(predictor[0])
209 output_samples[out++] = predictor[0];
210 if (s->channels == 2) {
fead30d4 211 predictor[1] = AV_RL16(&buf[in]);
3ef8be2b 212 in += 2;
b10529b4
MM
213 SE_16BIT(predictor[1])
214 output_samples[out++] = predictor[1];
3ef8be2b
MM
215 }
216
217 while (in < buf_size) {
218 predictor[channel_number] += interplay_delta_table[buf[in++]];
aee481ce 219 predictor[channel_number] = av_clip_int16(predictor[channel_number]);
3ef8be2b
MM
220 output_samples[out++] = predictor[channel_number];
221
222 /* toggle channel */
223 channel_number ^= s->channels - 1;
224 }
225
3ef8be2b 226 break;
9937e686
MM
227
228 case CODEC_ID_XAN_DPCM:
229 in = 0;
230 shift[0] = shift[1] = 4;
fead30d4 231 predictor[0] = AV_RL16(&buf[in]);
9937e686
MM
232 in += 2;
233 SE_16BIT(predictor[0]);
234 if (s->channels == 2) {
fead30d4 235 predictor[1] = AV_RL16(&buf[in]);
9937e686
MM
236 in += 2;
237 SE_16BIT(predictor[1]);
238 }
239
240 while (in < buf_size) {
241 byte = buf[in++];
242 diff = (byte & 0xFC) << 8;
243 if ((byte & 0x03) == 3)
244 shift[channel_number]++;
245 else
246 shift[channel_number] -= (2 * (byte & 3));
247 /* saturate the shifter to a lower limit of 0 */
248 if (shift[channel_number] < 0)
249 shift[channel_number] = 0;
250
251 diff >>= shift[channel_number];
252 predictor[channel_number] += diff;
253
aee481ce 254 predictor[channel_number] = av_clip_int16(predictor[channel_number]);
9937e686
MM
255 output_samples[out++] = predictor[channel_number];
256
257 /* toggle channel */
258 channel_number ^= s->channels - 1;
259 }
260 break;
d08d7142
MM
261 case CODEC_ID_SOL_DPCM:
262 in = 0;
263 if (avctx->codec_tag != 3) {
a2085a7e
KS
264 if(*data_size/4 < buf_size)
265 return -1;
d08d7142
MM
266 while (in < buf_size) {
267 int n1, n2;
268 n1 = (buf[in] >> 4) & 0xF;
269 n2 = buf[in++] & 0xF;
270 s->sample[0] += s->sol_table[n1];
efeb298e 271 if (s->sample[0] < 0) s->sample[0] = 0;
d08d7142
MM
272 if (s->sample[0] > 255) s->sample[0] = 255;
273 output_samples[out++] = (s->sample[0] - 128) << 8;
274 s->sample[s->channels - 1] += s->sol_table[n2];
275 if (s->sample[s->channels - 1] < 0) s->sample[s->channels - 1] = 0;
276 if (s->sample[s->channels - 1] > 255) s->sample[s->channels - 1] = 255;
277 output_samples[out++] = (s->sample[s->channels - 1] - 128) << 8;
278 }
279 } else {
280 while (in < buf_size) {
281 int n;
282 n = buf[in++];
283 if (n & 0x80) s->sample[channel_number] -= s->sol_table[n & 0x7F];
284 else s->sample[channel_number] += s->sol_table[n & 0x7F];
aee481ce 285 s->sample[channel_number] = av_clip_int16(s->sample[channel_number]);
d08d7142
MM
286 output_samples[out++] = s->sample[channel_number];
287 /* toggle channel */
288 channel_number ^= s->channels - 1;
289 }
290 }
291 break;
3ef8be2b
MM
292 }
293
294 *data_size = out * sizeof(short);
295 return buf_size;
296}
297
0f3af710 298#define DPCM_DECODER(id, name, long_name_) \
ad8df79b
AJ
299AVCodec name ## _decoder = { \
300 #name, \
301 CODEC_TYPE_AUDIO, \
302 id, \
303 sizeof(DPCMContext), \
304 dpcm_decode_init, \
305 NULL, \
306 NULL, \
307 dpcm_decode_frame, \
fe4bf374 308 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
3ef8be2b
MM
309};
310
0f3af710
SS
311DPCM_DECODER(CODEC_ID_INTERPLAY_DPCM, interplay_dpcm, "Interplay DPCM");
312DPCM_DECODER(CODEC_ID_ROQ_DPCM, roq_dpcm, "id RoQ DPCM");
313DPCM_DECODER(CODEC_ID_SOL_DPCM, sol_dpcm, "Sol DPCM");
314DPCM_DECODER(CODEC_ID_XAN_DPCM, xan_dpcm, "Xan DPCM");