shader: Unify shader stage types

1 files changed, 9 insertions, 2 deletions

diff --git a/src/shader_recompiler/stage.h b/src/shader_recompiler/stage.h
index 7d4f2c0bb..5c1c8d8fc 100644
--- a/src/shader_recompiler/stage.h
+++ b/src/shader_recompiler/stage.h
@@ -9,13 +9,20 @@
 namespace Shader {
 
 enum class Stage : u32 {
-    Compute,
-    VertexA,
     VertexB,
     TessellationControl,
     TessellationEval,
     Geometry,
     Fragment,
+
+    Compute,
+
+    VertexA,
 };
+constexpr u32 MaxStageTypes = 6;
+
+[[nodiscard]] constexpr Stage StageFromIndex(size_t index) noexcept {
+    return static_cast<Stage>(static_cast<size_t>(Stage::VertexB) + index);
+}
 
 } // namespace Shader