Statistics
| Branch: | Revision:

ffmpeg / libavcodec / adpcm.c @ d764e3ec

History | View | Annotate | Download (64.6 KB)

1
/*
2
 * ADPCM codecs
3
 * Copyright (c) 2001-2003 The ffmpeg Project
4
 *
5
 * This file is part of FFmpeg.
6
 *
7
 * FFmpeg is free software; you can redistribute it and/or
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
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
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
18
 * License along with FFmpeg; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21
#include "avcodec.h"
22
#include "get_bits.h"
23
#include "put_bits.h"
24
#include "bytestream.h"
25

    
26
/**
27
 * @file
28
 * ADPCM codecs.
29
 * First version by Francois Revol (revol@free.fr)
30
 * Fringe ADPCM codecs (e.g., DK3, DK4, Westwood)
31
 *   by Mike Melanson (melanson@pcisys.net)
32
 * CD-ROM XA ADPCM codec by BERO
33
 * EA ADPCM decoder by Robin Kay (komadori@myrealbox.com)
34
 * EA ADPCM R1/R2/R3 decoder by Peter Ross (pross@xvid.org)
35
 * EA IMA EACS decoder by Peter Ross (pross@xvid.org)
36
 * EA IMA SEAD decoder by Peter Ross (pross@xvid.org)
37
 * EA ADPCM XAS decoder by Peter Ross (pross@xvid.org)
38
 * MAXIS EA ADPCM decoder by Robert Marston (rmarston@gmail.com)
39
 * THP ADPCM decoder by Marco Gerards (mgerards@xs4all.nl)
40
 *
41
 * Features and limitations:
42
 *
43
 * Reference documents:
44
 * http://www.pcisys.net/~melanson/codecs/simpleaudio.html
45
 * http://www.geocities.com/SiliconValley/8682/aud3.txt
46
 * http://openquicktime.sourceforge.net/plugins.htm
47
 * XAnim sources (xa_codec.c) http://www.rasnaimaging.com/people/lapus/download.html
48
 * http://www.cs.ucla.edu/~leec/mediabench/applications.html
49
 * SoX source code http://home.sprynet.com/~cbagwell/sox.html
50
 *
51
 * CD-ROM XA:
52
 * http://ku-www.ss.titech.ac.jp/~yatsushi/xaadpcm.html
53
 * vagpack & depack http://homepages.compuserve.de/bITmASTER32/psx-index.html
54
 * readstr http://www.geocities.co.jp/Playtown/2004/
55
 */
56

    
57
#define BLKSIZE 1024
58

    
59
/* step_table[] and index_table[] are from the ADPCM reference source */
60
/* This is the index table: */
61
static const int index_table[16] = {
62
    -1, -1, -1, -1, 2, 4, 6, 8,
63
    -1, -1, -1, -1, 2, 4, 6, 8,
64
};
65

    
66
/**
67
 * This is the step table. Note that many programs use slight deviations from
68
 * this table, but such deviations are negligible:
69
 */
70
static const int step_table[89] = {
71
    7, 8, 9, 10, 11, 12, 13, 14, 16, 17,
72
    19, 21, 23, 25, 28, 31, 34, 37, 41, 45,
73
    50, 55, 60, 66, 73, 80, 88, 97, 107, 118,
74
    130, 143, 157, 173, 190, 209, 230, 253, 279, 307,
75
    337, 371, 408, 449, 494, 544, 598, 658, 724, 796,
76
    876, 963, 1060, 1166, 1282, 1411, 1552, 1707, 1878, 2066,
77
    2272, 2499, 2749, 3024, 3327, 3660, 4026, 4428, 4871, 5358,
78
    5894, 6484, 7132, 7845, 8630, 9493, 10442, 11487, 12635, 13899,
79
    15289, 16818, 18500, 20350, 22385, 24623, 27086, 29794, 32767
80
};
81

    
82
/* These are for MS-ADPCM */
83
/* AdaptationTable[], AdaptCoeff1[], and AdaptCoeff2[] are from libsndfile */
84
static const int AdaptationTable[] = {
85
        230, 230, 230, 230, 307, 409, 512, 614,
86
        768, 614, 512, 409, 307, 230, 230, 230
87
};
88

    
89
/** Divided by 4 to fit in 8-bit integers */
90
static const uint8_t AdaptCoeff1[] = {
91
        64, 128, 0, 48, 60, 115, 98
92
};
93

    
94
/** Divided by 4 to fit in 8-bit integers */
95
static const int8_t AdaptCoeff2[] = {
96
        0, -64, 0, 16, 0, -52, -58
97
};
98

    
99
/* These are for CD-ROM XA ADPCM */
100
static const int xa_adpcm_table[5][2] = {
101
   {   0,   0 },
102
   {  60,   0 },
103
   { 115, -52 },
104
   {  98, -55 },
105
   { 122, -60 }
106
};
107

    
108
static const int ea_adpcm_table[] = {
109
    0, 240, 460, 392, 0, 0, -208, -220, 0, 1,
110
    3, 4, 7, 8, 10, 11, 0, -1, -3, -4
111
};
112

    
113
// padded to zero where table size is less then 16
114
static const int swf_index_tables[4][16] = {
115
    /*2*/ { -1, 2 },
116
    /*3*/ { -1, -1, 2, 4 },
117
    /*4*/ { -1, -1, -1, -1, 2, 4, 6, 8 },
118
    /*5*/ { -1, -1, -1, -1, -1, -1, -1, -1, 1, 2, 4, 6, 8, 10, 13, 16 }
119
};
120

    
121
static const int yamaha_indexscale[] = {
122
    230, 230, 230, 230, 307, 409, 512, 614,
123
    230, 230, 230, 230, 307, 409, 512, 614
124
};
125

    
126
static const int yamaha_difflookup[] = {
127
    1, 3, 5, 7, 9, 11, 13, 15,
128
    -1, -3, -5, -7, -9, -11, -13, -15
129
};
130

    
131
/* end of tables */
132

    
133
typedef struct ADPCMChannelStatus {
134
    int predictor;
135
    short int step_index;
136
    int step;
137
    /* for encoding */
138
    int prev_sample;
139

    
140
    /* MS version */
141
    short sample1;
142
    short sample2;
143
    int coeff1;
144
    int coeff2;
145
    int idelta;
146
} ADPCMChannelStatus;
147

    
148
typedef struct TrellisPath {
149
    int nibble;
150
    int prev;
151
} TrellisPath;
152

    
153
typedef struct TrellisNode {
154
    uint32_t ssd;
155
    int path;
156
    int sample1;
157
    int sample2;
158
    int step;
159
} TrellisNode;
160

    
161
typedef struct ADPCMContext {
162
    ADPCMChannelStatus status[6];
163
    TrellisPath *paths;
164
    TrellisNode *node_buf;
165
    TrellisNode **nodep_buf;
166
    uint8_t *trellis_hash;
167
} ADPCMContext;
168

    
169
#define FREEZE_INTERVAL 128
170

    
171
/* XXX: implement encoding */
172

    
173
#if CONFIG_ENCODERS
174
static av_cold int adpcm_encode_init(AVCodecContext *avctx)
175
{
176
    ADPCMContext *s = avctx->priv_data;
177
    uint8_t *extradata;
178
    int i;
179
    if (avctx->channels > 2)
180
        return -1; /* only stereo or mono =) */
181

    
182
    if(avctx->trellis && (unsigned)avctx->trellis > 16U){
183
        av_log(avctx, AV_LOG_ERROR, "invalid trellis size\n");
184
        return -1;
185
    }
186

    
187
    if (avctx->trellis) {
188
        int frontier = 1 << avctx->trellis;
189
        int max_paths =  frontier * FREEZE_INTERVAL;
190
        FF_ALLOC_OR_GOTO(avctx, s->paths,     max_paths * sizeof(*s->paths), error);
191
        FF_ALLOC_OR_GOTO(avctx, s->node_buf,  2 * frontier * sizeof(*s->node_buf), error);
192
        FF_ALLOC_OR_GOTO(avctx, s->nodep_buf, 2 * frontier * sizeof(*s->nodep_buf), error);
193
        FF_ALLOC_OR_GOTO(avctx, s->trellis_hash, 65536 * sizeof(*s->trellis_hash), error);
194
    }
195

    
196
    switch(avctx->codec->id) {
197
    case CODEC_ID_ADPCM_IMA_WAV:
198
        avctx->frame_size = (BLKSIZE - 4 * avctx->channels) * 8 / (4 * avctx->channels) + 1; /* each 16 bits sample gives one nibble */
199
                                                             /* and we have 4 bytes per channel overhead */
200
        avctx->block_align = BLKSIZE;
201
        /* seems frame_size isn't taken into account... have to buffer the samples :-( */
202
        break;
203
    case CODEC_ID_ADPCM_IMA_QT:
204
        avctx->frame_size = 64;
205
        avctx->block_align = 34 * avctx->channels;
206
        break;
207
    case CODEC_ID_ADPCM_MS:
208
        avctx->frame_size = (BLKSIZE - 7 * avctx->channels) * 2 / avctx->channels + 2; /* each 16 bits sample gives one nibble */
209
                                                             /* and we have 7 bytes per channel overhead */
210
        avctx->block_align = BLKSIZE;
211
        avctx->extradata_size = 32;
212
        extradata = avctx->extradata = av_malloc(avctx->extradata_size);
213
        if (!extradata)
214
            return AVERROR(ENOMEM);
215
        bytestream_put_le16(&extradata, avctx->frame_size);
216
        bytestream_put_le16(&extradata, 7); /* wNumCoef */
217
        for (i = 0; i < 7; i++) {
218
            bytestream_put_le16(&extradata, AdaptCoeff1[i] * 4);
219
            bytestream_put_le16(&extradata, AdaptCoeff2[i] * 4);
220
        }
221
        break;
222
    case CODEC_ID_ADPCM_YAMAHA:
223
        avctx->frame_size = BLKSIZE * avctx->channels;
224
        avctx->block_align = BLKSIZE;
225
        break;
226
    case CODEC_ID_ADPCM_SWF:
227
        if (avctx->sample_rate != 11025 &&
228
            avctx->sample_rate != 22050 &&
229
            avctx->sample_rate != 44100) {
230
            av_log(avctx, AV_LOG_ERROR, "Sample rate must be 11025, 22050 or 44100\n");
231
            goto error;
232
        }
233
        avctx->frame_size = 512 * (avctx->sample_rate / 11025);
234
        break;
235
    default:
236
        goto error;
237
    }
238

    
239
    avctx->coded_frame= avcodec_alloc_frame();
240
    avctx->coded_frame->key_frame= 1;
241

    
242
    return 0;
243
error:
244
    av_freep(&s->paths);
245
    av_freep(&s->node_buf);
246
    av_freep(&s->nodep_buf);
247
    av_freep(&s->trellis_hash);
248
    return -1;
249
}
250

    
251
static av_cold int adpcm_encode_close(AVCodecContext *avctx)
252
{
253
    ADPCMContext *s = avctx->priv_data;
254
    av_freep(&avctx->coded_frame);
255
    av_freep(&s->paths);
256
    av_freep(&s->node_buf);
257
    av_freep(&s->nodep_buf);
258
    av_freep(&s->trellis_hash);
259

    
260
    return 0;
261
}
262

    
263

    
264
static inline unsigned char adpcm_ima_compress_sample(ADPCMChannelStatus *c, short sample)
265
{
266
    int delta = sample - c->prev_sample;
267
    int nibble = FFMIN(7, abs(delta)*4/step_table[c->step_index]) + (delta<0)*8;
268
    c->prev_sample += ((step_table[c->step_index] * yamaha_difflookup[nibble]) / 8);
269
    c->prev_sample = av_clip_int16(c->prev_sample);
270
    c->step_index = av_clip(c->step_index + index_table[nibble], 0, 88);
271
    return nibble;
272
}
273

    
274
static inline unsigned char adpcm_ms_compress_sample(ADPCMChannelStatus *c, short sample)
275
{
276
    int predictor, nibble, bias;
277

    
278
    predictor = (((c->sample1) * (c->coeff1)) + ((c->sample2) * (c->coeff2))) / 64;
279

    
280
    nibble= sample - predictor;
281
    if(nibble>=0) bias= c->idelta/2;
282
    else          bias=-c->idelta/2;
283

    
284
    nibble= (nibble + bias) / c->idelta;
285
    nibble= av_clip(nibble, -8, 7)&0x0F;
286

    
287
    predictor += (signed)((nibble & 0x08)?(nibble - 0x10):(nibble)) * c->idelta;
288

    
289
    c->sample2 = c->sample1;
290
    c->sample1 = av_clip_int16(predictor);
291

    
292
    c->idelta = (AdaptationTable[(int)nibble] * c->idelta) >> 8;
293
    if (c->idelta < 16) c->idelta = 16;
294

    
295
    return nibble;
296
}
297

    
298
static inline unsigned char adpcm_yamaha_compress_sample(ADPCMChannelStatus *c, short sample)
299
{
300
    int nibble, delta;
301

    
302
    if(!c->step) {
303
        c->predictor = 0;
304
        c->step = 127;
305
    }
306

    
307
    delta = sample - c->predictor;
308

    
309
    nibble = FFMIN(7, abs(delta)*4/c->step) + (delta<0)*8;
310

    
311
    c->predictor += ((c->step * yamaha_difflookup[nibble]) / 8);
312
    c->predictor = av_clip_int16(c->predictor);
313
    c->step = (c->step * yamaha_indexscale[nibble]) >> 8;
314
    c->step = av_clip(c->step, 127, 24567);
315

    
316
    return nibble;
317
}
318

    
319
static void adpcm_compress_trellis(AVCodecContext *avctx, const short *samples,
320
                                   uint8_t *dst, ADPCMChannelStatus *c, int n)
