libopencore-amr.c 11.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
/*
 * AMR Audio decoder stub
 * Copyright (c) 2003 the ffmpeg project
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * 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
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

22
#include "libavutil/avstring.h"
23
#include "libavutil/channel_layout.h"
24
#include "libavutil/common.h"
25
#include "libavutil/opt.h"
26
#include "avcodec.h"
27 28
#include "audio_frame_queue.h"
#include "internal.h"
29

30
static int amr_decode_fix_avctx(AVCodecContext *avctx)
31
{
32
    const int is_amr_wb = 1 + (avctx->codec_id == AV_CODEC_ID_AMR_WB);
33

34 35
    if (!avctx->sample_rate)
        avctx->sample_rate = 8000 * is_amr_wb;
36

37
    if (avctx->channels > 1) {
38
        avpriv_report_missing_feature(avctx, "multi-channel AMR");
39 40
        return AVERROR_PATCHWELCOME;
    }
41

42 43 44
    avctx->channels       = 1;
    avctx->channel_layout = AV_CH_LAYOUT_MONO;
    avctx->sample_fmt     = AV_SAMPLE_FMT_S16;
45
    return 0;
46 47 48 49 50 51 52 53
}

#if CONFIG_LIBOPENCORE_AMRNB

#include <opencore-amrnb/interf_dec.h>
#include <opencore-amrnb/interf_enc.h>

typedef struct AMRContext {
54
    AVClass *av_class;
55 56
    void *dec_state;
    void *enc_state;
57
    int   enc_bitrate;
58
    int   enc_mode;
59
    int   enc_dtx;
60
    int   enc_last_frame;
61
    AudioFrameQueue afq;
62 63
} AMRContext;

64
#if CONFIG_LIBOPENCORE_AMRNB_DECODER
65 66
static av_cold int amr_nb_decode_init(AVCodecContext *avctx)
{
67
    AMRContext *s  = avctx->priv_data;
68 69 70 71
    int ret;

    if ((ret = amr_decode_fix_avctx(avctx)) < 0)
        return ret;
72

73 74
    s->dec_state   = Decoder_Interface_init();
    if (!s->dec_state) {
75
        av_log(avctx, AV_LOG_ERROR, "Decoder_Interface_init error\n");
76 77 78 79 80 81 82 83 84 85
        return -1;
    }

    return 0;
}

static av_cold int amr_nb_decode_close(AVCodecContext *avctx)
{
    AMRContext *s = avctx->priv_data;

86
    Decoder_Interface_exit(s->dec_state);
87

88 89 90 91
    return 0;
}

static int amr_nb_decode_frame(AVCodecContext *avctx, void *data,
92
                               int *got_frame_ptr, AVPacket *avpkt)
93
{
94
    AVFrame *frame     = data;
95 96
    const uint8_t *buf = avpkt->data;
    int buf_size       = avpkt->size;
97
    AMRContext *s      = avctx->priv_data;
98 99
    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;
100
    int packet_size, ret;
101

102
    av_dlog(avctx, "amr_decode_frame buf=%p buf_size=%d frame_count=%d!!\n",
103
            buf, buf_size, avctx->frame_number);
104

105
    /* get output buffer */
106
    frame->nb_samples = 160;
107
    if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
108
        return ret;
109

110
    dec_mode    = (buf[0] >> 3) & 0x000F;
111 112 113 114 115
    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);
116
        return AVERROR_INVALIDDATA;
117 118
    }

119 120
    av_dlog(avctx, "packet_size=%d buf= 0x%X %X %X %X\n",
              packet_size, buf[0], buf[1], buf[2], buf[3]);
121
    /* call decoder */
122
    Decoder_Interface_Decode(s->dec_state, buf, (short *)frame->data[0], 0);
123

124
    *got_frame_ptr = 1;
125 126 127 128

    return packet_size;
}

129
AVCodec ff_libopencore_amrnb_decoder = {
130 131
    .name           = "libopencore_amrnb",
    .type           = AVMEDIA_TYPE_AUDIO,
132
    .id             = AV_CODEC_ID_AMR_NB,
133 134 135 136
    .priv_data_size = sizeof(AMRContext),
    .init           = amr_nb_decode_init,
    .close          = amr_nb_decode_close,
    .decode         = amr_nb_decode_frame,
137
    .capabilities   = CODEC_CAP_DR1,
138
    .long_name      = NULL_IF_CONFIG_SMALL("OpenCORE AMR-NB (Adaptive Multi-Rate Narrow-Band)"),
139
};
140
#endif /* CONFIG_LIBOPENCORE_AMRNB_DECODER */
141

142
#if CONFIG_LIBOPENCORE_AMRNB_ENCODER
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
/* Common code for fixed and float version*/
typedef struct AMR_bitrates {
    int       rate;
    enum Mode mode;
} AMR_bitrates;

