From 52b05e5db08bd4360a157609d8529e01ad35cfd9 Mon Sep 17 00:00:00 2001
From: Rutherther <rutherther@proton.me>
Date: Sat, 28 Oct 2023 18:56:08 +0200
Subject: [PATCH] feat: add control_unit wrapper over instruction_decoder

---
 src/control_unit.sv | 94 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100755 src/control_unit.sv

diff --git a/src/control_unit.sv b/src/control_unit.sv
new file mode 100755
index 0000000..60ed34e
--- /dev/null
+++ b/src/control_unit.sv
@@ -0,0 +1,94 @@
+import cpu_types::*;
+
+module control_unit(
+  input [31:0]  instruction,
+
+  // TODO: mask for: memory in and out
+  // don't forget zero extension, sign extension...
+
+  // whether to save alu to memory
+  output        memory_we,
+
+  // going to pc (pc+4/imm, alu)
+  output        pc_source_t pc_src,
+
+  // if alu output zero(negated if zero_negate), jump (pc source pc + imm instead od pc + 4)
+  output        jump_instruction,
+  output        jump_negate_zero,
+
+  output [31:0] immediate,
+
+  // register file or pc
+  output        alu_1_source_t alu_src_1,
+  // register file or immediate
+  output        alu_2_source_t alu_src_2,
+
+  // going to alu
+  output [2:0]  alu_op,
+  output        alu_signed,
+  output        alu_negate,
+  output        alu_add_one,
+
+  // going to register file
+  output [4:0]  reg_rs1,
+  output [4:0]  reg_rs2,
+  // alu or memory
+  output        reg_rd_source_t reg_rd_src,
+  output [4:0]  reg_rd,
+  output        reg_we
+);
+
+  wire use_immediate;
+  wire load_memory;
+  wire load_pc, store_pc;
+  wire conditional_jump, unconditional_jump;
+
+  wire [2:0] alu_reg_op, alu_jump_op;
+  wire       alu_reg_add_one, alu_reg_negate, alu_reg_signed;
+  wire       alu_jump_add_one, alu_jump_negate;
+
+  instruction_decoder decoder(
+    .instruction(instruction),
+
+    .store_memory(memory_we),
+
+    .load_memory(load_memory),
+    .load_memory_size(),
+    .load_memory_sign_extension(),
+
+    .load_pc(load_pc),
+    .store_pc(store_pc),
+
+    .alu_reg_op(alu_reg_op),
+    .alu_reg_add_one(alu_reg_add_one),
+    .alu_reg_negate(alu_reg_negate),
+    .alu_reg_signed(alu_reg_signed),
+
+    .alu_jump_op(alu_jump_op),
+    .alu_jump_add_one(alu_jump_add_one),
+    .alu_jump_negate(alu_jump_negate),
+    .jump_negate_zero(jump_negate_zero),
+
+    .conditional_jump(conditional_jump),
+    .unconditional_jump(unconditional_jump),
+
+    .immediate(immediate),
+    .use_immediate(use_immediate),
+
+    .reg_rs1(reg_rs1),
+    .reg_rs2(reg_rs2),
+    .reg_rd(reg_rd),
+    .reg_we(reg_we)
+  );
+
+  assign alu_op = conditional_jump ? alu_jump_op : alu_reg_op;
+  assign alu_add_one = conditional_jump ? alu_jump_add_one : alu_reg_add_one;
+  assign alu_negate = conditional_jump ? alu_jump_negate : alu_reg_negate;
+  assign alu_signed = conditional_jump ? 0 : alu_reg_signed;
+
+  assign pc_src = unconditional_jump ? PC_ALU : PC_PLUS;
+  assign alu_src_1 = load_pc ? PC : REG_FILE_RS1;
+  assign reg_rd_src = store_pc ? RD_PC_PLUS : (load_memory ? RD_MEMORY : RD_ALU);
+  assign alu_src_2 = use_immediate ? IMMEDIATE : REG_FILE_RS2;
+
+endmodule
-- 
2.48.1