// The OS/K Team licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

#include <pc/arch.h>

char *cond_suffixes[] =
{
    "-",
    "c", "o", "z", "s", "p",
    "a", "ae", "b", "be",
    "g", "ge", "l", "le",
    "cxz",
    "?"
};

#ifndef _ATT_STYLE
#define _ATT_STYLE 0
#endif

static void dump_acc(ctx_t *ctx, acc_t *p);

void dump_instr(ctx_t *ctx,
                instr_t *in,
                acc_t *p1,
                acc_t *p2,
                bool lock,
                bool rep,
                uint cond,
                ulong pc)
{
    log("0x%lX:\t", pc);

    if (lock)
        log("lock");

    log("%s", in->name);

    if (rep)
        log(".rep");

    if (cond)
    {
        log(".");

        if (cond & (1 << 4))
        {
            cond &= ~(1 << 4);
            log("n");
        }

        assert(cond <= sizeof(cond_suffixes)/sizeof(char *));

        log("%s", cond_suffixes[cond]);
    }
    
    if (!rep) log("\t\t");

    else
        log("\t");

    if (p1)
    {
        dump_acc(ctx, p1);
        if (p2) {
            log(", ");
            dump_acc(ctx, p2);
        }
    }

    log("\n");
}

void dump_acc(ctx_t *ctx, acc_t *p)
{
    uint mfmt;
    sym_t *sym;

    if (p->type == A_REG)
        log("%s", ctx->r[p->reg].name);

    else if (p->type == A_IMM64)
    {
        if (p->val < 0xA)
            log("%lu", p->val);

        else
        {
            sym = find_sym_by_addr(p->val);

            if (sym)
                log("$%s(0x%lX)", sym->name, sym->addr);
            else
                log("0x%lX", p->val);
        }
    }

    else
    {
        log("%c[", getmempref(p->mlen));

        mfmt = p->type & AM_MFMT_MASK;
        
        if (mfmt == AM_IMM64)
            log("0x%lX]", p->addr);
  
        else if (mfmt == AM_RR)
        {
            if (p->reg1 && p->reg2)
                log("%s+%s]", ctx->r[p->reg1].name, ctx->r[p->reg2].name);
            
            else log("%s]", ctx->r[p->reg1 ? p->reg1 : p->reg2].name);
        }

        else if (mfmt == AM_RRI)
        {
            if (p->reg1 && p->reg2)
                log("%s+%s+%hd]", ctx->r[p->reg1].name,
                                 ctx->r[p->reg2].name, p->imm2);
            
            else log("%s+%hd]",
                    ctx->r[p->reg1 ? p->reg1 : p->reg2].name, p->imm2);
        }

        else if (mfmt == AM_RRII)
        {
            if (p->reg1)
                log("%s+%s*%u+%hd]",
                        ctx->r[p->reg1].name,
                        ctx->r[p->reg2].name,
                        p->imm1, p->imm2);
            else
                log("%s*%u+%hd]",
                        ctx->r[p->reg2].name,
                        p->imm1, p->imm2);
        }
    }
}