X-Git-Url: http://git.tdb.fi/?p=libs%2Fgl.git;a=blobdiff_plain;f=source%2Fcore%2Fmodule.h;h=51924d606668db500e7b871f72f7677ae538190d;hp=9ac54d328ee42a86b97fed32e82383a171138739;hb=3efe3bab1c8290bd49a957ebec0ad97e58a35fcf;hpb=271760e6099bf5f4ad90894697dab911c236a0a3

diff --git a/source/core/module.h b/source/core/module.h
index 9ac54d32..51924d60 100644
--- a/source/core/module.h
+++ b/source/core/module.h
@@ -6,6 +6,7 @@
 #include <vector>
 #include <msp/io/base.h>
 #include "datatype.h"
+#include "module_backend.h"
 #include "glsl/compiler.h"
 #include "glsl/sourcemap.h"
 
@@ -21,6 +22,14 @@ public:
 
 class Resources;
 
+/**
+Base class for shader modules.  Internal representation depends on the
+concrete type.
+
+Modules can be loaded from files.
+
+Applications normally use the Program class to access shaders.
+*/
 class Module
 {
 public:
@@ -37,13 +46,26 @@ public:
 
 	virtual Format get_format() const = 0;
 
+	/** Sets the module's content from GLSL source code. */
 	void set_source(const std::string &);
-	void load_source(IO::Base &, Resources *, const std::string &);
+
+	/** Loads GLSL source from a file or other I/O object.  Any import
+	statements are resolved using res. */
+	void load_source(IO::Base &, Resources *res, const std::string &name);
+
+	/** Loads GLSL source from a file or other I/O object.  Only builtin
+	shader fragments can be imported. */
 	void load_source(IO::Base &, const std::string &);
+
 private:
 	virtual void compile(SL::Compiler &) = 0;
+
+	SL::Features create_features() const;
 };
 
+/**
+A shader module in GLSL source code format.
+*/
 class GlslModule: public Module
 {
 private:
@@ -61,8 +83,19 @@ public:
 	const SL::SourceMap &get_source_map() const { return source_map; }
 };
 
-class SpirVModule: public Module
+/**
+A shader module in SPIR-V binary format.
+
+When the module's contents are set from GLSL source, it will be automatically
+compiled to SPIR-V.  Pre-compiled SPIR-V modules can also be loaded.
+
+Afterwards reflection data is available, providing information about variables
+forming the module's interface.
+*/
+class SpirVModule: public Module, public SpirVModuleBackend
 {
+	friend SpirVModuleBackend;
+
 public:
 	enum Stage
 	{
@@ -76,7 +109,8 @@ public:
 		UNIFORM_CONSTANT = 0,
 		INPUT = 1,
 		UNIFORM = 2,
-		OUTPUT = 3
+		OUTPUT = 3,
+		PUSH_CONSTANT = 9
 	};
 
 	enum BuiltinSemantic
@@ -95,6 +129,7 @@ public:
 	struct EntryPoint
 	{
 		std::string name;
+		unsigned id = 0;
 		Stage stage = VERTEX;
 		std::vector<const Variable *> globals;
 	};
@@ -106,7 +141,6 @@ public:
 		const Structure *struct_type = 0;
 		unsigned offset = 0;
 		unsigned array_size = 0;
-		const Constant *array_size_spec = 0;
 		unsigned array_stride = 0;
 		unsigned matrix_stride = 0;
 		BuiltinSemantic builtin = NOT_BUILTIN;
@@ -115,20 +149,21 @@ 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;
-		const Constant *array_size_spec = 0;
+		StorageClass storage = static_cast<StorageClass>(-1);
 		unsigned array_size = 0;
 		int location = -1;
-		int descriptor_set = -1;
+		unsigned descriptor_set = 0;
 		int binding = -1;
 		BuiltinSemantic builtin = NOT_BUILTIN;
 
@@ -138,6 +173,7 @@ public:
 	struct Constant
 	{
 		std::string name;
+		unsigned id = 0;
 		int constant_id = -1;
 		DataType type = VOID;
 		union
@@ -147,12 +183,20 @@ public:
 		};
 	};
 
+	struct InstructionBlock
+	{
+		unsigned id = 0;
+		bool negate_condition = false;
+		const Constant *condition = 0;
+		std::vector<const Variable *> accessed_variables;
+		std::vector<const InstructionBlock *> successors;
+	};
+
 private:
 	struct TypeInfo
 	{
 		DataType type = VOID;
 		const Structure *struct_type = 0;
-		const Constant *array_size_spec = 0;
 		unsigned array_size = 0;
 		unsigned array_stride = 0;
 		StorageClass storage = static_cast<StorageClass>(-1);
@@ -168,6 +212,10 @@ private:
 		std::map<unsigned, EntryPoint> entry_points;
 		std::map<unsigned, Structure> structs;
 		std::map<unsigned, Variable> variables;
+		std::map<unsigned, InstructionBlock> blocks;
+		std::map<unsigned, unsigned> access_chain_bases;
+		Constant true_condition;
+		InstructionBlock *current_block = 0;
 
 		static std::uint32_t get_opcode(std::uint32_t);
 		static CodeIterator get_op_end(const CodeIterator &);
@@ -190,8 +238,13 @@ private:
 		void reflect_pointer_type(CodeIterator);
 		void reflect_constant(CodeIterator);
 		void reflect_variable(CodeIterator);
+		void reflect_access(CodeIterator);
+		void reflect_access_chain(CodeIterator);
 		void reflect_decorate(CodeIterator);
 		void reflect_member_decorate(CodeIterator);
+		void reflect_label(CodeIterator);
+		void reflect_branch(CodeIterator);
+		void reflect_branch_conditional(CodeIterator);
 	};
 
 	std::vector<std::uint32_t> code;
@@ -199,17 +252,13 @@ private:
 	std::vector<Structure> structs;
 	std::vector<Variable> variables;
 	std::vector<Constant> spec_constants;
-
-public:
-	SpirVModule() = default;
-	SpirVModule(const SpirVModule &);
-	SpirVModule &operator=(const SpirVModule &);
-private:
-	void remap_pointers_from(const SpirVModule &);
+	std::vector<InstructionBlock> blocks;
+	bool specializable = false;
 
 public:
 	virtual Format get_format() const { return SPIR_V; }
 
+	/** Loads a SPIR-V binary from a file or other I/O object. */
 	void load_code(IO::Base &);
 private:
 	virtual void compile(SL::Compiler &);
@@ -220,6 +269,15 @@ public:
 	const std::vector<EntryPoint> &get_entry_points() const { return entry_points; }
 	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;
 };
 
 } // namespace GL