]> git.tdb.fi Git - libs/gl.git/blob - source/glsl/visitor.h
Ignore function parameters in InterfaceGenerator
[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(MemberAccess &) { }
24         virtual void visit(UnaryExpression &) { }
25         virtual void visit(BinaryExpression &) { }
26         virtual void visit(Assignment &);
27         virtual void visit(FunctionCall &) { }
28         virtual void visit(ExpressionStatement &) { }
29         virtual void visit(Import &) { }
30         virtual void visit(Precision &) { }
31         virtual void visit(Layout &) { }
32         virtual void visit(InterfaceLayout &) { }
33         virtual void visit(StructDeclaration &) { }
34         virtual void visit(VariableDeclaration &) { }
35         virtual void visit(InterfaceBlock &) { }
36         virtual void visit(FunctionDeclaration &) { }
37         virtual void visit(Conditional &) { }
38         virtual void visit(Iteration &) { }
39         virtual void visit(Passthrough &) { }
40         virtual void visit(Return &) { }
41         virtual void visit(Jump &) { }
42 };
43
44 class TraversingVisitor: public NodeVisitor
45 {
46 protected:
47         Block *current_block;
48
49         TraversingVisitor(): current_block(0) { }
50
51 public:
52         virtual void enter(Block &) { }
53         virtual void visit(Block &);
54         virtual void visit(ParenthesizedExpression &);
55         virtual void visit(MemberAccess &);
56         virtual void visit(UnaryExpression &);
57         virtual void visit(BinaryExpression &);
58         virtual void visit(FunctionCall &);
59         virtual void visit(ExpressionStatement &);
60         virtual void visit(InterfaceLayout &);
61         virtual void visit(StructDeclaration &);
62         virtual void visit(VariableDeclaration &);
63         virtual void visit(InterfaceBlock &);
64         virtual void visit(FunctionDeclaration &);
65         virtual void visit(Conditional &);
66         virtual void visit(Iteration &);
67         virtual void visit(Passthrough &);
68         virtual void visit(Return &);
69 };
70
71 template<typename T>
72 class NodeGatherer: private TraversingVisitor
73 {
74 private:
75         std::vector<T *> nodes;
76
77 public:
78         const std::vector<T *> &apply(Stage &s) { s.content.visit(*this); return nodes; }
79
80 private:
81         virtual void visit(T &n) { nodes.push_back(&n); }
82 };
83
84 class NodeRemover: private TraversingVisitor
85 {
86 private:
87         Stage *stage;
88         const std::set<Node *> *to_remove;
89         std::vector<Block *> blocks;
90         bool anonymous;
91         bool recursive_remove;
92
93 public:
94         NodeRemover();
95
96         void apply(Stage &, const std::set<Node *> &);
97
98 private:
99         void remove_variable(std::map<std::string, VariableDeclaration *> &, VariableDeclaration &);
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