1 #ifndef MSP_GL_MODULE_H_
2 #define MSP_GL_MODULE_H_
7 #include <msp/io/base.h>
9 #include "module_backend.h"
10 #include "glsl/compiler.h"
11 #include "glsl/sourcemap.h"
16 class invalid_module: public std::runtime_error
19 invalid_module(const std::string &w): runtime_error(w) { }
20 virtual ~invalid_module() throw() { }
26 Base class for shader modules. Internal representation depends on the
29 Modules can be loaded from files.
31 Applications normally use the Program class to access shaders.
45 virtual ~Module() = default;
47 virtual Format get_format() const = 0;
49 /** Sets the module's content from GLSL source code. */
50 void set_source(const std::string &);
52 /** Loads GLSL source from a file or other I/O object. Any import
53 statements are resolved using res. */
54 void load_source(IO::Base &, Resources *res, const std::string &name);
56 /** Loads GLSL source from a file or other I/O object. Only builtin
57 shader fragments can be imported. */
58 void load_source(IO::Base &, const std::string &);
61 virtual void compile(SL::Compiler &) = 0;
63 SL::Features create_features() const;
66 virtual void set_debug_name(const std::string &) { }
70 A shader module in GLSL source code format.
72 class GlslModule: public Module
75 std::string prepared_source;
76 SL::SourceMap source_map;
79 virtual Format get_format() const { return GLSL; }
82 virtual void compile(SL::Compiler &);
85 const std::string &get_prepared_source() const { return prepared_source; }
86 const SL::SourceMap &get_source_map() const { return source_map; }
90 A shader module in SPIR-V binary format.
92 When the module's contents are set from GLSL source, it will be automatically
93 compiled to SPIR-V. Pre-compiled SPIR-V modules can also be loaded.
95 Afterwards reflection data is available, providing information about variables
96 forming the module's interface.
98 class SpirVModule: public Module, public SpirVModuleBackend
100 friend SpirVModuleBackend;
113 UNIFORM_CONSTANT = 0,
137 Stage stage = VERTEX;
138 std::vector<const Variable *> globals;
139 LinAl::Vector<unsigned, 3> compute_local_size;
145 DataType type = VOID;
146 const Structure *struct_type = 0;
148 unsigned array_size = 0;
149 unsigned array_stride = 0;
150 unsigned matrix_stride = 0;
151 BuiltinSemantic builtin = NOT_BUILTIN;
159 std::vector<StructMember> members;
166 DataType type = VOID;
167 const Structure *struct_type = 0;
168 StorageClass storage = static_cast<StorageClass>(-1);
169 unsigned array_size = 0;
171 unsigned descriptor_set = 0;
173 BuiltinSemantic builtin = NOT_BUILTIN;
175 bool operator==(const Variable &) const;
182 int constant_id = -1;
183 DataType type = VOID;
191 struct InstructionBlock
194 bool negate_condition = false;
195 const Constant *condition = 0;
196 std::vector<const Variable *> accessed_variables;
197 std::vector<const InstructionBlock *> successors;
203 DataType type = VOID;
204 const Structure *struct_type = 0;
205 unsigned array_size = 0;
206 unsigned array_stride = 0;
207 StorageClass storage = static_cast<StorageClass>(-1);
212 typedef std::vector<std::uint32_t>::const_iterator CodeIterator;
214 std::map<unsigned, std::string> names;
215 std::map<unsigned, Constant> constants;
216 std::map<unsigned, TypeInfo> types;
217 std::map<unsigned, EntryPoint> entry_points;
218 std::map<unsigned, Structure> structs;
219 std::map<unsigned, Variable> variables;
220 std::map<unsigned, InstructionBlock> blocks;
221 std::map<unsigned, unsigned> access_chain_bases;
222 Constant true_condition;
223 InstructionBlock *current_block = 0;
225 static std::uint32_t get_opcode(std::uint32_t);
226 static CodeIterator get_op_end(const CodeIterator &);
227 static std::string read_string(CodeIterator &, const CodeIterator &);
229 void reflect_code(const std::vector<std::uint32_t> &);
230 void reflect_name(CodeIterator);
231 void reflect_member_name(CodeIterator);
232 void reflect_entry_point(CodeIterator);
233 void reflect_execution_mode(CodeIterator);
234 void reflect_void_type(CodeIterator);
235 void reflect_bool_type(CodeIterator);
236 void reflect_int_type(CodeIterator);
237 void reflect_float_type(CodeIterator);
238 void reflect_vector_type(CodeIterator);
239 void reflect_matrix_type(CodeIterator);
240 void reflect_image_type(CodeIterator);
241 void reflect_sampled_image_type(CodeIterator);
242 void reflect_array_type(CodeIterator);
243 void reflect_struct_type(CodeIterator);
244 void reflect_pointer_type(CodeIterator);
245 void reflect_constant(CodeIterator);
246 void reflect_variable(CodeIterator);
247 void reflect_access(CodeIterator);
248 void reflect_access_chain(CodeIterator);
249 void reflect_decorate(CodeIterator);
250 void reflect_member_decorate(CodeIterator);
251 void reflect_label(CodeIterator);
252 void reflect_branch(CodeIterator);
253 void reflect_branch_conditional(CodeIterator);
256 std::vector<std::uint32_t> code;
257 std::vector<EntryPoint> entry_points;
258 std::vector<Structure> structs;
259 std::vector<Variable> variables;
260 std::vector<Constant> spec_constants;
261 std::vector<InstructionBlock> blocks;
262 bool specializable = false;
265 virtual Format get_format() const { return SPIR_V; }
267 /** Loads a SPIR-V binary from a file or other I/O object. */
268 void load_code(IO::Base &);
270 virtual void compile(SL::Compiler &);
274 const std::vector<std::uint32_t> &get_code() const { return code; }
275 const std::vector<EntryPoint> &get_entry_points() const { return entry_points; }
276 const std::vector<Variable> &get_variables() const { return variables; }
277 const std::vector<Constant> &get_spec_constants() const { return spec_constants; }
278 const std::vector<InstructionBlock> &get_blocks() const { return blocks; }
279 bool is_specializable() const { return specializable; }
281 /** Creates a new module which is a specialized version of this one. */
282 SpirVModule *specialize(const std::map<std::string, int> &) const;
285 std::vector<const InstructionBlock *> collect_visited_blocks(const std::map<unsigned, int> &) const;
286 void collect_visited_blocks(unsigned, std::vector<std::uint8_t> &) const;
289 void set_debug_name(const std::string &n) { SpirVModuleBackend::set_debug_name(n); }