pypol/_islhelper.c

   1 #include <Python.h>
   2
   3 #include <isl/constraint.h>
   4 #include <isl/set.h>
   5
   6
   7 struct pointer_list {
   8     int cursor;
   9     PyObject *pointers;
  10 };
  11
  12 typedef struct pointer_list pointer_list;
  13
  14
  15 static int pointer_list_append_constraint(isl_constraint *c, void *user) {
  16     pointer_list *list;
  17     PyObject *value;
  18
  19     list = (pointer_list *) user;
  20     value = PyLong_FromVoidPtr(c);
  21     if (value == NULL) {
  22         return -1;
  23     }
  24     PyList_SET_ITEM(list->pointers, list->cursor++, value);
  25     return 0;
  26 }
  27
  28 static PyObject * isl_basic_set_constraints(PyObject *self, PyObject* args) {
  29     long ptr;
  30     isl_basic_set *bset;
  31     int n;
  32     PyObject *pointers;
  33     pointer_list *list;
  34
  35     if (!PyArg_ParseTuple(args, "l", &ptr)) {
  36         return NULL;
  37     }
  38     bset = (isl_basic_set *) ptr;
  39     n = isl_basic_set_n_constraint(bset);
  40     if (n == -1) {
  41         PyErr_SetString(PyExc_RuntimeError,
  42             "an error occurred in isl_basic_set_n_constraint");
  43         return NULL;
  44     }
  45     pointers = PyList_New(n);
  46     if (pointers == NULL) {
  47         return NULL;
  48     }
  49     list = malloc(sizeof(pointer_list));
  50     if (list == NULL) {
  51         Py_DECREF(pointers);
  52         return PyErr_NoMemory();
  53     }
  54     list->cursor = 0;
  55     list->pointers = pointers;
  56     n = isl_basic_set_foreach_constraint(bset, pointer_list_append_constraint,
  57         list);
  58     free(list);
  59     if (n == -1) {
  60         PyErr_SetString(PyExc_RuntimeError,
  61             "an error occurred in isl_basic_set_foreach_constraint");
  62         Py_DECREF(pointers);
  63         return NULL;
  64     }
  65     return pointers;
  66 }
  67
  68
  69 static int pointer_list_append_basic_set(isl_basic_set *bset, void *user) {
  70     pointer_list *list;
  71     PyObject *value;
  72
  73     list = (pointer_list *) user;
  74     value = PyLong_FromVoidPtr(bset);
  75     if (value == NULL) {
  76         return -1;
  77     }
  78     PyList_SET_ITEM(list->pointers, list->cursor++, value);
  79     return 0;
  80 }
  81
  82 static PyObject * isl_set_basic_sets(PyObject *self, PyObject *args) {
  83     long ptr;
  84     isl_set *set;
  85     int n;
  86     PyObject *pointers;
  87     pointer_list *list;
  88
  89     if (!PyArg_ParseTuple(args, "l", &ptr)) {
  90         return NULL;
  91     }
  92     set = (isl_set *) ptr;
  93     n = isl_set_n_basic_set(set);
  94     if (n == -1) {
  95         PyErr_SetString(PyExc_RuntimeError,
  96             "an error occurred in isl_set_n_basic_set");
  97         return NULL;
  98     }
  99     pointers = PyList_New(n);
 100     if (pointers == NULL) {
 101         return NULL;
 102     }
 103     list = malloc(sizeof(pointer_list));
 104     if (list == NULL) {
 105         Py_DECREF(pointers);
 106         return PyErr_NoMemory();
 107     }
 108     list->cursor = 0;
 109     list->pointers = pointers;
 110     n = isl_set_foreach_basic_set(set, pointer_list_append_basic_set, list);
 111     free(list);
 112     if (n == -1) {
 113         PyErr_SetString(PyExc_RuntimeError,
 114             "an error occurred in isl_set_foreach_basic_set");
 115         Py_DECREF(pointers);
 116         return NULL;
 117     }
 118     return pointers;
 119 }
 120
 121
 122 static int pointer_list_append_point(isl_point *point, void *user) {
 123     PyObject *pointers, *value;
 124
 125     pointers = (PyObject *) user;
 126     value = PyLong_FromVoidPtr(point);
 127     if (value == NULL) {
 128         return -1;
 129     }
 130     return PyList_Append(pointers, value);
 131 }
 132
 133 static PyObject * isl_set_points(PyObject *self, PyObject *args) {
 134     long ptr;
 135     isl_set *set;
 136     int n;
 137     PyObject *pointers;
 138
 139     if (!PyArg_ParseTuple(args, "l", &ptr)) {
 140         return NULL;
 141     }
 142     set = (isl_set *) ptr;
 143     pointers = PyList_New(0);
 144     if (pointers == NULL) {
 145         return NULL;
 146     }
 147     n = isl_set_foreach_point(set, pointer_list_append_point, pointers);
 148     if (n == -1) {
 149         PyErr_SetString(PyExc_RuntimeError,
 150             "an error occurred in isl_set_foreach_point");
 151         Py_DECREF(pointers);
 152         return NULL;
 153     }
 154     return pointers;
 155 }
 156
 157
 158 static PyMethodDef _islhelper_methods[] = {
 159     {"isl_basic_set_constraints", isl_basic_set_constraints, METH_VARARGS, NULL},
 160     {"isl_set_basic_sets", isl_set_basic_sets, METH_VARARGS, NULL},
 161     {"isl_set_points", isl_set_points, METH_VARARGS, NULL},
 162     {NULL, NULL, 0, NULL}
 163 };
 164
 165 static struct PyModuleDef _islhelpermodule = {
 166     PyModuleDef_HEAD_INIT,
 167     "_islhelper",
 168     NULL,
 169     0,
 170     _islhelper_methods
 171 };
 172
 173 PyMODINIT_FUNC PyInit__islhelper(void) {
 174     PyObject *m;
 175     m = PyModule_Create(&_islhelpermodule);
 176     if (m == NULL) {
 177         return NULL;
 178     }
 179
 180     if (PyModule_AddObject(m, "dim_set", PyLong_FromLong(isl_dim_set)) == -1) {
 181         return NULL;
 182     }
 183
 184     return m;
 185 }