1 #ifndef MSP_GL_SL_OPTIMIZE_H_
2 #define MSP_GL_SL_OPTIMIZE_H_
13 class InlineableFunctionLocator: private TraversingVisitor
16 std::map<FunctionDeclaration *, unsigned> refcounts;
17 std::set<FunctionDeclaration *> inlineable;
18 FunctionDeclaration *in_function;
21 InlineableFunctionLocator();
23 const std::set<FunctionDeclaration *> &apply(Stage &s) { visit(s.content); return inlineable; }
26 virtual void visit(FunctionCall &);
27 virtual void visit(FunctionDeclaration &);
28 using TraversingVisitor::visit;
31 class FunctionInliner: private TraversingVisitor
34 std::set<FunctionDeclaration *> inlineable;
35 unsigned extract_result;
36 RefPtr<Expression> inline_result;
39 FunctionInliner(const std::set<FunctionDeclaration *> &);
41 void apply(Stage &s) { visit(s.content); }
44 void visit_and_inline(RefPtr<Expression> &);
46 virtual void visit(Block &);
47 virtual void visit(UnaryExpression &);
48 virtual void visit(BinaryExpression &);
49 virtual void visit(MemberAccess &);
50 virtual void visit(FunctionCall &);
51 virtual void visit(VariableDeclaration &);
52 virtual void visit(Return &);
53 using TraversingVisitor::visit;
56 class ConstantConditionEliminator: private BlockModifier
61 ExpressionEvaluator::ValueMap variable_values;
64 ConstantConditionEliminator();
66 void apply(Stage &s) { visit(s.content); }
69 virtual void visit(Block &);
70 virtual void visit(UnaryExpression &);
71 virtual void visit(Assignment &);
72 virtual void visit(VariableDeclaration &);
73 virtual void visit(Conditional &);
74 virtual void visit(Iteration &);
75 using BlockModifier::visit;
78 class UnusedVariableLocator: private TraversingVisitor
84 std::vector<Node *> assignments;
85 bool conditionally_assigned;
91 typedef std::map<VariableDeclaration *, VariableInfo> BlockVariableMap;
93 std::set<Node *> unused_nodes;
94 std::map<VariableDeclaration *, Node *> aggregates;
96 std::vector<BlockVariableMap> variables;
97 Assignment *assignment;
98 bool assignment_target;
99 bool assign_to_subscript;
103 UnusedVariableLocator();
105 const std::set<Node *> &apply(Stage &);
108 virtual void visit(VariableReference &);
109 virtual void visit(MemberAccess &);
110 virtual void visit(BinaryExpression &);
111 virtual void visit(Assignment &);
112 void record_assignment(VariableDeclaration &, Node &, bool);
113 void clear_assignments(VariableInfo &, bool);
114 virtual void visit(ExpressionStatement &);
115 virtual void visit(StructDeclaration &);
116 virtual void visit(VariableDeclaration &);
117 virtual void visit(InterfaceBlock &);
118 virtual void visit(FunctionDeclaration &);
119 void merge_down_variables();
120 virtual void visit(Conditional &);
121 virtual void visit(Iteration &);
122 using TraversingVisitor::visit;
125 class UnusedFunctionLocator: private TraversingVisitor
128 std::set<Node *> unused_nodes;
129 std::set<FunctionDeclaration *> used_definitions;
132 const std::set<Node *> &apply(Stage &s) { visit(s.content); return unused_nodes; }
135 virtual void visit(FunctionCall &);
136 virtual void visit(FunctionDeclaration &);
137 using TraversingVisitor::visit;