annotate native/PyPluginObject.cpp @ 102:216ed5a72c36

Some preliminary changes toward py2/3 compatibility (but the module load stuff remains quite a big deal)
author Chris Cannam
date Tue, 10 Feb 2015 12:28:53 +0000
parents 4bed6bf67243
children cf56111935fa
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@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@102 116 (infodict, "api_version", PyLong_FromLong(plugin->getVampApiVersion()));
Chris@34 117 PyDict_SetItemString
Chris@102 118 (infodict, "plugin_version", 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@83 149 (paramdict, "min_value", PyFloat_FromDouble(pl[i].minValue));
Chris@34 150 PyDict_SetItemString
Chris@83 151 (paramdict, "max_value", PyFloat_FromDouble(pl[i].maxValue));
Chris@34 152 PyDict_SetItemString
Chris@83 153 (paramdict, "default_value", PyFloat_FromDouble(pl[i].defaultValue));
Chris@34 154 if (pl[i].isQuantized) {
Chris@34 155 PyDict_SetItemString
Chris@83 156 (paramdict, "is_quantized", Py_True);
Chris@34 157 PyDict_SetItemString
Chris@83 158 (paramdict, "quantize_step", PyFloat_FromDouble(pl[i].quantizeStep));
Chris@34 159 if (!pl[i].valueNames.empty()) {
Chris@34 160 PyDict_SetItemString
Chris@83 161 (paramdict, "value_names", conv.PyValue_From_StringVector(pl[i].valueNames));
Chris@34 162 }
Chris@34 163 } else {
Chris@34 164 PyDict_SetItemString
Chris@83 165 (paramdict, "is_quantized", 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@102 207 (outdict, "has_fixed_bin_count", PyLong_FromLong(desc.hasFixedBinCount));
Chris@93 208 if (desc.hasFixedBinCount) {
Chris@93 209 PyDict_SetItemString
Chris@102 210 (outdict, "bin_count", PyLong_FromLong(desc.binCount));
Chris@93 211 if (!desc.binNames.empty()) {
Chris@93 212 PyDict_SetItemString
Chris@93 213 (outdict, "bin_names", 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@86 220 (outdict, "has_known_extents", Py_True);
Chris@86 221 PyDict_SetItemString
Chris@86 222 (outdict, "min_value", PyFloat_FromDouble(desc.minValue));
Chris@86 223 PyDict_SetItemString
Chris@86 224 (outdict, "max_value", PyFloat_FromDouble(desc.maxValue));
Chris@86 225 } else {
Chris@86 226 PyDict_SetItemString
Chris@86 227 (outdict, "has_known_extents", Py_False);
Chris@86 228 }
Chris@86 229 if (desc.isQuantized) {
Chris@86 230 PyDict_SetItemString
Chris@86 231 (outdict, "is_quantized", Py_True);
Chris@86 232 PyDict_SetItemString
Chris@86 233 (outdict, "quantize_step", PyFloat_FromDouble(desc.quantizeStep));
Chris@86 234 } else {
Chris@86 235 PyDict_SetItemString
Chris@86 236 (outdict, "is_quantized", Py_False);
Chris@86 237 }
Chris@86 238 }
Chris@86 239 PyDict_SetItemString
Chris@102 240 (outdict, "sample_type", PyLong_FromLong((int)desc.sampleType));
Chris@86 241 PyDict_SetItemString
Chris@86 242 (outdict, "sample_rate", PyFloat_FromDouble(desc.sampleRate));
Chris@86 243 PyDict_SetItemString
Chris@86 244 (outdict, "has_duration", desc.hasDuration ? Py_True : Py_False);
Chris@87 245 PyDict_SetItemString
Chris@102 246 (outdict, "output_index", 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@82 702 {(char *)"input_domain", T_INT, offsetof(PyPluginObject, inputDomain), READONLY,
Chris@81 703 (char *)"input_domain -> 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