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