source/glsl/spirv.h

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