Chris@0
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
Chris@0
|
2
|
Chris@26
|
3 /*
|
Chris@26
|
4 VampyHost
|
Chris@26
|
5
|
Chris@26
|
6 Use Vamp audio analysis plugins in Python
|
Chris@26
|
7
|
Chris@26
|
8 Gyorgy Fazekas and Chris Cannam
|
Chris@26
|
9 Centre for Digital Music, Queen Mary, University of London
|
Chris@26
|
10 Copyright 2008-2014 Queen Mary, University of London
|
Chris@26
|
11
|
Chris@26
|
12 Permission is hereby granted, free of charge, to any person
|
Chris@26
|
13 obtaining a copy of this software and associated documentation
|
Chris@26
|
14 files (the "Software"), to deal in the Software without
|
Chris@26
|
15 restriction, including without limitation the rights to use, copy,
|
Chris@26
|
16 modify, merge, publish, distribute, sublicense, and/or sell copies
|
Chris@26
|
17 of the Software, and to permit persons to whom the Software is
|
Chris@26
|
18 furnished to do so, subject to the following conditions:
|
Chris@26
|
19
|
Chris@26
|
20 The above copyright notice and this permission notice shall be
|
Chris@26
|
21 included in all copies or substantial portions of the Software.
|
Chris@26
|
22
|
Chris@26
|
23 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
Chris@26
|
24 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
Chris@26
|
25 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
Chris@26
|
26 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR
|
Chris@26
|
27 ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
|
Chris@26
|
28 CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
Chris@26
|
29 WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
Chris@26
|
30
|
Chris@26
|
31 Except as contained in this notice, the names of the Centre for
|
Chris@26
|
32 Digital Music; Queen Mary, University of London; and the authors
|
Chris@26
|
33 shall not be used in advertising or otherwise to promote the sale,
|
Chris@26
|
34 use or other dealings in this Software without prior written
|
Chris@26
|
35 authorization.
|
Chris@26
|
36 */
|
Chris@26
|
37
|
Chris@31
|
38 #include "PyPluginObject.h"
|
Chris@14
|
39
|
Chris@14
|
40 // define a unique API pointer
|
Chris@27
|
41 #define PY_ARRAY_UNIQUE_SYMBOL VAMPYHOST_ARRAY_API
|
Chris@14
|
42 #define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
|
Chris@31
|
43 #define NO_IMPORT_ARRAY
|
Chris@14
|
44 #include "numpy/arrayobject.h"
|
Chris@14
|
45
|
Chris@102
|
46 #if PY_MAJOR_VERSION < 3
|
Chris@102
|
47 #include "intobject.h"
|
Chris@102
|
48 #endif
|
Chris@102
|
49
|
Chris@33
|
50 #include "structmember.h"
|
Chris@33
|
51
|
Chris@48
|
52 #include "FloatConversion.h"
|
Chris@29
|
53 #include "VectorConversion.h"
|
Chris@16
|
54 #include "PyRealTime.h"
|
Chris@0
|
55
|
Chris@0
|
56 #include <string>
|
Chris@31
|
57 #include <vector>
|
Chris@33
|
58 #include <cstddef>
|
Chris@49
|
59 #include <set>
|
Chris@0
|
60
|
Chris@0
|
61 using namespace std;
|
Chris@0
|
62 using namespace Vamp;
|
Chris@0
|
63
|
Chris@31
|
64 static
|
Chris@21
|
65 PyPluginObject *
|
Chris@21
|
66 getPluginObject(PyObject *pyPluginHandle)
|
Chris@21
|
67 {
|
Chris@21
|
68 PyPluginObject *pd = 0;
|
Chris@21
|
69 if (PyPlugin_Check(pyPluginHandle)) {
|
Chris@21
|
70 pd = (PyPluginObject *)pyPluginHandle;
|
Chris@16
|
71 }
|
Chris@16
|
72 if (!pd || !pd->plugin) {
|
Chris@16
|
73 PyErr_SetString(PyExc_AttributeError,
|
Chris@39
|
74 "Invalid or already deleted plugin handle.");
|
Chris@16
|
75 return 0;
|
Chris@0
|
76 } else {
|
Chris@16
|
77 return pd;
|
Chris@0
|
78 }
|
Chris@0
|
79 }
|
Chris@0
|
80
|
Chris@34
|
81 static
|
Chris@34
|
82 PyObject *
|
Chris@34
|
83 pystr(const string &s)
|
Chris@34
|
84 {
|
Chris@102
|
85 #if PY_MAJOR_VERSION < 3
|
Chris@34
|
86 return PyString_FromString(s.c_str());
|
Chris@102
|
87 #else
|
Chris@102
|
88 return PyUnicode_FromString(s.c_str());
|
Chris@102
|
89 #endif
|
Chris@102
|
90 }
|
Chris@102
|
91
|
Chris@102
|
92 static
|
Chris@102
|
93 string
|
Chris@102
|
94 py2str(PyObject *obj)
|
Chris@102
|
95 {
|
Chris@102
|
96 #if PY_MAJOR_VERSION < 3
|
Chris@102
|
97 return PyString_AsString(obj);
|
Chris@102
|
98 #else
|
Chris@102
|
99 return PyBytes_AsString(PyUnicode_AsUTF8String(obj));
|
Chris@102
|
100 #endif
|
Chris@34
|
101 }
|
Chris@34
|
102
|
Chris@31
|
103 PyObject *
|
Chris@31
|
104 PyPluginObject_From_Plugin(Plugin *plugin)
|
Chris@0
|
105 {
|
Chris@31
|
106 PyPluginObject *pd =
|
Chris@31
|
107 (PyPluginObject *)PyType_GenericAlloc(&Plugin_Type, 0);
|
Chris@21
|
108 pd->plugin = plugin;
|
Chris@21
|
109 pd->isInitialised = false;
|
Chris@21
|
110 pd->channels = 0;
|
Chris@21
|
111 pd->blockSize = 0;
|
Chris@21
|
112 pd->stepSize = 0;
|
Chris@34
|
113
|
Chris@34
|
114 PyObject *infodict = PyDict_New();
|
Chris@34
|
115 PyDict_SetItemString
|
Chris@109
|
116 (infodict, "apiVersion", PyLong_FromLong(plugin->getVampApiVersion()));
|
Chris@34
|
117 PyDict_SetItemString
|
Chris@109
|
118 (infodict, "pluginVersion", PyLong_FromLong(plugin->getPluginVersion()));
|
Chris@34
|
119 PyDict_SetItemString
|
Chris@34
|
120 (infodict, "identifier", pystr(plugin->getIdentifier()));
|
Chris@34
|
121 PyDict_SetItemString
|
Chris@34
|
122 (infodict, "name", pystr(plugin->getName()));
|
Chris@34
|
123 PyDict_SetItemString
|
Chris@34
|
124 (infodict, "description", pystr(plugin->getDescription()));
|
Chris@34
|
125 PyDict_SetItemString
|
Chris@34
|
126 (infodict, "maker", pystr(plugin->getMaker()));
|
Chris@34
|
127 PyDict_SetItemString
|
Chris@34
|
128 (infodict, "copyright", pystr(plugin->getCopyright()));
|
Chris@34
|
129 pd->info = infodict;
|
Chris@34
|
130
|
Chris@82
|
131 pd->inputDomain = plugin->getInputDomain();
|
Chris@34
|
132
|
Chris@35
|
133 VectorConversion conv;
|
Chris@35
|
134
|
Chris@34
|
135 Plugin::ParameterList pl = plugin->getParameterDescriptors();
|
Chris@34
|
136 PyObject *params = PyList_New(pl.size());
|
Chris@34
|
137
|
Chris@34
|
138 for (int i = 0; i < (int)pl.size(); ++i) {
|
Chris@34
|
139 PyObject *paramdict = PyDict_New();
|
Chris@34
|
140 PyDict_SetItemString
|
Chris@34
|
141 (paramdict, "identifier", pystr(pl[i].identifier));
|
Chris@34
|
142 PyDict_SetItemString
|
Chris@34
|
143 (paramdict, "name", pystr(pl[i].name));
|
Chris@34
|
144 PyDict_SetItemString
|
Chris@34
|
145 (paramdict, "description", pystr(pl[i].description));
|
Chris@34
|
146 PyDict_SetItemString
|
Chris@34
|
147 (paramdict, "unit", pystr(pl[i].unit));
|
Chris@34
|
148 PyDict_SetItemString
|
Chris@109
|
149 (paramdict, "minValue", PyFloat_FromDouble(pl[i].minValue));
|
Chris@34
|
150 PyDict_SetItemString
|
Chris@109
|
151 (paramdict, "maxValue", PyFloat_FromDouble(pl[i].maxValue));
|
Chris@34
|
152 PyDict_SetItemString
|
Chris@109
|
153 (paramdict, "defaultValue", PyFloat_FromDouble(pl[i].defaultValue));
|
Chris@34
|
154 if (pl[i].isQuantized) {
|
Chris@34
|
155 PyDict_SetItemString
|
Chris@109
|
156 (paramdict, "isQuantized", Py_True);
|
Chris@34
|
157 PyDict_SetItemString
|
Chris@109
|
158 (paramdict, "quantizeStep", PyFloat_FromDouble(pl[i].quantizeStep));
|
Chris@34
|
159 if (!pl[i].valueNames.empty()) {
|
Chris@34
|
160 PyDict_SetItemString
|
Chris@109
|
161 (paramdict, "valueNames", conv.PyValue_From_StringVector(pl[i].valueNames));
|
Chris@34
|
162 }
|
Chris@34
|
163 } else {
|
Chris@34
|
164 PyDict_SetItemString
|
Chris@109
|
165 (paramdict, "isQuantized", Py_False);
|
Chris@34
|
166 }
|
Chris@34
|
167
|
Chris@34
|
168 PyList_SET_ITEM(params, i, paramdict);
|
Chris@34
|
169 }
|
Chris@34
|
170
|
Chris@34
|
171 pd->parameters = params;
|
Chris@39
|
172
|
Chris@39
|
173 Plugin::ProgramList prl = plugin->getPrograms();
|
Chris@39
|
174 PyObject *progs = PyList_New(prl.size());
|
Chris@39
|
175
|
Chris@39
|
176 for (int i = 0; i < (int)prl.size(); ++i) {
|
Chris@39
|
177 PyList_SET_ITEM(progs, i, pystr(prl[i]));
|
Chris@39
|
178 }
|
Chris@39
|
179
|
Chris@39
|
180 pd->programs = progs;
|
Chris@37
|
181
|
Chris@37
|
182 return (PyObject *)pd;
|
Chris@37
|
183 }
|
Chris@35
|
184
|
Chris@37
|
185 static void
|
Chris@37
|
186 PyPluginObject_dealloc(PyPluginObject *self)
|
Chris@37
|
187 {
|
Chris@37
|
188 delete self->plugin;
|
Chris@37
|
189 PyObject_Del(self);
|
Chris@37
|
190 }
|
Chris@37
|
191
|
Chris@37
|
192 static PyObject *
|
Chris@87
|
193 convertOutput(const Plugin::OutputDescriptor &desc, int ix)
|
Chris@86
|
194 {
|
Chris@93
|
195 VectorConversion conv;
|
Chris@93
|
196
|
Chris@86
|
197 PyObject *outdict = PyDict_New();
|
Chris@86
|
198 PyDict_SetItemString
|
Chris@86
|
199 (outdict, "identifier", pystr(desc.identifier));
|
Chris@86
|
200 PyDict_SetItemString
|
Chris@86
|
201 (outdict, "name", pystr(desc.name));
|
Chris@86
|
202 PyDict_SetItemString
|
Chris@86
|
203 (outdict, "description", pystr(desc.description));
|
Chris@86
|
204 PyDict_SetItemString
|
Chris@93
|
205 (outdict, "unit", pystr(desc.unit));
|
Chris@93
|
206 PyDict_SetItemString
|
Chris@109
|
207 (outdict, "hasFixedBinCount", PyLong_FromLong(desc.hasFixedBinCount));
|
Chris@93
|
208 if (desc.hasFixedBinCount) {
|
Chris@93
|
209 PyDict_SetItemString
|
Chris@109
|
210 (outdict, "binCount", PyLong_FromLong(desc.binCount));
|
Chris@93
|
211 if (!desc.binNames.empty()) {
|
Chris@93
|
212 PyDict_SetItemString
|
Chris@109
|
213 (outdict, "binNames", conv.PyValue_From_StringVector(desc.binNames));
|
Chris@93
|
214 }
|
Chris@93
|
215 }
|
Chris@93
|
216 if (!desc.hasFixedBinCount ||
|
Chris@93
|
217 (desc.hasFixedBinCount && (desc.binCount > 0))) {
|
Chris@86
|
218 if (desc.hasKnownExtents) {
|
Chris@86
|
219 PyDict_SetItemString
|
Chris@109
|
220 (outdict, "hasKnownExtents", Py_True);
|
Chris@86
|
221 PyDict_SetItemString
|
Chris@109
|
222 (outdict, "minValue", PyFloat_FromDouble(desc.minValue));
|
Chris@86
|
223 PyDict_SetItemString
|
Chris@109
|
224 (outdict, "maxValue", PyFloat_FromDouble(desc.maxValue));
|
Chris@86
|
225 } else {
|
Chris@86
|
226 PyDict_SetItemString
|
Chris@109
|
227 (outdict, "hasKnownExtents", Py_False);
|
Chris@86
|
228 }
|
Chris@86
|
229 if (desc.isQuantized) {
|
Chris@86
|
230 PyDict_SetItemString
|
Chris@109
|
231 (outdict, "isQuantized", Py_True);
|
Chris@86
|
232 PyDict_SetItemString
|
Chris@109
|
233 (outdict, "quantizeStep", PyFloat_FromDouble(desc.quantizeStep));
|
Chris@86
|
234 } else {
|
Chris@86
|
235 PyDict_SetItemString
|
Chris@109
|
236 (outdict, "isQuantized", Py_False);
|
Chris@86
|
237 }
|
Chris@86
|
238 }
|
Chris@86
|
239 PyDict_SetItemString
|
Chris@109
|
240 (outdict, "sampleType", PyLong_FromLong((int)desc.sampleType));
|
Chris@86
|
241 PyDict_SetItemString
|
Chris@109
|
242 (outdict, "sampleRate", PyFloat_FromDouble(desc.sampleRate));
|
Chris@86
|
243 PyDict_SetItemString
|
Chris@109
|
244 (outdict, "hasDuration", desc.hasDuration ? Py_True : Py_False);
|
Chris@87
|
245 PyDict_SetItemString
|
Chris@109
|
246 (outdict, "outputIndex", PyLong_FromLong(ix));
|
Chris@86
|
247 return outdict;
|
Chris@86
|
248 }
|
Chris@86
|
249
|
Chris@86
|
250 static PyObject *
|
Chris@86
|
251 get_output(PyObject *self, PyObject *args)
|
Chris@86
|
252 {
|
Chris@86
|
253 PyPluginObject *pd = getPluginObject(self);
|
Chris@86
|
254 if (!pd) return 0;
|
Chris@86
|
255
|
Chris@86
|
256 int n = -1;
|
Chris@86
|
257 PyObject *pyId = 0;
|
Chris@86
|
258
|
Chris@86
|
259 if (!PyArg_ParseTuple(args, "n", &n) &&
|
Chris@86
|
260 !PyArg_ParseTuple(args, "S", &pyId)) {
|
Chris@86
|
261 PyErr_SetString(PyExc_TypeError,
|
Chris@86
|
262 "get_output takes either output id (string) or output index (int) argument");
|
Chris@86
|
263 return 0;
|
Chris@86
|
264 }
|
Chris@86
|
265
|
Chris@87
|
266 PyErr_Clear();
|
Chris@87
|
267
|
Chris@86
|
268 Plugin::OutputList ol = pd->plugin->getOutputDescriptors();
|
Chris@86
|
269
|
Chris@86
|
270 if (pyId) {
|
Chris@102
|
271 string id = py2str(pyId);
|
Chris@86
|
272 for (int i = 0; i < int(ol.size()); ++i) {
|
Chris@86
|
273 if (ol[i].identifier == id) {
|
Chris@87
|
274 return convertOutput(ol[i], i);
|
Chris@86
|
275 }
|
Chris@86
|
276 }
|
Chris@86
|
277 } else {
|
Chris@86
|
278 if (n >= 0 && n < int(ol.size())) {
|
Chris@87
|
279 return convertOutput(ol[n], n);
|
Chris@86
|
280 }
|
Chris@86
|
281 }
|
Chris@86
|
282
|
Chris@102
|
283 PyErr_SetString(PyExc_Exception,
|
Chris@86
|
284 "unknown output id or output index out of range");
|
Chris@86
|
285 return 0;
|
Chris@86
|
286 }
|
Chris@86
|
287
|
Chris@86
|
288 static PyObject *
|
Chris@80
|
289 get_outputs(PyObject *self, PyObject *args)
|
Chris@37
|
290 {
|
Chris@37
|
291 PyPluginObject *pd = getPluginObject(self);
|
Chris@37
|
292 if (!pd) return 0;
|
Chris@37
|
293 Plugin::OutputList ol = pd->plugin->getOutputDescriptors();
|
Chris@35
|
294 PyObject *outputs = PyList_New(ol.size());
|
Chris@35
|
295
|
Chris@35
|
296 for (int i = 0; i < (int)ol.size(); ++i) {
|
Chris@87
|
297 PyObject *outdict = convertOutput(ol[i], i);
|
Chris@35
|
298 PyList_SET_ITEM(outputs, i, outdict);
|
Chris@35
|
299 }
|
Chris@35
|
300
|
Chris@37
|
301 return outputs;
|
Chris@33
|
302 }
|
Chris@33
|
303
|
Chris@0
|
304 static PyObject *
|
Chris@39
|
305 initialise(PyObject *self, PyObject *args)
|
Chris@0
|
306 {
|
luis@7
|
307 size_t channels, blockSize, stepSize;
|
Chris@0
|
308
|
Chris@23
|
309 if (!PyArg_ParseTuple (args, "nnn",
|
Chris@39
|
310 (size_t) &channels,
|
Chris@39
|
311 (size_t) &stepSize,
|
Chris@39
|
312 (size_t) &blockSize)) {
|
Chris@39
|
313 PyErr_SetString(PyExc_TypeError,
|
Chris@39
|
314 "initialise() takes channel count, step size, and block size arguments");
|
Chris@39
|
315 return 0;
|
Chris@0
|
316 }
|
Chris@0
|
317
|
Chris@23
|
318 PyPluginObject *pd = getPluginObject(self);
|
Chris@16
|
319 if (!pd) return 0;
|
Chris@0
|
320
|
Chris@16
|
321 pd->channels = channels;
|
Chris@16
|
322 pd->stepSize = stepSize;
|
Chris@16
|
323 pd->blockSize = blockSize;
|
Chris@0
|
324
|
Chris@16
|
325 if (!pd->plugin->initialise(channels, stepSize, blockSize)) {
|
Chris@39
|
326 cerr << "Failed to initialise native plugin adapter with channels = " << channels << ", stepSize = " << stepSize << ", blockSize = " << blockSize << endl;
|
Chris@39
|
327 PyErr_SetString(PyExc_TypeError,
|
Chris@39
|
328 "Plugin initialization failed");
|
Chris@39
|
329 return 0;
|
Chris@6
|
330 }
|
Chris@0
|
331
|
Chris@16
|
332 pd->isInitialised = true;
|
luis@7
|
333
|
Chris@0
|
334 return Py_True;
|
Chris@0
|
335 }
|
Chris@0
|
336
|
Chris@0
|
337 static PyObject *
|
Chris@39
|
338 reset(PyObject *self, PyObject *)
|
Chris@18
|
339 {
|
Chris@23
|
340 PyPluginObject *pd = getPluginObject(self);
|
Chris@18
|
341 if (!pd) return 0;
|
Chris@18
|
342
|
Chris@18
|
343 if (!pd->isInitialised) {
|
Chris@102
|
344 PyErr_SetString(PyExc_Exception,
|
Chris@18
|
345 "Plugin has not been initialised");
|
Chris@18
|
346 return 0;
|
Chris@18
|
347 }
|
Chris@18
|
348
|
Chris@18
|
349 pd->plugin->reset();
|
Chris@18
|
350 return Py_True;
|
Chris@18
|
351 }
|
Chris@18
|
352
|
Chris@49
|
353 static bool
|
Chris@49
|
354 hasParameter(PyPluginObject *pd, string id)
|
Chris@49
|
355 {
|
Chris@49
|
356 PluginBase::ParameterList pl = pd->plugin->getParameterDescriptors();
|
Chris@49
|
357 for (int i = 0; i < (int)pl.size(); ++i) {
|
Chris@49
|
358 if (pl[i].identifier == id) {
|
Chris@49
|
359 return true;
|
Chris@49
|
360 }
|
Chris@49
|
361 }
|
Chris@49
|
362 return false;
|
Chris@49
|
363 }
|
Chris@49
|
364
|
Chris@18
|
365 static PyObject *
|
Chris@80
|
366 get_parameter_value(PyObject *self, PyObject *args)
|
Chris@20
|
367 {
|
Chris@20
|
368 PyObject *pyParam;
|
Chris@20
|
369
|
Chris@23
|
370 if (!PyArg_ParseTuple(args, "S", &pyParam)) {
|
Chris@39
|
371 PyErr_SetString(PyExc_TypeError,
|
Chris@80
|
372 "get_parameter_value() takes parameter id (string) argument");
|
Chris@39
|
373 return 0; }
|
Chris@20
|
374
|
Chris@23
|
375 PyPluginObject *pd = getPluginObject(self);
|
Chris@20
|
376 if (!pd) return 0;
|
Chris@20
|
377
|
Chris@102
|
378 string param = py2str(pyParam);
|
Chris@49
|
379
|
Chris@49
|
380 if (!hasParameter(pd, param)) {
|
Chris@102
|
381 PyErr_SetString(PyExc_Exception,
|
Chris@49
|
382 (string("Unknown parameter id \"") + param + "\"").c_str());
|
Chris@49
|
383 return 0;
|
Chris@49
|
384 }
|
Chris@49
|
385
|
Chris@49
|
386 float value = pd->plugin->getParameter(param);
|
Chris@20
|
387 return PyFloat_FromDouble(double(value));
|
Chris@20
|
388 }
|
Chris@20
|
389
|
Chris@20
|
390 static PyObject *
|
Chris@80
|
391 set_parameter_value(PyObject *self, PyObject *args)
|
Chris@20
|
392 {
|
Chris@20
|
393 PyObject *pyParam;
|
Chris@20
|
394 float value;
|
Chris@20
|
395
|
Chris@23
|
396 if (!PyArg_ParseTuple(args, "Sf", &pyParam, &value)) {
|
Chris@39
|
397 PyErr_SetString(PyExc_TypeError,
|
Chris@80
|
398 "set_parameter_value() takes parameter id (string), and value (float) arguments");
|
Chris@39
|
399 return 0; }
|
Chris@20
|
400
|
Chris@23
|
401 PyPluginObject *pd = getPluginObject(self);
|
Chris@20
|
402 if (!pd) return 0;
|
Chris@20
|
403
|
Chris@102
|
404 string param = py2str(pyParam);
|
Chris@49
|
405
|
Chris@49
|
406 if (!hasParameter(pd, param)) {
|
Chris@102
|
407 PyErr_SetString(PyExc_Exception,
|
Chris@49
|
408 (string("Unknown parameter id \"") + param + "\"").c_str());
|
Chris@49
|
409 return 0;
|
Chris@49
|
410 }
|
Chris@49
|
411
|
Chris@49
|
412 pd->plugin->setParameter(param, value);
|
Chris@20
|
413 return Py_True;
|
Chris@20
|
414 }
|
Chris@20
|
415
|
Chris@39
|
416 static PyObject *
|
Chris@80
|
417 set_parameter_values(PyObject *self, PyObject *args)
|
Chris@48
|
418 {
|
Chris@48
|
419 PyObject *dict;
|
Chris@48
|
420
|
Chris@48
|
421 if (!PyArg_ParseTuple(args, "O", &dict)) {
|
Chris@48
|
422 PyErr_SetString(PyExc_TypeError,
|
Chris@80
|
423 "set_parameter_values() takes dict argument");
|
Chris@48
|
424 return 0; }
|
Chris@48
|
425
|
Chris@48
|
426 if (!PyDict_Check(dict)) {
|
Chris@48
|
427 PyErr_SetString(PyExc_TypeError,
|
Chris@80
|
428 "set_parameter_values() takes dict argument");
|
Chris@48
|
429 return 0; }
|
Chris@48
|
430
|
Chris@48
|
431 PyPluginObject *pd = getPluginObject(self);
|
Chris@48
|
432 if (!pd) return 0;
|
Chris@48
|
433
|
Chris@49
|
434 PluginBase::ParameterList pl = pd->plugin->getParameterDescriptors();
|
Chris@49
|
435 set<string> paramIds;
|
Chris@49
|
436 for (int i = 0; i < (int)pl.size(); ++i) {
|
Chris@49
|
437 paramIds.insert(pl[i].identifier);
|
Chris@49
|
438 }
|
Chris@49
|
439
|
Chris@48
|
440 Py_ssize_t pos = 0;
|
Chris@48
|
441 PyObject *key, *value;
|
Chris@48
|
442 while (PyDict_Next(dict, &pos, &key, &value)) {
|
Chris@102
|
443 #if PY_MAJOR_VERSION >= 3
|
Chris@102
|
444 if (!key || !PyUnicode_CheckExact(key)) {
|
Chris@102
|
445 #else
|
Chris@48
|
446 if (!key || !PyString_CheckExact(key)) {
|
Chris@102
|
447 #endif
|
Chris@48
|
448 PyErr_SetString(PyExc_TypeError,
|
Chris@48
|
449 "Parameter dict keys must all have string type");
|
Chris@48
|
450 return 0;
|
Chris@48
|
451 }
|
Chris@48
|
452 if (!value || !FloatConversion::check(value)) {
|
Chris@48
|
453 PyErr_SetString(PyExc_TypeError,
|
Chris@48
|
454 "Parameter dict values must be convertible to float");
|
Chris@48
|
455 return 0;
|
Chris@48
|
456 }
|
Chris@102
|
457 string param = py2str(key);
|
Chris@49
|
458 if (paramIds.find(param) == paramIds.end()) {
|
Chris@102
|
459 PyErr_SetString(PyExc_Exception,
|
Chris@49
|
460 (string("Unknown parameter id \"") + param + "\"").c_str());
|
Chris@49
|
461 return 0;
|
Chris@49
|
462 }
|
Chris@49
|
463 pd->plugin->setParameter(param, FloatConversion::convert(value));
|
Chris@48
|
464 }
|
Chris@48
|
465
|
Chris@48
|
466 return Py_True;
|
Chris@48
|
467 }
|
Chris@48
|
468
|
Chris@48
|
469 static PyObject *
|
Chris@80
|
470 select_program(PyObject *self, PyObject *args)
|
Chris@39
|
471 {
|
Chris@39
|
472 PyObject *pyParam;
|
Chris@39
|
473
|
Chris@39
|
474 if (!PyArg_ParseTuple(args, "S", &pyParam)) {
|
Chris@39
|
475 PyErr_SetString(PyExc_TypeError,
|
Chris@80
|
476 "select_program() takes parameter id (string) argument");
|
Chris@39
|
477 return 0; }
|
Chris@39
|
478
|
Chris@39
|
479 PyPluginObject *pd = getPluginObject(self);
|
Chris@39
|
480 if (!pd) return 0;
|
Chris@39
|
481
|
Chris@102
|
482 pd->plugin->selectProgram(py2str(pyParam));
|
Chris@39
|
483 return Py_True;
|
Chris@39
|
484 }
|
Chris@39
|
485
|
Chris@35
|
486 static
|
Chris@35
|
487 PyObject *
|
Chris@35
|
488 convertFeatureSet(const Plugin::FeatureSet &fs)
|
Chris@35
|
489 {
|
Chris@35
|
490 VectorConversion conv;
|
Chris@35
|
491
|
Chris@35
|
492 PyObject *pyFs = PyDict_New();
|
Chris@35
|
493
|
Chris@35
|
494 for (Plugin::FeatureSet::const_iterator fsi = fs.begin();
|
Chris@35
|
495 fsi != fs.end(); ++fsi) {
|
Chris@35
|
496
|
Chris@35
|
497 int fno = fsi->first;
|
Chris@35
|
498 const Plugin::FeatureList &fl = fsi->second;
|
Chris@35
|
499
|
Chris@35
|
500 if (!fl.empty()) {
|
Chris@35
|
501
|
Chris@35
|
502 PyObject *pyFl = PyList_New(fl.size());
|
Chris@35
|
503
|
Chris@35
|
504 for (int fli = 0; fli < (int)fl.size(); ++fli) {
|
Chris@35
|
505
|
Chris@35
|
506 const Plugin::Feature &f = fl[fli];
|
Chris@35
|
507 PyObject *pyF = PyDict_New();
|
Chris@35
|
508
|
Chris@35
|
509 if (f.hasTimestamp) {
|
Chris@35
|
510 PyDict_SetItemString
|
Chris@35
|
511 (pyF, "timestamp", PyRealTime_FromRealTime(f.timestamp));
|
Chris@35
|
512 }
|
Chris@35
|
513 if (f.hasDuration) {
|
Chris@35
|
514 PyDict_SetItemString
|
Chris@35
|
515 (pyF, "duration", PyRealTime_FromRealTime(f.duration));
|
Chris@35
|
516 }
|
Chris@35
|
517
|
Chris@35
|
518 PyDict_SetItemString
|
Chris@35
|
519 (pyF, "label", pystr(f.label));
|
Chris@35
|
520
|
Chris@35
|
521 if (!f.values.empty()) {
|
Chris@35
|
522 PyDict_SetItemString
|
Chris@35
|
523 (pyF, "values", conv.PyArray_From_FloatVector(f.values));
|
Chris@35
|
524 }
|
Chris@35
|
525
|
Chris@35
|
526 PyList_SET_ITEM(pyFl, fli, pyF);
|
Chris@35
|
527 }
|
Chris@35
|
528
|
Chris@102
|
529 PyObject *pyN = PyLong_FromLong(fno);
|
Chris@35
|
530 PyDict_SetItem(pyFs, pyN, pyFl);
|
Chris@35
|
531 }
|
Chris@35
|
532 }
|
Chris@35
|
533
|
Chris@35
|
534 return pyFs;
|
Chris@35
|
535 }
|
Chris@35
|
536
|
Chris@40
|
537 static vector<vector<float> >
|
Chris@40
|
538 convertPluginInput(PyObject *pyBuffer, int channels, int blockSize)
|
Chris@40
|
539 {
|
Chris@40
|
540 vector<vector<float> > data;
|
Chris@40
|
541
|
Chris@40
|
542 VectorConversion conv;
|
Chris@40
|
543
|
Chris@40
|
544 if (PyArray_CheckExact(pyBuffer)) {
|
Chris@40
|
545
|
Chris@40
|
546 data = conv.Py2DArray_To_FloatVector(pyBuffer);
|
Chris@40
|
547
|
Chris@40
|
548 if (conv.error) {
|
Chris@40
|
549 PyErr_SetString(PyExc_TypeError, conv.getError().str().c_str());
|
Chris@40
|
550 return data;
|
Chris@40
|
551 }
|
Chris@40
|
552
|
Chris@41
|
553 if ((int)data.size() != channels) {
|
Chris@41
|
554 // cerr << "Wrong number of channels: got " << data.size() << ", expected " << channels << endl;
|
Chris@41
|
555 PyErr_SetString(PyExc_TypeError, "Wrong number of channels");
|
Chris@41
|
556 return vector<vector<float> >();
|
Chris@41
|
557 }
|
Chris@41
|
558
|
Chris@40
|
559 } else {
|
Chris@40
|
560
|
Chris@40
|
561 if (!PyList_Check(pyBuffer)) {
|
Chris@43
|
562 PyErr_SetString(PyExc_TypeError, "List of NumPy arrays or lists of numbers required for process input");
|
Chris@40
|
563 return data;
|
Chris@40
|
564 }
|
Chris@43
|
565
|
Chris@40
|
566 if (PyList_GET_SIZE(pyBuffer) != channels) {
|
Chris@41
|
567 // cerr << "Wrong number of channels: got " << PyList_GET_SIZE(pyBuffer) << ", expected " << channels << endl;
|
Chris@40
|
568 PyErr_SetString(PyExc_TypeError, "Wrong number of channels");
|
Chris@40
|
569 return data;
|
Chris@40
|
570 }
|
Chris@40
|
571
|
Chris@40
|
572 for (int c = 0; c < channels; ++c) {
|
Chris@40
|
573 PyObject *cbuf = PyList_GET_ITEM(pyBuffer, c);
|
Chris@40
|
574 data.push_back(conv.PyValue_To_FloatVector(cbuf));
|
Chris@43
|
575 if (conv.error) {
|
Chris@43
|
576 PyErr_SetString(PyExc_TypeError, conv.getError().str().c_str());
|
Chris@43
|
577 return vector<vector<float> >();
|
Chris@43
|
578 }
|
Chris@40
|
579 }
|
Chris@41
|
580 }
|
Chris@40
|
581
|
Chris@41
|
582 for (int c = 0; c < channels; ++c) {
|
Chris@41
|
583 if ((int)data[c].size() != blockSize) {
|
Chris@41
|
584 // cerr << "Wrong number of samples on channel " << c << ": expected " << blockSize << " (plugin's block size), got " << data[c].size() << endl;
|
Chris@46
|
585 PyErr_SetString(PyExc_TypeError, "Wrong number of samples for process block");
|
Chris@41
|
586 return vector<vector<float> >();
|
Chris@40
|
587 }
|
Chris@40
|
588 }
|
Chris@40
|
589
|
Chris@40
|
590 return data;
|
Chris@40
|
591 }
|
Chris@40
|
592
|
Chris@20
|
593 static PyObject *
|
Chris@80
|
594 process_block(PyObject *self, PyObject *args)
|
Chris@0
|
595 {
|
Chris@0
|
596 PyObject *pyBuffer;
|
Chris@0
|
597 PyObject *pyRealTime;
|
Chris@0
|
598
|
Chris@23
|
599 if (!PyArg_ParseTuple(args, "OO",
|
Chris@39
|
600 &pyBuffer, // Audio data
|
Chris@39
|
601 &pyRealTime)) { // TimeStamp
|
Chris@39
|
602 PyErr_SetString(PyExc_TypeError,
|
Chris@80
|
603 "process_block() takes buffer (2D array or list of arrays, one row per channel) and timestamp (RealTime) arguments");
|
Chris@39
|
604 return 0; }
|
Chris@0
|
605
|
Chris@0
|
606 if (!PyRealTime_Check(pyRealTime)) {
|
Chris@40
|
607 PyErr_SetString(PyExc_TypeError, "Valid timestamp required.");
|
Chris@39
|
608 return 0; }
|
Chris@0
|
609
|
Chris@23
|
610 PyPluginObject *pd = getPluginObject(self);
|
Chris@16
|
611 if (!pd) return 0;
|
Chris@0
|
612
|
Chris@0
|
613 if (!pd->isInitialised) {
|
Chris@102
|
614 PyErr_SetString(PyExc_Exception,
|
Chris@39
|
615 "Plugin has not been initialised.");
|
Chris@39
|
616 return 0;
|
Chris@16
|
617 }
|
Chris@0
|
618
|
Chris@40
|
619 int channels = pd->channels;
|
Chris@40
|
620 vector<vector<float> > data =
|
Chris@40
|
621 convertPluginInput(pyBuffer, channels, pd->blockSize);
|
Chris@40
|
622 if (data.empty()) return 0;
|
Chris@0
|
623
|
Chris@4
|
624 float **inbuf = new float *[channels];
|
Chris@4
|
625 for (int c = 0; c < channels; ++c) {
|
Chris@12
|
626 inbuf[c] = &data[c][0];
|
Chris@4
|
627 }
|
Chris@12
|
628 RealTime timeStamp = *PyRealTime_AsRealTime(pyRealTime);
|
Chris@18
|
629 Plugin::FeatureSet fs = pd->plugin->process(inbuf, timeStamp);
|
Chris@4
|
630 delete[] inbuf;
|
Chris@0
|
631
|
Chris@35
|
632 return convertFeatureSet(fs);
|
Chris@35
|
633 }
|
Chris@0
|
634
|
Chris@35
|
635 static PyObject *
|
Chris@80
|
636 get_remaining_features(PyObject *self, PyObject *)
|
Chris@35
|
637 {
|
Chris@35
|
638 PyPluginObject *pd = getPluginObject(self);
|
Chris@35
|
639 if (!pd) return 0;
|
Chris@18
|
640
|
Chris@35
|
641 if (!pd->isInitialised) {
|
Chris@102
|
642 PyErr_SetString(PyExc_Exception,
|
Chris@39
|
643 "Plugin has not been initialised.");
|
Chris@39
|
644 return 0;
|
Chris@35
|
645 }
|
Chris@18
|
646
|
Chris@35
|
647 Plugin::FeatureSet fs = pd->plugin->getRemainingFeatures();
|
Chris@18
|
648
|
Chris@35
|
649 return convertFeatureSet(fs);
|
Chris@0
|
650 }
|
Chris@0
|
651
|
Chris@23
|
652 static PyObject *
|
Chris@80
|
653 get_preferred_block_size(PyObject *self, PyObject *)
|
Chris@37
|
654 {
|
Chris@37
|
655 PyPluginObject *pd = getPluginObject(self);
|
Chris@37
|
656 if (!pd) return 0;
|
Chris@102
|
657 return PyLong_FromLong(pd->plugin->getPreferredBlockSize());
|
Chris@37
|
658 }
|
Chris@37
|
659
|
Chris@37
|
660 static PyObject *
|
Chris@80
|
661 get_preferred_step_size(PyObject *self, PyObject *)
|
Chris@37
|
662 {
|
Chris@37
|
663 PyPluginObject *pd = getPluginObject(self);
|
Chris@37
|
664 if (!pd) return 0;
|
Chris@102
|
665 return PyLong_FromLong(pd->plugin->getPreferredStepSize());
|
Chris@37
|
666 }
|
Chris@37
|
667
|
Chris@37
|
668 static PyObject *
|
Chris@80
|
669 get_min_channel_count(PyObject *self, PyObject *)
|
Chris@37
|
670 {
|
Chris@37
|
671 PyPluginObject *pd = getPluginObject(self);
|
Chris@37
|
672 if (!pd) return 0;
|
Chris@102
|
673 return PyLong_FromLong(pd->plugin->getMinChannelCount());
|
Chris@37
|
674 }
|
Chris@37
|
675
|
Chris@37
|
676 static PyObject *
|
Chris@80
|
677 get_max_channel_count(PyObject *self, PyObject *)
|
Chris@37
|
678 {
|
Chris@37
|
679 PyPluginObject *pd = getPluginObject(self);
|
Chris@37
|
680 if (!pd) return 0;
|
Chris@102
|
681 return PyLong_FromLong(pd->plugin->getMaxChannelCount());
|
Chris@37
|
682 }
|
Chris@37
|
683
|
Chris@37
|
684 static PyObject *
|
Chris@39
|
685 unload(PyObject *self, PyObject *)
|
Chris@23
|
686 {
|
Chris@23
|
687 PyPluginObject *pd = getPluginObject(self);
|
Chris@23
|
688 if (!pd) return 0;
|
Chris@23
|
689
|
Chris@23
|
690 delete pd->plugin;
|
Chris@32
|
691 pd->plugin = 0; // This is checked by getPluginObject, so we avoid
|
Chris@32
|
692 // blowing up if called repeatedly
|
Chris@23
|
693
|
Chris@23
|
694 return Py_True;
|
Chris@23
|
695 }
|
Chris@23
|
696
|
Chris@33
|
697 static PyMemberDef PyPluginObject_members[] =
|
Chris@33
|
698 {
|
Chris@34
|
699 {(char *)"info", T_OBJECT, offsetof(PyPluginObject, info), READONLY,
|
Chris@39
|
700 (char *)"info -> A read-only dictionary of plugin metadata."},
|
Chris@34
|
701
|
Chris@110
|
702 {(char *)"inputDomain", T_INT, offsetof(PyPluginObject, inputDomain), READONLY,
|
Chris@110
|
703 (char *)"inputDomain -> The format of input audio required by the plugin, either vampyhost.TIME_DOMAIN or vampyhost.FREQUENCY_DOMAIN."},
|
Chris@34
|
704
|
Chris@34
|
705 {(char *)"parameters", T_OBJECT, offsetof(PyPluginObject, parameters), READONLY,
|
Chris@39
|
706 (char *)"parameters -> A list of metadata dictionaries describing the plugin's configurable parameters."},
|
Chris@39
|
707
|
Chris@39
|
708 {(char *)"programs", T_OBJECT, offsetof(PyPluginObject, programs), READONLY,
|
Chris@39
|
709 (char *)"programs -> A list of the programs available for this plugin, if any."},
|
Chris@33
|
710
|
Chris@33
|
711 {0, 0}
|
Chris@33
|
712 };
|
Chris@33
|
713
|
Chris@21
|
714 static PyMethodDef PyPluginObject_methods[] =
|
Chris@21
|
715 {
|
Chris@80
|
716 {"get_outputs", get_outputs, METH_NOARGS,
|
Chris@80
|
717 "get_outputs() -> Obtain the output descriptors for all of the plugin's outputs."},
|
Chris@37
|
718
|
Chris@86
|
719 {"get_output", get_output, METH_VARARGS,
|
Chris@86
|
720 "get_output(out) -> Obtain the output descriptor for a single output, by either id (string) or index (int)."},
|
Chris@86
|
721
|
Chris@80
|
722 {"get_parameter_value", get_parameter_value, METH_VARARGS,
|
Chris@80
|
723 "get_parameter_value(identifier) -> Return the value of the parameter with the given identifier."},
|
Chris@23
|
724
|
Chris@80
|
725 {"set_parameter_value", set_parameter_value, METH_VARARGS,
|
Chris@80
|
726 "set_parameter_value(identifier, value) -> Set the parameter with the given identifier to the given value."},
|
Chris@37
|
727
|
Chris@80
|
728 {"set_parameter_values", set_parameter_values, METH_VARARGS,
|
Chris@80
|
729 "set_parameter_values(dict) -> Set multiple parameters to values corresponding to the key/value pairs in the dict. Any parameters not mentioned in the dict are unchanged."},
|
Chris@48
|
730
|
Chris@80
|
731 {"select_program", select_program, METH_VARARGS,
|
Chris@80
|
732 "select_program(name) -> Select the processing program with the given name."},
|
Chris@39
|
733
|
Chris@80
|
734 {"get_preferred_block_size", get_preferred_block_size, METH_VARARGS,
|
Chris@80
|
735 "get_preferred_block_size() -> Return the plugin's preferred processing block size, or 0 if the plugin accepts any block size."},
|
Chris@37
|
736
|
Chris@80
|
737 {"get_preferred_step_size", get_preferred_step_size, METH_VARARGS,
|
Chris@80
|
738 "get_preferred_step_size() -> Return the plugin's preferred processing step size, or 0 if the plugin allows the host to select. If this is 0, the host should normally choose the same step as block size for time-domain plugins, or half the block size for frequency-domain plugins."},
|
Chris@37
|
739
|
Chris@80
|
740 {"get_min_channel_count", get_min_channel_count, METH_VARARGS,
|
Chris@80
|
741 "get_min_channel_count() -> Return the minimum number of channels of audio data the plugin accepts as input."},
|
Chris@37
|
742
|
Chris@80
|
743 {"get_max_channel_count", get_max_channel_count, METH_VARARGS,
|
Chris@80
|
744 "get_max_channel_count() -> Return the maximum number of channels of audio data the plugin accepts as input."},
|
Chris@33
|
745
|
Chris@39
|
746 {"initialise", initialise, METH_VARARGS,
|
Chris@80
|
747 "initialise(channels, stepSize, blockSize) -> Initialise the plugin for the given number of channels and processing frame sizes. This must be called before process_block() can be used."},
|
Chris@23
|
748
|
Chris@39
|
749 {"reset", reset, METH_NOARGS,
|
Chris@39
|
750 "reset() -> Reset the plugin after processing, to prepare for another processing run with the same parameters."},
|
Chris@23
|
751
|
Chris@80
|
752 {"process_block", process_block, METH_VARARGS,
|
Chris@80
|
753 "process_block(block, timestamp) -> Provide one processing frame to the plugin, with its timestamp, and obtain any features that were extracted immediately from this frame."},
|
Chris@23
|
754
|
Chris@80
|
755 {"get_remaining_features", get_remaining_features, METH_NOARGS,
|
Chris@80
|
756 "get_remaining_features() -> Obtain any features extracted at the end of processing."},
|
Chris@35
|
757
|
Chris@39
|
758 {"unload", unload, METH_NOARGS,
|
Chris@39
|
759 "unload() -> Dispose of the plugin. You cannot use the plugin object again after calling this. Note that unloading also happens automatically when the plugin object's reference count reaches zero; this function is only necessary if you wish to ensure the native part of the plugin is disposed of before then."},
|
Chris@23
|
760
|
Chris@21
|
761 {0, 0}
|
Chris@21
|
762 };
|
Chris@21
|
763
|
Chris@21
|
764 /* Doc:: 10.3 Type Objects */ /* static */
|
Chris@21
|
765 PyTypeObject Plugin_Type =
|
Chris@21
|
766 {
|
Chris@21
|
767 PyObject_HEAD_INIT(NULL)
|
Chris@39
|
768 0, /*ob_size*/
|
Chris@39
|
769 "vampyhost.Plugin", /*tp_name*/
|
Chris@39
|
770 sizeof(PyPluginObject), /*tp_basicsize*/
|
Chris@39
|
771 0, /*tp_itemsize*/
|
Chris@21
|
772 (destructor)PyPluginObject_dealloc, /*tp_dealloc*/
|
Chris@39
|
773 0, /*tp_print*/
|
Chris@39
|
774 0, /*tp_getattr*/
|
Chris@39
|
775 0, /*tp_setattr*/
|
Chris@39
|
776 0, /*tp_compare*/
|
Chris@39
|
777 0, /*tp_repr*/
|
Chris@39
|
778 0, /*tp_as_number*/
|
Chris@39
|
779 0, /*tp_as_sequence*/
|
Chris@39
|
780 0, /*tp_as_mapping*/
|
Chris@39
|
781 0, /*tp_hash*/
|
Chris@39
|
782 0, /*tp_call*/
|
Chris@39
|
783 0, /*tp_str*/
|
Chris@39
|
784 PyObject_GenericGetAttr, /*tp_getattro*/
|
Chris@39
|
785 PyObject_GenericSetAttr, /*tp_setattro*/
|
Chris@39
|
786 0, /*tp_as_buffer*/
|
Chris@39
|
787 Py_TPFLAGS_DEFAULT, /*tp_flags*/
|
Chris@40
|
788 "Plugin object, providing a low-level API for running a Vamp plugin.", /*tp_doc*/
|
Chris@39
|
789 0, /*tp_traverse*/
|
Chris@39
|
790 0, /*tp_clear*/
|
Chris@39
|
791 0, /*tp_richcompare*/
|
Chris@39
|
792 0, /*tp_weaklistoffset*/
|
Chris@39
|
793 0, /*tp_iter*/
|
Chris@39
|
794 0, /*tp_iternext*/
|
Chris@39
|
795 PyPluginObject_methods, /*tp_methods*/
|
Chris@39
|
796 PyPluginObject_members, /*tp_members*/
|
Chris@39
|
797 0, /*tp_getset*/
|
Chris@39
|
798 0, /*tp_base*/
|
Chris@39
|
799 0, /*tp_dict*/
|
Chris@39
|
800 0, /*tp_descr_get*/
|
Chris@39
|
801 0, /*tp_descr_set*/
|
Chris@39
|
802 0, /*tp_dictoffset*/
|
Chris@39
|
803 0, /*tp_init*/
|
Chris@39
|
804 PyType_GenericAlloc, /*tp_alloc*/
|
Chris@39
|
805 0, /*tp_new*/
|
Chris@39
|
806 PyObject_Del, /*tp_free*/
|
Chris@39
|
807 0, /*tp_is_gc*/
|
Chris@21
|
808 };
|
Chris@0
|
809
|