119 lines
3.3 KiB
Python
119 lines
3.3 KiB
Python
|
import gdb
|
||
|
import re
|
||
|
|
||
|
# GDB Pretty Printers for most isl objects
|
||
|
class IslObjectPrinter:
|
||
|
"""Print an isl object"""
|
||
|
|
||
|
def __init__(self, val, type):
|
||
|
self.val = val
|
||
|
self.type = type
|
||
|
|
||
|
def to_string(self):
|
||
|
# Cast val to a void pointer to stop gdb using this pretty
|
||
|
# printer for the pointer which would lead to an infinite loop.
|
||
|
void_ptr = gdb.lookup_type("void").pointer()
|
||
|
value = str(self.val.cast(void_ptr))
|
||
|
printer = gdb.parse_and_eval(
|
||
|
"isl_printer_to_str(isl_" + str(self.type) + "_get_ctx(" + value + "))"
|
||
|
)
|
||
|
printer = gdb.parse_and_eval(
|
||
|
"isl_printer_print_"
|
||
|
+ str(self.type)
|
||
|
+ "("
|
||
|
+ str(printer)
|
||
|
+ ", "
|
||
|
+ value
|
||
|
+ ")"
|
||
|
)
|
||
|
string = gdb.parse_and_eval("(char*)isl_printer_get_str(" + str(printer) + ")")
|
||
|
gdb.parse_and_eval("isl_printer_free(" + str(printer) + ")")
|
||
|
return string
|
||
|
|
||
|
def display_hint(self):
|
||
|
return "string"
|
||
|
|
||
|
|
||
|
class IslIntPrinter:
|
||
|
"""Print an isl_int"""
|
||
|
|
||
|
def __init__(self, val):
|
||
|
self.val = val
|
||
|
|
||
|
def to_string(self):
|
||
|
# Cast val to a void pointer to stop gdb using this pretty
|
||
|
# printer for the pointer which would lead to an infinite loop.
|
||
|
void_ptr = gdb.lookup_type("void").pointer()
|
||
|
value = str(self.val.cast(void_ptr))
|
||
|
|
||
|
context = gdb.parse_and_eval("isl_ctx_alloc()")
|
||
|
printer = gdb.parse_and_eval("isl_printer_to_str(" + str(context) + ")")
|
||
|
printer = gdb.parse_and_eval(
|
||
|
"isl_printer_print_isl_int(" + str(printer) + ", " + value + ")"
|
||
|
)
|
||
|
string = gdb.parse_and_eval("(char*)isl_printer_get_str(" + str(printer) + ")")
|
||
|
gdb.parse_and_eval("isl_printer_free(" + str(printer) + ")")
|
||
|
gdb.parse_and_eval("isl_ctx_free(" + str(context) + ")")
|
||
|
return string
|
||
|
|
||
|
def display_hint(self):
|
||
|
return "string"
|
||
|
|
||
|
|
||
|
class IslPrintCommand(gdb.Command):
|
||
|
"""Print an isl value."""
|
||
|
|
||
|
def __init__(self):
|
||
|
super(IslPrintCommand, self).__init__("islprint", gdb.COMMAND_OBSCURE)
|
||
|
|
||
|
def invoke(self, arg, from_tty):
|
||
|
arg = gdb.parse_and_eval(arg)
|
||
|
printer = str_lookup_function(arg)
|
||
|
|
||
|
if printer == None:
|
||
|
print("No isl printer for this type")
|
||
|
return
|
||
|
|
||
|
print(printer.to_string())
|
||
|
|
||
|
|
||
|
IslPrintCommand()
|
||
|
|
||
|
|
||
|
def str_lookup_function(val):
|
||
|
if val.type.code != gdb.TYPE_CODE_PTR:
|
||
|
if str(val.type) == "isl_int":
|
||
|
return IslIntPrinter(val)
|
||
|
else:
|
||
|
return None
|
||
|
|
||
|
lookup_tag = val.type.target()
|
||
|
regex = re.compile("^isl_(.*)$")
|
||
|
|
||
|
if lookup_tag == None:
|
||
|
return None
|
||
|
|
||
|
m = regex.match(str(lookup_tag))
|
||
|
|
||
|
if m:
|
||
|
# Those types of printers defined in isl.
|
||
|
if m.group(1) in [
|
||
|
"basic_set",
|
||
|
"set",
|
||
|
"union_set",
|
||
|
"basic_map",
|
||
|
"map",
|
||
|
"union_map",
|
||
|
"qpolynomial",
|
||
|
"pw_qpolynomial",
|
||
|
"pw_qpolynomial_fold",
|
||
|
"union_pw_qpolynomial",
|
||
|
"union_pw_qpolynomial_fold",
|
||
|
]:
|
||
|
return IslObjectPrinter(val, m.group(1))
|
||
|
return None
|
||
|
|
||
|
|
||
|
# Do not register the pretty printer.
|
||
|
# gdb.current_objfile().pretty_printers.append(str_lookup_function)
|