#include <vector>
#include <msp/io/base.h>
#include "datatype.h"
+#include "module_backend.h"
#include "glsl/compiler.h"
#include "glsl/sourcemap.h"
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:
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:
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
{
UNIFORM_CONSTANT = 0,
INPUT = 1,
UNIFORM = 2,
- OUTPUT = 3
+ OUTPUT = 3,
+ PUSH_CONSTANT = 9
};
enum BuiltinSemantic
struct EntryPoint
{
std::string name;
+ unsigned id = 0;
Stage stage = VERTEX;
std::vector<const Variable *> globals;
};
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;
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;
struct Constant
{
std::string name;
+ unsigned id = 0;
int constant_id = -1;
DataType type = VOID;
union
};
};
+ 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);
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 &);
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;
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;
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 &);
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; }
+
+ /** 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