from lib.sata.common import *
from lib.sata.link.cont import SATACONTInserter, SATACONTRemover

from lib.sata.test.common import *

class ContPacket(list):
	def __init__(self, data=[]):
		self.ongoing = False
		self.done = False
		for d in data:
			self.append(d)

class ContStreamer(PacketStreamer):
	def __init__(self):
		PacketStreamer.__init__(self, phy_description(32), ContPacket)

	def do_simulation(self, selfp):
		PacketStreamer.do_simulation(self, selfp)
		selfp.source.charisk = 0
		# Note: for simplicity we generate charisk by detecting
		# primitives in data
		for k, v in primitives.items():
			try:
				if self.source_data == v:
					selfp.source.charisk = 0b0001
			except:
				pass

class ContLogger(PacketLogger):
	def __init__(self):
		PacketLogger.__init__(self, phy_description(32), ContPacket)

class TB(Module):
	def __init__(self):
		self.streamer = ContStreamer()
		self.streamer_randomizer = Randomizer(phy_description(32), level=50)
		self.inserter = SATACONTInserter(phy_description(32))
		self.remover = SATACONTRemover(phy_description(32))
		self.logger_randomizer = Randomizer(phy_description(32), level=50)
		self.logger = ContLogger()

		self.pipeline = Pipeline(
			self.streamer,
			self.streamer_randomizer,
			self.inserter,
			self.remover,
			self.logger_randomizer,
			self.logger
		)

	def gen_simulation(self, selfp):
		test_packet = ContPacket([
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["ALIGN"],
			primitives["ALIGN"],
			primitives["SYNC"],
			primitives["SYNC"],
			#primitives["SYNC"],
			0x00000000,
			0x00000001,
			0x00000002,
			0x00000003,
			0x00000004,
			0x00000005,
			0x00000006,
			0x00000007,
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["ALIGN"],
			primitives["ALIGN"],
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["SYNC"],
			primitives["SYNC"]]*4
			)
		streamer_packet = ContPacket(test_packet)
		yield from self.streamer.send(streamer_packet)
		yield from self.logger.receive(len(test_packet))
		#for d in self.logger.packet:
		#	print("%08x" %d)

		# check results
		s, l, e = check(streamer_packet, self.logger.packet)
		print("shift "+ str(s) + " / length " + str(l) + " / errors " + str(e))


if __name__ == "__main__":
	run_simulation(TB(), ncycles=1024, vcd_name="my.vcd", keep_files=True)