`include "util.svh"

module core
    #(  ADDR_BITS   = 15
    ,   DATA_BITS   = 12
    )
    (   input   bit clk
    ,   input   bit reset

    ,   input   bit [2:0] switch_df
    ,   input   bit [2:0] switch_if
    ,   input   bit [ADDR_BITS-3-1:0] switch_sr
    ,   input   bit switch_start
    ,   input   bit switch_load_add
    ,   input   bit switch_dep
    ,   input   bit switch_exam
    ,   input   bit switch_cont
    ,   input   bit switch_stop
    ,   input   bit switch_sing_step
    ,   input   bit switch_sing_inst

    // verilator lint_off UNDRIVEN
    ,   output  bit [ADDR_BITS-3-1:0] led_pc
    ,   output  bit [ADDR_BITS-3-1:0] led_memaddr
    ,   output  bit [DATA_BITS-1:0] led_memdata
    ,   output  bit [DATA_BITS-1:0] led_acc
    ,   output  bit [DATA_BITS-1:0] led_mq
    ,   output  bit led_and
    ,   output  bit led_tad
    ,   output  bit led_isz
    ,   output  bit led_dca
    ,   output  bit led_jms
    ,   output  bit led_jmp
    ,   output  bit led_iot
    ,   output  bit led_opr
    ,   output  bit led_fetch
    ,   output  bit led_execute
    ,   output  bit led_defer
    ,   output  bit led_word_count
    ,   output  bit led_current_address
    ,   output  bit led_break
    ,   output  bit led_ion
    ,   output  bit led_pause
    ,   output  bit led_run
    ,   output  bit [4:0] led_step_counter
    ,   output  bit [2:0] led_df
    ,   output  bit [2:0] led_if
    ,   output  bit led_link
    // verilator lint_on UNDRIVEN
    );

bit run;
bit int_enable;
bit int_delay;
bit int_request;

bit switch_start_observed;
bit switch_load_add_observed;
bit switch_dep_observed;
bit switch_exam_observed;
bit switch_cont_observed;

assign led_run = run;
assign led_ion = int_enable;

bit mem_ready;
bit mem_valid;
bit mem_write;
bit [ADDR_BITS-1:0] mem_address;
bit [DATA_BITS-1:0] mem_write_data;
assign led_current_address = mem_valid;

assign led_memaddr = mem_address[ADDR_BITS-3-1:0];

bit mem_read_valid;
bit [DATA_BITS-1:0] mem_read_data;

mem
    #(  .ADDR_BITS(ADDR_BITS)
    ,   .DATA_BITS(DATA_BITS)
    ,   .INIT_FILE("mem/focal69.loaded.hex")
//    ,   .INIT_FILE("build/mem/hello.hex")
    )
    memory
    (   .clk(clk)
    ,   .reset(reset)

    ,   .ready(mem_ready)
    ,   .valid(mem_valid)
    ,   .address(mem_address)
    ,   .write(mem_write)
    ,   .write_data(mem_write_data)

    ,   .read_valid(mem_read_valid)
    ,   .read_data(mem_read_data)
    );

bit rx_ready;
bit rx_valid;
bit [7:0] rx_data;

bit tx_ready;
bit tx_valid;
bit [7:0] tx_data;

jtag_uart
    #(  .INSTANCE(0)
    ) uart0
    (   .clk(clk)
    ,   .reset(reset)

    ,   .rx_ready(rx_ready)
    ,   .rx_valid(rx_valid)
    ,   .rx_data(rx_data)

    ,   .tx_ready(tx_ready)
    ,   .tx_valid(tx_valid)
    ,   .tx_data(tx_data)
    );

bit [ADDR_BITS-3-1:7] page;

bit [2:0] data_field;
bit [2:0] data_field_saved;
bit [2:0] inst_field;
bit [2:0] inst_field_buffer;
bit [2:0] inst_field_saved;
bit [ADDR_BITS-3-1:0] pc;
assign led_pc = pc;
bit [2:0] opcode;
bit [8:0] operand;
bit [DATA_BITS-1:0] acc;
bit link;

assign led_df = data_field;
assign led_if = inst_field;

assign led_acc = acc;
assign led_link = link;

assign led_and = opcode == 0;
assign led_tad = opcode == 1;
assign led_isz = opcode == 2;
assign led_dca = opcode == 3;
assign led_jms = opcode == 4;
assign led_jmp = opcode == 5;
assign led_iot = opcode == 6;
assign led_opr = opcode == 7;

bit tti_int_enable;
bit tti_valid;
bit [DATA_BITS-1:0] tti_data;

bit i;
bit z;
bit [6:0] wip;
bit [ADDR_BITS-1:0] address;

assign int_request =
    (tti_int_enable && tti_valid)
    ;

enum
    {   FETCH
    ,   DECODE
    ,   INDIRECT
    ,   INDIRECTED
    ,   PREINC
    ,   AGEN
    ,   EXEC
    ,   MEMWAIT
    ,   HALT
    ,   DEPOSIT
    } state;

assign led_fetch = state == FETCH;
assign led_execute = state == DECODE || state == AGEN || state == EXEC;
assign led_defer = state == INDIRECT || state == INDIRECTED || state == PREINC;
assign led_pause = state == MEMWAIT || state == HALT;

always_ff @(posedge clk) begin
    if (reset) begin
`ifdef SYNTHESIS
        run = 0;
