from migen.fhdl.structure import * from migen.fhdl.structure import _Operator from migen.flow.actor import * from migen.corelogic.record import * from migen.corelogic import divider class _SimpleBinary(CombinatorialActor): def __init__(self, op, bv_op, bv_r): self.op = op CombinatorialActor.__init__(self, ("operands", Sink, [("a", bv_op), ("b", bv_op)]), ("result", Source, [("r", bv_r)])) def get_process_fragment(self): return Fragment([ self.token("result").r.eq(_Operator(self.op, [self.token("operands").a, self.token("operands").b])) ]) class Add(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "+", bv, BV(bv.width+1, bv.signed)) class Sub(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "-", bv, BV(bv.width+1, bv.signed)) class Mul(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "*", bv, BV(2*bv.width, bv.signed)) class And(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "&", bv, bv) class Xor(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "^", bv, bv) class Or(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "|", bv, bv) class LT(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "<", bv, BV(1)) class LE(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "<=", bv, BV(1)) class EQ(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "==", bv, BV(1)) class NE(_SimpleBinary): def __init__(self, bv): _SimpleBinary.__init__(self, "!=", bv, BV(1)) class DivMod(SequentialActor): def __init__(self, width): self.div = divider.Divider(width) SequentialActor.__init__(self, width, ("operands", Sink, [("dividend", self.div.dividend_i), ("divisor", self.div.divisor_i)]), ("result", Source, [("quotient", self.div.quotient_o), ("remainder", self.div.remainder_o)])) def get_process_fragment(self): return self.div.get_fragment() + Fragment([self.div.start_i.eq(self.trigger)])