ref: 635f1edc98fe8f8ce0f327e8151e3afa0c99a73c
dir: /plugins/foo_mp4/foo_mp4.cpp/
/*
** FAAD2 - Freeware Advanced Audio (AAC) Decoder including SBR decoding
** Copyright (C) 2003 M. Bakker, Ahead Software AG, http://www.nero.com
**
** This program 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.
**
** This program 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 this program; if not, write to the Free Software
** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
**
** Any non-GPL usage of this software or parts of this software is strictly
** forbidden.
**
** Commercial non-GPL licensing of this software is possible.
** For more info contact Ahead Software through Mpeg4AAClicense@nero.com.
**
** $Id: foo_mp4.cpp,v 1.61 2003/09/24 18:11:22 ca5e Exp $
**/
#include <mp4.h>
#include <faad.h>
#include "foobar2000/SDK/foobar2000.h"
#include "foobar2000/foo_input_std/id3v2_hacks.h"
//#define DBG_OUT(A) OutputDebugString(A)
#define DBG_OUT(A)
#if 0
char *STRIP_REVISION(const char *str)
{
char *tmp = strchr(str, ' ');
int len = lstrlen(tmp)-2;
if (len > 0 && tmp)
tmp[len] = '\0';
else
tmp = "000";
return tmp;
}
#endif
DECLARE_COMPONENT_VERSION ("MPEG-4 AAC decoder",
"1.60",
"Based on FAAD2 v" FAAD2_VERSION "\nCopyright (C) 2002-2003 http://www.audiocoding.com" );
static const char *object_type_string(int type)
{
static const char *types[31] = {
"AAC Main",
"AAC LC",
"AAC SSR",
"AAC LTP",
"AAC HE",
"AAC Scalable",
"TwinVQ",
"CELP",
"HVXC",
"Reserved",
"Reserved",
"TTSI",
"Main synthetic",
"Wavetable synthesis",
"General MIDI",
"Algorithmic Synthesis and Audio FX",
"ER AAC LC",
"Reserved",
"ER AAC LTP",
"ER AAC scalable",
"ER TwinVQ",
"ER BSAC",
"ER AAC LD",
"ER CELP",
"ER HVXC",
"ER HILN",
"ER Parametric",
"Reserved",
"Reserved",
"Reserved",
"Reserved",
};
if (type<1 || type>31) return NULL;
return types[type-1];
}
class input_mp4 : public input
{
public:
virtual bool test_filename(const char * fn,const char * ext)
{
return (!stricmp(ext,"MP4") || !stricmp(ext,"M4A"));
}
virtual bool open(reader *r, file_info *info, unsigned flags)
{
unsigned __int8 *buffer;
unsigned __int32 buffer_size;
unsigned __int8 channels;
unsigned __int32 samplerate;
faacDecConfigurationPtr config;
mp4AudioSpecificConfig mp4ASC;
m_reader = r;
if (!m_reader->can_seek())
{
console::error("MP4 file needs seeking.");
return 0;
}
hDecoder = faacDecOpen();
if (!hDecoder)
{
console::error("Failed to open FAAD2 library.");
return 0;
}
config = faacDecGetCurrentConfiguration(hDecoder);
config->outputFormat = FAAD_FMT_DOUBLE;
faacDecSetConfiguration(hDecoder, config);
hFile = MP4ReadCb(0, open_cb, close_cb, read_cb, write_cb,
setpos_cb, getpos_cb, filesize_cb, (void*)m_reader);
if (hFile == MP4_INVALID_FILE_HANDLE)
{
console::error("Failed to open MP4 file.");
return 0;
}
track = GetAACTrack(hFile);
if (track < 0)
{
console::error("No valid AAC track found.");
return 0;
}
buffer = NULL;
buffer_size = 0;
MP4GetTrackESConfiguration(hFile, track, &buffer, &buffer_size);
if (!buffer)
{
console::error("Unable to read track specific configuration.");
return 0;
}
AudioSpecificConfig((unsigned char*)buffer, buffer_size, &mp4ASC);
int rc = faacDecInit2(hDecoder, (unsigned char*)buffer, buffer_size,
(unsigned long*)&samplerate, (unsigned char*)&channels);
if (buffer) free(buffer);
if (rc < 0)
{
console::error("Unable to initialise FAAD2 library.");
return 0;
}
numSamples = MP4GetTrackNumberOfSamples(hFile, track);
sampleId = 1;
m_timescale = MP4GetTrackTimeScale(hFile, track);
m_samplerate = samplerate;
m_seekto = 0;
m_framesize = 1024;
if (mp4ASC.frameLengthFlag == 1) m_framesize = 960;
useAacLength = false;
MP4Duration trackDuration = MP4GetTrackDuration(hFile, track);
m_length = (double)(__int64)trackDuration / (double)m_timescale;
info->set_length(m_length);
info->info_set_int("bitrate",(__int64)(1.0/1000.0 *
(double)(__int64)MP4GetTrackIntegerProperty(hFile,
track, "mdia.minf.stbl.stsd.mp4a.esds.decConfigDescr.avgBitrate") + 0.5));
info->info_set_int("channels", (__int64)channels);
info->info_set_int("samplerate", (__int64)samplerate);
const char *profile_str = object_type_string(mp4ASC.objectTypeIndex);
if (profile_str)
info->info_set("aac_profile", profile_str);
if (mp4ASC.sbr_present_flag == 1) {
info->info_set("codec", "AAC+SBR");
m_framesize *= 2;
} else {
info->info_set("codec", "AAC");
}
if (flags & OPEN_FLAG_GET_INFO)
ReadMP4Tag(info);
return 1;
}
input_mp4()
{
hFile = MP4_INVALID_FILE_HANDLE;
hDecoder = NULL;
m_eof = false;
}
~input_mp4()
{
if (hFile != MP4_INVALID_FILE_HANDLE)
MP4Close(hFile);
if (hDecoder)
faacDecClose(hDecoder);
}
virtual int run(audio_chunk * chunk)
{
faacDecFrameInfo frameInfo;
audio_sample *sample_buffer;
MP4Timestamp sample_pos;
MP4Duration sample_dur;
unsigned __int64 sample_count;
unsigned __int64 delay = 0;
bool initial = (sampleId == 1);
do {
if (m_eof || (sampleId > numSamples)) return 0;
if (sampleId == MP4_INVALID_SAMPLE_ID)
{
console::error("Invalid sampleId.");
return 0;//-1;
}
do {
unsigned char *buffer = NULL;
unsigned __int32 buffer_size = 0;
delay = 0;
MP4ReadSample(hFile, track, sampleId,
(unsigned __int8**)&buffer, &buffer_size,
&sample_pos, &sample_dur, NULL, NULL);
sampleId++;
sample_buffer = (audio_sample*)faacDecDecode(hDecoder, &frameInfo, buffer, buffer_size);
if (buffer) free(buffer);
if (useAacLength || (m_timescale != m_samplerate)) {
sample_count = frameInfo.channels ? frameInfo.samples/frameInfo.channels : 0;
} else {
sample_count = sample_dur;
if (!useAacLength && !initial && m_seekto<sample_pos && (sampleId < numSamples/2) && (sample_dur*frameInfo.channels != frameInfo.samples))
{
console::info("MP4 seems to have incorrect frame duration, using values from AAC data");
useAacLength = true;
}
}
if (initial && (sample_count < m_framesize) && frameInfo.channels)
delay = (frameInfo.samples/frameInfo.channels) - sample_count;
if (frameInfo.error || !sample_buffer)
{
const char *msg;
if (frameInfo.error)
msg = faacDecGetErrorMessage(frameInfo.error);
else
msg = "faacDecDecode() error";
console::warning(msg);
if (sampleId > numSamples) return 0;//-1;
console::info("Skipping frame");
sample_count = 0;
}
} while (frameInfo.error || frameInfo.samples == 0 || frameInfo.channels == 0 || sample_count == 0);
unsigned __int64 skip = (sample_pos < m_seekto) ? (m_seekto - sample_pos) : 0;
if (skip < sample_count)
{
if (frameInfo.channels == 6 && frameInfo.num_lfe_channels)
{
//channel order for 5.1: L/R/C/LF/BL/BR
audio_sample r1, r2, r3, r4, r5, r6;
for (unsigned int i = 0; i < frameInfo.samples; i += frameInfo.channels)
{
r1 = sample_buffer[i];
r2 = sample_buffer[i+1];
r3 = sample_buffer[i+2];
r4 = sample_buffer[i+3];
r5 = sample_buffer[i+4];
r6 = sample_buffer[i+5];
sample_buffer[i] = r2;
sample_buffer[i+1] = r3;
sample_buffer[i+2] = r1;
sample_buffer[i+3] = r6;
sample_buffer[i+4] = r4;
sample_buffer[i+5] = r5;
}
}
unsigned int samples = (unsigned int)(sample_count - skip);
chunk->set_data((audio_sample*)sample_buffer + (unsigned int)(skip+delay)*frameInfo.channels,
samples, frameInfo.channels, frameInfo.samplerate);
m_seekto = 0;
}
} while (sample_pos + sample_dur <= m_seekto);
return 1;
}
virtual set_info_t set_info(reader *r, const file_info * info)
{
m_reader = r;
hFile = MP4ModifyCb(0, 0, open_cb, close_cb, read_cb, write_cb,
setpos_cb, getpos_cb, filesize_cb, (void*)m_reader);
if (hFile == MP4_INVALID_FILE_HANDLE) return SET_INFO_FAILURE;
MP4MetadataDelete(hFile);
/* replay gain writing */
const char *p = NULL;
p = info->info_get("TOOL");
if (p)
MP4SetMetadataTool(hFile, p);
p = info->info_get("REPLAYGAIN_TRACK_PEAK");
if (p)
MP4SetMetadataFreeForm(hFile, "REPLAYGAIN_TRACK_PEAK", (unsigned __int8*)p, strlen(p));
p = info->info_get("REPLAYGAIN_TRACK_GAIN");
if (p)
MP4SetMetadataFreeForm(hFile, "REPLAYGAIN_TRACK_GAIN", (unsigned __int8*)p, strlen(p));
p = info->info_get("REPLAYGAIN_ALBUM_PEAK");
if (p)
MP4SetMetadataFreeForm(hFile, "REPLAYGAIN_ALBUM_PEAK", (unsigned __int8*)p, strlen(p));
p = info->info_get("REPLAYGAIN_ALBUM_GAIN");
if (p)
MP4SetMetadataFreeForm(hFile, "REPLAYGAIN_ALBUM_GAIN", (unsigned __int8*)p, strlen(p));
int numItems = info->meta_get_count();
if (numItems > 0)
{
for (int i = 0; i < numItems; i++)
{
char *pName = (char*)info->meta_enum_name(i);
const char *val = info->meta_enum_value(i);
if (!val) continue;
if (stricmp(pName, "TITLE") == 0)
{
MP4SetMetadataName(hFile, val);
} else if (stricmp(pName, "ARTIST") == 0) {
MP4SetMetadataArtist(hFile, val);
} else if (stricmp(pName, "WRITER") == 0) {
MP4SetMetadataWriter(hFile, val);
} else if (stricmp(pName, "ALBUM") == 0) {
MP4SetMetadataAlbum(hFile, val);
} else if (stricmp(pName, "YEAR") == 0 || stricmp(pName, "DATE") == 0) {
MP4SetMetadataYear(hFile, val);
} else if (stricmp(pName, "COMMENT") == 0) {
MP4SetMetadataComment(hFile, val);
} else if (stricmp(pName, "GENRE") == 0) {
MP4SetMetadataGenre(hFile, val);
} else if (stricmp(pName, "TRACKNUMBER") == 0) {
unsigned __int16 trkn = 0, tot = 0;
sscanf(val, "%d", &trkn);
MP4SetMetadataTrack(hFile, trkn, tot);
} else if (stricmp(pName, "DISKNUMBER") == 0 || stricmp(pName, "DISC") == 0) {
unsigned __int16 disk = 0, tot = 0;
sscanf(val, "%d", &disk);
MP4SetMetadataDisk(hFile, disk, tot);
} else if (stricmp(pName, "COMPILATION") == 0) {
unsigned __int8 cpil = 0;
sscanf(val, "%d", &cpil);
MP4SetMetadataCompilation(hFile, cpil);
} else if (stricmp(pName, "TEMPO") == 0) {
unsigned __int16 tempo = 0;
sscanf(val, "%d", &tempo);
MP4SetMetadataTempo(hFile, tempo);
} else {
MP4SetMetadataFreeForm(hFile, pName, (unsigned __int8*)val, strlen(val));
}
}
}
/* end */
return SET_INFO_SUCCESS;
}
virtual bool seek(double seconds)
{
if (seconds >= m_length) {
m_eof = true;
return true;
}
unsigned int frame = (unsigned int)((double)m_framesize * ((double)m_samplerate / (double)m_timescale) + 0.5);
if (frame == 0) frame = 1;
m_seekto = (unsigned __int64)(seconds * m_timescale + 0.5) + frame;
MP4Duration target = m_seekto - frame;
while (1) {
MP4Duration duration = MP4ConvertToTrackDuration(hFile, track, target, m_timescale);
sampleId = MP4GetSampleIdFromTime(hFile, track, duration, 0);
if (sampleId == MP4_INVALID_SAMPLE_ID) return false;
MP4Timestamp position = MP4GetSampleTime(hFile, track, sampleId);
if (position <= m_seekto) break;
if (target == 0) return false;
if (target > frame) target -= frame; else target = 0;
}
faacDecPostSeekReset(hDecoder, -1);
return true;
}
virtual bool is_our_content_type(const char *url, const char *type)
{
return !stricmp(type, "audio/mp4") || !stricmp(type, "audio/x-mp4");
}
private:
reader *m_reader;
faacDecHandle hDecoder;
MP4FileHandle hFile;
MP4SampleId sampleId, numSamples;
MP4TrackId track;
unsigned int m_timescale;
unsigned int m_samplerate;
unsigned int m_framesize;
unsigned __int64 m_seekto;
double m_length;
bool m_eof;
bool useAacLength;
int ReadMP4Tag(file_info *info)
{
unsigned __int32 valueSize = 0;
unsigned __int8* pValue;
char* pName;
int i = 0;
do {
valueSize = 0;
MP4GetMetadataByIndex(hFile, i, (const char**)&pName, &pValue, &valueSize);
if (valueSize > 0)
{
char* val = (char*)malloc((valueSize+1)*sizeof(char));
memset(val, 0, (valueSize+1)*sizeof(char));
memcpy(val, pValue, valueSize*sizeof(char));
if (pName[0] == '�')
{
if (memcmp(pName, "�nam", 4) == 0)
{
info->meta_add("TITLE", val);
} else if (memcmp(pName, "�ART", 4) == 0) {
info->meta_add("ARTIST", val);
} else if (memcmp(pName, "�wrt", 4) == 0) {
info->meta_add("WRITER", val);
} else if (memcmp(pName, "�alb", 4) == 0) {
info->meta_add("ALBUM", val);
} else if (memcmp(pName, "�day", 4) == 0) {
info->meta_add("DATE", val);
} else if (memcmp(pName, "�too", 4) == 0) {
info->info_set("tool", val);
} else if (memcmp(pName, "�cmt", 4) == 0) {
info->meta_add("COMMENT", val);
} else if (memcmp(pName, "�gen", 4) == 0) {
info->meta_add("GENRE", val);
} else {
info->meta_add(pName, val);
}
} else if (memcmp(pName, "gnre", 4) == 0) {
char *t=0;
MP4GetMetadataGenre(hFile, &t);
info->meta_add("GENRE", t);
} else if (memcmp(pName, "trkn", 4) == 0) {
unsigned __int16 trkn = 0, tot = 0;
char t[200];
MP4GetMetadataTrack(hFile, &trkn, &tot);
wsprintf(t, "%d", trkn);
info->meta_add("TRACKNUMBER", t);
} else if (memcmp(pName, "disk", 4) == 0) {
unsigned __int16 disk = 0, tot = 0;
char t[200];
MP4GetMetadataDisk(hFile, &disk, &tot);
wsprintf(t, "%d", disk);
//info->meta_add("DISKNUMBER", t);
info->meta_add("DISC", t);
} else if (memcmp(pName, "cpil", 4) == 0) {
unsigned __int8 cpil = 0;
char t[200];
MP4GetMetadataCompilation(hFile, &cpil);
wsprintf(t, "%d", cpil);
info->meta_add("COMPILATION", t);
} else if (memcmp(pName, "tmpo", 4) == 0) {
unsigned __int16 tempo = 0;
char t[200];
MP4GetMetadataTempo(hFile, &tempo);
wsprintf(t, "%d BPM", tempo);
info->meta_add("TEMPO", t);
} else if (memcmp(pName, "NDFL", 4) == 0) {
/* Removed */
} else {
float f = 0;
if (!stricmp(pName, "REPLAYGAIN_TRACK_PEAK"))
{
sscanf(val, "%f", &f);
info->info_set_replaygain_track_peak((double)f);
} else if (!stricmp(pName, "REPLAYGAIN_TRACK_GAIN")) {
sscanf(val, "%f", &f);
info->info_set_replaygain_track_gain((double)f);
} else if (!stricmp(pName, "REPLAYGAIN_ALBUM_PEAK")) {
sscanf(val, "%f", &f);
info->info_set_replaygain_album_peak((double)f);
} else if (!stricmp(pName, "REPLAYGAIN_ALBUM_GAIN")) {
sscanf(val, "%f", &f);
info->info_set_replaygain_album_gain((double)f);
} else {
info->meta_add(pName, val);
}
}
}
i++;
} while (valueSize > 0);
return 1;
}
int GetAACTrack(MP4FileHandle infile)
{
/* find AAC track */
int i, rc;
int numTracks = MP4GetNumberOfTracks(infile, NULL, /* subType */ 0);
for (i = 0; i < numTracks; i++)
{
MP4TrackId trackId = MP4FindTrackId(infile, i, NULL, /* subType */ 0);
const char* trackType = MP4GetTrackType(infile, trackId);
if (!strcmp(trackType, MP4_AUDIO_TRACK_TYPE))
{
unsigned char *buff = NULL;
int buff_size = 0;
mp4AudioSpecificConfig mp4ASC;
MP4GetTrackESConfiguration(infile, trackId,
(unsigned __int8**)&buff, (unsigned __int32*)&buff_size);
if (buff)
{
rc = AudioSpecificConfig(buff, buff_size, &mp4ASC);
free(buff);
if (rc < 0)
return -1;
return trackId;
}
}
}
/* can't decode this */
return -1;
}
/* file callback stuff */
static unsigned __int32 open_cb(const char *pName,
const char *mode, void *userData)
{
DBG_OUT("open_cb");
return 1;
}
static void close_cb(void *userData)
{
DBG_OUT("close_cb");
return;
}
static unsigned __int32 read_cb(void *pBuffer, unsigned int nBytesToRead,
void *userData)
{
DBG_OUT("read_cb");
reader *r = (reader*)userData;
return r->read(pBuffer, nBytesToRead);
}
static unsigned __int32 write_cb(void *pBuffer, unsigned int nBytesToWrite,
void *userData)
{
DBG_OUT("write_cb");
reader *r = (reader*)userData;
return r->write(pBuffer, nBytesToWrite);
}
static __int64 getpos_cb(void *userData)
{
DBG_OUT("getpos_cb");
reader *r = (reader*)userData;
return r->get_position();
}
static __int32 setpos_cb(unsigned __int32 pos, void *userData)
{
DBG_OUT("setpos_cb");
reader *r = (reader*)userData;
return !(r->seek(pos));
}
static __int64 filesize_cb(void *userData)
{
DBG_OUT("filesize_cb");
reader *r = (reader*)userData;
return r->get_length();
}
};
struct seek_list
{
seek_list *next;
__int64 offset;
};
class input_aac : public input
{
public:
virtual bool test_filename(const char * fn,const char * ext)
{
return !stricmp(ext,"AAC");
}
virtual bool open(reader *r, file_info *info, unsigned flags)
{
int tagsize = 0, tmp = 0;
int bread = 0;
double length = 1.;
__int64 bitrate = 128;
unsigned char channels = 0;
unsigned long samplerate = 0;
int sbr = 0;
int header_type = 0;
int profile = 0;
m_reader = r;
tagsize = (int)id3v2_calc_size(m_reader);
if (tagsize<0) return 0;
if (!(m_aac_buffer = (unsigned char*)malloc(768*6)))
{
console::error("Memory allocation error.");
return 0;
}
for (int init=0; init<2; init++)
{
faacDecConfigurationPtr config;
hDecoder = faacDecOpen();
if (!hDecoder)
{
console::error("Failed to open FAAD2 library.");
return 0;
}
config = faacDecGetCurrentConfiguration(hDecoder);
config->outputFormat = FAAD_FMT_DOUBLE;
faacDecSetConfiguration(hDecoder, config);
memset(m_aac_buffer, 0, 768*6);
bread = m_reader->read(m_aac_buffer, 768*6);
m_aac_bytes_into_buffer = bread;
m_aac_bytes_consumed = 0;
m_file_offset = 0;
m_at_eof = (bread != 768*6) ? 1 : 0;
if (init==0)
{
faacDecFrameInfo frameInfo;
fill_buffer();
if ((m_aac_bytes_consumed = faacDecInit(hDecoder,
m_aac_buffer, m_aac_bytes_into_buffer,
&samplerate, &channels)) < 0)
{
console::error("Can't initialize decoder library.");
return 0;
}
advance_buffer(m_aac_bytes_consumed);
do {
memset(&frameInfo, 0, sizeof(faacDecFrameInfo));
fill_buffer();
faacDecDecode(hDecoder, &frameInfo, m_aac_buffer, m_aac_bytes_into_buffer);
} while (!frameInfo.samples && !frameInfo.error);
if (frameInfo.error)
{
console::error(faacDecGetErrorMessage(frameInfo.error));
return 0;
}
m_samplerate = frameInfo.samplerate;
m_framesize = (frameInfo.channels != 0) ? frameInfo.samples/frameInfo.channels : 0;
sbr = frameInfo.sbr;
profile = frameInfo.object_type;
if (sbr) profile = 5; // temporary fix
header_type = frameInfo.header_type;
faacDecClose(hDecoder);
m_reader->seek(tagsize);
}
}
m_head = (struct seek_list*)malloc(sizeof(struct seek_list));
m_tail = m_head;
m_tail->next = NULL;
m_header_type = 0;
if ((m_aac_buffer[0] == 0xFF) && ((m_aac_buffer[1] & 0xF6) == 0xF0))
{
if (m_reader->can_seek())
{
adts_parse(&bitrate, &length);
m_reader->seek(tagsize);
bread = m_reader->read(m_aac_buffer, 768*6);
if (bread != 768*6)
m_at_eof = 1;
else
m_at_eof = 0;
m_aac_bytes_into_buffer = bread;
m_aac_bytes_consumed = 0;
m_header_type = 1;
}
} else if (memcmp(m_aac_buffer, "ADIF", 4) == 0) {
int skip_size = (m_aac_buffer[4] & 0x80) ? 9 : 0;
bitrate = ((unsigned int)(m_aac_buffer[4 + skip_size] & 0x0F)<<19) |
((unsigned int)m_aac_buffer[5 + skip_size]<<11) |
((unsigned int)m_aac_buffer[6 + skip_size]<<3) |
((unsigned int)m_aac_buffer[7 + skip_size] & 0xE0);
length = (double)m_reader->get_length();
if (length == -1.)
{
length = 1;
} else {
length = ((double)length*8.)/((double)bitrate) + 0.5;
}
bitrate = (__int64)((double)bitrate/1000.0 + 0.5);
m_header_type = 2;
}
if (!m_reader->can_seek())
{
length = 0;
}
fill_buffer();
if ((m_aac_bytes_consumed = faacDecInit(hDecoder,
m_aac_buffer, m_aac_bytes_into_buffer,
&samplerate, &channels)) < 0)
{
console::error("Can't initialize decoder library.");
return 0;
}
advance_buffer(m_aac_bytes_consumed);
m_length = length;
if (flags & OPEN_FLAG_GET_INFO) {
const char *profile_str = object_type_string(profile);
const char *header_str = NULL;
info->info_set_int("bitrate", bitrate);
info->info_set_int("channels", (__int64)channels);
info->info_set_int("samplerate", (__int64)m_samplerate);
if (profile_str)
info->info_set("aac_profile", profile_str);
if (header_type == RAW)
header_str = "RAW";
else if (header_type == ADIF)
header_str = "ADIF";
else if (header_type == ADTS)
header_str = "ADTS";
if (header_str)
info->info_set("aac_header_type", header_str);
if (sbr)
info->info_set("codec", "AAC+SBR");
else
info->info_set("codec", "AAC");
}
tag_reader::g_run_multi(m_reader, info, "ape|id3v2|lyrics3|id3v1");
const char *p = info->meta_get("samples");
if (p) {
m_samples = (unsigned __int64)_atoi64(p);
info->info_set("samples", p);
info->meta_remove_field("samples");
} else {
p = info->info_get("samples");
if (p) m_samples = (unsigned __int64)_atoi64(p);
}
if (m_samples > 0) {
m_length = (double)(signed __int64)m_samples/(double)samplerate;
}
info->set_length(m_length);
return 1;
}
input_aac()
{
m_head = NULL;
m_tail = NULL;
m_samplerate = 0;
hDecoder = NULL;
m_aac_buffer = NULL;
m_samples = 0;
m_samplepos = 0;
m_seekskip = 0;
m_eof = false;
}
~input_aac()
{
struct seek_list *target = m_head;
if (hDecoder)
faacDecClose(hDecoder);
if (m_aac_buffer)
free(m_aac_buffer);
while (target)
{
struct seek_list *tmp = target;
target = target->next;
if (tmp) free(tmp);
}
}
virtual int run(audio_chunk * chunk)
{
while (1) {
if (m_eof || (m_samples > 0 && m_samplepos >= m_samples)) return 0; // gapless playback
if (m_aac_bytes_into_buffer == 0) return 0;
faacDecFrameInfo frameInfo;
audio_sample *sample_buffer = 0;
memset(&frameInfo, 0, sizeof(faacDecFrameInfo));
do
{
fill_buffer();
if (m_aac_bytes_into_buffer != 0)
{
sample_buffer = (audio_sample*)faacDecDecode(hDecoder, &frameInfo,
m_aac_buffer, m_aac_bytes_into_buffer);
if (m_header_type != 1)
{
m_tail->offset = m_file_offset;
m_tail->next = (struct seek_list*)malloc(sizeof(struct seek_list));
m_tail = m_tail->next;
m_tail->next = NULL;
}
advance_buffer(frameInfo.bytesconsumed);
}
if (frameInfo.error || !sample_buffer)
{
if (!frameInfo.error) return 0; // EOF
const char *msg = faacDecGetErrorMessage(frameInfo.error);
if (msg) console::error(msg);
return 0; //-1;
}
if (m_aac_bytes_into_buffer == 0) break;
} while (!frameInfo.samples || !frameInfo.channels);
if (!frameInfo.samples || !frameInfo.channels) return 0;
unsigned int samples = frameInfo.samples/frameInfo.channels;
m_samplerate = frameInfo.samplerate;
m_framesize = samples;
if (m_samples > 0) { // gapless playback
if (m_samplepos + samples > m_samples) samples = (unsigned int)(m_samples - m_samplepos);
}
m_samplepos += samples;
if ((unsigned)m_seekskip < samples) {
if (frameInfo.channels == 6 && frameInfo.num_lfe_channels)
{
//channel order for 5.1: L/R/C/LF/BL/BR
audio_sample r1, r2, r3, r4, r5, r6;
for (unsigned int i = 0; i < frameInfo.samples; i += frameInfo.channels)
{
r1 = sample_buffer[i];
r2 = sample_buffer[i+1];
r3 = sample_buffer[i+2];
r4 = sample_buffer[i+3];
r5 = sample_buffer[i+4];
r6 = sample_buffer[i+5];
sample_buffer[i] = r2;
sample_buffer[i+1] = r3;
sample_buffer[i+2] = r1;
sample_buffer[i+3] = r6;
sample_buffer[i+4] = r4;
sample_buffer[i+5] = r5;
}
}
samples -= m_seekskip;
if (chunk)
{
chunk->set_data((audio_sample*)sample_buffer + m_seekskip*frameInfo.channels,
samples, frameInfo.channels, frameInfo.samplerate);
}
m_seekskip = 0;
break;
} else {
m_seekskip -= samples;
}
}
return 1;
}
virtual set_info_t set_info(reader *r,const file_info *_info)
{
file_info_i_full info;
info.copy(_info);
const char *p = info.info_get("samples");
if (p) {
info.meta_add("samples", p);
info.info_remove_field("samples");
}
return tag_writer::g_run(r,&info,"ape") ? SET_INFO_SUCCESS : SET_INFO_FAILURE;
}
virtual bool seek(double seconds)
{
unsigned int i, frames;
int bread;
struct seek_list *target = m_head;
if (seconds >= m_length) {
m_eof = true;
return true;
}
double cur_pos_sec = (double)(__int64)m_samplepos / (double)(__int64)m_samplerate;
if (m_reader->can_seek() && ((m_header_type == 1) || (seconds < cur_pos_sec)))
{
frames = (unsigned int)(seconds*((double)m_samplerate/(double)m_framesize));
if (frames > 1) frames--;
for (i = 0; i < frames; i++)
{
if (target->next)
target = target->next;
else
return false;
}
if (target->offset == 0 && frames > 0)
return false;
m_file_offset = target->offset;
m_reader->seek(m_file_offset);
bread = m_reader->read(m_aac_buffer, 768*6);
if (bread != 768*6)
m_at_eof = 1;
else
m_at_eof = 0;
m_aac_bytes_into_buffer = bread;
m_aac_bytes_consumed = 0;
m_file_offset += bread;
m_samplepos =(frames > 1) ? (unsigned __int64)(frames-1) * m_framesize : 0;
m_seekskip = (int)((unsigned __int64)(seconds * m_samplerate + 0.5) - m_samplepos);// + m_framesize;
if (m_seekskip < 0) return false; // should never happen
faacDecPostSeekReset(hDecoder, -1);
faacDecFrameInfo frameInfo;
memset(&frameInfo, 0, sizeof(faacDecFrameInfo));
fill_buffer();
faacDecDecode(hDecoder, &frameInfo, m_aac_buffer, m_aac_bytes_into_buffer);
return true;
} else {
if (seconds > cur_pos_sec)
{
frames = (unsigned int)((seconds - cur_pos_sec)*((double)m_samplerate/(double)m_framesize));
if (frames > 0)
{
for (i = 0; i < frames; i++)
{
if (!run(NULL))
return false;
}
}
m_seekskip = (int)((unsigned __int64)(seconds * m_samplerate + 0.5) - m_samplepos);
if (m_seekskip < 0) return false; // should never happen
faacDecPostSeekReset(hDecoder, -1);
}
return true;
}
return false;
}
virtual bool is_our_content_type(const char *url, const char *type)
{
return !stricmp(type, "audio/aac") || !stricmp(type, "audio/x-aac");
}
private:
reader *m_reader;
faacDecHandle hDecoder;
long m_aac_bytes_into_buffer;
long m_aac_bytes_consumed;
__int64 m_file_offset;
unsigned char *m_aac_buffer;
int m_at_eof;
unsigned long m_samplerate;
int m_header_type;
struct seek_list *m_head;
struct seek_list *m_tail;
unsigned __int64 m_samples;
unsigned int m_framesize;
unsigned __int64 m_samplepos;
int m_seekskip;
double m_length;
bool m_eof;
int fill_buffer()
{
int bread;
if (m_aac_bytes_consumed > 0)
{
if (m_aac_bytes_into_buffer)
{
memmove((void*)m_aac_buffer, (void*)(m_aac_buffer + m_aac_bytes_consumed),
m_aac_bytes_into_buffer*sizeof(unsigned char));
}
if (!m_at_eof)
{
bread = m_reader->read((void*)(m_aac_buffer + m_aac_bytes_into_buffer),
m_aac_bytes_consumed);
if (bread != m_aac_bytes_consumed)
m_at_eof = 1;
m_aac_bytes_into_buffer += bread;
}
m_aac_bytes_consumed = 0;
if (m_aac_bytes_into_buffer > 3)
{
if (memcmp(m_aac_buffer, "TAG", 3) == 0)
m_aac_bytes_into_buffer = 0;
}
if (m_aac_bytes_into_buffer > 11)
{
if (memcmp(m_aac_buffer, "LYRICSBEGIN", 11) == 0)
m_aac_bytes_into_buffer = 0;
}
if (m_aac_bytes_into_buffer > 8)
{
if (memcmp(m_aac_buffer, "APETAGEX", 8) == 0)
m_aac_bytes_into_buffer = 0;
}
}
return 1;
}
void advance_buffer(int bytes)
{
m_file_offset += bytes;
m_aac_bytes_consumed = bytes;
m_aac_bytes_into_buffer -= bytes;
}
int adts_parse(__int64 *bitrate, double *length)
{
static int sample_rates[] = {96000,88200,64000,48000,44100,32000,24000,22050,16000,12000,11025,8000};
int frames, frame_length;
int t_framelength = 0;
int samplerate;
double frames_per_sec, bytes_per_frame;
/* Read all frames to ensure correct time and bitrate */
for (frames = 0; /* */; frames++)
{
fill_buffer();
if (m_aac_bytes_into_buffer > 7)
{
/* check syncword */
if (!((m_aac_buffer[0] == 0xFF)&&((m_aac_buffer[1] & 0xF6) == 0xF0)))
break;
m_tail->offset = m_file_offset;
m_tail->next = (struct seek_list*)malloc(sizeof(struct seek_list));
m_tail = m_tail->next;
m_tail->next = NULL;
if (frames == 0)
samplerate = sample_rates[(m_aac_buffer[2]&0x3c)>>2];
frame_length = ((((unsigned int)m_aac_buffer[3] & 0x3)) << 11)
| (((unsigned int)m_aac_buffer[4]) << 3) | (m_aac_buffer[5] >> 5);
t_framelength += frame_length;
if (frame_length > m_aac_bytes_into_buffer)
break;
advance_buffer(frame_length);
} else {
break;
}
}
frames_per_sec = (double)samplerate/1024.0;
if (frames != 0)
bytes_per_frame = (double)t_framelength/(double)(frames*1000);
else
bytes_per_frame = 0;
*bitrate = (__int64)(8. * bytes_per_frame * frames_per_sec + 0.5);
if (frames_per_sec != 0)
*length = (double)frames/frames_per_sec;
else
*length = 1;
return 1;
}
};
static service_factory_t<input,input_mp4> foo_mp4;
static service_factory_t<input,input_aac> foo_aac;