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
  20 from . import _islhelper
  21 from ._islhelper import *
  22
  23
  24 __all__ = [
  25     'libisl',
  26     'isl_version',
  27     'mainctx',
  28     'isl_val_to_int',
  29     'isl_basic_set_to_str', 'isl_basic_set_constraints',
  30     'isl_set_to_str', 'isl_set_basic_sets',
  31     'isl_set_points',
  32     'isl_vertices_vertices',
  33     'isl_multi_aff_to_str',
  34 ]
  35
  36
  37 libisl = ctypes.CDLL(ctypes.util.find_library('isl'))
  38 libisl.isl_dim_set = _islhelper.dim_set
  39
  40 libisl.isl_version.restype = ctypes.c_char_p
  41 isl_version = libisl.isl_version().decode().strip()[len('isl-'):]
  42
  43
  44 mainctx = libisl.isl_ctx_alloc()
  45
  46
  47 libisl.isl_printer_get_str.restype = ctypes.c_char_p
  48
  49 def isl_val_to_int(islval):
  50     islpr = libisl.isl_printer_to_str(mainctx)
  51     islpr = libisl.isl_printer_print_val(islpr, islval)
  52     string = libisl.isl_printer_get_str(islpr).decode()
  53     return int(string)
  54
  55 def isl_basic_set_to_str(islbset):
  56     islpr = libisl.isl_printer_to_str(mainctx)
  57     islpr = libisl.isl_printer_print_basic_set(islpr, islbset)
  58     string = libisl.isl_printer_get_str(islpr).decode()
  59     return string
  60
  61 def isl_set_to_str(islset):
  62     islpr = libisl.isl_printer_to_str(mainctx)
  63     islpr = libisl.isl_printer_print_set(islpr, islset)
  64     string = libisl.isl_printer_get_str(islpr).decode()
  65     return string
  66
  67 def isl_multi_aff_to_str(islmaff):
  68     islpr = libisl.isl_printer_to_str(mainctx)
  69     islpr = libisl.isl_printer_print_multi_aff(islpr, islmaff)
  70     string = libisl.isl_printer_get_str(islpr).decode()
  71     return string