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 2006 Chris Cannam. 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 "SpectralCentroid.h" cannam@97: cannam@97: using std::string; cannam@97: using std::vector; cannam@97: using std::cerr; cannam@97: using std::endl; cannam@97: cannam@97: #include cannam@97: cannam@97: #ifdef __SUNPRO_CC cannam@97: #include cannam@97: #define isinf(x) (!finite(x)) cannam@97: #endif cannam@97: cannam@97: #ifdef WIN32 cannam@97: #define isnan(x) false cannam@97: #define isinf(x) false cannam@97: #endif cannam@97: cannam@97: SpectralCentroid::SpectralCentroid(float inputSampleRate) : cannam@97: Plugin(inputSampleRate), cannam@97: m_stepSize(0), cannam@97: m_blockSize(0) cannam@97: { cannam@97: } cannam@97: cannam@97: SpectralCentroid::~SpectralCentroid() cannam@97: { cannam@97: } cannam@97: cannam@97: string cannam@97: SpectralCentroid::getIdentifier() const cannam@97: { cannam@97: return "spectralcentroid"; cannam@97: } cannam@97: cannam@97: string cannam@97: SpectralCentroid::getName() const cannam@97: { cannam@97: return "Spectral Centroid"; cannam@97: } cannam@97: cannam@97: string cannam@97: SpectralCentroid::getDescription() const cannam@97: { cannam@97: return "Calculate the centroid frequency of the spectrum of the input signal"; cannam@97: } cannam@97: cannam@97: string cannam@97: SpectralCentroid::getMaker() const cannam@97: { cannam@97: return "Vamp SDK Example Plugins"; cannam@97: } cannam@97: cannam@97: int cannam@97: SpectralCentroid::getPluginVersion() const cannam@97: { cannam@97: return 2; cannam@97: } cannam@97: cannam@97: string cannam@97: SpectralCentroid::getCopyright() const cannam@97: { cannam@97: return "Freely redistributable (BSD license)"; cannam@97: } cannam@97: cannam@97: bool cannam@97: SpectralCentroid::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_stepSize = stepSize; cannam@97: m_blockSize = blockSize; cannam@97: cannam@97: return true; cannam@97: } cannam@97: cannam@97: void cannam@97: SpectralCentroid::reset() cannam@97: { cannam@97: } cannam@97: cannam@97: SpectralCentroid::OutputList cannam@97: SpectralCentroid::getOutputDescriptors() const cannam@97: { cannam@97: OutputList list; cannam@97: cannam@97: OutputDescriptor d; cannam@97: d.identifier = "logcentroid"; cannam@97: d.name = "Log Frequency Centroid"; cannam@97: d.description = "Centroid of the log weighted frequency spectrum"; cannam@97: d.unit = "Hz"; cannam@97: d.hasFixedBinCount = true; cannam@97: d.binCount = 1; 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: d.identifier = "linearcentroid"; cannam@97: d.name = "Linear Frequency Centroid"; cannam@97: d.description = "Centroid of the linear frequency spectrum"; cannam@97: list.push_back(d); cannam@97: cannam@97: return list; cannam@97: } cannam@97: cannam@97: SpectralCentroid::FeatureSet cannam@97: SpectralCentroid::process(const float *const *inputBuffers, Vamp::RealTime timestamp) cannam@97: { cannam@97: if (m_stepSize == 0) { cannam@97: cerr << "ERROR: SpectralCentroid::process: " cannam@97: << "SpectralCentroid has not been initialised" cannam@97: << endl; cannam@97: return FeatureSet(); cannam@97: } cannam@97: cannam@97: double numLin = 0.0, numLog = 0.0, denom = 0.0; cannam@97: cannam@97: for (size_t i = 1; i <= m_blockSize/2; ++i) { cannam@97: double freq = (double(i) * m_inputSampleRate) / m_blockSize; cannam@97: double real = inputBuffers[0][i*2]; cannam@97: double imag = inputBuffers[0][i*2 + 1]; cannam@97: double scalemag = sqrt(real * real + imag * imag) / (m_blockSize/2); cannam@97: numLin += freq * scalemag; cannam@97: numLog += log10f(freq) * scalemag; cannam@97: denom += scalemag; cannam@97: } cannam@97: cannam@97: FeatureSet returnFeatures; cannam@97: cannam@97: if (denom != 0.0) { cannam@97: float centroidLin = float(numLin / denom); cannam@97: float centroidLog = powf(10, float(numLog / denom)); cannam@97: cannam@97: Feature feature; cannam@97: feature.hasTimestamp = false; cannam@97: cannam@97: if (!isnan(centroidLog) && !isinf(centroidLog)) { cannam@97: feature.values.push_back(centroidLog); cannam@97: } cannam@97: returnFeatures[0].push_back(feature); cannam@97: cannam@97: feature.values.clear(); cannam@97: if (!isnan(centroidLin) && !isinf(centroidLin)) { cannam@97: feature.values.push_back(centroidLin); cannam@97: } cannam@97: returnFeatures[1].push_back(feature); cannam@97: } cannam@97: cannam@97: return returnFeatures; cannam@97: } cannam@97: cannam@97: SpectralCentroid::FeatureSet cannam@97: SpectralCentroid::getRemainingFeatures() cannam@97: { cannam@97: return FeatureSet(); cannam@97: } cannam@97: