From 801b4159de46a183ea1ccab5aeb3e87f1f9af0f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=B3zef=20Kucia?= <jkucia@codeweavers.com>
Date: Tue, 20 Jun 2017 14:14:44 +0200
Subject: [PATCH] libs/vkd3d-shader: Translate some ALU instructions.

---
 libs/vkd3d-shader/spirv.c                | 69 ++++++++++++++++++++++++
 libs/vkd3d-shader/vkd3d_shader_private.h |  2 +
 2 files changed, 71 insertions(+)

diff --git a/libs/vkd3d-shader/spirv.c b/libs/vkd3d-shader/spirv.c
index bb467ce1..f928e6c9 100644
--- a/libs/vkd3d-shader/spirv.c
+++ b/libs/vkd3d-shader/spirv.c
@@ -1358,6 +1358,68 @@ static void vkd3d_dxbc_compiler_emit_dcl_thread_group(struct vkd3d_dxbc_compiler
     vkd3d_spirv_set_local_size(builder, group_size->x, group_size->y, group_size->z);
 }
 
+static SpvOp vkd3d_dxbc_compiler_map_alu_instruction(const struct vkd3d_shader_instruction *instruction)
+{
+    static const struct
+    {
+        enum VKD3D_SHADER_INSTRUCTION_HANDLER handler_idx;
+        SpvOp spirv_op;
+    }
+    alu_ops[] =
+    {
+        {VKD3DSIH_ADD,  SpvOpFAdd},
+        {VKD3DSIH_AND,  SpvOpBitwiseAnd},
+        {VKD3DSIH_DIV,  SpvOpFDiv},
+        {VKD3DSIH_MUL,  SpvOpFMul},
+        {VKD3DSIH_UTOF, SpvOpConvertUToF},
+    };
+    unsigned int i;
+
+    for (i = 0; i < ARRAY_SIZE(alu_ops); ++i)
+    {
+        if (alu_ops[i].handler_idx == instruction->handler_idx)
+            return alu_ops[i].spirv_op;
+    }
+
+    return SpvOpMax;
+}
+
+static void vkd3d_dxbc_compiler_emit_alu_instruction(struct vkd3d_dxbc_compiler *compiler,
+        const struct vkd3d_shader_instruction *instruction)
+{
+    struct vkd3d_spirv_builder *builder = &compiler->spirv_builder;
+    const struct vkd3d_shader_dst_param *dst = instruction->dst;
+    const struct vkd3d_shader_src_param *src = instruction->src;
+    uint32_t src_ids[VKD3D_DXBC_MAX_SOURCE_COUNT];
+    unsigned int component_count;
+    uint32_t type_id, val_id;
+    unsigned int i;
+    SpvOp op;
+
+    op = vkd3d_dxbc_compiler_map_alu_instruction(instruction);
+    if (op == SpvOpMax)
+    {
+        FIXME("Unhandled ALU instruction %#x.\n", instruction->handler_idx);
+        return;
+    }
+
+    assert(instruction->dst_count == 1);
+    assert(instruction->src_count <= VKD3D_DXBC_MAX_SOURCE_COUNT);
+
+    component_count = vkd3d_write_mask_component_count(dst->write_mask);
+    type_id = vkd3d_spirv_get_type_id(builder,
+            vkd3d_component_type_from_data_type(dst->reg.data_type), component_count);
+
+    for (i = 0; i < instruction->src_count; ++i)
+        src_ids[i] = vkd3d_dxbc_compiler_emit_load_reg(compiler,
+                &src[i].reg, src[i].swizzle, dst->write_mask);
+
+    val_id = vkd3d_spirv_build_op_trv(builder, &builder->function_stream, op, type_id,
+            src_ids, instruction->src_count);
+
+    vkd3d_dxbc_compiler_emit_store_reg(compiler, &dst->reg, dst->write_mask, val_id);
+}
+
 static void vkd3d_dxbc_compiler_emit_mov(struct vkd3d_dxbc_compiler *compiler,
         const struct vkd3d_shader_instruction *instruction)
 {
@@ -1395,6 +1457,13 @@ void vkd3d_dxbc_compiler_handle_instruction(struct vkd3d_dxbc_compiler *compiler
         case VKD3DSIH_MOV:
             vkd3d_dxbc_compiler_emit_mov(compiler, instruction);
             break;
+        case VKD3DSIH_ADD:
+        case VKD3DSIH_AND:
+        case VKD3DSIH_DIV:
+        case VKD3DSIH_MUL:
+        case VKD3DSIH_UTOF:
+            vkd3d_dxbc_compiler_emit_alu_instruction(compiler, instruction);
+            break;
         case VKD3DSIH_RET:
             vkd3d_dxbc_compiler_emit_return(compiler, instruction);
             break;
diff --git a/libs/vkd3d-shader/vkd3d_shader_private.h b/libs/vkd3d-shader/vkd3d_shader_private.h
index 050b57e6..e654a4f5 100644
--- a/libs/vkd3d-shader/vkd3d_shader_private.h
+++ b/libs/vkd3d-shader/vkd3d_shader_private.h
@@ -878,4 +878,6 @@ static inline unsigned int vkd3d_swizzle_get_component(DWORD swizzle,
     return (swizzle >> (2 * idx)) & 0x3;
 }
 
+#define VKD3D_DXBC_MAX_SOURCE_COUNT 6
+
 #endif  /* __VKD3D_SHADER_PRIVATE_H */