129 lines
3.7 KiB
Python
129 lines
3.7 KiB
Python
import warnings
|
|
|
|
from amaranth.hdl import *
|
|
from amaranth.sim import *
|
|
with warnings.catch_warnings():
|
|
warnings.filterwarnings(action="ignore", category=DeprecationWarning)
|
|
from amaranth.lib.coding import *
|
|
|
|
from .utils import *
|
|
|
|
|
|
class EncoderTestCase(FHDLTestCase):
|
|
def test_basic(self):
|
|
enc = Encoder(4)
|
|
def process():
|
|
self.assertEqual((yield enc.n), 1)
|
|
self.assertEqual((yield enc.o), 0)
|
|
|
|
yield enc.i.eq(0b0001)
|
|
self.assertEqual((yield enc.n), 0)
|
|
self.assertEqual((yield enc.o), 0)
|
|
|
|
yield enc.i.eq(0b0100)
|
|
self.assertEqual((yield enc.n), 0)
|
|
self.assertEqual((yield enc.o), 2)
|
|
|
|
yield enc.i.eq(0b0110)
|
|
self.assertEqual((yield enc.n), 1)
|
|
self.assertEqual((yield enc.o), 0)
|
|
|
|
sim = Simulator(enc)
|
|
sim.add_testbench(process)
|
|
sim.run()
|
|
|
|
|
|
class PriorityEncoderTestCase(FHDLTestCase):
|
|
def test_basic(self):
|
|
enc = PriorityEncoder(4)
|
|
def process():
|
|
self.assertEqual((yield enc.n), 1)
|
|
self.assertEqual((yield enc.o), 0)
|
|
|
|
yield enc.i.eq(0b0001)
|
|
self.assertEqual((yield enc.n), 0)
|
|
self.assertEqual((yield enc.o), 0)
|
|
|
|
yield enc.i.eq(0b0100)
|
|
self.assertEqual((yield enc.n), 0)
|
|
self.assertEqual((yield enc.o), 2)
|
|
|
|
yield enc.i.eq(0b0110)
|
|
self.assertEqual((yield enc.n), 0)
|
|
self.assertEqual((yield enc.o), 1)
|
|
|
|
sim = Simulator(enc)
|
|
sim.add_testbench(process)
|
|
sim.run()
|
|
|
|
|
|
class DecoderTestCase(FHDLTestCase):
|
|
def test_basic(self):
|
|
dec = Decoder(4)
|
|
def process():
|
|
self.assertEqual((yield dec.o), 0b0001)
|
|
|
|
yield dec.i.eq(1)
|
|
self.assertEqual((yield dec.o), 0b0010)
|
|
|
|
yield dec.i.eq(3)
|
|
self.assertEqual((yield dec.o), 0b1000)
|
|
|
|
yield dec.n.eq(1)
|
|
self.assertEqual((yield dec.o), 0b0000)
|
|
|
|
sim = Simulator(dec)
|
|
sim.add_testbench(process)
|
|
sim.run()
|
|
|
|
|
|
class ReversibleSpec(Elaboratable):
|
|
def __init__(self, encoder_cls, decoder_cls, i_width, args):
|
|
self.encoder_cls = encoder_cls
|
|
self.decoder_cls = decoder_cls
|
|
self.coder_args = args
|
|
self.i = Signal(i_width)
|
|
|
|
def elaborate(self, platform):
|
|
m = Module()
|
|
enc, dec = self.encoder_cls(*self.coder_args), self.decoder_cls(*self.coder_args)
|
|
m.submodules += enc, dec
|
|
m.d.comb += [
|
|
enc.i.eq(self.i),
|
|
dec.i.eq(enc.o),
|
|
Assert(enc.i == dec.o)
|
|
]
|
|
return m
|
|
|
|
|
|
class HammingDistanceSpec(Elaboratable):
|
|
def __init__(self, distance, encoder_cls, i_width, args):
|
|
self.distance = distance
|
|
self.encoder_cls = encoder_cls
|
|
self.coder_args = args
|
|
self.i1 = Signal(i_width)
|
|
self.i2 = Signal(i_width)
|
|
|
|
def elaborate(self, platform):
|
|
m = Module()
|
|
enc1, enc2 = self.encoder_cls(*self.coder_args), self.encoder_cls(*self.coder_args)
|
|
m.submodules += enc1, enc2
|
|
m.d.comb += [
|
|
enc1.i.eq(self.i1),
|
|
enc2.i.eq(self.i2),
|
|
Assume(enc1.i + 1 == enc2.i),
|
|
Assert(sum(enc1.o ^ enc2.o) == self.distance)
|
|
]
|
|
return m
|
|
|
|
|
|
class GrayCoderTestCase(FHDLTestCase):
|
|
def test_reversible(self):
|
|
spec = ReversibleSpec(encoder_cls=GrayEncoder, decoder_cls=GrayDecoder, i_width=16,
|
|
args=(16,))
|
|
self.assertFormal(spec, [spec.i], mode="prove")
|
|
|
|
def test_distance(self):
|
|
spec = HammingDistanceSpec(distance=1, encoder_cls=GrayEncoder, i_width=16, args=(16,))
|
|
self.assertFormal(spec, [spec.i1, spec.i2], mode="prove")
|