From 3efe3bab1c8290bd49a957ebec0ad97e58a35fcf Mon Sep 17 00:00:00 2001
From: Mikko Rasa <tdb@tdb.fi>
Date: Fri, 11 Mar 2022 11:22:23 +0200
Subject: [PATCH] Create specialized versions of SPIR-V modules with default
 spec values

It's important to remove unused resources in this case too.
---
 source/core/module.cpp  | 2 ++
 source/core/module.h    | 2 ++
 source/core/program.cpp | 2 +-
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/source/core/module.cpp b/source/core/module.cpp
index 636fd13c..11d11af5 100644
--- a/source/core/module.cpp
+++ b/source/core/module.cpp
@@ -237,6 +237,8 @@ void SpirVModule::reflect()
 	{
 		auto i = spec_indices.find(b.condition);
 		b.condition = (i!=spec_indices.end() ? &spec_constants[i->second] : 0);
+		if(b.condition)
+			specializable = true;
 
 		for(const Variable *&v: b.accessed_variables)
 		{
diff --git a/source/core/module.h b/source/core/module.h
index 2a4cac6b..51924d60 100644
--- a/source/core/module.h
+++ b/source/core/module.h
@@ -253,6 +253,7 @@ private:
 	std::vector<Variable> variables;
 	std::vector<Constant> spec_constants;
 	std::vector<InstructionBlock> blocks;
+	bool specializable = false;
 
 public:
 	virtual Format get_format() const { return SPIR_V; }
@@ -269,6 +270,7 @@ public:
 	const std::vector<Variable> &get_variables() const { return variables; }
 	const std::vector<Constant> &get_spec_constants() const { return spec_constants; }
 	const std::vector<InstructionBlock> &get_blocks() const { return blocks; }
+	bool is_specializable() const { return specializable; }
 
 	/** Creates a new module which is a specialized version of this one. */
 	SpirVModule *specialize(const std::map<std::string, int> &) const;
diff --git a/source/core/program.cpp b/source/core/program.cpp
index 88187221..e4250b6c 100644
--- a/source/core/program.cpp
+++ b/source/core/program.cpp
@@ -39,7 +39,7 @@ void Program::add_stages(const Module &mod, const map<string, int> &spec_values)
 		add_glsl_stages(static_cast<const GlslModule &>(mod), spec_values);
 		break;
 	case Module::SPIR_V:
-		if(!spec_values.empty())
+		if(static_cast<const SpirVModule &>(mod).is_specializable())
 		{
 			specialized_spirv = static_cast<const SpirVModule &>(mod).specialize(spec_values);
 			final_module = specialized_spirv;
-- 
2.43.0