Use standard fixed-size integer types

[libs/gl.git] / source / core / module.h

diff --git a/source/core/module.h b/source/core/module.h
index f65f739aeb1e2de9a6a8a640b473a9312655dc3d..5ea0149d2074e498e11e58ec97cfdd072b1ab642 100644 (file)
--- a/source/core/module.h
+++ b/source/core/module.h
@@ -3,34 +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:
-       void compile(SL::Compiler &);
+       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