← Back to branch summary

~ubuntu-branches/ubuntu/lucid/ardour/lucid-proposed

« back to all changes in this revision

Viewing changes to libs/vamp-plugins/SpectralCentroid.cpp

  • Committer: Bazaar Package Importer
  • Author(s): Free Ekanayaka
  • Date: 2008-02-07 20:51:55 UTC
  • mfrom: (1.1.13 upstream)
  • Revision ID: james.westby@ubuntu.com-20080207205155-xzi5lq5s2o1okvx8
Tags: 1:2.3.1-1
http://bugs.debian.org/463803
* New upstream release
* debian/control:
   - build-depend on libfftw3-dev (Closes: #463803)
   - added Homepage, Vcs-Svn, Vcs-Browser
* debian/patches
   - dropped 50-soundtouch.patch (fixed upstream)
   - updated 80_ardourino.patch
   - dropped unused patches from source package

Show diffs side-by-side

added added

removed removed

Lines of Context:
libs/vamp-plugins/SpectralCentroid.cpp
 
1
/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
 
2
 
 
3
/*
 
4
    Vamp
 
5
 
 
6
    An API for audio analysis and feature extraction plugins.
 
7
 
 
8
    Centre for Digital Music, Queen Mary, University of London.
 
9
    Copyright 2006 Chris Cannam.
 
10
  
 
11
    Permission is hereby granted, free of charge, to any person
 
12
    obtaining a copy of this software and associated documentation
 
13
    files (the "Software"), to deal in the Software without
 
14
    restriction, including without limitation the rights to use, copy,
 
15
    modify, merge, publish, distribute, sublicense, and/or sell copies
 
16
    of the Software, and to permit persons to whom the Software is
 
17
    furnished to do so, subject to the following conditions:
 
18
 
 
19
    The above copyright notice and this permission notice shall be
 
20
    included in all copies or substantial portions of the Software.
 
21
 
 
22
    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 
23
    EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 
24
    MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 
25
    NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR
 
26
    ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
 
27
    CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 
28
    WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
29
 
 
30
    Except as contained in this notice, the names of the Centre for
 
31
    Digital Music; Queen Mary, University of London; and Chris Cannam
 
32
    shall not be used in advertising or otherwise to promote the sale,
 
33
    use or other dealings in this Software without prior written
 
34
    authorization.
 
35
*/
 
36
 
 
37
#include "SpectralCentroid.h"
 
38
 
 
39
using std::string;
 
40
using std::vector;
 
41
using std::cerr;
 
42
using std::endl;
 
43
 
 
44
#include <cmath>
 
45
 
 
46
 
 
47
SpectralCentroid::SpectralCentroid(float inputSampleRate) :
 
48
    Plugin(inputSampleRate),
 
49
    m_stepSize(0),
 
50
    m_blockSize(0)
 
51
{
 
52
}
 
53
 
 
54
SpectralCentroid::~SpectralCentroid()
 
55
{
 
56
}
 
57
 
 
58
string
 
59
SpectralCentroid::getIdentifier() const
 
60
{
 
61
    return "spectralcentroid";
 
62
}
 
63
 
 
64
string
 
65
SpectralCentroid::getName() const
 
66
{
 
67
    return "Spectral Centroid";
 
68
}
 
69
 
 
70
string
 
71
SpectralCentroid::getDescription() const
 
72
{
 
73
    return "Calculate the centroid frequency of the spectrum of the input signal";
 
74
}
 
75
 
 
76
string
 
77
SpectralCentroid::getMaker() const
 
78
{
 
79
    return "Vamp SDK Example Plugins";
 
80
}
 
81
 
 
82
int
 
83
SpectralCentroid::getPluginVersion() const
 
84
{
 
85
    return 2;
 
86
}
 
87
 
 
88
string
 
89
SpectralCentroid::getCopyright() const
 
90
{
 
91
    return "Freely redistributable (BSD license)";
 
92
}
 
93
 
 
94
bool
 
95
SpectralCentroid::initialise(size_t channels, size_t stepSize, size_t blockSize)
 
96
{
 
97
    if (channels < getMinChannelCount() ||
 
98
        channels > getMaxChannelCount()) return false;
 
99
 
 
100
    m_stepSize = stepSize;
 
101
    m_blockSize = blockSize;
 
102
 
 
103
    return true;
 
104
}
 
105
 
 
106
void
 
107
SpectralCentroid::reset()
 
108
{
 
109
}
 
110
 
 
111
SpectralCentroid::OutputList
 
112
SpectralCentroid::getOutputDescriptors() const
 
113
{
 
114
    OutputList list;
 
115
 
 
116
    OutputDescriptor d;
 
117
    d.identifier = "logcentroid";
 
118
    d.name = "Log Frequency Centroid";
 
119
    d.description = "Centroid of the log weighted frequency spectrum";
 
120
    d.unit = "Hz";
 
121
    d.hasFixedBinCount = true;
 
122
    d.binCount = 1;
 
123
    d.hasKnownExtents = false;
 
124
    d.isQuantized = false;
 
125
    d.sampleType = OutputDescriptor::OneSamplePerStep;
 
126
    list.push_back(d);
 
127
 
 
128
    d.identifier = "linearcentroid";
 
129
    d.name = "Linear Frequency Centroid";
 
130
    d.description = "Centroid of the linear frequency spectrum";
 
131
    list.push_back(d);
 
132
 
 
133
    return list;
 
134
}
 
135
 
 
136
SpectralCentroid::FeatureSet
 
137
SpectralCentroid::process(const float *const *inputBuffers, Vamp::RealTime)
 
138
{
 
139
    if (m_stepSize == 0) {
 
140
        cerr << "ERROR: SpectralCentroid::process: "
 
141
             << "SpectralCentroid has not been initialised"
 
142
             << endl;
 
143
        return FeatureSet();
 
144
    }
 
145
 
 
146
    double numLin = 0.0, numLog = 0.0, denom = 0.0;
 
147
 
 
148
    for (size_t i = 1; i <= m_blockSize/2; ++i) {
 
149
        double freq = (double(i) * m_inputSampleRate) / m_blockSize;
 
150
        double real = inputBuffers[0][i*2];
 
151
        double imag = inputBuffers[0][i*2 + 1];
 
152
        double power = sqrt(real * real + imag * imag) / (m_blockSize/2);
 
153
        numLin += freq * power;
 
154
        numLog += log10f(freq) * power;
 
155
        denom += power;
 
156
    }
 
157
 
 
158
    FeatureSet returnFeatures;
 
159
 
 
160
//    std::cerr << "power " << denom << ", block size " << m_blockSize << std::endl;
 
161
 
 
162
    if (denom != 0.0) {
 
163
        float centroidLin = float(numLin / denom);
 
164
        float centroidLog = powf(10, float(numLog / denom));
 
165
 
 
166
        Feature feature;
 
167
        feature.hasTimestamp = false;
 
168
        if (!std::isnan(centroidLog) && !std::isinf(centroidLog)) {
 
169
            feature.values.push_back(centroidLog);
 
170
        }
 
171
        returnFeatures[0].push_back(feature);
 
172
 
 
173
        feature.values.clear();
 
174
        if (!std::isnan(centroidLin) && !std::isinf(centroidLin)) {
 
175
            feature.values.push_back(centroidLin);
 
176
        }
 
177
        returnFeatures[1].push_back(feature);
 
178
    }
 
179
 
 
180
    return returnFeatures;
 
181
}
 
182
 
 
183
SpectralCentroid::FeatureSet
 
184
SpectralCentroid::getRemainingFeatures()
 
185
{
 
186
    return FeatureSet();
 
187
}
 
188