/* Match desired bitrate */
static int get_bitrate_mode(int bitrate, void *log_ctx)
{
    /* make the correspondance between bitrate and mode */
    static const AMR_bitrates rates[] = {
        { 4750, MR475 }, { 5150, MR515 }, {  5900, MR59  }, {  6700, MR67  },
        { 7400, MR74 },  { 7950, MR795 }, { 10200, MR102 }, { 12200, MR122 }
    };
    int i, best = -1, min_diff = 0;
    char log_buf[200];

    for (i = 0; i < 8; i++) {
        if (rates[i].rate == bitrate)
            return rates[i].mode;
        if (best < 0 || abs(rates[i].rate - bitrate) < min_diff) {
            best     = i;
            min_diff = abs(rates[i].rate - bitrate);
        }
    }
    /* no bitrate matching exactly, log a warning */
    snprintf(log_buf, sizeof(log_buf), "bitrate not supported: use one of ");
    for (i = 0; i < 8; i++)
        av_strlcatf(log_buf, sizeof(log_buf), "%.2fk, ", rates[i].rate    / 1000.f);
    av_strlcatf(log_buf, sizeof(log_buf), "using %.2fk", rates[best].rate / 1000.f);
    av_log(log_ctx, AV_LOG_WARNING, "%s\n", log_buf);

    return best;
}

static const AVOption options[] = {
    { "dtx", "Allow DTX (generate comfort noise)", offsetof(AMRContext, enc_dtx), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_ENCODING_PARAM },
    { NULL }
};

183
static const AVClass amrnb_class = {
184 185 186
    "libopencore_amrnb", av_default_item_name, options, LIBAVUTIL_VERSION_INT
};

187 188 189 190
static av_cold int amr_nb_encode_init(AVCodecContext *avctx)
{
    AMRContext *s = avctx->priv_data;

191
    if (avctx->sample_rate != 8000 && avctx->strict_std_compliance > FF_COMPLIANCE_UNOFFICIAL) {
192
        av_log(avctx, AV_LOG_ERROR, "Only 8000Hz sample rate supported\n");
193
        return AVERROR(ENOSYS);
194 195 196 197
    }

    if (avctx->channels != 1) {
        av_log(avctx, AV_LOG_ERROR, "Only mono supported\n");
198
        return AVERROR(ENOSYS);
199 200 201
    }

    avctx->frame_size  = 160;
202
    avctx->delay       =  50;
203
    ff_af_queue_init(avctx, &s->afq);
204

205
    s->enc_state = Encoder_Interface_init(s->enc_dtx);
206
    if (!s->enc_state) {
207
        av_log(avctx, AV_LOG_ERROR, "Encoder_Interface_init error\n");
208
        av_freep(&avctx->coded_frame);
209 210 211
        return -1;
    }

212 213
    s->enc_mode    = get_bitrate_mode(avctx->bit_rate, avctx);
    s->enc_bitrate = avctx->bit_rate;
214 215 216 217 218 219 220 221

    return 0;
}

static av_cold int amr_nb_encode_close(AVCodecContext *avctx)
{
    AMRContext *s = avctx->priv_data;

222
    Encoder_Interface_exit(s->enc_state);
223
    ff_af_queue_close(&s->afq);
224 225 226
    return 0;
}

227 228
static int amr_nb_encode_frame(AVCodecContext *avctx, AVPacket *avpkt,
                               const AVFrame *frame, int *got_packet_ptr)
229 230
{
    AMRContext *s = avctx->priv_data;
231
    int written, ret;
232
    int16_t *flush_buf = NULL;
233
    const int16_t *samples = frame ? (const int16_t *)frame->data[0] : NULL;
234

235 236 237
    if (s->enc_bitrate != avctx->bit_rate) {
        s->enc_mode    = get_bitrate_mode(avctx->bit_rate, avctx);
        s->enc_bitrate = avctx->bit_rate;
238 239
    }

240
    if ((ret = ff_alloc_packet2(avctx, avpkt, 32)) < 0)
241 242 243 244 245
        return ret;

    if (frame) {
        if (frame->nb_samples < avctx->frame_size) {
            flush_buf = av_mallocz(avctx->frame_size * sizeof(*flush_buf));
246 247
            if (!flush_buf)
                return AVERROR(ENOMEM);
248
            memcpy(flush_buf, samples, frame->nb_samples * sizeof(*flush_buf));
249
            samples = flush_buf;
250
            if (frame->nb_samples < avctx->frame_size - avctx->delay)
251 252
                s->enc_last_frame = -1;
        }
253
        if ((ret = ff_af_queue_add(&s->afq, frame)) < 0) {
254 255 256
            av_freep(&flush_buf);
            return ret;
        }
257 258 259
    } else {
        if (s->enc_last_frame < 0)
            return 0;
260
        flush_buf = av_mallocz(avctx->frame_size * sizeof(*flush_buf));
261 262 263 264 265 266 267
        if (!flush_buf)
            return AVERROR(ENOMEM);
        samples = flush_buf;
        s->enc_last_frame = -1;
    }

    written = Encoder_Interface_Encode(s->enc_state, s->enc_mode, samples,
268
                                       avpkt->data, 0);
269
    av_dlog(avctx, "amr_nb_encode_frame encoded %u bytes, bitrate %u, first byte was %#02x\n",
270
            written, s->enc_mode, avpkt->data[0]);
271

272 273 274 275 276 277
    /* Get the next frame pts/duration */
    ff_af_queue_remove(&s->afq, avctx->frame_size, &avpkt->pts,
                       &avpkt->duration);

    avpkt->size = written;
    *got_packet_ptr = 1;
278
    av_freep(&flush_buf);
279
    return 0;
280 281
}

