X-Git-Url: http://git.tdb.fi/?p=libs%2Fgl.git;a=blobdiff_plain;f=source%2Fcore%2Fmodule.h;h=5ea0149d2074e498e11e58ec97cfdd072b1ab642;hp=b1c0f8abebda67ac56742507277bae9ccb92af22;hb=3a1b9cbe2441ae670a97541dc8ccb0a2860c8302;hpb=dccad64b2ec82249d850b9745614042171499972

diff --git a/source/core/module.h b/source/core/module.h
index b1c0f8ab..5ea0149d 100644
--- a/source/core/module.h
+++ b/source/core/module.h
@@ -3,30 +3,221 @@
 
 #include <string>
 #include <msp/io/base.h>
+#include "datatype.h"
+#include "gl.h"
+#include "glsl/compiler.h"
 #include "glsl/sourcemap.h"
 
 namespace Msp {
 namespace GL {
 
+class invalid_module: public std::runtime_error
+{
+public:
+	invalid_module(const std::string &w): runtime_error(w) { }
+	virtual ~invalid_module() throw() { }
+};
+
 class Resources;
 
 class Module
 {
-private:
-	std::string prepared_source;
-	SL::SourceMap source_map;
+public:
+	enum Format
+	{
+		GLSL,
+		SPIR_V
+	};
 
+protected:
+	Module() { }
 public:
-	Module();
+	virtual ~Module() { }
+
+	virtual Format get_format() const = 0;
 
 	void set_source(const std::string &);
 	void load_source(IO::Base &, Resources *, const std::string &);
 	void load_source(IO::Base &, const std::string &);
+private:
+	virtual void compile(SL::Compiler &) = 0;
+};
 
+class GlslModule: public Module
+{
+private:
+	std::string prepared_source;
+	SL::SourceMap source_map;
+
+public:
+	virtual Format get_format() const { return GLSL; }
+
+private:
+	virtual void compile(SL::Compiler &);
+
+public:
 	const std::string &get_prepared_source() const { return prepared_source; }
 	const SL::SourceMap &get_source_map() const { return source_map; }
 };
 
+class SpirVModule: public Module
+{
+public:
+	enum Stage
+	{
+		VERTEX = 0,
+		GEOMETRY = 3,
+		FRAGMENT = 4
+	};
+
+	enum StorageClass
+	{
+		UNIFORM_CONSTANT = 0,
+		INPUT = 1,
+		UNIFORM = 2,
+		OUTPUT = 3
+	};
+
+	struct Constant;
+	struct Structure;
+	struct Variable;
+
+	struct EntryPoint
+	{
+		std::string name;
+		Stage stage;
+		std::vector<const Variable *> globals;
+
+		EntryPoint();
+	};
+
+	struct StructMember
+	{
+		std::string name;
+		DataType type;
+		const Structure *struct_type;
+		unsigned offset;
+		unsigned array_size;
+		const Constant *array_size_spec;
+		unsigned array_stride;
+		unsigned matrix_stride;
+
+		StructMember();
+	};
+
+	struct Structure
+	{
+		std::string name;
+		std::vector<StructMember> members;
+		unsigned size;
+	};
+
+	struct Variable
+	{
+		std::string name;
+		StorageClass storage;
+		DataType type;
+		const Structure *struct_type;
+		const Constant *array_size_spec;
+		unsigned array_size;
+		int location;
+		int descriptor_set;
+		int binding;
+
+		Variable();
+
+		bool operator==(const Variable &) const;
+	};
+
+	struct Constant
+	{
+		std::string name;
+		int constant_id;
+		DataType type;
+		union
+		{
+			int i_value;
+			float f_value;
+		};
+	};
+
+private:
+	struct TypeInfo
+	{
+		DataType type;
+		const Structure *struct_type;
+		const Constant *array_size_spec;
+		unsigned array_size;
+		unsigned array_stride;
+		StorageClass storage;
+
+		TypeInfo();
+	};
+
+	struct Reflection
+	{
+		typedef std::vector<std::uint32_t>::const_iterator CodeIterator;
+
+		std::map<unsigned, std::string> names;
+		std::map<unsigned, Constant> constants;
+		std::map<unsigned, TypeInfo> types;
+		std::map<unsigned, EntryPoint> entry_points;
+		std::map<unsigned, Structure> structs;
+		std::map<unsigned, Variable> variables;
+
+		static std::uint32_t get_opcode(std::uint32_t);
+		static CodeIterator get_op_end(const CodeIterator &);
+		static std::string read_string(CodeIterator &, const CodeIterator &);
+
+		void reflect_code(const std::vector<std::uint32_t> &);
+		void reflect_name(CodeIterator);
+		void reflect_member_name(CodeIterator);
+		void reflect_entry_point(CodeIterator);
+		void reflect_void_type(CodeIterator);
+		void reflect_bool_type(CodeIterator);
+		void reflect_int_type(CodeIterator);
+		void reflect_float_type(CodeIterator);
+		void reflect_vector_type(CodeIterator);
+		void reflect_matrix_type(CodeIterator);
+		void reflect_image_type(CodeIterator);
+		void reflect_sampled_image_type(CodeIterator);
+		void reflect_array_type(CodeIterator);
+		void reflect_struct_type(CodeIterator);
+		void reflect_pointer_type(CodeIterator);
+		void reflect_constant(CodeIterator);
+		void reflect_variable(CodeIterator);
+		void reflect_decorate(CodeIterator);
+		void reflect_member_decorate(CodeIterator);
+	};
+
+	std::vector<std::uint32_t> code;
+	std::vector<EntryPoint> entry_points;
+	std::vector<Structure> structs;
+	std::vector<Variable> variables;
+	std::vector<Constant> spec_constants;
+
+public:
+	SpirVModule() { }
+	SpirVModule(const SpirVModule &);
+	SpirVModule &operator=(const SpirVModule &);
+private:
+	void remap_pointers_from(const SpirVModule &);
+
+public:
+	virtual Format get_format() const { return SPIR_V; }
+
+	void load_code(IO::Base &);
+private:
+	virtual void compile(SL::Compiler &);
+	void reflect();
+
+public:
+	const std::vector<std::uint32_t> &get_code() const { return code; }
+	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; }
+};
+
 } // namespace GL
 } // namespace Msp