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@1122: #include "ReadOnlyWaveFileModel.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@921: #include "base/Preferences.h"
Chris@921: 
Chris@147: #include <QFileInfo>
Chris@314: #include <QTextStream>
Chris@147: 
Chris@147: #include <iostream>
Chris@1218: //#include <unistd.h>
Chris@572: #include <cmath>
Chris@147: #include <sndfile.h>
Chris@147: 
Chris@147: #include <cassert>
Chris@147: 
Chris@1096: using namespace std;
Chris@1096: 
Chris@236: //#define DEBUG_WAVE_FILE_MODEL 1
Chris@236: 
Chris@179: PowerOfSqrtTwoZoomConstraint
Chris@1122: ReadOnlyWaveFileModel::m_zoomConstraint;
Chris@179: 
Chris@1122: ReadOnlyWaveFileModel::ReadOnlyWaveFileModel(FileSource source, sv_samplerate_t targetRate) :
Chris@316:     m_source(source),
Chris@316:     m_path(source.getLocation()),
Chris@971:     m_reader(0),
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@921:         bool normalise = Preferences::getInstance()->getNormaliseAudio();
Chris@327:         m_reader = AudioFileReaderFactory::createThreadingReader
Chris@921:             (m_source, targetRate, normalise);
Chris@316:         if (m_reader) {
Chris@1122:             SVDEBUG << "ReadOnlyWaveFileModel::ReadOnlyWaveFileModel: 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@1122: ReadOnlyWaveFileModel::ReadOnlyWaveFileModel(FileSource source, AudioFileReader *reader) :
Chris@316:     m_source(source),
Chris@316:     m_path(source.getLocation()),
Chris@971:     m_reader(0),
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@1122: ReadOnlyWaveFileModel::~ReadOnlyWaveFileModel()
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@1122: ReadOnlyWaveFileModel::isOK() const
Chris@147: {
Chris@147:     return m_reader && m_reader->isOK();
Chris@147: }
Chris@147: 
Chris@147: bool
Chris@1122: ReadOnlyWaveFileModel::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@1096:             else *completion = 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@1122:     SVDEBUG << "ReadOnlyWaveFileModel::isReady(): ready = " << ready << ", completion = " << (completion ? *completion : -1) << endl;
Chris@236: #endif
Chris@147:     return ready;
Chris@147: }
Chris@147: 
Chris@1038: sv_frame_t
Chris@1122: ReadOnlyWaveFileModel::getFrameCount() const
Chris@147: {
Chris@147:     if (!m_reader) return 0;
Chris@147:     return m_reader->getFrameCount();
Chris@147: }
Chris@147: 
Chris@929: int
Chris@1122: ReadOnlyWaveFileModel::getChannelCount() const
Chris@147: {
Chris@147:     if (!m_reader) return 0;
Chris@147:     return m_reader->getChannelCount();
Chris@147: }
Chris@147: 
Chris@1040: sv_samplerate_t
Chris@1122: ReadOnlyWaveFileModel::getSampleRate() const 
Chris@147: {
Chris@147:     if (!m_reader) return 0;
Chris@147:     return m_reader->getSampleRate();
Chris@147: }
Chris@147: 
Chris@1040: sv_samplerate_t
Chris@1122: ReadOnlyWaveFileModel::getNativeRate() const 
Chris@297: {
Chris@297:     if (!m_reader) return 0;
Chris@1040:     sv_samplerate_t rate = m_reader->getNativeRate();
Chris@297:     if (rate == 0) rate = getSampleRate();
Chris@297:     return rate;
Chris@297: }
Chris@297: 
Chris@333: QString
Chris@1122: ReadOnlyWaveFileModel::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@1122: ReadOnlyWaveFileModel::getMaker() const
Chris@333: {
Chris@333:     if (m_reader) return m_reader->getMaker();
Chris@333:     return "";
Chris@333: }
Chris@345: 
Chris@345: QString
Chris@1122: ReadOnlyWaveFileModel::getLocation() const
Chris@345: {
Chris@345:     if (m_reader) return m_reader->getLocation();
Chris@345:     return "";
Chris@345: }
Chris@1010: 
Chris@1010: QString
Chris@1122: ReadOnlyWaveFileModel::getLocalFilename() const
Chris@1010: {
Chris@1010:     if (m_reader) return m_reader->getLocalFilename();
Chris@1010:     return "";
Chris@1010: }
Chris@333:     
Chris@1096: vector<float>
Chris@1124: ReadOnlyWaveFileModel::getData(int channel, sv_frame_t start, sv_frame_t count) const
Chris@147: {
Chris@1100:     // Read directly from the file.  This is used for e.g. audio
Chris@1100:     // playback or input to transforms.
Chris@147: 
Chris@429: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@1151:     cout << "ReadOnlyWaveFileModel::getData[" << this << "]: " << channel << ", " << start << ", " << count << endl;
Chris@429: #endif
Chris@429: 
Chris@1100:     int channels = getChannelCount();
Chris@1100: 
Chris@1100:     if (channel >= channels) {
Chris@1100:         cerr << "ERROR: WaveFileModel::getData: channel ("
Chris@1100:              << channel << ") >= channel count (" << channels << ")"
Chris@1100:              << endl;
Chris@1100:         return {};
Chris@1100:     }
Chris@1100: 
Chris@1096:     if (!m_reader || !m_reader->isOK() || count == 0) {
Chris@1096:         return {};
Chris@1096:     }
Chris@1096: 
Chris@363:     if (start >= m_startFrame) {
Chris@300:         start -= m_startFrame;
Chris@300:     } else {
Chris@300:         if (count <= m_startFrame - start) {
Chris@1100:             return {};
Chris@300:         } else {
Chris@300:             count -= (m_startFrame - start);
Chris@300:             start = 0;
Chris@300:         }
Chris@147:     }
Chris@147: 
Chris@1100:     vector<float> interleaved = m_reader->getInterleavedFrames(start, count);
Chris@1100:     if (channels == 1) return interleaved;
Chris@1100: 
Chris@1100:     sv_frame_t obtained = interleaved.size() / channels;
Chris@1100:     
Chris@1100:     vector<float> result(obtained, 0.f);
Chris@1100:     
Chris@1096:     if (channel != -1) {
Chris@1096:         // get a single channel
Chris@1100:         for (int i = 0; i < obtained; ++i) {
Chris@1100:             result[i] = interleaved[i * channels + channel];
Chris@1100:         }
Chris@1100:     } else {
Chris@1100:         // channel == -1, mix down all channels
Chris@1280:         for (int i = 0; i < obtained; ++i) {
Chris@1280:             for (int c = 0; c < channels; ++c) {
Chris@1100:                 result[i] += interleaved[i * channels + c];
Chris@1100:             }
Chris@1086:         }
Chris@300:     }
Chris@147: 
Chris@1096:     return result;
Chris@147: }
Chris@147: 
Chris@1096: vector<vector<float>>
Chris@1122: ReadOnlyWaveFileModel::getMultiChannelData(int fromchannel, int tochannel,
Chris@1124:                                            sv_frame_t start, sv_frame_t count) const
Chris@363: {
Chris@1100:     // Read directly from the file.  This is used for e.g. audio
Chris@1100:     // playback or input to transforms.
Chris@1100: 
Chris@429: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@1151:     cout << "ReadOnlyWaveFileModel::getData[" << this << "]: " << fromchannel << "," << tochannel << ", " << start << ", " << count << endl;
Chris@429: #endif
Chris@429: 
Chris@929:     int channels = getChannelCount();
Chris@363: 
Chris@363:     if (fromchannel > tochannel) {
Chris@1122:         cerr << "ERROR: ReadOnlyWaveFileModel::getData: fromchannel ("
Chris@363:                   << fromchannel << ") > tochannel (" << tochannel << ")"
Chris@843:                   << endl;
Chris@1096:         return {};
Chris@363:     }
Chris@363: 
Chris@363:     if (tochannel >= channels) {
Chris@1122:         cerr << "ERROR: ReadOnlyWaveFileModel::getData: tochannel ("
Chris@363:                   << tochannel << ") >= channel count (" << channels << ")"
Chris@843:                   << endl;
Chris@1096:         return {};
Chris@363:     }
Chris@363: 
Chris@1096:     if (!m_reader || !m_reader->isOK() || count == 0) {
Chris@1096:         return {};
Chris@363:     }
Chris@363: 
Chris@929:     int reqchannels = (tochannel - fromchannel) + 1;
Chris@363: 
Chris@363:     if (start >= m_startFrame) {
Chris@363:         start -= m_startFrame;
Chris@363:     } else {
Chris@363:         if (count <= m_startFrame - start) {
Chris@1096:             return {};
Chris@363:         } else {
Chris@363:             count -= (m_startFrame - start);
Chris@363:             start = 0;
Chris@363:         }
Chris@363:     }
Chris@363: 
Chris@1096:     vector<float> interleaved = m_reader->getInterleavedFrames(start, count);
Chris@1096:     if (channels == 1) return { interleaved };
Chris@1096: 
Chris@1096:     sv_frame_t obtained = interleaved.size() / channels;
Chris@1096:     vector<vector<float>> result(reqchannels, vector<float>(obtained, 0.f));
Chris@1096: 
Chris@1096:     for (int c = fromchannel; c <= tochannel; ++c) {
Chris@1096:         int destc = c - fromchannel;
Chris@1096:         for (int i = 0; i < obtained; ++i) {
Chris@1096:             result[destc][i] = interleaved[i * channels + c];
Chris@363:         }
Chris@363:     }
Chris@1096:     
Chris@1096:     return result;
Chris@363: }
Chris@363: 
Chris@929: int
Chris@1122: ReadOnlyWaveFileModel::getSummaryBlockSize(int desired) const
Chris@377: {
Chris@377:     int cacheType = 0;
Chris@377:     int power = m_zoomConstraint.getMinCachePower();
Chris@929:     int 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@1122: ReadOnlyWaveFileModel::getSummaries(int channel, sv_frame_t start, sv_frame_t count,
Chris@1151:                                     RangeBlock &ranges, int &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@929:     int roundedBlockSize = m_zoomConstraint.getNearestBlockSize
Chris@377:         (blockSize, cacheType, power, ZoomConstraint::RoundDown);
Chris@147: 
Chris@929:     int 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@1041:             m_directRead = m_reader->getInterleavedFrames(start, count);
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@1038: 	sv_frame_t i = 0, got = 0;
Chris@147: 
Chris@300: 	while (i < count) {
Chris@147: 
Chris@1038: 	    sv_frame_t index = i * channels + channel;
Chris@1038: 	    if (index >= (sv_frame_t)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@1038:                 ranges.push_back(Range(min, max, total / float(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@1038:             ranges.push_back(Range(min, max, total / float(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@1038: 	sv_frame_t cacheBlock, div;
Chris@1152: 
Chris@1152:         cacheBlock = (sv_frame_t(1) << m_zoomConstraint.getMinCachePower());
Chris@1152: 	if (cacheType == 1) {
Chris@1152: 	    cacheBlock = sv_frame_t(double(cacheBlock) * sqrt(2.) + 0.01);
Chris@147: 	}
Chris@1152:         div = blockSize / cacheBlock;
Chris@147: 
Chris@1038: 	sv_frame_t startIndex = start / cacheBlock;
Chris@1038: 	sv_frame_t endIndex = (start + count) / cacheBlock;
Chris@147: 
Chris@147: 	float max = 0.0, min = 0.0, total = 0.0;
Chris@1038: 	sv_frame_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@1038: 	    sv_frame_t index = (i + startIndex) * channels + channel;
Chris@1152: 	    if (!in_range_for(cache, index)) 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@1038: 		ranges.push_back(Range(min, max, total / float(got)));
Chris@147:                 min = max = total = 0.0f;
Chris@300:                 got = 0;
Chris@147: 	    }
Chris@147: 	}
Chris@147: 		
Chris@300: 	if (got > 0) {
Chris@1038:             ranges.push_back(Range(min, max, total / float(got)));
Chris@147: 	}
Chris@147:     }
Chris@147: 
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@1151:     cerr << "returning " << ranges.size() << " ranges" << endl;
Chris@236: #endif
Chris@225:     return;
Chris@147: }
Chris@147: 
Chris@1122: ReadOnlyWaveFileModel::Range
Chris@1122: ReadOnlyWaveFileModel::getSummary(int channel, sv_frame_t start, sv_frame_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@929:     int 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@1038:     sv_frame_t blockStart = (start / blockSize) * blockSize;
Chris@1038:     sv_frame_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@929:         for (int i = 0; i < (int)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@1096:         range.setMin(min(range.min(), startRange.min()));
Chris@1096:         range.setMax(max(range.max(), startRange.max()));
Chris@1096:         range.setAbsmean(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@1096:         range.setMin(min(range.min(), endRange.min()));
Chris@1096:         range.setMax(max(range.max(), endRange.max()));
Chris@1096:         range.setAbsmean(min(range.absmean(), endRange.absmean()));
Chris@147:     }
Chris@147: 
Chris@147:     return range;
Chris@147: }
Chris@147: 
Chris@147: void
Chris@1122: ReadOnlyWaveFileModel::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@1122:     SVDEBUG << "ReadOnlyWaveFileModel::fillCache: started fill thread" << endl;
Chris@236: #endif
Chris@147: }   
Chris@147: 
Chris@147: void
Chris@1122: ReadOnlyWaveFileModel::fillTimerTimedOut()
Chris@147: {
Chris@147:     if (m_fillThread) {
Chris@1038: 	sv_frame_t fillExtent = m_fillThread->getFillExtent();
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@1122:         SVDEBUG << "ReadOnlyWaveFileModel::fillTimerTimedOut: extent = " << fillExtent << endl;
Chris@236: #endif
Chris@147: 	if (fillExtent > m_lastFillExtent) {
Chris@931: 	    emit modelChangedWithin(m_lastFillExtent, fillExtent);
Chris@147: 	    m_lastFillExtent = fillExtent;
Chris@147: 	}
Chris@147:     } else {
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@1122:         SVDEBUG << "ReadOnlyWaveFileModel::fillTimerTimedOut: no thread" << endl;
Chris@236: #endif
Chris@147: 	emit modelChanged();
Chris@147:     }
Chris@147: }
Chris@147: 
Chris@147: void
Chris@1122: ReadOnlyWaveFileModel::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@931:         emit modelChangedWithin(m_lastFillExtent, getEndFrame());
Chris@267:     }
Chris@147:     emit modelChanged();
Chris@411:     emit ready();
Chris@236: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@1122:     SVDEBUG << "ReadOnlyWaveFileModel::cacheFilled" << endl;
Chris@236: #endif
Chris@175: }
Chris@175: 
Chris@175: void
Chris@1122: ReadOnlyWaveFileModel::RangeCacheFillThread::run()
Chris@147: {
Chris@929:     int cacheBlockSize[2];
Chris@179:     cacheBlockSize[0] = (1 << m_model.m_zoomConstraint.getMinCachePower());
Chris@1038:     cacheBlockSize[1] = (int((1 << m_model.m_zoomConstraint.getMinCachePower()) *
Chris@608:                                         sqrt(2.) + 0.01));
Chris@147:     
Chris@1038:     sv_frame_t frame = 0;
Chris@1151:     const sv_frame_t readBlockSize = 32768;
Chris@1096:     vector<float> block;
Chris@147: 
Chris@147:     if (!m_model.isOK()) return;
Chris@147:     
Chris@929:     int 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@1151: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@1151:             cerr << "ReadOnlyWaveFileModel::fill: Waiting for channels..." << endl;
Chris@1151: #endif
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@929:     int 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@1151:         m_model.m_mutex.lock();
Chris@147: 
Chris@176:         while (frame < m_frameCount) {
Chris@147: 
Chris@1151:             m_model.m_mutex.unlock();
Chris@1151: 
Chris@1151: #ifdef DEBUG_WAVE_FILE_MODEL
Chris@1151:             cerr << "ReadOnlyWaveFileModel::fill inner loop: frame = " << frame << ", count = " << m_frameCount << ", blocksize " << readBlockSize << endl;
Chris@1151: #endif
Chris@265: 
Chris@1220:             if (updating && (frame + readBlockSize > m_frameCount)) {
Chris@1220:                 m_model.m_mutex.lock(); // must be locked on exiting loop
Chris@1220:                 break;
Chris@1220:             }
Chris@176: 
Chris@1041:             block = m_model.m_reader->getInterleavedFrames(frame, readBlockSize);
Chris@176: 
Chris@1151:             sv_frame_t gotBlockSize = block.size() / channels;
Chris@265: 
Chris@1151:             m_model.m_mutex.lock();
Chris@1151: 
Chris@1151:             for (sv_frame_t i = 0; i < gotBlockSize; ++i) {
Chris@147: 		
Chris@411:                 for (int ch = 0; ch < channels; ++ch) {
Chris@147: 
Chris@1038:                     sv_frame_t index = channels * i + ch;
Chris@176:                     float sample = block[index];
Chris@176:                     
Chris@1151:                     for (int cacheType = 0; cacheType < 2; ++cacheType) {
Chris@1053:                         sv_frame_t rangeIndex = ch * 2 + cacheType;
Chris@1053:                         range[rangeIndex].sample(sample);
Chris@410:                         means[rangeIndex] += fabsf(sample);
Chris@176:                     }
Chris@176:                 }
Chris@1042: 
Chris@1053:                 for (int cacheType = 0; cacheType < 2; ++cacheType) {
Chris@232: 
Chris@1053:                     if (++count[cacheType] == cacheBlockSize[cacheType]) {
Chris@410:                         
Chris@929:                         for (int ch = 0; ch < int(channels); ++ch) {
Chris@1053:                             int rangeIndex = ch * 2 + cacheType;
Chris@1053:                             means[rangeIndex] = means[rangeIndex] / float(count[cacheType]);
Chris@410:                             range[rangeIndex].setAbsmean(means[rangeIndex]);
Chris@1053:                             m_model.m_cache[cacheType].push_back(range[rangeIndex]);
Chris@176:                             range[rangeIndex] = Range();
Chris@411:                             means[rangeIndex] = 0.f;
Chris@176:                         }
Chris@232: 
Chris@1053:                         count[cacheType] = 0;
Chris@176:                     }
Chris@176:                 }
Chris@147:                 
Chris@176:                 ++frame;
Chris@147:             }
Chris@1151: 
Chris@176:             if (m_model.m_exiting) break;
Chris@176:             m_fillExtent = frame;
Chris@147:         }
Chris@147: 
Chris@1151:         m_model.m_mutex.unlock();
Chris@1151:             
Chris@176:         first = false;
Chris@177:         if (m_model.m_exiting) break;
Chris@187:         if (updating) {
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@1053:         for (int cacheType = 0; cacheType < 2; ++cacheType) {
Chris@232: 
Chris@1053:             if (count[cacheType] > 0) {
Chris@232: 
Chris@929:                 for (int ch = 0; ch < int(channels); ++ch) {
Chris@1053:                     int rangeIndex = ch * 2 + cacheType;
Chris@1053:                     means[rangeIndex] = means[rangeIndex] / float(count[cacheType]);
Chris@410:                     range[rangeIndex].setAbsmean(means[rangeIndex]);
Chris@1053:                     m_model.m_cache[cacheType].push_back(range[rangeIndex]);
Chris@177:                     range[rangeIndex] = Range();
Chris@411:                     means[rangeIndex] = 0.f;
Chris@177:                 }
Chris@232: 
Chris@1053:                 count[cacheType] = 0;
Chris@147:             }
Chris@177:             
Chris@1053:             const Range &rr = *m_model.m_cache[cacheType].begin();
Chris@1053:             MUNLOCK(&rr, m_model.m_cache[cacheType].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@1053:     for (int cacheType = 0; cacheType < 2; ++cacheType) {
Chris@1053:         cerr << "Cache type " << cacheType << " now contains " << m_model.m_cache[cacheType].size() << " ranges" << endl;
Chris@236:     }
Chris@236: #endif
Chris@147: }
Chris@147: 
Chris@163: void
Chris@1122: ReadOnlyWaveFileModel::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: