libspeex: convert to new channel layout API

Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
Anton Khirnov 2019-06-03 17:54:20 +02:00 committed by James Almer
parent ffdf7269a5
commit 97d9a32938
2 changed files with 25 additions and 17 deletions

View File

@ -42,7 +42,7 @@ static av_cold int libspeex_decode_init(AVCodecContext *avctx)
LibSpeexContext *s = avctx->priv_data;
const SpeexMode *mode;
SpeexHeader *header = NULL;
int spx_mode;
int spx_mode, channels;
if (avctx->extradata && avctx->extradata_size >= 80) {
header = speex_packet_to_header(avctx->extradata,
@ -68,7 +68,7 @@ static av_cold int libspeex_decode_init(AVCodecContext *avctx)
spx_mode = 0;
} else if (header) {
avctx->sample_rate = header->rate;
avctx->channels = header->nb_channels;
channels = header->nb_channels;
spx_mode = header->mode;
speex_header_free(header);
} else {
@ -94,14 +94,15 @@ static av_cold int libspeex_decode_init(AVCodecContext *avctx)
if (!avctx->sample_rate)
avctx->sample_rate = 8000 << spx_mode;
if (avctx->channels < 1 || avctx->channels > 2) {
if (channels < 1 || channels > 2) {
/* libspeex can handle mono or stereo if initialized as stereo */
av_log(avctx, AV_LOG_ERROR, "Invalid channel count: %d.\n"
"Decoding as stereo.\n", avctx->channels);
avctx->channels = 2;
"Decoding as stereo.\n", channels);
channels = 2;
}
avctx->channel_layout = avctx->channels == 2 ? AV_CH_LAYOUT_STEREO :
AV_CH_LAYOUT_MONO;
av_channel_layout_uninit(&avctx->ch_layout);
avctx->ch_layout = channels == 2 ? (AVChannelLayout)AV_CHANNEL_LAYOUT_STEREO :
(AVChannelLayout)AV_CHANNEL_LAYOUT_MONO;
speex_bits_init(&s->bits);
s->dec_state = speex_decoder_init(mode);
@ -110,7 +111,7 @@ static av_cold int libspeex_decode_init(AVCodecContext *avctx)
return -1;
}
if (avctx->channels == 2) {
if (channels == 2) {
SpeexCallback callback;
callback.callback_id = SPEEX_INBAND_STEREO;
callback.func = speex_std_stereo_request_handler;
@ -163,7 +164,7 @@ static int libspeex_decode_frame(AVCodecContext *avctx, void *data,
av_log(avctx, AV_LOG_ERROR, "Error decoding Speex frame.\n");
return AVERROR_INVALIDDATA;
}
if (avctx->channels == 2)
if (avctx->ch_layout.nb_channels == 2)
speex_decode_stereo_int(output, s->frame_size, &s->stereo);
*got_frame_ptr = 1;

View File

@ -28,8 +28,8 @@
* order to control various encoding parameters.
*
* Channels
* Speex only supports mono or stereo, so avctx->channels must be set to
* 1 or 2.
* Speex only supports mono or stereo, so avctx->ch_layout.nb_channels must
* be set to 1 or 2.
*
* Sample Rate / Encoding Mode
* Speex has 3 modes, each of which uses a specific sample rate.
@ -114,7 +114,7 @@ static av_cold void print_enc_params(AVCodecContext *avctx,
{
const char *mode_str = "unknown";
av_log(avctx, AV_LOG_DEBUG, "channels: %d\n", avctx->channels);
av_log(avctx, AV_LOG_DEBUG, "channels: %d\n", avctx->ch_layout.nb_channels);
switch (s->header.mode) {
case SPEEX_MODEID_NB: mode_str = "narrowband"; break;
case SPEEX_MODEID_WB: mode_str = "wideband"; break;
@ -146,15 +146,16 @@ static av_cold void print_enc_params(AVCodecContext *avctx,
static av_cold int encode_init(AVCodecContext *avctx)
{
LibSpeexEncContext *s = avctx->priv_data;
int channels = avctx->ch_layout.nb_channels;
const SpeexMode *mode;
uint8_t *header_data;
int header_size;
int32_t complexity;
/* channels */
if (avctx->channels < 1 || avctx->channels > 2) {
if (channels < 1 || channels > 2) {
av_log(avctx, AV_LOG_ERROR, "Invalid channels (%d). Only stereo and "
"mono are supported\n", avctx->channels);
"mono are supported\n", channels);
return AVERROR(EINVAL);
}
@ -175,7 +176,7 @@ static av_cold int encode_init(AVCodecContext *avctx)
av_log(avctx, AV_LOG_ERROR, "Error initializing libspeex\n");
return -1;
}
speex_init_header(&s->header, avctx->sample_rate, avctx->channels, mode);
speex_init_header(&s->header, avctx->sample_rate, channels, mode);
/* rate control method and parameters */
if (avctx->flags & AV_CODEC_FLAG_QSCALE) {
@ -210,7 +211,7 @@ static av_cold int encode_init(AVCodecContext *avctx)
}
/* stereo side information adds about 800 bps to the base bitrate */
/* TODO: this should be calculated exactly */
avctx->bit_rate = s->header.bitrate + (avctx->channels == 2 ? 800 : 0);
avctx->bit_rate = s->header.bitrate + (channels == 2 ? 800 : 0);
}
/* VAD is activated with VBR or can be turned on by itself */
@ -275,7 +276,7 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *avpkt,
if (samples) {
/* encode Speex frame */
if (avctx->channels == 2)
if (avctx->ch_layout.nb_channels == 2)
speex_encode_stereo_int(samples, s->header.frame_size, &s->bits);
speex_encode_int(s->enc_state, samples, &s->bits);
s->pkt_frame_count++;
@ -359,9 +360,15 @@ const AVCodec ff_libspeex_encoder = {
.capabilities = AV_CODEC_CAP_DELAY,
.sample_fmts = (const enum AVSampleFormat[]){ AV_SAMPLE_FMT_S16,
AV_SAMPLE_FMT_NONE },
#if FF_API_OLD_CHANNEL_LAYOUT
.channel_layouts = (const uint64_t[]){ AV_CH_LAYOUT_MONO,
AV_CH_LAYOUT_STEREO,
0 },
#endif
.ch_layouts = (const AVChannelLayout[]) { AV_CHANNEL_LAYOUT_MONO,
AV_CHANNEL_LAYOUT_STEREO,
{ 0 },
},
.supported_samplerates = (const int[]){ 8000, 16000, 32000, 0 },
.priv_class = &speex_class,
.defaults = defaults,