# Capstone Python bindings, by Nguyen Anh Quynnh <aquynh@gmail.com>

import ctypes
from . import copy_ctypes_list
from .systemz_const import *

# define the API
class SystemZOpMem(ctypes.Structure):
    _fields_ = (
        ('am', ctypes.c_int),
        ('base', ctypes.c_uint8),
        ('index', ctypes.c_uint8),
        ('length', ctypes.c_uint64),
        ('disp', ctypes.c_int64),
    )

class SystemZOpValue(ctypes.Union):
    _fields_ = (
        ('reg', ctypes.c_uint),
        ('imm', ctypes.c_int64),
        ('mem', SystemZOpMem),
    )

class SystemZOp(ctypes.Structure):
    _fields_ = (
        ('type', ctypes.c_uint),
        ('value', SystemZOpValue),
        ('access', ctypes.c_int),
        ('imm_width', ctypes.c_uint8),
    )

    @property
    def imm(self):
        return self.value.imm

    @property
    def reg(self):
        return self.value.reg

    @property
    def mem(self):
        return self.value.mem


class CsSystemZ(ctypes.Structure):
    _fields_ = (
        ('cc', ctypes.c_uint),
        ('format', ctypes.c_int),
        ('op_count', ctypes.c_uint8),
        ('operands', SystemZOp * 6),
    )

def get_arch_info(a):
    return a.cc, a.format, copy_ctypes_list(a.operands[:a.op_count])

