/* Copyright (C) 2014-2015 Ben Kurtovic <ben.kurtovic@gmail.com>
   Released under the terms of the MIT License. See LICENSE for details. */

#pragma once

#include <stdint.h>

#define MAX_SYMBOL_SIZE 256

typedef enum {
    AT_REGISTER,
    AT_IMMEDIATE,
    AT_INDIRECT,
    AT_INDEXED,
    AT_LABEL,
    AT_CONDITION
} ASMArgType;

typedef enum {
    REG_A, REG_F, REG_B, REG_C, REG_D, REG_E, REG_H, REG_L, REG_I, REG_R,
    REG_AF, REG_BC, REG_DE, REG_HL, REG_IX, REG_IY,
    REG_PC, REG_SP,
    REG_AF_, REG_IXH, REG_IXL, REG_IYH, REG_IYL
} ASMArgRegister;

typedef enum {
    IMM_U16 = 0x01,  // unsigned 16-bit: [0, 65535]
    IMM_U8  = 0x02,  // unsigned 8-bit: [0, 255]
    IMM_S8  = 0x04,  // signed 8-bit: [-128, 127]
    IMM_REL = 0x08,  // relative offset: [-126, 129]
    IMM_BIT = 0x10,  // bit index: [0, 7]
    IMM_RST = 0x20,  // RST page 0 addr: {0x00, 0x08, 0x10, 0x18, ..., 0x38}
    IMM_IM  = 0x40   // interrupt mode: [0, 2]
} ASMArgImmType;

typedef struct {
    ASMArgImmType mask;
    uint16_t uval;
    int16_t sval;
} ASMArgImmediate;

typedef struct {
    char text[MAX_SYMBOL_SIZE];
} ASMArgLabel;

typedef struct {
    ASMArgType type;
    union {
        ASMArgRegister reg;
        ASMArgImmediate imm;
        ASMArgLabel label;
    } addr;
} ASMArgIndirect;

typedef struct {
    ASMArgRegister reg;
    int8_t offset;
} ASMArgIndexed;

typedef enum {
    COND_NZ, COND_N, COND_NC, COND_C, COND_PO, COND_PE, COND_P, COND_M
} ASMArgCondition;

typedef struct {
    ASMArgType type;
    union {
        ASMArgRegister reg;
        ASMArgImmediate imm;
        ASMArgIndirect indirect;
        ASMArgIndexed index;
        ASMArgLabel label;
        ASMArgCondition cond;
    } data;
} ASMInstArg;