Chris@147: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@147:
Chris@147: /*
Chris@147: Sonic Visualiser
Chris@147: An audio file viewer and annotation editor.
Chris@147: Centre for Digital Music, Queen Mary, University of London.
Chris@202: This file copyright 2006 Chris Cannam and QMUL.
Chris@147:
Chris@147: This program is free software; you can redistribute it and/or
Chris@147: modify it under the terms of the GNU General Public License as
Chris@147: published by the Free Software Foundation; either version 2 of the
Chris@147: License, or (at your option) any later version. See the file
Chris@147: COPYING included with this distribution for more information.
Chris@147: */
Chris@147:
Chris@150: #include "WaveFileModel.h"
Chris@147:
Chris@147: #include "fileio/AudioFileReader.h"
Chris@147: #include "fileio/AudioFileReaderFactory.h"
Chris@147:
Chris@150: #include "system/System.h"
Chris@147:
Chris@147: #include <QFileInfo>
Chris@314: #include <QTextStream>
Chris@147:
Chris@147: #include <iostream>
Chris@147: #include <unistd.h>
Chris@572: #include <cmath>
Chris@147: #include <sndfile.h>
Chris@147:
Chris@147: #include <cassert>
Chris@147:
Chris@236: //#define DEBUG_WAVE_FILE_MODEL 1
Chris@236:
Chris@179: PowerOfSqrtTwoZoomConstraint
Chris@179: WaveFileModel::m_zoomConstraint;
Chris@179:
Chris@317: WaveFileModel::WaveFileModel(FileSource source, size_t targetRate) :
Chris@316: m_source(source),
Chris@316: m_path(source.getLocation()),
Chris@175: m_myReader(true),
Chris@300: m_startFrame(0),
Chris@147: m_fillThread(0),
Chris@147: m_updateTimer(0),
Chris@147: m_lastFillExtent(0),
Chris@752: m_exiting(false),
Chris@752: m_lastDirectReadStart(0),
Chris@752: m_lastDirectReadCount(0)
Chris@147: {
Chris@316: m_source.waitForData();
Chris@316: if (m_source.isOK()) {
Chris@327: m_reader = AudioFileReaderFactory::createThreadingReader
Chris@327: (m_source, targetRate);
Chris@316: if (m_reader) {
Chris@690: SVDEBUG << "WaveFileModel::WaveFileModel: reader rate: "
Chris@687: << m_reader->getSampleRate() << endl;
Chris@316: }
Chris@316: }
Chris@292: if (m_reader) setObjectName(m_reader->getTitle());
Chris@316: if (objectName() == "") setObjectName(QFileInfo(m_path).fileName());
Chris@175: if (isOK()) fillCache();
Chris@175: }
Chris@175:
Chris@317: WaveFileModel::WaveFileModel(FileSource source, AudioFileReader *reader) :
Chris@316: m_source(source),
Chris@316: m_path(source.getLocation()),
Chris@175: m_myReader(false),
Chris@300: m_startFrame(0),
Chris@175: m_fillThread(0),
Chris@175: m_updateTimer(0),
Chris@175: m_lastFillExtent(0),
Chris@175: m_exiting(false)
Chris@175: {
Chris@175: m_reader = reader;
Chris@292: if (m_reader) setObjectName(m_reader->getTitle());
Chris@316: if (objectName() == "") setObjectName(QFileInfo(m_path).fileName());
Chris@187: fillCache();
Chris@147: }
Chris@147:
Chris@147: WaveFileModel::~WaveFileModel()
Chris@147: {
Chris@147: m_exiting = true;
Chris@147: if (m_fillThread) m_fillThread->wait();
Chris@175: if (m_myReader) delete m_reader;
Chris@147: m_reader = 0;
Chris@147: }
Chris@147:
Chris@147: bool
Chris@147: WaveFileModel::isOK() const
Chris@147: {
Chris@147: return m_reader && m_reader->isOK();
Chris@147: }
Chris@147:
Chris@147: bool
Chris@147: WaveFileModel::isReady(int *completion) const
Chris@147: {
Chris@147: bool ready = (isOK() && (m_fillThread == 0));
Chris@147: double c = double(m_lastFillExtent) / double(getEndFrame() - getStartFrame());
Chris@266: static int prevCompletion = 0;
Chris@265: if (completion) {
Chris@265: *completion = int(c * 100.0 + 0.01);
Chris@265: if (m_reader) {
Chris@265: int decodeCompletion = m_reader->getDecodeCompletion();
Chris@266: if (decodeCompletion < 90) *completion = decodeCompletion;
Chris@266: else *completion = std::min(*completion, decodeCompletion);
Chris@265: }
Chris@266: if (*completion != 0 &&
Chris@266: *completion != 100 &&
Chris@266: prevCompletion != 0 &&
Chris@266: prevCompletion > *completion) {
Chris@266: // just to avoid completion going backwards
Chris@266: *completion = prevCompletion;
Chris@266: }
Chris@266: prevCompletion = *completion;
Chris@265: }
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@690: SVDEBUG << "WaveFileModel::isReady(): ready = " << ready << ", completion = " << (completion ? *completion : -1) << endl;
Chris@236: #endif
Chris@147: return ready;
Chris@147: }
Chris@147:
Chris@147: Model *
Chris@147: WaveFileModel::clone() const
Chris@147: {
Chris@316: WaveFileModel *model = new WaveFileModel(m_source);
Chris@147: return model;
Chris@147: }
Chris@147:
Chris@147: size_t
Chris@147: WaveFileModel::getFrameCount() const
Chris@147: {
Chris@147: if (!m_reader) return 0;
Chris@147: return m_reader->getFrameCount();
Chris@147: }
Chris@147:
Chris@147: size_t
Chris@147: WaveFileModel::getChannelCount() const
Chris@147: {
Chris@147: if (!m_reader) return 0;
Chris@147: return m_reader->getChannelCount();
Chris@147: }
Chris@147:
Chris@147: size_t
Chris@147: WaveFileModel::getSampleRate() const
Chris@147: {
Chris@147: if (!m_reader) return 0;
Chris@147: return m_reader->getSampleRate();
Chris@147: }
Chris@147:
Chris@147: size_t
Chris@297: WaveFileModel::getNativeRate() const
Chris@297: {
Chris@297: if (!m_reader) return 0;
Chris@297: size_t rate = m_reader->getNativeRate();
Chris@297: if (rate == 0) rate = getSampleRate();
Chris@297: return rate;
Chris@297: }
Chris@297:
Chris@333: QString
Chris@333: WaveFileModel::getTitle() const
Chris@333: {
Chris@333: QString title;
Chris@333: if (m_reader) title = m_reader->getTitle();
Chris@333: if (title == "") title = objectName();
Chris@333: return title;
Chris@333: }
Chris@333:
Chris@333: QString
Chris@333: WaveFileModel::getMaker() const
Chris@333: {
Chris@333: if (m_reader) return m_reader->getMaker();
Chris@333: return "";
Chris@333: }
Chris@345:
Chris@345: QString
Chris@345: WaveFileModel::getLocation() const
Chris@345: {
Chris@345: if (m_reader) return m_reader->getLocation();
Chris@345: return "";
Chris@345: }
Chris@333:
Chris@297: size_t
Chris@300: WaveFileModel::getData(int channel, size_t start, size_t count,
Chris@300: float *buffer) const
Chris@147: {
Chris@147: // Always read these directly from the file.
Chris@147: // This is used for e.g. audio playback.
Chris@147: // Could be much more efficient (although compiler optimisation will help)
Chris@147:
Chris@429: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@843: cout << "WaveFileModel::getData[" << this << "]: " << channel << ", " << start << ", " << count << ", " << buffer << endl;
Chris@429: #endif
Chris@429:
Chris@363: if (start >= m_startFrame) {
Chris@300: start -= m_startFrame;
Chris@300: } else {
Chris@300: for (size_t i = 0; i < count; ++i) buffer[i] = 0.f;
Chris@300: if (count <= m_startFrame - start) {
Chris@300: return 0;
Chris@300: } else {
Chris@300: count -= (m_startFrame - start);
Chris@300: start = 0;
Chris@300: }
Chris@147: }
Chris@147:
Chris@300: if (!m_reader || !m_reader->isOK() || count == 0) {
Chris@300: for (size_t i = 0; i < count; ++i) buffer[i] = 0.f;
Chris@300: return 0;
Chris@300: }
Chris@147:
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@690: // SVDEBUG << "WaveFileModel::getValues(" << channel << ", "
Chris@687: // << start << ", " << end << "): calling reader" << endl;
Chris@236: #endif
Chris@175:
Chris@363: int channels = getChannelCount();
Chris@363:
Chris@363: SampleBlock frames(count * channels);
Chris@300: m_reader->getInterleavedFrames(start, count, frames);
Chris@147:
Chris@147: size_t i = 0;
Chris@147:
Chris@363: int ch0 = channel, ch1 = channel;
Chris@147: if (channel == -1) {
Chris@147: ch0 = 0;
Chris@147: ch1 = channels - 1;
Chris@147: }
Chris@147:
Chris@300: while (i < count) {
Chris@147:
Chris@147: buffer[i] = 0.0;
Chris@147:
Chris@147: for (int ch = ch0; ch <= ch1; ++ch) {
Chris@147:
Chris@147: size_t index = i * channels + ch;
Chris@147: if (index >= frames.size()) break;
Chris@147:
Chris@147: float sample = frames[index];
Chris@147: buffer[i] += sample;
Chris@147: }
Chris@147:
Chris@147: ++i;
Chris@147: }
Chris@147:
Chris@147: return i;
Chris@147: }
Chris@147:
Chris@147: size_t
Chris@300: WaveFileModel::getData(int channel, size_t start, size_t count,
Chris@300: double *buffer) const
Chris@147: {
Chris@429: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@843: cout << "WaveFileModel::getData(double)[" << this << "]: " << channel << ", " << start << ", " << count << ", " << buffer << endl;
Chris@429: #endif
Chris@429:
Chris@300: if (start > m_startFrame) {
Chris@300: start -= m_startFrame;
Chris@300: } else {
Chris@300: for (size_t i = 0; i < count; ++i) buffer[i] = 0.0;
Chris@300: if (count <= m_startFrame - start) {
Chris@300: return 0;
Chris@300: } else {
Chris@300: count -= (m_startFrame - start);
Chris@300: start = 0;
Chris@300: }
Chris@147: }
Chris@147:
Chris@300: if (!m_reader || !m_reader->isOK() || count == 0) {
Chris@300: for (size_t i = 0; i < count; ++i) buffer[i] = 0.0;
Chris@300: return 0;
Chris@300: }
Chris@147:
Chris@363: int channels = getChannelCount();
Chris@363:
Chris@363: SampleBlock frames(count * channels);
Chris@300: m_reader->getInterleavedFrames(start, count, frames);
Chris@147:
Chris@147: size_t i = 0;
Chris@147:
Chris@363: int ch0 = channel, ch1 = channel;
Chris@147: if (channel == -1) {
Chris@147: ch0 = 0;
Chris@147: ch1 = channels - 1;
Chris@147: }
Chris@147:
Chris@300: while (i < count) {
Chris@147:
Chris@147: buffer[i] = 0.0;
Chris@147:
Chris@147: for (int ch = ch0; ch <= ch1; ++ch) {
Chris@147:
Chris@147: size_t index = i * channels + ch;
Chris@147: if (index >= frames.size()) break;
Chris@147:
Chris@147: float sample = frames[index];
Chris@147: buffer[i] += sample;
Chris@147: }
Chris@147:
Chris@147: ++i;
Chris@147: }
Chris@147:
Chris@147: return i;
Chris@147: }
Chris@147:
Chris@363: size_t
Chris@363: WaveFileModel::getData(size_t fromchannel, size_t tochannel,
Chris@363: size_t start, size_t count,
Chris@363: float **buffer) const
Chris@363: {
Chris@429: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@843: cout << "WaveFileModel::getData[" << this << "]: " << fromchannel << "," << tochannel << ", " << start << ", " << count << ", " << buffer << endl;
Chris@429: #endif
Chris@429:
Chris@363: size_t channels = getChannelCount();
Chris@363:
Chris@363: if (fromchannel > tochannel) {
Chris@843: cerr << "ERROR: WaveFileModel::getData: fromchannel ("
Chris@363: << fromchannel << ") > tochannel (" << tochannel << ")"
Chris@843: << endl;
Chris@363: return 0;
Chris@363: }
Chris@363:
Chris@363: if (tochannel >= channels) {
Chris@843: cerr << "ERROR: WaveFileModel::getData: tochannel ("
Chris@363: << tochannel << ") >= channel count (" << channels << ")"
Chris@843: << endl;
Chris@363: return 0;
Chris@363: }
Chris@363:
Chris@363: if (fromchannel == tochannel) {
Chris@363: return getData(fromchannel, start, count, buffer[0]);
Chris@363: }
Chris@363:
Chris@363: size_t reqchannels = (tochannel - fromchannel) + 1;
Chris@363:
Chris@363: // Always read these directly from the file.
Chris@363: // This is used for e.g. audio playback.
Chris@363: // Could be much more efficient (although compiler optimisation will help)
Chris@363:
Chris@363: if (start >= m_startFrame) {
Chris@363: start -= m_startFrame;
Chris@363: } else {
Chris@363: for (size_t c = 0; c < reqchannels; ++c) {
Chris@363: for (size_t i = 0; i < count; ++i) buffer[c][i] = 0.f;
Chris@363: }
Chris@363: if (count <= m_startFrame - start) {
Chris@363: return 0;
Chris@363: } else {
Chris@363: count -= (m_startFrame - start);
Chris@363: start = 0;
Chris@363: }
Chris@363: }
Chris@363:
Chris@363: if (!m_reader || !m_reader->isOK() || count == 0) {
Chris@363: for (size_t c = 0; c < reqchannels; ++c) {
Chris@363: for (size_t i = 0; i < count; ++i) buffer[c][i] = 0.f;
Chris@363: }
Chris@363: return 0;
Chris@363: }
Chris@363:
Chris@363: SampleBlock frames(count * channels);
Chris@363: m_reader->getInterleavedFrames(start, count, frames);
Chris@363:
Chris@363: size_t i = 0;
Chris@363:
Chris@363: int ch0 = fromchannel, ch1 = tochannel;
Chris@363:
Chris@363: size_t index = 0, available = frames.size();
Chris@363:
Chris@363: while (i < count) {
Chris@363:
Chris@363: if (index >= available) break;
Chris@363:
Chris@363: size_t destc = 0;
Chris@363:
Chris@363: for (size_t c = 0; c < channels; ++c) {
Chris@363:
Chris@363: if (c >= fromchannel && c <= tochannel) {
Chris@363: buffer[destc][i] = frames[index];
Chris@363: ++destc;
Chris@363: }
Chris@363:
Chris@363: ++index;
Chris@363: }
Chris@363:
Chris@363: ++i;
Chris@363: }
Chris@363:
Chris@363: return i;
Chris@363: }
Chris@363:
Chris@377: size_t
Chris@377: WaveFileModel::getSummaryBlockSize(size_t desired) const
Chris@377: {
Chris@377: int cacheType = 0;
Chris@377: int power = m_zoomConstraint.getMinCachePower();
Chris@377: size_t roundedBlockSize = m_zoomConstraint.getNearestBlockSize
Chris@377: (desired, cacheType, power, ZoomConstraint::RoundDown);
Chris@377: if (cacheType != 0 && cacheType != 1) {
Chris@377: // We will be reading directly from file, so can satisfy any
Chris@377: // blocksize requirement
Chris@377: return desired;
Chris@377: } else {
Chris@377: return roundedBlockSize;
Chris@377: }
Chris@377: }
Chris@377:
Chris@225: void
Chris@300: WaveFileModel::getSummaries(size_t channel, size_t start, size_t count,
Chris@300: RangeBlock &ranges, size_t &blockSize) const
Chris@147: {
Chris@225: ranges.clear();
Chris@225: if (!isOK()) return;
Chris@377: ranges.reserve((count / blockSize) + 1);
Chris@147:
Chris@300: if (start > m_startFrame) start -= m_startFrame;
Chris@300: else if (count <= m_startFrame - start) return;
Chris@300: else {
Chris@300: count -= (m_startFrame - start);
Chris@300: start = 0;
Chris@147: }
Chris@147:
Chris@147: int cacheType = 0;
Chris@179: int power = m_zoomConstraint.getMinCachePower();
Chris@377: size_t roundedBlockSize = m_zoomConstraint.getNearestBlockSize
Chris@377: (blockSize, cacheType, power, ZoomConstraint::RoundDown);
Chris@147:
Chris@147: size_t channels = getChannelCount();
Chris@147:
Chris@147: if (cacheType != 0 && cacheType != 1) {
Chris@147:
Chris@147: // We need to read directly from the file. We haven't got
Chris@147: // this cached. Hope the requested area is small. This is
Chris@147: // not optimal -- we'll end up reading the same frames twice
Chris@147: // for stereo files, in two separate calls to this method.
Chris@147: // We could fairly trivially handle this for most cases that
Chris@147: // matter by putting a single cache in getInterleavedFrames
Chris@147: // for short queries.
Chris@147:
Chris@377: m_directReadMutex.lock();
Chris@377:
Chris@377: if (m_lastDirectReadStart != start ||
Chris@377: m_lastDirectReadCount != count ||
Chris@377: m_directRead.empty()) {
Chris@377:
Chris@377: m_reader->getInterleavedFrames(start, count, m_directRead);
Chris@377: m_lastDirectReadStart = start;
Chris@377: m_lastDirectReadCount = count;
Chris@377: }
Chris@377:
Chris@147: float max = 0.0, min = 0.0, total = 0.0;
Chris@300: size_t i = 0, got = 0;
Chris@147:
Chris@300: while (i < count) {
Chris@147:
Chris@147: size_t index = i * channels + channel;
Chris@377: if (index >= m_directRead.size()) break;
Chris@147:
Chris@377: float sample = m_directRead[index];
Chris@300: if (sample > max || got == 0) max = sample;
Chris@300: if (sample < min || got == 0) min = sample;
Chris@147: total += fabsf(sample);
Chris@838:
Chris@147: ++i;
Chris@300: ++got;
Chris@147:
Chris@300: if (got == blockSize) {
Chris@300: ranges.push_back(Range(min, max, total / got));
Chris@147: min = max = total = 0.0f;
Chris@300: got = 0;
Chris@147: }
Chris@147: }
Chris@147:
Chris@377: m_directReadMutex.unlock();
Chris@377:
Chris@300: if (got > 0) {
Chris@300: ranges.push_back(Range(min, max, total / got));
Chris@147: }
Chris@147:
Chris@225: return;
Chris@147:
Chris@147: } else {
Chris@147:
Chris@147: QMutexLocker locker(&m_mutex);
Chris@147:
Chris@147: const RangeBlock &cache = m_cache[cacheType];
Chris@147:
Chris@377: blockSize = roundedBlockSize;
Chris@377:
Chris@147: size_t cacheBlock, div;
Chris@147:
Chris@147: if (cacheType == 0) {
Chris@179: cacheBlock = (1 << m_zoomConstraint.getMinCachePower());
Chris@147: div = (1 << power) / cacheBlock;
Chris@147: } else {
Chris@608: cacheBlock = ((unsigned int)((1 << m_zoomConstraint.getMinCachePower()) * sqrt(2.) + 0.01));
Chris@608: div = ((unsigned int)((1 << power) * sqrt(2.) + 0.01)) / cacheBlock;
Chris@147: }
Chris@147:
Chris@147: size_t startIndex = start / cacheBlock;
Chris@300: size_t endIndex = (start + count) / cacheBlock;
Chris@147:
Chris@147: float max = 0.0, min = 0.0, total = 0.0;
Chris@300: size_t i = 0, got = 0;
Chris@147:
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@300: cerr << "blockSize is " << blockSize << ", cacheBlock " << cacheBlock << ", start " << start << ", count " << count << " (frame count " << getFrameCount() << "), power is " << power << ", div is " << div << ", startIndex " << startIndex << ", endIndex " << endIndex << endl;
Chris@236: #endif
Chris@147:
Chris@300: for (i = 0; i <= endIndex - startIndex; ) {
Chris@147:
Chris@147: size_t index = (i + startIndex) * channels + channel;
Chris@147: if (index >= cache.size()) break;
Chris@147:
Chris@147: const Range &range = cache[index];
Chris@410: if (range.max() > max || got == 0) max = range.max();
Chris@410: if (range.min() < min || got == 0) min = range.min();
Chris@410: total += range.absmean();
Chris@147:
Chris@147: ++i;
Chris@300: ++got;
Chris@147:
Chris@300: if (got == div) {
Chris@300: ranges.push_back(Range(min, max, total / got));
Chris@147: min = max = total = 0.0f;
Chris@300: got = 0;
Chris@147: }
Chris@147: }
Chris@147:
Chris@300: if (got > 0) {
Chris@300: ranges.push_back(Range(min, max, total / got));
Chris@147: }
Chris@147: }
Chris@147:
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@690: SVDEBUG << "returning " << ranges.size() << " ranges" << endl;
Chris@236: #endif
Chris@225: return;
Chris@147: }
Chris@147:
Chris@147: WaveFileModel::Range
Chris@300: WaveFileModel::getSummary(size_t channel, size_t start, size_t count) const
Chris@147: {
Chris@147: Range range;
Chris@147: if (!isOK()) return range;
Chris@147:
Chris@300: if (start > m_startFrame) start -= m_startFrame;
Chris@300: else if (count <= m_startFrame - start) return range;
Chris@300: else {
Chris@300: count -= (m_startFrame - start);
Chris@300: start = 0;
Chris@147: }
Chris@147:
Chris@147: size_t blockSize;
Chris@300: for (blockSize = 1; blockSize <= count; blockSize *= 2);
Chris@300: if (blockSize > 1) blockSize /= 2;
Chris@147:
Chris@147: bool first = false;
Chris@147:
Chris@147: size_t blockStart = (start / blockSize) * blockSize;
Chris@300: size_t blockEnd = ((start + count) / blockSize) * blockSize;
Chris@147:
Chris@147: if (blockStart < start) blockStart += blockSize;
Chris@147:
Chris@147: if (blockEnd > blockStart) {
Chris@225: RangeBlock ranges;
Chris@300: getSummaries(channel, blockStart, blockEnd - blockStart, ranges, blockSize);
Chris@147: for (size_t i = 0; i < ranges.size(); ++i) {
Chris@410: if (first || ranges[i].min() < range.min()) range.setMin(ranges[i].min());
Chris@410: if (first || ranges[i].max() > range.max()) range.setMax(ranges[i].max());
Chris@410: if (first || ranges[i].absmean() < range.absmean()) range.setAbsmean(ranges[i].absmean());
Chris@147: first = false;
Chris@147: }
Chris@147: }
Chris@147:
Chris@147: if (blockStart > start) {
Chris@300: Range startRange = getSummary(channel, start, blockStart - start);
Chris@410: range.setMin(std::min(range.min(), startRange.min()));
Chris@410: range.setMax(std::max(range.max(), startRange.max()));
Chris@410: range.setAbsmean(std::min(range.absmean(), startRange.absmean()));
Chris@147: }
Chris@147:
Chris@300: if (blockEnd < start + count) {
Chris@300: Range endRange = getSummary(channel, blockEnd, start + count - blockEnd);
Chris@410: range.setMin(std::min(range.min(), endRange.min()));
Chris@410: range.setMax(std::max(range.max(), endRange.max()));
Chris@410: range.setAbsmean(std::min(range.absmean(), endRange.absmean()));
Chris@147: }
Chris@147:
Chris@147: return range;
Chris@147: }
Chris@147:
Chris@147: void
Chris@147: WaveFileModel::fillCache()
Chris@147: {
Chris@147: m_mutex.lock();
Chris@188:
Chris@147: m_updateTimer = new QTimer(this);
Chris@147: connect(m_updateTimer, SIGNAL(timeout()), this, SLOT(fillTimerTimedOut()));
Chris@147: m_updateTimer->start(100);
Chris@188:
Chris@147: m_fillThread = new RangeCacheFillThread(*this);
Chris@147: connect(m_fillThread, SIGNAL(finished()), this, SLOT(cacheFilled()));
Chris@188:
Chris@147: m_mutex.unlock();
Chris@147: m_fillThread->start();
Chris@188:
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@690: SVDEBUG << "WaveFileModel::fillCache: started fill thread" << endl;
Chris@236: #endif
Chris@147: }
Chris@147:
Chris@147: void
Chris@147: WaveFileModel::fillTimerTimedOut()
Chris@147: {
Chris@147: if (m_fillThread) {
Chris@147: size_t fillExtent = m_fillThread->getFillExtent();
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@690: SVDEBUG << "WaveFileModel::fillTimerTimedOut: extent = " << fillExtent << endl;
Chris@236: #endif
Chris@147: if (fillExtent > m_lastFillExtent) {
Chris@147: emit modelChanged(m_lastFillExtent, fillExtent);
Chris@147: m_lastFillExtent = fillExtent;
Chris@147: }
Chris@147: } else {
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@690: SVDEBUG << "WaveFileModel::fillTimerTimedOut: no thread" << endl;
Chris@236: #endif
Chris@147: emit modelChanged();
Chris@147: }
Chris@147: }
Chris@147:
Chris@147: void
Chris@147: WaveFileModel::cacheFilled()
Chris@147: {
Chris@147: m_mutex.lock();
Chris@147: delete m_fillThread;
Chris@147: m_fillThread = 0;
Chris@147: delete m_updateTimer;
Chris@147: m_updateTimer = 0;
Chris@147: m_mutex.unlock();
Chris@267: if (getEndFrame() > m_lastFillExtent) {
Chris@267: emit modelChanged(m_lastFillExtent, getEndFrame());
Chris@267: }
Chris@147: emit modelChanged();
Chris@411: emit ready();
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@690: SVDEBUG << "WaveFileModel::cacheFilled" << endl;
Chris@236: #endif
Chris@175: }
Chris@175:
Chris@175: void
Chris@147: WaveFileModel::RangeCacheFillThread::run()
Chris@147: {
Chris@147: size_t cacheBlockSize[2];
Chris@179: cacheBlockSize[0] = (1 << m_model.m_zoomConstraint.getMinCachePower());
Chris@179: cacheBlockSize[1] = ((unsigned int)((1 << m_model.m_zoomConstraint.getMinCachePower()) *
Chris@608: sqrt(2.) + 0.01));
Chris@147:
Chris@147: size_t frame = 0;
Chris@411: int readBlockSize = 16384;
Chris@290: SampleBlock block;
Chris@147:
Chris@147: if (!m_model.isOK()) return;
Chris@147:
Chris@147: size_t channels = m_model.getChannelCount();
Chris@187: bool updating = m_model.m_reader->isUpdating();
Chris@187:
Chris@187: if (updating) {
Chris@187: while (channels == 0 && !m_model.m_exiting) {
Chris@690: // SVDEBUG << "WaveFileModel::fill: Waiting for channels..." << endl;
Chris@187: sleep(1);
Chris@187: channels = m_model.getChannelCount();
Chris@187: }
Chris@187: }
Chris@147:
Chris@147: Range *range = new Range[2 * channels];
Chris@410: float *means = new float[2 * channels];
Chris@147: size_t count[2];
Chris@147: count[0] = count[1] = 0;
Chris@411: for (int i = 0; i < 2 * channels; ++i) {
Chris@411: means[i] = 0.f;
Chris@411: }
Chris@176:
Chris@176: bool first = true;
Chris@176:
Chris@176: while (first || updating) {
Chris@176:
Chris@176: updating = m_model.m_reader->isUpdating();
Chris@187: m_frameCount = m_model.getFrameCount();
Chris@175:
Chris@690: // SVDEBUG << "WaveFileModel::fill: frame = " << frame << ", count = " << m_frameCount << endl;
Chris@147:
Chris@176: while (frame < m_frameCount) {
Chris@147:
Chris@690: // SVDEBUG << "WaveFileModel::fill inner loop: frame = " << frame << ", count = " << m_frameCount << ", blocksize " << readBlockSize << endl;
Chris@265:
Chris@176: if (updating && (frame + readBlockSize > m_frameCount)) break;
Chris@176:
Chris@176: m_model.m_reader->getInterleavedFrames(frame, readBlockSize, block);
Chris@176:
Chris@843: // cerr << "block is " << block.size() << endl;
Chris@265:
Chris@411: for (int i = 0; i < readBlockSize; ++i) {
Chris@147:
Chris@232: if (channels * i + channels > block.size()) break;
Chris@232:
Chris@411: for (int ch = 0; ch < channels; ++ch) {
Chris@147:
Chris@411: int index = channels * i + ch;
Chris@176: float sample = block[index];
Chris@176:
Chris@411: for (int ct = 0; ct < 2; ++ct) { // cache type
Chris@176:
Chris@411: int rangeIndex = ch * 2 + ct;
Chris@176:
Chris@410: if (sample > range[rangeIndex].max() || count[ct] == 0) {
Chris@410: range[rangeIndex].setMax(sample);
Chris@176: }
Chris@410: if (sample < range[rangeIndex].min() || count[ct] == 0) {
Chris@410: range[rangeIndex].setMin(sample);
Chris@176: }
Chris@410:
Chris@410: means[rangeIndex] += fabsf(sample);
Chris@176: }
Chris@176: }
Chris@147:
Chris@176: QMutexLocker locker(&m_model.m_mutex);
Chris@232:
Chris@147: for (size_t ct = 0; ct < 2; ++ct) {
Chris@232:
Chris@176: if (++count[ct] == cacheBlockSize[ct]) {
Chris@410:
Chris@176: for (size_t ch = 0; ch < size_t(channels); ++ch) {
Chris@176: size_t rangeIndex = ch * 2 + ct;
Chris@410: means[rangeIndex] /= count[ct];
Chris@410: range[rangeIndex].setAbsmean(means[rangeIndex]);
Chris@176: m_model.m_cache[ct].push_back(range[rangeIndex]);
Chris@176: range[rangeIndex] = Range();
Chris@411: means[rangeIndex] = 0.f;
Chris@176: }
Chris@232:
Chris@176: count[ct] = 0;
Chris@176: }
Chris@176: }
Chris@147:
Chris@176: ++frame;
Chris@147: }
Chris@147:
Chris@176: if (m_model.m_exiting) break;
Chris@176:
Chris@176: m_fillExtent = frame;
Chris@147: }
Chris@147:
Chris@843: // cerr << "WaveFileModel: inner loop ended" << endl;
Chris@265:
Chris@176: first = false;
Chris@177: if (m_model.m_exiting) break;
Chris@187: if (updating) {
Chris@843: // cerr << "sleeping..." << endl;
Chris@187: sleep(1);
Chris@187: }
Chris@147: }
Chris@147:
Chris@177: if (!m_model.m_exiting) {
Chris@177:
Chris@177: QMutexLocker locker(&m_model.m_mutex);
Chris@232:
Chris@177: for (size_t ct = 0; ct < 2; ++ct) {
Chris@232:
Chris@177: if (count[ct] > 0) {
Chris@232:
Chris@177: for (size_t ch = 0; ch < size_t(channels); ++ch) {
Chris@177: size_t rangeIndex = ch * 2 + ct;
Chris@410: means[rangeIndex] /= count[ct];
Chris@410: range[rangeIndex].setAbsmean(means[rangeIndex]);
Chris@177: m_model.m_cache[ct].push_back(range[rangeIndex]);
Chris@177: range[rangeIndex] = Range();
Chris@411: means[rangeIndex] = 0.f;
Chris@177: }
Chris@232:
Chris@177: count[ct] = 0;
Chris@147: }
Chris@177:
Chris@177: const Range &rr = *m_model.m_cache[ct].begin();
Chris@177: MUNLOCK(&rr, m_model.m_cache[ct].capacity() * sizeof(Range));
Chris@147: }
Chris@147: }
Chris@147:
Chris@410: delete[] means;
Chris@147: delete[] range;
Chris@147:
Chris@175: m_fillExtent = m_frameCount;
Chris@236:
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@236: for (size_t ct = 0; ct < 2; ++ct) {
Chris@236: cerr << "Cache type " << ct << " now contains " << m_model.m_cache[ct].size() << " ranges" << endl;
Chris@236: }
Chris@236: #endif
Chris@147: }
Chris@147:
Chris@163: void
Chris@163: WaveFileModel::toXml(QTextStream &out,
Chris@163: QString indent,
Chris@163: QString extraAttributes) const
Chris@163: {
Chris@163: Model::toXml(out, indent,
Chris@163: QString("type=\"wavefile\" file=\"%1\" %2")
Chris@279: .arg(encodeEntities(m_path)).arg(extraAttributes));
Chris@163: }
Chris@163:
Chris@147: