from insn import *

class rvfi_insn_UJ(rvfi_insn):
    def __init__(self):
        super(rvfi_insn_UJ, self).__init__()
        self.insn_padding = Signal(32)
        self.insn_imm = Signal(32)
        self.insn_rd = Signal(5)
        self.insn_opcode = Signal(7)
        self.misa_ok = Signal(1)
        self.ialign16 = Signal(1)
    def ports(self):
        return super(rvfi_insn_UJ, self).ports()
    def elaborate(self, platform):
        m = super(rvfi_insn_UJ, self).elaborate(platform)

        # UJ-type instruction format
        m.d.comb += self.insn_padding.eq(self.rvfi_insn >> 32)
        m.d.comb += self.insn_imm.eq(Value.as_signed(Cat(self.rvfi_insn[21:31], self.rvfi_insn[20], self.rvfi_insn[12:20], self.rvfi_insn[31]) << 1))
        m.d.comb += self.insn_rd.eq(self.rvfi_insn[7:12])
        m.d.comb += self.insn_opcode.eq(self.rvfi_insn[:7])

        m.d.comb += self.misa_ok.eq(1)
        m.d.comb += self.ialign16.eq(0)

        # Default assignments
        m.d.comb += self.spec_rs1_addr.eq(0)
        m.d.comb += self.spec_rs2_addr.eq(0)
        m.d.comb += self.spec_mem_addr.eq(0)
        m.d.comb += self.spec_mem_rmask.eq(0)
        m.d.comb += self.spec_mem_wmask.eq(0)
        m.d.comb += self.spec_mem_wdata.eq(0)

        return m