lbajardsilogic@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
lbajardsilogic@0: 
lbajardsilogic@0: /*
lbajardsilogic@0:     Sonic Visualiser
lbajardsilogic@0:     An audio file viewer and annotation editor.
lbajardsilogic@0:     Centre for Digital Music, Queen Mary, University of London.
lbajardsilogic@0:     This file copyright 2006 Chris Cannam and QMUL.
lbajardsilogic@0:     
lbajardsilogic@0:     This program is free software; you can redistribute it and/or
lbajardsilogic@0:     modify it under the terms of the GNU General Public License as
lbajardsilogic@0:     published by the Free Software Foundation; either version 2 of the
lbajardsilogic@0:     License, or (at your option) any later version.  See the file
lbajardsilogic@0:     COPYING included with this distribution for more information.
lbajardsilogic@0: */
lbajardsilogic@0: 
lbajardsilogic@0: #include "WaveFileModel.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "fileio/AudioFileReader.h"
lbajardsilogic@0: #include "fileio/AudioFileReaderFactory.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "system/System.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <QMessageBox>
lbajardsilogic@0: #include <QFileInfo>
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: //#include <unistd.h>
lbajardsilogic@0: #include <math.h>
lbajardsilogic@0: #include <sndfile.h>
lbajardsilogic@0: 
lbajardsilogic@0: #include <cassert>
lbajardsilogic@0: 
lbajardsilogic@0: //#define DEBUG_WAVE_FILE_MODEL 1
lbajardsilogic@0: 
lbajardsilogic@33: #include "main/MainWindow.h"
lbajardsilogic@33: 
lbajardsilogic@0: using std::cerr;
lbajardsilogic@0: using std::endl;
lbajardsilogic@0: 
lbajardsilogic@0: PowerOfSqrtTwoZoomConstraint
lbajardsilogic@0: WaveFileModel::m_zoomConstraint;
lbajardsilogic@0: 
lbajardsilogic@0: WaveFileModel::WaveFileModel(QString path) :
lbajardsilogic@0:     m_path(path),
lbajardsilogic@0:     m_myReader(true),
lbajardsilogic@0:     m_fillThread(0),
lbajardsilogic@0:     m_updateTimer(0),
lbajardsilogic@0:     m_lastFillExtent(0),
lbajardsilogic@0:     m_exiting(false)
lbajardsilogic@0: {
lbajardsilogic@33: 	MainWindow::instance()->setCursor(Qt::BusyCursor);
lbajardsilogic@33: 	MainWindow::instance()->statusBar()->showMessage(tr("Opening audio file..."));
lbajardsilogic@0:     m_reader = AudioFileReaderFactory::createReader(path);
lbajardsilogic@0:     setObjectName(QFileInfo(path).fileName());
lbajardsilogic@0:     if (isOK()) fillCache();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: WaveFileModel::WaveFileModel(QString path, QString originalLocation) :
lbajardsilogic@0:     m_path(originalLocation),
lbajardsilogic@0:     m_myReader(true),
lbajardsilogic@0:     m_fillThread(0),
lbajardsilogic@0:     m_updateTimer(0),
lbajardsilogic@0:     m_lastFillExtent(0),
lbajardsilogic@0:     m_exiting(false)
lbajardsilogic@0: {
lbajardsilogic@0:     m_reader = AudioFileReaderFactory::createReader(path);
lbajardsilogic@0:     setObjectName(QFileInfo(originalLocation).fileName());
lbajardsilogic@0:     if (isOK()) fillCache();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: WaveFileModel::WaveFileModel(QString path, AudioFileReader *reader) :
lbajardsilogic@0:     m_path(path),
lbajardsilogic@0:     m_myReader(false),
lbajardsilogic@0:     m_fillThread(0),
lbajardsilogic@0:     m_updateTimer(0),
lbajardsilogic@0:     m_lastFillExtent(0),
lbajardsilogic@0:     m_exiting(false)
lbajardsilogic@0: {
lbajardsilogic@0:     m_reader = reader;
lbajardsilogic@0:     setObjectName(QFileInfo(path).fileName());
lbajardsilogic@0:     fillCache();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: WaveFileModel::~WaveFileModel()
lbajardsilogic@0: {
lbajardsilogic@0:     m_exiting = true;
lbajardsilogic@0:     if (m_fillThread) m_fillThread->wait();
lbajardsilogic@0:     if (m_myReader) delete m_reader;
lbajardsilogic@0:     m_reader = 0;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: WaveFileModel::isOK() const
lbajardsilogic@0: {
lbajardsilogic@0:     return m_reader && m_reader->isOK();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: WaveFileModel::isReady(int *completion) const
lbajardsilogic@0: {
lbajardsilogic@0:     bool ready = (isOK() && (m_fillThread == 0));
lbajardsilogic@0:     double c = double(m_lastFillExtent) / double(getEndFrame() - getStartFrame());
lbajardsilogic@0:     if (completion) *completion = int(c * 100.0 + 0.01);
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL
lbajardsilogic@0:     std::cerr << "WaveFileModel::isReady(): ready = " << ready << ", completion = " << (completion ? *completion : -1) << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0:     return ready;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: Model *
lbajardsilogic@0: WaveFileModel::clone() const
lbajardsilogic@0: {
lbajardsilogic@0:     WaveFileModel *model = new WaveFileModel(m_path);
lbajardsilogic@0:     return model;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: size_t
lbajardsilogic@0: WaveFileModel::getFrameCount() const
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_reader) return 0;
lbajardsilogic@0:     return m_reader->getFrameCount();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: size_t
lbajardsilogic@0: WaveFileModel::getChannelCount() const
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_reader) return 0;
lbajardsilogic@0:     return m_reader->getChannelCount();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: size_t
lbajardsilogic@0: WaveFileModel::getSampleRate() const 
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_reader) return 0;
lbajardsilogic@0:     return m_reader->getSampleRate();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: size_t
lbajardsilogic@0: WaveFileModel::getValues(int channel, size_t start, size_t end,
lbajardsilogic@0: 			 float *buffer) const
lbajardsilogic@0: {
lbajardsilogic@0:     // Always read these directly from the file. 
lbajardsilogic@0:     // This is used for e.g. audio playback.
lbajardsilogic@0:     // Could be much more efficient (although compiler optimisation will help)
lbajardsilogic@0: 
lbajardsilogic@0:     if (end < start) {
lbajardsilogic@0: 	std::cerr << "ERROR: WaveFileModel::getValues[float]: end < start ("
lbajardsilogic@0: 		  << end << " < " << start << ")" << std::endl;
lbajardsilogic@0: 	assert(end >= start);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_reader || !m_reader->isOK()) return 0;
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL
lbajardsilogic@0: //    std::cerr << "WaveFileModel::getValues(" << channel << ", "
lbajardsilogic@0: //              << start << ", " << end << "): calling reader" << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0:     SampleBlock frames;
lbajardsilogic@0:     m_reader->getInterleavedFrames(start, end - start, frames);
lbajardsilogic@0: 
lbajardsilogic@0:     size_t i = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     int ch0 = channel, ch1 = channel, channels = getChannelCount();
lbajardsilogic@0:     if (channel == -1) {
lbajardsilogic@0: 	ch0 = 0;
lbajardsilogic@0: 	ch1 = channels - 1;
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     while (i < end - start) {
lbajardsilogic@0: 
lbajardsilogic@0: 	buffer[i] = 0.0;
lbajardsilogic@0: 
lbajardsilogic@0: 	for (int ch = ch0; ch <= ch1; ++ch) {
lbajardsilogic@0: 
lbajardsilogic@0: 	    size_t index = i * channels + ch;
lbajardsilogic@0: 	    if (index >= frames.size()) break;
lbajardsilogic@0:             
lbajardsilogic@0: 	    float sample = frames[index];
lbajardsilogic@0: 	    buffer[i] += sample;
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	++i;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     return i;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: size_t
lbajardsilogic@0: WaveFileModel::getValues(int channel, size_t start, size_t end,
lbajardsilogic@0: 			 double *buffer) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (end < start) {
lbajardsilogic@0: 	std::cerr << "ERROR: WaveFileModel::getValues[double]: end < start ("
lbajardsilogic@0: 		  << end << " < " << start << ")" << std::endl;
lbajardsilogic@0: 	assert(end >= start);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_reader || !m_reader->isOK()) return 0;
lbajardsilogic@0: 
lbajardsilogic@0:     SampleBlock frames;
lbajardsilogic@0:     m_reader->getInterleavedFrames(start, end - start, frames);
lbajardsilogic@0: 
lbajardsilogic@0:     size_t i = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     int ch0 = channel, ch1 = channel, channels = getChannelCount();
lbajardsilogic@0:     if (channel == -1) {
lbajardsilogic@0: 	ch0 = 0;
lbajardsilogic@0: 	ch1 = channels - 1;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     while (i < end - start) {
lbajardsilogic@0: 
lbajardsilogic@0: 	buffer[i] = 0.0;
lbajardsilogic@0: 
lbajardsilogic@0: 	for (int ch = ch0; ch <= ch1; ++ch) {
lbajardsilogic@0: 
lbajardsilogic@0: 	    size_t index = i * channels + ch;
lbajardsilogic@0: 	    if (index >= frames.size()) break;
lbajardsilogic@0:             
lbajardsilogic@0: 	    float sample = frames[index];
lbajardsilogic@0: 	    buffer[i] += sample;
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	++i;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     return i;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WaveFileModel::getRanges(size_t channel, size_t start, size_t end,
lbajardsilogic@0:                          RangeBlock &ranges, size_t &blockSize) const
lbajardsilogic@0: {
lbajardsilogic@0:     ranges.clear();
lbajardsilogic@0:     if (!isOK()) return;
lbajardsilogic@0: 
lbajardsilogic@0:     if (end <= start) {
lbajardsilogic@0: 	std::cerr << "WARNING: Internal error: end <= start in WaveFileModel::getRanges (end = " << end << ", start = " << start << ", blocksize = " << blockSize << ")" << std::endl;
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     int cacheType = 0;
lbajardsilogic@0:     int power = m_zoomConstraint.getMinCachePower();
lbajardsilogic@0:     blockSize = m_zoomConstraint.getNearestBlockSize
lbajardsilogic@0:         (blockSize, cacheType, power, ZoomConstraint::RoundUp);
lbajardsilogic@0: 
lbajardsilogic@0:     size_t channels = getChannelCount();
lbajardsilogic@0: 
lbajardsilogic@0:     if (cacheType != 0 && cacheType != 1) {
lbajardsilogic@0: 
lbajardsilogic@0: 	// We need to read directly from the file.  We haven't got
lbajardsilogic@0: 	// this cached.  Hope the requested area is small.  This is
lbajardsilogic@0: 	// not optimal -- we'll end up reading the same frames twice
lbajardsilogic@0: 	// for stereo files, in two separate calls to this method.
lbajardsilogic@0: 	// We could fairly trivially handle this for most cases that
lbajardsilogic@0: 	// matter by putting a single cache in getInterleavedFrames
lbajardsilogic@0: 	// for short queries.
lbajardsilogic@0: 
lbajardsilogic@0: 	SampleBlock frames;
lbajardsilogic@0: 	m_reader->getInterleavedFrames(start, end - start, frames);
lbajardsilogic@0: 	float max = 0.0, min = 0.0, total = 0.0;
lbajardsilogic@0: 	size_t i = 0, count = 0;
lbajardsilogic@0: 
lbajardsilogic@0: 	while (i < end - start) {
lbajardsilogic@0: 
lbajardsilogic@0: 	    size_t index = i * channels + channel;
lbajardsilogic@0: 	    if (index >= frames.size()) break;
lbajardsilogic@0:             
lbajardsilogic@0: 	    float sample = frames[index];
lbajardsilogic@0:             if (sample > max || count == 0) max = sample;
lbajardsilogic@0: 	    if (sample < min || count == 0) min = sample;
lbajardsilogic@0:             total += fabsf(sample);
lbajardsilogic@0: 	    
lbajardsilogic@0: 	    ++i;
lbajardsilogic@0:             ++count;
lbajardsilogic@0:             
lbajardsilogic@0:             if (count == blockSize) {
lbajardsilogic@0:                 ranges.push_back(Range(min, max, total / count));
lbajardsilogic@0:                 min = max = total = 0.0f;
lbajardsilogic@0:                 count = 0;
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	if (count > 0) {
lbajardsilogic@0:             ranges.push_back(Range(min, max, total / count));
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	return;
lbajardsilogic@0: 
lbajardsilogic@0:     } else {
lbajardsilogic@0: 
lbajardsilogic@0: 	QMutexLocker locker(&m_mutex);
lbajardsilogic@0:     
lbajardsilogic@0: 	const RangeBlock &cache = m_cache[cacheType];
lbajardsilogic@0: 
lbajardsilogic@0: 	size_t cacheBlock, div;
lbajardsilogic@0:         
lbajardsilogic@0: 	if (cacheType == 0) {
lbajardsilogic@0: 	    cacheBlock = (1 << m_zoomConstraint.getMinCachePower());
lbajardsilogic@0:             div = (1 << power) / cacheBlock;
lbajardsilogic@0: 	} else {
lbajardsilogic@0: 	    cacheBlock = ((unsigned int)((1 << m_zoomConstraint.getMinCachePower()) * sqrt(2.0) + 0.01));
lbajardsilogic@0:             div = ((unsigned int)((1 << power) * sqrt(2.0) + 0.01)) / cacheBlock;
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	size_t startIndex = start / cacheBlock;
lbajardsilogic@0: 	size_t endIndex = end / cacheBlock;
lbajardsilogic@0: 
lbajardsilogic@0: 	float max = 0.0, min = 0.0, total = 0.0;
lbajardsilogic@0: 	size_t i = 0, count = 0;
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL
lbajardsilogic@0: 	cerr << "blockSize is " << blockSize << ", cacheBlock " << cacheBlock << ", start " << start << ", end " << end << " (frame count " << getFrameCount() << "), power is " << power << ", div is " << div << ", startIndex " << startIndex << ", endIndex " << endIndex << endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: 
lbajardsilogic@0: 	for (i = 0; i < endIndex - startIndex; ) {
lbajardsilogic@0:         
lbajardsilogic@0: 	    size_t index = (i + startIndex) * channels + channel;
lbajardsilogic@0: 	    if (index >= cache.size()) break;
lbajardsilogic@0:             
lbajardsilogic@0:             const Range &range = cache[index];
lbajardsilogic@0:             if (range.max > max || count == 0) max = range.max;
lbajardsilogic@0:             if (range.min < min || count == 0) min = range.min;
lbajardsilogic@0:             total += range.absmean;
lbajardsilogic@0:             
lbajardsilogic@0: 	    ++i;
lbajardsilogic@0:             ++count;
lbajardsilogic@0:             
lbajardsilogic@0: 	    if (count == div) {
lbajardsilogic@0: 		ranges.push_back(Range(min, max, total / count));
lbajardsilogic@0:                 min = max = total = 0.0f;
lbajardsilogic@0:                 count = 0;
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	}
lbajardsilogic@0: 		
lbajardsilogic@0: 	if (count > 0) {
lbajardsilogic@0:             ranges.push_back(Range(min, max, total / count));
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL
lbajardsilogic@0:     cerr << "returning " << ranges.size() << " ranges" << endl;
lbajardsilogic@0: #endif
lbajardsilogic@0:     return;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: WaveFileModel::Range
lbajardsilogic@0: WaveFileModel::getRange(size_t channel, size_t start, size_t end) const
lbajardsilogic@0: {
lbajardsilogic@0:     Range range;
lbajardsilogic@0:     if (!isOK()) return range;
lbajardsilogic@0: 
lbajardsilogic@0:     if (end <= start) {
lbajardsilogic@0: 	std::cerr << "WARNING: Internal error: end <= start in WaveFileModel::getRange (end = " << end << ", start = " << start << ")" << std::endl;
lbajardsilogic@0: 	return range;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     size_t blockSize;
lbajardsilogic@0:     for (blockSize = 1; blockSize <= end - start; blockSize *= 2);
lbajardsilogic@0:     blockSize /= 2;
lbajardsilogic@0: 
lbajardsilogic@0:     bool first = false;
lbajardsilogic@0: 
lbajardsilogic@0:     size_t blockStart = (start / blockSize) * blockSize;
lbajardsilogic@0:     size_t blockEnd = (end / blockSize) * blockSize;
lbajardsilogic@0: 
lbajardsilogic@0:     if (blockStart < start) blockStart += blockSize;
lbajardsilogic@0:         
lbajardsilogic@0:     if (blockEnd > blockStart) {
lbajardsilogic@0:         RangeBlock ranges;
lbajardsilogic@0:         getRanges(channel, blockStart, blockEnd, ranges, blockSize);
lbajardsilogic@0:         for (size_t i = 0; i < ranges.size(); ++i) {
lbajardsilogic@0:             if (first || ranges[i].min < range.min) range.min = ranges[i].min;
lbajardsilogic@0:             if (first || ranges[i].max > range.max) range.max = ranges[i].max;
lbajardsilogic@0:             if (first || ranges[i].absmean < range.absmean) range.absmean = ranges[i].absmean;
lbajardsilogic@0:             first = false;
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (blockStart > start) {
lbajardsilogic@0:         Range startRange = getRange(channel, start, blockStart);
lbajardsilogic@190:         range.min = MIN(range.min, startRange.min);
lbajardsilogic@190:         range.max = MAX(range.max, startRange.max);
lbajardsilogic@190:         range.absmean = MIN(range.absmean, startRange.absmean);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (blockEnd < end) {
lbajardsilogic@0:         Range endRange = getRange(channel, blockEnd, end);
lbajardsilogic@190:         range.min = MIN(range.min, endRange.min);
lbajardsilogic@190:         range.max = MAX(range.max, endRange.max);
lbajardsilogic@190:         range.absmean = MIN(range.absmean, endRange.absmean);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     return range;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WaveFileModel::fillCache()
lbajardsilogic@0: {
lbajardsilogic@0:     m_mutex.lock();
lbajardsilogic@0: 
lbajardsilogic@0:     m_updateTimer = new QTimer(this);
lbajardsilogic@0:     connect(m_updateTimer, SIGNAL(timeout()), this, SLOT(fillTimerTimedOut()));
lbajardsilogic@0:     m_updateTimer->start(100);
lbajardsilogic@0: 
lbajardsilogic@0:     m_fillThread = new RangeCacheFillThread(*this);
lbajardsilogic@0:     connect(m_fillThread, SIGNAL(finished()), this, SLOT(cacheFilled()));
lbajardsilogic@0: 
lbajardsilogic@0:     m_mutex.unlock();
lbajardsilogic@0:     m_fillThread->start();
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL
lbajardsilogic@0:     std::cerr << "WaveFileModel::fillCache: started fill thread" << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: }   
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WaveFileModel::fillTimerTimedOut()
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_fillThread) {
lbajardsilogic@0: 	size_t fillExtent = m_fillThread->getFillExtent();
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL
lbajardsilogic@0:         cerr << "WaveFileModel::fillTimerTimedOut: extent = " << fillExtent << endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: 	if (fillExtent > m_lastFillExtent) {
lbajardsilogic@0: 	    emit modelChanged(m_lastFillExtent, fillExtent);
lbajardsilogic@0: 	    m_lastFillExtent = fillExtent;
lbajardsilogic@0: 	}
lbajardsilogic@0:     } else {
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL
lbajardsilogic@0:         cerr << "WaveFileModel::fillTimerTimedOut: no thread" << std::endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: 	emit modelChanged();
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WaveFileModel::cacheFilled()
lbajardsilogic@0: {
lbajardsilogic@33: 	MainWindow::instance()->setCursor(Qt::ArrowCursor);
lbajardsilogic@33: 	MainWindow::instance()->statusBar()->clearMessage();
lbajardsilogic@0:     m_mutex.lock();
lbajardsilogic@0:     delete m_fillThread;
lbajardsilogic@0:     m_fillThread = 0;
lbajardsilogic@0:     delete m_updateTimer;
lbajardsilogic@0:     m_updateTimer = 0;
lbajardsilogic@0:     m_mutex.unlock();
lbajardsilogic@0:     emit modelChanged();
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL
lbajardsilogic@0:     cerr << "WaveFileModel::cacheFilled" << endl;
lbajardsilogic@0: #endif
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WaveFileModel::RangeCacheFillThread::run()
lbajardsilogic@0: {
lbajardsilogic@0:     size_t cacheBlockSize[2];
lbajardsilogic@0:     cacheBlockSize[0] = (1 << m_model.m_zoomConstraint.getMinCachePower());
lbajardsilogic@0:     cacheBlockSize[1] = ((unsigned int)((1 << m_model.m_zoomConstraint.getMinCachePower()) *
lbajardsilogic@0:                                         sqrt(2.0) + 0.01));
lbajardsilogic@0:     
lbajardsilogic@0:     size_t frame = 0;
lbajardsilogic@0:     size_t readBlockSize = 16384;
lbajardsilogic@0:     SampleBlock block;
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_model.isOK()) return;
lbajardsilogic@0:     
lbajardsilogic@0:     size_t channels = m_model.getChannelCount();
lbajardsilogic@0:     bool updating = m_model.m_reader->isUpdating();
lbajardsilogic@0: 
lbajardsilogic@0:     if (updating) {
lbajardsilogic@0:         while (channels == 0 && !m_model.m_exiting) {
lbajardsilogic@0: //            std::cerr << "WaveFileModel::fill: Waiting for channels..." << std::endl;
lbajardsilogic@0:             sleep(1);
lbajardsilogic@0:             channels = m_model.getChannelCount();
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     Range *range = new Range[2 * channels];
lbajardsilogic@0:     size_t count[2];
lbajardsilogic@0:     count[0] = count[1] = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     bool first = true;
lbajardsilogic@0: 
lbajardsilogic@0:     while (first || updating) {
lbajardsilogic@0: 
lbajardsilogic@0:         updating = m_model.m_reader->isUpdating();
lbajardsilogic@0:         m_frameCount = m_model.getFrameCount();
lbajardsilogic@0: 
lbajardsilogic@0: //        std::cerr << "WaveFileModel::fill: frame = " << frame << ", count = " << m_frameCount << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:         while (frame < m_frameCount) {
lbajardsilogic@0: 
lbajardsilogic@0:             if (updating && (frame + readBlockSize > m_frameCount)) break;
lbajardsilogic@0: 
lbajardsilogic@0:             m_model.m_reader->getInterleavedFrames(frame, readBlockSize, block);
lbajardsilogic@0: 
lbajardsilogic@0:             for (size_t i = 0; i < readBlockSize; ++i) {
lbajardsilogic@0: 		
lbajardsilogic@0:                 if (channels * i + channels > block.size()) break;
lbajardsilogic@0: 
lbajardsilogic@0:                 for (size_t ch = 0; ch < size_t(channels); ++ch) {
lbajardsilogic@0: 
lbajardsilogic@0:                     size_t index = channels * i + ch;
lbajardsilogic@0:                     float sample = block[index];
lbajardsilogic@0:                     
lbajardsilogic@0:                     for (size_t ct = 0; ct < 2; ++ct) { // cache type
lbajardsilogic@0:                         
lbajardsilogic@0:                         size_t rangeIndex = ch * 2 + ct;
lbajardsilogic@0:                         
lbajardsilogic@0:                         if (sample > range[rangeIndex].max || count[ct] == 0) {
lbajardsilogic@0:                             range[rangeIndex].max = sample;
lbajardsilogic@0:                         }
lbajardsilogic@0:                         if (sample < range[rangeIndex].min || count[ct] == 0) {
lbajardsilogic@0:                             range[rangeIndex].min = sample;
lbajardsilogic@0:                         }
lbajardsilogic@0:                         range[rangeIndex].absmean += fabsf(sample);
lbajardsilogic@0:                     }
lbajardsilogic@0:                 }
lbajardsilogic@0:                 
lbajardsilogic@0:                 QMutexLocker locker(&m_model.m_mutex);
lbajardsilogic@0: 
lbajardsilogic@0:                 for (size_t ct = 0; ct < 2; ++ct) {
lbajardsilogic@0: 
lbajardsilogic@0:                     if (++count[ct] == cacheBlockSize[ct]) {
lbajardsilogic@0: 
lbajardsilogic@0:                         for (size_t ch = 0; ch < size_t(channels); ++ch) {
lbajardsilogic@0:                             size_t rangeIndex = ch * 2 + ct;
lbajardsilogic@0:                             range[rangeIndex].absmean /= count[ct];
lbajardsilogic@0:                             m_model.m_cache[ct].push_back(range[rangeIndex]);
lbajardsilogic@0:                             range[rangeIndex] = Range();
lbajardsilogic@0:                         }
lbajardsilogic@0: 
lbajardsilogic@0:                         count[ct] = 0;
lbajardsilogic@0:                     }
lbajardsilogic@0:                 }
lbajardsilogic@0:                 
lbajardsilogic@0:                 ++frame;
lbajardsilogic@0:             }
lbajardsilogic@0:             
lbajardsilogic@0:             if (m_model.m_exiting) break;
lbajardsilogic@0:             
lbajardsilogic@0:             m_fillExtent = frame;
lbajardsilogic@0:         }
lbajardsilogic@0: 
lbajardsilogic@0:         first = false;
lbajardsilogic@0:         if (m_model.m_exiting) break;
lbajardsilogic@0:         if (updating) {
lbajardsilogic@0:             sleep(1);
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_model.m_exiting) {
lbajardsilogic@0: 
lbajardsilogic@0:         QMutexLocker locker(&m_model.m_mutex);
lbajardsilogic@0: 
lbajardsilogic@0:         for (size_t ct = 0; ct < 2; ++ct) {
lbajardsilogic@0: 
lbajardsilogic@0:             if (count[ct] > 0) {
lbajardsilogic@0: 
lbajardsilogic@0:                 for (size_t ch = 0; ch < size_t(channels); ++ch) {
lbajardsilogic@0:                     size_t rangeIndex = ch * 2 + ct;
lbajardsilogic@0:                     range[rangeIndex].absmean /= count[ct];
lbajardsilogic@0:                     m_model.m_cache[ct].push_back(range[rangeIndex]);
lbajardsilogic@0:                     range[rangeIndex] = Range();
lbajardsilogic@0:                 }
lbajardsilogic@0: 
lbajardsilogic@0:                 count[ct] = 0;
lbajardsilogic@0:             }
lbajardsilogic@0:             
lbajardsilogic@0:             const Range &rr = *m_model.m_cache[ct].begin();
lbajardsilogic@0:             MUNLOCK(&rr, m_model.m_cache[ct].capacity() * sizeof(Range));
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     delete[] range;
lbajardsilogic@0: 
lbajardsilogic@0:     m_fillExtent = m_frameCount;
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef DEBUG_WAVE_FILE_MODEL        
lbajardsilogic@0:     for (size_t ct = 0; ct < 2; ++ct) {
lbajardsilogic@0:         cerr << "Cache type " << ct << " now contains " << m_model.m_cache[ct].size() << " ranges" << endl;
lbajardsilogic@0:     }
lbajardsilogic@0: #endif
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WaveFileModel::toXml(QTextStream &out,
lbajardsilogic@0:                      QString indent,
lbajardsilogic@0:                      QString extraAttributes) const
lbajardsilogic@0: {
lbajardsilogic@0:     Model::toXml(out, indent,
lbajardsilogic@0:                  QString("type=\"wavefile\" file=\"%1\" %2")
lbajardsilogic@0:                  .arg(m_path).arg(extraAttributes));
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: WaveFileModel::toXmlString(QString indent,
lbajardsilogic@0: 			   QString extraAttributes) const
lbajardsilogic@0: {
lbajardsilogic@0:     return Model::toXmlString(indent,
lbajardsilogic@0: 			      QString("type=\"wavefile\" file=\"%1\" %2")
lbajardsilogic@0: 			      .arg(m_path).arg(extraAttributes));
lbajardsilogic@0: }
lbajardsilogic@0: