from migen.fhdl.std import *
from migen.bus import dfi, asmibus

from milkymist.asmicon.refresher import *
from milkymist.asmicon.bankmachine import *
from milkymist.asmicon.multiplexer import *

class PhySettings:
	def __init__(self, dfi_d, nphases, rdphase, wrphase):
		self.dfi_d = dfi_d
		self.nphases = nphases
		self.rdphase = rdphase
		self.wrphase = wrphase

class GeomSettings:
	def __init__(self, bank_a, row_a, col_a):
		self.bank_a = bank_a
		self.row_a = row_a
		self.col_a = col_a
		self.mux_a = max(row_a, col_a)

class TimingSettings:
	def __init__(self, tRP, tRCD, tWR, tREFI, tRFC, CL, rd_delay, read_time, write_time, slot_time=0):
		self.tRP = tRP
		self.tRCD = tRCD
		self.tWR = tWR
		self.tREFI = tREFI
		self.tRFC = tRFC
		
		self.CL = CL
		self.rd_delay = rd_delay
		
		self.read_time = read_time
		self.write_time = write_time
		self.slot_time = slot_time

class ASMIcon(Module):
	def __init__(self, phy_settings, geom_settings, timing_settings, full_selector=False):
		self.phy_settings = phy_settings
		self.geom_settings = geom_settings
		self.timing_settings = timing_settings
		self.full_selector = full_selector
		
		self.dfi = dfi.Interface(self.geom_settings.mux_a,
			self.geom_settings.bank_a,
			self.phy_settings.dfi_d,
			self.phy_settings.nphases)
		burst_length = self.phy_settings.nphases*2
		self.address_align = log2_int(burst_length)
		aw = self.geom_settings.bank_a + self.geom_settings.row_a + self.geom_settings.col_a - self.address_align
		dw = self.phy_settings.dfi_d*self.phy_settings.nphases
		self.submodules.hub = asmibus.Hub(aw, dw, self.timing_settings.slot_time)
	
	def do_finalize(self):
		slots = self.hub.get_slots()
		self.submodules.refresher = Refresher(self.geom_settings.mux_a, self.geom_settings.bank_a,
			self.timing_settings.tRP, self.timing_settings.tREFI, self.timing_settings.tRFC)
		self.submodules.bank_machines = [BankMachine(self.geom_settings, self.timing_settings, self.address_align, i, slots, self.full_selector)
			for i in range(2**self.geom_settings.bank_a)]
		self.submodules.multiplexer = Multiplexer(self.phy_settings, self.geom_settings, self.timing_settings,
			self.bank_machines, self.refresher,
			self.dfi, self.hub)