comparison transform/FeatureExtractionPluginTransform.cpp @ 67:eb530055ed55

* Add support for frequency-domain inputs to Vamp plugins

 #include "plugin/FeatureExtractionPluginFactory.h"
 #include "plugin/PluginXml.h"
 #include "vamp-sdk/Plugin.h"
 
 #include "base/Model.h"
+#include "base/Window.h"
 #include "model/SparseOneDimensionalModel.h"
 #include "model/SparseTimeValueModel.h"
 #include "model/DenseThreeDimensionalModel.h"
 #include "model/DenseTimeValueModel.h"
+
+#include <fftw3.h>
 
 #include <iostream>
 
 FeatureExtractionPluginTransform::FeatureExtractionPluginTransform(Model *inputModel,
 								   QString pluginId,

     m_plugin->initialise(channelCount, stepSize, blockSize);
 
     float **buffers = new float*[channelCount];
     for (size_t ch = 0; ch < channelCount; ++ch) {
 	buffers[ch] = new float[blockSize];
+    }
+
+    double *fftInput = 0;
+    fftw_complex *fftOutput = 0;
+    fftw_plan fftPlan = 0;
+    Window<double> windower(HanningWindow, blockSize);
+
+    if (m_plugin->getInputDomain() == Vamp::Plugin::FrequencyDomain) {
+
+        fftInput = (double *)fftw_malloc(blockSize * sizeof(double));
+        fftOutput = (fftw_complex *)fftw_malloc(blockSize * sizeof(fftw_complex));
+        fftPlan = fftw_plan_dft_r2c_1d(blockSize, fftInput, fftOutput,
+                                       FFTW_ESTIMATE);
+        if (!fftPlan) {
+            std::cerr << "ERROR: FeatureExtractionPluginTransform::run(): fftw_plan failed! Results will be garbage" << std::endl;
+        }
     }
 
     size_t startFrame = m_input->getStartFrame();
     size_t   endFrame = m_input->getEndFrame();
     size_t blockFrame = startFrame;

 		    buffers[ch][got++] = 0.0;
 		}
 	    }
 	}
 
+        if (fftPlan) {
+            for (size_t ch = 0; ch < channelCount; ++ch) {
+                for (size_t i = 0; i < blockSize; ++i) {
+                    fftInput[i] = buffers[ch][i];
+                }
+                windower.cut(fftInput);
+                for (size_t i = 0; i < blockSize/2; ++i) {
+                    double temp = fftInput[i];
+                    fftInput[i] = fftInput[i + blockSize/2];
+                    fftInput[i + blockSize/2] = temp;
+                }
+                fftw_execute(fftPlan);
+                for (size_t i = 0; i < blockSize/2; ++i) {
+                    buffers[ch][i*2] = fftOutput[i][0];
+                    buffers[ch][i*2 + 1] = fftOutput[i][1];
+                }
+            }
+        }
+
 	Vamp::Plugin::FeatureSet features = m_plugin->process
 	    (buffers, Vamp::RealTime::frame2RealTime(blockFrame, sampleRate));
 
 	for (size_t fi = 0; fi < features[m_outputFeatureNo].size(); ++fi) {
 	    Vamp::Plugin::Feature feature =

 	    setCompletion(completion);
 	    prevCompletion = completion;
 	}
 
 	blockFrame += stepSize;
+    }
+
+    if (fftPlan) {
+        fftw_destroy_plan(fftPlan);
+        fftw_free(fftInput);
+        fftw_free(fftOutput);
     }
 
     Vamp::Plugin::FeatureSet features = m_plugin->getRemainingFeatures();
 
     for (size_t fi = 0; fi < features[m_outputFeatureNo].size(); ++fi) {