amaranth/nmigen/lib/io.py
2019-01-14 21:39:19 +00:00

48 lines
1.3 KiB
Python

from .. import *
__all__ = ["TSTriple", "Tristate"]
class TSTriple:
def __init__(self, shape=None, min=None, max=None, reset_o=0, reset_oe=0, reset_i=0,
name=None):
self.o = Signal(shape, min=min, max=max, reset=reset_o,
name=None if name is None else name + "_o")
self.oe = Signal(reset=reset_oe,
name=None if name is None else name + "_oe")
self.i = Signal(shape, min=min, max=max, reset=reset_i,
name=None if name is None else name + "_i")
def __len__(self):
return len(self.o)
def get_fragment(self, platform):
return Fragment()
def get_tristate(self, io):
return Tristate(self, io)
class Tristate:
def __init__(self, triple, io):
self.triple = triple
self.io = io
def get_fragment(self, platform):
if hasattr(platform, "get_tristate"):
return platform.get_tristate(self.triple, self.io)
m = Module()
m.d.comb += self.triple.i.eq(self.io)
m.submodules += Instance("$tribuf",
p_WIDTH=len(self.io),
i_EN=self.triple.oe,
i_A=self.triple.o,
o_Y=self.io,
)
f = m.lower(platform)
f.flatten = True
return f