comparison VampTestPlugin.cpp @ 18:014cce47e998

Add an output that returns some actual feature of the input channels (also counting the channels, effectively)

 
 
 #include "VampTestPlugin.h"
 
 #include <sstream>
+#include <cmath>
 
 using std::stringstream;
 
 using Vamp::RealTime;
 
 VampTestPlugin::VampTestPlugin(float inputSampleRate) :
     Plugin(inputSampleRate),
     m_produceOutput(true),
     m_n(0),
+    m_channels(1),
     m_stepSize(0),
     m_blockSize(0)
 {
     for (int i = 0; i < 10; ++i) {
 	m_instants.push_back(RealTime::fromSeconds(1.5 * i));

 }
 
 size_t
 VampTestPlugin::getMaxChannelCount() const
 {
-    return 1;
+    return 10;
 }
 
 VampTestPlugin::ParameterList
 VampTestPlugin::getParameterDescriptors() const
 {

     d.sampleRate = 0;
     d.hasDuration = true;
     m_outputNumbers[d.identifier] = n++;
     list.push_back(d);
 
+    d.identifier = "rmss";
+    d.name = "RMS of Input Channels";
+    d.description = "RMS levels of each input channel";
+    d.unit = "";
+    d.hasFixedBinCount = true;
+    d.binCount = m_channels;
+    d.hasKnownExtents = false;
+    d.isQuantized = false;
+    d.sampleType = OutputDescriptor::OneSamplePerStep;
+    d.hasDuration = false;
+    m_outputNumbers[d.identifier] = n++;
+    list.push_back(d);
+
     return list;
 }
 
 bool
 VampTestPlugin::initialise(size_t channels, size_t stepSize, size_t blockSize)
 {
     if (channels < getMinChannelCount() ||
 	channels > getMaxChannelCount()) return false;
 
+    m_channels = channels;
     m_stepSize = stepSize;
     m_blockSize = blockSize;
 
     return true;
 }

 VampTestPlugin::FeatureSet
 VampTestPlugin::process(const float *const *inputBuffers, RealTime timestamp)
 {
     if (!m_produceOutput) return FeatureSet();
     FeatureSet fs = featuresFrom(timestamp, false);
+
+    Feature f;
+    for (int c = 0; c < m_channels; ++c) {
+	float sum = 0.f;
+	for (int i = 0; i < m_blockSize; ++i) {
+	    sum += inputBuffers[c][i] * inputBuffers[c][i];
+	}
+	float rms = sqrtf(sum / m_blockSize);
+	f.values.push_back(rms);
+    }
+    fs[m_outputNumbers["rmss"]].push_back(f);
+    
     return fs;
 }
 
 VampTestPlugin::FeatureSet
 VampTestPlugin::getRemainingFeatures()