annotate native/PyPluginObject.cpp @ 124:bea7cf4126b5

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