2017-06-19 09:05:53 -07:00
|
|
|
/*
|
|
|
|
* Copyright 2017 Józef Kucia for CodeWeavers
|
|
|
|
*
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
|
|
* License as published by the Free Software Foundation; either
|
|
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This library is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
* License along with this library; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "vkd3d_shader_private.h"
|
|
|
|
|
|
|
|
#include "spirv/1.0/spirv.h"
|
2017-06-19 09:05:53 -07:00
|
|
|
#ifdef HAVE_SPIRV_TOOLS
|
|
|
|
# include "spirv-tools/libspirv.h"
|
|
|
|
#endif /* HAVE_SPIRV_TOOLS */
|
|
|
|
|
|
|
|
#ifdef HAVE_SPIRV_TOOLS
|
|
|
|
|
|
|
|
static void vkd3d_spirv_dump(const struct vkd3d_shader_code *spirv)
|
|
|
|
{
|
|
|
|
const static uint32_t options
|
|
|
|
= SPV_BINARY_TO_TEXT_OPTION_FRIENDLY_NAMES | SPV_BINARY_TO_TEXT_OPTION_INDENT;
|
|
|
|
spv_diagnostic diagnostic = NULL;
|
|
|
|
spv_text text = NULL;
|
|
|
|
spv_context context;
|
|
|
|
spv_result_t ret;
|
|
|
|
|
|
|
|
context = spvContextCreate(SPV_ENV_VULKAN_1_0);
|
|
|
|
|
|
|
|
if (!(ret = spvBinaryToText(context, spirv->code, spirv->size / sizeof(uint32_t),
|
|
|
|
options, &text, &diagnostic)))
|
|
|
|
{
|
|
|
|
const char *str, *current = text->str;
|
|
|
|
while ((str = strchr(current, '\n')))
|
|
|
|
{
|
|
|
|
TRACE("%.*s\n", (int)(str - current), current);
|
|
|
|
current = str + 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
TRACE("Failed to convert SPIR-V to binary text, ret %d.\n", ret);
|
|
|
|
TRACE("Diagnostic message: %s.\n", debugstr_a(diagnostic->error));
|
|
|
|
}
|
|
|
|
|
|
|
|
spvTextDestroy(text);
|
|
|
|
spvDiagnosticDestroy(diagnostic);
|
|
|
|
spvContextDestroy(context);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_validate(const struct vkd3d_shader_code *spirv)
|
|
|
|
{
|
|
|
|
spv_diagnostic diagnostic = NULL;
|
|
|
|
spv_context context;
|
|
|
|
spv_result_t ret;
|
|
|
|
|
|
|
|
context = spvContextCreate(SPV_ENV_VULKAN_1_0);
|
|
|
|
|
|
|
|
if ((ret = spvValidateBinary(context, spirv->code, spirv->size / sizeof(uint32_t),
|
|
|
|
&diagnostic)))
|
|
|
|
{
|
|
|
|
TRACE("Failed to validate SPIR-V binary, ret %d.\n", ret);
|
|
|
|
TRACE("Diagnostic message: %s.\n", debugstr_a(diagnostic->error));
|
|
|
|
}
|
|
|
|
|
|
|
|
spvDiagnosticDestroy(diagnostic);
|
|
|
|
spvContextDestroy(context);
|
|
|
|
}
|
|
|
|
|
|
|
|
#else
|
2017-06-19 09:05:53 -07:00
|
|
|
|
|
|
|
static void vkd3d_spirv_dump(const struct vkd3d_shader_code *spirv) {}
|
|
|
|
static void vkd3d_spirv_validate(const struct vkd3d_shader_code *spirv) {}
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
#endif /* HAVE_SPIRV_TOOLS */
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
struct vkd3d_spirv_stream
|
|
|
|
{
|
|
|
|
uint32_t *words;
|
|
|
|
size_t capacity;
|
|
|
|
size_t word_count;
|
|
|
|
};
|
|
|
|
|
|
|
|
static void vkd3d_spirv_stream_init(struct vkd3d_spirv_stream *stream)
|
|
|
|
{
|
|
|
|
stream->capacity = 256;
|
|
|
|
if (!(stream->words = vkd3d_calloc(stream->capacity, sizeof(*stream->words))))
|
|
|
|
stream->capacity = 0;
|
|
|
|
stream->word_count = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_stream_free(struct vkd3d_spirv_stream *stream)
|
|
|
|
{
|
|
|
|
vkd3d_free(stream->words);
|
|
|
|
}
|
|
|
|
|
|
|
|
static bool vkd3d_spirv_stream_append(struct vkd3d_spirv_stream *dst_stream,
|
|
|
|
const struct vkd3d_spirv_stream *src_stream)
|
|
|
|
{
|
|
|
|
if (!vkd3d_array_reserve((void **)&dst_stream->words, &dst_stream->capacity,
|
|
|
|
dst_stream->word_count + src_stream->word_count, sizeof(*dst_stream->words)))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
assert(dst_stream->word_count + src_stream->word_count <= dst_stream->capacity);
|
|
|
|
memcpy(&dst_stream->words[dst_stream->word_count], src_stream->words,
|
|
|
|
src_stream->word_count * sizeof(*src_stream->words));
|
|
|
|
dst_stream->word_count += src_stream->word_count;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
struct vkd3d_spirv_builder
|
|
|
|
{
|
|
|
|
uint64_t capability_mask;
|
|
|
|
SpvExecutionModel execution_model;
|
|
|
|
|
|
|
|
uint32_t current_id;
|
2017-06-19 09:05:53 -07:00
|
|
|
uint32_t main_function_id;
|
|
|
|
uint32_t type_id[VKD3D_TYPE_COUNT][VKD3D_VEC4_SIZE];
|
|
|
|
|
|
|
|
struct vkd3d_spirv_stream debug_stream; /* debug instructions */
|
|
|
|
struct vkd3d_spirv_stream annotation_stream; /* decoration instructions */
|
|
|
|
struct vkd3d_spirv_stream global_stream; /* types, constants, global variables */
|
|
|
|
struct vkd3d_spirv_stream function_stream; /* function definitions */
|
|
|
|
|
|
|
|
union
|
|
|
|
{
|
|
|
|
struct
|
|
|
|
{
|
|
|
|
uint32_t local_size[3];
|
|
|
|
} compute;
|
|
|
|
} u;
|
2017-06-19 09:05:53 -07:00
|
|
|
};
|
|
|
|
|
|
|
|
static void vkd3d_spirv_enable_capability(struct vkd3d_spirv_builder *builder,
|
|
|
|
SpvCapability cap)
|
|
|
|
{
|
|
|
|
assert(cap < sizeof(builder->capability_mask) * CHAR_BIT);
|
|
|
|
builder->capability_mask |= 1ull << cap;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_set_execution_model(struct vkd3d_spirv_builder *builder,
|
|
|
|
SpvExecutionModel model)
|
|
|
|
{
|
|
|
|
builder->execution_model = model;
|
|
|
|
|
|
|
|
switch (model)
|
|
|
|
{
|
|
|
|
case SpvExecutionModelVertex:
|
|
|
|
case SpvExecutionModelFragment:
|
|
|
|
case SpvExecutionModelGLCompute:
|
|
|
|
vkd3d_spirv_enable_capability(builder, SpvCapabilityShader);
|
|
|
|
break;
|
|
|
|
case SpvExecutionModelTessellationControl:
|
|
|
|
case SpvExecutionModelTessellationEvaluation:
|
|
|
|
vkd3d_spirv_enable_capability(builder, SpvCapabilityTessellation);
|
|
|
|
break;
|
|
|
|
case SpvExecutionModelGeometry:
|
|
|
|
vkd3d_spirv_enable_capability(builder, SpvCapabilityGeometry);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
ERR("Unhandled execution model %#x.\n", model);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-06-19 10:49:11 -07:00
|
|
|
static void vkd3d_spirv_set_local_size(struct vkd3d_spirv_builder *builder,
|
|
|
|
unsigned int x, unsigned int y, unsigned int z)
|
|
|
|
{
|
|
|
|
assert(builder->execution_model == SpvExecutionModelGLCompute);
|
|
|
|
|
|
|
|
builder->u.compute.local_size[0] = x;
|
|
|
|
builder->u.compute.local_size[1] = y;
|
|
|
|
builder->u.compute.local_size[2] = z;
|
|
|
|
}
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
static uint32_t vkd3d_spirv_opcode_word(SpvOp op, unsigned int word_count)
|
|
|
|
{
|
|
|
|
assert(!(op & ~SpvOpCodeMask));
|
|
|
|
return (word_count << SpvWordCountShift) | op;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_word(struct vkd3d_spirv_stream *stream, uint32_t word)
|
|
|
|
{
|
|
|
|
if (!vkd3d_array_reserve((void **)&stream->words, &stream->capacity,
|
|
|
|
stream->word_count + 1, sizeof(*stream->words)))
|
|
|
|
return;
|
|
|
|
|
|
|
|
stream->words[stream->word_count++] = word;
|
|
|
|
}
|
|
|
|
|
|
|
|
static unsigned int vkd3d_spirv_string_word_count(const char *str)
|
|
|
|
{
|
|
|
|
return align(strlen(str) + 1, sizeof(uint32_t)) / sizeof(uint32_t);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_string(struct vkd3d_spirv_stream *stream,
|
|
|
|
const char *str, unsigned int word_count)
|
|
|
|
{
|
|
|
|
unsigned int word_idx, i;
|
|
|
|
const char *ptr = str;
|
|
|
|
|
|
|
|
for (word_idx = 0; word_idx < word_count; ++word_idx)
|
|
|
|
{
|
|
|
|
uint32_t word = 0;
|
|
|
|
for (i = 0; i < sizeof(uint32_t) && *ptr; ++i)
|
|
|
|
word |= (uint32_t)*ptr++ << (8 * i);
|
|
|
|
vkd3d_spirv_build_word(stream, word);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* vkd3d_spirv_build_op[1-3][v]()
|
|
|
|
* vkd3d_spirv_build_op_[t][r][1-3][v]()
|
|
|
|
*
|
|
|
|
* t - result type
|
|
|
|
* r - result id
|
|
|
|
* 1-3 - the number of operands
|
|
|
|
* v - variable number of operands
|
|
|
|
*/
|
|
|
|
static void vkd3d_spirv_build_op(struct vkd3d_spirv_stream *stream, SpvOp op)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_word(stream, vkd3d_spirv_opcode_word(op, 1));
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op1(struct vkd3d_spirv_stream *stream,
|
|
|
|
SpvOp op, uint32_t operand)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_word(stream, vkd3d_spirv_opcode_word(op, 2));
|
|
|
|
vkd3d_spirv_build_word(stream, operand);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op1v(struct vkd3d_spirv_stream *stream,
|
|
|
|
SpvOp op, uint32_t operand0, const uint32_t *operands, unsigned int operand_count)
|
|
|
|
{
|
|
|
|
unsigned int i;
|
|
|
|
vkd3d_spirv_build_word(stream, vkd3d_spirv_opcode_word(op, 2 + operand_count));
|
|
|
|
vkd3d_spirv_build_word(stream, operand0);
|
|
|
|
for (i = 0; i < operand_count; ++i)
|
|
|
|
vkd3d_spirv_build_word(stream, operands[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op2v(struct vkd3d_spirv_stream *stream,
|
|
|
|
SpvOp op, uint32_t operand0, uint32_t operand1,
|
|
|
|
const uint32_t *operands, unsigned int operand_count)
|
|
|
|
{
|
|
|
|
unsigned int i;
|
|
|
|
vkd3d_spirv_build_word(stream, vkd3d_spirv_opcode_word(op, 3 + operand_count));
|
|
|
|
vkd3d_spirv_build_word(stream, operand0);
|
|
|
|
vkd3d_spirv_build_word(stream, operand1);
|
|
|
|
for (i = 0; i < operand_count; ++i)
|
|
|
|
vkd3d_spirv_build_word(stream, operands[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op2(struct vkd3d_spirv_stream *stream,
|
|
|
|
SpvOp op, uint32_t operand0, uint32_t operand1)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op2v(stream, op, operand0, operand1, NULL, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_rv(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_spirv_stream *stream, SpvOp op,
|
|
|
|
const uint32_t *operands, unsigned int operand_count)
|
|
|
|
{
|
|
|
|
uint32_t result_id = builder->current_id++;
|
|
|
|
vkd3d_spirv_build_op1v(stream, op, result_id, operands, operand_count);
|
|
|
|
return result_id;
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_r(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_spirv_stream *stream, SpvOp op)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_rv(builder, stream, op, NULL, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_r1(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_spirv_stream *stream, SpvOp op, uint32_t operand0)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_rv(builder, stream, op, &operand0, 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_r2(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_spirv_stream *stream, SpvOp op, uint32_t operand0, uint32_t operand1)
|
|
|
|
{
|
|
|
|
uint32_t operands[] = {operand0, operand1};
|
|
|
|
return vkd3d_spirv_build_op_rv(builder, stream, op, operands, ARRAY_SIZE(operands));
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_r1v(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_spirv_stream *stream, SpvOp op, uint32_t operand0,
|
|
|
|
const uint32_t *operands, unsigned int operand_count)
|
|
|
|
{
|
|
|
|
uint32_t result_id = builder->current_id++;
|
|
|
|
vkd3d_spirv_build_op2v(stream, op, result_id, operand0, operands, operand_count);
|
|
|
|
return result_id;
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_trv(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_spirv_stream *stream, SpvOp op, uint32_t result_type,
|
|
|
|
const uint32_t *operands, unsigned int operand_count)
|
|
|
|
{
|
|
|
|
uint32_t result_id = builder->current_id++;
|
|
|
|
vkd3d_spirv_build_op2v(stream, op, result_type, result_id, operands, operand_count);
|
|
|
|
return result_id;
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_tr2(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_spirv_stream *stream, SpvOp op, uint32_t result_type,
|
|
|
|
uint32_t operand0, uint32_t operand1)
|
|
|
|
{
|
|
|
|
uint32_t operands[] = {operand0, operand1};
|
|
|
|
return vkd3d_spirv_build_op_trv(builder, stream, op, result_type,
|
|
|
|
operands, ARRAY_SIZE(operands));
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op_capability(struct vkd3d_spirv_stream *stream,
|
|
|
|
SpvCapability cap)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_op1(stream, SpvOpCapability, cap);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op_memory_model(struct vkd3d_spirv_stream *stream,
|
|
|
|
SpvAddressingModel addressing_model, SpvMemoryModel memory_model)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_op2(stream, SpvOpMemoryModel, addressing_model, memory_model);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op_entry_point(struct vkd3d_spirv_stream *stream,
|
|
|
|
SpvExecutionModel model, uint32_t function_id, const char *name,
|
|
|
|
uint32_t *interface_list, unsigned int interface_size)
|
|
|
|
{
|
|
|
|
unsigned int i, name_size = vkd3d_spirv_string_word_count(name);
|
|
|
|
vkd3d_spirv_build_word(stream, vkd3d_spirv_opcode_word(SpvOpEntryPoint, 3 + name_size + interface_size));
|
|
|
|
vkd3d_spirv_build_word(stream, model);
|
|
|
|
vkd3d_spirv_build_word(stream, function_id);
|
|
|
|
vkd3d_spirv_build_string(stream, name, name_size);
|
|
|
|
for (i = 0; i < interface_size; ++i)
|
|
|
|
vkd3d_spirv_build_word(stream, interface_list[i]);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op_execution_mode(struct vkd3d_spirv_stream *stream,
|
|
|
|
uint32_t entry_point, SpvExecutionMode mode, uint32_t *literals, uint32_t literal_count)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_op2v(stream, SpvOpExecutionMode, entry_point, mode, literals, literal_count);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op_name(struct vkd3d_spirv_builder *builder,
|
|
|
|
uint32_t id, const char *name)
|
|
|
|
{
|
|
|
|
unsigned int name_size = vkd3d_spirv_string_word_count(name);
|
|
|
|
struct vkd3d_spirv_stream *stream = &builder->debug_stream;
|
|
|
|
vkd3d_spirv_build_word(stream, vkd3d_spirv_opcode_word(SpvOpName, 2 + name_size));
|
|
|
|
vkd3d_spirv_build_word(stream, id);
|
|
|
|
vkd3d_spirv_build_string(stream, name, name_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_type_void(struct vkd3d_spirv_builder *builder)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_r(builder, &builder->global_stream, SpvOpTypeVoid);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_type_float(struct vkd3d_spirv_builder *builder,
|
|
|
|
uint32_t width)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_r1(builder, &builder->global_stream, SpvOpTypeFloat, width);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_type_int(struct vkd3d_spirv_builder *builder,
|
|
|
|
uint32_t width, uint32_t signedness)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_r2(builder, &builder->global_stream, SpvOpTypeInt, width, signedness);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_type_vector(struct vkd3d_spirv_builder *builder,
|
|
|
|
uint32_t component_type, uint32_t component_count)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_r2(builder, &builder->global_stream,
|
|
|
|
SpvOpTypeVector, component_type, component_count);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_type_function(struct vkd3d_spirv_builder *builder,
|
|
|
|
uint32_t return_type, uint32_t *param_types, unsigned int param_count)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_r1v(builder, &builder->global_stream,
|
|
|
|
SpvOpTypeFunction, return_type, param_types, param_count);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_function(struct vkd3d_spirv_builder *builder,
|
|
|
|
uint32_t result_type, uint32_t function_control, uint32_t function_type)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_tr2(builder, &builder->function_stream,
|
|
|
|
SpvOpFunction, result_type, function_control, function_type);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op_function_end(struct vkd3d_spirv_builder *builder)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_op(&builder->function_stream, SpvOpFunctionEnd);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_build_op_label(struct vkd3d_spirv_builder *builder)
|
|
|
|
{
|
|
|
|
return vkd3d_spirv_build_op_r(builder, &builder->function_stream, SpvOpLabel);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_op_return(struct vkd3d_spirv_builder *builder)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_op(&builder->function_stream, SpvOpReturn);
|
|
|
|
}
|
|
|
|
|
|
|
|
static uint32_t vkd3d_spirv_get_type_id(struct vkd3d_spirv_builder *builder,
|
|
|
|
enum vkd3d_component_type component_type, unsigned int component_count)
|
|
|
|
{
|
|
|
|
uint32_t id, scalar_id;
|
|
|
|
|
|
|
|
assert(0 < component_count && component_count <= VKD3D_VEC4_SIZE);
|
|
|
|
|
|
|
|
if (!(id = builder->type_id[component_type][component_count - 1]))
|
|
|
|
{
|
|
|
|
if (component_count == 1)
|
|
|
|
{
|
|
|
|
switch (component_type)
|
|
|
|
{
|
|
|
|
case VKD3D_TYPE_VOID:
|
|
|
|
id = vkd3d_spirv_build_op_type_void(builder);
|
|
|
|
break;
|
|
|
|
case VKD3D_TYPE_FLOAT:
|
|
|
|
id = vkd3d_spirv_build_op_type_float(builder, 32);
|
|
|
|
break;
|
|
|
|
case VKD3D_TYPE_INT:
|
|
|
|
case VKD3D_TYPE_UINT:
|
|
|
|
id = vkd3d_spirv_build_op_type_int(builder, 32, component_type == VKD3D_TYPE_INT);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
FIXME("Unhandled component type %#x.\n", component_type);
|
|
|
|
id = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
assert(component_type != VKD3D_TYPE_VOID);
|
|
|
|
scalar_id = vkd3d_spirv_get_type_id(builder, component_type, 1);
|
|
|
|
id = vkd3d_spirv_build_op_type_vector(builder, scalar_id, component_count);
|
|
|
|
}
|
|
|
|
|
|
|
|
builder->type_id[component_type][component_count - 1] = id;
|
|
|
|
}
|
|
|
|
|
|
|
|
return id;
|
|
|
|
}
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
static void vkd3d_spirv_builder_init(struct vkd3d_spirv_builder *builder)
|
|
|
|
{
|
2017-06-19 09:05:53 -07:00
|
|
|
uint32_t void_id, function_type_id;
|
|
|
|
|
|
|
|
vkd3d_spirv_stream_init(&builder->debug_stream);
|
|
|
|
vkd3d_spirv_stream_init(&builder->annotation_stream);
|
|
|
|
vkd3d_spirv_stream_init(&builder->global_stream);
|
|
|
|
vkd3d_spirv_stream_init(&builder->function_stream);
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
builder->current_id = 1;
|
2017-06-19 09:05:53 -07:00
|
|
|
|
|
|
|
void_id = vkd3d_spirv_get_type_id(builder, VKD3D_TYPE_VOID, 1);
|
|
|
|
function_type_id = vkd3d_spirv_build_op_type_function(builder, void_id, NULL, 0);
|
|
|
|
|
|
|
|
builder->main_function_id = vkd3d_spirv_build_op_function(builder, void_id,
|
|
|
|
SpvFunctionControlMaskNone, function_type_id);
|
|
|
|
vkd3d_spirv_build_op_name(builder, builder->main_function_id, "main");
|
|
|
|
vkd3d_spirv_build_op_label(builder);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_builder_free(struct vkd3d_spirv_builder *builder)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_stream_free(&builder->debug_stream);
|
|
|
|
vkd3d_spirv_stream_free(&builder->annotation_stream);
|
|
|
|
vkd3d_spirv_stream_free(&builder->global_stream);
|
|
|
|
vkd3d_spirv_stream_free(&builder->function_stream);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void vkd3d_spirv_build_execution_mode_declarations(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_spirv_stream *stream)
|
|
|
|
{
|
|
|
|
if (builder->execution_model == SpvExecutionModelGLCompute)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_op_execution_mode(stream, builder->main_function_id, SpvExecutionModeLocalSize,
|
|
|
|
builder->u.compute.local_size, ARRAY_SIZE(builder->u.compute.local_size));
|
|
|
|
}
|
2017-06-19 09:05:53 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
static bool vkd3d_spirv_compile_module(struct vkd3d_spirv_builder *builder,
|
|
|
|
struct vkd3d_shader_code *spirv)
|
|
|
|
{
|
2017-06-19 09:05:53 -07:00
|
|
|
uint64_t capability_mask = builder->capability_mask;
|
|
|
|
struct vkd3d_spirv_stream stream;
|
|
|
|
uint32_t *code;
|
|
|
|
unsigned int i;
|
|
|
|
size_t size;
|
|
|
|
|
|
|
|
vkd3d_spirv_build_op_function_end(builder);
|
|
|
|
|
|
|
|
vkd3d_spirv_stream_init(&stream);
|
|
|
|
|
|
|
|
vkd3d_spirv_build_word(&stream, SpvMagicNumber);
|
|
|
|
vkd3d_spirv_build_word(&stream, SpvVersion);
|
|
|
|
vkd3d_spirv_build_word(&stream, 0); /* generator */
|
|
|
|
vkd3d_spirv_build_word(&stream, builder->current_id); /* bound */
|
|
|
|
vkd3d_spirv_build_word(&stream, 0); /* schema, reserved */
|
|
|
|
|
|
|
|
for (i = 0; capability_mask; ++i)
|
|
|
|
{
|
|
|
|
if (capability_mask & 1)
|
|
|
|
vkd3d_spirv_build_op_capability(&stream, i);
|
|
|
|
capability_mask >>= 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
vkd3d_spirv_build_op_memory_model(&stream, SpvAddressingModelLogical, SpvMemoryModelGLSL450);
|
|
|
|
vkd3d_spirv_build_op_entry_point(&stream, builder->execution_model, builder->main_function_id,
|
|
|
|
"main", NULL, 0);
|
|
|
|
|
|
|
|
vkd3d_spirv_build_execution_mode_declarations(builder, &stream);
|
|
|
|
|
|
|
|
vkd3d_spirv_stream_append(&stream, &builder->debug_stream);
|
|
|
|
vkd3d_spirv_stream_append(&stream, &builder->annotation_stream);
|
|
|
|
vkd3d_spirv_stream_append(&stream, &builder->global_stream);
|
|
|
|
vkd3d_spirv_stream_append(&stream, &builder->function_stream);
|
|
|
|
|
|
|
|
if (!(code = vkd3d_calloc(stream.word_count, sizeof(*code))))
|
|
|
|
{
|
|
|
|
vkd3d_spirv_stream_free(&stream);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
size = stream.word_count * sizeof(*code);
|
|
|
|
memcpy(code, stream.words, size);
|
|
|
|
vkd3d_spirv_stream_free(&stream);
|
|
|
|
|
|
|
|
spirv->code = code;
|
|
|
|
spirv->size = size;
|
2017-06-19 09:05:53 -07:00
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
return true;
|
2017-06-19 09:05:53 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
struct vkd3d_dxbc_compiler
|
|
|
|
{
|
|
|
|
struct vkd3d_spirv_builder spirv_builder;
|
|
|
|
|
|
|
|
uint32_t options;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct vkd3d_dxbc_compiler *vkd3d_dxbc_compiler_create(const struct vkd3d_shader_version *shader_version,
|
|
|
|
uint32_t compiler_options)
|
|
|
|
{
|
|
|
|
struct vkd3d_dxbc_compiler *compiler;
|
|
|
|
|
|
|
|
if (!(compiler = vkd3d_malloc(sizeof(*compiler))))
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
memset(compiler, 0, sizeof(*compiler));
|
|
|
|
vkd3d_spirv_builder_init(&compiler->spirv_builder);
|
|
|
|
compiler->options = compiler_options;
|
|
|
|
|
|
|
|
switch (shader_version->type)
|
|
|
|
{
|
|
|
|
case VKD3D_SHADER_TYPE_VERTEX:
|
|
|
|
vkd3d_spirv_set_execution_model(&compiler->spirv_builder, SpvExecutionModelVertex);
|
|
|
|
break;
|
|
|
|
case VKD3D_SHADER_TYPE_HULL:
|
|
|
|
vkd3d_spirv_set_execution_model(&compiler->spirv_builder, SpvExecutionModelTessellationControl);
|
|
|
|
break;
|
|
|
|
case VKD3D_SHADER_TYPE_DOMAIN:
|
|
|
|
vkd3d_spirv_set_execution_model(&compiler->spirv_builder, SpvExecutionModelTessellationEvaluation);
|
|
|
|
break;
|
|
|
|
case VKD3D_SHADER_TYPE_GEOMETRY:
|
|
|
|
vkd3d_spirv_set_execution_model(&compiler->spirv_builder, SpvExecutionModelGeometry);
|
|
|
|
break;
|
|
|
|
case VKD3D_SHADER_TYPE_PIXEL:
|
|
|
|
vkd3d_spirv_set_execution_model(&compiler->spirv_builder, SpvExecutionModelFragment);
|
|
|
|
break;
|
|
|
|
case VKD3D_SHADER_TYPE_COMPUTE:
|
|
|
|
vkd3d_spirv_set_execution_model(&compiler->spirv_builder, SpvExecutionModelGLCompute);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
ERR("Invalid shader type %#x.\n", shader_version->type);
|
|
|
|
}
|
|
|
|
|
|
|
|
return compiler;
|
|
|
|
}
|
|
|
|
|
2017-06-19 10:49:11 -07:00
|
|
|
static void vkd3d_dxbc_compiler_emit_dcl_thread_group(struct vkd3d_dxbc_compiler *compiler,
|
|
|
|
const struct vkd3d_shader_instruction *instruction)
|
|
|
|
{
|
|
|
|
const struct vkd3d_shader_thread_group_size *group_size = &instruction->declaration.thread_group_size;
|
|
|
|
struct vkd3d_spirv_builder *builder = &compiler->spirv_builder;
|
|
|
|
|
|
|
|
vkd3d_spirv_set_local_size(builder, group_size->x, group_size->y, group_size->z);
|
|
|
|
}
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
static void vkd3d_dxbc_compiler_emit_return(struct vkd3d_dxbc_compiler *compiler,
|
|
|
|
const struct vkd3d_shader_instruction *instruction)
|
|
|
|
{
|
|
|
|
vkd3d_spirv_build_op_return(&compiler->spirv_builder);
|
|
|
|
}
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
void vkd3d_dxbc_compiler_handle_instruction(struct vkd3d_dxbc_compiler *compiler,
|
|
|
|
const struct vkd3d_shader_instruction *instruction)
|
|
|
|
{
|
2017-06-19 09:05:53 -07:00
|
|
|
switch (instruction->handler_idx)
|
|
|
|
{
|
2017-06-19 10:49:11 -07:00
|
|
|
case VKD3DSIH_DCL_THREAD_GROUP:
|
|
|
|
vkd3d_dxbc_compiler_emit_dcl_thread_group(compiler, instruction);
|
|
|
|
break;
|
2017-06-19 09:05:53 -07:00
|
|
|
case VKD3DSIH_RET:
|
|
|
|
vkd3d_dxbc_compiler_emit_return(compiler, instruction);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
FIXME("Unhandled instruction %#x.\n", instruction->handler_idx);
|
|
|
|
}
|
2017-06-19 09:05:53 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
bool vkd3d_dxbc_compiler_generate_spirv(struct vkd3d_dxbc_compiler *compiler,
|
|
|
|
struct vkd3d_shader_code *spirv)
|
|
|
|
{
|
|
|
|
if (!vkd3d_spirv_compile_module(&compiler->spirv_builder, spirv))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if (TRACE_ON())
|
|
|
|
{
|
|
|
|
vkd3d_spirv_dump(spirv);
|
|
|
|
vkd3d_spirv_validate(spirv);
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
void vkd3d_dxbc_compiler_destroy(struct vkd3d_dxbc_compiler *compiler)
|
|
|
|
{
|
2017-06-19 09:05:53 -07:00
|
|
|
vkd3d_spirv_builder_free(&compiler->spirv_builder);
|
|
|
|
|
2017-06-19 09:05:53 -07:00
|
|
|
vkd3d_free(compiler);
|
|
|
|
}
|