2010-01-30 16:57:40 +00:00
|
|
|
/*
|
|
|
|
* This file is part of MPlayer.
|
|
|
|
*
|
|
|
|
* MPlayer is free software; you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* MPlayer 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 General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License along
|
|
|
|
* with MPlayer; if not, write to the Free Software Foundation, Inc.,
|
|
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
*/
|
|
|
|
|
2002-03-25 21:06:01 +00:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <unistd.h>
|
2011-08-21 19:47:59 +00:00
|
|
|
#include <stdbool.h>
|
2002-03-25 21:06:01 +00:00
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
#include <libavcodec/avcodec.h>
|
|
|
|
|
2011-08-21 19:47:59 +00:00
|
|
|
#include "talloc.h"
|
|
|
|
|
2002-03-25 21:06:01 +00:00
|
|
|
#include "config.h"
|
|
|
|
#include "mp_msg.h"
|
2010-06-02 15:38:24 +00:00
|
|
|
#include "options.h"
|
2002-03-25 21:06:01 +00:00
|
|
|
|
|
|
|
#include "ad_internal.h"
|
2007-12-10 16:53:30 +00:00
|
|
|
#include "libaf/reorder_ch.h"
|
2002-03-25 21:06:01 +00:00
|
|
|
|
2006-12-07 11:58:07 +00:00
|
|
|
#include "mpbswap.h"
|
2002-03-25 21:06:01 +00:00
|
|
|
|
2008-04-26 13:48:06 +00:00
|
|
|
static const ad_info_t info =
|
2002-03-25 21:06:01 +00:00
|
|
|
{
|
2011-08-21 19:13:49 +00:00
|
|
|
"FFmpeg/libavcodec audio decoders",
|
|
|
|
"ffmpeg",
|
|
|
|
"Nick Kurshev",
|
|
|
|
"ffmpeg.sf.net",
|
|
|
|
""
|
2002-03-25 21:06:01 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
LIBAD_EXTERN(ffmpeg)
|
|
|
|
|
2011-08-21 19:47:59 +00:00
|
|
|
struct priv {
|
|
|
|
AVCodecContext *avctx;
|
|
|
|
bool old_packet;
|
|
|
|
};
|
|
|
|
|
2002-03-25 21:06:01 +00:00
|
|
|
static int preinit(sh_audio_t *sh)
|
|
|
|
{
|
2011-08-21 19:13:49 +00:00
|
|
|
sh->audio_out_minsize = AVCODEC_MAX_AUDIO_FRAME_SIZE;
|
|
|
|
return 1;
|
2002-03-25 21:06:01 +00:00
|
|
|
}
|
|
|
|
|
2010-11-21 12:52:08 +00:00
|
|
|
/* Prefer playing audio with the samplerate given in container data
|
|
|
|
* if available, but take number the number of channels and sample format
|
|
|
|
* from the codec, since if the codec isn't using the correct values for
|
|
|
|
* those everything breaks anyway.
|
|
|
|
*/
|
2011-08-21 19:13:49 +00:00
|
|
|
static int setup_format(sh_audio_t *sh_audio,
|
|
|
|
const AVCodecContext *lavc_context)
|
2010-07-15 17:59:46 +00:00
|
|
|
{
|
|
|
|
int sample_format = sh_audio->sample_format;
|
|
|
|
switch (lavc_context->sample_fmt) {
|
2011-08-21 19:13:49 +00:00
|
|
|
case SAMPLE_FMT_U8: sample_format = AF_FORMAT_U8; break;
|
|
|
|
case SAMPLE_FMT_S16: sample_format = AF_FORMAT_S16_NE; break;
|
|
|
|
case SAMPLE_FMT_S32: sample_format = AF_FORMAT_S32_NE; break;
|
|
|
|
case SAMPLE_FMT_FLT: sample_format = AF_FORMAT_FLOAT_NE; break;
|
|
|
|
default:
|
|
|
|
mp_msg(MSGT_DECAUDIO, MSGL_FATAL, "Unsupported sample format\n");
|
2010-07-15 17:59:46 +00:00
|
|
|
}
|
|
|
|
|
2010-11-21 12:52:08 +00:00
|
|
|
bool broken_srate = false;
|
|
|
|
int samplerate = lavc_context->sample_rate;
|
|
|
|
int container_samplerate = sh_audio->container_out_samplerate;
|
|
|
|
if (!container_samplerate && sh_audio->wf)
|
|
|
|
container_samplerate = sh_audio->wf->nSamplesPerSec;
|
|
|
|
if (lavc_context->codec_id == CODEC_ID_AAC
|
|
|
|
&& samplerate == 2 * container_samplerate)
|
|
|
|
broken_srate = true;
|
|
|
|
else if (container_samplerate)
|
|
|
|
samplerate = container_samplerate;
|
|
|
|
|
2010-07-15 17:59:46 +00:00
|
|
|
if (lavc_context->channels != sh_audio->channels ||
|
|
|
|
samplerate != sh_audio->samplerate ||
|
|
|
|
sample_format != sh_audio->sample_format) {
|
2011-08-21 19:13:49 +00:00
|
|
|
sh_audio->channels = lavc_context->channels;
|
|
|
|
sh_audio->samplerate = samplerate;
|
2010-07-15 17:59:46 +00:00
|
|
|
sh_audio->sample_format = sample_format;
|
2011-08-21 19:13:49 +00:00
|
|
|
sh_audio->samplesize = af_fmt2bits(sh_audio->sample_format) / 8;
|
2010-07-15 17:59:46 +00:00
|
|
|
if (broken_srate)
|
|
|
|
mp_msg(MSGT_DECAUDIO, MSGL_WARN,
|
|
|
|
"Ignoring broken container sample rate for AAC with SBR\n");
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2002-03-25 21:06:01 +00:00
|
|
|
static int init(sh_audio_t *sh_audio)
|
|
|
|
{
|
2010-06-02 15:38:24 +00:00
|
|
|
struct MPOpts *opts = sh_audio->opts;
|
2002-04-03 21:01:15 +00:00
|
|
|
AVCodecContext *lavc_context;
|
|
|
|
AVCodec *lavc_codec;
|
2002-04-01 13:10:03 +00:00
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
mp_msg(MSGT_DECAUDIO, MSGL_V, "FFmpeg's libavcodec audio codec\n");
|
2009-07-06 23:26:13 +00:00
|
|
|
|
2010-08-22 23:38:40 +00:00
|
|
|
lavc_codec = avcodec_find_decoder_by_name(sh_audio->codec->dll);
|
2011-08-21 19:13:49 +00:00
|
|
|
if (!lavc_codec) {
|
|
|
|
mp_tmsg(MSGT_DECAUDIO, MSGL_ERR,
|
|
|
|
"Cannot find codec '%s' in libavcodec...\n",
|
|
|
|
sh_audio->codec->dll);
|
|
|
|
return 0;
|
2002-03-25 21:06:01 +00:00
|
|
|
}
|
2009-07-06 23:26:13 +00:00
|
|
|
|
2011-08-21 19:47:59 +00:00
|
|
|
struct priv *ctx = talloc_zero(NULL, struct priv);
|
|
|
|
sh_audio->context = ctx;
|
2002-10-28 00:40:42 +00:00
|
|
|
lavc_context = avcodec_alloc_context();
|
2011-08-21 19:47:59 +00:00
|
|
|
ctx->avctx = lavc_context;
|
2002-10-28 00:40:42 +00:00
|
|
|
|
2010-06-02 15:38:24 +00:00
|
|
|
lavc_context->drc_scale = opts->drc_level;
|
2007-08-27 15:51:04 +00:00
|
|
|
lavc_context->sample_rate = sh_audio->samplerate;
|
|
|
|
lavc_context->bit_rate = sh_audio->i_bps * 8;
|
2011-08-21 19:13:49 +00:00
|
|
|
if (sh_audio->wf) {
|
|
|
|
lavc_context->channels = sh_audio->wf->nChannels;
|
|
|
|
lavc_context->sample_rate = sh_audio->wf->nSamplesPerSec;
|
|
|
|
lavc_context->bit_rate = sh_audio->wf->nAvgBytesPerSec * 8;
|
|
|
|
lavc_context->block_align = sh_audio->wf->nBlockAlign;
|
|
|
|
lavc_context->bits_per_coded_sample = sh_audio->wf->wBitsPerSample;
|
2002-12-28 00:40:11 +00:00
|
|
|
}
|
2010-10-31 05:26:40 +00:00
|
|
|
lavc_context->request_channels = opts->audio_output_channels;
|
2003-03-08 02:05:14 +00:00
|
|
|
lavc_context->codec_tag = sh_audio->format; //FOURCC
|
2011-04-19 23:59:45 +00:00
|
|
|
lavc_context->codec_type = AVMEDIA_TYPE_AUDIO;
|
2002-12-28 00:40:11 +00:00
|
|
|
lavc_context->codec_id = lavc_codec->id; // not sure if required, imho not --A'rpi
|
2002-10-28 00:40:42 +00:00
|
|
|
|
|
|
|
/* alloc extra data */
|
2002-12-28 00:40:11 +00:00
|
|
|
if (sh_audio->wf && sh_audio->wf->cbSize > 0) {
|
2005-12-19 19:38:28 +00:00
|
|
|
lavc_context->extradata = av_mallocz(sh_audio->wf->cbSize + FF_INPUT_BUFFER_PADDING_SIZE);
|
2002-10-28 00:40:42 +00:00
|
|
|
lavc_context->extradata_size = sh_audio->wf->cbSize;
|
2010-09-12 11:44:42 +00:00
|
|
|
memcpy(lavc_context->extradata, sh_audio->wf + 1,
|
2002-10-28 00:40:42 +00:00
|
|
|
lavc_context->extradata_size);
|
|
|
|
}
|
|
|
|
|
2005-02-01 19:19:40 +00:00
|
|
|
// for QDM2
|
2011-08-21 19:13:49 +00:00
|
|
|
if (sh_audio->codecdata_len && sh_audio->codecdata &&
|
|
|
|
!lavc_context->extradata) {
|
2011-02-02 07:13:14 +00:00
|
|
|
lavc_context->extradata = av_malloc(sh_audio->codecdata_len +
|
|
|
|
FF_INPUT_BUFFER_PADDING_SIZE);
|
2005-02-01 19:19:40 +00:00
|
|
|
lavc_context->extradata_size = sh_audio->codecdata_len;
|
2009-07-06 23:26:13 +00:00
|
|
|
memcpy(lavc_context->extradata, (char *)sh_audio->codecdata,
|
|
|
|
lavc_context->extradata_size);
|
2005-02-01 19:19:40 +00:00
|
|
|
}
|
|
|
|
|
2002-03-25 21:06:01 +00:00
|
|
|
/* open it */
|
2002-04-03 21:01:15 +00:00
|
|
|
if (avcodec_open(lavc_context, lavc_codec) < 0) {
|
2011-08-21 19:13:49 +00:00
|
|
|
mp_tmsg(MSGT_DECAUDIO, MSGL_ERR, "Could not open codec.\n");
|
2011-08-21 20:04:20 +00:00
|
|
|
uninit(sh_audio);
|
2002-03-25 21:06:01 +00:00
|
|
|
return 0;
|
|
|
|
}
|
2011-08-21 19:13:49 +00:00
|
|
|
mp_msg(MSGT_DECAUDIO, MSGL_V, "INFO: libavcodec \"%s\" init OK!\n",
|
|
|
|
lavc_codec->name);
|
2009-07-06 23:26:13 +00:00
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
if (sh_audio->format == 0x3343414D) {
|
|
|
|
// MACE 3:1
|
|
|
|
sh_audio->ds->ss_div = 2 * 3; // 1 samples/packet
|
|
|
|
sh_audio->ds->ss_mul = 2 * sh_audio->wf->nChannels; // 1 byte*ch/packet
|
|
|
|
} else if (sh_audio->format == 0x3643414D) {
|
|
|
|
// MACE 6:1
|
|
|
|
sh_audio->ds->ss_div = 2 * 6; // 1 samples/packet
|
|
|
|
sh_audio->ds->ss_mul = 2 * sh_audio->wf->nChannels; // 1 byte*ch/packet
|
|
|
|
}
|
2002-03-25 21:06:01 +00:00
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
// Decode at least 1 byte: (to get header filled)
|
|
|
|
for (int tries = 0;;) {
|
|
|
|
int x = decode_audio(sh_audio, sh_audio->a_buffer, 1,
|
|
|
|
sh_audio->a_buffer_size);
|
|
|
|
if (x > 0) {
|
|
|
|
sh_audio->a_buffer_len = x;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (++tries >= 5) {
|
|
|
|
mp_msg(MSGT_DECAUDIO, MSGL_ERR,
|
|
|
|
"ad_ffmpeg: initial decode failed\n");
|
2011-08-21 20:04:20 +00:00
|
|
|
uninit(sh_audio);
|
2011-08-21 19:13:49 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
2002-03-25 21:06:01 +00:00
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
sh_audio->i_bps = lavc_context->bit_rate / 8;
|
|
|
|
if (sh_audio->wf && sh_audio->wf->nAvgBytesPerSec)
|
|
|
|
sh_audio->i_bps = sh_audio->wf->nAvgBytesPerSec;
|
2010-07-15 17:59:46 +00:00
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
switch (lavc_context->sample_fmt) {
|
|
|
|
case SAMPLE_FMT_U8:
|
|
|
|
case SAMPLE_FMT_S16:
|
|
|
|
case SAMPLE_FMT_S32:
|
|
|
|
case SAMPLE_FMT_FLT:
|
|
|
|
break;
|
|
|
|
default:
|
2011-08-21 20:04:20 +00:00
|
|
|
uninit(sh_audio);
|
2011-08-21 19:13:49 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return 1;
|
2002-03-25 21:06:01 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void uninit(sh_audio_t *sh)
|
|
|
|
{
|
2011-08-21 19:47:59 +00:00
|
|
|
struct priv *ctx = sh->context;
|
2011-08-21 20:04:20 +00:00
|
|
|
if (!ctx)
|
|
|
|
return;
|
2011-08-21 19:47:59 +00:00
|
|
|
AVCodecContext *lavc_context = ctx->avctx;
|
2002-10-28 00:40:42 +00:00
|
|
|
|
2011-08-21 20:04:20 +00:00
|
|
|
if (lavc_context) {
|
|
|
|
if (avcodec_close(lavc_context) < 0)
|
|
|
|
mp_tmsg(MSGT_DECVIDEO, MSGL_ERR, "Could not close codec.\n");
|
|
|
|
av_freep(&lavc_context->extradata);
|
|
|
|
av_freep(&lavc_context);
|
|
|
|
}
|
2011-08-21 19:47:59 +00:00
|
|
|
talloc_free(ctx);
|
2011-08-21 20:04:20 +00:00
|
|
|
sh->context = NULL;
|
2002-03-25 21:06:01 +00:00
|
|
|
}
|
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
static int control(sh_audio_t *sh, int cmd, void *arg, ...)
|
2002-03-25 21:06:01 +00:00
|
|
|
{
|
2011-08-21 19:47:59 +00:00
|
|
|
struct priv *ctx = sh->context;
|
2011-08-21 19:13:49 +00:00
|
|
|
switch (cmd) {
|
2004-02-18 15:23:41 +00:00
|
|
|
case ADCTRL_RESYNC_STREAM:
|
2011-08-21 19:47:59 +00:00
|
|
|
avcodec_flush_buffers(ctx->avctx);
|
2010-01-24 15:19:14 +00:00
|
|
|
ds_clear_parser(sh->ds);
|
2011-08-21 19:47:59 +00:00
|
|
|
ctx->old_packet = false;
|
2011-08-21 19:13:49 +00:00
|
|
|
return CONTROL_TRUE;
|
2004-02-18 15:23:41 +00:00
|
|
|
}
|
|
|
|
return CONTROL_UNKNOWN;
|
2002-03-25 21:06:01 +00:00
|
|
|
}
|
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
static int decode_audio(sh_audio_t *sh_audio, unsigned char *buf, int minlen,
|
|
|
|
int maxlen)
|
2002-03-25 21:06:01 +00:00
|
|
|
{
|
2011-08-21 19:47:59 +00:00
|
|
|
struct priv *ctx = sh_audio->context;
|
|
|
|
AVCodecContext *avctx = ctx->avctx;
|
2011-08-21 19:13:49 +00:00
|
|
|
|
|
|
|
unsigned char *start = NULL;
|
|
|
|
int y, len = -1;
|
|
|
|
while (len < minlen) {
|
|
|
|
AVPacket pkt;
|
|
|
|
int len2 = maxlen;
|
|
|
|
double pts;
|
|
|
|
int x = ds_get_packet_pts(sh_audio->ds, &start, &pts);
|
|
|
|
if (x <= 0) {
|
|
|
|
start = NULL;
|
|
|
|
x = 0;
|
|
|
|
ds_parse(sh_audio->ds, &start, &x, MP_NOPTS_VALUE, 0);
|
|
|
|
if (x <= 0)
|
|
|
|
break; // error
|
|
|
|
} else {
|
|
|
|
int in_size = x;
|
|
|
|
int consumed = ds_parse(sh_audio->ds, &start, &x, pts, 0);
|
|
|
|
sh_audio->ds->buffer_pos -= in_size - consumed;
|
|
|
|
}
|
|
|
|
av_init_packet(&pkt);
|
|
|
|
pkt.data = start;
|
|
|
|
pkt.size = x;
|
2011-08-21 19:47:59 +00:00
|
|
|
if (pts != MP_NOPTS_VALUE && !ctx->old_packet) {
|
2011-08-21 19:13:49 +00:00
|
|
|
sh_audio->pts = pts;
|
|
|
|
sh_audio->pts_bytes = 0;
|
|
|
|
}
|
|
|
|
y = avcodec_decode_audio3(avctx, (int16_t *)buf, &len2, &pkt);
|
|
|
|
// LATM may need many packets to find mux info
|
|
|
|
if (y == AVERROR(EAGAIN))
|
|
|
|
continue;
|
|
|
|
if (y < 0) {
|
|
|
|
mp_msg(MSGT_DECAUDIO, MSGL_V, "lavc_audio: error\n");
|
|
|
|
break;
|
|
|
|
}
|
2011-08-21 19:47:59 +00:00
|
|
|
if (!sh_audio->parser && y < x) {
|
2011-08-21 19:13:49 +00:00
|
|
|
sh_audio->ds->buffer_pos += y - x; // put back data (HACK!)
|
2011-08-21 19:47:59 +00:00
|
|
|
ctx->old_packet = true;
|
|
|
|
}
|
2011-08-21 19:13:49 +00:00
|
|
|
if (len2 > 0) {
|
|
|
|
if (avctx->channels >= 5) {
|
|
|
|
int samplesize = av_get_bits_per_sample_format(
|
|
|
|
avctx->sample_fmt) / 8;
|
|
|
|
reorder_channel_nch(buf, AF_CHANNEL_LAYOUT_LAVC_DEFAULT,
|
|
|
|
AF_CHANNEL_LAYOUT_MPLAYER_DEFAULT,
|
|
|
|
avctx->channels,
|
|
|
|
len2 / samplesize, samplesize);
|
|
|
|
}
|
|
|
|
if (len < 0)
|
|
|
|
len = len2;
|
|
|
|
else
|
|
|
|
len += len2;
|
|
|
|
buf += len2;
|
|
|
|
maxlen -= len2;
|
|
|
|
sh_audio->pts_bytes += len2;
|
|
|
|
}
|
|
|
|
mp_dbg(MSGT_DECAUDIO, MSGL_DBG2, "Decoded %d -> %d \n", y, len2);
|
2010-07-15 17:59:46 +00:00
|
|
|
|
2011-08-21 19:13:49 +00:00
|
|
|
if (setup_format(sh_audio, avctx))
|
2010-07-15 17:59:46 +00:00
|
|
|
break;
|
2002-03-25 21:06:01 +00:00
|
|
|
}
|
2011-08-21 19:13:49 +00:00
|
|
|
return len;
|
2002-03-25 21:06:01 +00:00
|
|
|
}
|