Source code for hwtLib.examples.statements.constCondition

from hwt.code import If
from hwt.hdl.types.bits import Bits
from hwt.interfaces.std import Signal, VectSignal, Clk
from hwt.simulator.simTestCase import SimTestCase
from hwt.synthesizer.unit import Unit
from hwtSimApi.constants import CLK_PERIOD


class ConstCondition(Unit):
    def _declr(self):
        self.clk = Clk()
        self.a = VectSignal(2)
        self.b = VectSignal(2)
        self.c = Signal()._m()

    def _impl(self):
        one = self._sig('one', Bits(1))
        intermed = self._reg('intermed', Bits(1))
        one(1)
        self.c(intermed)
        If(one._eq(1),
            If(self.a._eq(1),
                intermed(1),
            ),
            If(self.b._eq(1),
                intermed(0),
            ),
            If(self.b._eq(2),
                intermed(1),
            ),
        )


class ConstConditionTC(SimTestCase):

    @classmethod
    def setUpClass(cls):
        cls.u = ConstCondition()
        cls.compileSim(cls.u)

    def test_reg_update(self):
        u = self.u
        u.a._ag.data.extend([0, 1, 0, 0, 0])
        u.b._ag.data.extend([0, 0, 1, 2, 1])
        self.runSim(10 * CLK_PERIOD)
        self.assertValSequenceEqual(u.c._ag.data, [None, None, 1, 0, 1, 0, 0, 0, 0, 0])


def main():
    import unittest
    from hwt.synthesizer.utils import to_rtl_str
    u = ConstCondition()
    print(to_rtl_str(u))

    testLoader = unittest.TestLoader()
    suite = testLoader.loadTestsFromTestCase(ConstConditionTC)
    runner = unittest.TextTestRunner(verbosity=3)
    runner.run(suite)


if __name__ == '__main__':
    main()