X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fglsl%2Foptimize.cpp;h=7bbe14da382b6dbd54fb23ce113187f7e8616de3;hb=ff8a0248f1a3c0c1f48f670867bc9106c898c55b;hp=dfb17b26799bc2845d063ad9618356f2f80be8d9;hpb=05597fbb3671dfed4776bc5223958c85e780345e;p=libs%2Fgl.git

diff --git a/source/glsl/optimize.cpp b/source/glsl/optimize.cpp
index dfb17b26..7bbe14da 100644
--- a/source/glsl/optimize.cpp
+++ b/source/glsl/optimize.cpp
@@ -39,11 +39,16 @@ void InlineableFunctionLocator::visit(FunctionDeclaration &func)
 }
 
 
-FunctionInliner::FunctionInliner(const set<FunctionDeclaration *> &in):
-	inlineable(in),
+FunctionInliner::FunctionInliner():
 	extract_result(0)
 { }
 
+void FunctionInliner::apply(Stage &stage)
+{
+	inlineable = InlineableFunctionLocator().apply(stage);
+	visit(stage.content);
+}
+
 void FunctionInliner::visit_and_inline(RefPtr<Expression> &ptr)
 {
 	inline_result = 0;
@@ -198,14 +203,14 @@ void ConstantConditionEliminator::visit(Iteration &iter)
 }
 
 
-UnusedVariableLocator::VariableInfo::VariableInfo():
+UnusedVariableRemover::VariableInfo::VariableInfo():
 	local(false),
 	conditionally_assigned(false),
 	referenced(false)
 { }
 
 
-UnusedVariableLocator::UnusedVariableLocator():
+UnusedVariableRemover::UnusedVariableRemover():
 	aggregate(0),
 	assignment(0),
 	assignment_target(false),
@@ -213,7 +218,7 @@ UnusedVariableLocator::UnusedVariableLocator():
 	global_scope(true)
 { }
 
-const set<Node *> &UnusedVariableRemover::apply(Stage &stage)
+bool UnusedVariableRemover::apply(Stage &stage)
 {
 	variables.push_back(BlockVariableMap());
 	visit(stage.content);
@@ -230,10 +235,12 @@ const set<Node *> &UnusedVariableRemover::apply(Stage &stage)
 	}
 	variables.pop_back();
 
-	return unused_nodes;
+	NodeRemover().apply(stage, unused_nodes);
+
+	return !unused_nodes.empty();
 }
 
-void UnusedVariableLocator::visit(VariableReference &var)
+void UnusedVariableRemover::visit(VariableReference &var)
 {
 	map<VariableDeclaration *, Node *>::iterator i = aggregates.find(var.declaration);
 	if(i!=aggregates.end())
@@ -247,13 +254,13 @@ void UnusedVariableLocator::visit(VariableReference &var)
 	}
 }
 
-void UnusedVariableLocator::visit(MemberAccess &memacc)
+void UnusedVariableRemover::visit(MemberAccess &memacc)
 {
 	TraversingVisitor::visit(memacc);
 	unused_nodes.erase(memacc.declaration);
 }
 
