[PATCH 0/1] MR92: vkd3d-shader/hlsl: Write SM4 break instructions.

[PATCH 1/1] vkd3d-shader/hlsl: Write SM4 break instructions.

[Giovanni Mascellani]

From: Giovanni Mascellani

---
 libs/vkd3d-shader/hlsl.c          | 15 +++++++++++++++
 libs/vkd3d-shader/hlsl.h          |  2 ++
 libs/vkd3d-shader/hlsl_sm4.c      | 23 +++++++++++++++++++++++
 tests/function-return.shader_test | 16 ++++++++--------
 tests/return.shader_test          | 30 +++++++++++++++---------------
 5 files changed, 63 insertions(+), 23 deletions(-)

diff --git a/libs/vkd3d-shader/hlsl.c b/libs/vkd3d-shader/hlsl.c
index 97713343..aa6aa701 100644
--- a/libs/vkd3d-shader/hlsl.c
+++ b/libs/vkd3d-shader/hlsl.c
@@ -1861,6 +1861,21 @@ const char *hlsl_node_type_to_string(enum hlsl_ir_node_type type)
     return names[type];
 }

+const char *hlsl_jump_type_to_string(enum hlsl_ir_jump_type type)
+{
+    static const char * const names[] =
+    {
+        "HLSL_IR_JUMP_BREAK",
+        "HLSL_IR_JUMP_CONTINUE",
+        "HLSL_IR_JUMP_DISCARD",
+        "HLSL_IR_JUMP_RETURN",
+    };
+
+    if (type >= ARRAY_SIZE(names))
+        return "Unexpected jump type";
+    return names[type];
+}
+
 static void dump_instr(struct hlsl_ctx *ctx, struct vkd3d_string_buffer *buffer, const struct hlsl_ir_node *instr);

static void dump_instr_list(struct hlsl_ctx *ctx, struct vkd3d_string_buffer *buffer, const struct list *list)
diff --git a/libs/vkd3d-shader/hlsl.h b/libs/vkd3d-shader/hlsl.h
index 53b47e40..81b1a61d 100644
--- a/libs/vkd3d-shader/hlsl.h
+++ b/libs/vkd3d-shader/hlsl.h
@@ -1003,6 +1003,8 @@ struct hlsl_ir_var *hlsl_get_var(struct hlsl_scope *scope, const char *name);
 struct hlsl_type *hlsl_get_element_type_from_path_index(struct hlsl_ctx *ctx, const struct hlsl_type *type,
         struct hlsl_ir_node *idx);

+const char *hlsl_jump_type_to_string(enum hlsl_ir_jump_type type);
+
 struct hlsl_type *hlsl_new_array_type(struct hlsl_ctx *ctx, struct hlsl_type *basic_type, unsigned int array_size);
 struct hlsl_ir_node *hlsl_new_binary_expr(struct hlsl_ctx *ctx, enum hlsl_ir_expr_op op, struct hlsl_ir_node *arg1,
         struct hlsl_ir_node *arg2);
diff --git a/libs/vkd3d-shader/hlsl_sm4.c b/libs/vkd3d-shader/hlsl_sm4.c
index 2e7c101f..691a0468 100644
--- a/libs/vkd3d-shader/hlsl_sm4.c
+++ b/libs/vkd3d-shader/hlsl_sm4.c
@@ -2118,6 +2118,25 @@ static void write_sm4_if(struct hlsl_ctx *ctx, struct vkd3d_bytecode_buffer *buf
     write_sm4_instruction(buffer, &instr);
 }

