f98ee40f4b
This is an ongoing series of commits that are reformatting our Python code. This catches the last of the python files to reformat. Since they where so few I bunched them together. Reformatting is done with `black`. If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run git checkout --ours <yourfile> and then reformat it with black. If you run into any problems, post to discourse about it and we will try to help. RFC Thread below: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: jhenderson, #libc, Mordante, sivachandra Differential Revision: https://reviews.llvm.org/D150784
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)
|