int get_stream_info(AAC_Dec codec)
{
codec->aac_stream_info = aacDecoder_GetStreamInfo(codec->handle);
LOGI("channel = %d\t sample_rate = %d\t", codec->aac_stream_info->numChannels, codec->aac_stream_info->sampleRate);
LOGI("frame_size = %d\t aot = %d\t bitrate = %d", codec->aac_stream_info->frameSize, codec->aac_stream_info->aot, codec->aac_stream_info->bitRate);
}
static int aac_get_stream_info(AVCodecContext *avctx)
{
AACContext *s = avctx->priv_data;
CStreamInfo *info = aacDecoder_GetStreamInfo(s->handle);
if (!info) {
av_log(avctx, AV_LOG_ERROR, "Unable to get stream info\n");
return AVERROR(EINVAL);
}
if (info->sampleRate <= 0) {
av_log(avctx, AV_LOG_ERROR, "Stream info not initialized\n");
return AVERROR(EINVAL);
}
avctx->channels = info->numChannels;
avctx->sample_rate = info->sampleRate;
avctx->frame_size = info->frameSize;
return 0;
}
void *io_thread_a2dp_sink_aac(void *arg) {
struct ba_transport *t = (struct ba_transport *)arg;
pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, NULL);
pthread_cleanup_push(CANCEL_ROUTINE(io_thread_release), t);
if (t->bt_fd == -1) {
error("Invalid BT socket: %d", t->bt_fd);
goto fail_open;
}
if (t->mtu_read <= 0) {
error("Invalid reading MTU: %zu", t->mtu_read);
goto fail_open;
}
HANDLE_AACDECODER handle;
AAC_DECODER_ERROR err;
if ((handle = aacDecoder_Open(TT_MP4_LATM_MCP1, 1)) == NULL) {
error("Couldn't open AAC decoder");
goto fail_open;
}
pthread_cleanup_push(CANCEL_ROUTINE(aacDecoder_Close), handle);
const unsigned int channels = transport_get_channels(t);
#ifdef AACDECODER_LIB_VL0
if ((err = aacDecoder_SetParam(handle, AAC_PCM_MIN_OUTPUT_CHANNELS, channels)) != AAC_DEC_OK) {
error("Couldn't set min output channels: %s", aacdec_strerror(err));
goto fail_init;
}
if ((err = aacDecoder_SetParam(handle, AAC_PCM_MAX_OUTPUT_CHANNELS, channels)) != AAC_DEC_OK) {
error("Couldn't set max output channels: %s", aacdec_strerror(err));
goto fail_init;
}
#else
if ((err = aacDecoder_SetParam(handle, AAC_PCM_OUTPUT_CHANNELS, channels)) != AAC_DEC_OK) {
error("Couldn't set output channels: %s", aacdec_strerror(err));
goto fail_init;
}
#endif
const size_t in_buffer_size = t->mtu_read;
const size_t out_buffer_size = 2048 * channels * sizeof(INT_PCM);
uint8_t *in_buffer = malloc(in_buffer_size);
int16_t *out_buffer = malloc(out_buffer_size);
pthread_cleanup_push(CANCEL_ROUTINE(free), in_buffer);
pthread_cleanup_push(CANCEL_ROUTINE(free), out_buffer);
if (in_buffer == NULL || out_buffer == NULL) {
error("Couldn't create data buffers: %s", strerror(ENOMEM));
goto fail;
}
struct pollfd pfds[] = {
{ t->event_fd, POLLIN, 0 },
{ -1, POLLIN, 0 },
};
debug("Starting IO loop: %s",
bluetooth_profile_to_string(t->profile, t->codec));
for (;;) {
pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
CStreamInfo *aacinf;
ssize_t len;
/* add BT socket to the poll if transport is active */
pfds[1].fd = t->state == TRANSPORT_ACTIVE ? t->bt_fd : -1;
if (poll(pfds, sizeof(pfds) / sizeof(*pfds), -1) == -1) {
error("Transport poll error: %s", strerror(errno));
goto fail;
}
if (pfds[0].revents & POLLIN) {
/* dispatch incoming event */
eventfd_t event;
eventfd_read(pfds[0].fd, &event);
continue;
}
if ((len = read(pfds[1].fd, in_buffer, in_buffer_size)) == -1) {
debug("BT read error: %s", strerror(errno));
continue;
}
pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, NULL);
/* it seems that zero is never returned... */
if (len == 0) {
debug("BT socket has been closed: %d", pfds[1].fd);
/* Prevent sending the release request to the BlueZ. If the socket has
* been closed, it means that BlueZ has already closed the connection. */
close(pfds[1].fd);
t->bt_fd = -1;
goto fail;
}
if (io_thread_open_pcm_write(&t->a2dp.pcm) == -1) {
if (errno != ENXIO)
error("Couldn't open FIFO: %s", strerror(errno));
continue;
}
const rtp_header_t *rtp_header = (rtp_header_t *)in_buffer;
uint8_t *rtp_latm = (uint8_t *)&rtp_header->csrc[rtp_header->cc];
size_t rtp_latm_len = len - ((void *)rtp_latm - (void *)rtp_header);
if (rtp_header->paytype != 96) {
warn("Unsupported RTP payload type: %u", rtp_header->paytype);
continue;
}
unsigned int data_len = rtp_latm_len;
unsigned int valid = rtp_latm_len;
if ((err = aacDecoder_Fill(handle, &rtp_latm, &data_len, &valid)) != AAC_DEC_OK)
error("AAC buffer fill error: %s", aacdec_strerror(err));
else if ((err = aacDecoder_DecodeFrame(handle, out_buffer, out_buffer_size, 0)) != AAC_DEC_OK)
error("AAC decode frame error: %s", aacdec_strerror(err));
else if ((aacinf = aacDecoder_GetStreamInfo(handle)) == NULL)
error("Couldn't get AAC stream info");
else {
const size_t size = aacinf->frameSize * aacinf->numChannels;
if (io_thread_write_pcm(&t->a2dp.pcm, out_buffer, size) == -1)
error("FIFO write error: %s", strerror(errno));
}
}
fail:
pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, NULL);
pthread_cleanup_pop(1);
pthread_cleanup_pop(1);
fail_init:
pthread_cleanup_pop(1);
fail_open:
pthread_cleanup_pop(1);
return NULL;
}
static int get_stream_info(AVCodecContext *avctx)
{
FDKAACDecContext *s = avctx->priv_data;
CStreamInfo *info = aacDecoder_GetStreamInfo(s->handle);
int channel_counts[0x24] = { 0 };
int i, ch_error = 0;
uint64_t ch_layout = 0;
if (!info) {
av_log(avctx, AV_LOG_ERROR, "Unable to get stream info\n");
return AVERROR_UNKNOWN;
}
if (info->sampleRate <= 0) {
av_log(avctx, AV_LOG_ERROR, "Stream info not initialized\n");
return AVERROR_UNKNOWN;
}
avctx->sample_rate = info->sampleRate;
avctx->frame_size = info->frameSize;
for (i = 0; i < info->numChannels; i++) {
AUDIO_CHANNEL_TYPE ctype = info->pChannelType[i];
if (ctype <= ACT_NONE || ctype >= FF_ARRAY_ELEMS(channel_counts)) {
av_log(avctx, AV_LOG_WARNING, "unknown channel type\n");
break;
}
channel_counts[ctype]++;
}
av_log(avctx, AV_LOG_DEBUG,
"%d channels - front:%d side:%d back:%d lfe:%d top:%d\n",
info->numChannels,
channel_counts[ACT_FRONT], channel_counts[ACT_SIDE],
channel_counts[ACT_BACK], channel_counts[ACT_LFE],
channel_counts[ACT_FRONT_TOP] + channel_counts[ACT_SIDE_TOP] +
channel_counts[ACT_BACK_TOP] + channel_counts[ACT_TOP]);
switch (channel_counts[ACT_FRONT]) {
case 4:
ch_layout |= AV_CH_LAYOUT_STEREO | AV_CH_FRONT_LEFT_OF_CENTER |
AV_CH_FRONT_RIGHT_OF_CENTER;
break;
case 3:
ch_layout |= AV_CH_LAYOUT_STEREO | AV_CH_FRONT_CENTER;
break;
case 2:
ch_layout |= AV_CH_LAYOUT_STEREO;
break;
case 1:
ch_layout |= AV_CH_FRONT_CENTER;
break;
default:
av_log(avctx, AV_LOG_WARNING,
"unsupported number of front channels: %d\n",
channel_counts[ACT_FRONT]);
ch_error = 1;
break;
}
if (channel_counts[ACT_SIDE] > 0) {
if (channel_counts[ACT_SIDE] == 2) {
ch_layout |= AV_CH_SIDE_LEFT | AV_CH_SIDE_RIGHT;
} else {
av_log(avctx, AV_LOG_WARNING,
"unsupported number of side channels: %d\n",
channel_counts[ACT_SIDE]);
ch_error = 1;
}
}
if (channel_counts[ACT_BACK] > 0) {
switch (channel_counts[ACT_BACK]) {
case 3:
ch_layout |= AV_CH_BACK_LEFT | AV_CH_BACK_RIGHT | AV_CH_BACK_CENTER;
break;
case 2:
ch_layout |= AV_CH_BACK_LEFT | AV_CH_BACK_RIGHT;
break;
case 1:
ch_layout |= AV_CH_BACK_CENTER;
break;
default:
av_log(avctx, AV_LOG_WARNING,
"unsupported number of back channels: %d\n",
channel_counts[ACT_BACK]);
ch_error = 1;
break;
}
}
if (channel_counts[ACT_LFE] > 0) {
if (channel_counts[ACT_LFE] == 1) {
ch_layout |= AV_CH_LOW_FREQUENCY;
} else {
av_log(avctx, AV_LOG_WARNING,
"unsupported number of LFE channels: %d\n",
channel_counts[ACT_LFE]);
ch_error = 1;
}
}
if (!ch_error &&
av_get_channel_layout_nb_channels(ch_layout) != info->numChannels) {
av_log(avctx, AV_LOG_WARNING, "unsupported channel configuration\n");
ch_error = 1;
}
if (ch_error)
avctx->channel_layout = 0;
else
avctx->channel_layout = ch_layout;
avctx->channels = info->numChannels;
return 0;
}
int main(int argc, char *argv[]) {
const char *infile, *outfile;
AVFormatContext *in = NULL;
AVStream *st = NULL;
void *wav = NULL;
int output_size, ret, i;
uint8_t *output_buf;
int16_t *decode_buf;
HANDLE_AACDECODER handle;
AAC_DECODER_ERROR err;
int frame_size = 0;
if (argc < 3) {
fprintf(stderr, "%s in.m4a out.wav\n", argv[0]);
return 1;
}
infile = argv[1];
outfile = argv[2];
av_register_all();
avformat_network_init();
ret = avformat_open_input(&in, infile, NULL, NULL);
if (ret < 0) {
char buf[100];
av_strerror(ret, buf, sizeof(buf));
fprintf(stderr, "%s: %s\n", infile, buf);
return 1;
}
for (i = 0; i < in->nb_streams && !st; i++) {
if (in->streams[i]->codec->codec_id == AV_CODEC_ID_AAC)
st = in->streams[i];
}
if (!st) {
fprintf(stderr, "No AAC stream found\n");
return 1;
}
if (!st->codec->extradata_size) {
fprintf(stderr, "No AAC ASC found\n");
return 1;
}
handle = aacDecoder_Open(TT_MP4_RAW, 1);
err = aacDecoder_ConfigRaw(handle, &st->codec->extradata, &st->codec->extradata_size);
if (err != AAC_DEC_OK) {
fprintf(stderr, "Unable to decode the ASC\n");
return 1;
}
output_size = 8*2*1024;
output_buf = (uint8_t*) malloc(output_size);
decode_buf = (int16_t*) malloc(output_size);
while (1) {
int i;
UINT valid;
AVPacket pkt = { 0 };
int ret = av_read_frame(in, &pkt);
if (ret < 0) {
if (ret == AVERROR(EAGAIN))
continue;
break;
}
if (pkt.stream_index != st->index) {
av_free_packet(&pkt);
continue;
}
valid = pkt.size;
err = aacDecoder_Fill(handle, &pkt.data, &pkt.size, &valid);
if (err != AAC_DEC_OK) {
fprintf(stderr, "Fill failed: %x\n", err);
break;
}
err = aacDecoder_DecodeFrame(handle, decode_buf, output_size, 0);
av_free_packet(&pkt);
if (err == AAC_DEC_NOT_ENOUGH_BITS)
continue;
if (err != AAC_DEC_OK) {
fprintf(stderr, "Decode failed: %x\n", err);
continue;
}
if (!wav) {
CStreamInfo *info = aacDecoder_GetStreamInfo(handle);
if (!info || info->sampleRate <= 0) {
fprintf(stderr, "No stream info\n");
break;
}
frame_size = info->frameSize * info->numChannels;
// Note, this probably doesn't return channels > 2 in the right order for wav
wav = wav_write_open(outfile, info->sampleRate, 16, info->numChannels);
if (!wav) {
perror(outfile);
break;
}
}
for (i = 0; i < frame_size; i++) {
uint8_t* out = &output_buf[2*i];
out[0] = decode_buf[i] & 0xff;
out[1] = decode_buf[i] >> 8;
}
wav_write_data(wav, output_buf, 2*frame_size);
}
free(output_buf);
free(decode_buf);
avformat_close_input(&in);
if (wav)
wav_write_close(wav);
aacDecoder_Close(handle);
return 0;
}