321
{
322
    //FIXME 6% faster if frontier is a compile-time constant
323
    ADPCMContext *s = avctx->priv_data;
324
    const int frontier = 1 << avctx->trellis;
325
    const int stride = avctx->channels;
326
    const int version = avctx->codec->id;
327
    TrellisPath *paths = s->paths, *p;
328
    TrellisNode *node_buf = s->node_buf;
329
    TrellisNode **nodep_buf = s->nodep_buf;
330
    TrellisNode **nodes = nodep_buf; // nodes[] is always sorted by .ssd
331
    TrellisNode **nodes_next = nodep_buf + frontier;
332
    int pathn = 0, froze = -1, i, j, k, generation = 0;
333
    uint8_t *hash = s->trellis_hash;
334
    memset(hash, 0xff, 65536 * sizeof(*hash));
335

    
336
    memset(nodep_buf, 0, 2 * frontier * sizeof(*nodep_buf));
337
    nodes[0] = node_buf + frontier;
338
    nodes[0]->ssd = 0;
339
    nodes[0]->path = 0;
340
    nodes[0]->step = c->step_index;
341
    nodes[0]->sample1 = c->sample1;
342
    nodes[0]->sample2 = c->sample2;
343
    if((version == CODEC_ID_ADPCM_IMA_WAV) || (version == CODEC_ID_ADPCM_IMA_QT) || (version == CODEC_ID_ADPCM_SWF))
344
        nodes[0]->sample1 = c->prev_sample;
345
    if(version == CODEC_ID_ADPCM_MS)
346
        nodes[0]->step = c->idelta;
347
    if(version == CODEC_ID_ADPCM_YAMAHA) {
348
        if(c->step == 0) {
349
            nodes[0]->step = 127;
350
            nodes[0]->sample1 = 0;
351
        } else {
352
            nodes[0]->step = c->step;
353
            nodes[0]->sample1 = c->predictor;
354
        }
355
    }
356

    
357
    for(i=0; i<n; i++) {
358
        TrellisNode *t = node_buf + frontier*(i&1);
359
        TrellisNode **u;
360
        int sample = samples[i*stride];
361
        int heap_pos = 0;
362
        memset(nodes_next, 0, frontier*sizeof(TrellisNode*));
363
        for(j=0; j<frontier && nodes[j]; j++) {
364
            // higher j have higher ssd already, so they're likely to yield a suboptimal next sample too
365
            const int range = (j < frontier/2) ? 1 : 0;
366
            const int step = nodes[j]->step;
367
            int nidx;
368
            if(version == CODEC_ID_ADPCM_MS) {
369
                const int predictor = ((nodes[j]->sample1 * c->coeff1) + (nodes[j]->sample2 * c->coeff2)) / 64;
370
                const int div = (sample - predictor) / step;
371
                const int nmin = av_clip(div-range, -8, 6);
372
                const int nmax = av_clip(div+range, -7, 7);
373
                for(nidx=nmin; nidx<=nmax; nidx++) {
374
                    const int nibble = nidx & 0xf;
375
                    int dec_sample = predictor + nidx * step;
376
#define STORE_NODE(NAME, STEP_INDEX)\
377
                    int d;\
378
                    uint32_t ssd;\
379
                    int pos;\
380
                    TrellisNode *u;\
381
                    uint8_t *h;\
382
                    dec_sample = av_clip_int16(dec_sample);\
383
                    d = sample - dec_sample;\
384
                    ssd = nodes[j]->ssd + d*d;\
385
                    /* Collapse any two states with the same previous sample value. \
386
                     * One could also distinguish states by step and by 2nd to last
387
                     * sample, but the effects of that are negligible.
388
                     * Since nodes in the previous generation are iterated
389
                     * through a heap, they're roughly ordered from better to
390
                     * worse, but not strictly ordered. Therefore, an earlier
391
                     * node with the same sample value is better in most cases
392
                     * (and thus the current is skipped), but not strictly
393
                     * in all cases. Only skipping samples where ssd >=
394
                     * ssd of the earlier node with the same sample gives
395
                     * slightly worse quality, though, for some reason. */ \
396
                    h = &hash[(uint16_t) dec_sample];\
397
                    if (*h == generation)\
398
                        goto next_##NAME;\
399
                    if (heap_pos < frontier) {\
400
                        pos = heap_pos++;\
401
                    } else {\
402
                        /* Try to replace one of the leaf nodes with the new \
403
                         * one, but try a different slot each time. */\
404
                        pos = (frontier >> 1) + (heap_pos++ & ((frontier >> 1) - 1));\
405
                        if (ssd > nodes_next[pos]->ssd)\
406
                            goto next_##NAME;\
407
                    }\
408
                    *h = generation;\
409
                    u = nodes_next[pos];\
410
                    if(!u) {\
411
                        assert(pathn < FREEZE_INTERVAL<<avctx->trellis);\
412
                        u = t++;\
413
                        nodes_next[pos] = u;\
414
                        u->path = pathn++;\
415
                    }\
416
                    u->ssd = ssd;\
417
                    u->step = STEP_INDEX;\
418
                    u->sample2 = nodes[j]->sample1;\
419
                    u->sample1 = dec_sample;\
420
                    paths[u->path].nibble = nibble;\
421
                    paths[u->path].prev = nodes[j]->path;\
422
                    /* Sift the newly inserted node down in the heap to \
423
                     * restore the heap property. */\
424
                    while (pos > 0) {\
425
                        int parent = (pos - 1) >> 1;\
426
                        if (nodes_next[parent]->ssd <= ssd)\
427
                            break;\
428
                        FFSWAP(TrellisNode*, nodes_next[parent], nodes_next[pos]);\
429
                        pos = parent;\
430
                    }\
431
                    next_##NAME:;
432
                    STORE_NODE(ms, FFMAX(16, (AdaptationTable[nibble] * step) >> 8));
433
                }
434
            } else if((version == CODEC_ID_ADPCM_IMA_WAV)|| (version == CODEC_ID_ADPCM_IMA_QT)|| (version == CODEC_ID_ADPCM_SWF)) {
435
#define LOOP_NODES(NAME, STEP_TABLE, STEP_INDEX)\
436
                const int predictor = nodes[j]->sample1;\
437
                const int div = (sample - predictor) * 4 / STEP_TABLE;\
438
                int nmin = av_clip(div-range, -7, 6);\
439
                int nmax = av_clip(div+range, -6, 7);\
440
                if(nmin<=0) nmin--; /* distinguish -0 from +0 */\
441
                if(nmax<0) nmax--;\
442
                for(nidx=nmin; nidx<=nmax; nidx++) {\
443
                    const int nibble = nidx<0 ? 7-nidx : nidx;\
444
                    int dec_sample = predictor + (STEP_TABLE * yamaha_difflookup[nibble]) / 8;\
445
                    STORE_NODE(NAME, STEP_INDEX);\
446
                }
447
                LOOP_NODES(ima, step_table[step], av_clip(step + index_table[nibble], 0, 88));
448
            } else { //CODEC_ID_ADPCM_YAMAHA
449
                LOOP_NODES(yamaha, step, av_clip((step * yamaha_indexscale[nibble]) >> 8, 127, 24567));
450
#undef LOOP_NODES
451
#undef STORE_NODE
452
            }
453
        }
454

    
455
        u = nodes;
456
        nodes = nodes_next;
457
        nodes_next = u;
458

    
459
        generation++;
460
        if (generation == 255) {
461
            memset(hash, 0xff, 65536 * sizeof(*hash));
462
            generation = 0;
463
        }
464

    
465
        // prevent overflow
466
        if(nodes[0]->ssd > (1<<28)) {
467
            for(j=1; j<frontier && nodes[j]; j++)
468
                nodes[j]->ssd -= nodes[0]->ssd;
469
            nodes[0]->ssd = 0;
470
        }
471

    
472
        // merge old paths to save memory
473
        if(i == froze + FREEZE_INTERVAL) {
474
            p = &paths[nodes[0]->path];
475
            for(k=i; k>froze; k--) {
476
                dst[k] = p->nibble;
477
                p = &paths[p->prev];
478
            }
479
            froze = i;
480
            pathn = 0;
481
            // other nodes might use paths that don't coincide with the frozen one.
482
            // checking which nodes do so is too slow, so just kill them all.
483
            // this also slightly improves quality, but I don't know why.
484
            memset(nodes+1, 0, (frontier-1)*sizeof(TrellisNode*));
485
        }
486
    }
487

    
488
    p = &paths[nodes[0]->path];
489
    for(i=n-1; i>froze; i--) {
490
        dst[i] = p->nibble;
491
        p = &paths[p->prev];
492
    }
493

    
494
    c->predictor = nodes[0]->sample1;
495
    c->sample1 = nodes[0]->sample1;
496
    c->sample2 = nodes[0]->sample2;
497
    c->step_index = nodes[0]->step;
498
    c->step = nodes[0]->step;
499
    c->idelta = nodes[0]->step;
500
}
501

    
502
static int adpcm_encode_frame(AVCodecContext *avctx,
503
                            unsigned char *frame, int buf_size, void *data)
