import numpy as np
from cell import Cell

class Table:
	def __init__(self, columns, rows):
		self.cells = [[Cell(c[0], r[1], c[1], r[0]) for c in columns] for r in rows]
		self.columns = columns
		self.rows = rows

	def set_edges(self, vertical, horizontal):
		for v in vertical:
			# find closest x coord
			# iterate over y coords and find closest points
			i = [i for i, t in enumerate(self.columns) if np.isclose(v[0], t[0], atol=2)]
			j = [j for j, t in enumerate(self.rows) if np.isclose(v[3], t[0], atol=2)]
			k = [k for k, t in enumerate(self.rows) if np.isclose(v[1], t[0], atol=2)]
			if not j:
				continue
			if i == [0]: # only left edge
				if k:
					I = i[0]
					J = j[0]
					K = k[0]
					while J < K:
						self.cells[J][I].left = True
						J += 1
				else:
					I = i[0]
					J = j[0]
					K = len(self.rows)
					while J < K:
						self.cells[J][I].left = True
						J += 1
			elif i == []: # only right edge
				if k:
					I = len(self.columns) - 1
					J = j[0]
					K = k[0]
					while J < K:
						self.cells[J][I].right = True
						J += 1
				else:
					I = len(self.columns) - 1
					J = j[0]
					K = len(self.rows)
					while J < K:
						self.cells[J][I].right = True
						J += 1
			else: # both left and right edges
				if k:
					I = i[0]
					J = j[0]
					K = k[0]
					while J < K:
						self.cells[J][I].left = True
						self.cells[J][I - 1].right = True
						J += 1
				else:
					I = i[0]
					J = j[0]
					K = len(self.rows)
					while J < K:
						self.cells[J][I].left = True
						self.cells[J][I - 1].right = True
						J += 1

		for h in horizontal:
			#  find closest y coord
			# iterate over x coords and find closest points
			i = [i for i, t in enumerate(self.rows) if np.isclose(h[1], t[0], atol=2)]
			j = [j for j, t in enumerate(self.columns) if np.isclose(h[0], t[0], atol=2)]
			k = [k for k, t in enumerate(self.columns) if np.isclose(h[2], t[0], atol=2)]
			if not j:
				continue
			if i == [0]: # only top edge
				if k:
					I = i[0]
					J = j[0]
					K = k[0]
					while J < K:
						self.cells[I][J].top = True
						J += 1
				else:
					I = i[0]
					J = j[0]
					K = len(self.columns)
					while J < K:
						self.cells[I][J].top = True
						J += 1
			elif i == []: # only bottom edge
				if k:
					I = len(self.rows) - 1
					J = j[0]
					K = k[0]
					while J < K:
						self.cells[I][J].bottom = True
						J += 1
				else:
					I = len(self.rows) - 1
					J = j[0]
					K = len(self.columns)
					while J < K:
						self.cells[I][J].bottom = True
						J += 1
			else: # both top and bottom edges
				if k:
					I = i[0]
					J = j[0]
					K = k[0]
					while J < K:
						self.cells[I][J].top = True
						self.cells[I - 1][J].bottom = True
						J += 1
				else:
					I = i[0]
					J = j[0]
					K = len(self.columns)
					while J < K:
						self.cells[I][J].top = True
						self.cells[I - 1][J].bottom = True
						J += 1

		return self

	def set_spanning(self):
		for i in range(len(self.cells)):
			for j in range(len(self.cells[i])):
				bound = self.cells[i][j].get_bounded_edges()
				if bound == 4:
					continue
				elif bound == 3:
					if not self.cells[i][j].left:
						if self.cells[i][j].right and self.cells[i][j].top and self.cells[i][j].bottom:
							self.cells[i][j].spanning_h = True
					elif not self.cells[i][j].right:
						if self.cells[i][j].left and self.cells[i][j].top and self.cells[i][j].bottom:
							self.cells[i][j].spanning_h = True
					elif not self.cells[i][j].top:
						if self.cells[i][j].left and self.cells[i][j].right and self.cells[i][j].bottom:
							self.cells[i][j].spanning_v = True
					elif not self.cells[i][j].bottom:
						if self.cells[i][j].left and self.cells[i][j].right and self.cells[i][j].top:
							self.cells[i][j].spanning_v = True
				elif bound == 2:
					if self.cells[i][j].left and self.cells[i][j].right:
						if not self.cells[i][j].top and not self.cells[i][j].bottom:
							self.cells[i][j].spanning_v = True
					elif self.cells[i][j].top and self.cells[i][j].bottom:
						if not self.cells[i][j].left and not self.cells[i][j].right:
							self.cells[i][j].spanning_h = True
		return self