report the shader stage when compiling

Signed-off-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>

1 files changed, 15 insertions, 1 deletions

diff --git a/run.c b/run.c
index 3448895..1d641a0 100644
--- a/run.c
+++ b/run.c
@@ -182,11 +182,25 @@ fail:
     return ret;
 }
 
+static const char *
+get_shader_stage_name(VkShaderStageFlagBits stage)
+{
+    switch (stage) {
+    case VK_SHADER_STAGE_VERTEX_BIT: return "VS";
+    case VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT: return "TCS";
+    case VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT: return "TES";
+    case VK_SHADER_STAGE_GEOMETRY_BIT: return "GS";
+    case VK_SHADER_STAGE_FRAGMENT_BIT: return "FS";
+    case VK_SHADER_STAGE_COMPUTE_BIT: return "CS";
+    default: return NULL;
+    }
+}
+
 static void
 print_shader_stats(const char *pipeline_name, VkShaderStageFlagBits stage,
                    const struct shader_stats *stats)
 {
-    printf("%s - ", pipeline_name);
+    printf("%s (%s) - ", pipeline_name, get_shader_stage_name(stage));
     printf("Shader Stats: ");
     printf("SGPRS: %d ", stats->num_sgprs);
     printf("VGPRS: %d ", stats->num_vgprs);