504
{
505
    int n, i, st;
506
    short *samples;
507
    unsigned char *dst;
508
    ADPCMContext *c = avctx->priv_data;
509
    uint8_t *buf;
510

    
511
    dst = frame;
512
    samples = (short *)data;
513
    st= avctx->channels == 2;
514
/*    n = (BLKSIZE - 4 * avctx->channels) / (2 * 8 * avctx->channels); */
515

    
516
    switch(avctx->codec->id) {
517
    case CODEC_ID_ADPCM_IMA_WAV:
518
        n = avctx->frame_size / 8;
519
            c->status[0].prev_sample = (signed short)samples[0]; /* XXX */
520
/*            c->status[0].step_index = 0; *//* XXX: not sure how to init the state machine */
521
            bytestream_put_le16(&dst, c->status[0].prev_sample);
522
            *dst++ = (unsigned char)c->status[0].step_index;
523
            *dst++ = 0; /* unknown */
524
            samples++;
525
            if (avctx->channels == 2) {
526
                c->status[1].prev_sample = (signed short)samples[0];
527
/*                c->status[1].step_index = 0; */
528
                bytestream_put_le16(&dst, c->status[1].prev_sample);
529
                *dst++ = (unsigned char)c->status[1].step_index;
530
                *dst++ = 0;
531
                samples++;
532
            }
533

    
534
            /* stereo: 4 bytes (8 samples) for left, 4 bytes for right, 4 bytes left, ... */
535
            if(avctx->trellis > 0) {
536
                FF_ALLOC_OR_GOTO(avctx, buf, 2*n*8, error);
537
                adpcm_compress_trellis(avctx, samples, buf, &c->status[0], n*8);
538
                if(avctx->channels == 2)
539
                    adpcm_compress_trellis(avctx, samples+1, buf + n*8, &c->status[1], n*8);
540
                for(i=0; i<n; i++) {
541
                    *dst++ = buf[8*i+0] | (buf[8*i+1] << 4);
542
                    *dst++ = buf[8*i+2] | (buf[8*i+3] << 4);
543
                    *dst++ = buf[8*i+4] | (buf[8*i+5] << 4);
544
                    *dst++ = buf[8*i+6] | (buf[8*i+7] << 4);
545
                    if (avctx->channels == 2) {
546
                        uint8_t *buf1 = buf + n*8;
547
                        *dst++ = buf1[8*i+0] | (buf1[8*i+1] << 4);
548
                        *dst++ = buf1[8*i+2] | (buf1[8*i+3] << 4);
549
                        *dst++ = buf1[8*i+4] | (buf1[8*i+5] << 4);
550
                        *dst++ = buf1[8*i+6] | (buf1[8*i+7] << 4);
551
                    }
552
                }
553
                av_free(buf);
554
            } else
555
            for (; n>0; n--) {
556
                *dst = adpcm_ima_compress_sample(&c->status[0], samples[0]);
557
                *dst |= adpcm_ima_compress_sample(&c->status[0], samples[avctx->channels]) << 4;
558
                dst++;
559
                *dst = adpcm_ima_compress_sample(&c->status[0], samples[avctx->channels * 2]);
560
                *dst |= adpcm_ima_compress_sample(&c->status[0], samples[avctx->channels * 3]) << 4;
561
                dst++;
562
                *dst = adpcm_ima_compress_sample(&c->status[0], samples[avctx->channels * 4]);
563
                *dst |= adpcm_ima_compress_sample(&c->status[0], samples[avctx->channels * 5]) << 4;
564
                dst++;
565
                *dst = adpcm_ima_compress_sample(&c->status[0], samples[avctx->channels * 6]);
566
                *dst |= adpcm_ima_compress_sample(&c->status[0], samples[avctx->channels * 7]) << 4;
567
                dst++;
568
                /* right channel */
569
                if (avctx->channels == 2) {
570
                    *dst = adpcm_ima_compress_sample(&c->status[1], samples[1]);
571
                    *dst |= adpcm_ima_compress_sample(&c->status[1], samples[3]) << 4;
572
                    dst++;
573
                    *dst = adpcm_ima_compress_sample(&c->status[1], samples[5]);
574
                    *dst |= adpcm_ima_compress_sample(&c->status[1], samples[7]) << 4;
575
                    dst++;
576
                    *dst = adpcm_ima_compress_sample(&c->status[1], samples[9]);
577
                    *dst |= adpcm_ima_compress_sample(&c->status[1], samples[11]) << 4;
578
                    dst++;
579
                    *dst = adpcm_ima_compress_sample(&c->status[1], samples[13]);
580
                    *dst |= adpcm_ima_compress_sample(&c->status[1], samples[15]) << 4;
581
                    dst++;
582
                }
583
                samples += 8 * avctx->channels;
584
            }
585
        break;
586
    case CODEC_ID_ADPCM_IMA_QT:
587
    {
588
        int ch, i;
589
        PutBitContext pb;
590
        init_put_bits(&pb, dst, buf_size*8);
591

    
592
        for(ch=0; ch<avctx->channels; ch++){
593
            put_bits(&pb, 9, (c->status[ch].prev_sample + 0x10000) >> 7);
594
            put_bits(&pb, 7, c->status[ch].step_index);
595
            if(avctx->trellis > 0) {
596
                uint8_t buf[64];
597
                adpcm_compress_trellis(avctx, samples+ch, buf, &c->status[ch], 64);
598
                for(i=0; i<64; i++)
599
                    put_bits(&pb, 4, buf[i^1]);
600
                c->status[ch].prev_sample = c->status[ch].predictor & ~0x7F;
601
            } else {
602
                for (i=0; i<64; i+=2){
603
                    int t1, t2;
604
                    t1 = adpcm_ima_compress_sample(&c->status[ch], samples[avctx->channels*(i+0)+ch]);
605
                    t2 = adpcm_ima_compress_sample(&c->status[ch], samples[avctx->channels*(i+1)+ch]);
606
                    put_bits(&pb, 4, t2);
607
                    put_bits(&pb, 4, t1);
608
                }
609
                c->status[ch].prev_sample &= ~0x7F;
610
            }
611
        }
612

    
613
        flush_put_bits(&pb);
614
        dst += put_bits_count(&pb)>>3;
615
        break;
616
    }
617
    case CODEC_ID_ADPCM_SWF:
618
    {
619
        int i;
620
        PutBitContext pb;
621
        init_put_bits(&pb, dst, buf_size*8);
622

    
623
        n = avctx->frame_size-1;
624

    
625
        //Store AdpcmCodeSize
626
        put_bits(&pb, 2, 2);                //Set 4bits flash adpcm format
627

    
628
        //Init the encoder state
629
        for(i=0; i<avctx->channels; i++){
630
            c->status[i].step_index = av_clip(c->status[i].step_index, 0, 63); // clip step so it fits 6 bits
631
            put_sbits(&pb, 16, samples[i]);
632
            put_bits(&pb, 6, c->status[i].step_index);
633
            c->status[i].prev_sample = (signed short)samples[i];
634
        }
635

    
636
        if(avctx->trellis > 0) {
637
            FF_ALLOC_OR_GOTO(avctx, buf, 2*n, error);
638
            adpcm_compress_trellis(avctx, samples+2, buf, &c->status[0], n);
639
            if (avctx->channels == 2)
640
                adpcm_compress_trellis(avctx, samples+3, buf+n, &c->status[1], n);
641
            for(i=0; i<n; i++) {
642
                put_bits(&pb, 4, buf[i]);
643
                if (avctx->channels == 2)
644
                    put_bits(&pb, 4, buf[n+i]);
645
            }
646
            av_free(buf);
647
        } else {
648
            for (i=1; i<avctx->frame_size; i++) {
649
                put_bits(&pb, 4, adpcm_ima_compress_sample(&c->status[0], samples[avctx->channels*i]));
650
                if (avctx->channels == 2)
651
                    put_bits(&pb, 4, adpcm_ima_compress_sample(&c->status[1], samples[2*i+1]));
652
            }
653
        }
654
        flush_put_bits(&pb);
655
        dst += put_bits_count(&pb)>>3;
656
        break;
657
    }
658
    case CODEC_ID_ADPCM_MS:
659
        for(i=0; i<avctx->channels; i++){
660
            int predictor=0;
661

    
662
            *dst++ = predictor;
663
            c->status[i].coeff1 = AdaptCoeff1[predictor];
664
            c->status[i].coeff2 = AdaptCoeff2[predictor];
665
        }
666
        for(i=0; i<avctx->channels; i++){
667
            if (c->status[i].idelta < 16)
668
                c->status[i].idelta = 16;
669

    
670
            bytestream_put_le16(&dst, c->status[i].idelta);
671
        }
672
        for(i=0; i<avctx->channels; i++){
673
            c->status[i].sample2= *samples++;
674
        }
675
        for(i=0; i<avctx->channels; i++){
676
            c->status[i].sample1= *samples++;
677

    
678
            bytestream_put_le16(&dst, c->status[i].sample1);
679
        }
680
        for(i=0; i<avctx->channels; i++)
681
            bytestream_put_le16(&dst, c->status[i].sample2);
682

    
683
        if(avctx->trellis > 0) {
684
            int n = avctx->block_align - 7*avctx->channels;
685
            FF_ALLOC_OR_GOTO(avctx, buf, 2*n, error);
686
            if(avctx->channels == 1) {
687
                adpcm_compress_trellis(avctx, samples, buf, &c->status[0], n);
688
                for(i=0; i<n; i+=2)
689
                    *dst++ = (buf[i] << 4) | buf[i+1];
690
            } else {
691
                adpcm_compress_trellis(avctx, samples, buf, &c->status[0], n);
692
                adpcm_compress_trellis(avctx, samples+1, buf+n, &c->status[1], n);
693
                for(i=0; i<n; i++)
694
                    *dst++ = (buf[i] << 4) | buf[n+i];
695
            }
696
            av_free(buf);
697
        } else
698
        for(i=7*avctx->channels; i<avctx->block_align; i++) {
699
            int nibble;
700
            nibble = adpcm_ms_compress_sample(&c->status[ 0], *samples++)<<4;
701
            nibble|= adpcm_ms_compress_sample(&c->status[st], *samples++);
702
            *dst++ = nibble;
703
        }
704
        break;
705
    case CODEC_ID_ADPCM_YAMAHA:
706
        n = avctx->frame_size / 2;
707
        if(avctx->trellis > 0) {
708
            FF_ALLOC_OR_GOTO(avctx, buf, 2*n*2, error);
709
            n *= 2;
710
            if(avctx->channels == 1) {
711
                adpcm_compress_trellis(avctx, samples, buf, &c->status[0], n);
712
                for(i=0; i<n; i+=2)
713
                    *dst++ = buf[i] | (buf[i+1] << 4);
714
            } else {
715
                adpcm_compress_trellis(avctx, samples, buf, &c->status[0], n);
716
                adpcm_compress_trellis(avctx, samples+1, buf+n, &c->status[1], n);
717
                for(i=0; i<n; i++)
718
                    *dst++ = buf[i] | (buf[n+i] << 4);
719
            }
720
            av_free(buf);
721
        } else
722
            for (n *= avctx->channels; n>0; n--) {
723
                int nibble;
724
                nibble  = adpcm_yamaha_compress_sample(&c->status[ 0], *samples++);
725
                nibble |= adpcm_yamaha_compress_sample(&c->status[st], *samples++) << 4;
726
                *dst++ = nibble;
727
            }
728
        break;
729
    default:
730
    error:
731
        return -1;
732
    }
733
    return dst - frame;
734
}
735
#endif //CONFIG_ENCODERS
736

    
737
static av_cold int adpcm_decode_init(AVCodecContext * avctx)
738
{
739
    ADPCMContext *c = avctx->priv_data;
740
    unsigned int max_channels = 2;
741

    
742
    switch(avctx->codec->id) {
743
    case CODEC_ID_ADPCM_EA_R1:
744
    case CODEC_ID_ADPCM_EA_R2:
745
    case CODEC_ID_ADPCM_EA_R3:
746
        max_channels = 6;
747
        break;
748
    }
749
    if(avctx->channels > max_channels){
750
        return -1;
751
    }
752

    
753
    switch(avctx->codec->id) {
754
    case CODEC_ID_ADPCM_CT:
755
        c->status[0].step = c->status[1].step = 511;
756
        break;
757
    case CODEC_ID_ADPCM_IMA_WAV:
758
        if (avctx->bits_per_coded_sample != 4) {
759
            av_log(avctx, AV_LOG_ERROR, "Only 4-bit ADPCM IMA WAV files are supported\n");
760
            return -1;
761
        }
762
        break;
763
    case CODEC_ID_ADPCM_IMA_WS:
764
        if (avctx->extradata && avctx->extradata_size == 2 * 4) {
765
            c->status[0].predictor = AV_RL32(avctx->extradata);
766
            c->status[1].predictor = AV_RL32(avctx->extradata + 4);
767
        }
768
        break;
769
    default:
770
        break;
771
    }
772
    avctx->sample_fmt = AV_SAMPLE_FMT_S16;
773
    return 0;
774
}
775

    
776
static inline short adpcm_ima_expand_nibble(ADPCMChannelStatus *c, char nibble, int shift)
777
{
778
    int step_index;
779
    int predictor;
780
    int sign, delta, diff, step;
781

    
782
    step = step_table[c->step_index];
783
    step_index = c->step_index + index_table[(unsigned)nibble];
784
    if (step_index < 0) step_index = 0;
785
    else if (step_index > 88) step_index = 88;
786

    
787
    sign = nibble & 8;
788
    delta = nibble & 7;
789
    /* perform direct multiplication instead of series of jumps proposed by
790
     * the reference ADPCM implementation since modern CPUs can do the mults
791
     * quickly enough */
792
    diff = ((2 * delta + 1) * step) >> shift;
793
    predictor = c->predictor;
794
    if (sign) predictor -= diff;
795
    else predictor += diff;
796

    
797
    c->predictor = av_clip_int16(predictor);
798
    c->step_index = step_index;
799

    
800
    return (short)c->predictor;
801
}
802

    
803
static inline short adpcm_ms_expand_nibble(ADPCMChannelStatus *c, char nibble)
804
{
805
    int predictor;
806

    
807
    predictor = (((c->sample1) * (c->coeff1)) + ((c->sample2) * (c->coeff2))) / 64;
808
    predictor += (signed)((nibble & 0x08)?(nibble - 0x10):(nibble)) * c->idelta;
809

    
810
    c->sample2 = c->sample1;
811
    c->sample1 = av_clip_int16(predictor);
812
    c->idelta = (AdaptationTable[(int)nibble] * c->idelta) >> 8;
813
    if (c->idelta < 16) c->idelta = 16;
814

    
815
    return c->sample1;
816
}
817

    
818
static inline short adpcm_ct_expand_nibble(ADPCMChannelStatus *c, char nibble)
819
{
820
    int sign, delta, diff;
821
    int new_step;
822

    
823
    sign = nibble & 8;
824
    delta = nibble & 7;
825
    /* perform direct multiplication instead of series of jumps proposed by
826
     * the reference ADPCM implementation since modern CPUs can do the mults
827
     * quickly enough */
828
    diff = ((2 * delta + 1) * c->step) >> 3;
829
    /* predictor update is not so trivial: predictor is multiplied on 254/256 before updating */
830
    c->predictor = ((c->predictor * 254) >> 8) + (sign ? -diff : diff);
831
    c->predictor = av_clip_int16(c->predictor);
832
    /* calculate new step and clamp it to range 511..32767 */
833
    new_step = (AdaptationTable[nibble & 7] * c->step) >> 8;
834
    c->step = av_clip(new_step, 511, 32767);
835

    
836
    return (short)c->predictor;
837
}
838

    
839
static inline short adpcm_sbpro_expand_nibble(ADPCMChannelStatus *c, char nibble, int size, int shift)
840
{
841
    int sign, delta, diff;
842

    
843
    sign = nibble & (1<<(size-1));
844
    delta = nibble & ((1<<(size-1))-1);
845
    diff = delta << (7 + c->step + shift);
846

    
847
    /* clamp result */
848
    c->predictor = av_clip(c->predictor + (sign ? -diff : diff), -16384,16256);
849

    
850
    /* calculate new step */
851
    if (delta >= (2*size - 3) && c->step < 3)
852
        c->step++;
853
    else if (delta == 0 && c->step > 0)
854
        c->step--;
855

    
856
    return (short) c->predictor;
857
}
858

    
859
static inline short adpcm_yamaha_expand_nibble(ADPCMChannelStatus *c, unsigned char nibble)
860
{
861
    if(!c->step) {
862
        c->predictor = 0;
863
        c->step = 127;
864
    }
865

    
866
    c->predictor += (c->step * yamaha_difflookup[nibble]) / 8;
867
    c->predictor = av_clip_int16(c->predictor);
868
    c->step = (c->step * yamaha_indexscale[nibble]) >> 8;
869
    c->step = av_clip(c->step, 127, 24567);
870
    return c->predictor;
871
}
872

    
873
static void xa_decode(short *out, const unsigned char *in,
874
    ADPCMChannelStatus *left, ADPCMChannelStatus *right, int inc)
