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)