libamr.c 12.1 KB
Newer Older
1 2 3 4
/*
 * AMR Audio decoder stub
 * Copyright (c) 2003 the ffmpeg project
 *
5 6 7
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
8 9
 * modify it under the terms of the GNU Lesser General Public
 * 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 14 15 16 17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * 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

22 23 24 25 26 27 28
 /** @file
 * Adaptive Multi-Rate (AMR) Audio decoder stub.
 *
 * This code implements both an AMR-NarrowBand (AMR-NB) and an AMR-WideBand
 * (AMR-WB) audio encoder/decoder through external reference code from
 * http://www.3gpp.org/. The license of the code from 3gpp is unclear so you
 * have to download the code separately. Two versions exists: One fixed-point
29 30 31 32
 * and one floating-point. For some reason the float encoder is significantly
 * faster at least on a P4 1.5GHz (0.9s instead of 9.9s on a 30s audio clip
 * at MR102). Both float and fixed point are supported for AMR-NB, but only
 * float for AMR-WB.
33 34 35 36 37
 *
 * \section AMR-NB
 *
 * \subsection Float
 * The float version (default) can be downloaded from:
38
 * http://www.3gpp.org/ftp/Specs/archive/26_series/26.104/26104-610.zip
39 40 41 42 43 44 45
 *
 * \subsection Specification
 * The specification for AMR-NB can be found in TS 26.071
 * (http://www.3gpp.org/ftp/Specs/html-info/26071.htm) and some other
 * info at http://www.3gpp.org/ftp/Specs/html-info/26-series.htm.
 *
 * \section AMR-WB
46
 *
47 48
 * \subsection Float
 * The reference code can be downloaded from:
49
 * http://www.3gpp.org/ftp/Specs/archive/26_series/26.204/26204-600.zip
50 51
 *
 * \subsection Specification
52 53 54
 * The specification for AMR-WB can be found in TS 26.171
 * (http://www.3gpp.org/ftp/Specs/html-info/26171.htm) and some other
 * info at http://www.3gpp.org/ftp/Specs/html-info/26-series.htm.
55
 *
Zdenek Kabelac's avatar
Zdenek Kabelac committed
56
 */
57

58 59
#include "avcodec.h"

60 61
#include <amrnb/interf_dec.h>
#include <amrnb/interf_enc.h>
Zdenek Kabelac's avatar
Zdenek Kabelac committed
62

63
static const char nb_bitrate_unsupported[] =
64
    "bitrate not supported: use one of 4.75k, 5.15k, 5.9k, 6.7k, 7.4k, 7.95k, 10.2k or 12.2k\n";
65
static const char wb_bitrate_unsupported[] =
66 67
    "bitrate not supported: use one of 6.6k, 8.85k, 12.65k, 14.25k, 15.85k, 18.25k, 19.85k, 23.05k, or 23.85k\n";

Zdenek Kabelac's avatar
Zdenek Kabelac committed
68 69 70
/* Common code for fixed and float version*/
typedef struct AMR_bitrates
{
Benoit Fouet's avatar
Benoit Fouet committed
71
    int rate;
Zdenek Kabelac's avatar
Zdenek Kabelac committed
72 73 74
    enum Mode mode;
} AMR_bitrates;

Benoit Fouet's avatar
Benoit Fouet committed
75 76
/* Match desired bitrate */
static int getBitrateMode(int bitrate)
Zdenek Kabelac's avatar
Zdenek Kabelac committed
77
{
Benoit Fouet's avatar
Benoit Fouet committed
78 79 80 81 82 83 84 85 86
    /* make the correspondance between bitrate and mode */
    AMR_bitrates rates[]={ {4750,MR475},
                           {5150,MR515},
                           {5900,MR59},
                           {6700,MR67},
                           {7400,MR74},
                           {7950,MR795},
                           {10200,MR102},
                           {12200,MR122},
Zdenek Kabelac's avatar
Zdenek Kabelac committed
87 88
                         };
    int i;
Benoit Fouet's avatar
Benoit Fouet committed
89

90
    for(i=0;i<8;i++)
Benoit Fouet's avatar
Benoit Fouet committed
91
        if(rates[i].rate==bitrate)
92
            return rates[i].mode;
Benoit Fouet's avatar
Benoit Fouet committed
93 94
    /* no bitrate matching, return an error */
    return -1;
Zdenek Kabelac's avatar
Zdenek Kabelac committed
95 96
}

