Chris@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@0: 
Chris@0: //include for python extension module: must be first
Chris@0: #include <Python.h>
Chris@0: #include <vampyhost.h>
Chris@0: #include <pyRealTime.h>
Chris@0: 
Chris@0: //includes for vamp host
Chris@1: #include "vamp-hostsdk/Plugin.h"
Chris@1: #include "vamp-hostsdk/PluginHostAdapter.h"
Chris@1: #include "vamp-hostsdk/PluginChannelAdapter.h"
Chris@1: #include "vamp-hostsdk/PluginInputDomainAdapter.h"
Chris@1: #include "vamp-hostsdk/PluginLoader.h"
Chris@0: //#include "vamp/vamp.h"
Chris@0: 
Chris@0: #include <iostream>
Chris@0: #include <fstream>
Chris@0: #include <set>
Chris@0: #include <sndfile.h>
Chris@0: 
Chris@0: #include <cstring>
Chris@0: #include <cstdlib>
Chris@0: #include <string>
Chris@0: 
Chris@0: #include "system.h"
Chris@0: 
Chris@0: #include <cmath>
Chris@0: 
Chris@0: 
Chris@0: using namespace std;
Chris@0: using namespace Vamp;
Chris@0: 
Chris@0: using Vamp::Plugin;
Chris@0: using Vamp::PluginHostAdapter;
Chris@0: using Vamp::RealTime;
Chris@0: using Vamp::HostExt::PluginLoader;
Chris@0: 
Chris@0: #define HOST_VERSION "1.1"
Chris@0: 
Chris@0: 
Chris@0: /* MODULE HELPER FUNCTIONS */
Chris@2: PyDoc_STRVAR(xx_foo_doc, "Some description"); //!!!
Chris@0: 
Chris@0: /*obtain C plugin handle and key from pyCobject */
Chris@2: bool getPluginHandle 
Chris@0: (PyObject *pyPluginHandle, Plugin **plugin, string **pKey=NULL) {
Chris@0: 
Chris@0:     //char errormsg[]="Wrong input argument: Plugin Handle required.";
Chris@0: 
Chris@0:     *plugin = NULL;
Chris@2:     if (!PyCObject_Check(pyPluginHandle)) return false;
Chris@0: 
Chris@0:     //try to convert to Plugin pointer
Chris@0:     Plugin *p = (Plugin*) PyCObject_AsVoidPtr(pyPluginHandle);
Chris@2:     if (!p) return false;	
Chris@0: 
Chris@0:     string pId;
Chris@0: 
Chris@0:     if (pKey) {
Chris@0: 	*pKey = (string*) PyCObject_GetDesc(pyPluginHandle);
Chris@2: 	if (!*pKey) return false;
Chris@0: 	pId = *(string*) *pKey;
Chris@0: 
Chris@0:     } else {
Chris@0: 
Chris@0: 	void *pKey = PyCObject_GetDesc(pyPluginHandle);
Chris@2: 	if (!pKey) return false;
Chris@0: 	pId = *(string*) pKey;
Chris@0:     }
Chris@0: 	
Chris@0:     string::size_type pos = pId.find(':');
Chris@2:     if (pos == string::npos) return false;
Chris@0: 
Chris@0:     pId = pId.substr(pId.rfind(':')+1);
Chris@0:     string identifier = p->getIdentifier();	
Chris@0: 			
Chris@2:     if (pId.compare(identifier)) return false;
Chris@0: 
Chris@0:     *plugin = p;
Chris@0:     return true;
Chris@0: }
Chris@0: 
Chris@0: /*
Chris@0:   ----------------------------------------------------------------
Chris@0: */
Chris@0: 
Chris@0: 
Chris@0: 
Chris@0: /*
Chris@0:   VAMPYHOST MAIN 
Chris@0:   --------------------------------------------------------------------- 
Chris@0: */
Chris@0: 
Chris@0: /* ENUMERATE PLUGINS*/
Chris@0: 
Chris@0: static PyObject *
Chris@0: vampyhost_enumeratePlugins(PyObject *self, PyObject *args)
Chris@0: {
Chris@0:     string retType;
Chris@0: 	
Chris@0:     if (!PyArg_ParseTuple(args, "|s:enumeratePlugins", &retType))
Chris@0: 	return NULL;
Chris@0: 
Chris@0:     //list available plugins
Chris@0:     PluginLoader *loader = PluginLoader::getInstance();
Chris@0:     vector<PluginLoader::PluginKey> plugins = loader->listPlugins();
Chris@0: 
Chris@0:     //library Map
Chris@0:     typedef multimap<string, PluginLoader::PluginKey> LibraryMap;
Chris@0:     LibraryMap libraryMap;
Chris@0: 
Chris@0:     //New list object
Chris@0:     PyObject *pyList = PyList_New(plugins.size());
Chris@0: 
Chris@0:     for (size_t i = 0; i < plugins.size(); ++i) {
Chris@0:         string path = loader->getLibraryPathForPlugin(plugins[i]);
Chris@0:         libraryMap.insert(LibraryMap::value_type(path, plugins[i]));
Chris@0: 
Chris@0: 	PyObject *pyPluginKey = PyString_FromString(plugins[i].c_str());
Chris@0: 	PyList_SET_ITEM(pyList,i,pyPluginKey);
Chris@0: 		
Chris@0:     }
Chris@0: 
Chris@0:     PyList_Sort(pyList);
Chris@0:     return pyList;
Chris@0: }
Chris@0: 
Chris@0: 
Chris@0: /* GET PLUGIN LIBRARY PATH*/
Chris@0: 
Chris@0: static PyObject *
Chris@0: vampyhost_getLibraryPath(PyObject *self, PyObject *args)
Chris@0: {
Chris@0:     PyObject *pyPluginKey;
Chris@0: 
Chris@0:     if (!PyArg_ParseTuple(args, "S", &pyPluginKey)) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"String input argument required: pluginKey");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     //convert to stl string
Chris@0:     string pluginKey(PyString_AS_STRING(pyPluginKey));
Chris@0: 
Chris@0:     //check pluginKey Validity
Chris@0:     string::size_type ki = pluginKey.find(':');
Chris@0:     if (ki == string::npos) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"String input argument required: pluginLibrary:Identifier"); 
Chris@0:        	return NULL;
Chris@0:     }
Chris@0: 
Chris@0:     PluginLoader *loader = PluginLoader::getInstance();
Chris@0:     string path = loader->getLibraryPathForPlugin(pluginKey);
Chris@0:     PyObject *pyPath = PyString_FromString(path.c_str());
Chris@0:     return pyPath;
Chris@0: }
Chris@0: 
Chris@0: 
Chris@0: /* GET PLUGIN CATEGORY*/
Chris@0: 
Chris@0: static PyObject *
Chris@0: vampyhost_getPluginCategory(PyObject *self, PyObject *args)
Chris@0: {
Chris@0:     PyObject *pyPluginKey;
Chris@0: 
Chris@0:     if (!PyArg_ParseTuple(args, "S", &pyPluginKey)) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"String input argument required: pluginKey");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     //convert to stl string
Chris@0:     string pluginKey(PyString_AS_STRING(pyPluginKey));
Chris@0: 
Chris@0:     //check pluginKey Validity
Chris@0:     string::size_type ki = pluginKey.find(':');
Chris@0:     if (ki == string::npos) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"String input argument required: pluginLibrary:Identifier"); 
Chris@0:        	return NULL;
Chris@0:     }
Chris@0: 
Chris@0:     PluginLoader *loader = PluginLoader::getInstance();
Chris@0:     PluginLoader::PluginCategoryHierarchy 
Chris@0: 	category = loader->getPluginCategory(pluginKey);
Chris@0:     string catstring;
Chris@0: 
Chris@0:     if (!category.empty()) {
Chris@0:         catstring = "";
Chris@0:         for (size_t ci = 0; ci < category.size(); ++ci) {
Chris@0: 	    catstring.append(category[ci]);
Chris@0:             catstring.append(" ");
Chris@0:         }
Chris@0: 	PyObject *pyCat = PyString_FromString(catstring.c_str());
Chris@0: 	return pyCat;
Chris@0:     }
Chris@0:     PyObject *pyCat = PyString_FromString("");
Chris@0:     return pyCat;
Chris@0: }
Chris@0: 
Chris@0: 
Chris@0: 
Chris@0: /* GET PLUGIN OUTPUT LIST*/
Chris@0: 
Chris@0: static PyObject *
Chris@0: vampyhost_getOutputList(PyObject *self, PyObject *args)
Chris@0: {
Chris@0:     PyObject *pyPluginHandle;
Chris@0:     string pluginKey;
Chris@0: 
Chris@0:     if (!PyArg_ParseTuple(args, "O", &pyPluginHandle)) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Invalid argument: plugin handle or plugin key required.");
Chris@0: 	return NULL; 
Chris@0:     }
Chris@0: 
Chris@0:     //check if we have a plugin key string or a handle object
Chris@0:     if (PyString_Check(pyPluginHandle) ) {
Chris@0: 
Chris@0: 	pluginKey.assign(PyString_AS_STRING(pyPluginHandle));
Chris@0: 	//check pluginKey Validity
Chris@0:     	string::size_type ki = pluginKey.find(':');
Chris@0:     	if (ki == string::npos) {
Chris@0: 	    PyErr_SetString(PyExc_TypeError,
Chris@0: 			    "String input argument required: pluginLibrary:Identifier"); 
Chris@0: 	    return NULL;
Chris@0:     	}
Chris@0: 
Chris@0:     } else {
Chris@0: 		
Chris@0: 	string *key;	
Chris@0: 	Plugin *plugin; 
Chris@0: 
Chris@0: 	if ( !getPluginHandle(pyPluginHandle, &plugin, &key) ) {
Chris@0: 	    PyErr_SetString(PyExc_TypeError,
Chris@0: 			    "Invalid or deleted plugin handle.");
Chris@0: 	    return NULL; }
Chris@0: 	pluginKey.assign(*key);
Chris@0:     } 
Chris@0: 	
Chris@0:     //This code creates new instance of the plugin anyway
Chris@0:     PluginLoader *loader = PluginLoader::getInstance();
Chris@0: 	
Chris@0:     //load plugin
Chris@0:     Plugin *plugin = loader->loadPlugin (pluginKey, 48000);
Chris@0:     if (!plugin) { 		
Chris@0: 	string pyerr("Failed to load plugin: "); pyerr += pluginKey;
Chris@0: 	PyErr_SetString(PyExc_TypeError,pyerr.c_str()); 
Chris@0: 	return NULL;
Chris@0:     }		
Chris@0: 
Chris@0:     Plugin::OutputList outputs = plugin->getOutputDescriptors();
Chris@0:     //Plugin::OutputDescriptor od;
Chris@0: 
Chris@0:     if (outputs.size()<1) {
Chris@0: 	string pyerr("Plugin has no output: "); pyerr += pluginKey;
Chris@0: 	PyErr_SetString(PyExc_TypeError,pyerr.c_str()); 
Chris@0: 	return NULL;
Chris@0:     }
Chris@0: 
Chris@0:     //New list object
Chris@0:     PyObject *pyList = PyList_New(outputs.size());
Chris@0: 
Chris@0:     for (size_t i = 0; i < outputs.size(); ++i) {
Chris@0: 	PyObject *pyOutputId = 
Chris@0: 	    PyString_FromString(outputs[i].identifier.c_str());
Chris@0: 	PyList_SET_ITEM(pyList,i,pyOutputId);		
Chris@0:     }
Chris@0: 
Chris@0:     delete plugin;
Chris@0:     return pyList;
Chris@0: }
Chris@0: 
Chris@0: 
Chris@0: 
Chris@0: /* LOAD PLUGIN */
Chris@0: 
Chris@0: static PyObject *
Chris@0: vampyhost_loadPlugin(PyObject *self, PyObject *args)
Chris@0: {
Chris@0:     PyObject *pyPluginKey;
Chris@0:     float inputSampleRate;
Chris@0: 
Chris@0:     if (!PyArg_ParseTuple(args, "Sf", 
Chris@0: 			  &pyPluginKey,
Chris@0: 			  &inputSampleRate)) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"String input argument required: pluginKey");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     //convert to stl string
Chris@0:     string pluginKey(PyString_AS_STRING(pyPluginKey));
Chris@0: 
Chris@0:     //check pluginKey Validity
Chris@0:     string::size_type ki = pluginKey.find(':');
Chris@0:     if (ki == string::npos) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"String input argument required: pluginLibrary:Identifier"); 
Chris@0:        	return NULL;
Chris@0:     }
Chris@0: 
Chris@0:     PluginLoader *loader = PluginLoader::getInstance();
Chris@0: 	
Chris@0:     //load plugin
Chris@0:     Plugin *plugin = loader->loadPlugin (pluginKey, inputSampleRate);
Chris@0:     if (!plugin) { 		
Chris@0: 	string pyerr("Failed to load plugin: "); pyerr += pluginKey;
Chris@0: 	PyErr_SetString(PyExc_TypeError,pyerr.c_str()); 
Chris@0: 	return NULL;
Chris@0:     }		
Chris@0:     //void *identifier = (void*) new string(pluginKey);
Chris@0:     PyPluginDescriptor *pd = new PyPluginDescriptor;
Chris@0: 	
Chris@0:     pd->key = pluginKey;
Chris@0:     pd->isInitialised = false;
Chris@0:     pd->inputSampleRate = inputSampleRate;
Chris@0: 	
Chris@0:     //New PyCObject
Chris@0:     //PyObject *pyPluginHandle = PyCObject_FromVoidPtrAndDesc( 
Chris@0:     //(void*) plugin, identifier, NULL);	
Chris@0: 
Chris@0:     PyObject *pyPluginHandle = PyCObject_FromVoidPtrAndDesc( 
Chris@0: 	(void*) plugin, (void*) pd, NULL);	
Chris@0: 		
Chris@0:     return pyPluginHandle;
Chris@0: }
Chris@0: 
Chris@0: 
Chris@0: 
Chris@0: /* UNLOAD PLUGIN */
Chris@0: 
Chris@0: static PyObject *
Chris@0: vampyhost_unloadPlugin(PyObject *self, PyObject *args)
Chris@0: {
Chris@0:     PyObject *pyPluginHandle;
Chris@0: 
Chris@0:     if (!PyArg_ParseTuple(args, "O", &pyPluginHandle)) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Wrong input argument: Plugin Handle required.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     string *key;	
Chris@0:     Plugin *plugin; 
Chris@0: 
Chris@0:     if ( !getPluginHandle(pyPluginHandle, &plugin, &key) ) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Invalid or already deleted plugin handle.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0: /*	Prevent repeated calls from causing segfault
Chris@0: 	sice it will fail type checking the 2nd time:						*/
Chris@0:     PyCObject_SetVoidPtr(pyPluginHandle,NULL);
Chris@0: 	
Chris@0:     PyPluginDescriptor *pd = (PyPluginDescriptor*) key;
Chris@0: 	
Chris@0:     delete plugin;
Chris@0:     delete pd;
Chris@0:     return pyPluginHandle;
Chris@0: 
Chris@0: }
Chris@0: 
Chris@0: 
Chris@0: /* INITIALISE PLUGIN */
Chris@0: 
Chris@0: static PyObject *
Chris@0: vampyhost_initialise(PyObject *self, PyObject *args)
Chris@0: {
Chris@0:     PyObject *pyPluginHandle;
Chris@0:     size_t channels,blockSize,stepSize;
Chris@0:     //PyObject pyInputSampleType;
Chris@0:     bool mixChannels = false;
Chris@0: 
Chris@0:     if (!PyArg_ParseTuple (args, "Oiii",  &pyPluginHandle, 
Chris@0: 			   (size_t) &channels, 
Chris@0: 			   (size_t) &stepSize, 
Chris@0: 			   (size_t) &blockSize)) 
Chris@0:     {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Wrong input arguments: requires a valid plugin handle,channels,stepSize,blockSize.");
Chris@0: 	return NULL; 
Chris@0:     }
Chris@0: 
Chris@0:     Plugin *plugin; 
Chris@0:     string *key; 
Chris@0: 
Chris@0:     if ( !getPluginHandle(pyPluginHandle, &plugin, &key) ) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Invalid plugin handle.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     // here we cast the void pointer as PyPluginDescriptor instead of string
Chris@0:     PyPluginDescriptor *plugDesc = (PyPluginDescriptor*) key;	
Chris@0: 
Chris@0:     plugDesc->channels = channels;
Chris@0:     plugDesc->stepSize = stepSize;
Chris@0:     plugDesc->blockSize = blockSize;
Chris@0:     plugDesc->inputSampleType = PyPluginDescriptor::int16;
Chris@0:     plugDesc->sampleSize = 2;
Chris@0:     plugDesc->mixChannels = mixChannels; 
Chris@0: 	
Chris@0:     size_t minch = plugin->getMinChannelCount();
Chris@0:     size_t maxch = plugin->getMaxChannelCount();
Chris@0:     if (mixChannels) channels = 1;
Chris@0: 
Chris@0:     /* TODO: DO WE WANT TO MIX IT DOWN? */
Chris@0:     if (maxch < channels || channels < minch) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Invalid number of channels.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     if (!plugin->initialise(channels, stepSize, blockSize)) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Plugin initialization failed.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     plugDesc->identifier = 
Chris@0: 	plugDesc->key.substr(plugDesc->key.rfind(':')+1);
Chris@0:     plugDesc->isInitialised = true;
Chris@0: 	
Chris@0:     return Py_True;
Chris@0: }
Chris@0: 
Chris@0: 
Chris@0: /* RUN PROCESS */
Chris@0: 
Chris@0: static PyObject *
Chris@0: vampyhost_process(PyObject *self, PyObject *args)
Chris@0: {
Chris@0:     PyObject *pyPluginHandle;
Chris@0:     PyObject *pyBuffer;
Chris@0:     PyObject *pyRealTime;
Chris@0: 
Chris@0:     if (!PyArg_ParseTuple(args, "OOO", 
Chris@0: 			  &pyPluginHandle,	// C object holding a pointer to a plugin and its descriptor
Chris@0: 			  &pyBuffer,			// Audio data
Chris@0: 			  &pyRealTime)) {		// TimeStamp
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Required: plugin handle, buffer, timestmap.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     if (!PyRealTime_Check(pyRealTime)) {
Chris@0: 	PyErr_SetString(PyExc_TypeError,"Valid timestamp required.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     // RealTime *rt = PyRealTime_AsPointer(pyRealTime);
Chris@0:     // if (!rt) return NULL;
Chris@0:     // cerr << ">>>sec: " << rt->sec << " nsec: " << rt->nsec << endl;
Chris@0:     // 
Chris@0:     // PyObject *rrt = PyRealTime_FromRealTime (rt);
Chris@0: 
Chris@0:     string *key;	
Chris@0:     Plugin *plugin; 
Chris@0: 
Chris@0:     if ( !getPluginHandle(pyPluginHandle, &plugin, &key) ) {
Chris@0: 	PyErr_SetString(PyExc_AttributeError,
Chris@0: 			"Invalid or already deleted plugin handle.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     PyPluginDescriptor *pd = (PyPluginDescriptor*) key;
Chris@0: 
Chris@0:     if (!pd->isInitialised) {
Chris@0: 	PyErr_SetString(PyExc_StandardError,
Chris@0: 			"Plugin has not been initialised.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     size_t channels =  pd->channels;	
Chris@0:     size_t blockSize = pd->blockSize;
Chris@0: 
Chris@0: /*
Chris@0:   Handle the case when we get the data as a character buffer
Chris@0:   Handle SampleFormats: int16, float32	
Chris@0: 
Chris@0: */		
Chris@0: 
Chris@0:     if (PyString_Check(pyBuffer)) {
Chris@0: 	cerr << ">>> String obj passed in." << endl;
Chris@0:     }
Chris@0: 
Chris@0:     size_t sample_size = sizeof(short);
Chris@0: 
Chris@0:     long buflen = (long) PyString_GET_SIZE(pyBuffer);
Chris@0: 
Chris@0:     size_t input_length = 
Chris@0: 	static_cast <size_t> (buflen/channels/sample_size);
Chris@0: 
Chris@0:     if (input_length == pd->blockSize) {
Chris@0: 	cerr << ">>> A full block has been passed in." << endl; }
Chris@0:     short *input = 
Chris@0: 	reinterpret_cast <short*> (PyString_AS_STRING(pyBuffer));
Chris@0: 
Chris@0:     //convert int16 PCM data to 32-bit floats
Chris@0:     float **plugbuf = new float*[channels];
Chris@0:     float normfact = 1.0f / static_cast <float> (SHRT_MAX);
Chris@0: 		
Chris@0:     for (size_t c = 0; c < channels; ++c) {
Chris@0: 
Chris@0: 	plugbuf[c] = new float[blockSize+2];
Chris@0:   
Chris@0:       	size_t j = 0;
Chris@0:         while (j < input_length) {
Chris@0: 	    plugbuf[c][j] = normfact *
Chris@0: 		static_cast <float> (input[j * channels + c]);
Chris@0: 	    ++j; 
Chris@0:         }
Chris@0:         while (j < blockSize) {
Chris@0:             plugbuf[c][j] = 0.0f;
Chris@0:             ++j;
Chris@0:         }
Chris@0:     }	
Chris@0: 
Chris@0:     const char *output = reinterpret_cast <const char*> (plugbuf[0]);
Chris@0:     Py_ssize_t len = (Py_ssize_t) channels*blockSize*4;
Chris@0: 	
Chris@0:     PyObject* pyReturnBuffer = 
Chris@0: 	PyString_FromStringAndSize(output,len);
Chris@0: 
Chris@0:     // long frame = 1;
Chris@0:     // unsigned int samplerate = (unsigned int) pd->inputSampleRate;
Chris@0: 
Chris@0:     RealTime timeStamp = *PyRealTime_AsPointer(pyRealTime);
Chris@0: 
Chris@0:     //Call process and store the output
Chris@0:     pd->output = plugin->process(
Chris@0: 	plugbuf, 
Chris@0: 	timeStamp);
Chris@0: 
Chris@0:     /* TODO:  DO SOMETHONG WITH THE FEATURE SET HERE */
Chris@0: /// convert to appropriate python objects, reuse types and conversion utilities from Vampy ...
Chris@0: 
Chris@0: 
Chris@0:     //We can safely delete here
Chris@0:     for(size_t k=0; k<channels; k++){
Chris@0: 	delete[] plugbuf[k];
Chris@0:     }
Chris@0:     delete[] plugbuf;
Chris@0: 
Chris@0:     return pyReturnBuffer;
Chris@0: 
Chris@0: }
Chris@0: 
Chris@0: /* GET / SET OUTPUT */
Chris@0: 
Chris@0: //getOutput(plugin,outputNo)
Chris@0: static PyObject *
Chris@0: vampyhost_getOutput(PyObject *self, PyObject *args) {
Chris@0: 
Chris@0:     PyObject *pyPluginHandle;
Chris@0: //	PyObject *pyBuffer;
Chris@0: //	PyObject *pyRealTime;
Chris@0:     PyObject *pyOutput;
Chris@0: 
Chris@0:     if (!PyArg_ParseTuple(args, "OO", 
Chris@0: 			  &pyPluginHandle,	// C object holding a pointer to a plugin and its descriptor
Chris@0: 			  &pyOutput)) {		// Output reference
Chris@0: 	PyErr_SetString(PyExc_TypeError,
Chris@0: 			"Required: plugin handle, buffer, timestmap.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     string *key;	
Chris@0:     Plugin *plugin; 
Chris@0: 
Chris@0:     if ( !getPluginHandle(pyPluginHandle, &plugin, &key) ) {
Chris@0: 	PyErr_SetString(PyExc_AttributeError,
Chris@0: 			"Invalid or already deleted plugin handle.");
Chris@0: 	return NULL; }
Chris@0: 
Chris@0:     PyPluginDescriptor *pd = (PyPluginDescriptor*) key;
Chris@0: 
Chris@0:     unsigned int outputNo = (unsigned int) PyInt_AS_LONG(pyOutput);
Chris@0: 
Chris@0:     //Get output list: but we don't need it
Chris@0:     //Plugin::FeatureList features = pd->output[outputNo];
Chris@0: 
Chris@0:     size_t outLength = pd->output[outputNo].size();
Chris@0: 
Chris@0:     //New PyList for the featurelist
Chris@0:     PyObject *pyFeatureList = PyList_New(outLength);
Chris@0: 
Chris@0:     for (size_t i = 0; i < outLength; ++i) {
Chris@0: 	// Test:
Chris@0: 	/*
Chris@0: 	  XxoObject *pyFeature = PyObject_New(XxoObject, &Xxo_Type); 
Chris@0: 	  if (pyFeature == NULL) break; //return NULL;
Chris@0: 
Chris@0: 	  pyFeature->x_attr = NULL;
Chris@0: 	  pyFeature->feature = &pd->output[outputNo][i];
Chris@0: 
Chris@0: 	  PyList_SET_ITEM(pyFeatureList,i,(PyObject*)pyFeature);		
Chris@0: 	*/
Chris@0:     }
Chris@0: 
Chris@0:     Py_INCREF(pyFeatureList);
Chris@0:     return pyFeatureList;
Chris@0: 
Chris@0: // EXPLAIN WHAT WE NEED TO DO HERE:
Chris@0: // We have the block output in pd->output
Chris@0: // FeatureSet[output] -> [Feature[x]] -> Feature.hasTimestamp = v	
Chris@0: // Vamp::Plugin::FeatureSet output; = pd->output	
Chris@0: // typedef std::vector<Feature> FeatureList;
Chris@0: // typedef std::map<int, FeatureList> FeatureSet; // key is output no
Chris@0: 
Chris@0:     // 	THIS IS FOR OUTPUT id LOOKUP LATER 
Chris@0:     //     Plugin::OutputList outputs = plugin->getOutputDescriptors();
Chris@0:     // 
Chris@0:     // if (outputs.size()<1) {
Chris@0:     // 	string pyerr("Plugin has no output: "); pyerr += pluginKey;
Chris@0:     // 	PyErr_SetString(PyExc_TypeError,pyerr.c_str()); 
Chris@0:     // 	return NULL;
Chris@0:     // }
Chris@0:     // 
Chris@0:     // //New list object
Chris@0:     // PyObject *pyList = PyList_New(outputs.size());
Chris@0:     // 
Chris@0:     //     for (size_t i = 0; i < outputs.size(); ++i) {
Chris@0:     // 	PyObject *pyOutputId = 
Chris@0:     // 	PyString_FromString(outputs[i].identifier.c_str());
Chris@0:     // 	PyList_SET_ITEM(pyList,i,pyOutputId);		
Chris@0:     //     }
Chris@0: 	
Chris@0: }
Chris@0: 
Chris@0: 
Chris@0: 
Chris@0: 
Chris@0: /* List of functions defined in this module */
Chris@0: //module methods table
Chris@0: static PyMethodDef vampyhost_methods[] = {
Chris@0: 
Chris@0:     {"enumeratePlugins",	vampyhost_enumeratePlugins,	METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     {"getLibraryPath",	vampyhost_getLibraryPath, METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     {"getPluginCategory",	vampyhost_getPluginCategory, METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     {"getOutputList",	vampyhost_getOutputList, METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     {"loadPlugin",	vampyhost_loadPlugin, METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     {"process",	vampyhost_process, METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     {"unloadPlugin",	vampyhost_unloadPlugin, METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     {"initialise",	vampyhost_initialise, METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     {"getOutput",	vampyhost_getOutput, METH_VARARGS,
Chris@0:      xx_foo_doc},
Chris@0: 
Chris@0:     /* Add RealTime Module Methods */
Chris@0: 
Chris@0:     {"frame2RealTime",	(PyCFunction)RealTime_frame2RealTime,	METH_VARARGS,
Chris@0:      PyDoc_STR("frame2RealTime((int64)frame, (uint32)sampleRate ) -> returns new RealTime object from frame.")},
Chris@0: 
Chris@0:     {"realtime",	(PyCFunction)RealTime_new,		METH_VARARGS,
Chris@0:      PyDoc_STR("realtime() -> returns new RealTime object")},
Chris@0: 			
Chris@0:     {NULL,		NULL}		/* sentinel */
Chris@0: };
Chris@0: 
Chris@0: //Documentation for our new module
Chris@0: PyDoc_STRVAR(module_doc, "This is a template module just for instruction.");
Chris@0: 
Chris@0: /* Initialization function for the module (*must* be called initxx) */
Chris@0: 
Chris@0: //module initialization (includes extern C {...} as necessary)
Chris@0: PyMODINIT_FUNC
Chris@0: initvampyhost(void)
Chris@0: {
Chris@0:     PyObject *m;
Chris@0: 
Chris@0:     /* Finalize the type object including setting type of the new type
Chris@0:      * object; doing it here is required for portability to Windows 
Chris@0:      * without requiring C++. */
Chris@0: 
Chris@0:     if (PyType_Ready(&RealTime_Type) < 0)
Chris@0: 	return;
Chris@0: //	PyModule_AddObject(m, "Real_Time", (PyObject *)&RealTime_Type);
Chris@0: 
Chris@0:     /* Create the module and add the functions */
Chris@0:     m = Py_InitModule3("vampyhost", vampyhost_methods, module_doc);
Chris@0:     if (m == NULL) return;
Chris@0: 
Chris@0:     // PyModule_AddObject(m, "realtime", (PyObject *)&RealTime_Type);
Chris@0: 
Chris@0: }