875
{
876
    int i, j;
877
    int shift,filter,f0,f1;
878
    int s_1,s_2;
879
    int d,s,t;
880

    
881
    for(i=0;i<4;i++) {
882

    
883
        shift  = 12 - (in[4+i*2] & 15);
884
        filter = in[4+i*2] >> 4;
885
        f0 = xa_adpcm_table[filter][0];
886
        f1 = xa_adpcm_table[filter][1];
887

    
888
        s_1 = left->sample1;
889
        s_2 = left->sample2;
890

    
891
        for(j=0;j<28;j++) {
892
            d = in[16+i+j*4];
893

    
894
            t = (signed char)(d<<4)>>4;
895
            s = ( t<<shift ) + ((s_1*f0 + s_2*f1+32)>>6);
896
            s_2 = s_1;
897
            s_1 = av_clip_int16(s);
898
            *out = s_1;
899
            out += inc;
900
        }
901

    
902
        if (inc==2) { /* stereo */
903
            left->sample1 = s_1;
904
            left->sample2 = s_2;
905
            s_1 = right->sample1;
906
            s_2 = right->sample2;
907
            out = out + 1 - 28*2;
908
        }
909

    
910
        shift  = 12 - (in[5+i*2] & 15);
911
        filter = in[5+i*2] >> 4;
912

    
913
        f0 = xa_adpcm_table[filter][0];
914
        f1 = xa_adpcm_table[filter][1];
915

    
916
        for(j=0;j<28;j++) {
917
            d = in[16+i+j*4];
918

    
919
            t = (signed char)d >> 4;
920
            s = ( t<<shift ) + ((s_1*f0 + s_2*f1+32)>>6);
921
            s_2 = s_1;
922
            s_1 = av_clip_int16(s);
923
            *out = s_1;
924
            out += inc;
925
        }
926

    
927
        if (inc==2) { /* stereo */
928
            right->sample1 = s_1;
929
            right->sample2 = s_2;
930
            out -= 1;
931
        } else {
932
            left->sample1 = s_1;
933
            left->sample2 = s_2;
934
        }
935
    }
936
}
937

    
938

    
939
/* DK3 ADPCM support macro */
940
#define DK3_GET_NEXT_NIBBLE() \
941
    if (decode_top_nibble_next) \
