binary_c_python.c

#include <Python.h>
#include "binary_c_python.h"

/*
 * binary_c/PYTHON API interface functions
 *
 * Remember: variables must be passed by references
 * (i.e. as pointers).
 *
 * See apitest.py for an expmple of how to use these functions.
 *
 * See also
 * http://www-h.eng.cam.ac.uk/help/tpl/languages/mixinglanguages.html
 */

/* list of variables used in the Py<>C interface */

/************************************************************/
/*
 * function prototypes : these are the functions
 * called by PYTHON code, without the trailing underscore.
 */
/************************************************************/


static char module_docstring[] MAYBE_UNUSED =
    "This module is a python wrapper around binary_c";
#ifdef __DEPRECATED
static char create_binary_docstring[] =
    "Allocate memory for a binary";
#endif
static char run_binary_docstring[] =
    "Run one binary using binary_c";
static char new_binary_system_docstring[] =
    "Return an object containing a binary, ready for evolution";
static char function_prototype_docstring[] =
    "The prototype for a binary_c python function";

static struct libbinary_c_store_t *store = NULL;

#ifdef __DEPRECATED
static PyObject* binary_c_create_binary(PyObject *self, PyObject *args);
#endif
static PyObject* binary_c_run_binary(PyObject *self, PyObject *args);
static PyObject* binary_c_function_prototype(PyObject *self, PyObject *args);
static PyObject* binary_c_new_binary_system(PyObject *self, PyObject *args);

    
/*
 * Python 3 interface is described at
 *
 * http://scipy-lectures.org/advanced/interfacing_with_c/interfacing_with_c.html
 */

static PyMethodDef module_methods[] = {
#ifdef __DEPRECATED
    {"create_binary", binary_c_create_binary, METH_VARARGS, create_binary_docstring},
#endif
    {"run_binary", binary_c_run_binary, METH_VARARGS, run_binary_docstring},
    {"function_prototype", binary_c_function_prototype, METH_VARARGS, function_prototype_docstring},
    {"new_system", binary_c_new_binary_system, METH_VARARGS, new_binary_system_docstring},
    
    {NULL, NULL, 0, NULL}
};

#if PY_MAJOR_VERSION >= 3

/* Python 3+ */

static struct PyModuleDef Py_binary_c =
{
    PyModuleDef_HEAD_INIT,
    "binary_c", /* name of module */
    "binary_c docs",          /* module documentation, may be NULL */
    -1,          /* size of per-interpreter state of the module, or -1 if the module keeps state in global variables. */
    module_methods
};

PyMODINIT_FUNC PyInit_binary_c(void)
{
    return PyModule_Create(&Py_binary_c);
}

#else

/*
 * Python pre-V3
 *
 * NOT TESTED THOROUGHLY!
 */


PyMODINIT_FUNC initbinary_c(void)
{
    PyObject *m = Py_InitModule3("binary_c", module_methods, module_docstring);
    if(m == NULL)
        return;
}
#endif // Python version check


#ifdef __DEPRECATED
static PyObject* binary_c_create_binary(PyObject *self, PyObject *args){

    double var1, var2;
    char * empty_str = "";
    int i;
    const int N = 1;

    /* Parse the input tuple */
    if(!PyArg_ParseTuple(args, "dd", &var1, &var2))
        return NULL;


    /* Binary structures */
    struct libbinary_c_stardata_t *stardata[N];
    struct libbinary_c_store_t *store = NULL;

    /* Allocate memory for binaries */
    for(i=0;i<N;i++){
        stardata[i] = NULL;
        binary_c_new_system(&stardata[i], NULL, NULL, &store, &empty_str, -1);
    }

    /* Return the evolved binary */
    PyObject *ret = Py_BuildValue("");

    return ret;
}
#endif


static PyObject* binary_c_new_binary_system(PyObject *self, PyObject *args)
{
    /* Binary structures */
    struct libbinary_c_stardata_t *stardata;

    /* Allocate memory for binaries */
    char * empty_str = "";
    stardata = NULL;
    binary_c_new_system(&stardata, NULL, NULL, &store, &empty_str, -1);
    
    /* Return an object containing the stardata */
    PyObject *ret = Py_BuildValue("");
    return ret;
}


static PyObject* binary_c_function_prototype(PyObject *self, PyObject *args)
{
    double var1, var2;

    /* Parse the input tuple */
    if(!PyArg_ParseTuple(args, "dd", &var1, &var2))
    {
        return NULL;
    }
    else
    {
        /* Return the evolved binary */
        PyObject *ret = Py_BuildValue("");
        return ret;
    }
}


static PyObject* binary_c_run_binary(PyObject *self, PyObject *args)
{
    /* Parse the input tuple */
    char *argstring;
    
    if(!PyArg_ParseTuple(args, "s", &argstring))
    {
        return NULL;
    }
    else
    {
        char * buffer;
        char * error_buffer;
        size_t nbytes;
        int out MAYBE_UNUSED = run_binary(argstring,
                                          &buffer,
                                          &error_buffer,
                                          &nbytes);
        /* copy the buffer to a python string */
        PyObject * return_string = Py_BuildValue("s", buffer);
        PyObject * return_error_string MAYBE_UNUSED = Py_BuildValue("s", error_buffer);

        if(error_buffer != NULL && strlen(error_buffer)>0)
        {
            fprintf(stderr,
                    "Error in binary_c run : %s\n",
                    error_buffer);
        }
        
        Safe_free(buffer);
        Safe_free(error_buffer);

        /* 
         * TODO
         * return the return_error_string as well!
         */
        return return_string;
    }
}