source/glsl/resolve.h

   1 #ifndef MSP_GL_SL_RESOLVE_H_
   2 #define MSP_GL_SL_RESOLVE_H_
   3
   4 #include <map>
   5 #include <set>
   6 #include <string>
   7 #include <vector>
   8 #include "visitor.h"
   9
  10 namespace Msp {
  11 namespace GL {
  12 namespace SL {
  13
  14 /** Forms links between nested blocks in the syntax tree. */
  15 class BlockHierarchyResolver: private TraversingVisitor
  16 {
  17 private:
  18         bool r_any_resolved = false;
  19
  20 public:
  21         bool apply(Stage &s) { r_any_resolved = false; s.content.visit(*this); return r_any_resolved; }
  22
  23 private:
  24         virtual void enter(Block &);
  25 };
  26
  27 /** Resolves types of variables and base types of other types. */
  28 class TypeResolver: private TraversingVisitor
  29 {
  30 private:
  31         Stage *stage = 0;
  32         std::map<TypeDeclaration *, TypeDeclaration *> alias_map;
  33         std::map<TypeDeclaration *, TypeDeclaration *> array_types;
  34         NodeList<Statement>::iterator type_insert_point;
  35         InterfaceBlock *iface_block = 0;
  36         bool r_any_resolved = false;
  37
  38 public:
  39         bool apply(Stage &);
  40
  41 private:
  42         TypeDeclaration *get_or_create_array_type(TypeDeclaration &);
  43         void resolve_type(TypeDeclaration *&, const std::string &, bool);
  44         virtual void visit(Block &);
  45         virtual void visit(BasicTypeDeclaration &);
  46         virtual void visit(ImageTypeDeclaration &);
  47         virtual void visit(StructDeclaration &);
  48         virtual void visit(VariableDeclaration &);
  49         virtual void visit(InterfaceBlock &);
  50         virtual void visit(FunctionDeclaration &);
  51 };
  52
  53 /** Resolves variable references.  Variable references which match the name
  54 of an interface block are turned into interface block references. */
  55 class VariableResolver: private TraversingVisitor
  56 {
  57 private:
  58         Stage *stage = 0;
  59         RefPtr<Expression> r_replacement_expr;
  60         bool r_any_resolved = false;
  61         bool record_target = false;
  62         bool r_self_referencing = false;
  63         Assignment::Target r_assignment_target;
  64         std::vector<VariableDeclaration *> redeclared_builtins;
  65         std::set<Node *> nodes_to_remove;
  66
  67 public:
  68         bool apply(Stage &);
  69
  70 private:
  71         virtual void enter(Block &);
  72         virtual void visit(RefPtr<Expression> &);
  73         void check_assignment_target(Statement *);
  74         virtual void visit(VariableReference &);
  75         virtual void visit(InterfaceBlockReference &);
  76         virtual void visit(MemberAccess &);
  77         virtual void visit(Swizzle &);
  78         virtual void visit(BinaryExpression &);
  79         virtual void visit(Assignment &);
  80         void merge_layouts(Layout &, const Layout &);
  81         virtual void visit(VariableDeclaration &);
  82         virtual void visit(InterfaceBlock &);
  83 };
  84
  85 /** Resolves types and lvalueness of expressions. */
  86 class ExpressionResolver: private TraversingVisitor
  87 {
  88 private:
  89         enum Compatibility
  90         {
  91                 NOT_COMPATIBLE,
  92                 LEFT_CONVERTIBLE,
  93                 RIGHT_CONVERTIBLE,
  94                 SAME_TYPE
  95         };
  96
  97         struct ArgumentInfo
  98         {
  99                 BasicTypeDeclaration *type = 0;
 100                 unsigned component_count = 0;
 101         };
 102
 103         Stage *stage = 0;
 104         std::vector<BasicTypeDeclaration *> basic_types;
 105         NodeList<Statement>::iterator insert_point;
 106         bool r_any_resolved = false;
 107
 108 public:
 109         bool apply(Stage &);
 110
 111 private:
 112         static Compatibility get_compatibility(BasicTypeDeclaration &, BasicTypeDeclaration &);
 113         BasicTypeDeclaration *find_type(BasicTypeDeclaration::Kind, unsigned, bool = true);
 114         BasicTypeDeclaration *find_type(BasicTypeDeclaration &, BasicTypeDeclaration::Kind, unsigned);
 115         void convert_to(RefPtr<Expression> &, BasicTypeDeclaration &);
 116         bool convert_to_element(RefPtr<Expression> &, BasicTypeDeclaration &);
 117         bool truncate_vector(RefPtr<Expression> &, unsigned);
 118         void resolve(Expression &, TypeDeclaration *, bool);
 119
 120         virtual void visit(Block &);
 121         virtual void visit(Literal &);
 122         virtual void visit(VariableReference &);
 123         virtual void visit(InterfaceBlockReference &);
 124         virtual void visit(MemberAccess &);
 125         virtual void visit(Swizzle &);
 126         virtual void visit(UnaryExpression &);
 127         void visit(BinaryExpression &, bool);
 128         virtual void visit(BinaryExpression &);
 129         virtual void visit(Assignment &);
 130         virtual void visit(TernaryExpression &);
 131         void visit_constructor(FunctionCall &);
 132         virtual void visit(FunctionCall &);
 133         virtual void visit(BasicTypeDeclaration &);
 134         virtual void visit(VariableDeclaration &);
 135 };
 136
 137 /** Resolves function declarations and calls. */
 138 class FunctionResolver: private TraversingVisitor
 139 {
 140 private:
 141         Stage *stage = 0;
 142         std::map<std::string, std::vector<FunctionDeclaration *> > declarations;
 143         bool r_any_resolved = false;
 144
 145 public:
 146         bool apply(Stage &);
 147
 148 private:
 149         static bool can_convert_arguments(const FunctionCall &, const FunctionDeclaration &);
 150
 151         virtual void visit(FunctionCall &);
 152         virtual void visit(FunctionDeclaration &);
 153 };
 154
 155 } // namespace SL
 156 } // namespace GL
 157 } // namespace Msp
 158
 159 #endif