942
    { \
943
        nibble = last_byte >> 4; \
944
        decode_top_nibble_next = 0; \
945
    } \
946
    else \
947
    { \
948
        last_byte = *src++; \
949
        if (src >= buf + buf_size) break; \
950
        nibble = last_byte & 0x0F; \
951
        decode_top_nibble_next = 1; \
952
    }
953

    
954
static int adpcm_decode_frame(AVCodecContext *avctx,
955
                            void *data, int *data_size,
956
                            AVPacket *avpkt)
957
{
958
    const uint8_t *buf = avpkt->data;
959
    int buf_size = avpkt->size;
960
    ADPCMContext *c = avctx->priv_data;
961
    ADPCMChannelStatus *cs;
962
    int n, m, channel, i;
963
    int block_predictor[2];
964
    short *samples;
965
    short *samples_end;
966
    const uint8_t *src;
967
    int st; /* stereo */
968

    
969
    /* DK3 ADPCM accounting variables */
970
    unsigned char last_byte = 0;
971
    unsigned char nibble;
972
    int decode_top_nibble_next = 0;
973
    int diff_channel;
974

    
975
    /* EA ADPCM state variables */
976
    uint32_t samples_in_chunk;
977
    int32_t previous_left_sample, previous_right_sample;
978
    int32_t current_left_sample, current_right_sample;
979
    int32_t next_left_sample, next_right_sample;
980
    int32_t coeff1l, coeff2l, coeff1r, coeff2r;
981
    uint8_t shift_left, shift_right;
982
    int count1, count2;
983
    int coeff[2][2], shift[2];//used in EA MAXIS ADPCM
984

    
985
    if (!buf_size)
986
        return 0;
987

    
988
    //should protect all 4bit ADPCM variants
989
    //8 is needed for CODEC_ID_ADPCM_IMA_WAV with 2 channels
990
    //
991
    if(*data_size/4 < buf_size + 8)
992
        return -1;
993

    
994
    samples = data;
995
    samples_end= samples + *data_size/2;
996
    *data_size= 0;
997
    src = buf;
998

    
999
    st = avctx->channels == 2 ? 1 : 0;
1000

    
1001
    switch(avctx->codec->id) {
1002
    case CODEC_ID_ADPCM_IMA_QT:
1003
        n = buf_size - 2*avctx->channels;
1004
        for (channel = 0; channel < avctx->channels; channel++) {
1005
            cs = &(c->status[channel]);
1006
            /* (pppppp) (piiiiiii) */
1007

    
1008
            /* Bits 15-7 are the _top_ 9 bits of the 16-bit initial predictor value */
1009
            cs->predictor = (*src++) << 8;
1010
            cs->predictor |= (*src & 0x80);
1011
            cs->predictor &= 0xFF80;
1012

    
1013
            /* sign extension */
1014
            if(cs->predictor & 0x8000)
1015
                cs->predictor -= 0x10000;
1016

    
1017
            cs->predictor = av_clip_int16(cs->predictor);
1018

    
1019
            cs->step_index = (*src++) & 0x7F;
1020

    
1021
            if (cs->step_index > 88){
1022
                av_log(avctx, AV_LOG_ERROR, "ERROR: step_index = %i\n", cs->step_index);
1023
                cs->step_index = 88;
1024
            }
1025

    
1026
            cs->step = step_table[cs->step_index];
1027

    
1028
            samples = (short*)data + channel;
1029

    
1030
            for(m=32; n>0 && m>0; n--, m--) { /* in QuickTime, IMA is encoded by chuncks of 34 bytes (=64 samples) */
1031
                *samples = adpcm_ima_expand_nibble(cs, src[0] & 0x0F, 3);
1032
                samples += avctx->channels;
1033
                *samples = adpcm_ima_expand_nibble(cs, src[0] >> 4  , 3);
1034
                samples += avctx->channels;
1035
                src ++;
1036
            }
1037
        }
1038
        if (st)
1039
            samples--;
1040
        break;
1041
    case CODEC_ID_ADPCM_IMA_WAV:
1042
        if (avctx->block_align != 0 && buf_size > avctx->block_align)
1043
            buf_size = avctx->block_align;
1044

    
1045
//        samples_per_block= (block_align-4*chanels)*8 / (bits_per_sample * chanels) + 1;
1046

    
1047
        for(i=0; i<avctx->channels; i++){
1048
            cs = &(c->status[i]);
1049
            cs->predictor = *samples++ = (int16_t)bytestream_get_le16(&src);
1050

    
1051
            cs->step_index = *src++;
1052
            if (cs->step_index > 88){
1053
                av_log(avctx, AV_LOG_ERROR, "ERROR: step_index = %i\n", cs->step_index);
1054
                cs->step_index = 88;
1055
            }
1056
            if (*src++) av_log(avctx, AV_LOG_ERROR, "unused byte should be null but is %d!!\n", src[-1]); /* unused */
1057
        }
1058

    
1059
        while(src < buf + buf_size){
1060
            for(m=0; m<4; m++){
1061
                for(i=0; i<=st; i++)
1062
                    *samples++ = adpcm_ima_expand_nibble(&c->status[i], src[4*i] & 0x0F, 3);
1063
                for(i=0; i<=st; i++)
1064
                    *samples++ = adpcm_ima_expand_nibble(&c->status[i], src[4*i] >> 4  , 3);
1065
                src++;
1066
            }
1067
            src += 4*st;
1068
        }
1069
        break;
1070
    case CODEC_ID_ADPCM_4XM:
1071
        cs = &(c->status[0]);
1072
        c->status[0].predictor= (int16_t)bytestream_get_le16(&src);
1073
        if(st){
1074
            c->status[1].predictor= (int16_t)bytestream_get_le16(&src);
1075
        }
1076
        c->status[0].step_index= (int16_t)bytestream_get_le16(&src);
1077
        if(st){
1078
            c->status[1].step_index= (int16_t)bytestream_get_le16(&src);
1079
        }
1080
        if (cs->step_index < 0) cs->step_index = 0;
1081
        if (cs->step_index > 88) cs->step_index = 88;
1082

    
1083
        m= (buf_size - (src - buf))>>st;
1084
        for(i=0; i<m; i++) {
1085
            *samples++ = adpcm_ima_expand_nibble(&c->status[0], src[i] & 0x0F, 4);
1086
            if (st)
1087
                *samples++ = adpcm_ima_expand_nibble(&c->status[1], src[i+m] & 0x0F, 4);
1088
            *samples++ = adpcm_ima_expand_nibble(&c->status[0], src[i] >> 4, 4);
1089
            if (st)
1090
                *samples++ = adpcm_ima_expand_nibble(&c->status[1], src[i+m] >> 4, 4);
1091
        }
1092

    
1093
        src += m<<st;
1094

    
1095
        break;
1096
    case CODEC_ID_ADPCM_MS:
1097
        if (avctx->block_align != 0 && buf_size > avctx->block_align)
1098
            buf_size = avctx->block_align;
1099
        n = buf_size - 7 * avctx->channels;
1100
        if (n < 0)
1101
            return -1;
1102
        block_predictor[0] = av_clip(*src++, 0, 6);
1103
        block_predictor[1] = 0;
1104
        if (st)
1105
            block_predictor[1] = av_clip(*src++, 0, 6);
1106
        c->status[0].idelta = (int16_t)bytestream_get_le16(&src);
1107
        if (st){
1108
            c->status[1].idelta = (int16_t)bytestream_get_le16(&src);
1109
        }
1110
        c->status[0].coeff1 = AdaptCoeff1[block_predictor[0]];
1111
        c->status[0].coeff2 = AdaptCoeff2[block_predictor[0]];
1112
        c->status[1].coeff1 = AdaptCoeff1[block_predictor[1]];
1113
        c->status[1].coeff2 = AdaptCoeff2[block_predictor[1]];
1114

    
1115
        c->status[0].sample1 = bytestream_get_le16(&src);
1116
        if (st) c->status[1].sample1 = bytestream_get_le16(&src);
1117
        c->status[0].sample2 = bytestream_get_le16(&src);
1118
        if (st) c->status[1].sample2 = bytestream_get_le16(&src);
1119

    
1120
        *samples++ = c->status[0].sample2;
1121
        if (st) *samples++ = c->status[1].sample2;
1122
        *samples++ = c->status[0].sample1;
1123
        if (st) *samples++ = c->status[1].sample1;
1124
        for(;n>0;n--) {
1125
            *samples++ = adpcm_ms_expand_nibble(&c->status[0 ], src[0] >> 4  );
1126
            *samples++ = adpcm_ms_expand_nibble(&c->status[st], src[0] & 0x0F);
1127
            src ++;
1128
        }
1129
        break;
1130
    case CODEC_ID_ADPCM_IMA_DK4:
1131
        if (avctx->block_align != 0 && buf_size > avctx->block_align)
1132
            buf_size = avctx->block_align;
1133

    
1134
        c->status[0].predictor  = (int16_t)bytestream_get_le16(&src);
1135
        c->status[0].step_index = *src++;
1136
        src++;
1137
        *samples++ = c->status[0].predictor;
1138
        if (st) {
1139
            c->status[1].predictor  = (int16_t)bytestream_get_le16(&src);
1140
            c->status[1].step_index = *src++;
1141
            src++;
1142
            *samples++ = c->status[1].predictor;
1143
        }
1144
        while (src < buf + buf_size) {
1145

    
1146
            /* take care of the top nibble (always left or mono channel) */
1147
            *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1148
                src[0] >> 4, 3);
1149

    
1150
            /* take care of the bottom nibble, which is right sample for
1151
             * stereo, or another mono sample */
1152
            if (st)
1153
                *samples++ = adpcm_ima_expand_nibble(&c->status[1],
1154
                    src[0] & 0x0F, 3);
1155
            else
1156
                *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1157
                    src[0] & 0x0F, 3);
1158

    
1159
            src++;
1160
        }
