Create specialized versions of SPIR-V modules with default spec values

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

diff --git a/source/core/module.h b/source/core/module.h
index 15c07fb0366bd79b0e2ee737f4c21e42b3ed58f8..51924d606668db500e7b871f72f7677ae538190d 100644 (file)
--- a/source/core/module.h
+++ b/source/core/module.h
@@ -94,6 +94,8 @@ forming the module's interface.
 */
 class SpirVModule: public Module, public SpirVModuleBackend
 {
+       friend SpirVModuleBackend;
+
 public:
        enum Stage
        {
@@ -127,6 +129,7 @@ public:
        struct EntryPoint
        {
                std::string name;
+               unsigned id = 0;
                Stage stage = VERTEX;
                std::vector<const Variable *> globals;
        };
@@ -146,16 +149,18 @@ 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;
+               StorageClass storage = static_cast<StorageClass>(-1);
                unsigned array_size = 0;
                int location = -1;
                unsigned descriptor_set = 0;
@@ -168,6 +173,7 @@ public:
        struct Constant
        {
                std::string name;
+               unsigned id = 0;
                int constant_id = -1;
                DataType type = VOID;
                union
@@ -179,8 +185,9 @@ public:
 
        struct InstructionBlock
        {
-               const Constant *condition = 0;
+               unsigned id = 0;
                bool negate_condition = false;
+               const Constant *condition = 0;
                std::vector<const Variable *> accessed_variables;
                std::vector<const InstructionBlock *> successors;
        };
@@ -246,6 +253,7 @@ private:
        std::vector<Variable> variables;
        std::vector<Constant> spec_constants;
        std::vector<InstructionBlock> blocks;
+       bool specializable = false;
 
 public:
        virtual Format get_format() const { return SPIR_V; }
@@ -262,6 +270,14 @@ public:
        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