X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fcore%2Fmodule.h;h=480a0669ffbb6275324a7ab2b3c5bb660fdba2eb;hb=560b11de262a6cae2c71d744598bf3fd435ab14f;hp=855d51957e8f9adb30a6e903d333186ebe5dbb25;hpb=99ca354f18119f82f1adeca100cd665a8f640317;p=libs%2Fgl.git

diff --git a/source/core/module.h b/source/core/module.h
index 855d5195..480a0669 100644
--- a/source/core/module.h
+++ b/source/core/module.h
@@ -61,6 +61,9 @@ private:
 	virtual void compile(SL::Compiler &) = 0;
 
 	SL::Features create_features() const;
+
+public:
+	virtual void set_debug_name(const std::string &) { }
 };
 
 /**
@@ -101,7 +104,8 @@ public:
 	{
 		VERTEX = 0,
 		GEOMETRY = 3,
-		FRAGMENT = 4
+		FRAGMENT = 4,
+		COMPUTE = 5
 	};
 
 	enum StorageClass
@@ -129,8 +133,10 @@ public:
 	struct EntryPoint
 	{
 		std::string name;
+		unsigned id = 0;
 		Stage stage = VERTEX;
 		std::vector<const Variable *> globals;
+		LinAl::Vector<unsigned, 3> compute_local_size;
 	};
 
 	struct StructMember
@@ -148,16 +154,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;
@@ -170,6 +178,7 @@ public:
 	struct Constant
 	{
 		std::string name;
+		unsigned id = 0;
 		int constant_id = -1;
 		DataType type = VOID;
 		union
@@ -181,8 +190,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;
 	};
@@ -220,6 +230,7 @@ private:
 		void reflect_name(CodeIterator);
 		void reflect_member_name(CodeIterator);
 		void reflect_entry_point(CodeIterator);
+		void reflect_execution_mode(CodeIterator);
 		void reflect_void_type(CodeIterator);
 		void reflect_bool_type(CodeIterator);
 		void reflect_int_type(CodeIterator);
@@ -248,6 +259,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; }
@@ -264,6 +276,17 @@ 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;
+
+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;
+
+public:
+	void set_debug_name(const std::string &n) { SpirVModuleBackend::set_debug_name(n); }
 };
 
 } // namespace GL