Andrew Voznytsa's avatar
Andrew Voznytsa committed
97 98 99 100 101 102 103 104 105 106 107
static void amr_decode_fix_avctx(AVCodecContext * avctx)
{
    const int is_amr_wb = 1 + (avctx->codec_id == CODEC_ID_AMR_WB);

    if(avctx->sample_rate == 0)
        avctx->sample_rate = 8000 * is_amr_wb;

    if(avctx->channels == 0)
        avctx->channels = 1;

    avctx->frame_size = 160 * is_amr_wb;
108
    avctx->sample_fmt = SAMPLE_FMT_S16;
Andrew Voznytsa's avatar
Andrew Voznytsa committed
109 110
}

111
#if CONFIG_LIBAMR_NB
Zdenek Kabelac's avatar
Zdenek Kabelac committed
112 113 114 115 116

typedef struct AMRContext {
    int frameCount;
    void * decState;
    int *enstate;
Benoit Fouet's avatar
Benoit Fouet committed
117
    int enc_bitrate;
Zdenek Kabelac's avatar
Zdenek Kabelac committed
118 119
} AMRContext;

120
static av_cold int amr_nb_decode_init(AVCodecContext * avctx)
Zdenek Kabelac's avatar
Zdenek Kabelac committed
121 122
{
    AMRContext *s = avctx->priv_data;
Benoit Fouet's avatar
Benoit Fouet committed
123

Zdenek Kabelac's avatar
Zdenek Kabelac committed
124 125 126 127
    s->frameCount=0;
    s->decState=Decoder_Interface_init();
    if(!s->decState)
    {
Michael Niedermayer's avatar
Michael Niedermayer committed
128
        av_log(avctx, AV_LOG_ERROR, "Decoder_Interface_init error\r\n");
Zdenek Kabelac's avatar
Zdenek Kabelac committed
129 130
        return -1;
    }
Andrew Voznytsa's avatar
Andrew Voznytsa committed
131 132 133 134 135 136 137 138 139

    amr_decode_fix_avctx(avctx);

    if(avctx->channels > 1)
    {
        av_log(avctx, AV_LOG_ERROR, "amr_nb: multichannel decoding not supported\n");
        return -1;
    }

Zdenek Kabelac's avatar
Zdenek Kabelac committed
140 141 142
    return 0;
}

143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194
static av_cold int amr_nb_decode_close(AVCodecContext * avctx)
{
    AMRContext *s = avctx->priv_data;

    Decoder_Interface_exit(s->decState);
    return 0;
}

static int amr_nb_decode_frame(AVCodecContext * avctx,
            void *data, int *data_size,
            AVPacket *avpkt)
{
    const uint8_t *buf = avpkt->data;
    int buf_size = avpkt->size;
    AMRContext *s = avctx->priv_data;
    const uint8_t*amrData=buf;
    static const uint8_t block_size[16]={ 12, 13, 15, 17, 19, 20, 26, 31, 5, 0, 0, 0, 0, 0, 0, 0 };
    enum Mode dec_mode;
    int packet_size;

    /* av_log(NULL,AV_LOG_DEBUG,"amr_decode_frame buf=%p buf_size=%d frameCount=%d!!\n",buf,buf_size,s->frameCount); */

    dec_mode = (buf[0] >> 3) & 0x000F;
    packet_size = block_size[dec_mode]+1;

    if(packet_size > buf_size) {
        av_log(avctx, AV_LOG_ERROR, "amr frame too short (%u, should be %u)\n", buf_size, packet_size);
        return -1;
    }

    s->frameCount++;
    /* av_log(NULL,AV_LOG_DEBUG,"packet_size=%d amrData= 0x%X %X %X %X\n",packet_size,amrData[0],amrData[1],amrData[2],amrData[3]); */
    /* call decoder */
    Decoder_Interface_Decode(s->decState, amrData, data, 0);
    *data_size=160*2;

    return packet_size;
}

