]> git.tdb.fi Git - libs/gl.git/blob - source/glsl/resolve.h
Minor cleanups and tweaks
[libs/gl.git] / 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<Statement *> 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         void redeclare_builtin(VariableDeclaration &, VariableDeclaration &);
82         virtual void visit(VariableDeclaration &);
83         virtual void visit(InterfaceBlock &);
84 };
85
86 /** Resolves types and lvalueness of expressions. */
87 class ExpressionResolver: private TraversingVisitor
88 {
89 private:
90         enum Compatibility
91         {
92                 NOT_COMPATIBLE,
93                 LEFT_CONVERTIBLE,
94                 RIGHT_CONVERTIBLE,
95                 SAME_TYPE
96         };
97
98         struct ArgumentInfo
99         {
100                 BasicTypeDeclaration *type = 0;
101                 unsigned component_count = 0;
102         };
103
104         Stage *stage = 0;
105         std::vector<BasicTypeDeclaration *> basic_types;
106         NodeList<Statement>::iterator insert_point;
107         bool r_any_resolved = false;
108
109 public:
110         bool apply(Stage &);
111
112 private:
113         static Compatibility get_compatibility(BasicTypeDeclaration &, BasicTypeDeclaration &);
114         BasicTypeDeclaration *find_type(BasicTypeDeclaration::Kind, unsigned, bool = true);
115         BasicTypeDeclaration *find_type(BasicTypeDeclaration &, BasicTypeDeclaration::Kind, unsigned);
116         void convert_to(RefPtr<Expression> &, BasicTypeDeclaration &);
117         bool convert_to_element(RefPtr<Expression> &, BasicTypeDeclaration &);
118         bool truncate_vector(RefPtr<Expression> &, unsigned);
119         void resolve(Expression &, TypeDeclaration *, bool);
120
121         virtual void visit(Block &);
122         virtual void visit(Literal &);
123         virtual void visit(VariableReference &);
124         virtual void visit(InterfaceBlockReference &);
125         virtual void visit(MemberAccess &);
126         virtual void visit(Swizzle &);
127         virtual void visit(UnaryExpression &);
128         void visit(BinaryExpression &, bool);
129         virtual void visit(BinaryExpression &);
130         virtual void visit(Assignment &);
131         virtual void visit(TernaryExpression &);
132         void visit_constructor(FunctionCall &);
133         virtual void visit(FunctionCall &);
134         virtual void visit(BasicTypeDeclaration &);
135         virtual void visit(VariableDeclaration &);
136 };
137
138 /** Resolves function declarations and calls. */
139 class FunctionResolver: private TraversingVisitor
140 {
141 private:
142         Stage *stage = 0;
143         std::map<std::string, std::vector<FunctionDeclaration *> > declarations;
144         bool r_any_resolved = false;
145
146 public:
147         bool apply(Stage &);
148
149 private:
150         static bool can_convert_arguments(const FunctionCall &, const FunctionDeclaration &);
151
152         virtual void visit(FunctionCall &);
153         virtual void visit(FunctionDeclaration &);
154 };
155
156 } // namespace SL
157 } // namespace GL
158 } // namespace Msp
159
160 #endif