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)