AVCodec libamr_nb_decoder =
{
    "libamr_nb",
    CODEC_TYPE_AUDIO,
    CODEC_ID_AMR_NB,
    sizeof(AMRContext),
    amr_nb_decode_init,
    NULL,
    amr_nb_decode_close,
    amr_nb_decode_frame,
    .long_name = NULL_IF_CONFIG_SMALL("libamr-nb Adaptive Multi-Rate (AMR) Narrow-Band"),
};

195
static av_cold int amr_nb_encode_init(AVCodecContext * avctx)
Zdenek Kabelac's avatar
Zdenek Kabelac committed
196 197
{
    AMRContext *s = avctx->priv_data;
Benoit Fouet's avatar
Benoit Fouet committed
198

Zdenek Kabelac's avatar
Zdenek Kabelac committed
199
    s->frameCount=0;
200

Zdenek Kabelac's avatar
Zdenek Kabelac committed
201 202
    if(avctx->sample_rate!=8000)
    {
203
        av_log(avctx, AV_LOG_ERROR, "Only 8000Hz sample rate supported\n");
Zdenek Kabelac's avatar
Zdenek Kabelac committed
204 205 206 207 208
        return -1;
    }

    if(avctx->channels!=1)
    {
209
        av_log(avctx, AV_LOG_ERROR, "Only mono supported\n");
Zdenek Kabelac's avatar
Zdenek Kabelac committed
210 211 212 213 214 215 216 217 218
        return -1;
    }

    avctx->frame_size=160;
    avctx->coded_frame= avcodec_alloc_frame();

    s->enstate=Encoder_Interface_init(0);
    if(!s->enstate)
    {
219
        av_log(avctx, AV_LOG_ERROR, "Encoder_Interface_init error\n");
Zdenek Kabelac's avatar
Zdenek Kabelac committed
220 221 222
        return -1;
    }

Benoit Fouet's avatar
Benoit Fouet committed
223 224
    if((s->enc_bitrate=getBitrateMode(avctx->bit_rate))<0)
    {
225
        av_log(avctx, AV_LOG_ERROR, nb_bitrate_unsupported);
Benoit Fouet's avatar
Benoit Fouet committed
226 227
        return -1;
    }
Zdenek Kabelac's avatar
Zdenek Kabelac committed
228 229 230 231

    return 0;
}

232
static av_cold int amr_nb_encode_close(AVCodecContext * avctx)
Zdenek Kabelac's avatar
Zdenek Kabelac committed
233 234
{
    AMRContext *s = avctx->priv_data;
Benoit Fouet's avatar
Benoit Fouet committed
235

Zdenek Kabelac's avatar
Zdenek Kabelac committed
236 237 238 239 240 241
    Encoder_Interface_exit(s->enstate);
    av_freep(&avctx->coded_frame);
    return 0;
}

static int amr_nb_encode_frame(AVCodecContext *avctx,
242
                            unsigned char *frame/*out*/, int buf_size, void *data/*in*/)
Zdenek Kabelac's avatar
Zdenek Kabelac committed
243
{
244
    AMRContext *s = avctx->priv_data;
Zdenek Kabelac's avatar
Zdenek Kabelac committed
245 246
    int written;

Benoit Fouet's avatar
Benoit Fouet committed
247 248
    if((s->enc_bitrate=getBitrateMode(avctx->bit_rate))<0)
    {
249
        av_log(avctx, AV_LOG_ERROR, nb_bitrate_unsupported);
Benoit Fouet's avatar
Benoit Fouet committed
250 251
        return -1;
    }
252

253 254 255 256
    written = Encoder_Interface_Encode(s->enstate,
        s->enc_bitrate,
        data,
        frame,
Zdenek Kabelac's avatar
Zdenek Kabelac committed
257
        0);
258
    /* av_log(NULL,AV_LOG_DEBUG,"amr_nb_encode_frame encoded %u bytes, bitrate %u, first byte was %#02x\n",written, s->enc_bitrate, frame[0] ); */
Zdenek Kabelac's avatar
Zdenek Kabelac committed
259 260 261 262

    return written;
}

