import os
from itertools import count

from jinja2 import Environment,FileSystemLoader


class DiagramPoint:
	def __init__(self,x,y,color="",label=""):
		self.x=x
		self.y=y
		self.color=color
		self.label=label

	def __repr__(self):
		return 'DiagramPoint({0},{1},"{2}","{3}")'.format(self.x,self.y,self.color,self.label)


class Base:
	highNumbers=True

	def __init__(self,start=0):
		self.overlays=[]
		self._letter="a"

		self._index=dict()
		self._indexGen=count(start)

		curDir=os.path.dirname(__file__)
		templateDir=os.path.join(curDir,"..","templ")
		self._env=Environment(loader=FileSystemLoader(templateDir))
		self._env.trim_blocks=True
		self._env.lstrip_blocks=True

	def addStone(self,x,y,color):
		assert (x,y) not in self._index
		self._index[(x,y)]=(next(self._indexGen),DiagramPoint(x,y,color))

	def addMove(self,x,y,color,label):
		if (not self.highNumbers) and isinstance(label,int) and label%100!=0:
			label%=100

		if (x,y) not in self._index:
			self._index[(x,y)]=(next(self._indexGen),DiagramPoint(x,y,color,label))
		else:
			(_,point)=self._index[(x,y)]
			if not point.label:
				point.label=self._letter
				self._letter=chr(ord(self._letter)+1)
			self.overlays.append((label, point.label))

	def addLabel(self,x,y,label):
		self._index[(x,y)]=(next(self._indexGen),DiagramPoint(x,y,"",label))

	def save(self,filename):
		notes=open(filename+".txt", 'w')
		notes.write("\n".join("{0} = {1}".format(a,b) for (a,b) in self.overlays))
		notes.close()