adpcmdec: use planar sample format for adpcm_ima_wav
[ffmpeg.git] / libavcodec / adpcm.c
1 /*
2  * Copyright (c) 2001-2003 The ffmpeg Project
3  *
4  * This file is part of Libav.
5  *
6  * Libav is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * Libav is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with Libav; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 #include "avcodec.h"
21 #include "get_bits.h"
22 #include "put_bits.h"
23 #include "bytestream.h"
24 #include "adpcm.h"
25 #include "adpcm_data.h"
26
27 /**
28  * @file
29  * ADPCM decoders
30  * First version by Francois Revol (revol@free.fr)
31  * Fringe ADPCM codecs (e.g., DK3, DK4, Westwood)
32  *   by Mike Melanson (melanson@pcisys.net)
33  * CD-ROM XA ADPCM codec by BERO
34  * EA ADPCM decoder by Robin Kay (komadori@myrealbox.com)
35  * EA ADPCM R1/R2/R3 decoder by Peter Ross (pross@xvid.org)
36  * EA IMA EACS decoder by Peter Ross (pross@xvid.org)
37  * EA IMA SEAD decoder by Peter Ross (pross@xvid.org)
38  * EA ADPCM XAS decoder by Peter Ross (pross@xvid.org)
39  * MAXIS EA ADPCM decoder by Robert Marston (rmarston@gmail.com)
40  * THP ADPCM decoder by Marco Gerards (mgerards@xs4all.nl)
41  *
42  * Features and limitations:
43  *
44  * Reference documents:
45  * http://wiki.multimedia.cx/index.php?title=Category:ADPCM_Audio_Codecs
46  * http://www.pcisys.net/~melanson/codecs/simpleaudio.html [dead]
47  * http://www.geocities.com/SiliconValley/8682/aud3.txt [dead]
48  * http://openquicktime.sourceforge.net/
49  * XAnim sources (xa_codec.c) http://xanim.polter.net/
50  * http://www.cs.ucla.edu/~leec/mediabench/applications.html [dead]
51  * SoX source code http://sox.sourceforge.net/
52  *
53  * CD-ROM XA:
54  * http://ku-www.ss.titech.ac.jp/~yatsushi/xaadpcm.html [dead]
55  * vagpack & depack http://homepages.compuserve.de/bITmASTER32/psx-index.html [dead]
56  * readstr http://www.geocities.co.jp/Playtown/2004/
57  */
58
59 /* These are for CD-ROM XA ADPCM */
60 static const int xa_adpcm_table[5][2] = {
61     {   0,   0 },
62     {  60,   0 },
63     { 115, -52 },
64     {  98, -55 },
65     { 122, -60 }
66 };
67
68 static const int ea_adpcm_table[] = {
69     0,  240,  460,  392,
70     0,    0, -208, -220,
71     0,    1,    3,    4,
72     7,    8,   10,   11,
73     0,   -1,   -3,   -4
74 };
75
76 // padded to zero where table size is less then 16
77 static const int swf_index_tables[4][16] = {
78     /*2*/ { -1, 2 },
79     /*3*/ { -1, -1, 2, 4 },
80     /*4*/ { -1, -1, -1, -1, 2, 4, 6, 8 },
81     /*5*/ { -1, -1, -1, -1, -1, -1, -1, -1, 1, 2, 4, 6, 8, 10, 13, 16 }
82 };
83
84 /* end of tables */
85
86 typedef struct ADPCMDecodeContext {
87     AVFrame frame;
88     ADPCMChannelStatus status[6];
89     int vqa_version;                /**< VQA version. Used for ADPCM_IMA_WS */
90 } ADPCMDecodeContext;
91
92 static av_cold int adpcm_decode_init(AVCodecContext * avctx)
93 {
94     ADPCMDecodeContext *c = avctx->priv_data;
95     unsigned int min_channels = 1;
96     unsigned int max_channels = 2;
97
98     switch(avctx->codec->id) {
99     case AV_CODEC_ID_ADPCM_EA:
100         min_channels = 2;
101         break;
102     case AV_CODEC_ID_ADPCM_EA_R1:
103     case AV_CODEC_ID_ADPCM_EA_R2:
104     case AV_CODEC_ID_ADPCM_EA_R3:
105     case AV_CODEC_ID_ADPCM_EA_XAS:
106         max_channels = 6;
107         break;
108     }
109     if (avctx->channels < min_channels || avctx->channels > max_channels) {
110         av_log(avctx, AV_LOG_ERROR, "Invalid number of channels\n");
111         return AVERROR(EINVAL);
112     }
113
114     switch(avctx->codec->id) {
115     case AV_CODEC_ID_ADPCM_CT:
116         c->status[0].step = c->status[1].step = 511;
117         break;
118     case AV_CODEC_ID_ADPCM_IMA_WAV:
119         if (avctx->bits_per_coded_sample != 4) {
120             av_log(avctx, AV_LOG_ERROR, "Only 4-bit ADPCM IMA WAV files are supported\n");
121             return -1;
122         }
123         break;
124     case AV_CODEC_ID_ADPCM_IMA_APC:
125         if (avctx->extradata && avctx->extradata_size >= 8) {
126             c->status[0].predictor = AV_RL32(avctx->extradata);
127             c->status[1].predictor = AV_RL32(avctx->extradata + 4);
128         }
129         break;
130     case AV_CODEC_ID_ADPCM_IMA_WS:
131         if (avctx->extradata && avctx->extradata_size >= 2)
132             c->vqa_version = AV_RL16(avctx->extradata);
133         break;
134     default:
135         break;
136     }
137
138     switch(avctx->codec->id) {
139         case AV_CODEC_ID_ADPCM_IMA_QT:
140         case AV_CODEC_ID_ADPCM_IMA_WAV:
141             avctx->sample_fmt = AV_SAMPLE_FMT_S16P;
142             break;
143         default:
144             avctx->sample_fmt = AV_SAMPLE_FMT_S16;
145     }
146
147     avcodec_get_frame_defaults(&c->frame);
148     avctx->coded_frame = &c->frame;
149
150     return 0;
151 }
152
153 static inline short adpcm_ima_expand_nibble(ADPCMChannelStatus *c, char nibble, int shift)
154 {
155     int step_index;
156     int predictor;
157     int sign, delta, diff, step;
158
159     step = ff_adpcm_step_table[c->step_index];
160     step_index = c->step_index + ff_adpcm_index_table[(unsigned)nibble];
161     step_index = av_clip(step_index, 0, 88);
162
163     sign = nibble & 8;
164     delta = nibble & 7;
165     /* perform direct multiplication instead of series of jumps proposed by
166      * the reference ADPCM implementation since modern CPUs can do the mults
167      * quickly enough */
168     diff = ((2 * delta + 1) * step) >> shift;
169     predictor = c->predictor;
170     if (sign) predictor -= diff;
171     else predictor += diff;
172
173     c->predictor = av_clip_int16(predictor);
174     c->step_index = step_index;
175
176     return (short)c->predictor;
177 }
178
179 static inline int adpcm_ima_qt_expand_nibble(ADPCMChannelStatus *c, int nibble, int shift)
180 {
181     int step_index;
182     int predictor;
183     int diff, step;
184
185     step = ff_adpcm_step_table[c->step_index];
186     step_index = c->step_index + ff_adpcm_index_table[nibble];
187     step_index = av_clip(step_index, 0, 88);
188
189     diff = step >> 3;
190     if (nibble & 4) diff += step;
191     if (nibble & 2) diff += step >> 1;
192     if (nibble & 1) diff += step >> 2;
193
194     if (nibble & 8)
195         predictor = c->predictor - diff;
196     else
197         predictor = c->predictor + diff;
198
199     c->predictor = av_clip_int16(predictor);
200     c->step_index = step_index;
201
202     return c->predictor;
203 }
204
205 static inline short adpcm_ms_expand_nibble(ADPCMChannelStatus *c, int nibble)
206 {
207     int predictor;
208
209     predictor = (((c->sample1) * (c->coeff1)) + ((c->sample2) * (c->coeff2))) / 64;
210     predictor += ((nibble & 0x08)?(nibble - 0x10):(nibble)) * c->idelta;
211
212     c->sample2 = c->sample1;
213     c->sample1 = av_clip_int16(predictor);
214     c->idelta = (ff_adpcm_AdaptationTable[(int)nibble] * c->idelta) >> 8;
215     if (c->idelta < 16) c->idelta = 16;
216
217     return c->sample1;
218 }
219
220 static inline short adpcm_ct_expand_nibble(ADPCMChannelStatus *c, char nibble)
221 {
222     int sign, delta, diff;
223     int new_step;
224
225     sign = nibble & 8;
226     delta = nibble & 7;
227     /* perform direct multiplication instead of series of jumps proposed by
228      * the reference ADPCM implementation since modern CPUs can do the mults
229      * quickly enough */
230     diff = ((2 * delta + 1) * c->step) >> 3;
231     /* predictor update is not so trivial: predictor is multiplied on 254/256 before updating */
232     c->predictor = ((c->predictor * 254) >> 8) + (sign ? -diff : diff);
233     c->predictor = av_clip_int16(c->predictor);
234     /* calculate new step and clamp it to range 511..32767 */
235     new_step = (ff_adpcm_AdaptationTable[nibble & 7] * c->step) >> 8;
236     c->step = av_clip(new_step, 511, 32767);
237
238     return (short)c->predictor;
239 }
240
241 static inline short adpcm_sbpro_expand_nibble(ADPCMChannelStatus *c, char nibble, int size, int shift)
242 {
243     int sign, delta, diff;
244
245     sign = nibble & (1<<(size-1));
246     delta = nibble & ((1<<(size-1))-1);
247     diff = delta << (7 + c->step + shift);
248
249     /* clamp result */
250     c->predictor = av_clip(c->predictor + (sign ? -diff : diff), -16384,16256);
251
252     /* calculate new step */
253     if (delta >= (2*size - 3) && c->step < 3)
254         c->step++;
255     else if (delta == 0 && c->step > 0)
256         c->step--;
257
258     return (short) c->predictor;
259 }
260
261 static inline short adpcm_yamaha_expand_nibble(ADPCMChannelStatus *c, unsigned char nibble)
262 {
263     if(!c->step) {
264         c->predictor = 0;
265         c->step = 127;
266     }
267
268     c->predictor += (c->step * ff_adpcm_yamaha_difflookup[nibble]) / 8;
269     c->predictor = av_clip_int16(c->predictor);
270     c->step = (c->step * ff_adpcm_yamaha_indexscale[nibble]) >> 8;
271     c->step = av_clip(c->step, 127, 24567);
272     return c->predictor;
273 }
274
275 static int xa_decode(AVCodecContext *avctx,
276                      short *out, const unsigned char *in,
277                      ADPCMChannelStatus *left, ADPCMChannelStatus *right, int inc)
278 {
279     int i, j;
280     int shift,filter,f0,f1;
281     int s_1,s_2;
282     int d,s,t;
283
284     for(i=0;i<4;i++) {
285
286         shift  = 12 - (in[4+i*2] & 15);
287         filter = in[4+i*2] >> 4;
288         if (filter > 4) {
289             av_log(avctx, AV_LOG_ERROR,
290                    "Invalid XA-ADPCM filter %d (max. allowed is 4)\n",
291                    filter);
292             return AVERROR_INVALIDDATA;
293         }
294         f0 = xa_adpcm_table[filter][0];
295         f1 = xa_adpcm_table[filter][1];
296
297         s_1 = left->sample1;
298         s_2 = left->sample2;
299
300         for(j=0;j<28;j++) {
301             d = in[16+i+j*4];
302
303             t = sign_extend(d, 4);
304             s = ( t<<shift ) + ((s_1*f0 + s_2*f1+32)>>6);
305             s_2 = s_1;
306             s_1 = av_clip_int16(s);
307             *out = s_1;
308             out += inc;
309         }
310
311         if (inc==2) { /* stereo */
312             left->sample1 = s_1;
313             left->sample2 = s_2;
314             s_1 = right->sample1;
315             s_2 = right->sample2;
316             out = out + 1 - 28*2;
317         }
318
319         shift  = 12 - (in[5+i*2] & 15);
320         filter = in[5+i*2] >> 4;
321         if (filter > 4) {
322             av_log(avctx, AV_LOG_ERROR,
323                    "Invalid XA-ADPCM filter %d (max. allowed is 4)\n",
324                    filter);
325             return AVERROR_INVALIDDATA;
326         }
327         f0 = xa_adpcm_table[filter][0];
328         f1 = xa_adpcm_table[filter][1];
329
330         for(j=0;j<28;j++) {
331             d = in[16+i+j*4];
332
333             t = sign_extend(d >> 4, 4);
334             s = ( t<<shift ) + ((s_1*f0 + s_2*f1+32)>>6);
335             s_2 = s_1;
336             s_1 = av_clip_int16(s);
337             *out = s_1;
338             out += inc;
339         }
340
341         if (inc==2) { /* stereo */
342             right->sample1 = s_1;
343             right->sample2 = s_2;
344             out -= 1;
345         } else {
346             left->sample1 = s_1;
347             left->sample2 = s_2;
348         }
349     }
350
351     return 0;
352 }
353
354 static void adpcm_swf_decode(AVCodecContext *avctx, const uint8_t *buf, int buf_size, int16_t *samples)
355 {
356     ADPCMDecodeContext *c = avctx->priv_data;
357     GetBitContext gb;
358     const int *table;
359     int k0, signmask, nb_bits, count;
360     int size = buf_size*8;
361     int i;
362
363     init_get_bits(&gb, buf, size);
364
365     //read bits & initial values
366     nb_bits = get_bits(&gb, 2)+2;
367     table = swf_index_tables[nb_bits-2];
368     k0 = 1 << (nb_bits-2);
369     signmask = 1 << (nb_bits-1);
370
371     while (get_bits_count(&gb) <= size - 22*avctx->channels) {
372         for (i = 0; i < avctx->channels; i++) {
373             *samples++ = c->status[i].predictor = get_sbits(&gb, 16);
374             c->status[i].step_index = get_bits(&gb, 6);
375         }
376
377         for (count = 0; get_bits_count(&gb) <= size - nb_bits*avctx->channels && count < 4095; count++) {
378             int i;
379
380             for (i = 0; i < avctx->channels; i++) {
381                 // similar to IMA adpcm
382                 int delta = get_bits(&gb, nb_bits);
383                 int step = ff_adpcm_step_table[c->status[i].step_index];
384                 long vpdiff = 0; // vpdiff = (delta+0.5)*step/4
385                 int k = k0;
386
387                 do {
388                     if (delta & k)
389                         vpdiff += step;
390                     step >>= 1;
391                     k >>= 1;
392                 } while(k);
393                 vpdiff += step;
394
395                 if (delta & signmask)
396                     c->status[i].predictor -= vpdiff;
397                 else
398                     c->status[i].predictor += vpdiff;
399
400                 c->status[i].step_index += table[delta & (~signmask)];
401
402                 c->status[i].step_index = av_clip(c->status[i].step_index, 0, 88);
403                 c->status[i].predictor = av_clip_int16(c->status[i].predictor);
404
405                 *samples++ = c->status[i].predictor;
406             }
407         }
408     }
409 }
410
411 /**
412  * Get the number of samples that will be decoded from the packet.
413  * In one case, this is actually the maximum number of samples possible to
414  * decode with the given buf_size.
415  *
416  * @param[out] coded_samples set to the number of samples as coded in the
417  *                           packet, or 0 if the codec does not encode the
418  *                           number of samples in each frame.
419  */
420 static int get_nb_samples(AVCodecContext *avctx, GetByteContext *gb,
421                           int buf_size, int *coded_samples)
422 {
423     ADPCMDecodeContext *s = avctx->priv_data;
424     int nb_samples        = 0;
425     int ch                = avctx->channels;
426     int has_coded_samples = 0;
427     int header_size;
428
429     *coded_samples = 0;
430
431     switch (avctx->codec->id) {
432     /* constant, only check buf_size */
433     case AV_CODEC_ID_ADPCM_EA_XAS:
434         if (buf_size < 76 * ch)
435             return 0;
436         nb_samples = 128;
437         break;
438     case AV_CODEC_ID_ADPCM_IMA_QT:
439         if (buf_size < 34 * ch)
440             return 0;
441         nb_samples = 64;
442         break;
443     /* simple 4-bit adpcm */
444     case AV_CODEC_ID_ADPCM_CT:
445     case AV_CODEC_ID_ADPCM_IMA_APC:
446     case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
447     case AV_CODEC_ID_ADPCM_IMA_WS:
448     case AV_CODEC_ID_ADPCM_YAMAHA:
449         nb_samples = buf_size * 2 / ch;
450         break;
451     }
452     if (nb_samples)
453         return nb_samples;
454
455     /* simple 4-bit adpcm, with header */
456     header_size = 0;
457     switch (avctx->codec->id) {
458         case AV_CODEC_ID_ADPCM_4XM:
459         case AV_CODEC_ID_ADPCM_IMA_ISS:     header_size = 4 * ch;      break;
460         case AV_CODEC_ID_ADPCM_IMA_AMV:     header_size = 8;           break;
461         case AV_CODEC_ID_ADPCM_IMA_SMJPEG:  header_size = 4;           break;
462     }
463     if (header_size > 0)
464         return (buf_size - header_size) * 2 / ch;
465
466     /* more complex formats */
467     switch (avctx->codec->id) {
468     case AV_CODEC_ID_ADPCM_EA:
469         has_coded_samples = 1;
470         *coded_samples  = bytestream2_get_le32(gb);
471         *coded_samples -= *coded_samples % 28;
472         nb_samples      = (buf_size - 12) / 30 * 28;
473         break;
474     case AV_CODEC_ID_ADPCM_IMA_EA_EACS:
475         has_coded_samples = 1;
476         *coded_samples = bytestream2_get_le32(gb);
477         nb_samples     = (buf_size - (4 + 8 * ch)) * 2 / ch;
478         break;
479     case AV_CODEC_ID_ADPCM_EA_MAXIS_XA:
480         nb_samples = (buf_size - ch) / ch * 2;
481         break;
482     case AV_CODEC_ID_ADPCM_EA_R1:
483     case AV_CODEC_ID_ADPCM_EA_R2:
484     case AV_CODEC_ID_ADPCM_EA_R3:
485         /* maximum number of samples */
486         /* has internal offsets and a per-frame switch to signal raw 16-bit */
487         has_coded_samples = 1;
488         switch (avctx->codec->id) {
489         case AV_CODEC_ID_ADPCM_EA_R1:
490             header_size    = 4 + 9 * ch;
491             *coded_samples = bytestream2_get_le32(gb);
492             break;
493         case AV_CODEC_ID_ADPCM_EA_R2:
494             header_size    = 4 + 5 * ch;
495             *coded_samples = bytestream2_get_le32(gb);
496             break;
497         case AV_CODEC_ID_ADPCM_EA_R3:
498             header_size    = 4 + 5 * ch;
499             *coded_samples = bytestream2_get_be32(gb);
500             break;
501         }
502         *coded_samples -= *coded_samples % 28;
503         nb_samples      = (buf_size - header_size) * 2 / ch;
504         nb_samples     -= nb_samples % 28;
505         break;
506     case AV_CODEC_ID_ADPCM_IMA_DK3:
507         if (avctx->block_align > 0)
508             buf_size = FFMIN(buf_size, avctx->block_align);
509         nb_samples = ((buf_size - 16) * 2 / 3 * 4) / ch;
510         break;
511     case AV_CODEC_ID_ADPCM_IMA_DK4:
512         if (avctx->block_align > 0)
513             buf_size = FFMIN(buf_size, avctx->block_align);
514         nb_samples = 1 + (buf_size - 4 * ch) * 2 / ch;
515         break;
516     case AV_CODEC_ID_ADPCM_IMA_WAV:
517         if (avctx->block_align > 0)
518             buf_size = FFMIN(buf_size, avctx->block_align);
519         nb_samples = 1 + (buf_size - 4 * ch) / (4 * ch) * 8;
520         break;
521     case AV_CODEC_ID_ADPCM_MS:
522         if (avctx->block_align > 0)
523             buf_size = FFMIN(buf_size, avctx->block_align);
524         nb_samples = 2 + (buf_size - 7 * ch) * 2 / ch;
525         break;
526     case AV_CODEC_ID_ADPCM_SBPRO_2:
527     case AV_CODEC_ID_ADPCM_SBPRO_3:
528     case AV_CODEC_ID_ADPCM_SBPRO_4:
529     {
530         int samples_per_byte;
531         switch (avctx->codec->id) {
532         case AV_CODEC_ID_ADPCM_SBPRO_2: samples_per_byte = 4; break;
533         case AV_CODEC_ID_ADPCM_SBPRO_3: samples_per_byte = 3; break;
534         case AV_CODEC_ID_ADPCM_SBPRO_4: samples_per_byte = 2; break;
535         }
536         if (!s->status[0].step_index) {
537             nb_samples++;
538             buf_size -= ch;
539         }
540         nb_samples += buf_size * samples_per_byte / ch;
541         break;
542     }
543     case AV_CODEC_ID_ADPCM_SWF:
544     {
545         int buf_bits       = buf_size * 8 - 2;
546         int nbits          = (bytestream2_get_byte(gb) >> 6) + 2;
547         int block_hdr_size = 22 * ch;
548         int block_size     = block_hdr_size + nbits * ch * 4095;
549         int nblocks        = buf_bits / block_size;
550         int bits_left      = buf_bits - nblocks * block_size;
551         nb_samples         = nblocks * 4096;
552         if (bits_left >= block_hdr_size)
553             nb_samples += 1 + (bits_left - block_hdr_size) / (nbits * ch);
554         break;
555     }
556     case AV_CODEC_ID_ADPCM_THP:
557         has_coded_samples = 1;
558         bytestream2_skip(gb, 4); // channel size
559         *coded_samples  = bytestream2_get_be32(gb);
560         *coded_samples -= *coded_samples % 14;
561         nb_samples      = (buf_size - 80) / (8 * ch) * 14;
562         break;
563     case AV_CODEC_ID_ADPCM_XA:
564         nb_samples = (buf_size / 128) * 224 / ch;
565         break;
566     }
567
568     /* validate coded sample count */
569     if (has_coded_samples && (*coded_samples <= 0 || *coded_samples > nb_samples))
570         return AVERROR_INVALIDDATA;
571
572     return nb_samples;
573 }
574
575 static int adpcm_decode_frame(AVCodecContext *avctx, void *data,
576                               int *got_frame_ptr, AVPacket *avpkt)
577 {
578     const uint8_t *buf = avpkt->data;
579     int buf_size = avpkt->size;
580     ADPCMDecodeContext *c = avctx->priv_data;
581     ADPCMChannelStatus *cs;
582     int n, m, channel, i;
583     short *samples;
584     int16_t **samples_p;
585     int st; /* stereo */
586     int count1, count2;
587     int nb_samples, coded_samples, ret;
588     GetByteContext gb;
589
590     bytestream2_init(&gb, buf, buf_size);
591     nb_samples = get_nb_samples(avctx, &gb, buf_size, &coded_samples);
592     if (nb_samples <= 0) {
593         av_log(avctx, AV_LOG_ERROR, "invalid number of samples in packet\n");
594         return AVERROR_INVALIDDATA;
595     }
596
597     /* get output buffer */
598     c->frame.nb_samples = nb_samples;
599     if ((ret = avctx->get_buffer(avctx, &c->frame)) < 0) {
600         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
601         return ret;
602     }
603     samples = (short *)c->frame.data[0];
604     samples_p = (int16_t **)c->frame.extended_data;
605
606     /* use coded_samples when applicable */
607     /* it is always <= nb_samples, so the output buffer will be large enough */
608     if (coded_samples) {
609         if (coded_samples != nb_samples)
610             av_log(avctx, AV_LOG_WARNING, "mismatch in coded sample count\n");
611         c->frame.nb_samples = nb_samples = coded_samples;
612     }
613
614     st = avctx->channels == 2 ? 1 : 0;
615
616     switch(avctx->codec->id) {
617     case AV_CODEC_ID_ADPCM_IMA_QT:
618         /* In QuickTime, IMA is encoded by chunks of 34 bytes (=64 samples).
619            Channel data is interleaved per-chunk. */
620         for (channel = 0; channel < avctx->channels; channel++) {
621             int predictor;
622             int step_index;
623             cs = &(c->status[channel]);
624             /* (pppppp) (piiiiiii) */
625
626             /* Bits 15-7 are the _top_ 9 bits of the 16-bit initial predictor value */
627             predictor = sign_extend(bytestream2_get_be16u(&gb), 16);
628             step_index = predictor & 0x7F;
629             predictor &= ~0x7F;
630
631             if (cs->step_index == step_index) {
632                 int diff = predictor - cs->predictor;
633                 if (diff < 0)
634                     diff = - diff;
635                 if (diff > 0x7f)
636                     goto update;
637             } else {
638             update:
639                 cs->step_index = step_index;
640                 cs->predictor = predictor;
641             }
642
643             if (cs->step_index > 88u){
644                 av_log(avctx, AV_LOG_ERROR, "ERROR: step_index[%d] = %i\n",
645                        channel, cs->step_index);
646                 return AVERROR_INVALIDDATA;
647             }
648
649             samples = samples_p[channel];
650
651             for (m = 0; m < 64; m += 2) {
652                 int byte = bytestream2_get_byteu(&gb);
653                 samples[m    ] = adpcm_ima_qt_expand_nibble(cs, byte & 0x0F, 3);
654                 samples[m + 1] = adpcm_ima_qt_expand_nibble(cs, byte >> 4  , 3);
655             }
656         }
657         break;
658     case AV_CODEC_ID_ADPCM_IMA_WAV:
659         for(i=0; i<avctx->channels; i++){
660             cs = &(c->status[i]);
661             cs->predictor = samples_p[i][0] = sign_extend(bytestream2_get_le16u(&gb), 16);
662
663             cs->step_index = sign_extend(bytestream2_get_le16u(&gb), 16);
664             if (cs->step_index > 88u){
665                 av_log(avctx, AV_LOG_ERROR, "ERROR: step_index[%d] = %i\n",
666                        i, cs->step_index);
667                 return AVERROR_INVALIDDATA;
668             }
669         }
670
671         for (n = 0; n < (nb_samples - 1) / 8; n++) {
672             for (i = 0; i < avctx->channels; i++) {
673                 cs = &c->status[i];
674                 samples = &samples_p[i][1 + n * 8];
675                 for (m = 0; m < 8; m += 2) {
676                     int v = bytestream2_get_byteu(&gb);
677                     samples[m    ] = adpcm_ima_expand_nibble(cs, v & 0x0F, 3);
678                     samples[m + 1] = adpcm_ima_expand_nibble(cs, v >> 4  , 3);
679                 }
680             }
681         }
682         break;
683     case AV_CODEC_ID_ADPCM_4XM:
684         for (i = 0; i < avctx->channels; i++)
685             c->status[i].predictor = sign_extend(bytestream2_get_le16u(&gb), 16);
686
687         for (i = 0; i < avctx->channels; i++) {
688             c->status[i].step_index = sign_extend(bytestream2_get_le16u(&gb), 16);
689             if (c->status[i].step_index > 88u) {
690                 av_log(avctx, AV_LOG_ERROR, "ERROR: step_index[%d] = %i\n",
691                        i, c->status[i].step_index);
692                 return AVERROR_INVALIDDATA;
693             }
694         }
695
696         for (i = 0; i < avctx->channels; i++) {
697             samples = (short *)c->frame.data[0] + i;
698             cs = &c->status[i];
699             for (n = nb_samples >> 1; n > 0; n--) {
700                 int v = bytestream2_get_byteu(&gb);
701                 *samples = adpcm_ima_expand_nibble(cs, v & 0x0F, 4);
702                 samples += avctx->channels;
703                 *samples = adpcm_ima_expand_nibble(cs, v >> 4  , 4);
704                 samples += avctx->channels;
705             }
706         }
707         break;
708     case AV_CODEC_ID_ADPCM_MS:
709     {
710         int block_predictor;
711
712         block_predictor = bytestream2_get_byteu(&gb);
713         if (block_predictor > 6) {
714             av_log(avctx, AV_LOG_ERROR, "ERROR: block_predictor[0] = %d\n",
715                    block_predictor);
716             return AVERROR_INVALIDDATA;
717         }
718         c->status[0].coeff1 = ff_adpcm_AdaptCoeff1[block_predictor];
719         c->status[0].coeff2 = ff_adpcm_AdaptCoeff2[block_predictor];
720         if (st) {
721             block_predictor = bytestream2_get_byteu(&gb);
722             if (block_predictor > 6) {
723                 av_log(avctx, AV_LOG_ERROR, "ERROR: block_predictor[1] = %d\n",
724                        block_predictor);
725                 return AVERROR_INVALIDDATA;
726             }
727             c->status[1].coeff1 = ff_adpcm_AdaptCoeff1[block_predictor];
728             c->status[1].coeff2 = ff_adpcm_AdaptCoeff2[block_predictor];
729         }
730         c->status[0].idelta = sign_extend(bytestream2_get_le16u(&gb), 16);
731         if (st){
732             c->status[1].idelta = sign_extend(bytestream2_get_le16u(&gb), 16);
733         }
734
735         c->status[0].sample1 = sign_extend(bytestream2_get_le16u(&gb), 16);
736         if (st) c->status[1].sample1 = sign_extend(bytestream2_get_le16u(&gb), 16);
737         c->status[0].sample2 = sign_extend(bytestream2_get_le16u(&gb), 16);
738         if (st) c->status[1].sample2 = sign_extend(bytestream2_get_le16u(&gb), 16);
739
740         *samples++ = c->status[0].sample2;
741         if (st) *samples++ = c->status[1].sample2;
742         *samples++ = c->status[0].sample1;
743         if (st) *samples++ = c->status[1].sample1;
744         for(n = (nb_samples - 2) >> (1 - st); n > 0; n--) {
745             int byte = bytestream2_get_byteu(&gb);
746             *samples++ = adpcm_ms_expand_nibble(&c->status[0 ], byte >> 4  );
747             *samples++ = adpcm_ms_expand_nibble(&c->status[st], byte & 0x0F);
748         }
749         break;
750     }
751     case AV_CODEC_ID_ADPCM_IMA_DK4:
752         for (channel = 0; channel < avctx->channels; channel++) {
753             cs = &c->status[channel];
754             cs->predictor  = *samples++ = sign_extend(bytestream2_get_le16u(&gb), 16);
755             cs->step_index = sign_extend(bytestream2_get_le16u(&gb), 16);
756             if (cs->step_index > 88u){
757                 av_log(avctx, AV_LOG_ERROR, "ERROR: step_index[%d] = %i\n",
758                        channel, cs->step_index);
759                 return AVERROR_INVALIDDATA;
760             }
761         }
762         for (n = nb_samples >> (1 - st); n > 0; n--) {
763             int v = bytestream2_get_byteu(&gb);
764             *samples++ = adpcm_ima_expand_nibble(&c->status[0 ], v >> 4  , 3);
765             *samples++ = adpcm_ima_expand_nibble(&c->status[st], v & 0x0F, 3);
766         }
767         break;
768     case AV_CODEC_ID_ADPCM_IMA_DK3:
769     {
770         int last_byte = 0;
771         int nibble;
772         int decode_top_nibble_next = 0;
773         int diff_channel;
774         const int16_t *samples_end = samples + avctx->channels * nb_samples;
775
776         bytestream2_skipu(&gb, 10);
777         c->status[0].predictor  = sign_extend(bytestream2_get_le16u(&gb), 16);
778         c->status[1].predictor  = sign_extend(bytestream2_get_le16u(&gb), 16);
779         c->status[0].step_index = bytestream2_get_byteu(&gb);
780         c->status[1].step_index = bytestream2_get_byteu(&gb);
781         if (c->status[0].step_index > 88u || c->status[1].step_index > 88u){
782             av_log(avctx, AV_LOG_ERROR, "ERROR: step_index = %i/%i\n",
783                    c->status[0].step_index, c->status[1].step_index);
784             return AVERROR_INVALIDDATA;
785         }
786         /* sign extend the predictors */
787         diff_channel = c->status[1].predictor;
788
789         /* DK3 ADPCM support macro */
790 #define DK3_GET_NEXT_NIBBLE() \
791     if (decode_top_nibble_next) { \
792         nibble = last_byte >> 4; \
793         decode_top_nibble_next = 0; \
794     } else { \
795         last_byte = bytestream2_get_byteu(&gb); \
796         nibble = last_byte & 0x0F; \
797         decode_top_nibble_next = 1; \
798     }
799
800         while (samples < samples_end) {
801
802             /* for this algorithm, c->status[0] is the sum channel and
803              * c->status[1] is the diff channel */
804
805             /* process the first predictor of the sum channel */
806             DK3_GET_NEXT_NIBBLE();
807             adpcm_ima_expand_nibble(&c->status[0], nibble, 3);
808
809             /* process the diff channel predictor */
810             DK3_GET_NEXT_NIBBLE();
811             adpcm_ima_expand_nibble(&c->status[1], nibble, 3);
812
813             /* process the first pair of stereo PCM samples */
814             diff_channel = (diff_channel + c->status[1].predictor) / 2;
815             *samples++ = c->status[0].predictor + c->status[1].predictor;
816             *samples++ = c->status[0].predictor - c->status[1].predictor;
817
818             /* process the second predictor of the sum channel */
819             DK3_GET_NEXT_NIBBLE();
820             adpcm_ima_expand_nibble(&c->status[0], nibble, 3);
821
822             /* process the second pair of stereo PCM samples */
823             diff_channel = (diff_channel + c->status[1].predictor) / 2;
824             *samples++ = c->status[0].predictor + c->status[1].predictor;
825             *samples++ = c->status[0].predictor - c->status[1].predictor;
826         }
827         break;
828     }
829     case AV_CODEC_ID_ADPCM_IMA_ISS:
830         for (channel = 0; channel < avctx->channels; channel++) {
831             cs = &c->status[channel];
832             cs->predictor  = sign_extend(bytestream2_get_le16u(&gb), 16);
833             cs->step_index = sign_extend(bytestream2_get_le16u(&gb), 16);
834             if (cs->step_index > 88u){
835                 av_log(avctx, AV_LOG_ERROR, "ERROR: step_index[%d] = %i\n",
836                        channel, cs->step_index);
837                 return AVERROR_INVALIDDATA;
838             }
839         }
840
841         for (n = nb_samples >> (1 - st); n > 0; n--) {
842             int v1, v2;
843             int v = bytestream2_get_byteu(&gb);
844             /* nibbles are swapped for mono */
845             if (st) {
846                 v1 = v >> 4;
847                 v2 = v & 0x0F;
848             } else {
849                 v2 = v >> 4;
850                 v1 = v & 0x0F;
851             }
852             *samples++ = adpcm_ima_expand_nibble(&c->status[0 ], v1, 3);
853             *samples++ = adpcm_ima_expand_nibble(&c->status[st], v2, 3);
854         }
855         break;
856     case AV_CODEC_ID_ADPCM_IMA_APC:
857         while (bytestream2_get_bytes_left(&gb) > 0) {
858             int v = bytestream2_get_byteu(&gb);
859             *samples++ = adpcm_ima_expand_nibble(&c->status[0],  v >> 4  , 3);
860             *samples++ = adpcm_ima_expand_nibble(&c->status[st], v & 0x0F, 3);
861         }
862         break;
863     case AV_CODEC_ID_ADPCM_IMA_WS:
864         if (c->vqa_version == 3) {
865             for (channel = 0; channel < avctx->channels; channel++) {
866                 int16_t *smp = samples + channel;
867
868                 for (n = nb_samples / 2; n > 0; n--) {
869                     int v = bytestream2_get_byteu(&gb);
870                     *smp = adpcm_ima_expand_nibble(&c->status[channel], v >> 4  , 3);
871                     smp += avctx->channels;
872                     *smp = adpcm_ima_expand_nibble(&c->status[channel], v & 0x0F, 3);
873                     smp += avctx->channels;
874                 }
875             }
876         } else {
877             for (n = nb_samples / 2; n > 0; n--) {
878                 for (channel = 0; channel < avctx->channels; channel++) {
879                     int v = bytestream2_get_byteu(&gb);
880                     *samples++  = adpcm_ima_expand_nibble(&c->status[channel], v >> 4  , 3);
881                     samples[st] = adpcm_ima_expand_nibble(&c->status[channel], v & 0x0F, 3);
882                 }
883                 samples += avctx->channels;
884             }
885         }
886         bytestream2_seek(&gb, 0, SEEK_END);
887         break;
888     case AV_CODEC_ID_ADPCM_XA:
889         while (bytestream2_get_bytes_left(&gb) >= 128) {
890             if ((ret = xa_decode(avctx, samples, buf + bytestream2_tell(&gb), &c->status[0],
891                                  &c->status[1], avctx->channels)) < 0)
892                 return ret;
893             bytestream2_skipu(&gb, 128);
894             samples += 28 * 8;
895         }
896         break;
897     case AV_CODEC_ID_ADPCM_IMA_EA_EACS:
898         for (i=0; i<=st; i++) {
899             c->status[i].step_index = bytestream2_get_le32u(&gb);
900             if (c->status[i].step_index > 88u) {
901                 av_log(avctx, AV_LOG_ERROR, "ERROR: step_index[%d] = %i\n",
902                        i, c->status[i].step_index);
903                 return AVERROR_INVALIDDATA;
904             }
905         }
906         for (i=0; i<=st; i++)
907             c->status[i].predictor  = bytestream2_get_le32u(&gb);
908
909         for (n = nb_samples >> (1 - st); n > 0; n--) {
910             int byte   = bytestream2_get_byteu(&gb);
911             *samples++ = adpcm_ima_expand_nibble(&c->status[0],  byte >> 4,   3);
912             *samples++ = adpcm_ima_expand_nibble(&c->status[st], byte & 0x0F, 3);
913         }
914         break;
915     case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
916         for (n = nb_samples >> (1 - st); n > 0; n--) {
917             int byte = bytestream2_get_byteu(&gb);
918             *samples++ = adpcm_ima_expand_nibble(&c->status[0],  byte >> 4,   6);
919             *samples++ = adpcm_ima_expand_nibble(&c->status[st], byte & 0x0F, 6);
920         }
921         break;
922     case AV_CODEC_ID_ADPCM_EA:
923     {
924         int previous_left_sample, previous_right_sample;
925         int current_left_sample, current_right_sample;
926         int next_left_sample, next_right_sample;
927         int coeff1l, coeff2l, coeff1r, coeff2r;
928         int shift_left, shift_right;
929
930         /* Each EA ADPCM frame has a 12-byte header followed by 30-byte pieces,
931            each coding 28 stereo samples. */
932
933         current_left_sample   = sign_extend(bytestream2_get_le16u(&gb), 16);
934         previous_left_sample  = sign_extend(bytestream2_get_le16u(&gb), 16);
935         current_right_sample  = sign_extend(bytestream2_get_le16u(&gb), 16);
936         previous_right_sample = sign_extend(bytestream2_get_le16u(&gb), 16);
937
938         for (count1 = 0; count1 < nb_samples / 28; count1++) {
939             int byte = bytestream2_get_byteu(&gb);
940             coeff1l = ea_adpcm_table[ byte >> 4       ];
941             coeff2l = ea_adpcm_table[(byte >> 4  ) + 4];
942             coeff1r = ea_adpcm_table[ byte & 0x0F];
943             coeff2r = ea_adpcm_table[(byte & 0x0F) + 4];
944
945             byte = bytestream2_get_byteu(&gb);
946             shift_left  = 20 - (byte >> 4);
947             shift_right = 20 - (byte & 0x0F);
948
949             for (count2 = 0; count2 < 28; count2++) {
950                 byte = bytestream2_get_byteu(&gb);
951                 next_left_sample  = sign_extend(byte >> 4, 4) << shift_left;
952                 next_right_sample = sign_extend(byte,      4) << shift_right;
953
954                 next_left_sample = (next_left_sample +
955                     (current_left_sample * coeff1l) +
956                     (previous_left_sample * coeff2l) + 0x80) >> 8;
957                 next_right_sample = (next_right_sample +
958                     (current_right_sample * coeff1r) +
959                     (previous_right_sample * coeff2r) + 0x80) >> 8;
960
961                 previous_left_sample = current_left_sample;
962                 current_left_sample = av_clip_int16(next_left_sample);
963                 previous_right_sample = current_right_sample;
964                 current_right_sample = av_clip_int16(next_right_sample);
965                 *samples++ = current_left_sample;
966                 *samples++ = current_right_sample;
967             }
968         }
969
970         bytestream2_skip(&gb, 2); // Skip terminating 0x0000
971
972         break;
973     }
974     case AV_CODEC_ID_ADPCM_EA_MAXIS_XA:
975     {
976         int coeff[2][2], shift[2];
977
978         for(channel = 0; channel < avctx->channels; channel++) {
979             int byte = bytestream2_get_byteu(&gb);
980             for (i=0; i<2; i++)
981                 coeff[channel][i] = ea_adpcm_table[(byte >> 4) + 4*i];
982             shift[channel] = 20 - (byte & 0x0F);
983         }
984         for (count1 = 0; count1 < nb_samples / 2; count1++) {
985             int byte[2];
986
987             byte[0] = bytestream2_get_byteu(&gb);
988             if (st) byte[1] = bytestream2_get_byteu(&gb);
989             for(i = 4; i >= 0; i-=4) { /* Pairwise samples LL RR (st) or LL LL (mono) */
990                 for(channel = 0; channel < avctx->channels; channel++) {
991                     int sample = sign_extend(byte[channel] >> i, 4) << shift[channel];
992                     sample = (sample +
993                              c->status[channel].sample1 * coeff[channel][0] +
994                              c->status[channel].sample2 * coeff[channel][1] + 0x80) >> 8;
995                     c->status[channel].sample2 = c->status[channel].sample1;
996                     c->status[channel].sample1 = av_clip_int16(sample);
997                     *samples++ = c->status[channel].sample1;
998                 }
999             }
1000         }
1001         bytestream2_seek(&gb, 0, SEEK_END);
1002         break;
1003     }
1004     case AV_CODEC_ID_ADPCM_EA_R1:
1005     case AV_CODEC_ID_ADPCM_EA_R2:
1006     case AV_CODEC_ID_ADPCM_EA_R3: {
1007         /* channel numbering
1008            2chan: 0=fl, 1=fr
1009            4chan: 0=fl, 1=rl, 2=fr, 3=rr
1010            6chan: 0=fl, 1=c,  2=fr, 3=rl,  4=rr, 5=sub */
1011         const int big_endian = avctx->codec->id == AV_CODEC_ID_ADPCM_EA_R3;
1012         int previous_sample, current_sample, next_sample;
1013         int coeff1, coeff2;
1014         int shift;
1015         unsigned int channel;
1016         uint16_t *samplesC;
1017         int count = 0;
1018         int offsets[6];
1019
1020         for (channel=0; channel<avctx->channels; channel++)
1021             offsets[channel] = (big_endian ? bytestream2_get_be32(&gb) :
1022                                              bytestream2_get_le32(&gb)) +
1023                                (avctx->channels + 1) * 4;
1024
1025         for (channel=0; channel<avctx->channels; channel++) {
1026             bytestream2_seek(&gb, offsets[channel], SEEK_SET);
1027             samplesC = samples + channel;
1028
1029             if (avctx->codec->id == AV_CODEC_ID_ADPCM_EA_R1) {
1030                 current_sample  = sign_extend(bytestream2_get_le16(&gb), 16);
1031                 previous_sample = sign_extend(bytestream2_get_le16(&gb), 16);
1032             } else {
1033                 current_sample  = c->status[channel].predictor;
1034                 previous_sample = c->status[channel].prev_sample;
1035             }
1036
1037             for (count1 = 0; count1 < nb_samples / 28; count1++) {
1038                 int byte = bytestream2_get_byte(&gb);
1039                 if (byte == 0xEE) {  /* only seen in R2 and R3 */
1040                     current_sample  = sign_extend(bytestream2_get_be16(&gb), 16);
1041                     previous_sample = sign_extend(bytestream2_get_be16(&gb), 16);
1042
1043                     for (count2=0; count2<28; count2++) {
1044                         *samplesC = sign_extend(bytestream2_get_be16(&gb), 16);
1045                         samplesC += avctx->channels;
1046                     }
1047                 } else {
1048                     coeff1 = ea_adpcm_table[ byte >> 4     ];
1049                     coeff2 = ea_adpcm_table[(byte >> 4) + 4];
1050                     shift = 20 - (byte & 0x0F);
1051
1052                     for (count2=0; count2<28; count2++) {
1053                         if (count2 & 1)
1054                             next_sample = sign_extend(byte,    4) << shift;
1055                         else {
1056                             byte = bytestream2_get_byte(&gb);
1057                             next_sample = sign_extend(byte >> 4, 4) << shift;
1058                         }
1059
1060                         next_sample += (current_sample  * coeff1) +
1061                                        (previous_sample * coeff2);
1062                         next_sample = av_clip_int16(next_sample >> 8);
1063
1064                         previous_sample = current_sample;
1065                         current_sample  = next_sample;
1066                         *samplesC = current_sample;
1067                         samplesC += avctx->channels;
1068                     }
1069                 }
1070             }
1071             if (!count) {
1072                 count = count1;
1073             } else if (count != count1) {
1074                 av_log(avctx, AV_LOG_WARNING, "per-channel sample count mismatch\n");
1075                 count = FFMAX(count, count1);
1076             }
1077
1078             if (avctx->codec->id != AV_CODEC_ID_ADPCM_EA_R1) {
1079                 c->status[channel].predictor   = current_sample;
1080                 c->status[channel].prev_sample = previous_sample;
1081             }
1082         }
1083
1084         c->frame.nb_samples = count * 28;
1085         bytestream2_seek(&gb, 0, SEEK_END);
1086         break;
1087     }
1088     case AV_CODEC_ID_ADPCM_EA_XAS:
1089         for (channel=0; channel<avctx->channels; channel++) {
1090             int coeff[2][4], shift[4];
1091             short *s2, *s = &samples[channel];
1092             for (n=0; n<4; n++, s+=32*avctx->channels) {
1093                 int val = sign_extend(bytestream2_get_le16u(&gb), 16);
1094                 for (i=0; i<2; i++)
1095                     coeff[i][n] = ea_adpcm_table[(val&0x0F)+4*i];
1096                 s[0] = val & ~0x0F;
1097
1098                 val = sign_extend(bytestream2_get_le16u(&gb), 16);
1099                 shift[n] = 20 - (val & 0x0F);
1100                 s[avctx->channels] = val & ~0x0F;
1101             }
1102
1103             for (m=2; m<32; m+=2) {
1104                 s = &samples[m*avctx->channels + channel];
1105                 for (n=0; n<4; n++, s+=32*avctx->channels) {
1106                     int byte = bytestream2_get_byteu(&gb);
1107                     for (s2=s, i=0; i<8; i+=4, s2+=avctx->channels) {
1108                         int level = sign_extend(byte >> (4 - i), 4) << shift[n];
1109                         int pred  = s2[-1*avctx->channels] * coeff[0][n]
1110                                   + s2[-2*avctx->channels] * coeff[1][n];
1111                         s2[0] = av_clip_int16((level + pred + 0x80) >> 8);
1112                     }
1113                 }
1114             }
1115         }
1116         break;
1117     case AV_CODEC_ID_ADPCM_IMA_AMV:
1118     case AV_CODEC_ID_ADPCM_IMA_SMJPEG:
1119         if (avctx->codec->id == AV_CODEC_ID_ADPCM_IMA_AMV) {
1120             c->status[0].predictor = sign_extend(bytestream2_get_le16u(&gb), 16);
1121             c->status[0].step_index = bytestream2_get_le16u(&gb);
1122             bytestream2_skipu(&gb, 4);
1123         } else {
1124             c->status[0].predictor = sign_extend(bytestream2_get_be16u(&gb), 16);
1125             c->status[0].step_index = bytestream2_get_byteu(&gb);
1126             bytestream2_skipu(&gb, 1);
1127         }
1128         if (c->status[0].step_index > 88u) {
1129             av_log(avctx, AV_LOG_ERROR, "ERROR: step_index = %i\n",
1130                    c->status[0].step_index);
1131             return AVERROR_INVALIDDATA;
1132         }
1133
1134         for (n = nb_samples >> (1 - st); n > 0; n--) {
1135             int hi, lo, v = bytestream2_get_byteu(&gb);
1136
1137             if (avctx->codec->id == AV_CODEC_ID_ADPCM_IMA_AMV) {
1138                 hi = v & 0x0F;
1139                 lo = v >> 4;
1140             } else {
1141                 lo = v & 0x0F;
1142                 hi = v >> 4;
1143             }
1144
1145             *samples++ = adpcm_ima_expand_nibble(&c->status[0], lo, 3);
1146             *samples++ = adpcm_ima_expand_nibble(&c->status[0], hi, 3);
1147         }
1148         break;
1149     case AV_CODEC_ID_ADPCM_CT:
1150         for (n = nb_samples >> (1 - st); n > 0; n--) {
1151             int v = bytestream2_get_byteu(&gb);
1152             *samples++ = adpcm_ct_expand_nibble(&c->status[0 ], v >> 4  );
1153             *samples++ = adpcm_ct_expand_nibble(&c->status[st], v & 0x0F);
1154         }
1155         break;
1156     case AV_CODEC_ID_ADPCM_SBPRO_4:
1157     case AV_CODEC_ID_ADPCM_SBPRO_3:
1158     case AV_CODEC_ID_ADPCM_SBPRO_2:
1159         if (!c->status[0].step_index) {
1160             /* the first byte is a raw sample */
1161             *samples++ = 128 * (bytestream2_get_byteu(&gb) - 0x80);
1162             if (st)
1163                 *samples++ = 128 * (bytestream2_get_byteu(&gb) - 0x80);
1164             c->status[0].step_index = 1;
1165             nb_samples--;
1166         }
1167         if (avctx->codec->id == AV_CODEC_ID_ADPCM_SBPRO_4) {
1168             for (n = nb_samples >> (1 - st); n > 0; n--) {
1169                 int byte = bytestream2_get_byteu(&gb);
1170                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1171                                                        byte >> 4,   4, 0);
1172                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[st],
1173                                                        byte & 0x0F, 4, 0);
1174             }
1175         } else if (avctx->codec->id == AV_CODEC_ID_ADPCM_SBPRO_3) {
1176             for (n = nb_samples / 3; n > 0; n--) {
1177                 int byte = bytestream2_get_byteu(&gb);
1178                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1179                                                         byte >> 5        , 3, 0);
1180                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1181                                                        (byte >> 2) & 0x07, 3, 0);
1182                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1183                                                         byte & 0x03,       2, 0);
1184             }
1185         } else {
1186             for (n = nb_samples >> (2 - st); n > 0; n--) {
1187                 int byte = bytestream2_get_byteu(&gb);
1188                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1189                                                         byte >> 6        , 2, 2);
1190                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[st],
1191                                                        (byte >> 4) & 0x03, 2, 2);
1192                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1193                                                        (byte >> 2) & 0x03, 2, 2);
1194                 *samples++ = adpcm_sbpro_expand_nibble(&c->status[st],
1195                                                         byte & 0x03,       2, 2);
1196             }
1197         }
1198         break;
1199     case AV_CODEC_ID_ADPCM_SWF:
1200         adpcm_swf_decode(avctx, buf, buf_size, samples);
1201         bytestream2_seek(&gb, 0, SEEK_END);
1202         break;
1203     case AV_CODEC_ID_ADPCM_YAMAHA:
1204         for (n = nb_samples >> (1 - st); n > 0; n--) {
1205             int v = bytestream2_get_byteu(&gb);
1206             *samples++ = adpcm_yamaha_expand_nibble(&c->status[0 ], v & 0x0F);
1207             *samples++ = adpcm_yamaha_expand_nibble(&c->status[st], v >> 4  );
1208         }
1209         break;
1210     case AV_CODEC_ID_ADPCM_THP:
1211     {
1212         int table[2][16];
1213         int prev[2][2];
1214         int ch;
1215
1216         for (i = 0; i < 2; i++)
1217             for (n = 0; n < 16; n++)
1218                 table[i][n] = sign_extend(bytestream2_get_be16u(&gb), 16);
1219
1220         /* Initialize the previous sample.  */
1221         for (i = 0; i < 2; i++)
1222             for (n = 0; n < 2; n++)
1223                 prev[i][n] = sign_extend(bytestream2_get_be16u(&gb), 16);
1224
1225         for (ch = 0; ch <= st; ch++) {
1226             samples = (short *)c->frame.data[0] + ch;
1227
1228             /* Read in every sample for this channel.  */
1229             for (i = 0; i < nb_samples / 14; i++) {
1230                 int byte = bytestream2_get_byteu(&gb);
1231                 int index = (byte >> 4) & 7;
1232                 unsigned int exp = byte & 0x0F;
1233                 int factor1 = table[ch][index * 2];
1234                 int factor2 = table[ch][index * 2 + 1];
1235
1236                 /* Decode 14 samples.  */
1237                 for (n = 0; n < 14; n++) {
1238                     int32_t sampledat;
1239
1240                     if (n & 1) {
1241                         sampledat = sign_extend(byte, 4);
1242                     } else {
1243                         byte = bytestream2_get_byteu(&gb);
1244                         sampledat = sign_extend(byte >> 4, 4);
1245                     }
1246
1247                     sampledat = ((prev[ch][0]*factor1
1248                                 + prev[ch][1]*factor2) >> 11) + (sampledat << exp);
1249                     *samples = av_clip_int16(sampledat);
1250                     prev[ch][1] = prev[ch][0];
1251                     prev[ch][0] = *samples++;
1252
1253                     /* In case of stereo, skip one sample, this sample
1254                        is for the other channel.  */
1255                     samples += st;
1256                 }
1257             }
1258         }
1259         break;
1260     }
1261
1262     default:
1263         return -1;
1264     }
1265
1266     *got_frame_ptr   = 1;
1267     *(AVFrame *)data = c->frame;
1268
1269     return bytestream2_tell(&gb);
1270 }
1271
1272
1273 static const enum AVSampleFormat sample_fmts_s16[]  = { AV_SAMPLE_FMT_S16,
1274                                                         AV_SAMPLE_FMT_NONE };
1275 static const enum AVSampleFormat sample_fmts_s16p[] = { AV_SAMPLE_FMT_S16,
1276                                                         AV_SAMPLE_FMT_NONE };
1277
1278 #define ADPCM_DECODER(id_, sample_fmts_, name_, long_name_) \
1279 AVCodec ff_ ## name_ ## _decoder = {                        \
1280     .name           = #name_,                               \
1281     .type           = AVMEDIA_TYPE_AUDIO,                   \
1282     .id             = id_,                                  \
1283     .priv_data_size = sizeof(ADPCMDecodeContext),           \
1284     .init           = adpcm_decode_init,                    \
1285     .decode         = adpcm_decode_frame,                   \
1286     .capabilities   = CODEC_CAP_DR1,                        \
1287     .long_name      = NULL_IF_CONFIG_SMALL(long_name_),     \
1288     .sample_fmts    = sample_fmts_,                         \
1289 }
1290
1291 /* Note: Do not forget to add new entries to the Makefile as well. */
1292 ADPCM_DECODER(AV_CODEC_ID_ADPCM_4XM,         sample_fmts_s16,  adpcm_4xm,         "ADPCM 4X Movie");
1293 ADPCM_DECODER(AV_CODEC_ID_ADPCM_CT,          sample_fmts_s16,  adpcm_ct,          "ADPCM Creative Technology");
1294 ADPCM_DECODER(AV_CODEC_ID_ADPCM_EA,          sample_fmts_s16,  adpcm_ea,          "ADPCM Electronic Arts");
1295 ADPCM_DECODER(AV_CODEC_ID_ADPCM_EA_MAXIS_XA, sample_fmts_s16,  adpcm_ea_maxis_xa, "ADPCM Electronic Arts Maxis CDROM XA");
1296 ADPCM_DECODER(AV_CODEC_ID_ADPCM_EA_R1,       sample_fmts_s16,  adpcm_ea_r1,       "ADPCM Electronic Arts R1");
1297 ADPCM_DECODER(AV_CODEC_ID_ADPCM_EA_R2,       sample_fmts_s16,  adpcm_ea_r2,       "ADPCM Electronic Arts R2");
1298 ADPCM_DECODER(AV_CODEC_ID_ADPCM_EA_R3,       sample_fmts_s16,  adpcm_ea_r3,       "ADPCM Electronic Arts R3");
1299 ADPCM_DECODER(AV_CODEC_ID_ADPCM_EA_XAS,      sample_fmts_s16,  adpcm_ea_xas,      "ADPCM Electronic Arts XAS");
1300 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_AMV,     sample_fmts_s16,  adpcm_ima_amv,     "ADPCM IMA AMV");
1301 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_APC,     sample_fmts_s16,  adpcm_ima_apc,     "ADPCM IMA CRYO APC");
1302 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_DK3,     sample_fmts_s16,  adpcm_ima_dk3,     "ADPCM IMA Duck DK3");
1303 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_DK4,     sample_fmts_s16,  adpcm_ima_dk4,     "ADPCM IMA Duck DK4");
1304 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_EA_EACS, sample_fmts_s16,  adpcm_ima_ea_eacs, "ADPCM IMA Electronic Arts EACS");
1305 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_EA_SEAD, sample_fmts_s16,  adpcm_ima_ea_sead, "ADPCM IMA Electronic Arts SEAD");
1306 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_ISS,     sample_fmts_s16,  adpcm_ima_iss,     "ADPCM IMA Funcom ISS");
1307 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_QT,      sample_fmts_s16p, adpcm_ima_qt,      "ADPCM IMA QuickTime");
1308 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_SMJPEG,  sample_fmts_s16,  adpcm_ima_smjpeg,  "ADPCM IMA Loki SDL MJPEG");
1309 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_WAV,     sample_fmts_s16p, adpcm_ima_wav,     "ADPCM IMA WAV");
1310 ADPCM_DECODER(AV_CODEC_ID_ADPCM_IMA_WS,      sample_fmts_s16,  adpcm_ima_ws,      "ADPCM IMA Westwood");
1311 ADPCM_DECODER(AV_CODEC_ID_ADPCM_MS,          sample_fmts_s16,  adpcm_ms,          "ADPCM Microsoft");
1312 ADPCM_DECODER(AV_CODEC_ID_ADPCM_SBPRO_2,     sample_fmts_s16,  adpcm_sbpro_2,     "ADPCM Sound Blaster Pro 2-bit");
1313 ADPCM_DECODER(AV_CODEC_ID_ADPCM_SBPRO_3,     sample_fmts_s16,  adpcm_sbpro_3,     "ADPCM Sound Blaster Pro 2.6-bit");
1314 ADPCM_DECODER(AV_CODEC_ID_ADPCM_SBPRO_4,     sample_fmts_s16,  adpcm_sbpro_4,     "ADPCM Sound Blaster Pro 4-bit");
1315 ADPCM_DECODER(AV_CODEC_ID_ADPCM_SWF,         sample_fmts_s16,  adpcm_swf,         "ADPCM Shockwave Flash");
1316 ADPCM_DECODER(AV_CODEC_ID_ADPCM_THP,         sample_fmts_s16,  adpcm_thp,         "ADPCM Nintendo Gamecube THP");
1317 ADPCM_DECODER(AV_CODEC_ID_ADPCM_XA,          sample_fmts_s16,  adpcm_xa,          "ADPCM CDROM XA");
1318 ADPCM_DECODER(AV_CODEC_ID_ADPCM_YAMAHA,      sample_fmts_s16,  adpcm_yamaha,      "ADPCM Yamaha");