`else
        run = 1;
`endif
        int_enable = 0;
        int_delay = 0;
        switch_start_observed = 0;
        switch_load_add_observed = 0;
        switch_dep_observed = 0;
        switch_exam_observed = 0;
        switch_cont_observed = 0;
        mem_valid = 0;
        rx_ready = 0;
        tx_valid = 0;
        tx_data = 0;
        data_field = 0;
        data_field_saved = 0;
        inst_field = 0;
        inst_field_buffer = 0;
        inst_field_saved = 0;
        pc = 'o200;
        acc = 0;
        link = 1;
        tti_int_enable = 0;
        tti_valid = 0;
        state = state.first;
    end else begin
        if (switch_start && !switch_start_observed) begin
            switch_start_observed = 1;
            run = 1;
            int_enable = 0;
            int_delay = 0;
            mem_valid = 0;
            acc = 0;
            link = 1;
            state = state.first;
        end

        if (!switch_start)
            switch_start_observed = 0;

        if (switch_load_add && !switch_load_add_observed) begin
            switch_load_add_observed = 1;
            data_field = switch_df;
            inst_field = switch_if;
            inst_field_buffer = switch_if;
            pc = switch_sr;
        end

        if (!switch_load_add)
            switch_load_add_observed = 0;

`ifdef HISTORIC_SWITCH_BEHAVIOUR
        if (switch_dep)
            switch_dep_observed = 1;

        if (!switch_dep && switch_dep_observed) begin
`else
        if (switch_dep && !switch_dep_observed) begin
            switch_dep_observed = 1;
