b0dffe70394e65a5366da5f2ae60bd84f729ac7b
[linpy.git] / 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