1161
        break;
1162
    case CODEC_ID_ADPCM_IMA_DK3:
1163
        if (avctx->block_align != 0 && buf_size > avctx->block_align)
1164
            buf_size = avctx->block_align;
1165

    
1166
        if(buf_size + 16 > (samples_end - samples)*3/8)
1167
            return -1;
1168

    
1169
        c->status[0].predictor  = (int16_t)AV_RL16(src + 10);
1170
        c->status[1].predictor  = (int16_t)AV_RL16(src + 12);
1171
        c->status[0].step_index = src[14];
1172
        c->status[1].step_index = src[15];
1173
        /* sign extend the predictors */
1174
        src += 16;
1175
        diff_channel = c->status[1].predictor;
1176

    
1177
        /* the DK3_GET_NEXT_NIBBLE macro issues the break statement when
1178
         * the buffer is consumed */
1179
        while (1) {
1180

    
1181
            /* for this algorithm, c->status[0] is the sum channel and
1182
             * c->status[1] is the diff channel */
1183

    
1184
            /* process the first predictor of the sum channel */
1185
            DK3_GET_NEXT_NIBBLE();
1186
            adpcm_ima_expand_nibble(&c->status[0], nibble, 3);
1187

    
1188
            /* process the diff channel predictor */
1189
            DK3_GET_NEXT_NIBBLE();
1190
            adpcm_ima_expand_nibble(&c->status[1], nibble, 3);
1191

    
1192
            /* process the first pair of stereo PCM samples */
1193
            diff_channel = (diff_channel + c->status[1].predictor) / 2;
1194
            *samples++ = c->status[0].predictor + c->status[1].predictor;
1195
            *samples++ = c->status[0].predictor - c->status[1].predictor;
1196

    
1197
            /* process the second predictor of the sum channel */
1198
            DK3_GET_NEXT_NIBBLE();
1199
            adpcm_ima_expand_nibble(&c->status[0], nibble, 3);
1200

    
1201
            /* process the second pair of stereo PCM samples */
1202
            diff_channel = (diff_channel + c->status[1].predictor) / 2;
1203
            *samples++ = c->status[0].predictor + c->status[1].predictor;
1204
            *samples++ = c->status[0].predictor - c->status[1].predictor;
1205
        }
1206
        break;
1207
    case CODEC_ID_ADPCM_IMA_ISS:
1208
        c->status[0].predictor  = (int16_t)AV_RL16(src + 0);
1209
        c->status[0].step_index = src[2];
1210
        src += 4;
1211
        if(st) {
1212
            c->status[1].predictor  = (int16_t)AV_RL16(src + 0);
1213
            c->status[1].step_index = src[2];
1214
            src += 4;
1215
        }
1216

    
1217
        while (src < buf + buf_size) {
1218

    
1219
            if (st) {
1220
                *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1221
                    src[0] >> 4  , 3);
1222
                *samples++ = adpcm_ima_expand_nibble(&c->status[1],
1223
                    src[0] & 0x0F, 3);
1224
            } else {
1225
                *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1226
                    src[0] & 0x0F, 3);
1227
                *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1228
                    src[0] >> 4  , 3);
1229
            }
1230

    
1231
            src++;
1232
        }
1233
        break;
1234
    case CODEC_ID_ADPCM_IMA_WS:
1235
        /* no per-block initialization; just start decoding the data */
1236
        while (src < buf + buf_size) {
1237

    
1238
            if (st) {
1239
                *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1240
                    src[0] >> 4  , 3);
1241
                *samples++ = adpcm_ima_expand_nibble(&c->status[1],
1242
                    src[0] & 0x0F, 3);
1243
            } else {
1244
                *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1245
                    src[0] >> 4  , 3);
1246
                *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1247
                    src[0] & 0x0F, 3);
1248
            }
1249

    
1250
            src++;
1251
        }
1252
        break;
1253
    case CODEC_ID_ADPCM_XA:
1254
        while (buf_size >= 128) {
1255
            xa_decode(samples, src, &c->status[0], &c->status[1],
1256
                avctx->channels);
1257
            src += 128;
1258
            samples += 28 * 8;
1259
            buf_size -= 128;
1260
        }
1261
        break;
1262
    case CODEC_ID_ADPCM_IMA_EA_EACS:
1263
        samples_in_chunk = bytestream_get_le32(&src) >> (1-st);
1264

    
1265
        if (samples_in_chunk > buf_size-4-(8<<st)) {
1266
            src += buf_size - 4;
1267
            break;
1268
        }
1269

    
1270
        for (i=0; i<=st; i++)
1271
            c->status[i].step_index = bytestream_get_le32(&src);
1272
        for (i=0; i<=st; i++)
1273
            c->status[i].predictor  = bytestream_get_le32(&src);
1274

    
1275
        for (; samples_in_chunk; samples_in_chunk--, src++) {
1276
            *samples++ = adpcm_ima_expand_nibble(&c->status[0],  *src>>4,   3);
1277
            *samples++ = adpcm_ima_expand_nibble(&c->status[st], *src&0x0F, 3);
1278
        }
1279
        break;
1280
    case CODEC_ID_ADPCM_IMA_EA_SEAD:
1281
        for (; src < buf+buf_size; src++) {
1282
            *samples++ = adpcm_ima_expand_nibble(&c->status[0], src[0] >> 4, 6);
1283
            *samples++ = adpcm_ima_expand_nibble(&c->status[st],src[0]&0x0F, 6);
1284
        }
1285
        break;
1286
    case CODEC_ID_ADPCM_EA:
1287
        if (buf_size < 4 || AV_RL32(src) >= ((buf_size - 12) * 2)) {
1288
            src += buf_size;
1289
            break;
1290
        }
1291
        samples_in_chunk = AV_RL32(src);
1292
        src += 4;
1293
        current_left_sample   = (int16_t)bytestream_get_le16(&src);
1294
        previous_left_sample  = (int16_t)bytestream_get_le16(&src);
1295
        current_right_sample  = (int16_t)bytestream_get_le16(&src);
1296
        previous_right_sample = (int16_t)bytestream_get_le16(&src);
1297

    
1298
        for (count1 = 0; count1 < samples_in_chunk/28;count1++) {
1299
            coeff1l = ea_adpcm_table[ *src >> 4       ];
1300
            coeff2l = ea_adpcm_table[(*src >> 4  ) + 4];
1301
            coeff1r = ea_adpcm_table[*src & 0x0F];
1302
            coeff2r = ea_adpcm_table[(*src & 0x0F) + 4];
1303
            src++;
1304

    
1305
            shift_left  = (*src >> 4  ) + 8;
1306
            shift_right = (*src & 0x0F) + 8;
1307
            src++;
1308

    
1309
            for (count2 = 0; count2 < 28; count2++) {
1310
                next_left_sample  = (int32_t)((*src & 0xF0) << 24) >> shift_left;
1311
                next_right_sample = (int32_t)((*src & 0x0F) << 28) >> shift_right;
1312
                src++;
1313

    
1314
                next_left_sample = (next_left_sample +
1315
                    (current_left_sample * coeff1l) +
1316
                    (previous_left_sample * coeff2l) + 0x80) >> 8;
1317
                next_right_sample = (next_right_sample +
1318
                    (current_right_sample * coeff1r) +
1319
                    (previous_right_sample * coeff2r) + 0x80) >> 8;
1320

    
1321
                previous_left_sample = current_left_sample;
1322
                current_left_sample = av_clip_int16(next_left_sample);
1323
                previous_right_sample = current_right_sample;
1324
                current_right_sample = av_clip_int16(next_right_sample);
1325
                *samples++ = (unsigned short)current_left_sample;
1326
                *samples++ = (unsigned short)current_right_sample;
1327
            }
1328
        }
1329

    
1330
        if (src - buf == buf_size - 2)
1331
            src += 2; // Skip terminating 0x0000
1332

    
1333
        break;
1334
    case CODEC_ID_ADPCM_EA_MAXIS_XA:
1335
        for(channel = 0; channel < avctx->channels; channel++) {
1336
            for (i=0; i<2; i++)
1337
                coeff[channel][i] = ea_adpcm_table[(*src >> 4) + 4*i];
1338
            shift[channel] = (*src & 0x0F) + 8;
1339
            src++;
1340
        }
1341
        for (count1 = 0; count1 < (buf_size - avctx->channels) / avctx->channels; count1++) {
1342
            for(i = 4; i >= 0; i-=4) { /* Pairwise samples LL RR (st) or LL LL (mono) */
1343
                for(channel = 0; channel < avctx->channels; channel++) {
1344
                    int32_t sample = (int32_t)(((*(src+channel) >> i) & 0x0F) << 0x1C) >> shift[channel];
1345
                    sample = (sample +
1346
                             c->status[channel].sample1 * coeff[channel][0] +
1347
                             c->status[channel].sample2 * coeff[channel][1] + 0x80) >> 8;
1348
                    c->status[channel].sample2 = c->status[channel].sample1;
1349
                    c->status[channel].sample1 = av_clip_int16(sample);
1350
                    *samples++ = c->status[channel].sample1;
1351
                }
1352
            }
1353
            src+=avctx->channels;
1354
        }
1355
        break;
1356
    case CODEC_ID_ADPCM_EA_R1:
1357
    case CODEC_ID_ADPCM_EA_R2:
