1 files changed, 26 insertions, 49 deletions
diff --git a/src/instructions/rvi.rs b/src/instructions/rvi.rs
index d8439fa..a348596 100644
--- a/src/instructions/rvi.rs
+++ b/src/instructions/rvi.rs
@@ -106,52 +106,29 @@ pub fn jalr(core: &mut Core, instr: Instruction) -> Result<(), ExceptionType> {
     Ok(())
 }
 
-pub fn beq(core: &mut Core, instr: Instruction) -> Result<(), ExceptionType> {
-    if core.reg_read(instr.rs1()) == core.reg_read(instr.rs2()) {
-        core.pc = core.pc.wrapping_add(instr.imm_b());
-    } else {
-        core.advance_pc();
-    }
-
-    Ok(())
-}
-
-pub fn bne(core: &mut Core, instr: Instruction) -> Result<(), ExceptionType> {
-    if core.reg_read(instr.rs1()) != core.reg_read(instr.rs2()) {
-        core.pc = core.pc.wrapping_add(instr.imm_b());
-    } else {
-        core.advance_pc();
-    }
-
-    Ok(())
-}
-
-pub fn blt(core: &mut Core, instr: Instruction) -> Result<(), ExceptionType> {
-    if (core.reg_read(instr.rs1()) as i64) < (core.reg_read(instr.rs2()) as i64) {
-        core.pc = core.pc.wrapping_add(instr.imm_b());
-    } else {
-        core.advance_pc();
-    }
-
-    Ok(())
-}
-
-pub fn bgeu(core: &mut Core, instr: Instruction) -> Result<(), ExceptionType> {
-    if core.reg_read(instr.rs1()) >= core.reg_read(instr.rs2()) {
-        core.pc = core.pc.wrapping_add(instr.imm_b());
-    } else {
-        core.advance_pc();
-    }
-
-    Ok(())
-}
-
-pub fn bltu(core: &mut Core, instr: Instruction) -> Result<(), ExceptionType> {
-    if core.reg_read(instr.rs1()) < core.reg_read(instr.rs2()) {
-        core.pc = core.pc.wrapping_add(instr.imm_b());
-    } else {
-        core.advance_pc();
-    }
-
-    Ok(())
-}
+macro_rules! instr_branch {
+    ($name:ident, $cond:expr) => {
+        pub fn $name(core: &mut Core, instr: Instruction) -> Result<(), ExceptionType> {
+            let a = core.reg_read(instr.rs1());
+            let b = core.reg_read(instr.rs2());
+            if $cond(a, b) {
+                core.pc = core.pc.wrapping_add(instr.imm_b());
+            } else {
+                core.advance_pc();
+            }
+            Ok(())
+        }
+    };
+}
+macro_rules! instr_branch_signed {
+    ($name:ident, $cond:expr) => {
+        instr_branch!($name, |a, b| $cond((a as i64), (b as i64)));
+    };
+}
+
+instr_branch!(beq, |a, b| a == b);
+instr_branch!(bne, |a, b| a != b);
+instr_branch!(bltu, |a, b| a < b);
+instr_branch!(bgeu, |a, b| a >= b);
+instr_branch_signed!(blt, |a, b| a < b);
+instr_branch_signed!(bge, |a, b| a >= b);