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 "RealTimePluginTransform.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "plugin/RealTimePluginFactory.h"
lbajardsilogic@0: #include "plugin/RealTimePluginInstance.h"
lbajardsilogic@0: #include "plugin/PluginXml.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "data/model/Model.h"
lbajardsilogic@0: #include "data/model/SparseTimeValueModel.h"
lbajardsilogic@0: #include "data/model/DenseTimeValueModel.h"
lbajardsilogic@0: #include "data/model/WritableWaveFileModel.h"
lbajardsilogic@0: #include "data/model/WaveFileModel.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: 
lbajardsilogic@0: RealTimePluginTransform::RealTimePluginTransform(Model *inputModel,
lbajardsilogic@0:                                                  QString pluginId,
lbajardsilogic@0:                                                  const ExecutionContext &context,
lbajardsilogic@0:                                                  QString configurationXml,
lbajardsilogic@0:                                                  QString units,
lbajardsilogic@0:                                                  int output) :
lbajardsilogic@0:     PluginTransform(inputModel, context),
lbajardsilogic@0:     m_pluginId(pluginId),
lbajardsilogic@0:     m_configurationXml(configurationXml),
lbajardsilogic@0:     m_units(units),
lbajardsilogic@0:     m_plugin(0),
lbajardsilogic@0:     m_outputNo(output)
lbajardsilogic@0: {
lbajardsilogic@0:     if (!m_context.blockSize) m_context.blockSize = 1024;
lbajardsilogic@0: 
lbajardsilogic@0: //    std::cerr << "RealTimePluginTransform::RealTimePluginTransform: plugin " << pluginId.toStdString() << ", output " << output << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:     RealTimePluginFactory *factory =
lbajardsilogic@0: 	RealTimePluginFactory::instanceFor(pluginId);
lbajardsilogic@0: 
lbajardsilogic@0:     if (!factory) {
lbajardsilogic@0: 	std::cerr << "RealTimePluginTransform: No factory available for plugin id \""
lbajardsilogic@0: 		  << pluginId.toStdString() << "\"" << std::endl;
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     DenseTimeValueModel *input = getInput();
lbajardsilogic@0:     if (!input) return;
lbajardsilogic@0: 
lbajardsilogic@0:     m_plugin = factory->instantiatePlugin(pluginId, 0, 0,
lbajardsilogic@0:                                           m_input->getSampleRate(),
lbajardsilogic@0:                                           m_context.blockSize,
lbajardsilogic@0:                                           input->getChannelCount());
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_plugin) {
lbajardsilogic@0: 	std::cerr << "RealTimePluginTransform: Failed to instantiate plugin \""
lbajardsilogic@0: 		  << pluginId.toStdString() << "\"" << std::endl;
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (configurationXml != "") {
lbajardsilogic@0:         PluginXml(m_plugin).setParametersFromXml(configurationXml);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_outputNo >= 0 &&
lbajardsilogic@0:         m_outputNo >= int(m_plugin->getControlOutputCount())) {
lbajardsilogic@0:         std::cerr << "RealTimePluginTransform: Plugin has fewer than desired " << m_outputNo << " control outputs" << std::endl;
lbajardsilogic@0:         return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_outputNo == -1) {
lbajardsilogic@0: 
lbajardsilogic@0:         size_t outputChannels = m_plugin->getAudioOutputCount();
lbajardsilogic@0:         if (outputChannels > input->getChannelCount()) {
lbajardsilogic@0:             outputChannels = input->getChannelCount();
lbajardsilogic@0:         }
lbajardsilogic@0: 
lbajardsilogic@0:         WritableWaveFileModel *model = new WritableWaveFileModel
lbajardsilogic@0:             (input->getSampleRate(), outputChannels);
lbajardsilogic@0: 
lbajardsilogic@0:         m_output = model;
lbajardsilogic@0: 
lbajardsilogic@0:     } else {
lbajardsilogic@0: 	
lbajardsilogic@0:         SparseTimeValueModel *model = new SparseTimeValueModel
lbajardsilogic@0:             (input->getSampleRate(), m_context.blockSize, 0.0, 0.0, false);
lbajardsilogic@0: 
lbajardsilogic@0:         if (units != "") model->setScaleUnits(units);
lbajardsilogic@0: 
lbajardsilogic@0:         m_output = model;
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: RealTimePluginTransform::~RealTimePluginTransform()
lbajardsilogic@0: {
lbajardsilogic@0:     delete m_plugin;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: DenseTimeValueModel *
lbajardsilogic@0: RealTimePluginTransform::getInput()
lbajardsilogic@0: {
lbajardsilogic@0:     DenseTimeValueModel *dtvm =
lbajardsilogic@0: 	dynamic_cast<DenseTimeValueModel *>(getInputModel());
lbajardsilogic@0:     if (!dtvm) {
lbajardsilogic@0: 	std::cerr << "RealTimePluginTransform::getInput: WARNING: Input model is not conformable to DenseTimeValueModel" << std::endl;
lbajardsilogic@0:     }
lbajardsilogic@0:     return dtvm;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: RealTimePluginTransform::run()
lbajardsilogic@0: {
lbajardsilogic@0:     DenseTimeValueModel *input = getInput();
lbajardsilogic@0:     if (!input) return;
lbajardsilogic@0: 
lbajardsilogic@0:     while (!input->isReady()) {
lbajardsilogic@0:         if (dynamic_cast<WaveFileModel *>(input)) break; // no need to wait
lbajardsilogic@0:         std::cerr << "RealTimePluginTransform::run: Waiting for input model to be ready..." << std::endl;
lbajardsilogic@0:         sleep(1);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     SparseTimeValueModel *stvm = dynamic_cast<SparseTimeValueModel *>(m_output);
lbajardsilogic@0:     WritableWaveFileModel *wwfm = dynamic_cast<WritableWaveFileModel *>(m_output);
lbajardsilogic@0:     if (!stvm && !wwfm) return;
lbajardsilogic@0: 
lbajardsilogic@0:     if (stvm && (m_outputNo >= int(m_plugin->getControlOutputCount()))) return;
lbajardsilogic@0: 
lbajardsilogic@0:     size_t sampleRate = input->getSampleRate();
lbajardsilogic@0:     size_t channelCount = input->getChannelCount();
lbajardsilogic@0:     if (!wwfm && m_context.channel != -1) channelCount = 1;
lbajardsilogic@0: 
lbajardsilogic@0:     size_t blockSize = m_plugin->getBufferSize();
lbajardsilogic@0: 
lbajardsilogic@0:     float **inbufs = m_plugin->getAudioInputBuffers();
lbajardsilogic@0: 
lbajardsilogic@0:     size_t startFrame = m_input->getStartFrame();
lbajardsilogic@0:     size_t   endFrame = m_input->getEndFrame();
lbajardsilogic@0:     size_t blockFrame = startFrame;
lbajardsilogic@0: 
lbajardsilogic@0:     size_t prevCompletion = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     size_t latency = m_plugin->getLatency();
lbajardsilogic@0: 
lbajardsilogic@0:     while (blockFrame < endFrame && !m_abandoned) {
lbajardsilogic@0: 
lbajardsilogic@0: 	size_t completion =
lbajardsilogic@0: 	    (((blockFrame - startFrame) / blockSize) * 99) /
lbajardsilogic@0: 	    (   (endFrame - startFrame) / blockSize);
lbajardsilogic@0: 
lbajardsilogic@0: 	size_t got = 0;
lbajardsilogic@0: 
lbajardsilogic@0: 	if (channelCount == 1) {
lbajardsilogic@0:             if (inbufs && inbufs[0]) {
lbajardsilogic@0:                 got = input->getValues
lbajardsilogic@0:                     (m_context.channel, blockFrame, blockFrame + blockSize, inbufs[0]);
lbajardsilogic@0:                 while (got < blockSize) {
lbajardsilogic@0:                     inbufs[0][got++] = 0.0;
lbajardsilogic@0:                 }          
lbajardsilogic@0:             }
lbajardsilogic@0:             for (size_t ch = 1; ch < m_plugin->getAudioInputCount(); ++ch) {
lbajardsilogic@0:                 for (size_t i = 0; i < blockSize; ++i) {
lbajardsilogic@0:                     inbufs[ch][i] = inbufs[0][i];
lbajardsilogic@0:                 }
lbajardsilogic@0:             }
lbajardsilogic@0: 	} else {
lbajardsilogic@0: 	    for (size_t ch = 0; ch < channelCount; ++ch) {
lbajardsilogic@0:                 if (inbufs && inbufs[ch]) {
lbajardsilogic@0:                     got = input->getValues
lbajardsilogic@0:                         (ch, blockFrame, blockFrame + blockSize, inbufs[ch]);
lbajardsilogic@0:                     while (got < blockSize) {
lbajardsilogic@0:                         inbufs[ch][got++] = 0.0;
lbajardsilogic@0:                     }
lbajardsilogic@0:                 }
lbajardsilogic@0: 	    }
lbajardsilogic@0:             for (size_t ch = channelCount; ch < m_plugin->getAudioInputCount(); ++ch) {
lbajardsilogic@0:                 for (size_t i = 0; i < blockSize; ++i) {
lbajardsilogic@0:                     inbufs[ch][i] = inbufs[ch % channelCount][i];
lbajardsilogic@0:                 }
lbajardsilogic@0:             }
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: /*
lbajardsilogic@0:         std::cerr << "Input for plugin: " << m_plugin->getAudioInputCount() << " channels "<< std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:         for (size_t ch = 0; ch < m_plugin->getAudioInputCount(); ++ch) {
lbajardsilogic@0:             std::cerr << "Input channel " << ch << std::endl;
lbajardsilogic@0:             for (size_t i = 0; i < 100; ++i) {
lbajardsilogic@0:                 std::cerr << inbufs[ch][i] << " ";
lbajardsilogic@0:                 if (isnan(inbufs[ch][i])) {
lbajardsilogic@0:                     std::cerr << "\n\nWARNING: NaN in audio input" << std::endl;
lbajardsilogic@0:                 }
lbajardsilogic@0:             }
lbajardsilogic@0:         }
lbajardsilogic@0: */
lbajardsilogic@0: 
lbajardsilogic@0:         m_plugin->run(Vamp::RealTime::frame2RealTime(blockFrame, sampleRate));
lbajardsilogic@0: 
lbajardsilogic@0:         if (stvm) {
lbajardsilogic@0: 
lbajardsilogic@0:             float value = m_plugin->getControlOutputValue(m_outputNo);
lbajardsilogic@0: 
lbajardsilogic@0:             size_t pointFrame = blockFrame;
lbajardsilogic@0:             if (pointFrame > latency) pointFrame -= latency;
lbajardsilogic@0:             else pointFrame = 0;
lbajardsilogic@0: 
lbajardsilogic@0:             stvm->addPoint(SparseTimeValueModel::Point
lbajardsilogic@0:                            (pointFrame, value, ""));
lbajardsilogic@0: 
lbajardsilogic@0:         } else if (wwfm) {
lbajardsilogic@0: 
lbajardsilogic@0:             float **outbufs = m_plugin->getAudioOutputBuffers();
lbajardsilogic@0: 
lbajardsilogic@0:             if (outbufs) {
lbajardsilogic@0: 
lbajardsilogic@0:                 if (blockFrame >= latency) {
lbajardsilogic@0:                     wwfm->addSamples(outbufs, blockSize);
lbajardsilogic@0:                 } else if (blockFrame + blockSize >= latency) {
lbajardsilogic@0:                     size_t offset = latency - blockFrame;
lbajardsilogic@0:                     size_t count = blockSize - offset;
lbajardsilogic@0:                     float **tmp = new float *[channelCount];
lbajardsilogic@0:                     for (size_t c = 0; c < channelCount; ++c) {
lbajardsilogic@0:                         tmp[c] = outbufs[c] + offset;
lbajardsilogic@0:                     }
lbajardsilogic@0:                     wwfm->addSamples(tmp, count);
lbajardsilogic@0:                     delete[] tmp;
lbajardsilogic@0:                 }
lbajardsilogic@0:             }
lbajardsilogic@0:         }
lbajardsilogic@0: 
lbajardsilogic@0: 	if (blockFrame == startFrame || completion > prevCompletion) {
lbajardsilogic@0: 	    if (stvm) stvm->setCompletion(completion);
lbajardsilogic@0: 	    if (wwfm) wwfm->setCompletion(completion);
lbajardsilogic@0: 	    prevCompletion = completion;
lbajardsilogic@0: 	}
lbajardsilogic@0:         
lbajardsilogic@0: 	blockFrame += blockSize;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_abandoned) return;
lbajardsilogic@0:     
lbajardsilogic@0:     if (stvm) stvm->setCompletion(100);
lbajardsilogic@0:     if (wwfm) wwfm->setCompletion(100);
lbajardsilogic@0: }
lbajardsilogic@0: