#/usr/bin/python
import random
import math
import time
import matplotlib.pyplot as plt

fig = plt.figure()

def plotShit(points, centroids=None):
	"""splits dimensions for matplotlib, scatterplots it"""
	if dim >= 2:
		x = [point[0] for point in points]
		y = [point[1] for point in points]
		if centroids:
			xc = [point[0] for point in centroids]
			yc = [point[1] for point in centroids]
		if dim >= 3:
			threedplot = fig.add_subplot(projection='3d')
			z = [point[2] for point in points]
			threedplot.scatter(x, y, z)
			if centroids:
				zc = [point[1] for point in centroids]
			threedplot.scatter(xc, yc, zc, s=190.0)
		else:
			plot = fig.add_subplot()
			plot.scatter(x, y)
			if centroids:
				plot.scatter(xc, yc)
	plt.show()

def printPixels(pixels):
	print(chr(27) + "[2J") #Clear
	print('┌' + '─'*len(pixels[0])+'┐', flush=False)
	for row in pixels:
		print('│', end='', flush=False)
		for cell in row:
			print((' ' if cell == 0 else f'\033[{31+cell}m{cell}\033[0m'), end='', flush=False)
		print('│')
	print('└' + '─'*len(pixels[0])+'┘', flush=False)


n = 2000 # initial points
dim = 2 # in 2 of 3 dimensies. Alleen zinnig voor 2, maar werkt ook voor 3
k = 10 # aantal clusters/kamers
domains = [(0,1000)]*dim

points = [tuple(round(random.uniform(lb, ub),2) for lb, ub in domains) for _ in range(n)]
centroids = random.sample(points, k)
epochs = 0
while epochs < 5:
	dists = [tuple(math.dist(p, c) for c in centroids) for p in points]
	point_cluster_index = list(map(lambda x: x.index(min(x)), dists))
	points_per_cluster = [[p for i, p in enumerate(points) if point_cluster_index[i] == j] for j in range(k)]
	centroids = [tuple(sum(col) / float(len(col)) for col in zip(*points_per_cluster[i])) for i in range(k)]
	epochs += 1
	#print(points)
	#print(centroids)
	#print(dists)
	#print(point_cluster_index)
	#print(points_per_cluster)
	#print("---")

plotShit(points, centroids)

width= 64
height = 24
#pixels=[[0]*width]*height #kut python
pixels=[[0]*width for _ in range(height)] # aanspreken als pixels[y][x] (rows columns, verwarrend).

# map cluster centers to pixels:
cluster_pixels = [(int(c[0]/domains[0][1]*width), int(c[1]/domains[1][1]*height))for c in centroids]
for i, c in enumerate(cluster_pixels):
	pixels[c[1]][c[0]] = i

epochs = 0
while epochs < 20:
	for i, c in enumerate(cluster_pixels):
		for y, row in enumerate(pixels):
			for x, cell in enumerate(row):
				if cell == i:
					nx = x-random.randrange(-1,2) # randrange heeft non-inclusive upperbound. leuk.
					ny = y-random.randrange(-1,2)
					if nx >= 0 and ny >= 0 and nx < width and ny < height and pixels[ny][nx] == 0:
						pixels[ny][nx] = i
	epochs += 1
	printPixels(pixels)