yading@10: /* yading@10: * Interface to libgsm for gsm encoding/decoding yading@10: * Copyright (c) 2005 Alban Bedel yading@10: * Copyright (c) 2006, 2007 Michel Bardiaux yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: /** yading@10: * @file yading@10: * Interface to libgsm for gsm encoding/decoding yading@10: */ yading@10: yading@10: // The idiosyncrasies of GSM-in-WAV are explained at http://kbs.cs.tu-berlin.de/~jutta/toast.html yading@10: yading@10: #include "config.h" yading@10: #if HAVE_GSM_H yading@10: #include yading@10: #else yading@10: #include yading@10: #endif yading@10: yading@10: #include "libavutil/channel_layout.h" yading@10: #include "libavutil/common.h" yading@10: #include "avcodec.h" yading@10: #include "internal.h" yading@10: #include "gsm.h" yading@10: yading@10: static av_cold int libgsm_encode_close(AVCodecContext *avctx) { yading@10: gsm_destroy(avctx->priv_data); yading@10: avctx->priv_data = NULL; yading@10: return 0; yading@10: } yading@10: yading@10: static av_cold int libgsm_encode_init(AVCodecContext *avctx) { yading@10: if (avctx->channels > 1) { yading@10: av_log(avctx, AV_LOG_ERROR, "Mono required for GSM, got %d channels\n", yading@10: avctx->channels); yading@10: return -1; yading@10: } yading@10: yading@10: if (avctx->sample_rate != 8000) { yading@10: av_log(avctx, AV_LOG_ERROR, "Sample rate 8000Hz required for GSM, got %dHz\n", yading@10: avctx->sample_rate); yading@10: if (avctx->strict_std_compliance > FF_COMPLIANCE_UNOFFICIAL) yading@10: return -1; yading@10: } yading@10: if (avctx->bit_rate != 13000 /* Official */ && yading@10: avctx->bit_rate != 13200 /* Very common */ && yading@10: avctx->bit_rate != 0 /* Unknown; a.o. mov does not set bitrate when decoding */ ) { yading@10: av_log(avctx, AV_LOG_ERROR, "Bitrate 13000bps required for GSM, got %dbps\n", yading@10: avctx->bit_rate); yading@10: if (avctx->strict_std_compliance > FF_COMPLIANCE_UNOFFICIAL) yading@10: return -1; yading@10: } yading@10: yading@10: avctx->priv_data = gsm_create(); yading@10: if (!avctx->priv_data) yading@10: goto error; yading@10: yading@10: switch(avctx->codec_id) { yading@10: case AV_CODEC_ID_GSM: yading@10: avctx->frame_size = GSM_FRAME_SIZE; yading@10: avctx->block_align = GSM_BLOCK_SIZE; yading@10: break; yading@10: case AV_CODEC_ID_GSM_MS: { yading@10: int one = 1; yading@10: gsm_option(avctx->priv_data, GSM_OPT_WAV49, &one); yading@10: avctx->frame_size = 2*GSM_FRAME_SIZE; yading@10: avctx->block_align = GSM_MS_BLOCK_SIZE; yading@10: } yading@10: } yading@10: yading@10: return 0; yading@10: error: yading@10: libgsm_encode_close(avctx); yading@10: return -1; yading@10: } yading@10: yading@10: static int libgsm_encode_frame(AVCodecContext *avctx, AVPacket *avpkt, yading@10: const AVFrame *frame, int *got_packet_ptr) yading@10: { yading@10: int ret; yading@10: gsm_signal *samples = (gsm_signal *)frame->data[0]; yading@10: struct gsm_state *state = avctx->priv_data; yading@10: yading@10: if ((ret = ff_alloc_packet2(avctx, avpkt, avctx->block_align)) < 0) yading@10: return ret; yading@10: yading@10: switch(avctx->codec_id) { yading@10: case AV_CODEC_ID_GSM: yading@10: gsm_encode(state, samples, avpkt->data); yading@10: break; yading@10: case AV_CODEC_ID_GSM_MS: yading@10: gsm_encode(state, samples, avpkt->data); yading@10: gsm_encode(state, samples + GSM_FRAME_SIZE, avpkt->data + 32); yading@10: } yading@10: yading@10: *got_packet_ptr = 1; yading@10: return 0; yading@10: } yading@10: yading@10: yading@10: #if CONFIG_LIBGSM_ENCODER yading@10: AVCodec ff_libgsm_encoder = { yading@10: .name = "libgsm", yading@10: .type = AVMEDIA_TYPE_AUDIO, yading@10: .id = AV_CODEC_ID_GSM, yading@10: .init = libgsm_encode_init, yading@10: .encode2 = libgsm_encode_frame, yading@10: .close = libgsm_encode_close, yading@10: .sample_fmts = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_S16, yading@10: AV_SAMPLE_FMT_NONE }, yading@10: .long_name = NULL_IF_CONFIG_SMALL("libgsm GSM"), yading@10: }; yading@10: #endif yading@10: #if CONFIG_LIBGSM_MS_ENCODER yading@10: AVCodec ff_libgsm_ms_encoder = { yading@10: .name = "libgsm_ms", yading@10: .type = AVMEDIA_TYPE_AUDIO, yading@10: .id = AV_CODEC_ID_GSM_MS, yading@10: .init = libgsm_encode_init, yading@10: .encode2 = libgsm_encode_frame, yading@10: .close = libgsm_encode_close, yading@10: .sample_fmts = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_S16, yading@10: AV_SAMPLE_FMT_NONE }, yading@10: .long_name = NULL_IF_CONFIG_SMALL("libgsm GSM Microsoft variant"), yading@10: }; yading@10: #endif yading@10: yading@10: typedef struct LibGSMDecodeContext { yading@10: struct gsm_state *state; yading@10: } LibGSMDecodeContext; yading@10: yading@10: static av_cold int libgsm_decode_init(AVCodecContext *avctx) { yading@10: LibGSMDecodeContext *s = avctx->priv_data; yading@10: yading@10: avctx->channels = 1; yading@10: avctx->channel_layout = AV_CH_LAYOUT_MONO; yading@10: if (!avctx->sample_rate) yading@10: avctx->sample_rate = 8000; yading@10: avctx->sample_fmt = AV_SAMPLE_FMT_S16; yading@10: yading@10: s->state = gsm_create(); yading@10: yading@10: switch(avctx->codec_id) { yading@10: case AV_CODEC_ID_GSM: yading@10: avctx->frame_size = GSM_FRAME_SIZE; yading@10: avctx->block_align = GSM_BLOCK_SIZE; yading@10: break; yading@10: case AV_CODEC_ID_GSM_MS: { yading@10: int one = 1; yading@10: gsm_option(s->state, GSM_OPT_WAV49, &one); yading@10: avctx->frame_size = 2 * GSM_FRAME_SIZE; yading@10: avctx->block_align = GSM_MS_BLOCK_SIZE; yading@10: } yading@10: } yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static av_cold int libgsm_decode_close(AVCodecContext *avctx) { yading@10: LibGSMDecodeContext *s = avctx->priv_data; yading@10: yading@10: gsm_destroy(s->state); yading@10: s->state = NULL; yading@10: return 0; yading@10: } yading@10: yading@10: static int libgsm_decode_frame(AVCodecContext *avctx, void *data, yading@10: int *got_frame_ptr, AVPacket *avpkt) yading@10: { yading@10: int i, ret; yading@10: LibGSMDecodeContext *s = avctx->priv_data; yading@10: AVFrame *frame = data; yading@10: uint8_t *buf = avpkt->data; yading@10: int buf_size = avpkt->size; yading@10: int16_t *samples; yading@10: yading@10: if (buf_size < avctx->block_align) { yading@10: av_log(avctx, AV_LOG_ERROR, "Packet is too small\n"); yading@10: return AVERROR_INVALIDDATA; yading@10: } yading@10: yading@10: /* get output buffer */ yading@10: frame->nb_samples = avctx->frame_size; yading@10: if ((ret = ff_get_buffer(avctx, frame, 0)) < 0) yading@10: return ret; yading@10: samples = (int16_t *)frame->data[0]; yading@10: yading@10: for (i = 0; i < avctx->frame_size / GSM_FRAME_SIZE; i++) { yading@10: if ((ret = gsm_decode(s->state, buf, samples)) < 0) yading@10: return -1; yading@10: buf += GSM_BLOCK_SIZE; yading@10: samples += GSM_FRAME_SIZE; yading@10: } yading@10: yading@10: *got_frame_ptr = 1; yading@10: yading@10: return avctx->block_align; yading@10: } yading@10: yading@10: static void libgsm_flush(AVCodecContext *avctx) { yading@10: LibGSMDecodeContext *s = avctx->priv_data; yading@10: int one = 1; yading@10: yading@10: gsm_destroy(s->state); yading@10: s->state = gsm_create(); yading@10: if (avctx->codec_id == AV_CODEC_ID_GSM_MS) yading@10: gsm_option(s->state, GSM_OPT_WAV49, &one); yading@10: } yading@10: yading@10: #if CONFIG_LIBGSM_DECODER yading@10: AVCodec ff_libgsm_decoder = { yading@10: .name = "libgsm", yading@10: .type = AVMEDIA_TYPE_AUDIO, yading@10: .id = AV_CODEC_ID_GSM, yading@10: .priv_data_size = sizeof(LibGSMDecodeContext), yading@10: .init = libgsm_decode_init, yading@10: .close = libgsm_decode_close, yading@10: .decode = libgsm_decode_frame, yading@10: .flush = libgsm_flush, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .long_name = NULL_IF_CONFIG_SMALL("libgsm GSM"), yading@10: }; yading@10: #endif yading@10: #if CONFIG_LIBGSM_MS_DECODER yading@10: AVCodec ff_libgsm_ms_decoder = { yading@10: .name = "libgsm_ms", yading@10: .type = AVMEDIA_TYPE_AUDIO, yading@10: .id = AV_CODEC_ID_GSM_MS, yading@10: .priv_data_size = sizeof(LibGSMDecodeContext), yading@10: .init = libgsm_decode_init, yading@10: .close = libgsm_decode_close, yading@10: .decode = libgsm_decode_frame, yading@10: .flush = libgsm_flush, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .long_name = NULL_IF_CONFIG_SMALL("libgsm GSM Microsoft variant"), yading@10: }; yading@10: #endif