linpy/islhelper.py

   1 # Copyright 2014 MINES ParisTech
   2 #
   3 # This file is part of LinPy.
   4 #
   5 # LinPy is free software: you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation, either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # LinPy is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with LinPy.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 import ctypes, ctypes.util
  19 import re
  20
  21 from . import _islhelper
  22 from ._islhelper import *
  23
  24
  25 __all__ = [
  26     'libisl',
  27     'isl_version',
  28     'mainctx',
  29     'isl_val_to_int',
  30     'isl_basic_set_to_str', 'isl_basic_set_constraints',
  31     'isl_set_to_str', 'isl_set_basic_sets',
  32     'isl_set_points',
  33     'isl_vertices_vertices',
  34     'isl_multi_aff_to_str',
  35 ]
  36
  37
  38 libisl = ctypes.CDLL(ctypes.util.find_library('isl'))
  39 libisl.isl_dim_set = _islhelper.dim_set
  40
  41 libisl.isl_version.restype = ctypes.c_char_p
  42 isl_version = libisl.isl_version().decode().strip()
  43 isl_version = re.sub(r'^isl-', '', isl_version)
  44
  45
  46 mainctx = libisl.isl_ctx_alloc()
  47
  48
  49 libisl.isl_printer_get_str.restype = ctypes.c_char_p
  50
  51 def isl_val_to_int(islval):
  52     islpr = libisl.isl_printer_to_str(mainctx)
  53     islpr = libisl.isl_printer_print_val(islpr, islval)
  54     string = libisl.isl_printer_get_str(islpr).decode()
  55     return int(string)
  56
  57 def isl_basic_set_to_str(islbset):
  58     islpr = libisl.isl_printer_to_str(mainctx)
  59     islpr = libisl.isl_printer_print_basic_set(islpr, islbset)
  60     string = libisl.isl_printer_get_str(islpr).decode()
  61     return string
  62
  63 def isl_set_to_str(islset):
  64     islpr = libisl.isl_printer_to_str(mainctx)
  65     islpr = libisl.isl_printer_print_set(islpr, islset)
  66     string = libisl.isl_printer_get_str(islpr).decode()
  67     return string
  68
  69 def isl_multi_aff_to_str(islmaff):
  70     islpr = libisl.isl_printer_to_str(mainctx)
  71     islpr = libisl.isl_printer_print_multi_aff(islpr, islmaff)
  72     string = libisl.isl_printer_get_str(islpr).decode()
  73     return string