from misoclib.com.liteeth.common import *
from misoclib.com.liteeth.mac.common import *
from misoclib.com.liteeth.mac.core import LiteEthMACCore
from misoclib.com.liteeth.mac.frontend.wishbone import LiteEthMACWishboneInterface


class LiteEthMAC(Module, AutoCSR):
    def __init__(self, phy, dw, interface="crossbar", endianness="big",
            with_preamble_crc=True):
        self.submodules.core = LiteEthMACCore(phy, dw, endianness, with_preamble_crc)
        self.csrs = []
        if interface == "crossbar":
            self.submodules.crossbar = LiteEthMACCrossbar()
            self.submodules.packetizer = LiteEthMACPacketizer()
            self.submodules.depacketizer = LiteEthMACDepacketizer()
            self.comb += [
                Record.connect(self.crossbar.master.source, self.packetizer.sink),
                Record.connect(self.packetizer.source, self.core.sink),
                Record.connect(self.core.source, self.depacketizer.sink),
                Record.connect(self.depacketizer.source, self.crossbar.master.sink)
            ]
        elif interface == "wishbone":
            self.submodules.interface = LiteEthMACWishboneInterface(dw, 2, 2)
            self.comb += Port.connect(self.interface, self.core)
            self.ev, self.bus = self.interface.sram.ev, self.interface.bus
            self.csrs = self.interface.get_csrs() + self.core.get_csrs()
        elif interface == "dma":
            raise NotImplementedError
        else:
            raise ValueError(interface + " not supported by LiteEthMac!")

    def get_csrs(self):
        return self.csrs