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