`include "cpu_pkg.svh"

import cpu_pkg::*;

module registers (
    input  logic clk_i,
    input  logic nreset_i,

    input  reg8_t reg8_wselect_i,
    input  logic  reg8_we_i,
    input  logic [ 7:0] reg8_wdata_i,

    input  reg16_t reg16_wselect_i,
    input  logic   reg16_we_i,
    input  logic [15:0] reg16_wdata_i,

    input  reg8_t reg8_rselect_i,
    input  reg16_t reg16_rselect_i,
    output logic [ 7:0] reg8_rdata_o,
    output logic [15:0] reg16_rdata_o
);

    logic [15:0] reg_r   [0:2];
    logic [15:0] reg_next;
    logic [ 1:0] reg_we  [0:2];

    generate for (genvar i = 0; i <= 2; i++) begin : gen_regs
        always_ff @(posedge clk_i or negedge nreset_i) begin
            if (~nreset_i) begin
                reg_r[i] <= '0;
            end else if (|reg_we[i]) begin
                reg_r[i]<= reg_next;
            end
        end

        assign reg_we[i][0] = (reg16_we_i & (reg16_wselect_i == i)) |
                              (reg8_we_i  & ({reg8_wselect_i, 1'b0} == i));
        assign reg_we[i][1] = (reg16_we_i & (reg16_wselect_i == i)) |
                              (reg8_we_i  & ({reg8_wselect_i, 1'b1} == i));
    end endgenerate

    assign reg_next = ({16{reg8_we_i}}  & {reg8_wdata_i, reg8_wdata_i}) |
                      ({16{reg16_we_i}} & reg16_wdata_i);

    assign reg8_rdata_o  = reg8_rselect_i[0] ? reg_r[reg8_rselect_i[2:1]][15:8] :
                                               reg_r[reg8_rselect_i[2:1]][ 7:0];
    assign reg16_rdata_o = reg_r[reg16_rselect_i];

endmodule : registers