annotate data/fileio/BQAFileReader.cpp @ 1587:b67f5b6a7978 bqaudiostream

Request metadata from bq audio stream
author Chris Cannam
date Tue, 15 Jan 2019 16:43:23 +0000
parents ff18abb88563
children c9c8c49b22d6
rev   line source
Chris@1583 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@1583 2
Chris@1583 3 /*
Chris@1583 4 Sonic Visualiser
Chris@1583 5 An audio file viewer and annotation editor.
Chris@1583 6 Centre for Digital Music, Queen Mary, University of London.
Chris@1583 7
Chris@1583 8 This program is free software; you can redistribute it and/or
Chris@1583 9 modify it under the terms of the GNU General Public License as
Chris@1583 10 published by the Free Software Foundation; either version 2 of the
Chris@1583 11 License, or (at your option) any later version. See the file
Chris@1583 12 COPYING included with this distribution for more information.
Chris@1583 13 */
Chris@1583 14
Chris@1583 15 #include "BQAFileReader.h"
Chris@1583 16
Chris@1583 17 #include <bqaudiostream/AudioReadStreamFactory.h>
Chris@1583 18 #include <bqaudiostream/AudioReadStream.h>
Chris@1583 19 #include <bqaudiostream/Exceptions.h>
Chris@1583 20
Chris@1583 21 #include "base/Profiler.h"
Chris@1583 22 #include "base/ProgressReporter.h"
Chris@1583 23
Chris@1583 24 #include <QFileInfo>
Chris@1583 25
Chris@1583 26 using namespace std;
Chris@1583 27
Chris@1583 28 BQAFileReader::BQAFileReader(FileSource source,
Chris@1583 29 DecodeMode decodeMode,
Chris@1583 30 CacheMode mode,
Chris@1583 31 sv_samplerate_t targetRate,
Chris@1583 32 bool normalised,
Chris@1583 33 ProgressReporter *reporter) :
Chris@1583 34 CodedAudioFileReader(mode, targetRate, normalised),
Chris@1583 35 m_source(source),
Chris@1583 36 m_path(source.getLocalFilename()),
Chris@1583 37 m_cancelled(false),
Chris@1583 38 m_completion(0),
Chris@1583 39 m_reporter(reporter),
Chris@1583 40 m_decodeThread(0)
Chris@1583 41 {
Chris@1583 42 SVDEBUG << "BQAFileReader: local path: \"" << m_path
Chris@1583 43 << "\", decode mode: " << decodeMode << " ("
Chris@1583 44 << (decodeMode == DecodeAtOnce ? "DecodeAtOnce" : "DecodeThreaded")
Chris@1583 45 << ")" << endl;
Chris@1583 46
Chris@1583 47 m_channelCount = 0;
Chris@1583 48 m_fileRate = 0;
Chris@1583 49
Chris@1583 50 Profiler profiler("BQAFileReader::BQAFileReader");
Chris@1583 51
Chris@1583 52 try {
Chris@1583 53 m_stream = breakfastquay::AudioReadStreamFactory::createReadStream
Chris@1583 54 (m_path.toUtf8().data());
Chris@1583 55 } catch (const std::exception &e) {
Chris@1583 56 m_error = e.what();
Chris@1583 57 m_stream = 0;
Chris@1583 58 return;
Chris@1583 59 }
Chris@1583 60
Chris@1584 61 m_channelCount = int(m_stream->getChannelCount());
Chris@1584 62 m_fileRate = sv_samplerate_t(m_stream->getSampleRate());
Chris@1587 63 m_title = QString::fromUtf8(m_stream->getTrackName().c_str());
Chris@1587 64 m_maker = QString::fromUtf8(m_stream->getArtistName().c_str());
Chris@1583 65
Chris@1583 66 initialiseDecodeCache();
Chris@1583 67
Chris@1583 68 if (decodeMode == DecodeAtOnce) {
Chris@1583 69
Chris@1583 70 if (m_reporter) {
Chris@1583 71 connect(m_reporter, SIGNAL(cancelled()), this, SLOT(cancelled()));
Chris@1583 72 m_reporter->setMessage
Chris@1583 73 (tr("Decoding %1...").arg(QFileInfo(m_path).fileName()));
Chris@1583 74 }
Chris@1583 75
Chris@1583 76 sv_frame_t blockSize = 65536;
Chris@1583 77 floatvec_t block(blockSize * m_channelCount, 0.f);
Chris@1583 78
Chris@1583 79 while (true) {
Chris@1583 80 try {
Chris@1583 81 sv_frame_t retrieved =
Chris@1583 82 m_stream->getInterleavedFrames(blockSize, block.data());
Chris@1583 83
Chris@1583 84 addSamplesToDecodeCache(block.data(), retrieved);
Chris@1583 85
Chris@1583 86 if (retrieved < blockSize) {
Chris@1583 87 break;
Chris@1583 88 }
Chris@1583 89 } catch (const breakfastquay::InvalidFileFormat &f) {
Chris@1583 90 m_error = f.what();
Chris@1583 91 break;
Chris@1583 92 }
Chris@1583 93
Chris@1583 94 if (m_cancelled) break;
Chris@1583 95 }
Chris@1583 96
Chris@1583 97 if (isDecodeCacheInitialised()) finishDecodeCache();
Chris@1583 98 endSerialised();
Chris@1583 99
Chris@1583 100 if (m_reporter) m_reporter->setProgress(100);
Chris@1583 101
Chris@1583 102 delete m_stream;
Chris@1583 103 m_stream = 0;
Chris@1583 104
Chris@1583 105 } else {
Chris@1583 106
Chris@1583 107 if (m_reporter) m_reporter->setProgress(100);
Chris@1583 108
Chris@1583 109 m_decodeThread = new DecodeThread(this);
Chris@1583 110 m_decodeThread->start();
Chris@1583 111 }
Chris@1583 112 }
Chris@1583 113
Chris@1583 114 BQAFileReader::~BQAFileReader()
Chris@1583 115 {
Chris@1583 116 if (m_decodeThread) {
Chris@1583 117 m_cancelled = true;
Chris@1583 118 m_decodeThread->wait();
Chris@1583 119 delete m_decodeThread;
Chris@1583 120 }
Chris@1583 121
Chris@1583 122 delete m_stream;
Chris@1583 123 }
Chris@1583 124
Chris@1583 125 void
Chris@1583 126 BQAFileReader::cancelled()
Chris@1583 127 {
Chris@1583 128 m_cancelled = true;
Chris@1583 129 }
Chris@1583 130
Chris@1583 131 void
Chris@1583 132 BQAFileReader::DecodeThread::run()
Chris@1583 133 {
Chris@1583 134 if (m_reader->m_cacheMode == CacheInTemporaryFile) {
Chris@1583 135 m_reader->startSerialised("BQAFileReader::Decode");
Chris@1583 136 }
Chris@1583 137
Chris@1583 138 sv_frame_t blockSize = 65536;
Chris@1583 139 floatvec_t block(blockSize * m_reader->getChannelCount(), 0.f);
Chris@1583 140
Chris@1583 141 while (true) {
Chris@1583 142 try {
Chris@1583 143 sv_frame_t retrieved =
Chris@1583 144 m_reader->m_stream->getInterleavedFrames
Chris@1583 145 (blockSize, block.data());
Chris@1583 146
Chris@1583 147 m_reader->addSamplesToDecodeCache(block.data(), retrieved);
Chris@1583 148
Chris@1583 149 if (retrieved < blockSize) {
Chris@1583 150 break;
Chris@1583 151 }
Chris@1583 152 } catch (const breakfastquay::InvalidFileFormat &f) {
Chris@1583 153 m_reader->m_error = f.what();
Chris@1583 154 break;
Chris@1583 155 }
Chris@1583 156
Chris@1583 157 if (m_reader->m_cancelled) break;
Chris@1583 158 }
Chris@1583 159
Chris@1583 160 if (m_reader->isDecodeCacheInitialised()) m_reader->finishDecodeCache();
Chris@1583 161 m_reader->m_completion = 100;
Chris@1583 162
Chris@1583 163 m_reader->endSerialised();
Chris@1583 164
Chris@1583 165 delete m_reader->m_stream;
Chris@1583 166 m_reader->m_stream = 0;
Chris@1583 167 }
Chris@1583 168
Chris@1583 169 void
Chris@1583 170 BQAFileReader::getSupportedExtensions(set<QString> &extensions)
Chris@1583 171 {
Chris@1583 172 vector<string> exts =
Chris@1583 173 breakfastquay::AudioReadStreamFactory::getSupportedFileExtensions();
Chris@1583 174 for (auto e: exts) {
Chris@1583 175 extensions.insert(QString::fromUtf8(e.c_str()));
Chris@1583 176 }
Chris@1583 177 }
Chris@1583 178
Chris@1583 179 bool
Chris@1583 180 BQAFileReader::supportsExtension(QString extension)
Chris@1583 181 {
Chris@1583 182 set<QString> extensions;
Chris@1583 183 getSupportedExtensions(extensions);
Chris@1583 184 return (extensions.find(extension.toLower()) != extensions.end());
Chris@1583 185 }
Chris@1583 186
Chris@1583 187 bool
Chris@1584 188 BQAFileReader::supportsContentType(QString)
Chris@1583 189 {
Chris@1583 190 //!!! todo
Chris@1583 191 return false;
Chris@1583 192 }
Chris@1583 193
Chris@1583 194 bool
Chris@1583 195 BQAFileReader::supports(FileSource &source)
Chris@1583 196 {
Chris@1583 197 return (supportsExtension(source.getExtension()) ||
Chris@1583 198 supportsContentType(source.getContentType()));
Chris@1583 199 }
Chris@1583 200