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)