source/glsl/spirv.h

   1 #ifndef MSP_GL_SL_SPIRV_H_
   2 #define MSP_GL_SL_SPIRV_H_
   3
   4 #include <map>
   5 #include <string>
   6 #include <vector>
   7 #include "spirvconstants.h"
   8 #include "spirvwriter.h"
   9 #include "visitor.h"
  10
  11 namespace Msp {
  12 namespace GL {
  13 namespace SL {
  14
  15 /** Creates SPIR-V binary from a module. */
  16 class SpirVGenerator: private NodeVisitor
  17 {
  18 private:
  19         typedef SpirVCapability Capability;
  20         typedef SpirVStorageClass StorageClass;
  21         typedef SpirVOpcode Opcode;
  22         typedef SpirVDecoration Decoration;
  23         typedef SpirVBuiltin BuiltinSemantic;
  24         typedef SpirVContent::Word Word;
  25         typedef SpirVWriter::Id Id;
  26
  27         struct BuiltinFunctionInfo
  28         {
  29                 char function[22];
  30                 char arg_types[5];
  31                 char extension[13];
  32                 Word opcode;
  33                 std::uint8_t arg_order[4];
  34                 Word capability;
  35                 void (SpirVGenerator::*handler)(FunctionCall &, const std::vector<Id> &);
  36         };
  37
  38         struct Declaration
  39         {
  40                 Id id;
  41                 Id type_id;
  42
  43                 Declaration(Id i, Id t): id(i), type_id(t) { }
  44         };
  45
  46         struct TypeKey
  47         {
  48                 Id type_id;
  49                 unsigned detail;
  50
  51                 TypeKey(Id i, unsigned d): type_id(i), detail(d) { }
  52                 TypeKey(BasicTypeDeclaration::Kind, bool);
  53
  54                 bool operator<(const TypeKey &) const;
  55         };
  56
  57         struct ConstantKey
  58         {
  59                 Id type_id;
  60                 union
  61                 {
  62                         int int_value;
  63                         float float_value;
  64                 };
  65
  66                 ConstantKey(Id t, int i): type_id(t), int_value(i) { }
  67                 ConstantKey(Id t, unsigned u): type_id(t), int_value(u) { }
  68                 ConstantKey(Id t, float f): type_id(t), float_value(f) { }
  69
  70                 bool operator<(const ConstantKey &) const;
  71         };
  72
  73         Stage *stage;
  74         FunctionDeclaration *current_function;
  75         std::vector<const InterfaceLayout *> interface_layouts;
  76         SpirVContent content;
  77         SpirVWriter writer;
  78         std::set<Capability> used_capabilities;
  79         std::map<std::string, Id> imported_extension_ids;
  80         std::map<Node *, Declaration> declared_ids;
  81         std::map<std::string, Id> declared_uniform_ids;
  82         std::map<TypeKey, Id> standard_type_ids;
  83         std::map<Id, Id> image_type_ids;
  84         std::map<TypeKey, Id> array_type_ids;
  85         std::map<TypeKey, Id> pointer_type_ids;
  86         std::map<std::string, Id> function_type_ids;
  87         std::map<ConstantKey, Id> constant_ids;
  88         std::map<const VariableDeclaration *, Id> variable_load_ids;
  89         Id next_id;
  90         Id r_expression_result_id;
  91         bool r_constant_result;
  92         bool constant_expression;
  93         bool spec_constant;
  94         bool reachable;
  95         bool composite_access;
  96         Id r_composite_base_id;
  97         Node *r_composite_base;
  98         std::vector<unsigned> r_composite_chain;
  99         Id assignment_source_id;
 100         Id loop_merge_block_id;
 101         Id loop_continue_target_id;
 102
 103         static const BuiltinFunctionInfo builtin_functions[];
 104
 105 public:
 106         SpirVGenerator();
 107
 108         void apply(Module &);
 109         const std::vector<Word> &get_code() const { return content.code; }
 110
 111 private:
 112         static StorageClass get_interface_storage(const std::string &, bool);
 113         static SpirVBuiltin get_builtin_semantic(const std::string &);
 114         void use_capability(Capability);
 115         Id import_extension(const std::string &);
 116         Id get_id(Node &) const;
 117         Id allocate_id(Node &, Id);
 118         Id allocate_forward_id(Node &);
 119         Id write_constant(Id, Word, bool);
 120         static ConstantKey get_constant_key(Id, const Variant &value);
 121         Id get_constant_id(Id, const Variant &value);
 122         Id get_vector_constant_id(Id, unsigned, Id);
 123         Id get_standard_type_id(BasicTypeDeclaration::Kind, unsigned, bool = true);
 124         bool is_scalar_type(Id, BasicTypeDeclaration::Kind) const;
 125         Id get_array_type_id(TypeDeclaration &, unsigned);
 126         Id get_pointer_type_id(Id, StorageClass);
 127         Id get_variable_type_id(const VariableDeclaration &);
 128         Id get_load_id(VariableDeclaration &);
 129         void prune_loads(Id);
 130         Id begin_expression(Opcode, Id, unsigned = 0);
 131         void end_expression(Opcode);
 132         Id write_expression(Opcode, Id, Id);
 133         Id write_expression(Opcode, Id, Id, Id);
 134         void write_deconstruct(Id, Id, Id *, unsigned);
 135         Id write_construct(Id, const Id *, unsigned);
 136
 137         virtual void visit(Block &);
 138         virtual void visit(Literal &);
 139         virtual void visit(VariableReference &);
 140         virtual void visit(InterfaceBlockReference &);
 141         void generate_composite_access(TypeDeclaration &);
 142         void visit_composite(Expression &, unsigned, TypeDeclaration &);
 143         void visit_isolated(Expression &);
 144         virtual void visit(MemberAccess &);
 145         virtual void visit(Swizzle &);
 146         virtual void visit(UnaryExpression &);
 147         virtual void visit(BinaryExpression &);
 148         virtual void visit(Assignment &);
 149         virtual void visit(TernaryExpression &);
 150         virtual void visit(FunctionCall &);
 151         void visit_constructor(FunctionCall &, const std::vector<Id> &, bool);
 152         void visit_builtin_matrix_comp_mult(FunctionCall &, const std::vector<Id> &);
 153         void visit_builtin_texture_query(FunctionCall &, const std::vector<Id> &);
 154         void visit_builtin_texture(FunctionCall &, const std::vector<Id> &);
 155         void visit_builtin_texel_fetch(FunctionCall &, const std::vector<Id> &);
 156         void visit_builtin_interpolate(FunctionCall &, const std::vector<Id> &);
 157         virtual void visit(ExpressionStatement &);
 158         virtual void visit(InterfaceLayout &);
 159         bool check_duplicate_type(TypeDeclaration &);
 160         bool check_standard_type(BasicTypeDeclaration &);
 161         virtual void visit(BasicTypeDeclaration &);
 162         virtual void visit(ImageTypeDeclaration &);
 163         virtual void visit(StructDeclaration &);
 164         virtual void visit(VariableDeclaration &);
 165         virtual void visit(InterfaceBlock &);
 166         void visit_entry_point(FunctionDeclaration &, Id);
 167         virtual void visit(FunctionDeclaration &);
 168         virtual void visit(Conditional &);
 169         virtual void visit(Iteration &);
 170         virtual void visit(Return &);
 171         virtual void visit(Jump &);
 172 };
 173
 174 } // namespace SL
 175 } // namespace GL
 176 } // namespace Msp
 177
 178 #endif