[PATCH 0/1] MR546: vkd3d-shader/spirv: Enable capability ShaderViewportIndexLayerEXT when needed.

[Giovanni Mascellani (＠giomasce)]

Capability Geometry allows to use the Layer builtin in geometry shaders.
For vertex shaders ShaderLayer should be used, but it's only available
starting from SPIR-V 1.5. ShaderViewportIndexLayerEXT can be used instead
with extension SPV_EXT_shader_viewport_index_layer.

[PATCH 1/1] vkd3d-shader/spirv: Enable capability ShaderViewportIndexLayerEXT when needed.

[Giovanni Mascellani]

From: Giovanni Mascellani

Capability Geometry allows to use the Layer builtin in geometry shaders.
For vertex shaders ShaderLayer should be used, but it's only available
starting from SPIR-V 1.5. ShaderViewportIndexLayerEXT can be used instead
with extension SPV_EXT_shader_viewport_index_layer.
---
 libs/vkd3d-shader/spirv.c | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/libs/vkd3d-shader/spirv.c b/libs/vkd3d-shader/spirv.c
index 8a4704e56..42ee42475 100644
--- a/libs/vkd3d-shader/spirv.c
+++ b/libs/vkd3d-shader/spirv.c
@@ -1949,6 +1949,8 @@ static bool vkd3d_spirv_compile_module(struct vkd3d_spirv_builder *builder,
         vkd3d_spirv_build_op_extension(&stream, "SPV_EXT_descriptor_indexing");
     if (vkd3d_spirv_capability_is_enabled(builder, SpvCapabilityStencilExportEXT))
         vkd3d_spirv_build_op_extension(&stream, "SPV_EXT_shader_stencil_export");
+    if (vkd3d_spirv_capability_is_enabled(builder, SpvCapabilityShaderViewportIndexLayerEXT))
+        vkd3d_spirv_build_op_extension(&stream, "SPV_EXT_shader_viewport_index_layer");

if (builder->ext_instr_set_glsl_450)
         vkd3d_spirv_build_op_ext_inst_import(&stream, builder->ext_instr_set_glsl_450, "GLSL.std.450");
@@ -4284,7 +4286,20 @@ static void spirv_compiler_decorate_builtin(struct spirv_compiler *compiler,
             spirv_compiler_emit_execution_mode(compiler, SpvExecutionModeDepthReplacing, NULL, 0);
             break;
         case SpvBuiltInLayer:
-            vkd3d_spirv_enable_capability(builder, SpvCapabilityGeometry);
+            switch (compiler->shader_type)
+            {
+                case VKD3D_SHADER_TYPE_GEOMETRY:
+                    vkd3d_spirv_enable_capability(builder, SpvCapabilityGeometry);
+                    break;
+
+                case VKD3D_SHADER_TYPE_VERTEX:
+                    vkd3d_spirv_enable_capability(builder, SpvCapabilityShaderViewportIndexLayerEXT);
+                    break;
+
+                default:
+                    FIXME("Unhandled use of a Layer builtin in a shader of type %u.\n", compiler->shader_type);
+                    break;
+            }
             break;
         case SpvBuiltInViewportIndex:
             vkd3d_spirv_enable_capability(builder, SpvCapabilityMultiViewport);