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 "WavFileWriter.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "model/DenseTimeValueModel.h"
lbajardsilogic@0: #include "base/Selection.h"
lbajardsilogic@0: #include "system/System.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <QFileInfo>
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: 
lbajardsilogic@0: WavFileWriter::WavFileWriter(QString path,
lbajardsilogic@0: 			     size_t sampleRate,
lbajardsilogic@0:                              size_t channels) :
lbajardsilogic@0:     m_path(path),
lbajardsilogic@0:     m_sampleRate(sampleRate),
lbajardsilogic@0:     m_channels(channels),
lbajardsilogic@0:     m_file(0)
lbajardsilogic@0: {
lbajardsilogic@0:     SF_INFO fileInfo;
lbajardsilogic@0:     fileInfo.samplerate = m_sampleRate;
lbajardsilogic@0:     fileInfo.channels = m_channels;
lbajardsilogic@0:     fileInfo.format = SF_FORMAT_WAV | SF_FORMAT_FLOAT;
lbajardsilogic@0:     
lbajardsilogic@0:     m_file = sf_open(m_path.toLocal8Bit(), SFM_WRITE, &fileInfo);
lbajardsilogic@0:     if (!m_file) {
lbajardsilogic@0: 	std::cerr << "WavFileWriter: Failed to open file ("
lbajardsilogic@0: 		  << sf_strerror(m_file) << ")" << std::endl;
lbajardsilogic@0: 	m_error = QString("Failed to open audio file '%1' for writing")
lbajardsilogic@0: 	    .arg(m_path);
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: WavFileWriter::~WavFileWriter()
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_file) close();
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: WavFileWriter::isOK() const
lbajardsilogic@0: {
lbajardsilogic@0:     return (m_error.isEmpty());
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: WavFileWriter::getError() const
lbajardsilogic@0: {
lbajardsilogic@0:     return m_error;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: WavFileWriter::writeModel(DenseTimeValueModel *source,
lbajardsilogic@0:                           MultiSelection *selection)
lbajardsilogic@0: {
lbajardsilogic@0:     if (source->getChannelCount() != m_channels) {
lbajardsilogic@0:         std::cerr << "WavFileWriter::writeModel: Wrong number of channels ("
lbajardsilogic@0:                   << source->getChannelCount()  << " != " << m_channels << ")"
lbajardsilogic@0:                   << std::endl;
lbajardsilogic@0:         m_error = QString("Failed to write model to audio file '%1'")
lbajardsilogic@0:             .arg(m_path);
lbajardsilogic@0:         return false;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_file) {
lbajardsilogic@0:         m_error = QString("Failed to write model to audio file '%1': File not open")
lbajardsilogic@0:             .arg(m_path);
lbajardsilogic@0: 	return false;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     bool ownSelection = false;
lbajardsilogic@0:     if (!selection) {
lbajardsilogic@0: 	selection = new MultiSelection;
lbajardsilogic@0: 	selection->setSelection(Selection(source->getStartFrame(),
lbajardsilogic@0: 					  source->getEndFrame()));
lbajardsilogic@0:         ownSelection = true;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     size_t bs = 2048;
lbajardsilogic@0:     float *ub = new float[bs]; // uninterleaved buffer (one channel)
lbajardsilogic@0:     float *ib = new float[bs * m_channels]; // interleaved buffer
lbajardsilogic@0: 
lbajardsilogic@0:     for (MultiSelection::SelectionList::const_iterator i =
lbajardsilogic@0: 	     selection->getSelections().begin();
lbajardsilogic@0: 	 i != selection->getSelections().end(); ++i) {
lbajardsilogic@0: 	
lbajardsilogic@0: 	size_t f0(i->getStartFrame()), f1(i->getEndFrame());
lbajardsilogic@0: 
lbajardsilogic@0: 	for (size_t f = f0; f < f1; f += bs) {
lbajardsilogic@0: 	    
lbajardsilogic@191: 	    size_t n = MIN(bs, f1 - f);
lbajardsilogic@0: 
lbajardsilogic@0: 	    for (int c = 0; c < int(m_channels); ++c) {
lbajardsilogic@0: 		source->getValues(c, f, f + n, ub);
lbajardsilogic@0: 		for (size_t i = 0; i < n; ++i) {
lbajardsilogic@0: 		    ib[i * m_channels + c] = ub[i];
lbajardsilogic@0: 		}
lbajardsilogic@0: 	    }	    
lbajardsilogic@0: 
lbajardsilogic@0: 	    sf_count_t written = sf_writef_float(m_file, ib, n);
lbajardsilogic@0: 
lbajardsilogic@0: 	    if (written < n) {
lbajardsilogic@0: 		m_error = QString("Only wrote %1 of %2 frames at file frame %3")
lbajardsilogic@0: 		    .arg(written).arg(n).arg(f);
lbajardsilogic@0: 		break;
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     delete[] ub;
lbajardsilogic@0:     delete[] ib;
lbajardsilogic@0:     if (ownSelection) delete selection;
lbajardsilogic@0: 
lbajardsilogic@0:     return isOK();
lbajardsilogic@0: }
lbajardsilogic@0: 	
lbajardsilogic@0: bool
lbajardsilogic@0: WavFileWriter::writeSamples(float **samples, size_t count)
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_file) {
lbajardsilogic@0:         m_error = QString("Failed to write model to audio file '%1': File not open")
lbajardsilogic@0:             .arg(m_path);
lbajardsilogic@0: 	return false;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     float *b = new float[count * m_channels];
lbajardsilogic@0:     for (size_t i = 0; i < count; ++i) {
lbajardsilogic@0:         for (size_t c = 0; c < m_channels; ++c) {
lbajardsilogic@0:             b[i * m_channels + c] = samples[c][i];
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     sf_count_t written = sf_writef_float(m_file, b, count);
lbajardsilogic@0: 
lbajardsilogic@0:     delete[] b;
lbajardsilogic@0: 
lbajardsilogic@0:     if (written < count) {
lbajardsilogic@0:         m_error = QString("Only wrote %1 of %2 frames")
lbajardsilogic@0:             .arg(written).arg(count);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     return isOK();
lbajardsilogic@0: }
lbajardsilogic@0:     
lbajardsilogic@0: bool
lbajardsilogic@0: WavFileWriter::close()
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_file) {
lbajardsilogic@0:         sf_close(m_file);
lbajardsilogic@0:         m_file = 0;
lbajardsilogic@0:     }
lbajardsilogic@0:     return true;
lbajardsilogic@0: }
lbajardsilogic@0: