X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fcore%2Fmodule.h;h=2a4cac6b06758faa45910ffa48c463b271ee07eb;hb=11a5d4f2a35dfdcf61e16d4fcfba22e20ea6a3db;hp=15c07fb0366bd79b0e2ee737f4c21e42b3ed58f8;hpb=fbfd4b52102e06398ff162be07d6799f005a04a8;p=libs%2Fgl.git

diff --git a/source/core/module.h b/source/core/module.h
index 15c07fb0..2a4cac6b 100644
--- a/source/core/module.h
+++ b/source/core/module.h
@@ -94,6 +94,8 @@ forming the module's interface.
 */
 class SpirVModule: public Module, public SpirVModuleBackend
 {
+	friend SpirVModuleBackend;
+
 public:
 	enum Stage
 	{
@@ -127,6 +129,7 @@ public:
 	struct EntryPoint
 	{
 		std::string name;
+		unsigned id = 0;
 		Stage stage = VERTEX;
 		std::vector<const Variable *> globals;
 	};
@@ -146,16 +149,18 @@ public:
 	struct Structure
 	{
 		std::string name;
-		std::vector<StructMember> members;
+		unsigned id = 0;
 		unsigned size = 0;
+		std::vector<StructMember> members;
 	};
 
 	struct Variable
 	{
 		std::string name;
-		StorageClass storage = static_cast<StorageClass>(-1);
+		unsigned id = 0;
 		DataType type = VOID;
 		const Structure *struct_type = 0;
+		StorageClass storage = static_cast<StorageClass>(-1);
 		unsigned array_size = 0;
 		int location = -1;
 		unsigned descriptor_set = 0;
@@ -168,6 +173,7 @@ public:
 	struct Constant
 	{
 		std::string name;
+		unsigned id = 0;
 		int constant_id = -1;
 		DataType type = VOID;
 		union
@@ -179,8 +185,9 @@ public:
 
 	struct InstructionBlock
 	{
-		const Constant *condition = 0;
+		unsigned id = 0;
 		bool negate_condition = false;
+		const Constant *condition = 0;
 		std::vector<const Variable *> accessed_variables;
 		std::vector<const InstructionBlock *> successors;
 	};
@@ -262,6 +269,13 @@ 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; }
+
+	/** Creates a new module which is a specialized version of this one. */
+	SpirVModule *specialize(const std::map<std::string, int> &) const;
+
+private:
+	std::vector<const InstructionBlock *> collect_visited_blocks(const std::map<unsigned, int> &) const;
+	void collect_visited_blocks(unsigned, std::vector<std::uint8_t> &) const;
 };
 
 } // namespace GL