282
AVCodec ff_libopencore_amrnb_encoder = {
283 284
    .name           = "libopencore_amrnb",
    .type           = AVMEDIA_TYPE_AUDIO,
285
    .id             = AV_CODEC_ID_AMR_NB,
286 287
    .priv_data_size = sizeof(AMRContext),
    .init           = amr_nb_encode_init,
288
    .encode2        = amr_nb_encode_frame,
289
    .close          = amr_nb_encode_close,
290
    .capabilities   = CODEC_CAP_DELAY | CODEC_CAP_SMALL_LAST_FRAME,
291 292
    .sample_fmts    = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_S16,
                                                     AV_SAMPLE_FMT_NONE },
293
    .long_name      = NULL_IF_CONFIG_SMALL("OpenCORE AMR-NB (Adaptive Multi-Rate Narrow-Band)"),
294
    .priv_class     = &amrnb_class,
295
};
296
#endif /* CONFIG_LIBOPENCORE_AMRNB_ENCODER */
297

298
#endif /* CONFIG_LIBOPENCORE_AMRNB */
299 300

/* -----------AMR wideband ------------*/
301
#if CONFIG_LIBOPENCORE_AMRWB_DECODER
302 303 304 305 306 307 308 309 310 311 312

#include <opencore-amrwb/dec_if.h>
#include <opencore-amrwb/if_rom.h>

typedef struct AMRWBContext {
    void  *state;
} AMRWBContext;

static av_cold int amr_wb_decode_init(AVCodecContext *avctx)
{
    AMRWBContext *s = avctx->priv_data;
313
    int ret;
314

315 316
    if ((ret = amr_decode_fix_avctx(avctx)) < 0)
        return ret;
317

318
    s->state        = D_IF_init();
319 320 321 322 323

    return 0;
}

static int amr_wb_decode_frame(AVCodecContext *avctx, void *data,
324
                               int *got_frame_ptr, AVPacket *avpkt)
325
{
326
    AVFrame *frame     = data;
327 328
    const uint8_t *buf = avpkt->data;
    int buf_size       = avpkt->size;
329
    AMRWBContext *s    = avctx->priv_data;
330 331
    int mode, ret;
    int packet_size;
332 333
    static const uint8_t block_size[16] = {18, 24, 33, 37, 41, 47, 51, 59, 61, 6, 6, 0, 0, 0, 1, 1};

334
    /* get output buffer */
335
    frame->nb_samples = 320;
336
    if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
337
        return ret;
338

339
    mode        = (buf[0] >> 3) & 0x000F;
340 341 342 343 344
    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);
345
        return AVERROR_INVALIDDATA;
346
    }
347 348 349 350
    if (!packet_size) {
        av_log(avctx, AV_LOG_ERROR, "amr packet_size invalid\n");
        return AVERROR_INVALIDDATA;
    }
351

352
    D_IF_decode(s->state, buf, (short *)frame->data[0], _good_frame);
353

354
    *got_frame_ptr = 1;
355

356 357 358 359 360 361 362 363 364 365 366
    return packet_size;
}

static int amr_wb_decode_close(AVCodecContext *avctx)
{
    AMRWBContext *s = avctx->priv_data;

    D_IF_exit(s->state);
    return 0;
}

367
AVCodec ff_libopencore_amrwb_decoder = {
368 369
    .name           = "libopencore_amrwb",
    .type           = AVMEDIA_TYPE_AUDIO,
370
    .id             = AV_CODEC_ID_AMR_WB,
371 372 373 374
    .priv_data_size = sizeof(AMRWBContext),
    .init           = amr_wb_decode_init,
    .close          = amr_wb_decode_close,
    .decode         = amr_wb_decode_frame,
375
    .capabilities   = CODEC_CAP_DR1,
376
    .long_name      = NULL_IF_CONFIG_SMALL("OpenCORE AMR-WB (Adaptive Multi-Rate Wide-Band)"),
377 378
};

379
#endif /* CONFIG_LIBOPENCORE_AMRWB_DECODER */