cannam@97: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */ cannam@97: cannam@97: /* cannam@97: Vamp cannam@97: cannam@97: An API for audio analysis and feature extraction plugins. cannam@97: cannam@97: Centre for Digital Music, Queen Mary, University of London. cannam@97: Copyright 2008 QMUL. cannam@97: cannam@97: Permission is hereby granted, free of charge, to any person cannam@97: obtaining a copy of this software and associated documentation cannam@97: files (the "Software"), to deal in the Software without cannam@97: restriction, including without limitation the rights to use, copy, cannam@97: modify, merge, publish, distribute, sublicense, and/or sell copies cannam@97: of the Software, and to permit persons to whom the Software is cannam@97: furnished to do so, subject to the following conditions: cannam@97: cannam@97: The above copyright notice and this permission notice shall be cannam@97: included in all copies or substantial portions of the Software. cannam@97: cannam@97: THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, cannam@97: EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF cannam@97: MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND cannam@97: NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR cannam@97: ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF cannam@97: CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION cannam@97: WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. cannam@97: cannam@97: Except as contained in this notice, the names of the Centre for cannam@97: Digital Music; Queen Mary, University of London; and Chris Cannam cannam@97: shall not be used in advertising or otherwise to promote the sale, cannam@97: use or other dealings in this Software without prior written cannam@97: authorization. cannam@97: */ cannam@97: cannam@97: #include "PowerSpectrum.h" cannam@97: cannam@97: using std::string; cannam@97: using std::cerr; cannam@97: using std::endl; cannam@97: cannam@97: #include cannam@97: cannam@97: PowerSpectrum::PowerSpectrum(float inputSampleRate) : cannam@97: Plugin(inputSampleRate), cannam@97: m_blockSize(0) cannam@97: { cannam@97: } cannam@97: cannam@97: PowerSpectrum::~PowerSpectrum() cannam@97: { cannam@97: } cannam@97: cannam@97: string cannam@97: PowerSpectrum::getIdentifier() const cannam@97: { cannam@97: return "powerspectrum"; cannam@97: } cannam@97: cannam@97: string cannam@97: PowerSpectrum::getName() const cannam@97: { cannam@97: return "Simple Power Spectrum"; cannam@97: } cannam@97: cannam@97: string cannam@97: PowerSpectrum::getDescription() const cannam@97: { cannam@97: return "Return the power spectrum of a signal"; cannam@97: } cannam@97: cannam@97: string cannam@97: PowerSpectrum::getMaker() const cannam@97: { cannam@97: return "Vamp SDK Example Plugins"; cannam@97: } cannam@97: cannam@97: int cannam@97: PowerSpectrum::getPluginVersion() const cannam@97: { cannam@97: return 1; cannam@97: } cannam@97: cannam@97: string cannam@97: PowerSpectrum::getCopyright() const cannam@97: { cannam@97: return "Freely redistributable (BSD license)"; cannam@97: } cannam@97: cannam@97: bool cannam@97: PowerSpectrum::initialise(size_t channels, size_t stepSize, size_t blockSize) cannam@97: { cannam@97: if (channels < getMinChannelCount() || cannam@97: channels > getMaxChannelCount()) return false; cannam@97: cannam@97: m_blockSize = blockSize; cannam@97: cannam@97: return true; cannam@97: } cannam@97: cannam@97: void cannam@97: PowerSpectrum::reset() cannam@97: { cannam@97: } cannam@97: cannam@97: PowerSpectrum::OutputList cannam@97: PowerSpectrum::getOutputDescriptors() const cannam@97: { cannam@97: OutputList list; cannam@97: cannam@97: OutputDescriptor d; cannam@97: d.identifier = "powerspectrum"; cannam@97: d.name = "Power Spectrum"; cannam@97: d.description = "Power values of the frequency spectrum bins calculated from the input signal"; cannam@97: d.unit = ""; cannam@97: d.hasFixedBinCount = true; cannam@97: if (m_blockSize == 0) { cannam@97: // Just so as not to return "1". This is the bin count that cannam@97: // would result from a block size of 1024, which is a likely cannam@97: // default -- but the host should always set the block size cannam@97: // before querying the bin count for certain. cannam@97: d.binCount = 513; cannam@97: } else { cannam@97: d.binCount = m_blockSize / 2 + 1; cannam@97: } cannam@97: d.hasKnownExtents = false; cannam@97: d.isQuantized = false; cannam@97: d.sampleType = OutputDescriptor::OneSamplePerStep; cannam@97: list.push_back(d); cannam@97: cannam@97: return list; cannam@97: } cannam@97: cannam@97: PowerSpectrum::FeatureSet cannam@97: PowerSpectrum::process(const float *const *inputBuffers, Vamp::RealTime timestamp) cannam@97: { cannam@97: FeatureSet fs; cannam@97: cannam@97: if (m_blockSize == 0) { cannam@97: cerr << "ERROR: PowerSpectrum::process: Not initialised" << endl; cannam@97: return fs; cannam@97: } cannam@97: cannam@97: size_t n = m_blockSize / 2 + 1; cannam@97: const float *fbuf = inputBuffers[0]; cannam@97: cannam@97: Feature feature; cannam@97: feature.hasTimestamp = false; cannam@97: feature.values.reserve(n); // optional cannam@97: cannam@97: for (size_t i = 0; i < n; ++i) { cannam@97: cannam@97: double real = fbuf[i * 2]; cannam@97: double imag = fbuf[i * 2 + 1]; cannam@97: cannam@97: feature.values.push_back(real * real + imag * imag); cannam@97: } cannam@97: cannam@97: fs[0].push_back(feature); cannam@97: cannam@97: return fs; cannam@97: } cannam@97: cannam@97: PowerSpectrum::FeatureSet cannam@97: PowerSpectrum::getRemainingFeatures() cannam@97: { cannam@97: return FeatureSet(); cannam@97: } cannam@97: