#include "m7700.hpp"

inline static void outreg(const int n) { out_register(ph.regNames[n]); }

static void outaddr(op_t &op, const bool replace_with_label = true) {
    bool ind = is_addr_ind(op);      // is operand indirect ?

    if (ind)    out_symbol('(');

    // if addressing mode is direct and the value of DR is unknown,
    // we have to print DR:x (where x is the "indexed" value)
    if (is_addr_dr_rel(op) && getSR(cmd.ea, rDR) == BADSEL) {
        outreg(rDR);
        out_symbol(':');
        OutValue(op, OOF_ADDR | OOFS_NOSIGN);
    }
    // otherwise ...
    else {
        char *b = get_name_expr(cmd.ea + op.offb, op.n, toEA(cmd.cs, op.addr), op.addr);
        if (b == NULL || !replace_with_label) {
            if (replace_with_label) out_tagon(COLOR_ERROR);
            OutValue(op, OOF_ADDR | OOFS_NOSIGN /*| OOFW_16*/);
            if (replace_with_label) out_tagoff(COLOR_ERROR);
        }
        else {
            OutLine(b);
        }
    }

    if (ind)    out_symbol(')');
}

static void outdispl(op_t &op) {
    if (is_displ_ind(op)) {
        out_symbol('(');
        outaddr(op, false);
        out_symbol(',');
        if (!(ash.uflag & UAS_INDX_NOSPACE)) OutChar(' ');
        outreg(op.reg);
        out_symbol(')');
    }
    else if (is_displ_ind_p1(op)) {
        out_symbol('(');
        outaddr(op, false);
        out_symbol(')');
        out_symbol(',');
        OutChar(' ');
        outreg(op.reg);
    }
    else {
        outaddr(op, false);
        out_symbol(',');
        OutChar(' ');
        outreg(op.reg);
    }
}

void header(void) {
    gen_cmt_line("Processor:            %s [%s]", inf.procName, device);
    gen_cmt_line("Target assembler:     %s", ash.name);

    if (ash.header != NULL) {
        for (const char **ptr = ash.header; *ptr != NULL; ptr++) {
            MakeLine(*ptr,0);
        }
    }

    if (ash.uflag & UAS_DEVICE_DIR) {
        switch (ptype) {
            case prc_m7700:     printf_line(-1, ".MCU M37700"); break;
            case prc_m7750:     printf_line(-1, ".MCU M37750"); break;
            default:            IDA_ERROR("invalid ptype in header()");
        }
    }
}

void footer(void) {
    char buf[MAXSTR];
    char *const end = buf + sizeof(buf);
    if (ash.end != NULL) {
        MakeNull();
        register char *p = tag_addstr(buf, end, COLOR_ASMDIR, ash.end);
        char name[MAXSTR];
        if (get_colored_name(BADADDR, inf.beginEA, name, sizeof(name)) != NULL) {
            APPCHAR(p, end, ' ');
            if (ash.uflag & UAS_END_WITHOUT_LABEL)
            {
              APPCHAR(p, end, ';');
              APPCHAR(p, end, ' ');
            }
            APPEND(p, end, name);
        }
        MakeLine(buf);
    }
    else {
        gen_cmt_line("end of file");
    }
}

static bool bitmask2list(op_t &op) {
    static const char *flags[] = {
        "N", "V", "m", "x", "D", "I", "Z", "C"
    };
    static const int regs[] = {
        rPS, rPG, rDT, rDR, rY, rX, rB, rA
    };

    enum { bitFLAGS, bitREGS } t;
    switch (cmd.itype) {
        case m7700_psh:
        case m7700_pul:
            t = bitREGS;
            break;

        case m7700_sep:
        case m7700_clp:
            t = bitFLAGS;
            break;

        default:
            return false;
    }

    if (op.value == 0)
        return false;

    bool ok = false;
    for (int tmp = op.value, i = 1, j = 0; j < 8; i *= 2, j++) {
        if (((tmp & i) >> j) != 1)
            continue;

        if (ok) {
            out_symbol(',');
            OutChar(' ');
        }

        switch (t) {
            case bitFLAGS:      out_register(flags[7 - j]); break;
            case bitREGS:       outreg(regs[7 - j]); break;
        }
        ok = true;
    }
    return true;
}

bool outop(op_t &op) {
    switch (op.type) {
        // register
        case o_reg:
            outreg(op.reg);
            break;

        // immediate
        case o_imm:
           {
                const ioport_t * port = find_sym(op.value);

                // this immediate is represented in the .cfg file
                if (port != NULL) {
                    // output the port name instead of the numeric value
                    out_line(port->name, COLOR_IMPNAME);
                }
                // otherwise, simply print the value
                else {
                    bool list_ok = false;

                    if (ash.uflag & UAS_BITMASK_LIST)
                        list_ok = bitmask2list(op);

                    if (!list_ok) {
                        if (!(is_imm_without_sharp(op)))
                            out_symbol('#');
                        OutValue(op, OOFW_IMM);
                    }
                }
            }
            break;

        // bit
        case o_bit:
            {
                const ioport_bit_t * port = NULL;

                if (op.n == 0 && (cmd.Op2.type == o_near || cmd.Op2.type == o_mem))
                    port = find_bit(cmd.Op2.addr, op.value);

                // this immediate is represented in the .cfg file
                if (port != NULL && port->name != NULL) {
                    // output the port name instead of the numeric value
                    out_line(port->name, COLOR_IMPNAME);
                }
                // otherwise, simply print the value
                else {
                    out_symbol('#');
                    OutValue(op, OOFW_IMM);
                }
            }
            break;

        // data / code memory address
        case o_near:
        case o_mem:
            outaddr(op);
            break;

        // displ
        case o_displ:
            outdispl(op);
            break;

        // ignore void operands
        case o_void:
            break;

        default:
            IDA_ERROR("Invalid op.type in outop()");
            break;
    }
    return 1;
}

void out(void) {
    char buf[MAXSTR];

    init_output_buffer(buf, sizeof(buf));

    //
    // print insn mnemonic
    //

    char postfix[3];
    postfix[0] = '\0';

    if (is_insn_long_format()) {
        qstrncpy(postfix, "l", sizeof(postfix));
    }

    OutMnem(8, postfix);

    //
    // print insn operands
    //

    out_one_operand(0);        // output the first operand

    if (cmd.Op2.type != o_void) {
        out_symbol(',');
        OutChar(' ');
        out_one_operand(1);
    }

    if (cmd.Op3.type != o_void) {
        out_symbol(',');
        OutChar(' ');
        out_one_operand(2);
    }

    // output a character representation of the immediate values
    // embedded in the instruction as comments
    if (isVoid(cmd.ea,uFlag,0)) OutImmChar(cmd.Op1);
    if (isVoid(cmd.ea,uFlag,1)) OutImmChar(cmd.Op2);
    if (isVoid(cmd.ea,uFlag,2)) OutImmChar(cmd.Op3);

    term_output_buffer();                   // terminate the output string
    gl_comm = 1;                            // ask to attach a possible user-
                                            // defined comment to it
    MakeLine(buf);                          // pass the generated line to the
                                            // kernel
}

// generate segment header
void gen_segm_header(ea_t ea) {
    segment_t *Sarea = getseg(ea);
    char *segname = get_segm_name(Sarea);

    if (ash.uflag & UAS_SEGM)
        printf_line(-1, COLSTR("SEGMENT %s", SCOLOR_ASMDIR), segname);
    else
        printf_line(-1, COLSTR(".SECTION %s", SCOLOR_ASMDIR), segname);

    unsigned long orgbase = ea - get_segm_para(Sarea);

    if (orgbase != 0)
        printf_line(-1, COLSTR("%s %s", SCOLOR_ASMDIR),
            ash.origin, btoa(orgbase));
}