From 379b305b4bc09799d53981fa6e5d9cbe6be99561 Mon Sep 17 00:00:00 2001
From: ReinUsesLisp <reinuseslisp@airmail.cc>
Date: Tue, 25 May 2021 02:34:08 -0300
Subject: [PATCH] glasm: Throw when there are register leaks

---
 src/shader_recompiler/backend/glasm/emit_glasm.cpp | 3 +++
 src/shader_recompiler/backend/glasm/reg_alloc.h    | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/src/shader_recompiler/backend/glasm/emit_glasm.cpp b/src/shader_recompiler/backend/glasm/emit_glasm.cpp
index 4aa3682c25..0e9dc06a60 100644
--- a/src/shader_recompiler/backend/glasm/emit_glasm.cpp
+++ b/src/shader_recompiler/backend/glasm/emit_glasm.cpp
@@ -271,6 +271,9 @@ void EmitCode(EmitContext& ctx, const IR::Program& program) {
             break;
         }
     }
+    if (!ctx.reg_alloc.IsEmpty()) {
+        throw LogicError("Register allocator is not empty");
+    }
 }
 
 void SetupOptions(const IR::Program& program, const Profile& profile,
diff --git a/src/shader_recompiler/backend/glasm/reg_alloc.h b/src/shader_recompiler/backend/glasm/reg_alloc.h
index b97c84146d..019e1bc0fc 100644
--- a/src/shader_recompiler/backend/glasm/reg_alloc.h
+++ b/src/shader_recompiler/backend/glasm/reg_alloc.h
@@ -128,6 +128,10 @@ public:
         return num_used_long_registers;
     }
 
+    [[nodiscard]] bool IsEmpty() const noexcept {
+        return register_use.none() && long_register_use.none();
+    }
+
     /// Returns true if the instruction is expected to be aliased to another
     static bool IsAliased(const IR::Inst& inst);