from migen.fhdl.std import *
from migen.fhdl.specials import Memory
from migen.bank.description import *
from migen.genlib.record import *

from litescope.common import *

class LiteScopeTerm(Module, AutoCSR):
	def __init__(self, width):
		self.width = width

		self.sink = Record(dat_layout(width))
		self.source = Record(hit_layout())

		self._trig = CSRStorage(width)
		self._mask = CSRStorage(width)

	###

		trig = self._trig.storage
		mask = self._mask.storage
		dat = self.sink.dat
		hit = self.source.hit

		self.comb += [
			hit.eq((dat & mask) == trig),
			self.source.stb.eq(self.sink.stb)
		]

class LiteScopeRangeDetector(Module, AutoCSR):
	def __init__(self, width):
		self.width = width

		self.sink = Record(dat_layout(width))
		self.source = Record(hit_layout())

		self._low = CSRStorage(width)
		self._high = CSRStorage(width)

	###

		low = self._low.storage
		high = self._high.storage
		dat = self.sink.dat
		hit = self.source.hit

		self.comb += [
			hit.eq((dat >= low) & (dat <= high)),
			self.source.stb.eq(self.sink.stb)
		]

class LiteScopeEdgeDetector(Module, AutoCSR):
	def __init__(self, width):
		self.width = width

		self.sink = Record(dat_layout(width))
		self.source = Record(hit_layout())

		self._rising_mask = CSRStorage(width)
		self._falling_mask = CSRStorage(width)
		self._both_mask = CSRStorage(width)

	###

		rising_mask = self._rising_mask.storage
		falling_mask = self._falling_mask.storage
		both_mask = self._both_mask.storage

		dat = self.sink.dat
		dat_d = Signal(width)
		rising_hit = Signal()
		falling_hit = Signal()
		both_hit = Signal()
		hit = self.source.hit

		self.sync += dat_d.eq(dat)

		self.comb += [
			rising_hit.eq(rising_mask & dat & ~dat_d),
			falling_hit.eq(rising_mask & ~dat & dat_d),
			both_hit.eq((both_mask & dat) != (both_mask & dat_d)),
			hit.eq(rising_hit | falling_hit | both_hit),
			self.source.stb.eq(self.sink.stb)
		]

class LiteScopeSum(Module, AutoCSR):
	def __init__(self, ports=4):

		self.sinks = [Record(hit_layout()) for p in range(ports)]
		self.source = Record(hit_layout())

		self._prog_we = CSRStorage()
		self._prog_adr = CSRStorage(ports) #FIXME
		self._prog_dat = CSRStorage()

		mem = Memory(1, 2**ports)
		lut_port = mem.get_port()
		prog_port = mem.get_port(write_capable=True)

		self.specials += mem, lut_port, prog_port

		###

		# Lut prog
		self.comb += [
			prog_port.we.eq(self._prog_we.storage),
			prog_port.adr.eq(self._prog_adr.storage),
			prog_port.dat_w.eq(self._prog_dat.storage)
		]

		# Lut read
		for i, sink in enumerate(self.sinks):
			self.comb += lut_port.adr[i].eq(sink.hit)

		# Drive source
		self.comb += [
			self.source.stb.eq(optree("&", [sink.stb for sink in self.sinks])),
			self.source.hit.eq(lut_port.dat_r),
		]


class LiteScopeTrigger(Module, AutoCSR):
	def __init__(self, width, ports):
		self.width = width
		self.ports = ports

		self.submodules.sum = LiteScopeSum(len(ports))
		for i, port in enumerate(ports):
			setattr(self.submodules, "port"+str(i), port)

		self.sink   = Record(dat_layout(width))
		self.source = self.sum.source

		###

		for i, port in enumerate(ports):
			self.comb += [
				self.sink.connect(port.sink),
				port.source.connect(self.sum.sinks[i])
			]