1358
    case CODEC_ID_ADPCM_EA_R3: {
1359
        /* channel numbering
1360
           2chan: 0=fl, 1=fr
1361
           4chan: 0=fl, 1=rl, 2=fr, 3=rr
1362
           6chan: 0=fl, 1=c,  2=fr, 3=rl,  4=rr, 5=sub */
1363
        const int big_endian = avctx->codec->id == CODEC_ID_ADPCM_EA_R3;
1364
        int32_t previous_sample, current_sample, next_sample;
1365
        int32_t coeff1, coeff2;
1366
        uint8_t shift;
1367
        unsigned int channel;
1368
        uint16_t *samplesC;
1369
        const uint8_t *srcC;
1370
        const uint8_t *src_end = buf + buf_size;
1371

    
1372
        samples_in_chunk = (big_endian ? bytestream_get_be32(&src)
1373
                                       : bytestream_get_le32(&src)) / 28;
1374
        if (samples_in_chunk > UINT32_MAX/(28*avctx->channels) ||
1375
            28*samples_in_chunk*avctx->channels > samples_end-samples) {
1376
            src += buf_size - 4;
1377
            break;
1378
        }
1379

    
1380
        for (channel=0; channel<avctx->channels; channel++) {
1381
            int32_t offset = (big_endian ? bytestream_get_be32(&src)
1382
                                         : bytestream_get_le32(&src))
1383
                           + (avctx->channels-channel-1) * 4;
1384

    
1385
            if ((offset < 0) || (offset >= src_end - src - 4)) break;
1386
            srcC  = src + offset;
1387
            samplesC = samples + channel;
1388

    
1389
            if (avctx->codec->id == CODEC_ID_ADPCM_EA_R1) {
1390
                current_sample  = (int16_t)bytestream_get_le16(&srcC);
1391
                previous_sample = (int16_t)bytestream_get_le16(&srcC);
1392
            } else {
1393
                current_sample  = c->status[channel].predictor;
1394
                previous_sample = c->status[channel].prev_sample;
1395
            }
1396

    
1397
            for (count1=0; count1<samples_in_chunk; count1++) {
1398
                if (*srcC == 0xEE) {  /* only seen in R2 and R3 */
1399
                    srcC++;
1400
                    if (srcC > src_end - 30*2) break;
1401
                    current_sample  = (int16_t)bytestream_get_be16(&srcC);
1402
                    previous_sample = (int16_t)bytestream_get_be16(&srcC);
1403

    
1404
                    for (count2=0; count2<28; count2++) {
1405
                        *samplesC = (int16_t)bytestream_get_be16(&srcC);
1406
                        samplesC += avctx->channels;
1407
                    }
1408
                } else {
1409
                    coeff1 = ea_adpcm_table[ *srcC>>4     ];
1410
                    coeff2 = ea_adpcm_table[(*srcC>>4) + 4];
1411
                    shift = (*srcC++ & 0x0F) + 8;
1412

    
1413
                    if (srcC > src_end - 14) break;
1414
                    for (count2=0; count2<28; count2++) {
1415
                        if (count2 & 1)
1416
                            next_sample = (int32_t)((*srcC++ & 0x0F) << 28) >> shift;
1417
                        else
1418
                            next_sample = (int32_t)((*srcC   & 0xF0) << 24) >> shift;
1419

    
1420
                        next_sample += (current_sample  * coeff1) +
1421
                                       (previous_sample * coeff2);
1422
                        next_sample = av_clip_int16(next_sample >> 8);
1423

    
1424
                        previous_sample = current_sample;
1425
                        current_sample  = next_sample;
1426
                        *samplesC = current_sample;
1427
                        samplesC += avctx->channels;
1428
                    }
1429
                }
1430
            }
1431

    
1432
            if (avctx->codec->id != CODEC_ID_ADPCM_EA_R1) {
1433
                c->status[channel].predictor   = current_sample;
1434
                c->status[channel].prev_sample = previous_sample;
1435
            }
1436
        }
1437

    
1438
        src = src + buf_size - (4 + 4*avctx->channels);
1439
        samples += 28 * samples_in_chunk * avctx->channels;
1440
        break;
1441
    }
1442
    case CODEC_ID_ADPCM_EA_XAS:
1443
        if (samples_end-samples < 32*4*avctx->channels
1444
            || buf_size < (4+15)*4*avctx->channels) {
1445
            src += buf_size;
1446
            break;
1447
        }
1448
        for (channel=0; channel<avctx->channels; channel++) {
1449
            int coeff[2][4], shift[4];
1450
            short *s2, *s = &samples[channel];
1451
            for (n=0; n<4; n++, s+=32*avctx->channels) {
1452
                for (i=0; i<2; i++)
1453
                    coeff[i][n] = ea_adpcm_table[(src[0]&0x0F)+4*i];
1454
                shift[n] = (src[2]&0x0F) + 8;
1455
                for (s2=s, i=0; i<2; i++, src+=2, s2+=avctx->channels)
1456
                    s2[0] = (src[0]&0xF0) + (src[1]<<8);
1457
            }
1458

    
1459
            for (m=2; m<32; m+=2) {
1460
                s = &samples[m*avctx->channels + channel];
1461
                for (n=0; n<4; n++, src++, s+=32*avctx->channels) {
1462
                    for (s2=s, i=0; i<8; i+=4, s2+=avctx->channels) {
1463
                        int level = (int32_t)((*src & (0xF0>>i)) << (24+i)) >> shift[n];
1464
                        int pred  = s2[-1*avctx->channels] * coeff[0][n]
1465
                                  + s2[-2*avctx->channels] * coeff[1][n];
1466
                        s2[0] = av_clip_int16((level + pred + 0x80) >> 8);
1467
                    }
1468
                }
1469
            }
1470
        }
1471
        samples += 32*4*avctx->channels;
1472
        break;
1473
    case CODEC_ID_ADPCM_IMA_AMV:
1474
    case CODEC_ID_ADPCM_IMA_SMJPEG:
1475
        c->status[0].predictor = (int16_t)bytestream_get_le16(&src);
1476
        c->status[0].step_index = bytestream_get_le16(&src);
1477

    
1478
        if (avctx->codec->id == CODEC_ID_ADPCM_IMA_AMV)
1479
            src+=4;
1480

    
1481
        while (src < buf + buf_size) {
1482
            char hi, lo;
1483
            lo = *src & 0x0F;
1484
            hi = *src >> 4;
1485

    
1486
            if (avctx->codec->id == CODEC_ID_ADPCM_IMA_AMV)
1487
                FFSWAP(char, hi, lo);
1488

    
1489
            *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1490
                lo, 3);
1491
            *samples++ = adpcm_ima_expand_nibble(&c->status[0],
1492
                hi, 3);
1493
            src++;
1494
        }
1495
        break;
1496
    case CODEC_ID_ADPCM_CT:
1497
        while (src < buf + buf_size) {
1498
            if (st) {
1499
                *samples++ = adpcm_ct_expand_nibble(&c->status[0],
1500
                    src[0] >> 4);
1501
                *samples++ = adpcm_ct_expand_nibble(&c->status[1],
1502
                    src[0] & 0x0F);
1503
            } else {
1504
                *samples++ = adpcm_ct_expand_nibble(&c->status[0],
1505
                    src[0] >> 4);
1506
                *samples++ = adpcm_ct_expand_nibble(&c->status[0],
1507
                    src[0] & 0x0F);
1508
            }
1509
            src++;
1510
        }
1511
        break;
1512
    case CODEC_ID_ADPCM_SBPRO_4:
1513
    case CODEC_ID_ADPCM_SBPRO_3:
1514
    case CODEC_ID_ADPCM_SBPRO_2:
1515
        if (!c->status[0].step_index) {
1516
            /* the first byte is a raw sample */
1517
            *samples++ = 128 * (*src++ - 0x80);
1518
            if (st)
1519
              *samples++ = 128 * (*src++ - 0x80);
1520
            c->status[0].step_index = 1;
1521
        }
1522
        if (avctx->codec->id == CODEC_ID_ADPCM_SBPRO_4) {
1523
            while (src < buf + buf_size) {
1524
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1525
                    src[0] >> 4, 4, 0);
1526
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[st],
1527
                    src[0] & 0x0F, 4, 0);
1528
                src++;
1529
            }
1530
        } else if (avctx->codec->id == CODEC_ID_ADPCM_SBPRO_3) {
1531
            while (src < buf + buf_size && samples + 2 < samples_end) {
1532
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1533
                     src[0] >> 5        , 3, 0);
1534
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1535
                    (src[0] >> 2) & 0x07, 3, 0);
1536
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1537
                    src[0] & 0x03, 2, 0);
1538
                src++;
1539
            }
1540
        } else {
1541
            while (src < buf + buf_size && samples + 3 < samples_end) {
1542
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1543
                     src[0] >> 6        , 2, 2);
1544
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[st],
1545
                    (src[0] >> 4) & 0x03, 2, 2);
1546
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[0],
1547
                    (src[0] >> 2) & 0x03, 2, 2);
1548
                *samples++ = adpcm_sbpro_expand_nibble(&c->status[st],
1549
                    src[0] & 0x03, 2, 2);
1550
                src++;
1551
            }
1552
        }
1553
        break;
1554
    case CODEC_ID_ADPCM_SWF:
1555
    {
1556
        GetBitContext gb;
1557
        const int *table;
1558
        int k0, signmask, nb_bits, count;
1559
        int size = buf_size*8;
1560

    
1561
        init_get_bits(&gb, buf, size);
1562

    
1563
        //read bits & initial values
1564
        nb_bits = get_bits(&gb, 2)+2;
1565
        //av_log(NULL,AV_LOG_INFO,"nb_bits: %d\n", nb_bits);
1566
        table = swf_index_tables[nb_bits-2];
1567
        k0 = 1 << (nb_bits-2);
1568
        signmask = 1 << (nb_bits-1);
1569

    
1570
        while (get_bits_count(&gb) <= size - 22*avctx->channels) {
1571
            for (i = 0; i < avctx->channels; i++) {
1572
                *samples++ = c->status[i].predictor = get_sbits(&gb, 16);
1573
                c->status[i].step_index = get_bits(&gb, 6);
1574
            }
1575

    
1576
            for (count = 0; get_bits_count(&gb) <= size - nb_bits*avctx->channels && count < 4095; count++) {
1577
                int i;
1578

    
1579
                for (i = 0; i < avctx->channels; i++) {
1580
                    // similar to IMA adpcm
1581
                    int delta = get_bits(&gb, nb_bits);
1582
                    int step = step_table[c->status[i].step_index];
1583
                    long vpdiff = 0; // vpdiff = (delta+0.5)*step/4
1584
                    int k = k0;
1585

    
1586
                    do {
1587
                        if (delta & k)
1588
                            vpdiff += step;
1589
                        step >>= 1;
1590
                        k >>= 1;
1591
                    } while(k);
1592
                    vpdiff += step;
1593

    
1594
                    if (delta & signmask)
1595
                        c->status[i].predictor -= vpdiff;
1596
                    else
1597
                        c->status[i].predictor += vpdiff;
1598

    
1599
                    c->status[i].step_index += table[delta & (~signmask)];
1600

    
1601
                    c->status[i].step_index = av_clip(c->status[i].step_index, 0, 88);
1602
                    c->status[i].predictor = av_clip_int16(c->status[i].predictor);
1603

    
1604
                    *samples++ = c->status[i].predictor;
1605
                    if (samples >= samples_end) {
1606
                        av_log(avctx, AV_LOG_ERROR, "allocated output buffer is too small\n");
1607
                        return -1;
1608
                    }
1609
                }
1610
            }
1611
        }
1612
        src += buf_size;
1613
        break;
1614
    }
