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.
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 "CodedAudioFileReader.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "WavFileReader.h"
lbajardsilogic@0: #include "base/TempDirectory.h"
lbajardsilogic@0: #include "base/Exceptions.h"
lbajardsilogic@0: #include "base/Profiler.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: #include <QDir>
lbajardsilogic@0: 
lbajardsilogic@0: CodedAudioFileReader::CodedAudioFileReader(CacheMode cacheMode) :
lbajardsilogic@0:     m_cacheMode(cacheMode),
lbajardsilogic@0:     m_initialised(false),
lbajardsilogic@0:     m_cacheFileWritePtr(0),
lbajardsilogic@0:     m_cacheFileReader(0),
lbajardsilogic@0:     m_cacheWriteBuffer(0),
lbajardsilogic@0:     m_cacheWriteBufferIndex(0),
lbajardsilogic@0:     m_cacheWriteBufferSize(16384)
lbajardsilogic@0: {
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: CodedAudioFileReader::~CodedAudioFileReader()
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_cacheFileWritePtr) sf_close(m_cacheFileWritePtr);
lbajardsilogic@0:     if (m_cacheFileReader) delete m_cacheFileReader;
lbajardsilogic@0:     if (m_cacheWriteBuffer) delete[] m_cacheWriteBuffer;
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_cacheFileName != "") {
lbajardsilogic@0:         if (!QFile(m_cacheFileName).remove()) {
lbajardsilogic@0:             std::cerr << "WARNING: CodedAudioFileReader::~CodedAudioFileReader: Failed to delete cache file \"" << m_cacheFileName.toStdString() << "\"" << std::endl;
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: CodedAudioFileReader::initialiseDecodeCache()
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_cacheMode == CacheInTemporaryFile) {
lbajardsilogic@0: 
lbajardsilogic@0:         m_cacheWriteBuffer = new float[m_cacheWriteBufferSize * m_channelCount];
lbajardsilogic@0:         m_cacheWriteBufferIndex = 0;
lbajardsilogic@0: 
lbajardsilogic@0:         try {
lbajardsilogic@0:             QDir dir(TempDirectory::getInstance()->getPath());
lbajardsilogic@0:             m_cacheFileName = dir.filePath(QString("decoded_%1.wav")
lbajardsilogic@0:                                            .arg((intptr_t)this));
lbajardsilogic@0: 
lbajardsilogic@0:             SF_INFO fileInfo;
lbajardsilogic@0:             fileInfo.samplerate = m_sampleRate;
lbajardsilogic@0:             fileInfo.channels = m_channelCount;
lbajardsilogic@0:             fileInfo.format = SF_FORMAT_WAV | SF_FORMAT_FLOAT;
lbajardsilogic@0:     
lbajardsilogic@0:             m_cacheFileWritePtr = sf_open(m_cacheFileName.toLocal8Bit(),
lbajardsilogic@0:                                           SFM_WRITE, &fileInfo);
lbajardsilogic@0: 
lbajardsilogic@0:             if (!m_cacheFileWritePtr) {
lbajardsilogic@0:                 std::cerr << "CodedAudioFileReader::initialiseDecodeCache: failed to open cache file \"" << m_cacheFileName.toStdString() << "\" (" << m_channelCount << " channels, sample rate " << m_sampleRate << " for writing, falling back to in-memory cache" << std::endl;
lbajardsilogic@0:                 m_cacheMode = CacheInMemory;
lbajardsilogic@0:             }
lbajardsilogic@0:         } catch (DirectoryCreationFailed f) {
lbajardsilogic@0:             std::cerr << "CodedAudioFileReader::initialiseDecodeCache: failed to create temporary directory! Falling back to in-memory cache" << std::endl;
lbajardsilogic@0:             m_cacheMode = CacheInMemory;
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_cacheMode == CacheInMemory) {
lbajardsilogic@0:         m_data.clear();
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     m_initialised = true;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: CodedAudioFileReader::addSampleToDecodeCache(float sample)
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_initialised) return;
lbajardsilogic@0: 
lbajardsilogic@0:     switch (m_cacheMode) {
lbajardsilogic@0: 
lbajardsilogic@0:     case CacheInTemporaryFile:
lbajardsilogic@0: 
lbajardsilogic@0:         m_cacheWriteBuffer[m_cacheWriteBufferIndex++] = sample;
lbajardsilogic@0: 
lbajardsilogic@0:         if (m_cacheWriteBufferIndex ==
lbajardsilogic@0:             m_cacheWriteBufferSize * m_channelCount) {
lbajardsilogic@0: 
lbajardsilogic@0:             //!!! check for return value! out of disk space, etc!
lbajardsilogic@0:             sf_writef_float(m_cacheFileWritePtr,
lbajardsilogic@0:                             m_cacheWriteBuffer,
lbajardsilogic@0:                             m_cacheWriteBufferSize);
lbajardsilogic@0: 
lbajardsilogic@0:             m_cacheWriteBufferIndex = 0;
lbajardsilogic@0:         }
lbajardsilogic@0:         break;
lbajardsilogic@0: 
lbajardsilogic@0:     case CacheInMemory:
lbajardsilogic@0:         m_data.push_back(sample);
lbajardsilogic@0:         break;
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: CodedAudioFileReader::finishDecodeCache()
lbajardsilogic@0: {
lbajardsilogic@0:     Profiler profiler("CodedAudioFileReader::finishDecodeCache", true);
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_initialised) {
lbajardsilogic@0:         std::cerr << "WARNING: CodedAudioFileReader::finishDecodeCache: Cache was never initialised!" << std::endl;
lbajardsilogic@0:         return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     switch (m_cacheMode) {
lbajardsilogic@0: 
lbajardsilogic@0:     case CacheInTemporaryFile:
lbajardsilogic@0: 
lbajardsilogic@0:         if (m_cacheWriteBufferIndex > 0) {
lbajardsilogic@0:             //!!! check for return value! out of disk space, etc!
lbajardsilogic@0:             sf_writef_float(m_cacheFileWritePtr,
lbajardsilogic@0:                             m_cacheWriteBuffer,
lbajardsilogic@0:                             m_cacheWriteBufferIndex / m_channelCount);
lbajardsilogic@0:         }
lbajardsilogic@0: 
lbajardsilogic@0:         if (m_cacheWriteBuffer) {
lbajardsilogic@0:             delete[] m_cacheWriteBuffer;
lbajardsilogic@0:             m_cacheWriteBuffer = 0;
lbajardsilogic@0:         }
lbajardsilogic@0: 
lbajardsilogic@0:         m_cacheWriteBufferIndex = 0;
lbajardsilogic@0: 
lbajardsilogic@0:         sf_close(m_cacheFileWritePtr);
lbajardsilogic@0:         m_cacheFileWritePtr = 0;
lbajardsilogic@0: 
lbajardsilogic@0:         m_cacheFileReader = new WavFileReader(m_cacheFileName);
lbajardsilogic@0: 
lbajardsilogic@0:         if (!m_cacheFileReader->isOK()) {
lbajardsilogic@0:             std::cerr << "ERROR: CodedAudioFileReader::finishDecodeCache: Failed to construct WAV file reader for temporary file: " << m_cacheFileReader->getError().toStdString() << std::endl;
lbajardsilogic@0:             delete m_cacheFileReader;
lbajardsilogic@0:             m_cacheFileReader = 0;
lbajardsilogic@0:         }
lbajardsilogic@0:         break;
lbajardsilogic@0: 
lbajardsilogic@0:     case CacheInMemory:
lbajardsilogic@0:         // nothing to do 
lbajardsilogic@0:         break;
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: CodedAudioFileReader::getInterleavedFrames(size_t start, size_t count,
lbajardsilogic@0:                                            SampleBlock &frames) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_initialised) return;
lbajardsilogic@0: 
lbajardsilogic@0:     switch (m_cacheMode) {
lbajardsilogic@0: 
lbajardsilogic@0:     case CacheInTemporaryFile:
lbajardsilogic@0:         if (m_cacheFileReader) {
lbajardsilogic@0:             m_cacheFileReader->getInterleavedFrames(start, count, frames);
lbajardsilogic@0:         }
lbajardsilogic@0:         break;
lbajardsilogic@0: 
lbajardsilogic@0:     case CacheInMemory:
lbajardsilogic@0:     {
lbajardsilogic@0:         frames.clear();
lbajardsilogic@0:         if (!isOK()) return;
lbajardsilogic@0:         if (count == 0) return;
lbajardsilogic@0: 
lbajardsilogic@0:         // slownessabounds
lbajardsilogic@0: 
lbajardsilogic@0:         for (size_t i = start; i < start + count; ++i) {
lbajardsilogic@0:             for (size_t ch = 0; ch < m_channelCount; ++ch) {
lbajardsilogic@0:                 size_t index = i * m_channelCount + ch;
lbajardsilogic@0:                 if (index >= m_data.size()) return;
lbajardsilogic@0:                 frames.push_back(m_data[index]);
lbajardsilogic@0:             }
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: