29 lines
670 B
Python
29 lines
670 B
Python
|
from nmigen import *
|
||
|
|
|
||
|
|
from ..isa import Funct3
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ["LogicUnit"]
|
||
|
|
|
||
|
|
|
||
|
|
class LogicUnit(Elaboratable):
|
||
|
|
def __init__(self):
|
||
|
|
self.op = Signal(3)
|
||
|
|
self.src1 = Signal(32)
|
||
|
|
self.src2 = Signal(32)
|
||
|
|
|
||
|
|
self.result = Signal(32)
|
||
|
|
|
||
|
|
def elaborate(self, platform):
|
||
|
|
m = Module()
|
||
|
|
|
||
|
|
with m.Switch(self.op):
|
||
|
|
with m.Case(Funct3.XOR):
|
||
|
|
m.d.comb += self.result.eq(self.src1 ^ self.src2)
|
||
|
|
with m.Case(Funct3.OR):
|
||
|
|
m.d.comb += self.result.eq(self.src1 | self.src2)
|
||
|
|
with m.Case(Funct3.AND):
|
||
|
|
m.d.comb += self.result.eq(self.src1 & self.src2)
|
||
|
|
|
||
|
|
return m
|