cannam@3
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
cannam@3
|
2
|
cannam@3
|
3 /*
|
cannam@3
|
4 Vamp
|
cannam@3
|
5
|
cannam@3
|
6 An API for audio analysis and feature extraction plugins.
|
cannam@3
|
7
|
cannam@3
|
8 Centre for Digital Music, Queen Mary, University of London.
|
cannam@3
|
9 Copyright 2006 Chris Cannam.
|
cannam@3
|
10
|
cannam@3
|
11 Permission is hereby granted, free of charge, to any person
|
cannam@3
|
12 obtaining a copy of this software and associated documentation
|
cannam@3
|
13 files (the "Software"), to deal in the Software without
|
cannam@3
|
14 restriction, including without limitation the rights to use, copy,
|
cannam@3
|
15 modify, merge, publish, distribute, sublicense, and/or sell copies
|
cannam@3
|
16 of the Software, and to permit persons to whom the Software is
|
cannam@3
|
17 furnished to do so, subject to the following conditions:
|
cannam@3
|
18
|
cannam@3
|
19 The above copyright notice and this permission notice shall be
|
cannam@3
|
20 included in all copies or substantial portions of the Software.
|
cannam@3
|
21
|
cannam@3
|
22 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
cannam@3
|
23 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
cannam@3
|
24 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
cannam@6
|
25 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR
|
cannam@3
|
26 ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
|
cannam@3
|
27 CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
cannam@3
|
28 WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
cannam@3
|
29
|
cannam@3
|
30 Except as contained in this notice, the names of the Centre for
|
cannam@3
|
31 Digital Music; Queen Mary, University of London; and Chris Cannam
|
cannam@3
|
32 shall not be used in advertising or otherwise to promote the sale,
|
cannam@3
|
33 use or other dealings in this Software without prior written
|
cannam@3
|
34 authorization.
|
cannam@3
|
35 */
|
cannam@3
|
36
|
cannam@3
|
37 #ifndef _VAMP_PLUGIN_H_
|
cannam@3
|
38 #define _VAMP_PLUGIN_H_
|
cannam@3
|
39
|
cannam@3
|
40 #include "PluginBase.h"
|
cannam@3
|
41 #include "RealTime.h"
|
cannam@3
|
42
|
cannam@3
|
43 #include <string>
|
cannam@3
|
44 #include <vector>
|
cannam@3
|
45 #include <map>
|
cannam@3
|
46
|
cannam@3
|
47 namespace Vamp {
|
cannam@3
|
48
|
cannam@3
|
49 /**
|
cannam@3
|
50 * Vamp::Plugin is a base class for plugin instance classes
|
cannam@3
|
51 * that provide feature extraction from audio or related data.
|
cannam@3
|
52 *
|
cannam@3
|
53 * In most cases, the input will be audio and the output will be a
|
cannam@3
|
54 * stream of derived data at a lower sampling resolution than the
|
cannam@3
|
55 * input.
|
cannam@3
|
56 *
|
cannam@3
|
57 * Note that this class inherits several abstract methods from
|
cannam@3
|
58 * PluginBase, that must be implemented by the subclass.
|
cannam@3
|
59 */
|
cannam@3
|
60
|
cannam@3
|
61 /**
|
cannam@3
|
62 * Plugin Lifecycle
|
cannam@3
|
63 * ================
|
cannam@3
|
64 *
|
cannam@3
|
65 * Feature extraction plugins are managed differently from real-time
|
cannam@3
|
66 * plugins (such as VST effects). The main difference is that the
|
cannam@3
|
67 * parameters for a feature extraction plugin are configured before
|
cannam@3
|
68 * the plugin is used, and do not change during use.
|
cannam@3
|
69 *
|
cannam@3
|
70 * 1. Host constructs the plugin, passing it the input sample rate.
|
cannam@3
|
71 * The plugin may do basic initialisation, but should not do anything
|
cannam@3
|
72 * computationally expensive at this point.
|
cannam@3
|
73 *
|
cannam@3
|
74 * 2. Host may query the plugin's available outputs.
|
cannam@3
|
75 *
|
cannam@3
|
76 * 3. Host queries programs and parameter descriptors, and may set
|
cannam@3
|
77 * some or all of them. Parameters that are not explicitly set should
|
cannam@3
|
78 * take their default values as specified in the parameter descriptor.
|
cannam@3
|
79 * When a program is set, the parameter values may change and the host
|
cannam@3
|
80 * will re-query them to check.
|
cannam@3
|
81 *
|
cannam@3
|
82 * 4. Host queries the preferred step size, block size, number of
|
cannam@3
|
83 * channels, and the number of values per feature for the plugin's
|
cannam@3
|
84 * outputs. These may all vary depending on the parameter values.
|
cannam@3
|
85 * (Note however that you cannot make the number of distinct outputs
|
cannam@3
|
86 * dependent on parameter values; nor can you make any of these depend
|
cannam@3
|
87 * on the number of input channels.)
|
cannam@3
|
88 *
|
cannam@3
|
89 * 5. Plugin is properly initialised with a call to initialise. This
|
cannam@3
|
90 * fixes the step size, block size, and number of channels, as well as
|
cannam@3
|
91 * all of the parameter and program settings. If the values passed in
|
cannam@3
|
92 * to initialise do not match the plugin's advertised preferred values
|
cannam@3
|
93 * from step 4, the plugin may refuse to initialise and return false
|
cannam@3
|
94 * (although if possible it should accept the new values).
|
cannam@3
|
95 *
|
cannam@3
|
96 * 6. Host will repeatedly call the process method to pass in blocks
|
cannam@3
|
97 * of input data. This method may return features extracted from that
|
cannam@3
|
98 * data (if the plugin is causal).
|
cannam@3
|
99 *
|
cannam@3
|
100 * 7. Host will call getRemainingFeatures exactly once, after all the
|
cannam@3
|
101 * input data has been processed. This may return any non-causal or
|
cannam@3
|
102 * leftover features.
|
cannam@3
|
103 *
|
cannam@3
|
104 * 8. At any point after initialise was called, the host may
|
cannam@3
|
105 * optionally call the reset method and restart processing. (This
|
cannam@3
|
106 * does not mean it can change the parameters, which are fixed from
|
cannam@3
|
107 * initialise until destruction.)
|
cannam@3
|
108 *
|
cannam@3
|
109 * A plugin does not need to handle the case where setParameter or
|
cannam@3
|
110 * selectProgram is called after initialise has been called. It's the
|
cannam@3
|
111 * host's responsibility not to do that.
|
cannam@3
|
112 */
|
cannam@3
|
113
|
cannam@3
|
114 class Plugin : public PluginBase
|
cannam@3
|
115 {
|
cannam@3
|
116 public:
|
cannam@20
|
117 virtual ~Plugin() { }
|
cannam@20
|
118
|
cannam@3
|
119 /**
|
cannam@3
|
120 * Initialise a plugin to prepare it for use with the given number
|
cannam@3
|
121 * of input channels, step size (window increment, in sample
|
cannam@3
|
122 * frames) and block size (window size, in sample frames).
|
cannam@3
|
123 *
|
cannam@3
|
124 * The input sample rate should have been already specified at
|
cannam@3
|
125 * construction time.
|
cannam@3
|
126 *
|
cannam@3
|
127 * Return true for successful initialisation, false if the number
|
cannam@3
|
128 * of input channels, step size and/or block size cannot be
|
cannam@3
|
129 * supported.
|
cannam@3
|
130 */
|
cannam@3
|
131 virtual bool initialise(size_t inputChannels,
|
cannam@3
|
132 size_t stepSize,
|
cannam@3
|
133 size_t blockSize) = 0;
|
cannam@3
|
134
|
cannam@3
|
135 /**
|
cannam@3
|
136 * Reset the plugin after use, to prepare it for another clean
|
cannam@3
|
137 * run. Not called for the first initialisation (i.e. initialise
|
cannam@3
|
138 * must also do a reset).
|
cannam@3
|
139 */
|
cannam@3
|
140 virtual void reset() = 0;
|
cannam@3
|
141
|
cannam@3
|
142 enum InputDomain { TimeDomain, FrequencyDomain };
|
cannam@3
|
143
|
cannam@3
|
144 /**
|
cannam@3
|
145 * Get the plugin's required input domain. If this is TimeDomain,
|
cannam@3
|
146 * the samples provided to the process() function (below) will be
|
cannam@3
|
147 * in the time domain, as for a traditional audio processing
|
cannam@3
|
148 * plugin. If this is FrequencyDomain, the host will carry out a
|
cannam@3
|
149 * windowed FFT of size equal to the negotiated block size on the
|
cannam@3
|
150 * data before passing the frequency bin data in to process().
|
cannam@3
|
151 * The plugin does not get to choose the window type -- the host
|
cannam@3
|
152 * will either let the user do so, or will use a Hanning window.
|
cannam@3
|
153 */
|
cannam@3
|
154 virtual InputDomain getInputDomain() const = 0;
|
cannam@3
|
155
|
cannam@3
|
156 /**
|
cannam@8
|
157 * Get the preferred block size (window size -- the number of
|
cannam@8
|
158 * sample frames passed in each block to the process() function).
|
cannam@8
|
159 * This should be called before initialise().
|
cannam@8
|
160 *
|
cannam@8
|
161 * A plugin that can handle any block size may return 0. The
|
cannam@8
|
162 * final block size will be set in the initialise() call.
|
cannam@8
|
163 */
|
cannam@8
|
164 virtual size_t getPreferredBlockSize() const { return 0; }
|
cannam@8
|
165
|
cannam@8
|
166 /**
|
cannam@3
|
167 * Get the preferred step size (window increment -- the distance
|
cannam@3
|
168 * in sample frames between the start frames of consecutive blocks
|
cannam@3
|
169 * passed to the process() function) for the plugin. This should
|
cannam@3
|
170 * be called before initialise().
|
cannam@8
|
171 *
|
cannam@8
|
172 * A plugin may return 0 if it has no particular interest in the
|
cannam@8
|
173 * step size. In this case, the host should make the step size
|
cannam@8
|
174 * equal to the block size if the plugin is accepting input in the
|
cannam@8
|
175 * time domain. If the plugin is accepting input in the frequency
|
cannam@8
|
176 * domain, the host may use any step size. The final step size
|
cannam@8
|
177 * will be set in the initialise() call.
|
cannam@3
|
178 */
|
cannam@8
|
179 virtual size_t getPreferredStepSize() const { return 0; }
|
cannam@3
|
180
|
cannam@3
|
181 /**
|
cannam@3
|
182 * Get the minimum supported number of input channels.
|
cannam@3
|
183 */
|
cannam@3
|
184 virtual size_t getMinChannelCount() const { return 1; }
|
cannam@3
|
185
|
cannam@3
|
186 /**
|
cannam@3
|
187 * Get the maximum supported number of input channels.
|
cannam@3
|
188 */
|
cannam@3
|
189 virtual size_t getMaxChannelCount() const { return 1; }
|
cannam@3
|
190
|
cannam@3
|
191 struct OutputDescriptor
|
cannam@3
|
192 {
|
cannam@3
|
193 /**
|
cannam@3
|
194 * The name of the output, in computer-usable form. Should be
|
cannam@3
|
195 * reasonably short and without whitespace or punctuation.
|
cannam@3
|
196 */
|
cannam@3
|
197 std::string name;
|
cannam@3
|
198
|
cannam@3
|
199 /**
|
cannam@3
|
200 * The human-readable name of the output.
|
cannam@3
|
201 */
|
cannam@3
|
202 std::string description;
|
cannam@3
|
203
|
cannam@3
|
204 /**
|
cannam@3
|
205 * The unit of the output, in human-readable form.
|
cannam@3
|
206 */
|
cannam@3
|
207 std::string unit;
|
cannam@3
|
208
|
cannam@3
|
209 /**
|
cannam@9
|
210 * True if the output has the same number of values per sample
|
cannam@9
|
211 * for every output sample. Outputs for which this is false
|
cannam@3
|
212 * are unlikely to be very useful in a general-purpose host.
|
cannam@3
|
213 */
|
cannam@9
|
214 bool hasFixedBinCount;
|
cannam@3
|
215
|
cannam@3
|
216 /**
|
cannam@3
|
217 * The number of values per result of the output. Undefined
|
cannam@9
|
218 * if hasFixedBinCount is false. If this is zero, the output
|
cannam@3
|
219 * is point data (i.e. only the time of each output is of
|
cannam@3
|
220 * interest, the value list will be empty).
|
cannam@3
|
221 */
|
cannam@9
|
222 size_t binCount;
|
cannam@3
|
223
|
cannam@3
|
224 /**
|
cannam@9
|
225 * The names of each of the bins, if appropriate. This is
|
cannam@3
|
226 * always optional.
|
cannam@3
|
227 */
|
cannam@9
|
228 std::vector<std::string> binNames;
|
cannam@3
|
229
|
cannam@3
|
230 /**
|
cannam@9
|
231 * True if the results in each output bin fall within a fixed
|
cannam@9
|
232 * numeric range (minimum and maximum values). Undefined if
|
cannam@9
|
233 * binCount is zero.
|
cannam@3
|
234 */
|
cannam@3
|
235 bool hasKnownExtents;
|
cannam@3
|
236
|
cannam@3
|
237 /**
|
cannam@3
|
238 * Minimum value of the results in the output. Undefined if
|
cannam@9
|
239 * hasKnownExtents is false or binCount is zero.
|
cannam@3
|
240 */
|
cannam@3
|
241 float minValue;
|
cannam@3
|
242
|
cannam@3
|
243 /**
|
cannam@3
|
244 * Maximum value of the results in the output. Undefined if
|
cannam@9
|
245 * hasKnownExtents is false or binCount is zero.
|
cannam@3
|
246 */
|
cannam@3
|
247 float maxValue;
|
cannam@3
|
248
|
cannam@3
|
249 /**
|
cannam@3
|
250 * True if the output values are quantized to a particular
|
cannam@9
|
251 * resolution. Undefined if binCount is zero.
|
cannam@3
|
252 */
|
cannam@3
|
253 bool isQuantized;
|
cannam@3
|
254
|
cannam@3
|
255 /**
|
cannam@3
|
256 * Quantization resolution of the output values (e.g. 1.0 if
|
cannam@3
|
257 * they are all integers). Undefined if isQuantized is false
|
cannam@9
|
258 * or binCount is zero.
|
cannam@3
|
259 */
|
cannam@3
|
260 float quantizeStep;
|
cannam@3
|
261
|
cannam@3
|
262 enum SampleType {
|
cannam@3
|
263
|
cannam@3
|
264 /// Results from each process() align with that call's block start
|
cannam@3
|
265 OneSamplePerStep,
|
cannam@3
|
266
|
cannam@3
|
267 /// Results are evenly spaced in time (sampleRate specified below)
|
cannam@3
|
268 FixedSampleRate,
|
cannam@3
|
269
|
cannam@3
|
270 /// Results are unevenly spaced and have individual timestamps
|
cannam@3
|
271 VariableSampleRate
|
cannam@3
|
272 };
|
cannam@3
|
273
|
cannam@3
|
274 /**
|
cannam@3
|
275 * Positioning in time of the output results.
|
cannam@3
|
276 */
|
cannam@3
|
277 SampleType sampleType;
|
cannam@3
|
278
|
cannam@3
|
279 /**
|
cannam@17
|
280 * Sample rate of the output results, as samples per second.
|
cannam@17
|
281 * Undefined if sampleType is OneSamplePerStep.
|
cannam@3
|
282 *
|
cannam@3
|
283 * If sampleType is VariableSampleRate and this value is
|
cannam@3
|
284 * non-zero, then it may be used to calculate a resolution for
|
cannam@17
|
285 * the output (i.e. the "duration" of each sample, in time,
|
cannam@17
|
286 * will be 1/sampleRate seconds). It's recommended to set
|
cannam@17
|
287 * this to zero if that behaviour is not desired.
|
cannam@3
|
288 */
|
cannam@3
|
289 float sampleRate;
|
cannam@3
|
290 };
|
cannam@3
|
291
|
cannam@3
|
292 typedef std::vector<OutputDescriptor> OutputList;
|
cannam@3
|
293
|
cannam@3
|
294 /**
|
cannam@3
|
295 * Get the outputs of this plugin. An output's index in this list
|
cannam@3
|
296 * is used as its numeric index when looking it up in the
|
cannam@3
|
297 * FeatureSet returned from the process() call.
|
cannam@3
|
298 */
|
cannam@3
|
299 virtual OutputList getOutputDescriptors() const = 0;
|
cannam@3
|
300
|
cannam@3
|
301 struct Feature
|
cannam@3
|
302 {
|
cannam@3
|
303 /**
|
cannam@3
|
304 * True if an output feature has its own timestamp. This is
|
cannam@3
|
305 * mandatory if the output has VariableSampleRate, and is
|
cannam@3
|
306 * likely to be disregarded otherwise.
|
cannam@3
|
307 */
|
cannam@3
|
308 bool hasTimestamp;
|
cannam@3
|
309
|
cannam@3
|
310 /**
|
cannam@3
|
311 * Timestamp of the output feature. This is mandatory if the
|
cannam@3
|
312 * output has VariableSampleRate, and is likely to be
|
cannam@3
|
313 * disregarded otherwise. Undefined if hasTimestamp is false.
|
cannam@3
|
314 */
|
cannam@3
|
315 RealTime timestamp;
|
cannam@3
|
316
|
cannam@3
|
317 /**
|
cannam@3
|
318 * Results for a single sample of this feature. If the output
|
cannam@9
|
319 * hasFixedBinCount, there must be the same number of values
|
cannam@9
|
320 * as the output's binCount count.
|
cannam@3
|
321 */
|
cannam@3
|
322 std::vector<float> values;
|
cannam@3
|
323
|
cannam@3
|
324 /**
|
cannam@3
|
325 * Label for the sample of this feature.
|
cannam@3
|
326 */
|
cannam@3
|
327 std::string label;
|
cannam@3
|
328 };
|
cannam@3
|
329
|
cannam@3
|
330 typedef std::vector<Feature> FeatureList;
|
cannam@3
|
331 typedef std::map<int, FeatureList> FeatureSet; // key is output no
|
cannam@3
|
332
|
cannam@3
|
333 /**
|
cannam@3
|
334 * Process a single block of input data.
|
cannam@3
|
335 *
|
cannam@3
|
336 * If the plugin's inputDomain is TimeDomain, inputBuffers will
|
cannam@3
|
337 * point to one array of floats per input channel, and each of
|
cannam@3
|
338 * these arrays will contain blockSize consecutive audio samples
|
cannam@9
|
339 * (the host will zero-pad as necessary). The timestamp will be
|
cannam@9
|
340 * the real time in seconds of the start of the supplied block of
|
cannam@9
|
341 * samples.
|
cannam@3
|
342 *
|
cannam@3
|
343 * If the plugin's inputDomain is FrequencyDomain, inputBuffers
|
cannam@3
|
344 * will point to one array of floats per input channel, and each
|
cannam@3
|
345 * of these arrays will contain blockSize/2 consecutive pairs of
|
cannam@3
|
346 * real and imaginary component floats corresponding to bins
|
cannam@9
|
347 * 0..(blockSize/2-1) of the FFT output. The timestamp will be
|
cannam@9
|
348 * the real time in seconds of the centre of the FFT input window
|
cannam@9
|
349 * (i.e. the very first block passed to process might contain the
|
cannam@9
|
350 * FFT of half a block of zero samples and the first half-block of
|
cannam@9
|
351 * the actual data, with a timestamp of zero).
|
cannam@3
|
352 *
|
cannam@3
|
353 * Return any features that have become available after this
|
cannam@3
|
354 * process call. (These do not necessarily have to fall within
|
cannam@3
|
355 * the process block, except for OneSamplePerStep outputs.)
|
cannam@3
|
356 */
|
cannam@3
|
357 virtual FeatureSet process(float **inputBuffers,
|
cannam@3
|
358 RealTime timestamp) = 0;
|
cannam@3
|
359
|
cannam@3
|
360 /**
|
cannam@3
|
361 * After all blocks have been processed, calculate and return any
|
cannam@3
|
362 * remaining features derived from the complete input.
|
cannam@3
|
363 */
|
cannam@3
|
364 virtual FeatureSet getRemainingFeatures() = 0;
|
cannam@3
|
365
|
cannam@3
|
366 virtual std::string getType() const { return "Feature Extraction Plugin"; }
|
cannam@3
|
367
|
cannam@3
|
368 protected:
|
cannam@3
|
369 Plugin(float inputSampleRate) :
|
cannam@3
|
370 m_inputSampleRate(inputSampleRate) { }
|
cannam@3
|
371
|
cannam@3
|
372 float m_inputSampleRate;
|
cannam@3
|
373 };
|
cannam@3
|
374
|
cannam@3
|
375 }
|
cannam@3
|
376
|
cannam@3
|
377 #endif
|
cannam@3
|
378
|
cannam@3
|
379
|
cannam@3
|
380
|