adamstark@14: #include <iostream>
adamstark@14: #include <Python.h>
adamstark@14: #include "../../src/OnsetDetectionFunction.h"
adamstark@14: #include "../../src/BTrack.h"
adamstark@14: #include <numpy/arrayobject.h>
adamstark@14: 
adamstark@14: static PyObject * btrack_onsetdf(PyObject *dummy, PyObject *args) 
adamstark@14: {
adamstark@14:     PyObject *arg1=NULL;
adamstark@14:     PyObject *arr1=NULL;
adamstark@14:     
adamstark@14:     if (!PyArg_ParseTuple(args, "O", &arg1)) 
adamstark@14:     {
adamstark@14:         return NULL;
adamstark@14:     }
adamstark@14:     
adamstark@14:     arr1 = PyArray_FROM_OTF(arg1, NPY_DOUBLE, NPY_IN_ARRAY); 
adamstark@14:     if (arr1 == NULL) 
adamstark@14:     {
adamstark@14:         return NULL;
adamstark@14:     }
adamstark@14: 
adamstark@14: 
adamstark@14:     
adamstark@14:     ////////// GET INPUT DATA ///////////////////
adamstark@14:     
adamstark@14:     // get data as array
adamstark@14:     double* data = (double*) PyArray_DATA(arr1);
adamstark@14:     
adamstark@14:     // get array size
adamstark@15:     long signal_length = PyArray_Size((PyObject*)arr1);
adamstark@14:     //int k = (int) theSize;
adamstark@14:     
adamstark@14:     // get data type 
adamstark@15:     //char type = PyArray_DESCR(arr1)->type;
adamstark@14:     
adamstark@14:     ////////// BEGIN PROCESS ///////////////////
adamstark@14:     int hsize = 512;
adamstark@14:     int fsize = 1024;
adamstark@14:     int df_type = 6;
adamstark@14:     int numframes;
adamstark@14:     double buffer[hsize];	// buffer to hold one hopsize worth of audio samples
adamstark@14: 
adamstark@14:     
adamstark@14:     // get number of audio frames, given the hop size and signal length
adamstark@14: 	numframes = (int) floor(((double) signal_length) / ((double) hsize));
adamstark@14:     
adamstark@14:     OnsetDetectionFunction onset(hsize,fsize,df_type,1);
adamstark@14: 
adamstark@14:     double df[numframes];
adamstark@14:     
adamstark@14: 
adamstark@14:     
adamstark@14:     ///////////////////////////////////////////
adamstark@14: 	//////// Begin Processing Loop ////////////
adamstark@14: 	
adamstark@14: 	for (int i=0;i < numframes;i++)
adamstark@14: 	{		
adamstark@14: 		// add new samples to frame
adamstark@14: 		for (int n = 0;n < hsize;n++)
adamstark@14: 		{
adamstark@14: 			buffer[n] = data[(i*hsize)+n];
adamstark@14: 		}
adamstark@14: 		
adamstark@14: 		df[i] = onset.getDFsample(buffer);
adamstark@14: 		
adamstark@14: 	}
adamstark@14: 	
adamstark@14: 	///////// End Processing Loop /////////////
adamstark@14: 	///////////////////////////////////////////
adamstark@14:     
adamstark@14:     
adamstark@14:     ////////// END PROCESS ///////////////////
adamstark@14:     
adamstark@14:     
adamstark@14:     
adamstark@14:     ////////// CREATE ARRAY AND RETURN IT ///////////////////
adamstark@14:     int nd=1;
adamstark@14:     npy_intp m= numframes;
adamstark@14:     //double fArray[5] = {0,1,2,3,4};
adamstark@14:     
adamstark@14:     PyObject* c=PyArray_SimpleNew(nd, &m, NPY_DOUBLE);
adamstark@14:     
adamstark@14:     void *arr_data = PyArray_DATA((PyArrayObject*)c);
adamstark@14:         
adamstark@14:     memcpy(arr_data, df, PyArray_ITEMSIZE((PyArrayObject*) c) * m); 
adamstark@14:     
adamstark@14:      
adamstark@14:     Py_DECREF(arr1);  
adamstark@14:     Py_INCREF(Py_None); 
adamstark@14:     //return Py_None;
adamstark@14:     
adamstark@14:     return (PyObject *)c;
adamstark@14:     
adamstark@14:     //return Py_BuildValue("c", type);
adamstark@14:     //return Py_BuildValue("d", sum);
adamstark@14:     //return Py_BuildValue("i", k);
adamstark@14: /*    
adamstark@14: fail:
adamstark@14:     Py_XDECREF(arr1); 
adamstark@14:     Py_XDECREF(arr2); 
adamstark@14:     PyArray_XDECREF_ERR(oarr); 
adamstark@14:     return NULL;*/
adamstark@14: }
adamstark@14: 
adamstark@14: 
adamstark@14: static PyObject * btrack_btrack(PyObject *dummy, PyObject *args) 
adamstark@14: {
adamstark@14:     PyObject *arg1=NULL;
adamstark@14:     PyObject *arr1=NULL;
adamstark@14:     
adamstark@14:     if (!PyArg_ParseTuple(args, "O", &arg1)) 
adamstark@14:     {
adamstark@14:         return NULL;
adamstark@14:     }
adamstark@14:     
adamstark@14:     arr1 = PyArray_FROM_OTF(arg1, NPY_DOUBLE, NPY_IN_ARRAY); 
adamstark@14:     if (arr1 == NULL) 
adamstark@14:     {
adamstark@14:         return NULL;
adamstark@14:     }
adamstark@14:     
adamstark@14:     
adamstark@14:     
adamstark@14:     ////////// GET INPUT DATA ///////////////////
adamstark@14:     
adamstark@14:     // get data as array
adamstark@14:     double* data = (double*) PyArray_DATA(arr1);
adamstark@14:     
adamstark@14:     // get array size
adamstark@15:     long signal_length = PyArray_Size((PyObject*)arr1);
adamstark@14:     //int k = (int) theSize;
adamstark@14:     
adamstark@14:     // get data type 
adamstark@15:     //char type = PyArray_DESCR(arr1)->type;
adamstark@14:     
adamstark@14:     ////////// BEGIN PROCESS ///////////////////
adamstark@14:     int hsize = 512;
adamstark@14:     int fsize = 1024;
adamstark@14:     int df_type = 6;
adamstark@14:     int numframes;
adamstark@14:     double buffer[hsize];	// buffer to hold one hopsize worth of audio samples
adamstark@14:     
adamstark@14:     
adamstark@14:     // get number of audio frames, given the hop size and signal length
adamstark@14: 	numframes = (int) floor(((double) signal_length) / ((double) hsize));
adamstark@14:     
adamstark@14:     OnsetDetectionFunction onset(hsize,fsize,df_type,1);
adamstark@14:     BTrack b;
adamstark@14:     
adamstark@14:     b.initialise((int) hsize);	// initialise beat tracker
adamstark@14: 	
adamstark@14: 	// set parameters
adamstark@14:     //b.setparams(0.9,5);
adamstark@14:     
adamstark@14:     double df[numframes];
adamstark@14:     double beats[5000];
adamstark@14:     int beatnum = 0;
adamstark@14:     float df_val;
adamstark@14:     
adamstark@14:     ///////////////////////////////////////////
adamstark@14: 	//////// Begin Processing Loop ////////////
adamstark@14: 	
adamstark@14: 	for (int i=0;i < numframes;i++)
adamstark@14: 	{		
adamstark@14: 		// add new samples to frame
adamstark@14: 		for (int n = 0;n < hsize;n++)
adamstark@14: 		{
adamstark@14: 			buffer[n] = data[(i*hsize)+n];
adamstark@14: 		}
adamstark@14: 		
adamstark@14: 		df[i] = onset.getDFsample(buffer);
adamstark@14:         
adamstark@14:         df_val = (float) (df[i] + 0.0001);
adamstark@14:                 
adamstark@14: 		b.process(df_val);				// process df sample in beat tracker
adamstark@14: 		
adamstark@14: 		if (b.playbeat == 1)
adamstark@14: 		{
adamstark@14: 			beats[beatnum] = (((double) hsize) / 44100) * ((double) i);
adamstark@14: 			beatnum = beatnum + 1;	
adamstark@14: 		}
adamstark@14: 		
adamstark@14: 	}
adamstark@14: 	
adamstark@14: 	///////// End Processing Loop /////////////
adamstark@14: 	///////////////////////////////////////////
adamstark@14:     
adamstark@14:     
adamstark@14:     ////////// END PROCESS ///////////////////
adamstark@14:     
adamstark@14:     double beats_out[beatnum];          // create output array
adamstark@14:     
adamstark@14:     // copy beats into output array
adamstark@14:     for (int i = 0;i < beatnum;i++)     
adamstark@14:     {
adamstark@14:         beats_out[i] = beats[i];
adamstark@14:     }
adamstark@14:     
adamstark@14:     
adamstark@14:     
adamstark@14:     ////////// CREATE ARRAY AND RETURN IT ///////////////////
adamstark@14:     int nd=1;
adamstark@14:     npy_intp m= beatnum;
adamstark@14:     //double fArray[5] = {0,1,2,3,4};
adamstark@14:     
adamstark@14:     PyObject* c=PyArray_SimpleNew(nd, &m, NPY_DOUBLE);
adamstark@14:     
adamstark@14:     void *arr_data = PyArray_DATA((PyArrayObject*)c);
adamstark@14:     
adamstark@14:     memcpy(arr_data, beats_out, PyArray_ITEMSIZE((PyArrayObject*) c) * m); 
adamstark@14:     
adamstark@14:     
adamstark@14:     Py_DECREF(arr1);  
adamstark@14:     Py_INCREF(Py_None); 
adamstark@14:     //return Py_None;
adamstark@14:     
adamstark@14:     return (PyObject *)c;
adamstark@14:     
adamstark@14:     //return Py_BuildValue("c", type);
adamstark@14:     //return Py_BuildValue("d", sum);
adamstark@14:     //return Py_BuildValue("i", k);
adamstark@14:     /*    
adamstark@14:      fail:
adamstark@14:      Py_XDECREF(arr1); 
adamstark@14:      Py_XDECREF(arr2); 
adamstark@14:      PyArray_XDECREF_ERR(oarr); 
adamstark@14:      return NULL;*/
adamstark@14: }
adamstark@14: 
adamstark@14: static PyObject * btrack_btrack_df(PyObject *dummy, PyObject *args) 
adamstark@14: {
adamstark@14:     PyObject *arg1=NULL;
adamstark@14:     PyObject *arr1=NULL;
adamstark@14:     
adamstark@14:     if (!PyArg_ParseTuple(args, "O", &arg1)) 
adamstark@14:     {
adamstark@14:         return NULL;
adamstark@14:     }
adamstark@14:     
adamstark@14:     arr1 = PyArray_FROM_OTF(arg1, NPY_DOUBLE, NPY_IN_ARRAY); 
adamstark@14:     if (arr1 == NULL) 
adamstark@14:     {
adamstark@14:         return NULL;
adamstark@14:     }
adamstark@14:     
adamstark@14:     
adamstark@14:     
adamstark@14:     ////////// GET INPUT DATA ///////////////////
adamstark@14:     
adamstark@14:     // get data as array
adamstark@14:     double* data = (double*) PyArray_DATA(arr1);
adamstark@14:     
adamstark@14:     // get array size
adamstark@15:     long numframes = PyArray_Size((PyObject*)arr1);
adamstark@14:     //int k = (int) theSize;
adamstark@14:     
adamstark@14:     // get data type 
adamstark@15:     //char type = PyArray_DESCR(arr1)->type;
adamstark@14:     
adamstark@14:     ////////// BEGIN PROCESS ///////////////////
adamstark@14:     int hsize = 512;
adamstark@14: 
adamstark@14:     BTrack b;
adamstark@14:     
adamstark@14:     b.initialise((int) hsize);	// initialise beat tracker
adamstark@14: 	
adamstark@14: 	// set parameters
adamstark@14:     //b.setparams(0.9,5);
adamstark@14:     
adamstark@14:     double beats[5000];
adamstark@14:     int beatnum = 0;
adamstark@14:     float df_val;
adamstark@14:     
adamstark@14:     ///////////////////////////////////////////
adamstark@14: 	//////// Begin Processing Loop ////////////
adamstark@14: 	
adamstark@15: 	for (long i=0;i < numframes;i++)
adamstark@14: 	{		
adamstark@14:         df_val = (float) (data[i] + 0.0001);
adamstark@14:         
adamstark@14: 		b.process(df_val);				// process df sample in beat tracker
adamstark@14: 		
adamstark@14: 		if (b.playbeat == 1)
adamstark@14: 		{
adamstark@14: 			beats[beatnum] = (((double) hsize) / 44100) * ((double) i);
adamstark@14: 			beatnum = beatnum + 1;	
adamstark@14: 		}
adamstark@14: 		
adamstark@14: 	}
adamstark@14: 	
adamstark@14: 	///////// End Processing Loop /////////////
adamstark@14: 	///////////////////////////////////////////
adamstark@14:     
adamstark@14:     
adamstark@14:     ////////// END PROCESS ///////////////////
adamstark@14:     
adamstark@14:     double beats_out[beatnum];          // create output array
adamstark@14:     
adamstark@14:     
adamstark@14:     // copy beats into output array
adamstark@14:     for (int i = 0;i < beatnum;i++)     
adamstark@14:     {
adamstark@14:         beats_out[i] = beats[i];
adamstark@14:     }
adamstark@14:     
adamstark@14:     
adamstark@14:     ////////// CREATE ARRAY AND RETURN IT ///////////////////
adamstark@14:     int nd=1;
adamstark@14:     npy_intp m= beatnum;
adamstark@14:     //double fArray[5] = {0,1,2,3,4};
adamstark@14:     
adamstark@14:     PyObject* c=PyArray_SimpleNew(nd, &m, NPY_DOUBLE);
adamstark@14:     
adamstark@14:     void *arr_data = PyArray_DATA((PyArrayObject*)c);
adamstark@14:     
adamstark@14:     memcpy(arr_data, beats_out, PyArray_ITEMSIZE((PyArrayObject*) c) * m); 
adamstark@14:     
adamstark@14:     
adamstark@14:     Py_DECREF(arr1);  
adamstark@14:     Py_INCREF(Py_None); 
adamstark@14:     //return Py_None;
adamstark@14:     
adamstark@14:     return (PyObject *)c;
adamstark@14:     
adamstark@14:     //return Py_BuildValue("c", type);
adamstark@14:     //return Py_BuildValue("d", sum);
adamstark@14:     //return Py_BuildValue("i", k);
adamstark@14:     /*    
adamstark@14:      fail:
adamstark@14:      Py_XDECREF(arr1); 
adamstark@14:      Py_XDECREF(arr2); 
adamstark@14:      PyArray_XDECREF_ERR(oarr); 
adamstark@14:      return NULL;*/
adamstark@14: }
adamstark@14: 
adamstark@14: 
adamstark@14: 
adamstark@14: static PyMethodDef btrack_methods[] = {
adamstark@14:     { "onsetdf",btrack_onsetdf,METH_VARARGS,"onset detection function"},
adamstark@14:     { "btrack",btrack_btrack,METH_VARARGS,"beat tracker"},
adamstark@14:     { "btrack_df",btrack_btrack_df,METH_VARARGS,"beat tracker with detection function input"},
adamstark@14:     {NULL, NULL, 0, NULL} /* Sentinel */
adamstark@14: };
adamstark@14: 
adamstark@14: PyMODINIT_FUNC initbtrack(void)
adamstark@14: {
adamstark@14:     (void)Py_InitModule("btrack", btrack_methods);
adamstark@14:     import_array();
adamstark@14: }
adamstark@14: 
adamstark@14: int main(int argc, char *argv[])
adamstark@14: {
adamstark@14:     /* Pass argv[0] to the Python interpreter */
adamstark@14:     Py_SetProgramName(argv[0]);
adamstark@14:     
adamstark@14:     /* Initialize the Python interpreter.  Required. */
adamstark@14:     Py_Initialize();
adamstark@14:     
adamstark@14:     /* Add a static module */
adamstark@14:     initbtrack();
adamstark@14: }