#include "ibm.h"

#include "x86.h"
#include "x86_flags.h"
#include "386_common.h"
#include "codegen.h"
#include "codegen_accumulate.h"
#include "codegen_ir.h"
#include "codegen_ops.h"
#include "codegen_ops_mmx_shift.h"
#include "codegen_ops_helpers.h"

uint32_t ropPSxxW_imm(codeblock_t *block, ir_data_t *ir, uint8_t opcode, uint32_t fetchdat, uint32_t op_32, uint32_t op_pc) {
        int reg = fetchdat & 7;
        int op = fetchdat & 0x38;
        int shift = fastreadb(cs + op_pc + 1);

        uop_MMX_ENTER(ir);
        codegen_mark_code_present(block, cs + op_pc, 1);
        switch (op) {
        case 0x10: /*PSRLW*/
                uop_PSRLW_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        case 0x20: /*PSRAW*/
                uop_PSRAW_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        case 0x30: /*PSLLW*/
                uop_PSLLW_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        default:
                return 0;
        }

        codegen_mark_code_present(block, cs + op_pc + 1, 1);
        return op_pc + 2;
}
uint32_t ropPSxxD_imm(codeblock_t *block, ir_data_t *ir, uint8_t opcode, uint32_t fetchdat, uint32_t op_32, uint32_t op_pc) {
        int reg = fetchdat & 7;
        int op = fetchdat & 0x38;
        int shift = fastreadb(cs + op_pc + 1);

        uop_MMX_ENTER(ir);
        codegen_mark_code_present(block, cs + op_pc, 1);
        switch (op) {
        case 0x10: /*PSRLD*/
                uop_PSRLD_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        case 0x20: /*PSRAD*/
                uop_PSRAD_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        case 0x30: /*PSLLD*/
                uop_PSLLD_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        default:
                return 0;
        }

        codegen_mark_code_present(block, cs + op_pc + 1, 1);
        return op_pc + 2;
}
uint32_t ropPSxxQ_imm(codeblock_t *block, ir_data_t *ir, uint8_t opcode, uint32_t fetchdat, uint32_t op_32, uint32_t op_pc) {
        int reg = fetchdat & 7;
        int op = fetchdat & 0x38;
        int shift = fastreadb(cs + op_pc + 1);

        uop_MMX_ENTER(ir);
        codegen_mark_code_present(block, cs + op_pc, 1);
        switch (op) {
        case 0x10: /*PSRLQ*/
                uop_PSRLQ_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        case 0x20: /*PSRAQ*/
                uop_PSRAQ_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        case 0x30: /*PSLLQ*/
                uop_PSLLQ_IMM(ir, IREG_MM(reg), IREG_MM(reg), shift);
                break;
        default:
                return 0;
        }

        codegen_mark_code_present(block, cs + op_pc + 1, 1);
        return op_pc + 2;
}