-void UnusedVariableLocator::visit(BinaryExpression &binary)
+void UnusedVariableRemover::visit(BinaryExpression &binary)
 {
 	if(binary.oper=="[")
 	{
@@ -267,7 +274,7 @@ void UnusedVariableLocator::visit(BinaryExpression &binary)
 		TraversingVisitor::visit(binary);
 }
 
-void UnusedVariableLocator::visit(Assignment &assign)
+void UnusedVariableRemover::visit(Assignment &assign)
 {
 	{
 		assign_to_subscript = false;
@@ -278,7 +285,7 @@ void UnusedVariableLocator::visit(Assignment &assign)
 	assignment = &assign;
 }
 
-void UnusedVariableLocator::record_assignment(VariableDeclaration &var, Node &node, bool chained)
+void UnusedVariableRemover::record_assignment(VariableDeclaration &var, Node &node, bool chained)
 {
 	VariableInfo &var_info = variables.back()[&var];
 	if(!chained)
@@ -287,7 +294,7 @@ void UnusedVariableLocator::record_assignment(VariableDeclaration &var, Node &no
 	var_info.conditionally_assigned = false;
 }
 
-void UnusedVariableLocator::clear_assignments(VariableInfo &var_info, bool mark_unused)
+void UnusedVariableRemover::clear_assignments(VariableInfo &var_info, bool mark_unused)
 {
 	if(mark_unused)
 	{
@@ -297,7 +304,7 @@ void UnusedVariableLocator::clear_assignments(VariableInfo &var_info, bool mark_
 	var_info.assignments.clear();
 }
 
-void UnusedVariableLocator::visit(ExpressionStatement &expr)
+void UnusedVariableRemover::visit(ExpressionStatement &expr)
 {
 	assignment = 0;
 	TraversingVisitor::visit(expr);
@@ -305,14 +312,14 @@ void UnusedVariableLocator::visit(ExpressionStatement &expr)
 		record_assignment(*assignment->target_declaration, expr, (assignment->self_referencing || assign_to_subscript));
 }
 
-void UnusedVariableLocator::visit(StructDeclaration &strct)
+void UnusedVariableRemover::visit(StructDeclaration &strct)
 {
 	SetForScope<Node *> set(aggregate, &strct);
 	unused_nodes.insert(&strct);
 	TraversingVisitor::visit(strct);
 }
 
-void UnusedVariableLocator::visit(VariableDeclaration &var)
+void UnusedVariableRemover::visit(VariableDeclaration &var)
 {
 	if(aggregate)
 		aggregates[&var] = aggregate;
@@ -326,14 +333,14 @@ void UnusedVariableLocator::visit(VariableDeclaration &var)
 	TraversingVisitor::visit(var);
 }
 
-void UnusedVariableLocator::visit(InterfaceBlock &iface)
+void UnusedVariableRemover::visit(InterfaceBlock &iface)
 {
 	SetForScope<Node *> set(aggregate, &iface);
 	unused_nodes.insert(&iface);
 	TraversingVisitor::visit(iface);
 }
 
-void UnusedVariableLocator::visit(FunctionDeclaration &func)
+void UnusedVariableRemover::visit(FunctionDeclaration &func)
 {
 	variables.push_back(BlockVariableMap());
 
@@ -352,7 +359,7 @@ void UnusedVariableLocator::visit(FunctionDeclaration &func)
 	merge_down_variables();
 }
 
-void UnusedVariableLocator::merge_down_variables()
+void UnusedVariableRemover::merge_down_variables()
 {
 	BlockVariableMap &parent_variables = variables[variables.size()-2];
 	BlockVariableMap &block_variables = variables.back();
@@ -381,7 +388,7 @@ void UnusedVariableLocator::merge_down_variables()
 	variables.pop_back();
 }
 
-void UnusedVariableLocator::visit(Conditional &cond)
+void UnusedVariableRemover::visit(Conditional &cond)
 {
 	cond.condition->visit(*this);
 	variables.push_back(BlockVariableMap());
@@ -414,7 +421,7 @@ void UnusedVariableLocator::visit(Conditional &cond)
 	merge_down_variables();
 }
 
-void UnusedVariableLocator::visit(Iteration &iter)
+void UnusedVariableRemover::visit(Iteration &iter)
 {
 	variables.push_back(BlockVariableMap());
 	TraversingVisitor::visit(iter);
@@ -428,7 +435,14 @@ void UnusedVariableLocator::visit(Iteration &iter)
 }
 
 
-void UnusedFunctionLocator::visit(FunctionCall &call)
+bool UnusedFunctionRemover::apply(Stage &stage)
+{
+	visit(stage.content);
+	NodeRemover().apply(stage, unused_nodes);
+	return !unused_nodes.empty();
+}
+
+void UnusedFunctionRemover::visit(FunctionCall &call)
 {
 	TraversingVisitor::visit(call);
 
@@ -437,7 +451,7 @@ void UnusedFunctionLocator::visit(FunctionCall &call)
 		used_definitions.insert(call.declaration->definition);
 }
 
-void UnusedFunctionLocator::visit(FunctionDeclaration &func)
+void UnusedFunctionRemover::visit(FunctionDeclaration &func)
 {
 	TraversingVisitor::visit(func);