/***************************************************************************
* Copyright (C) 2011-2021 by Ilya Kotov *
* forkotov02@ya.ru *
* *
* 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., *
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
***************************************************************************/
#include <QObject>
#include <QFile>
#include <QRegularExpression>
#include <qmmp/cueparser.h>
#include <qmmp/decoderfactory.h>
#include "decoder_ffmpeg.h"
#include "decoder_ffmpegm4b.h"
DecoderFFmpegM4b::DecoderFFmpegM4b(DecoderFactory *factory, const QString &url) : Decoder(),
m_url(url), m_factory(factory)
{}
DecoderFFmpegM4b::~DecoderFFmpegM4b()
{
if(m_decoder)
delete m_decoder;
m_decoder = nullptr;
if(m_buf)
delete [] m_buf;
m_buf = nullptr;
if(m_input)
m_input->deleteLater();
m_input = nullptr;
for(ChapterInfo &ch : m_chapters)
{
delete ch.info;
ch.info = nullptr;
}
}
bool DecoderFFmpegM4b::initialize()
{
QString filePath = m_url;
if(!m_url.startsWith("m4b://"))
{
qWarning("DecoderFFmpegM4b: invalid url.");
return false;
}
filePath.remove("m4b://");
filePath.remove(QRegularExpression("#\\d+$"));
m_track = m_url.section("#", -1).toInt();
AVFormatContext *in = nullptr;
#ifdef Q_OS_WIN
if(avformat_open_input(&in, filePath.toUtf8().constData(), nullptr, nullptr) < 0)
#else
if(avformat_open_input(&in, filePath.toLocal8Bit().constData(), nullptr, nullptr) < 0)
#endif
{
qDebug("DecoderFFmpegM4b: unable to open file");
return false;
}
avformat_find_stream_info(in, nullptr);
if(in->nb_chapters <= 1)
{
avformat_close_input(&in);
qWarning("DecoderFFmpegM4b: unable to find chapters");
return false;
}
if(m_track > int(in->nb_chapters) || m_track < 1)
{
avformat_close_input(&in);
qWarning("DecoderFFmpegM4b: invalid track number");
return false;
}
QList<TrackInfo *> tracks = m_factory->createPlayList(filePath, TrackInfo::AllParts, nullptr);
if(tracks.isEmpty() || tracks.count() != int(in->nb_chapters))
{
qDeleteAll(tracks);
avformat_close_input(&in);
qWarning("DecoderFFmpegM4b: unable to find tracks");
return false;
}
for(int i = 0; i < tracks.count(); ++i)
{
AVChapter *chapter = in->chapters[i];
ChapterInfo chapterInfo = {
.info = tracks[i],
.offset = chapter->start * chapter->time_base.num * 1000 / chapter->time_base.den,
.duration = (chapter->end - chapter->start) * chapter->time_base.num * 1000 / chapter->time_base.den,
.url = QString("m4b://%1#%2").arg(filePath).arg(i + 1)
};
m_chapters << chapterInfo;
}
tracks.clear();
avformat_close_input(&in);
m_input = new QFile(filePath);
if(!m_input->open(QIODevice::ReadOnly))
{
qWarning("DecoderFFmpegM4b: unable to open file; error: %s", qPrintable(m_input->errorString()));
return false;
}
m_duration = m_chapters[m_track - 1].duration;
m_offset = m_chapters[m_track - 1].offset;
m_decoder = new DecoderFFmpeg(filePath, m_input);
if(!m_decoder->initialize())
{
qDeleteAll(tracks);
qWarning("DecoderFFapCUE: invalid audio file");
return false;
}
m_decoder->seek(m_offset);
configure(m_decoder->audioParameters());
m_trackSize = audioParameters().sampleRate() * audioParameters().channels() *
audioParameters().sampleSize() * m_duration / 1000;
m_written = 0;
m_frameSize = audioParameters().sampleSize() * audioParameters().channels();
setReplayGainInfo(m_decoder->replayGainInfo()); //send ReplayGaing info
addMetaData(m_chapters[m_track - 1].info->metaData()); //send metadata
return true;
}
qint64 DecoderFFmpegM4b::totalTime() const
{
return m_decoder ? m_duration : 0;
}
void DecoderFFmpegM4b::seek(qint64 pos)
{
m_decoder->seek(m_offset + pos);
m_written = audioParameters().sampleRate() *
audioParameters().channels() *
audioParameters().sampleSize() * pos/1000;
}
qint64 DecoderFFmpegM4b::read(unsigned char *data, qint64 size)
{
if(m_trackSize - m_written < m_frameSize) //end of cue track
return 0;
qint64 len = 0;
if(m_buf) //read remaining data first
{
len = qMin(m_bufSize, size);
memmove(data, m_buf, len);
if(size >= m_bufSize)
{
delete[] m_buf;
m_buf = nullptr;
m_bufSize = 0;
}
else
memmove(m_buf, m_buf + len, size - len);
}
else
len = m_decoder->read(data, size);
if(len <= 0) //end of file
return 0;
if(len + m_written <= m_trackSize)
{
m_written += len;
return len;
}
qint64 len2 = qMax(qint64(0), m_trackSize - m_written);
len2 = (len2 / m_frameSize) * m_frameSize; //integer number of samples
m_written += len2;
//save data of the next track
if(m_buf)
delete[] m_buf;
m_bufSize = len - len2;
m_buf = new char[m_bufSize];
memmove(m_buf, data + len2, m_bufSize);
return len2;
}
int DecoderFFmpegM4b::bitrate() const
{
return m_decoder->bitrate();
}
const QString DecoderFFmpegM4b::nextURL() const
{
if(m_track + 1 <= m_chapters.count())
return m_chapters[m_track].url;
else
return QString();
}
void DecoderFFmpegM4b::next()
{
if(m_track + 1 <= m_chapters.count())
{
m_track++;
m_duration = m_chapters[m_track - 1].duration;
m_offset = m_chapters[m_track - 1].offset;
m_trackSize = audioParameters().sampleRate() *
audioParameters().channels() *
audioParameters().sampleSize() * m_duration/1000;
addMetaData(m_chapters[m_track - 1].info->metaData());
setReplayGainInfo(m_decoder->replayGainInfo());
m_written = 0;
}
}
