]> git.tdb.fi Git - libs/gl.git/blob - source/glsl/visitor.h
Remove the View and Framebuffer constructors from Sequence
[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 /** Base class for all node visitors. */
13 class NodeVisitor
14 {
15 protected:
16         NodeVisitor() { }
17 public:
18         virtual ~NodeVisitor() { }
19
20         virtual void visit(Block &) { }
21         virtual void visit(Literal &) { }
22         virtual void visit(VariableReference &) { }
23         virtual void visit(InterfaceBlockReference &) { }
24         virtual void visit(MemberAccess &) { }
25         virtual void visit(Swizzle &) { }
26         virtual void visit(UnaryExpression &) { }
27         virtual void visit(BinaryExpression &) { }
28         virtual void visit(Assignment &) { }
29         virtual void visit(TernaryExpression &) { }
30         virtual void visit(FunctionCall &) { }
31         virtual void visit(ExpressionStatement &) { }
32         virtual void visit(Import &) { }
33         virtual void visit(Precision &) { }
34         virtual void visit(Layout &) { }
35         virtual void visit(InterfaceLayout &) { }
36         virtual void visit(BasicTypeDeclaration &) { }
37         virtual void visit(ImageTypeDeclaration &) { }
38         virtual void visit(StructDeclaration &) { }
39         virtual void visit(VariableDeclaration &) { }
40         virtual void visit(InterfaceBlock &) { }
41         virtual void visit(FunctionDeclaration &) { }
42         virtual void visit(Conditional &) { }
43         virtual void visit(Iteration &) { }
44         virtual void visit(Passthrough &) { }
45         virtual void visit(Return &) { }
46         virtual void visit(Jump &) { }
47 };
48
49 /** An intermediate base visitor class which traverses the syntax tree. */
50 class TraversingVisitor: public NodeVisitor
51 {
52 protected:
53         Block *current_block;
54
55         TraversingVisitor(): current_block(0) { }
56
57 public:
58         virtual void enter(Block &) { }
59         virtual void visit(Block &);
60         virtual void visit(RefPtr<Expression> &);
61         virtual void visit(MemberAccess &);
62         virtual void visit(Swizzle &);
63         virtual void visit(UnaryExpression &);
64         virtual void visit(BinaryExpression &);
65         virtual void visit(Assignment &);
66         virtual void visit(TernaryExpression &);
67         virtual void visit(FunctionCall &);
68         virtual void visit(ExpressionStatement &);
69         virtual void visit(InterfaceLayout &);
70         virtual void visit(StructDeclaration &);
71         virtual void visit(VariableDeclaration &);
72         virtual void visit(InterfaceBlock &);
73         virtual void visit(FunctionDeclaration &);
74         virtual void visit(Conditional &);
75         virtual void visit(Iteration &);
76         virtual void visit(Passthrough &);
77         virtual void visit(Return &);
78 };
79
80 /** Removes a set of nodes from the syntax tree. */
81 class NodeRemover: private TraversingVisitor
82 {
83 private:
84         Stage *stage;
85         const std::set<Node *> *to_remove;
86         bool recursive_remove;
87
88 public:
89         NodeRemover();
90
91         void apply(Stage &, const std::set<Node *> &);
92
93 private:
94         template<typename T>
95         void remove_from_map(std::map<std::string, T *> &, const std::string &, T &);
96
97         virtual void visit(Block &);
98         void visit(TypeDeclaration &);
99         virtual void visit(BasicTypeDeclaration &t) { visit(static_cast<TypeDeclaration &>(t)); }
100         virtual void visit(ImageTypeDeclaration &t) { visit(static_cast<TypeDeclaration &>(t)); }
101         virtual void visit(StructDeclaration &t) { visit(static_cast<TypeDeclaration &>(t)); }
102         virtual void visit(VariableDeclaration &);
103         virtual void visit(InterfaceBlock &);
104         virtual void visit(FunctionDeclaration &);
105         virtual void visit(Iteration &);
106 };
107
108 /** Reorders a set of nodes so they appear before another node.  Only nodes
109 on the same hierarchly level as the target node are reordered. */
110 class NodeReorderer: private TraversingVisitor
111 {
112 private:
113         Node *reorder_before;
114         const std::set<Node *> *to_reorder;
115
116 public:
117         NodeReorderer();
118
119         void apply(Stage &, Node &, const std::set<Node *> &);
120
121 private:
122         virtual void visit(Block &);
123 };
124
125 } // namespace SL
126 } // namespace GL
127 } // namespace Msp
128
129 #endif