From 395bed3a0af90a53be44e81eadd06f4931c8e933 Mon Sep 17 00:00:00 2001
From: ReinUsesLisp <reinuseslisp@airmail.cc>
Date: Wed, 23 Jun 2021 02:41:00 -0300
Subject: shader: Unify shader stage types

---
 src/shader_recompiler/stage.h | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/shader_recompiler')

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
-- 
cgit v1.2.3