]> git.tdb.fi Git - libs/gl.git/blob - source/glsl/visitor.h
Support specialization constants in the GLSL compiler
[libs/gl.git] / source / glsl / visitor.h
1 #ifndef MSP_GL_SL_VISITOR_H_
2 #define MSP_GL_SL_VISITOR_H_
3
4 #include <set>
5 #include <vector>
6 #include "syntax.h"
7
8 namespace Msp {
9 namespace GL {
10 namespace SL {
11
12 class NodeVisitor
13 {
14 protected:
15         NodeVisitor() { }
16 public:
17         virtual ~NodeVisitor() { }
18
19         virtual void visit(Block &) { }
20         virtual void visit(Literal &) { }
21         virtual void visit(ParenthesizedExpression &) { }
22         virtual void visit(VariableReference &) { }
23         virtual void visit(InterfaceBlockReference &) { }
24         virtual void visit(MemberAccess &) { }
25         virtual void visit(UnaryExpression &) { }
26         virtual void visit(BinaryExpression &) { }
27         virtual void visit(Assignment &);
28         virtual void visit(FunctionCall &) { }
29         virtual void visit(ExpressionStatement &) { }
30         virtual void visit(Import &) { }
31         virtual void visit(Precision &) { }
32         virtual void visit(Layout &) { }
33         virtual void visit(InterfaceLayout &) { }
34         virtual void visit(StructDeclaration &) { }
35         virtual void visit(VariableDeclaration &) { }
36         virtual void visit(InterfaceBlock &) { }
37         virtual void visit(FunctionDeclaration &) { }
38         virtual void visit(Conditional &) { }
39         virtual void visit(Iteration &) { }
40         virtual void visit(Passthrough &) { }
41         virtual void visit(Return &) { }
42         virtual void visit(Jump &) { }
43 };
44
45 class TraversingVisitor: public NodeVisitor
46 {
47 protected:
48         Block *current_block;
49
50         TraversingVisitor(): current_block(0) { }
51
52 public:
53         virtual void enter(Block &) { }
54         virtual void visit(Block &);
55         virtual void visit(ParenthesizedExpression &);
56         virtual void visit(MemberAccess &);
57         virtual void visit(UnaryExpression &);
58         virtual void visit(BinaryExpression &);
59         virtual void visit(FunctionCall &);
60         virtual void visit(ExpressionStatement &);
61         virtual void visit(InterfaceLayout &);
62         virtual void visit(StructDeclaration &);
63         virtual void visit(VariableDeclaration &);
64         virtual void visit(InterfaceBlock &);
65         virtual void visit(FunctionDeclaration &);
66         virtual void visit(Conditional &);
67         virtual void visit(Iteration &);
68         virtual void visit(Passthrough &);
69         virtual void visit(Return &);
70 };
71
72 template<typename T>
73 class NodeGatherer: private TraversingVisitor
74 {
75 private:
76         std::vector<T *> nodes;
77
78 public:
79         const std::vector<T *> &apply(Stage &s) { s.content.visit(*this); return nodes; }
80
81 private:
82         virtual void visit(T &n) { nodes.push_back(&n); }
83 };
84
85 class NodeRemover: private TraversingVisitor
86 {
87 private:
88         Stage *stage;
89         const std::set<Node *> *to_remove;
90         bool recursive_remove;
91
92 public:
93         NodeRemover();
94
95         void apply(Stage &, const std::set<Node *> &);
96
97 private:
98         template<typename T>
99         void remove_from_map(std::map<std::string, T *> &, const std::string &, T &);
100
101         virtual void visit(Block &);
102         virtual void visit(StructDeclaration &);
103         virtual void visit(VariableDeclaration &);
104         virtual void visit(InterfaceBlock &);
105         virtual void visit(Iteration &);
106 };
107
108 } // namespace SL
109 } // namespace GL
110 } // namespace Msp
111
112 #endif