from litescope.host.dump import *

class PythonDump(Dump):
	def __init__(self, init_dump=None):
		Dump.__init__(self)
		if init_dump:
			self.vars = init_dump.vars

	def generate_data(self):
		r = "dump = {\n"
		for var in self.vars:
			r += "\"" + var.name + "\""
			r += " : "
			r += str(var.values)
			r += ",\n"
		r += "}"
		return r

	def write(self, filename):
		f = open(filename, "w")
		f.write(self.generate_data())
		f.close()

	def read(self, filename):
		raise NotImplementedError("Python files can not (yet) be read, please contribute!")

if __name__ == '__main__':
	dump = PythonDump()
	dump.add(Var("foo1", 1, [0,1,0,1,0,1]))
	dump.add(Var("foo2", 2, [1,0,1,0,1,0,1,0,1,0,1,0,1,0,1,0,1,0]))
	ramp = [i%128 for i in range(1024)]
	dump.add(Var("ramp", 16, ramp))
	dump.write("dump.py")