263
AVCodec libamr_nb_encoder =
Zdenek Kabelac's avatar
Zdenek Kabelac committed
264
{
265
    "libamr_nb",
Zdenek Kabelac's avatar
Zdenek Kabelac committed
266 267 268 269 270 271 272
    CODEC_TYPE_AUDIO,
    CODEC_ID_AMR_NB,
    sizeof(AMRContext),
    amr_nb_encode_init,
    amr_nb_encode_frame,
    amr_nb_encode_close,
    NULL,
273
    .sample_fmts = (enum SampleFormat[]){SAMPLE_FMT_S16,SAMPLE_FMT_NONE},
274
    .long_name = NULL_IF_CONFIG_SMALL("libamr-nb Adaptive Multi-Rate (AMR) Narrow-Band"),
Zdenek Kabelac's avatar
Zdenek Kabelac committed
275
};
276

277 278
#endif

279
/* -----------AMR wideband ------------*/
280
#if CONFIG_LIBAMR_WB
281 282

#ifdef _TYPEDEF_H
Diego Biurrun's avatar
typo  
Diego Biurrun committed
283
//To avoid duplicate typedefs from typedef in amr-nb
284 285 286
#define typedef_h
#endif

287 288
#include <amrwb/dec_if.h>
#include <amrwb/if_rom.h>
289 290 291 292

/* Common code for fixed and float version*/
typedef struct AMRWB_bitrates
{
Benoit Fouet's avatar
Benoit Fouet committed
293
    int rate;
294 295 296
    int mode;
} AMRWB_bitrates;

297 298 299 300 301 302 303
typedef struct AMRWBContext {
    int frameCount;
    void *state;
    int mode;
    Word16 allow_dtx;
} AMRWBContext;

304 305
#if CONFIG_LIBAMR_WB_ENCODER

306 307
#include <amrwb/enc_if.h>

308 309
static int getWBBitrateMode(int bitrate)
{
Benoit Fouet's avatar
Benoit Fouet committed
310 311 312 313 314 315 316 317 318 319
    /* make the correspondance between bitrate and mode */
    AMRWB_bitrates rates[]={ {6600,0},
                           {8850,1},
                           {12650,2},
                           {14250,3},
                           {15850,4},
                           {18250,5},
                           {19850,6},
                           {23050,7},
                           {23850,8},
320 321 322 323
                         };
    int i;

    for(i=0;i<9;i++)
Benoit Fouet's avatar
Benoit Fouet committed
324
        if(rates[i].rate==bitrate)
325
            return rates[i].mode;
Benoit Fouet's avatar
Benoit Fouet committed
326 327
    /* no bitrate matching, return an error */
    return -1;
328 329
}

330
static av_cold int amr_wb_encode_init(AVCodecContext * avctx)
331
{
332
    AMRWBContext *s = avctx->priv_data;
Benoit Fouet's avatar
Benoit Fouet committed
333

334
    s->frameCount=0;
335

336 337
    if(avctx->sample_rate!=16000)
    {
338
        av_log(avctx, AV_LOG_ERROR, "Only 16000Hz sample rate supported\n");
339 340 341 342 343
        return -1;
    }

    if(avctx->channels!=1)
    {
344
        av_log(avctx, AV_LOG_ERROR, "Only mono supported\n");
345 346 347
        return -1;
    }

Benoit Fouet's avatar
Benoit Fouet committed
348 349
    if((s->mode=getWBBitrateMode(avctx->bit_rate))<0)
    {
350
        av_log(avctx, AV_LOG_ERROR, wb_bitrate_unsupported);
Benoit Fouet's avatar
Benoit Fouet committed
351 352 353
        return -1;
    }

354 355 356 357 358 359 360 361 362 363 364
    avctx->frame_size=320;
    avctx->coded_frame= avcodec_alloc_frame();

    s->state = E_IF_init();
    s->allow_dtx=0;

    return 0;
}

static int amr_wb_encode_close(AVCodecContext * avctx)
{
365
    AMRWBContext *s = avctx->priv_data;
Benoit Fouet's avatar
Benoit Fouet committed
366

367 368 369 370 371 372 373
    E_IF_exit(s->state);
    av_freep(&avctx->coded_frame);
    s->frameCount++;
    return 0;
}

