plot.py

from hypercomplex import Order, Names

import argparse as ap
import matplotlib as mpl
import matplotlib.pyplot as plt
import seaborn as sea
import numpy as np

# Color Maps: https://matplotlib.org/stable/tutorials/colors/colormaps.html

def plot(**options):

	def option(name, default, **options):

		if name in options:

			return options[name]

		return default

	showneg = option("negatives", True, **options)
	poscmap = option("colormap", "RdBu_r", **options)
	negcmap = option("ncolormap", "PiYG_r", **options)
	diverge = option("diverging", True, **options)
	figsize = option("figsize", 8.0, **options)
	figdpis = option("figdpi", 100.0, **options)
	filename = option("filename", "P{order}.{filetype}", **options)
	filetype = option("filetype", "png", **options)
	order = option("order", None, **options)
	named = option("named", None, **options)
	save = option("save", False, **options)
	show = option("show", True, **options)

	if named != None:

		self = Names.get(named, None)

	elif order != None:

		self = Order.get(order, None)

	else:

		self = None

	if self == None or (hasattr(self, "order") and self.order > 8):

		raise NotImplementedError

	sea.set_style("white")

	size = self.dimensions
	matrix = self.matrix(asindex=True)
	figure, axis = plt.subplots(figsize=(figsize, figsize), dpi=figdpis)
	numcolors = 2 * size + 1 if diverge else size
	positives = sea.color_palette(poscmap, numcolors)
	negatives = sea.color_palette(negcmap, numcolors)
	rectangle = mpl.patches.Rectangle
	extras = dict(snap=False, lw=1, zorder=1)

	for (x, y), value in np.ndenumerate(matrix):

		location = (y, size - x - 1)
		value = value + size if diverge else abs(value) - 1
		color = negatives[value] if showneg and not diverge and value < 0 else positives[value]
		patch = rectangle(location, 1, 1, edgecolor=color, facecolor=color, **extras)
		axis.add_patch(patch)

	axis.get_xaxis().set_ticks([])
	axis.get_yaxis().set_ticks([])
	axis.set_xlim(0, size)
	axis.set_ylim(0, size)

	plt.tight_layout()

	if save:

		output = ((filename).format(order=self.order, filetype=filetype))

		plt.savefig(output)

	elif show:

		plt.show()

if __name__ == "__main__":

	parser = ap.ArgumentParser()

	parser.add_argument("-o", "--order", type=int, default=2)
	parser.add_argument("-f", "--filename", type=str, default="P{order}.{filetype}")
	parser.add_argument("-t", "--filetype", type=str, default="png")
	parser.add_argument("-s", "--figsize", type=float, default=8.0)
	parser.add_argument("-r", "--figdpi", type=int, default=100)
	parser.add_argument("-c", "--colormap", type=str, default="RdBu_r")
	parser.add_argument("-x", "--ncolormap", type=str, default="PiYG_r")
	parser.add_argument("-n", "--named", type=str)

	parser.add_argument("--negatives", action="store_true", default=True)
	parser.add_argument("--diverging", action="store_true", default=True)
	parser.add_argument("--save", action="store_true", default=False)
	parser.add_argument("--show", action="store_true", default=True)

	args, urgs = parser.parse_known_args()

	plot(**vars(args))