Include Python.h headers first in _isl.c
[linpy.git] / pypol / isl.py
index ddf7ea0..a295e9f 100644 (file)
@@ -1,18 +1,14 @@
-"""
-note: for islpy
-isl format: basic set: ("{[x, y] : x >= 0 and x < 5 and y >= 0 and y < x+4 }")
-"""
-
 import ctypes, ctypes.util
 import ctypes, ctypes.util
-import functools
-import math
-import numbers
-import operator
-import re
 
 from . import _isl
 
 
 
 from . import _isl
 
 
+__all__ = [
+    'Context',
+    'BasicSet',
+]
+
+
 libisl = ctypes.CDLL(ctypes.util.find_library('isl'))
 
 libisl.isl_printer_get_str.restype = ctypes.c_char_p
 libisl = ctypes.CDLL(ctypes.util.find_library('isl'))
 
 libisl.isl_printer_get_str.restype = ctypes.c_char_p
@@ -21,7 +17,9 @@ libisl.isl_dim_set = _isl.isl_dim_set
 
 class IslObject:
 
 
 class IslObject:
 
-    __slots__ = ('_ptr')
+    __slots__ = (
+        '_ptr',
+    )
 
     def __init__(self, ptr):
         self._ptr = ptr
 
     def __init__(self, ptr):
         self._ptr = ptr
@@ -34,7 +32,8 @@ class IslObject:
 class Context(IslObject):
 
     def __init__(self):
 class Context(IslObject):
 
     def __init__(self):
-        self._ptr = libisl.isl_ctx_alloc()
+        ptr = libisl.isl_ctx_alloc()
+        super().__init__(ptr)
 
     #comment out so does not delete itself after being created
     #def __del__(self):
 
     #comment out so does not delete itself after being created
     #def __del__(self):
@@ -58,3 +57,6 @@ class BasicSet(IslObject):
 
     def __del__(self):
         libisl.isl_basic_set_free(self)
 
     def __del__(self):
         libisl.isl_basic_set_free(self)
+
+    def constraints(self):
+        return _isl.isl_basic_set_constraints(self._ptr)