72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
|
"""
|
||
|
|
CS 412 Tiling Problem Visualizer
|
||
|
|
Author: Molloy (Feb 2023)
|
||
|
|
|
||
|
|
Input: from stdin the matrix of tiles
|
||
|
|
|
||
|
|
Output: produces a visualization of the tiling
|
||
|
|
|
||
|
|
NOTE: This program assigns tiles a random color. It is
|
||
|
|
possible that the color of two adj tiles will be very similar.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
from matplotlib import colors
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
import random
|
||
|
|
|
||
|
|
|
||
|
|
def stdinToMatrix():
|
||
|
|
"""
|
||
|
|
Read in the matrix from stdin and return a 2D list
|
||
|
|
"""
|
||
|
|
matrix = []
|
||
|
|
|
||
|
|
row = input().split()
|
||
|
|
rowsize = len(row) # compute # of rows (since numrows == numcols)
|
||
|
|
matrix.append([int(i) for i in row])
|
||
|
|
|
||
|
|
for _ in range(rowsize - 1):
|
||
|
|
matrix.append([int(i) for i in input().split()])
|
||
|
|
|
||
|
|
return matrix
|
||
|
|
|
||
|
|
|
||
|
|
def draw_tiling(matrix):
|
||
|
|
"""
|
||
|
|
Compute and display the plot
|
||
|
|
"""
|
||
|
|
|
||
|
|
# create set of unique tile numbers
|
||
|
|
tiling_indices = set([x for row in matrix for x in row])
|
||
|
|
|
||
|
|
colorlist = [(0,0,0)] # initial empty space
|
||
|
|
|
||
|
|
"""
|
||
|
|
Adjacent tiles were sometimes too close in color. The first two colors
|
||
|
|
in the RGB tuple are now "seeded" by the tile number in hopes of
|
||
|
|
reducing the likelihood of "close" colors being adjacent.
|
||
|
|
"""
|
||
|
|
for tile in tiling_indices:
|
||
|
|
if tile != -1:
|
||
|
|
newColor = ((tile*17)%255 + 10, (tile*37)%255 + 10,random.randint(10,255))
|
||
|
|
colorlist.append(newColor)
|
||
|
|
|
||
|
|
cmap = colors.ListedColormap(colorlist)
|
||
|
|
|
||
|
|
bounds = [-1] + list(range(len(tiling_indices))) # map tilenumbers to colors
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
|
||
|
|
ax.imshow(matrix, cmap=cmap)
|
||
|
|
ax.grid(which='major', axis='both', linestyle='-', color='k', linewidth=2)
|
||
|
|
|
||
|
|
ax.set_xticks(np.arange(-.5, len(matrix[0]), 1));
|
||
|
|
ax.set_yticks(np.arange(-.5, len(matrix[0]), 1));
|
||
|
|
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
matrix = stdinToMatrix()
|
||
|
|
draw_tiling(matrix)
|