diff options
Diffstat (limited to 'src/plugins/Input/ffmpeg/decoder_ffmpegm4b.cpp')
| -rw-r--r-- | src/plugins/Input/ffmpeg/decoder_ffmpegm4b.cpp | 231 |
1 files changed, 231 insertions, 0 deletions
diff --git a/src/plugins/Input/ffmpeg/decoder_ffmpegm4b.cpp b/src/plugins/Input/ffmpeg/decoder_ffmpegm4b.cpp new file mode 100644 index 000000000..ee3425d72 --- /dev/null +++ b/src/plugins/Input/ffmpeg/decoder_ffmpegm4b.cpp @@ -0,0 +1,231 @@ +/*************************************************************************** + * 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; + } +} |