binary_c_python.c

#include <Python.h>
#include "/home/andrews/Research/code/binary_c_newest/src/API/binary_c_API.h"
// #include "/home/andrews/Desktop/binary_c_old/src/API/binary_c_API.h"
#include "binary_c_api.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[] =
  "This module is a python wrapper around binary_c";
static char create_binary_docstring[] =
  "Allocate memory for a binary";
static char run_binary_docstring[] =
  "Run one binary using binary_c";
static char function_prototype_docstring[] =
  "The prototype for a binary_c python function";


static PyObject* binary_c_create_binary(PyObject *self, PyObject *args);
static PyObject* binary_c_run_binary(PyObject *self, PyObject *args);
static PyObject* binary_c_function_prototype(PyObject *self, PyObject *args);


static PyMethodDef module_methods[] = {
    {"create_binary", binary_c_create_binary, METH_VARARGS, create_binary_docstring},
    {"run_binary", binary_c_run_binary, METH_VARARGS, run_binary_docstring},
    {"function_prototype", binary_c_function_prototype, METH_VARARGS, function_prototype_docstring},
    {NULL, NULL, 0, NULL}
};


PyMODINIT_FUNC initbinary_c(void)
{
    PyObject *m = Py_InitModule3("binary_c", module_methods, module_docstring);
    if (m == NULL)
        return;
}



static PyObject* binary_c_create_binary(PyObject *self, PyObject *args){

  double var1, var2;
  char* empty_str = "";
  int i;
  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;
}



static PyObject* binary_c_function_prototype(PyObject *self, PyObject *args){

  double var1, var2;
  char* empty_str = "";


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



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

  return ret;
}





static PyObject* binary_c_run_binary(PyObject *self, PyObject *args){

  int out, ktype_1, ktype_2, comenv_count;
  int evol_flag, dco_flag;
  double m1, m2, orbital_period, eccentricity, metallicity, maxt;
  double sn_kick_magnitude_1, sn_kick_theta_1, sn_kick_phi_1;
  double sn_kick_magnitude_2, sn_kick_theta_2, sn_kick_phi_2;
  double m1_out, m2_out, orbital_separation_out, eccentricity_out;
  double system_velocity, L_x, time_SN_1, time_SN_2, time_current;
  char evol_hist[1000000];

  /* Parse the input tuple */
  // if (!PyArg_ParseTuple(args, "ddddddd", &m1, &m2, &orbital_period, &eccentricity, &metallicity, &maxt, &sn_kick_magnitude_1))
  if (!PyArg_ParseTuple(args, "ddddddddddddii", &m1, &m2, &orbital_period, &eccentricity, &metallicity, &maxt,
                        &sn_kick_magnitude_1, &sn_kick_theta_1, &sn_kick_phi_1,
                        &sn_kick_magnitude_2, &sn_kick_theta_2, &sn_kick_phi_2,
                        &evol_flag, &dco_flag))
    return NULL;

  // evol_flag = 0;

  out =  run_binary(m1, m2, orbital_period, eccentricity, metallicity, maxt,
                    sn_kick_magnitude_1, sn_kick_theta_1, sn_kick_phi_1,
                    sn_kick_magnitude_2, sn_kick_theta_2, sn_kick_phi_2,
                    &m1_out, &m2_out, &orbital_separation_out, &eccentricity_out,
                    &system_velocity, &L_x, &time_SN_1, &time_SN_2, &time_current,
                    &ktype_1, &ktype_2, &comenv_count, evol_flag, evol_hist, dco_flag);


  /* Return variables */
  PyObject *ret = Py_BuildValue("dddddddddiiis", m1_out, m2_out, orbital_separation_out, eccentricity_out,
                                system_velocity, L_x, time_SN_1, time_SN_2, time_current,
                                ktype_1, ktype_2, comenv_count, evol_hist);

  /* Free evol_hist memory */
  // free(evol_hist);

  return ret;
}