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 "WavFileReader.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: 
lbajardsilogic@0: #include <QMutexLocker>
lbajardsilogic@0: 
lbajardsilogic@0: WavFileReader::WavFileReader(QString path, bool fileUpdating) :
lbajardsilogic@0:     m_file(0),
lbajardsilogic@0:     m_path(path),
lbajardsilogic@0:     m_buffer(0),
lbajardsilogic@0:     m_bufsiz(0),
lbajardsilogic@0:     m_lastStart(0),
lbajardsilogic@0:     m_lastCount(0),
lbajardsilogic@0:     m_updating(fileUpdating)
lbajardsilogic@0: {
lbajardsilogic@0:     m_frameCount = 0;
lbajardsilogic@0:     m_channelCount = 0;
lbajardsilogic@0:     m_sampleRate = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     m_fileInfo.format = 0;
lbajardsilogic@0:     m_fileInfo.frames = 0;
lbajardsilogic@0:     m_file = sf_open(m_path.toLocal8Bit(), SFM_READ, &m_fileInfo);
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_file || (!fileUpdating && m_fileInfo.channels <= 0)) {
lbajardsilogic@0: 	std::cerr << "WavFileReader::initialize: Failed to open file ("
lbajardsilogic@0: 		  << sf_strerror(m_file) << ")" << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0: 	if (m_file) {
lbajardsilogic@0: 	    m_error = QString("Couldn't load audio file '%1':\n%2")
lbajardsilogic@0: 		.arg(m_path).arg(sf_strerror(m_file));
lbajardsilogic@0: 	} else {
lbajardsilogic@0: 	    m_error = QString("Failed to open audio file '%1'")
lbajardsilogic@0: 		.arg(m_path);
lbajardsilogic@0: 	}
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_fileInfo.channels > 0) {
lbajardsilogic@0:         m_frameCount = m_fileInfo.frames;
lbajardsilogic@0:         m_channelCount = m_fileInfo.channels;
lbajardsilogic@0:         m_sampleRate = m_fileInfo.samplerate;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0: //    std::cerr << "WavFileReader: Frame count " << m_frameCount << ", channel count " << m_channelCount << ", sample rate " << m_sampleRate << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: WavFileReader::~WavFileReader()
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_file) sf_close(m_file);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WavFileReader::updateFrameCount()
lbajardsilogic@0: {
lbajardsilogic@0:     QMutexLocker locker(&m_mutex);
lbajardsilogic@0: 
lbajardsilogic@0:     size_t prevCount = m_fileInfo.frames;
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_file) {
lbajardsilogic@0:         sf_close(m_file);
lbajardsilogic@0:         m_file = sf_open(m_path.toLocal8Bit(), SFM_READ, &m_fileInfo);
lbajardsilogic@0:         if (!m_file || m_fileInfo.channels <= 0) {
lbajardsilogic@0:             std::cerr << "WavFileReader::updateFrameCount: Failed to open file ("
lbajardsilogic@0:                       << sf_strerror(m_file) << ")" << std::endl;
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0: //    std::cerr << "WavFileReader::updateFrameCount: now " << m_fileInfo.frames << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:     m_frameCount = m_fileInfo.frames;
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_channelCount == 0) {
lbajardsilogic@0:         m_channelCount = m_fileInfo.channels;
lbajardsilogic@0:         m_sampleRate = m_fileInfo.samplerate;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_frameCount != prevCount) {
lbajardsilogic@0: //        std::cerr << "frameCountChanged" << std::endl;
lbajardsilogic@0:         emit frameCountChanged();
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WavFileReader::updateDone()
lbajardsilogic@0: {
lbajardsilogic@0:     updateFrameCount();
lbajardsilogic@0:     m_updating = false;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WavFileReader::getInterleavedFrames(size_t start, size_t count,
lbajardsilogic@0: 				    SampleBlock &results) const
lbajardsilogic@0: {
lbajardsilogic@0:     if (count == 0) return;
lbajardsilogic@0:     results.clear();
lbajardsilogic@0: 
lbajardsilogic@0:     QMutexLocker locker(&m_mutex);
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_file || !m_channelCount) {
lbajardsilogic@0:         return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if ((long)start >= m_fileInfo.frames) {
lbajardsilogic@0: //        std::cerr << "WavFileReader::getInterleavedFrames: " << start
lbajardsilogic@0: //                  << " > " << m_fileInfo.frames << std::endl;
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (long(start + count) > m_fileInfo.frames) {
lbajardsilogic@0: 	count = m_fileInfo.frames - start;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     sf_count_t readCount = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     if (start != m_lastStart || count != m_lastCount) {
lbajardsilogic@0: 
lbajardsilogic@0: 	if (sf_seek(m_file, start, SEEK_SET) < 0) {
lbajardsilogic@0: //            std::cerr << "sf_seek failed" << std::endl;
lbajardsilogic@0: 	    return;
lbajardsilogic@0: 	}
lbajardsilogic@0: 	
lbajardsilogic@0: 	if (count * m_fileInfo.channels > m_bufsiz) {
lbajardsilogic@0: //	    std::cerr << "WavFileReader: Reallocating buffer for " << count
lbajardsilogic@0: //		      << " frames, " << m_fileInfo.channels << " channels: "
lbajardsilogic@0: //		      << m_bufsiz << " floats" << std::endl;
lbajardsilogic@0: 	    m_bufsiz = count * m_fileInfo.channels;
lbajardsilogic@0: 	    delete[] m_buffer;
lbajardsilogic@0: 	    m_buffer = new float[m_bufsiz];
lbajardsilogic@0: 	}
lbajardsilogic@0: 	
lbajardsilogic@0: 	if ((readCount = sf_readf_float(m_file, m_buffer, count)) < 0) {
lbajardsilogic@0: //            std::cerr << "sf_readf_float failed" << std::endl;
lbajardsilogic@0: 	    return;
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	m_lastStart = start;
lbajardsilogic@0: 	m_lastCount = readCount;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t i = 0; i < count * m_fileInfo.channels; ++i) {
lbajardsilogic@0:         if (i >= m_bufsiz) {
lbajardsilogic@0:             std::cerr << "INTERNAL ERROR: WavFileReader::getInterleavedFrames: " << i << " >= " << m_bufsiz << std::endl;
lbajardsilogic@0:         }
lbajardsilogic@0: 	results.push_back(m_buffer[i]);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     return;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: WavFileReader::getSupportedExtensions(std::set<QString> &extensions)
lbajardsilogic@0: {
lbajardsilogic@0:     int count;
lbajardsilogic@0: 
lbajardsilogic@0:     if (sf_command(0, SFC_GET_FORMAT_MAJOR_COUNT, &count, sizeof(count))) {
lbajardsilogic@0:         extensions.insert("wav");
lbajardsilogic@0:         extensions.insert("aiff");
lbajardsilogic@0:         extensions.insert("aif");
lbajardsilogic@0:         return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     SF_FORMAT_INFO info;
lbajardsilogic@0:     for (int i = 0; i < count; ++i) {
lbajardsilogic@0:         info.format = i;
lbajardsilogic@0:         if (!sf_command(0, SFC_GET_FORMAT_MAJOR, &info, sizeof(info))) {
lbajardsilogic@0:             extensions.insert(info.extension);
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: }