cannam@1: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
cannam@1: 
cannam@1: /*
cannam@1:     Vamp
cannam@1: 
cannam@1:     An API for audio analysis and feature extraction plugins.
cannam@1: 
cannam@1:     Centre for Digital Music, Queen Mary, University of London.
cannam@1:     Copyright 2006 Chris Cannam.
cannam@16:     FFT code from Don Cross's public domain FFT implementation.
cannam@1:   
cannam@1:     Permission is hereby granted, free of charge, to any person
cannam@1:     obtaining a copy of this software and associated documentation
cannam@1:     files (the "Software"), to deal in the Software without
cannam@1:     restriction, including without limitation the rights to use, copy,
cannam@1:     modify, merge, publish, distribute, sublicense, and/or sell copies
cannam@1:     of the Software, and to permit persons to whom the Software is
cannam@1:     furnished to do so, subject to the following conditions:
cannam@1: 
cannam@1:     The above copyright notice and this permission notice shall be
cannam@1:     included in all copies or substantial portions of the Software.
cannam@1: 
cannam@1:     THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
cannam@1:     EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
cannam@1:     MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
cannam@6:     NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR
cannam@1:     ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
cannam@1:     CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
cannam@1:     WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
cannam@1: 
cannam@1:     Except as contained in this notice, the names of the Centre for
cannam@1:     Digital Music; Queen Mary, University of London; and Chris Cannam
cannam@1:     shall not be used in advertising or otherwise to promote the sale,
cannam@1:     use or other dealings in this Software without prior written
cannam@1:     authorization.
cannam@1: */
cannam@1: 
cannam@16: #include "PluginHostAdapter.h"
cannam@1: #include "vamp.h"
cannam@1: 
cannam@16: #include <iostream>
cannam@16: #include <sndfile.h>
cannam@1: 
cannam@1: #include "system.h"
cannam@1: 
cannam@16: using std::cout;
cannam@16: using std::cerr;
cannam@16: using std::endl;
cannam@16: using std::string;
cannam@16: 
cannam@16: void printFeatures(int, int, int, Vamp::Plugin::FeatureSet);
cannam@16: void transformInput(float *, size_t);
cannam@16: void fft(unsigned int, bool, double *, double *, double *, double *);
cannam@16: 
cannam@1: /*
cannam@16:     A very simple Vamp plugin host.  Given the name of a plugin
cannam@16:     library and the name of a sound file on the command line, it loads
cannam@16:     the first plugin in the library and runs it on the sound file,
cannam@16:     dumping the plugin's first output to stdout.
cannam@1: */
cannam@1: 
cannam@1: int main(int argc, char **argv)
cannam@1: {
cannam@16:     if (argc < 2 || argc > 4) {
cannam@16:         cerr << "Usage: " << argv[0] << " pluginlibrary.so[:plugin] [file.wav] [outputno]" << endl;
cannam@1:         return 2;
cannam@1:     }
cannam@1: 
cannam@16:     cerr << endl << argv[0] << ": Running..." << endl;
cannam@1: 
cannam@16:     string soname = argv[1];
cannam@16:     string plugname = "";
cannam@16:     string wavname;
cannam@16:     if (argc >= 3) wavname = argv[2];
cannam@16: 
cannam@16:     if (soname.find(":") >= 0) {
cannam@16:         plugname = soname.substr(soname.find(":") + 1);
cannam@16:         soname = soname.substr(0, soname.find(":"));
cannam@16:     }
cannam@1: 
cannam@1:     void *libraryHandle = DLOPEN(soname, RTLD_LAZY);
cannam@1: 
cannam@1:     if (!libraryHandle) {
cannam@16:         cerr << argv[0] << ": Failed to open plugin library " 
cannam@16:                   << soname << ": " << DLERROR() << endl;
cannam@1:         return 1;
cannam@1:     }
cannam@1: 
cannam@16:     cerr << argv[0] << ": Opened plugin library " << soname << endl;
cannam@1: 
cannam@1:     VampGetPluginDescriptorFunction fn = (VampGetPluginDescriptorFunction)
cannam@1:         DLSYM(libraryHandle, "vampGetPluginDescriptor");
cannam@1:     
cannam@1:     if (!fn) {
cannam@16:         cerr << argv[0] << ": No Vamp descriptor function in library "
cannam@16:                   << soname << endl;
cannam@1:         DLCLOSE(libraryHandle);
cannam@1:         return 1;
cannam@1:     }
cannam@1: 
cannam@16:     cerr << argv[0] << ": Found plugin descriptor function" << endl;
cannam@1: 
cannam@1:     int index = 0;
cannam@16:     int plugnumber = -1;
cannam@1:     const VampPluginDescriptor *descriptor = 0;
cannam@1: 
cannam@1:     while ((descriptor = fn(index))) {
cannam@1: 
cannam@16:         Vamp::PluginHostAdapter plugin(descriptor, 48000);
cannam@16:         cerr << argv[0] << ": Plugin " << (index+1)
cannam@16:                   << " is \"" << plugin.getName() << "\"" << endl;
cannam@16: 
cannam@16:         if (plugin.getName() == plugname) plugnumber = index;
cannam@1:         
cannam@1:         ++index;
cannam@1:     }
cannam@1: 
cannam@16:     cerr << argv[0] << ": Done\n" << endl;
cannam@16: 
cannam@16:     if (wavname == "") {
cannam@16:         DLCLOSE(libraryHandle);
cannam@16:         return 0;
cannam@16:     }
cannam@16: 
cannam@16:     if (plugnumber < 0) {
cannam@16:         if (plugname != "") {
cannam@16:             cerr << "ERROR: No such plugin as " << plugname << " in library"
cannam@16:                  << endl;
cannam@16:             DLCLOSE(libraryHandle);
cannam@16:             return 0;
cannam@16:         } else {
cannam@16:             plugnumber = 0;
cannam@16:         }
cannam@16:     }
cannam@16: 
cannam@16:     descriptor = fn(plugnumber);
cannam@16:     if (!descriptor) {
cannam@16:         DLCLOSE(libraryHandle);
cannam@16:         return 0;
cannam@16:     }
cannam@16:     
cannam@16:     SNDFILE *sndfile;
cannam@16:     SF_INFO sfinfo;
cannam@16:     memset(&sfinfo, 0, sizeof(SF_INFO));
cannam@16: 
cannam@16:     sndfile = sf_open(wavname.c_str(), SFM_READ, &sfinfo);
cannam@16:     if (!sndfile) {
cannam@16: 	cerr << "ERROR: Failed to open input file \"" << wavname << "\": "
cannam@16: 	     << sf_strerror(sndfile) << endl;
cannam@16:         DLCLOSE(libraryHandle);
cannam@16: 	return 1;
cannam@16:     }
cannam@16: 
cannam@16:     Vamp::PluginHostAdapter *plugin =
cannam@16:         new Vamp::PluginHostAdapter(descriptor, sfinfo.samplerate);
cannam@16: 
cannam@16:     cerr << "Running " << plugin->getName() << "..." << endl;
cannam@16: 
cannam@16:     int blockSize = plugin->getPreferredBlockSize();
cannam@16:     int stepSize = plugin->getPreferredStepSize();
cannam@16: 
cannam@16:     cerr << "Preferred block size = " << blockSize << ", step size = "
cannam@16:               << stepSize << endl;
cannam@16: 
cannam@16:     if (blockSize == 0) blockSize = 1024;
cannam@16:     if (stepSize == 0) stepSize = blockSize;
cannam@16: 
cannam@16:     int channels = sfinfo.channels;
cannam@16: 
cannam@16:     float *filebuf = new float[blockSize * channels];
cannam@16:     float **plugbuf = new float*[channels];
cannam@16:     for (int c = 0; c < channels; ++c) plugbuf[c] = new float[blockSize];
cannam@16: 
cannam@16:     cerr << "Using block size = " << blockSize << ", step size = "
cannam@16:               << stepSize << endl;
cannam@16: 
cannam@16:     int minch = plugin->getMinChannelCount();
cannam@16:     int maxch = plugin->getMaxChannelCount();
cannam@16:     cerr << "Plugin accepts " << minch << " -> " << maxch << " channel(s)" << endl;
cannam@16: 
cannam@16:     Vamp::Plugin::OutputList outputs = plugin->getOutputDescriptors();
cannam@16:     Vamp::Plugin::OutputDescriptor od;
cannam@16: 
cannam@16:     int output = 0;
cannam@16:     if (argc == 4) output = atoi(argv[3]);
cannam@16: 
cannam@16:     bool mix = false;
cannam@16: 
cannam@16:     if (minch > channels || maxch < channels) {
cannam@16:         if (minch == 1) {
cannam@16:             cerr << "WARNING: Sound file has " << channels << " channels, mixing down to 1" << endl;
cannam@16:             mix = true;
cannam@16:             channels = 1;
cannam@16:         } else {
cannam@16:             cerr << "ERROR: Sound file has " << channels << " channels, out of range for plugin" << endl;
cannam@16:             goto done;
cannam@16:         }
cannam@16:     }
cannam@16: 
cannam@16:     if (outputs.empty()) {
cannam@16: 	cerr << "Plugin has no outputs!" << endl;
cannam@16:         goto done;
cannam@16:     }
cannam@16: 
cannam@16:     if (int(outputs.size()) <= output) {
cannam@16: 	cerr << "Output " << output << " requested, but plugin has only " << outputs.size() << " output(s)" << endl;
cannam@16:         goto done;
cannam@16:     }        
cannam@16: 
cannam@16:     od = outputs[output];
cannam@16:     cerr << "Output is " << od.name << endl;
cannam@16: 
cannam@16:     plugin->initialise(channels, stepSize, blockSize);
cannam@16: 
cannam@16:     for (size_t i = 0; i < sfinfo.frames; i += stepSize) {
cannam@16: 
cannam@16:         int count;
cannam@16: 
cannam@16:         if (sf_seek(sndfile, i, SEEK_SET) < 0) {
cannam@16:             cerr << "ERROR: sf_seek failed: " << sf_strerror(sndfile) << endl;
cannam@16:             break;
cannam@16:         }
cannam@16:         
cannam@16:         if ((count = sf_readf_float(sndfile, filebuf, blockSize)) < 0) {
cannam@16:             cerr << "ERROR: sf_readf_float failed: " << sf_strerror(sndfile) << endl;
cannam@16:             break;
cannam@16:         }
cannam@16: 
cannam@16:         for (int c = 0; c < channels; ++c) {
cannam@16:             for (int j = 0; j < blockSize; ++j) {
cannam@16:                 plugbuf[c][j] = 0.0f;
cannam@16:             }
cannam@16:         }
cannam@16: 
cannam@16:         for (int c = 0; c < sfinfo.channels; ++c) {
cannam@16:             int tc = c;
cannam@16:             if (mix) tc = 0;
cannam@16:             for (int j = 0; j < blockSize && j < count; ++j) {
cannam@16:                 plugbuf[tc][j] += filebuf[j * channels + c];
cannam@16:             }
cannam@16: 
cannam@16:             if (plugin->getInputDomain() == Vamp::Plugin::FrequencyDomain) {
cannam@16:                 transformInput(plugbuf[tc], blockSize);
cannam@16:             }
cannam@16:         }
cannam@16: 
cannam@16:         printFeatures
cannam@16:             (i, sfinfo.samplerate, output, plugin->process
cannam@16:              (plugbuf, Vamp::RealTime::frame2RealTime(i, sfinfo.samplerate)));
cannam@16:     }
cannam@16: 
cannam@16:     printFeatures(sfinfo.frames, sfinfo.samplerate, output,
cannam@16:                   plugin->getRemainingFeatures());
cannam@16: 
cannam@16: done:
cannam@16:     delete plugin;
cannam@1: 
cannam@1:     DLCLOSE(libraryHandle);
cannam@16:     sf_close(sndfile);
cannam@1:     return 0;
cannam@1: }
cannam@1: 
cannam@16: void
cannam@16: printFeatures(int frame, int sr, int output, Vamp::Plugin::FeatureSet features)
cannam@16: {
cannam@16:     for (unsigned int i = 0; i < features[output].size(); ++i) {
cannam@16:         Vamp::RealTime rt = Vamp::RealTime::frame2RealTime(frame, sr);
cannam@16:         if (features[output][i].hasTimestamp) {
cannam@16:             rt = features[output][i].timestamp;
cannam@16:         }
cannam@16:         cout << rt.toString() << ":";
cannam@16:         for (unsigned int j = 0; j < features[output][i].values.size(); ++j) {
cannam@16:             cout << " " << features[output][i].values[j];
cannam@16:         }
cannam@16:         cout << endl;
cannam@16:     }
cannam@16: }
cannam@16: 
cannam@16: void
cannam@16: transformInput(float *buffer, size_t size)
cannam@16: {
cannam@16:     double *inbuf = new double[size * 2];
cannam@16:     double *outbuf = new double[size * 2];
cannam@16: 
cannam@16:     // Copy across with Hanning window
cannam@16:     for (size_t i = 0; i < size; ++i) {
cannam@16:         inbuf[i] = double(buffer[i]) * (0.50 - 0.50 * cos(2 * M_PI * i / size));
cannam@16:         inbuf[i + size] = 0.0;
cannam@16:     }
cannam@16:     
cannam@16:     for (size_t i = 0; i < size/2; ++i) {
cannam@16:         double temp = inbuf[i];
cannam@16:         inbuf[i] = inbuf[i + size/2];
cannam@16:         inbuf[i + size/2] = temp;
cannam@16:     }
cannam@16: 
cannam@16:     fft(size, false, inbuf, inbuf + size, outbuf, outbuf + size);
cannam@16: 
cannam@16:     for (size_t i = 0; i < size/2; ++i) {
cannam@16:         buffer[i * 2] = outbuf[i];
cannam@16:         buffer[i * 2 + 1] = outbuf[i + size];
cannam@16:     }
cannam@16:     
cannam@16:     delete inbuf;
cannam@16:     delete outbuf;
cannam@16: }
cannam@16: 
cannam@16: void
cannam@16: fft(unsigned int n, bool inverse, double *ri, double *ii, double *ro, double *io)
cannam@16: {
cannam@16:     if (!ri || !ro || !io) return;
cannam@16: 
cannam@16:     unsigned int bits;
cannam@16:     unsigned int i, j, k, m;
cannam@16:     unsigned int blockSize, blockEnd;
cannam@16: 
cannam@16:     double tr, ti;
cannam@16: 
cannam@16:     if (n < 2) return;
cannam@16:     if (n & (n-1)) return;
cannam@16: 
cannam@16:     double angle = 2.0 * M_PI;
cannam@16:     if (inverse) angle = -angle;
cannam@16: 
cannam@16:     for (i = 0; ; ++i) {
cannam@16: 	if (n & (1 << i)) {
cannam@16: 	    bits = i;
cannam@16: 	    break;
cannam@16: 	}
cannam@16:     }
cannam@16: 
cannam@16:     static unsigned int tableSize = 0;
cannam@16:     static int *table = 0;
cannam@16: 
cannam@16:     if (tableSize != n) {
cannam@16: 
cannam@16: 	delete[] table;
cannam@16: 
cannam@16: 	table = new int[n];
cannam@16: 
cannam@16: 	for (i = 0; i < n; ++i) {
cannam@16: 	
cannam@16: 	    m = i;
cannam@16: 
cannam@16: 	    for (j = k = 0; j < bits; ++j) {
cannam@16: 		k = (k << 1) | (m & 1);
cannam@16: 		m >>= 1;
cannam@16: 	    }
cannam@16: 
cannam@16: 	    table[i] = k;
cannam@16: 	}
cannam@16: 
cannam@16: 	tableSize = n;
cannam@16:     }
cannam@16: 
cannam@16:     if (ii) {
cannam@16: 	for (i = 0; i < n; ++i) {
cannam@16: 	    ro[table[i]] = ri[i];
cannam@16: 	    io[table[i]] = ii[i];
cannam@16: 	}
cannam@16:     } else {
cannam@16: 	for (i = 0; i < n; ++i) {
cannam@16: 	    ro[table[i]] = ri[i];
cannam@16: 	    io[table[i]] = 0.0;
cannam@16: 	}
cannam@16:     }
cannam@16: 
cannam@16:     blockEnd = 1;
cannam@16: 
cannam@16:     for (blockSize = 2; blockSize <= n; blockSize <<= 1) {
cannam@16: 
cannam@16: 	double delta = angle / (double)blockSize;
cannam@16: 	double sm2 = -sin(-2 * delta);
cannam@16: 	double sm1 = -sin(-delta);
cannam@16: 	double cm2 = cos(-2 * delta);
cannam@16: 	double cm1 = cos(-delta);
cannam@16: 	double w = 2 * cm1;
cannam@16: 	double ar[3], ai[3];
cannam@16: 
cannam@16: 	for (i = 0; i < n; i += blockSize) {
cannam@16: 
cannam@16: 	    ar[2] = cm2;
cannam@16: 	    ar[1] = cm1;
cannam@16: 
cannam@16: 	    ai[2] = sm2;
cannam@16: 	    ai[1] = sm1;
cannam@16: 
cannam@16: 	    for (j = i, m = 0; m < blockEnd; j++, m++) {
cannam@16: 
cannam@16: 		ar[0] = w * ar[1] - ar[2];
cannam@16: 		ar[2] = ar[1];
cannam@16: 		ar[1] = ar[0];
cannam@16: 
cannam@16: 		ai[0] = w * ai[1] - ai[2];
cannam@16: 		ai[2] = ai[1];
cannam@16: 		ai[1] = ai[0];
cannam@16: 
cannam@16: 		k = j + blockEnd;
cannam@16: 		tr = ar[0] * ro[k] - ai[0] * io[k];
cannam@16: 		ti = ar[0] * io[k] + ai[0] * ro[k];
cannam@16: 
cannam@16: 		ro[k] = ro[j] - tr;
cannam@16: 		io[k] = io[j] - ti;
cannam@16: 
cannam@16: 		ro[j] += tr;
cannam@16: 		io[j] += ti;
cannam@16: 	    }
cannam@16: 	}
cannam@16: 
cannam@16: 	blockEnd = blockSize;
cannam@16:     }
cannam@16: 
cannam@16:     if (inverse) {
cannam@16: 
cannam@16: 	double denom = (double)n;
cannam@16: 
cannam@16: 	for (i = 0; i < n; i++) {
cannam@16: 	    ro[i] /= denom;
cannam@16: 	    io[i] /= denom;
cannam@16: 	}
cannam@16:     }
cannam@16: }
cannam@16: 
cannam@16: