implement branch instructions

Author: outercloudstudio

src/main/scala/RISCV/Main.scala

@@ -456,6 +456,102 @@ class Main() extends Module {
 
                     printf("[JALR] RS1: %d Rd: %d Immediate: %b\n", decoder.io.rs1, decoder.io.rd, decoder.io.immediate);
                 }
+
+                // BEQ
+                is("b000_1100011".U) { 
+                    registers.io.read_address_a := decoder.io.rs1;
+                    registers.io.read_address_b := decoder.io.rs2;
+
+                    when(registers.io.out_a === registers.io.out_b) {
+                        program_pointer := (program_pointer.zext + decoder.io.immediate.asSInt).asUInt;
+                    }.otherwise {
+                        program_pointer := program_pointer + 1.U;
+                    }
+                    
+                    stage := 0.U;
+
+                    printf("[BEQ] Rs1: %d Rs2: %d Immediate: %b\n", decoder.io.rs1, decoder.io.rs2, decoder.io.immediate);
+                }
+
+                // BNEQ
+                is("b001_1100011".U) { 
+                    registers.io.read_address_a := decoder.io.rs1;
+                    registers.io.read_address_b := decoder.io.rs2;
+
+                    when(registers.io.out_a =/= registers.io.out_b) {
+                        program_pointer := (program_pointer.zext + decoder.io.immediate.asSInt).asUInt;
+                    }.otherwise {
+                        program_pointer := program_pointer + 1.U;
+                    }
+                    
+                    stage := 0.U;
+
+                    printf("[BNEQ] Rs1: %d Rs2: %d Immediate: %b\n", decoder.io.rs1, decoder.io.rs2, decoder.io.immediate);
+                }
+
+                // BLT
+                is("b100_1100011".U) { 
+                    registers.io.read_address_a := decoder.io.rs1;
+                    registers.io.read_address_b := decoder.io.rs2;
+
+                    when(registers.io.out_a.asSInt < registers.io.out_b.asSInt) {
+                        program_pointer := (program_pointer.zext + decoder.io.immediate.asSInt).asUInt;
+                    }.otherwise {
+                        program_pointer := program_pointer + 1.U;
+                    }
+                    
+                    stage := 0.U;
+
+                    printf("[BLT] Rs1: %d Rs2: %d Immediate: %b\n", decoder.io.rs1, decoder.io.rs2, decoder.io.immediate);
+                }
+
+                // BGE
+                is("b101_1100011".U) { 
+                    registers.io.read_address_a := decoder.io.rs1;
+                    registers.io.read_address_b := decoder.io.rs2;
+
+                    when(registers.io.out_a.asSInt >= registers.io.out_b.asSInt) {
+                        program_pointer := (program_pointer.zext + decoder.io.immediate.asSInt).asUInt;
+                    }.otherwise {
+                        program_pointer := program_pointer + 1.U;
+                    }
+                    
+                    stage := 0.U;
+
+                    printf("[BGE] Rs1: %d Rs2: %d Immediate: %b\n", decoder.io.rs1, decoder.io.rs2, decoder.io.immediate);
+                }
+
+                // BLTU
+                is("b110_1100011".U) { 
+                    registers.io.read_address_a := decoder.io.rs1;
+                    registers.io.read_address_b := decoder.io.rs2;
+
+                    when(registers.io.out_a < registers.io.out_b) {
+                        program_pointer := (program_pointer.zext + decoder.io.immediate.asSInt).asUInt;
+                    }.otherwise {
+                        program_pointer := program_pointer + 1.U;
+                    }
+                    
+                    stage := 0.U;
+
+                    printf("[BLTU] Rs1: %d Rs2: %d Immediate: %b\n", decoder.io.rs1, decoder.io.rs2, decoder.io.immediate);
+                }
+
+                // BGEU
+                is("b111_1100011".U) { 
+                    registers.io.read_address_a := decoder.io.rs1;
+                    registers.io.read_address_b := decoder.io.rs2;
+
+                    when(registers.io.out_a >= registers.io.out_b) {
+                        program_pointer := (program_pointer.zext + decoder.io.immediate.asSInt).asUInt;
+                    }.otherwise {
+                        program_pointer := program_pointer + 1.U;
+                    }
+                    
+                    stage := 0.U;
+
+                    printf("[BGEU] Rs1: %d Rs2: %d Immediate: %b\n", decoder.io.rs1, decoder.io.rs2, decoder.io.immediate);
+                }
             }
         }
 

src/test/scala/RISCV/MainSpec.scala

@@ -15,23 +15,17 @@ class MainSpec extends AnyFreeSpec with Matchers with ChiselSim {
       dut.clock.step(1);
 
       dut.io.debug_write.poke(true.B);
-      dut.io.debug_write_data.poke("b000000000000_00001_000_00001_1100111".U(32.W));
+      dut.io.debug_write_data.poke("b000000000111_00000_000_00010_0010011".U(32.W));
       dut.io.debug_write_addressess.poke(1.U);
 
       dut.clock.step(1);
 
       dut.io.debug_write.poke(true.B);
-      dut.io.debug_write_data.poke("b000000000110_00000_000_00010_0010011".U(32.W));
+      dut.io.debug_write_data.poke("b0000000_00010_00001_001_00010_1100011".U(32.W));
       dut.io.debug_write_addressess.poke(2.U);
 
       dut.clock.step(1);
 
-      dut.io.debug_write.poke(true.B);
-      dut.io.debug_write_data.poke("b000000000111_00000_000_00010_0010011".U(32.W));
-      dut.io.debug_write_addressess.poke(3.U);
-
-      dut.clock.step(1);
-
       dut.io.debug_write.poke(false.B);
       dut.io.execute.poke(true.B);