Reflect the control flow graph from SPIR-V and check variable accesses

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

diff --git a/source/core/module.h b/source/core/module.h
index c669af0506ab64d26b25bf504ec28533e7e1cf92..15c07fb0366bd79b0e2ee737f4c21e42b3ed58f8 100644 (file)
--- a/source/core/module.h
+++ b/source/core/module.h
@@ -177,6 +177,14 @@ public:
                };
        };
 
+       struct InstructionBlock
+       {
+               const Constant *condition = 0;
+               bool negate_condition = false;
+               std::vector<const Variable *> accessed_variables;
+               std::vector<const InstructionBlock *> successors;
+       };
+
 private:
        struct TypeInfo
        {
@@ -197,6 +205,10 @@ private:
                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 &);
@@ -219,8 +231,13 @@ private:
                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;
@@ -228,6 +245,7 @@ private:
        std::vector<Structure> structs;
        std::vector<Variable> variables;
        std::vector<Constant> spec_constants;
+       std::vector<InstructionBlock> blocks;
 
 public:
        virtual Format get_format() const { return SPIR_V; }
@@ -243,6 +261,7 @@ public:
        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; }
 };
 
 } // namespace GL