`endif
            state = DEPOSIT;
            mem_valid = 1;
            mem_address = {inst_field, pc};
            mem_write = 1;
            mem_write_data = switch_sr;
            led_memdata = mem_write_data;
            run = 1;
        end

        if (!switch_dep)
            switch_dep_observed = 0;

        if (switch_exam && !switch_exam_observed) begin
            if (!run) begin
                switch_exam_observed = 1;
                state = FETCH;
                run = 1;
            end
        end

        if (!switch_exam)
            switch_exam_observed = 0;

        if (switch_cont && !switch_cont_observed) begin
            switch_cont_observed = 1;
            run = 1;
        end

        if (!switch_cont)
            switch_cont_observed = 0;

        if (switch_stop)
            run = 0;

        if (`lag(tx_ready)) tx_valid = 0;
        if (rx_ready && `lag(rx_valid)) begin
            tti_valid = 1;
            tti_data = {4'b0, 1'b1, `lag(rx_data[6:0])};
        end

        if (run) begin
            case (state)
                FETCH: begin
                    mem_valid = 1;
                    mem_address = {inst_field, pc};
                    mem_write = 0;
                    if (`lag(mem_ready)) begin
                        state = DECODE;
                        page = pc[ADDR_BITS-3-1:7];
                        ++pc;
                    end
                end

                DECODE: begin
                    automatic bit go;
                    go = 0;
                    mem_valid = 0;
                    mem_write = 0;
                    if (!switch_exam_observed && (int_enable && int_request)) begin
                        int_enable = 0;
                        int_delay = 0;
                        data_field_saved = data_field;
                        inst_field_saved = inst_field;
                        data_field = 0;
                        inst_field = 0;
                        inst_field_buffer = 0;
                        --pc;
                        opcode = 'b100;
                        operand = 'b000000000;
                        go = 1;
                    end else if (`lag(mem_read_valid)) begin
                        state = FETCH;
                        led_memdata = `lag(mem_read_data);
                        {opcode, operand} = `lag(mem_read_data);
                        if (switch_exam_observed) begin
                            run = 0;
                        end else begin
                            go = 1;
                        end
                    end
                    if (go) begin
                        int_enable = int_delay;
    //                    $display("%o: decode %o %o", pc-1, opcode, operand);
                        // verilator lint_off WIDTH
    //                    $display("%o  %b %o  0000", 14'(pc-1), link, acc);
                        // verilator lint_on WIDTH
                        {i, z, wip} = operand;
                        if (z)
                            address = {inst_field, page, wip};
                        else
                            address = {inst_field, 5'b0, wip};
                        case (opcode)
                            'o0, 'o1, 'o2: state = i ? INDIRECT : AGEN;
                            'o3, 'o4: state = i ? INDIRECT : EXEC;
                            'o5: begin
                                if (i) begin
                                    state = INDIRECT;
                                end else begin
                                    pc = address[ADDR_BITS-3-1:0];
                                    inst_field = inst_field_buffer;
                                end
                            end
                            'o7: begin
                                casez (operand)
                                    'b0????????: begin
                                        automatic bit cla, cll, cma, cml, rar, ral, bsw, iac;
                                        {cla, cll, cma, cml, rar, ral, bsw, iac} = operand[7:0];
                                        if (cla) acc = 0;
                                        if (cll) link = 0;
                                        if (cma) acc = ~acc;
                                        if (cml) link = ~link;
                                        if (iac) {link, acc} += 1;
                                        if (rar && !ral) begin
                                            {link, acc} = {acc[0], link, acc[11:1]};
                                            if (bsw) {link, acc} = {acc[0], link, acc[11:1]};
                                        end
                                        if (ral && !rar) begin
                                            {link, acc} = {acc, link};
                                            if (bsw) {link, acc} = {acc, link};
                                        end
                                        if (bsw && !(rar || ral)) acc = {acc[5:0], acc[11:6]};
                                    end
                                    'b1????0??0: begin
                                        automatic bit cla, sma, sza, snl, osr, hlt;
                                        automatic bit skip;
                                        {cla, sma, sza, snl, osr, hlt} = {operand[7:4], operand[2:1]};
                                        skip = 0;
                                        if (sma && acc[11]) skip = 1;
                                        if (sza && acc == 0) skip = 1;
                                        if (snl && link != 0) skip = 1;
                                        if (skip) pc++;
                                        if (cla) acc = 0;
                                        if (osr) begin
                                            $display("%o: unsupported front panel switch test", pc);
                                            $finish;
                                        end
                                        if (hlt) state = HALT;
                                    end
                                    'b1????1??0: begin
                                        automatic bit cla, spa, sna, szl, osr, hlt;
                                        automatic bit skip;
                                        {cla, spa, sna, szl, osr, hlt} = {operand[7:4], operand[2:1]};
                                        skip = 1;
                                        if (spa && acc[11]) skip = 0;
                                        if (sna && acc == 0) skip = 0;
                                        if (szl && link != 0) skip = 0;
                                        if (skip) pc++;
                                        if (cla) acc = 0;
                                        if (osr) begin
                                            $display("%o: unsupported front panel switch test", pc);
                                            $finish;
                                        end
                                        if (hlt) state = HALT;
                                    end
                                    default: begin
                                        $display("%o: decoded unknown opcode %o %o", pc-1, opcode, operand);
                                        $finish;
                                    end
                                endcase
                            end
                            'o6: begin
                                case (operand[8:6])
                                    'o0: begin
                                        case (operand[5:3])
                                            'o0: begin
                                                case (operand[2:0])
                                                    'o0: begin
                                                        if (int_enable)
                                                            ++pc;
                                                        int_enable = 0;
                                                        int_delay = 0;
                                                    end
                                                    'o1: int_delay = 1;
                                                    'o2: begin
                                                        int_enable = 0;
                                                        int_delay = 0;
                                                    end
                                                    'o3: begin
                                                        if (int_request)
                                                            ++pc;
                                                    end
                                                    'o4: acc = {link, 1'b0/*gt*/, int_request, 1'b0/*ii*/, int_enable, 1'b0/*u*/, 3'b0/*if*/, 3'b0/*df*/};
                                                    'o5: begin
                                                        link = acc[11];
                                                        if (acc[7]) begin
                                                            int_delay = 1;
                                                        end else begin
                                                            int_enable = 0;
                                                            int_delay = 0;
                                                        end
                                                    end
                                                    'o7: begin
                                                        int_enable = 0;
                                                        int_delay = 0;
                                                        acc = 0;
                                                        link = 1;
                                                        tx_valid = 0;
                                                        tti_valid = 0;
                                                    end
                                                    default: $display("%o: unsupported 600%o op", pc-1, operand[2:0]);
                                                endcase
                                            end
                                            'o3: begin
                                                case (operand[2:0])
                                                    'o1: if (tti_valid) pc++;
                                                    'o5: begin
                                                        tti_int_enable = acc[0];
                                                    end
                                                    'o6: begin
                                                        acc = tti_data;
                                                        tti_valid = 0;
                                                    end
                                                    default: begin
                                                        $display("%o: unsupported keyboard op %o", pc-1, operand[2:0]);
                                                        $finish;
                                                    end
                                                endcase
                                            end
                                            'o4: begin
                                                case (operand[2:0])
                                                    'o1: if (!tx_valid) pc++;
                                                    'o6: begin
                                                        tx_valid = 1;
                                                        tx_data = {1'b0, acc[6:0]};
                                                    end
                                                    default: begin
                                                        $display("%o: unsupported tty op %o", pc-1, operand[2:0]);
                                                        $finish;
                                                    end
                                                endcase
                                            end
                                            default: begin
                                                $display("%o: unsupported device %o (operation %o)", pc-1, operand[8:3], operand[2:0]);
                                            end
                                        endcase
                                    end
                                    'o2: begin
                                        if (operand[0])
                                            data_field = operand[5:3];
                                        if (operand[1])
                                            inst_field_buffer = operand[5:3];
                                        if (operand[2]) begin
                                            if (operand[3] && operand[4])
                                                acc = acc | {6'b0, inst_field_saved, data_field_saved};
                                            else if (operand[3])
                                                acc = acc | {6'b0, data_field, 3'b0};
                                            else if (operand[4])
                                                acc = acc | {6'b0, inst_field, 3'b0};
                                            if (operand[5]) begin
                                                data_field = data_field_saved;
                                                inst_field_buffer = inst_field_saved;
                                            end
                                        end
                                    end
                                    default: begin
                                        $display("%o: unsupported device %o (operation %o)", pc-1, operand[8:3], operand[2:0]);
                                    end
                                endcase
                            end
                            default: begin
                                $display("%o: decoded unknown opcode %o %o", pc-1, opcode, operand);
                                $finish;
                            end
                        endcase
                    end
                end

                INDIRECT: begin
                    mem_valid = 1;
                    mem_write = 0;
                    mem_address = address;
                    state = `lag(mem_ready) ? INDIRECTED : INDIRECT;
                end

                INDIRECTED: begin
                    if (`lag(mem_ready)) begin
                        mem_valid = 0;
                        mem_write = 0;
                    end
                    if (`lag(mem_read_valid)) begin
                        if (address[7:3] == 5'b00001) begin
                            led_memdata = `lag(mem_read_data);
                            address = {3'b0, `lag(mem_read_data)};
                            address += 1;
                            address[ADDR_BITS-1:ADDR_BITS-3] = data_field;
                            state = PREINC;
                        end else begin
                            led_memdata = `lag(mem_read_data);
                            address = {data_field, `lag(mem_read_data)};
                            case (opcode)
                                'o0, 'o1, 'o2: state = AGEN;
                                'o3, 'o4: state = EXEC;
                                'o5: begin
                                    pc = address[ADDR_BITS-3-1:0];
                                    inst_field = inst_field_buffer;
                                    state = FETCH;
                                end
                            endcase
                        end
                    end
                end

                PREINC: begin
                    mem_valid = 1;
                    mem_write = 1;
                    mem_write_data = address[DATA_BITS-1:0];
                    led_memdata = mem_write_data;
                    case (opcode)
                        'o0, 'o1, 'o2: state = `lag(mem_ready) ? AGEN : PREINC;
                        'o3, 'o4, 'o5: state = `lag(mem_ready) ? EXEC : PREINC;
                    endcase
                end

                AGEN: begin
                    mem_valid = 1;
                    case (opcode)
                        'o0, 'o1, 'o2: mem_write = 0;
                    endcase
                    mem_address = address;
                    state = `lag(mem_ready) ? EXEC : AGEN;
                end

                EXEC: begin
                    automatic bit stall = 0;
                    if (`lag(mem_ready)) begin
                        mem_valid = 0;
                        mem_write = 0;
                    end else if (mem_valid) begin
                        stall = 1;
                    end
                    case (opcode)
                        'o0, 'o1, 'o2: if (! `lag(mem_read_valid)) stall = 1;
                    endcase
                    if (! stall) begin
                        state = FETCH;
                        case (opcode)
                            'o0: begin led_memdata = `lag(mem_read_data); acc &= `lag(mem_read_data); end
                            'o1: begin led_memdata = `lag(mem_read_data); {link, acc} += {1'b0, `lag(mem_read_data)}; end
                            'o2: begin
                                mem_valid = 1;
                                mem_address = address;
                                mem_write = 1;
                                mem_write_data = `lag(mem_read_data) + 1;
                                led_memdata = mem_write_data;
                                if (mem_write_data == 0) ++pc;
                                state = MEMWAIT;
                            end
                            'o3: begin
                                mem_valid = 1;
                                mem_address = address;
                                mem_write = 1;
                                mem_write_data = acc;
                                led_memdata = mem_write_data;
                                acc = 0;
                                state = MEMWAIT;
                            end
                            'o4: begin
                                mem_valid = 1;
                                mem_address = address;
                                mem_write = 1;
                                mem_write_data = pc[DATA_BITS-1:0];
                                led_memdata = mem_write_data;
                                pc = address[ADDR_BITS-3-1:0] + 1;
                                inst_field = inst_field_buffer;
                                state = MEMWAIT;
                            end
                            'o5: begin
                                pc = address[ADDR_BITS-3-1:0];
                                inst_field = inst_field_buffer;
                            end
                        endcase
                    end
                end

                MEMWAIT: state = `lag(mem_ready) ? FETCH : MEMWAIT;

                HALT: begin
                    run = 0;
                    $display("\nhalt state reached");
                    $finish;
                end

                DEPOSIT: begin
                    if (`lag(mem_ready)) begin
                        state = FETCH;
                        page = pc[ADDR_BITS-3-1:7];
                        ++pc;
                        run = 0;
                    end
                end
            endcase

            if (switch_sing_step)
                run = 0;

            if (state == FETCH && switch_sing_inst)
                run = 0;

            rx_ready = !tti_valid;
        end
    end
end

endmodule