annotate transform/RealTimePluginTransform.cpp @ 39:f18093617b78

* Introduce WritableWaveFileModel, and use it as an output model for audio real-time plugin transforms. Updates aren't working correctly yet.
author Chris Cannam
date Tue, 03 Oct 2006 14:17:37 +0000
parents 8ad306d8a568
children 75c5951cf9d7
rev   line source
Chris@0 1
Chris@0 2 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@0 3
Chris@0 4 /*
Chris@0 5 Sonic Visualiser
Chris@0 6 An audio file viewer and annotation editor.
Chris@0 7 Centre for Digital Music, Queen Mary, University of London.
Chris@0 8 This file copyright 2006 Chris Cannam.
Chris@0 9
Chris@0 10 This program is free software; you can redistribute it and/or
Chris@0 11 modify it under the terms of the GNU General Public License as
Chris@0 12 published by the Free Software Foundation; either version 2 of the
Chris@0 13 License, or (at your option) any later version. See the file
Chris@0 14 COPYING included with this distribution for more information.
Chris@0 15 */
Chris@0 16
Chris@0 17 #include "RealTimePluginTransform.h"
Chris@0 18
Chris@0 19 #include "plugin/RealTimePluginFactory.h"
Chris@0 20 #include "plugin/RealTimePluginInstance.h"
Chris@0 21 #include "plugin/PluginXml.h"
Chris@0 22
Chris@1 23 #include "data/model/Model.h"
Chris@1 24 #include "data/model/SparseTimeValueModel.h"
Chris@1 25 #include "data/model/DenseTimeValueModel.h"
Chris@39 26 #include "data/model/WritableWaveFileModel.h"
Chris@0 27
Chris@0 28 #include <iostream>
Chris@0 29
Chris@0 30 RealTimePluginTransform::RealTimePluginTransform(Model *inputModel,
Chris@0 31 QString pluginId,
Chris@27 32 const ExecutionContext &context,
Chris@0 33 QString configurationXml,
Chris@0 34 QString units,
Chris@27 35 int output) :
Chris@27 36 PluginTransform(inputModel, context),
Chris@0 37 m_plugin(0),
Chris@27 38 m_outputNo(output)
Chris@0 39 {
Chris@27 40 if (!m_context.blockSize) m_context.blockSize = 1024;
Chris@26 41
Chris@0 42 std::cerr << "RealTimePluginTransform::RealTimePluginTransform: plugin " << pluginId.toStdString() << ", output " << output << std::endl;
Chris@0 43
Chris@0 44 RealTimePluginFactory *factory =
Chris@0 45 RealTimePluginFactory::instanceFor(pluginId);
Chris@0 46
Chris@0 47 if (!factory) {
Chris@0 48 std::cerr << "RealTimePluginTransform: No factory available for plugin id \""
Chris@0 49 << pluginId.toStdString() << "\"" << std::endl;
Chris@0 50 return;
Chris@0 51 }
Chris@0 52
Chris@0 53 DenseTimeValueModel *input = getInput();
Chris@0 54 if (!input) return;
Chris@0 55
Chris@0 56 m_plugin = factory->instantiatePlugin(pluginId, 0, 0, m_input->getSampleRate(),
Chris@27 57 m_context.blockSize,
Chris@0 58 input->getChannelCount());
Chris@0 59
Chris@0 60 if (!m_plugin) {
Chris@0 61 std::cerr << "RealTimePluginTransform: Failed to instantiate plugin \""
Chris@0 62 << pluginId.toStdString() << "\"" << std::endl;
Chris@0 63 return;
Chris@0 64 }
Chris@0 65
Chris@0 66 if (configurationXml != "") {
Chris@0 67 PluginXml(m_plugin).setParametersFromXml(configurationXml);
Chris@0 68 }
Chris@0 69
Chris@34 70 if (m_outputNo >= 0 && m_outputNo >= m_plugin->getControlOutputCount()) {
Chris@0 71 std::cerr << "RealTimePluginTransform: Plugin has fewer than desired " << m_outputNo << " control outputs" << std::endl;
Chris@0 72 return;
Chris@0 73 }
Chris@34 74
Chris@34 75 if (m_outputNo == -1) {
Chris@34 76
Chris@39 77 WritableWaveFileModel *model = new WritableWaveFileModel
Chris@39 78 (input->getSampleRate(), input->getChannelCount()); //!!!
Chris@39 79
Chris@39 80 m_output = model;
Chris@34 81
Chris@34 82 } else {
Chris@0 83
Chris@34 84 SparseTimeValueModel *model = new SparseTimeValueModel
Chris@34 85 (input->getSampleRate(), m_context.blockSize, 0.0, 0.0, false);
Chris@0 86
Chris@34 87 if (units != "") model->setScaleUnits(units);
Chris@0 88
Chris@34 89 m_output = model;
Chris@34 90 }
Chris@0 91 }
Chris@0 92
Chris@0 93 RealTimePluginTransform::~RealTimePluginTransform()
Chris@0 94 {
Chris@0 95 delete m_plugin;
Chris@0 96 }
Chris@0 97
Chris@0 98 DenseTimeValueModel *
Chris@0 99 RealTimePluginTransform::getInput()
Chris@0 100 {
Chris@0 101 DenseTimeValueModel *dtvm =
Chris@0 102 dynamic_cast<DenseTimeValueModel *>(getInputModel());
Chris@0 103 if (!dtvm) {
Chris@0 104 std::cerr << "RealTimePluginTransform::getInput: WARNING: Input model is not conformable to DenseTimeValueModel" << std::endl;
Chris@0 105 }
Chris@0 106 return dtvm;
Chris@0 107 }
Chris@0 108
Chris@0 109 void
Chris@0 110 RealTimePluginTransform::run()
Chris@0 111 {
Chris@0 112 DenseTimeValueModel *input = getInput();
Chris@0 113 if (!input) return;
Chris@0 114
Chris@39 115 SparseTimeValueModel *stvm = dynamic_cast<SparseTimeValueModel *>(m_output);
Chris@39 116 WritableWaveFileModel *wwfm = dynamic_cast<WritableWaveFileModel *>(m_output);
Chris@39 117 if (!stvm && !wwfm) return;
Chris@0 118
Chris@39 119 if (stvm && (m_outputNo >= m_plugin->getControlOutputCount())) return;
Chris@0 120
Chris@0 121 size_t sampleRate = input->getSampleRate();
Chris@0 122 int channelCount = input->getChannelCount();
Chris@27 123 if (m_context.channel != -1) channelCount = 1;
Chris@0 124
Chris@0 125 size_t blockSize = m_plugin->getBufferSize();
Chris@0 126
Chris@0 127 float **buffers = m_plugin->getAudioInputBuffers();
Chris@0 128
Chris@0 129 size_t startFrame = m_input->getStartFrame();
Chris@0 130 size_t endFrame = m_input->getEndFrame();
Chris@0 131 size_t blockFrame = startFrame;
Chris@0 132
Chris@0 133 size_t prevCompletion = 0;
Chris@0 134
Chris@39 135 size_t latency = m_plugin->getLatency();
Chris@39 136
Chris@0 137 int i = 0;
Chris@0 138
Chris@0 139 while (blockFrame < endFrame) {
Chris@0 140
Chris@0 141 size_t completion =
Chris@0 142 (((blockFrame - startFrame) / blockSize) * 99) /
Chris@0 143 ( (endFrame - startFrame) / blockSize);
Chris@0 144
Chris@0 145 size_t got = 0;
Chris@0 146
Chris@0 147 if (channelCount == 1) {
Chris@0 148 got = input->getValues
Chris@27 149 (m_context.channel, blockFrame, blockFrame + blockSize, buffers[0]);
Chris@0 150 while (got < blockSize) {
Chris@0 151 buffers[0][got++] = 0.0;
Chris@0 152 }
Chris@27 153 if (m_context.channel == -1 && channelCount > 1) {
Chris@0 154 // use mean instead of sum, as plugin input
Chris@0 155 for (size_t i = 0; i < got; ++i) {
Chris@0 156 buffers[0][i] /= channelCount;
Chris@0 157 }
Chris@0 158 }
Chris@0 159 } else {
Chris@0 160 for (size_t ch = 0; ch < channelCount; ++ch) {
Chris@0 161 got = input->getValues
Chris@0 162 (ch, blockFrame, blockFrame + blockSize, buffers[ch]);
Chris@0 163 while (got < blockSize) {
Chris@0 164 buffers[ch][got++] = 0.0;
Chris@0 165 }
Chris@0 166 }
Chris@0 167 }
Chris@0 168
Chris@0 169 m_plugin->run(Vamp::RealTime::frame2RealTime(blockFrame, sampleRate));
Chris@0 170
Chris@39 171 if (stvm) {
Chris@0 172
Chris@39 173 float value = m_plugin->getControlOutputValue(m_outputNo);
Chris@39 174
Chris@39 175 size_t pointFrame = blockFrame;
Chris@39 176 if (pointFrame > latency) pointFrame -= latency;
Chris@39 177 else pointFrame = 0;
Chris@39 178
Chris@39 179 stvm->addPoint(SparseTimeValueModel::Point
Chris@39 180 (pointFrame, value, ""));
Chris@39 181
Chris@39 182 } else if (wwfm) {
Chris@39 183
Chris@39 184 float **buffers = m_plugin->getAudioOutputBuffers();
Chris@39 185
Chris@39 186 if (blockFrame >= latency) {
Chris@39 187 wwfm->addSamples(buffers, blockSize);
Chris@39 188 } else if (blockFrame + blockSize >= latency) {
Chris@39 189 size_t offset = latency - blockFrame;
Chris@39 190 size_t count = blockSize - offset;
Chris@39 191 float **tmp = new float *[channelCount];
Chris@39 192 for (size_t c = 0; c < channelCount; ++c) {
Chris@39 193 tmp[c] = buffers[c] + offset;
Chris@39 194 }
Chris@39 195 wwfm->addSamples(tmp, count);
Chris@39 196 delete[] tmp;
Chris@39 197 }
Chris@39 198 }
Chris@0 199
Chris@0 200 if (blockFrame == startFrame || completion > prevCompletion) {
Chris@39 201 if (stvm) stvm->setCompletion(completion);
Chris@0 202 prevCompletion = completion;
Chris@0 203 }
Chris@0 204
Chris@0 205 blockFrame += blockSize;
Chris@0 206 }
Chris@0 207
Chris@39 208 if (stvm) stvm->setCompletion(100);
Chris@39 209 if (wwfm) wwfm->sync();
Chris@0 210 }
Chris@0 211