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;
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 TraversingVisitor
60 ExpressionEvaluator::ValueMap variable_values;
61 NodeList<Statement>::iterator insert_point;
62 std::set<Node *> nodes_to_remove;
65 ConstantConditionEliminator();
70 virtual void visit(Block &);
71 virtual void visit(UnaryExpression &);
72 virtual void visit(Assignment &);
73 virtual void visit(VariableDeclaration &);
74 virtual void visit(Conditional &);
75 virtual void visit(Iteration &);
76 using TraversingVisitor::visit;
79 class UnusedVariableRemover: private TraversingVisitor
85 std::vector<Node *> assignments;
86 bool conditionally_assigned;
92 typedef std::map<VariableDeclaration *, VariableInfo> BlockVariableMap;
94 std::set<Node *> unused_nodes;
95 std::map<VariableDeclaration *, Node *> aggregates;
97 std::vector<BlockVariableMap> variables;
98 Assignment *assignment;
99 bool assignment_target;
100 bool assign_to_subscript;
103 UnusedVariableRemover();
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 UnusedFunctionRemover: private TraversingVisitor
128 std::set<Node *> unused_nodes;
129 std::set<FunctionDeclaration *> used_definitions;
132 bool apply(Stage &s);
135 virtual void visit(FunctionCall &);
136 virtual void visit(FunctionDeclaration &);
137 using TraversingVisitor::visit;