1615
    case CODEC_ID_ADPCM_YAMAHA:
1616
        while (src < buf + buf_size) {
1617
            if (st) {
1618
                *samples++ = adpcm_yamaha_expand_nibble(&c->status[0],
1619
                        src[0] & 0x0F);
1620
                *samples++ = adpcm_yamaha_expand_nibble(&c->status[1],
1621
                        src[0] >> 4  );
1622
            } else {
1623
                *samples++ = adpcm_yamaha_expand_nibble(&c->status[0],
1624
                        src[0] & 0x0F);
1625
                *samples++ = adpcm_yamaha_expand_nibble(&c->status[0],
1626
                        src[0] >> 4  );
1627
            }
1628
            src++;
1629
        }
1630
        break;
1631
    case CODEC_ID_ADPCM_THP:
1632
    {
1633
        int table[2][16];
1634
        unsigned int samplecnt;
1635
        int prev[2][2];
1636
        int ch;
1637

    
1638
        if (buf_size < 80) {
1639
            av_log(avctx, AV_LOG_ERROR, "frame too small\n");
1640
            return -1;
1641
        }
1642

    
1643
        src+=4;
1644
        samplecnt = bytestream_get_be32(&src);
1645

    
1646
        for (i = 0; i < 32; i++)
1647
            table[0][i] = (int16_t)bytestream_get_be16(&src);
1648

    
1649
        /* Initialize the previous sample.  */
1650
        for (i = 0; i < 4; i++)
1651
            prev[0][i] = (int16_t)bytestream_get_be16(&src);
1652

    
1653
        if (samplecnt >= (samples_end - samples) /  (st + 1)) {
1654
            av_log(avctx, AV_LOG_ERROR, "allocated output buffer is too small\n");
1655
            return -1;
1656
        }
1657

    
1658
        for (ch = 0; ch <= st; ch++) {
1659
            samples = (unsigned short *) data + ch;
1660

    
1661
            /* Read in every sample for this channel.  */
1662
            for (i = 0; i < samplecnt / 14; i++) {
1663
                int index = (*src >> 4) & 7;
1664
                unsigned int exp = 28 - (*src++ & 15);
1665
                int factor1 = table[ch][index * 2];
1666
                int factor2 = table[ch][index * 2 + 1];
1667

    
1668
                /* Decode 14 samples.  */
1669
                for (n = 0; n < 14; n++) {
1670
                    int32_t sampledat;
1671
                    if(n&1) sampledat=  *src++    <<28;
1672
                    else    sampledat= (*src&0xF0)<<24;
1673

    
1674
                    sampledat = ((prev[ch][0]*factor1
1675
                                + prev[ch][1]*factor2) >> 11) + (sampledat>>exp);
1676
                    *samples = av_clip_int16(sampledat);
1677
                    prev[ch][1] = prev[ch][0];
1678
                    prev[ch][0] = *samples++;
1679

    
1680
                    /* In case of stereo, skip one sample, this sample
1681
                       is for the other channel.  */
1682
                    samples += st;
1683
                }
1684
            }
1685
        }
1686

    
1687
        /* In the previous loop, in case stereo is used, samples is
1688
           increased exactly one time too often.  */
1689
        samples -= st;
1690
        break;
1691
    }
1692

    
1693
    default:
1694
        return -1;
1695
    }
1696
    *data_size = (uint8_t *)samples - (uint8_t *)data;
1697
    return src - buf;
1698
}
1699

    
1700

    
1701

    
1702
#if CONFIG_ENCODERS
1703
#define ADPCM_ENCODER(id,name,long_name_)       \
1704
AVCodec name ## _encoder = {                    \
1705
    #name,                                      \
1706
    AVMEDIA_TYPE_AUDIO,                         \
1707
    id,                                         \
1708
    sizeof(ADPCMContext),                       \
1709
    adpcm_encode_init,                          \
1710
    adpcm_encode_frame,                         \
1711
    adpcm_encode_close,                         \
1712
    NULL,                                       \
1713
    .sample_fmts = (const enum AVSampleFormat[]){AV_SAMPLE_FMT_S16,AV_SAMPLE_FMT_NONE}, \
1714
    .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
1715
};
1716
#else
1717
#define ADPCM_ENCODER(id,name,long_name_)
1718
#endif
1719

    
1720
#if CONFIG_DECODERS
1721
#define ADPCM_DECODER(id,name,long_name_)       \
1722
AVCodec name ## _decoder = {                    \
1723
    #name,                                      \
1724
    AVMEDIA_TYPE_AUDIO,                         \
1725
    id,                                         \
1726
    sizeof(ADPCMContext),                       \
1727
    adpcm_decode_init,                          \
1728
    NULL,                                       \
1729
    NULL,                                       \
1730
    adpcm_decode_frame,                         \
1731
    .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
1732
};
1733
#else
1734
#define ADPCM_DECODER(id,name,long_name_)
1735
#endif
1736

    
1737
#define ADPCM_CODEC(id,name,long_name_)         \
1738
    ADPCM_ENCODER(id,name,long_name_) ADPCM_DECODER(id,name,long_name_)
1739

    
1740
/* Note: Do not forget to add new entries to the Makefile as well. */
1741
ADPCM_DECODER(CODEC_ID_ADPCM_4XM, adpcm_4xm, "ADPCM 4X Movie");
1742
ADPCM_DECODER(CODEC_ID_ADPCM_CT, adpcm_ct, "ADPCM Creative Technology");
1743
ADPCM_DECODER(CODEC_ID_ADPCM_EA, adpcm_ea, "ADPCM Electronic Arts");
1744
ADPCM_DECODER(CODEC_ID_ADPCM_EA_MAXIS_XA, adpcm_ea_maxis_xa, "ADPCM Electronic Arts Maxis CDROM XA");
1745
ADPCM_DECODER(CODEC_ID_ADPCM_EA_R1, adpcm_ea_r1, "ADPCM Electronic Arts R1");
1746
ADPCM_DECODER(CODEC_ID_ADPCM_EA_R2, adpcm_ea_r2, "ADPCM Electronic Arts R2");
1747
ADPCM_DECODER(CODEC_ID_ADPCM_EA_R3, adpcm_ea_r3, "ADPCM Electronic Arts R3");
1748
ADPCM_DECODER(CODEC_ID_ADPCM_EA_XAS, adpcm_ea_xas, "ADPCM Electronic Arts XAS");
1749
ADPCM_DECODER(CODEC_ID_ADPCM_IMA_AMV, adpcm_ima_amv, "ADPCM IMA AMV");
1750
ADPCM_DECODER(CODEC_ID_ADPCM_IMA_DK3, adpcm_ima_dk3, "ADPCM IMA Duck DK3");
1751
ADPCM_DECODER(CODEC_ID_ADPCM_IMA_DK4, adpcm_ima_dk4, "ADPCM IMA Duck DK4");
1752
ADPCM_DECODER(CODEC_ID_ADPCM_IMA_EA_EACS, adpcm_ima_ea_eacs, "ADPCM IMA Electronic Arts EACS");
1753
ADPCM_DECODER(CODEC_ID_ADPCM_IMA_EA_SEAD, adpcm_ima_ea_sead, "ADPCM IMA Electronic Arts SEAD");
1754
ADPCM_DECODER(CODEC_ID_ADPCM_IMA_ISS, adpcm_ima_iss, "ADPCM IMA Funcom ISS");
1755
ADPCM_CODEC  (CODEC_ID_ADPCM_IMA_QT, adpcm_ima_qt, "ADPCM IMA QuickTime");
1756
ADPCM_DECODER(CODEC_ID_ADPCM_IMA_SMJPEG, adpcm_ima_smjpeg, "ADPCM IMA Loki SDL MJPEG");
1757
ADPCM_CODEC  (CODEC_ID_ADPCM_IMA_WAV, adpcm_ima_wav, "ADPCM IMA WAV");
1758
ADPCM_DECODER(CODEC_ID_ADPCM_IMA_WS, adpcm_ima_ws, "ADPCM IMA Westwood");
1759
ADPCM_CODEC  (CODEC_ID_ADPCM_MS, adpcm_ms, "ADPCM Microsoft");
1760
ADPCM_DECODER(CODEC_ID_ADPCM_SBPRO_2, adpcm_sbpro_2, "ADPCM Sound Blaster Pro 2-bit");
1761
ADPCM_DECODER(CODEC_ID_ADPCM_SBPRO_3, adpcm_sbpro_3, "ADPCM Sound Blaster Pro 2.6-bit");
1762
ADPCM_DECODER(CODEC_ID_ADPCM_SBPRO_4, adpcm_sbpro_4, "ADPCM Sound Blaster Pro 4-bit");
1763
ADPCM_CODEC  (CODEC_ID_ADPCM_SWF, adpcm_swf, "ADPCM Shockwave Flash");
1764
ADPCM_DECODER(CODEC_ID_ADPCM_THP, adpcm_thp, "ADPCM Nintendo Gamecube THP");
1765
ADPCM_DECODER(CODEC_ID_ADPCM_XA, adpcm_xa, "ADPCM CDROM XA");
1766
ADPCM_CODEC  (CODEC_ID_ADPCM_YAMAHA, adpcm_yamaha, "ADPCM Yamaha");