static int amr_wb_encode_frame(AVCodecContext *avctx,
374
                            unsigned char *frame/*out*/, int buf_size, void *data/*in*/)
375
{
376
    AMRWBContext *s = avctx->priv_data;
377
    int size;
378

Benoit Fouet's avatar
Benoit Fouet committed
379 380
    if((s->mode=getWBBitrateMode(avctx->bit_rate))<0)
    {
381
        av_log(avctx, AV_LOG_ERROR, wb_bitrate_unsupported);
Benoit Fouet's avatar
Benoit Fouet committed
382 383
        return -1;
    }
384
    size = E_IF_encode(s->state, s->mode, data, frame, s->allow_dtx);
385 386 387
    return size;
}

388 389 390 391 392 393 394 395 396 397 398 399 400 401
AVCodec libamr_wb_encoder =
{
    "libamr_wb",
    CODEC_TYPE_AUDIO,
    CODEC_ID_AMR_WB,
    sizeof(AMRWBContext),
    amr_wb_encode_init,
    amr_wb_encode_frame,
    amr_wb_encode_close,
    NULL,
    .sample_fmts = (enum SampleFormat[]){SAMPLE_FMT_S16,SAMPLE_FMT_NONE},
    .long_name = NULL_IF_CONFIG_SMALL("libamr-wb Adaptive Multi-Rate (AMR) Wide-Band"),
};

402 403
#endif

404
static av_cold int amr_wb_decode_init(AVCodecContext * avctx)
405
{
406
    AMRWBContext *s = avctx->priv_data;
Benoit Fouet's avatar
Benoit Fouet committed
407

408 409
    s->frameCount=0;
    s->state = D_IF_init();
Andrew Voznytsa's avatar
Andrew Voznytsa committed
410 411 412 413 414 415 416 417 418

    amr_decode_fix_avctx(avctx);

    if(avctx->channels > 1)
    {
        av_log(avctx, AV_LOG_ERROR, "amr_wb: multichannel decoding not supported\n");
        return -1;
    }

419 420 421 422 423
    return 0;
}

static int amr_wb_decode_frame(AVCodecContext * avctx,
            void *data, int *data_size,
424
            AVPacket *avpkt)
425
{
426 427
    const uint8_t *buf = avpkt->data;
    int buf_size = avpkt->size;
428
    AMRWBContext *s = avctx->priv_data;
429
    const uint8_t*amrData=buf;
430 431
    int mode;
    int packet_size;
432
    static const uint8_t block_size[16] = {18, 23, 33, 37, 41, 47, 51, 59, 61, 6, 6, 0, 0, 0, 1, 1};
433

434
    if(buf_size==0)
435 436 437 438 439 440 441 442 443
        /* nothing to do */
        return 0;

    mode = (amrData[0] >> 3) & 0x000F;
    packet_size = block_size[mode];

    if(packet_size > buf_size) {
        av_log(avctx, AV_LOG_ERROR, "amr frame too short (%u, should be %u)\n", buf_size, packet_size+1);
        return -1;
444
    }
445

446 447 448 449
    s->frameCount++;
    D_IF_decode( s->state, amrData, data, _good_frame);
    *data_size=320*2;
    return packet_size;
450 451 452 453
}

static int amr_wb_decode_close(AVCodecContext * avctx)
{
454
    AMRWBContext *s = avctx->priv_data;
Benoit Fouet's avatar
Benoit Fouet committed
455

456 457 458 459
    D_IF_exit(s->state);
    return 0;
}

460
AVCodec libamr_wb_decoder =
461
{
462
    "libamr_wb",
463 464 465 466 467 468 469
    CODEC_TYPE_AUDIO,
    CODEC_ID_AMR_WB,
    sizeof(AMRWBContext),
    amr_wb_decode_init,
    NULL,
    amr_wb_decode_close,
    amr_wb_decode_frame,
470
    .long_name = NULL_IF_CONFIG_SMALL("libamr-wb Adaptive Multi-Rate (AMR) Wide-Band"),
471 472
};

473
#endif //CONFIG_LIBAMR_WB