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 cannam@16: #include 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: