diff data/fileio/WavFileReader.cpp @ 148:1a42221a1522

* Reorganising code base. This revision will not compile.
author Chris Cannam
date Mon, 31 Jul 2006 11:49:58 +0000
parents
children c03ec31005e1
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/data/fileio/WavFileReader.cpp	Mon Jul 31 11:49:58 2006 +0000
@@ -0,0 +1,112 @@
+/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
+
+/*
+    Sonic Visualiser
+    An audio file viewer and annotation editor.
+    Centre for Digital Music, Queen Mary, University of London.
+    This file copyright 2006 Chris Cannam.
+    
+    This program is free software; you can redistribute it and/or
+    modify it under the terms of the GNU General Public License as
+    published by the Free Software Foundation; either version 2 of the
+    License, or (at your option) any later version.  See the file
+    COPYING included with this distribution for more information.
+*/
+
+#include "WavFileReader.h"
+
+#include <iostream>
+
+WavFileReader::WavFileReader(QString path) :
+    m_file(0),
+    m_path(path),
+    m_buffer(0),
+    m_bufsiz(0),
+    m_lastStart(0),
+    m_lastCount(0)
+{
+    m_frameCount = 0;
+    m_channelCount = 0;
+    m_sampleRate = 0;
+
+    m_fileInfo.format = 0;
+    m_fileInfo.frames = 0;
+    m_file = sf_open(m_path.toLocal8Bit(), SFM_READ, &m_fileInfo);
+
+    if (!m_file || m_fileInfo.frames <= 0 || m_fileInfo.channels <= 0) {
+	std::cerr << "WavFileReader::initialize: Failed to open file ("
+		  << sf_strerror(m_file) << ")" << std::endl;
+
+	if (m_file) {
+	    m_error = QString("Couldn't load audio file '%1':\n%2")
+		.arg(m_path).arg(sf_strerror(m_file));
+	} else {
+	    m_error = QString("Failed to open audio file '%1'")
+		.arg(m_path);
+	}
+	return;
+    }
+
+    m_frameCount = m_fileInfo.frames;
+    m_channelCount = m_fileInfo.channels;
+    m_sampleRate = m_fileInfo.samplerate;
+}
+
+WavFileReader::~WavFileReader()
+{
+    if (m_file) sf_close(m_file);
+}
+
+void
+WavFileReader::getInterleavedFrames(size_t start, size_t count,
+				    SampleBlock &results) const
+{
+    results.clear();
+    if (!m_file || !m_channelCount) return;
+    if (count == 0) return;
+
+    if ((long)start >= m_fileInfo.frames) {
+	return;
+    }
+
+    if (long(start + count) > m_fileInfo.frames) {
+	count = m_fileInfo.frames - start;
+    }
+
+    sf_count_t readCount = 0;
+
+    m_mutex.lock();
+
+    if (start != m_lastStart || count != m_lastCount) {
+
+	if (sf_seek(m_file, start, SEEK_SET) < 0) {
+	    m_mutex.unlock();
+	    return;
+	}
+	
+	if (count * m_fileInfo.channels > m_bufsiz) {
+//	    std::cerr << "WavFileReader: Reallocating buffer for " << count
+//		      << " frames, " << m_fileInfo.channels << " channels: "
+//		      << m_bufsiz << " floats" << std::endl;
+	    m_bufsiz = count * m_fileInfo.channels;
+	    delete[] m_buffer;
+	    m_buffer = new float[m_bufsiz];
+	}
+	
+	if ((readCount = sf_readf_float(m_file, m_buffer, count)) < 0) {
+	    m_mutex.unlock();
+	    return;
+	}
+
+	m_lastStart = start;
+	m_lastCount = readCount;
+    }
+
+    for (size_t i = 0; i < count * m_fileInfo.channels; ++i) {
+	results.push_back(m_buffer[i]);
+    }
+
+    m_mutex.unlock();
+    return;
+}
+