+static void write_sm4_jump(struct hlsl_ctx *ctx,
+        struct vkd3d_bytecode_buffer *buffer, const struct hlsl_ir_jump *jump)
+{
+    struct sm4_instruction instr = {0};
+
+    switch (jump->type)
+    {
+        case HLSL_IR_JUMP_BREAK:
+            instr.opcode = VKD3D_SM4_OP_BREAK;
+            break;
+
+        default:
+            hlsl_fixme(ctx, &jump->node.loc, "Jump type %s.\n", hlsl_jump_type_to_string(jump->type));
+            return;
+    }
+
+    write_sm4_instruction(buffer, &instr);
+}
+
 static void write_sm4_load(struct hlsl_ctx *ctx,
         struct vkd3d_bytecode_buffer *buffer, const struct hlsl_ir_load *load)
 {
@@ -2375,6 +2394,10 @@ static void write_sm4_block(struct hlsl_ctx *ctx, struct vkd3d_bytecode_buffer *
                 write_sm4_if(ctx, buffer, hlsl_ir_if(instr));
                 break;

+            case HLSL_IR_JUMP:
+                write_sm4_jump(ctx, buffer, hlsl_ir_jump(instr));
+                break;
+
             case HLSL_IR_LOAD:
                 write_sm4_load(ctx, buffer, hlsl_ir_load(instr));
                 break;
diff --git a/tests/function-return.shader_test b/tests/function-return.shader_test
index 7d411998..cbd29749 100644
--- a/tests/function-return.shader_test
+++ b/tests/function-return.shader_test
@@ -145,7 +145,7 @@ uniform 0 float 0.9
 draw quad
 probe all rgba (1.0, 0.9, 1.0, 0.6) 1

-[pixel shader todo]
+[pixel shader]

float func(out float o)
 {
@@ -184,10 +184,10 @@ float4 main() : sv_target
 }

[test]
-todo draw quad
+draw quad
 probe all rgba (0.4, 0.3, 0.3, 0.9) 1

-[pixel shader todo]
+[pixel shader]

uniform float f;

@@ -239,23 +239,23 @@ float4 main() : sv_target

[test]
 uniform 0 float 0.0
-todo draw quad
+draw quad
 probe all rgba (0.3, 0.2, 0.3, 0.3) 1

uniform 0 float 0.1
-todo draw quad
+draw quad
 probe all rgba (0.3, 0.3, 0.3, 0.3) 1

uniform 0 float 0.3
-todo draw quad
+draw quad
 probe all rgba (0.3, 0.5, 0.3, 0.3) 1

uniform 0 float 0.7
-todo draw quad
+draw quad
 probe all rgba (0.3, 0.9, 0.7, 0.6) 1

uniform 0 float 0.9
-todo draw quad
+draw quad
 probe all rgba (0.4, 0.1, 0.7, 0.6) 1

[pixel shader todo]
diff --git a/tests/return.shader_test b/tests/return.shader_test
index 2660c6dd..3c9ea611 100644
--- a/tests/return.shader_test
+++ b/tests/return.shader_test
@@ -128,7 +128,7 @@ uniform 0 float 0.9
 draw quad
 probe all rgba (0.4, 0.5, 0.6, 0.7) 1

-[pixel shader todo]
+[pixel shader]

void main(out float4 ret : sv_target)
 {
@@ -143,10 +143,10 @@ void main(out float4 ret : sv_target)
 }

[test]
-todo draw quad
-todo probe all rgba (0.2, 0.4, 0.6, 0.8)
+draw quad
+probe all rgba (0.2, 0.4, 0.6, 0.8)

-[pixel shader todo]
+[pixel shader]

uniform float f;

@@ -166,26 +166,26 @@ void main(out float4 ret : sv_target)

[test]
 uniform 0 float 0.0
-todo draw quad
-todo probe all rgba (0.1, 0.1, 0.1, 0.1) 1
+draw quad
+probe all rgba (0.1, 0.1, 0.1, 0.1) 1

uniform 0 float 0.1
-todo draw quad
-todo probe all rgba (0.2, 0.2, 0.2, 0.2) 1
+draw quad
+probe all rgba (0.2, 0.2, 0.2, 0.2) 1

uniform 0 float 0.3
-todo draw quad
+draw quad
 todo probe all rgba (0.4, 0.4, 0.4, 0.4) 1

uniform 0 float 0.7
-todo draw quad
+draw quad
 todo probe all rgba (0.8, 0.8, 0.8, 0.8) 1

uniform 0 float 0.9
-todo draw quad
+draw quad
 todo probe all rgba (0.9, 0.9, 0.9, 0.9) 1

-[pixel shader todo]
+[pixel shader]

uniform float f;

@@ -211,10 +211,10 @@ void main(out float4 ret : sv_target)

[test]
 uniform 0 float 0.2
-todo draw quad
-todo probe all rgba (0.2, 0.2, 0.2, 0.2)
+draw quad
+probe all rgba (0.2, 0.2, 0.2, 0.2)
 uniform 0 float 0.8
-todo draw quad
+draw quad
 probe all rgba (0.5, 0.5, 0.5, 0.5)

[pixel shader todo]

[Zebediah Figura (＠zfigura)]

Zebediah Figura (@zfigura) commented about libs/vkd3d-shader/hlsl.c:
>  return names[type];
>
> }
>
> +const char *hlsl_jump_type_to_string(enum hlsl_ir_jump_type type)
> +{
> - static const char * const names[] =
> - {
> -
>    "HLSL_IR_JUMP_BREAK",
> -
>    "HLSL_IR_JUMP_CONTINUE",
> -
>    "HLSL_IR_JUMP_DISCARD",
> -
>    "HLSL_IR_JUMP_RETURN",
> - };
> -
> - if (type >= ARRAY_SIZE(names))
> -
>    return "Unexpected jump type";

I'd just make this an assert honestly?

[Zebediah Figura (＠zfigura)]

Zebediah Figura (@zfigura) commented about libs/vkd3d-shader/hlsl_sm4.c:
> +static void write_sm4_jump(struct hlsl_ctx *ctx,
> -
>    struct vkd3d_bytecode_buffer *buffer, const struct hlsl_ir_jump *jump)
>
> +{
> - struct sm4_instruction instr = {0};
> -
> - switch (jump->type)
> - {
> -
>    case HLSL_IR_JUMP_BREAK:
> -
>        instr.opcode = VKD3D_SM4_OP_BREAK;
> -
>        break;
> -
> -
>    default:
> -
>        hlsl_fixme(ctx, &jump->node.loc, "Jump type %s.\n", hlsl_jump_type_to_string(jump->type));
> -
>        return;
> - }

While we're at it I might add an unreachable case for RETURN. Doesn't need to block this patch though.