From c72c4ebe384a0d5a48c268d09449707bd86090a8 Mon Sep 17 00:00:00 2001
From: Mikko Rasa <tdb@tdb.fi>
Date: Fri, 9 Apr 2021 00:52:06 +0300
Subject: [PATCH] Visit the entire call graph in DependencyCollector

---
 source/glsl/reflect.cpp | 12 ++++++++++++
 source/glsl/reflect.h   |  2 ++
 2 files changed, 14 insertions(+)

diff --git a/source/glsl/reflect.cpp b/source/glsl/reflect.cpp
index e6355316..c38e3aec 100644
--- a/source/glsl/reflect.cpp
+++ b/source/glsl/reflect.cpp
@@ -279,7 +279,10 @@ void DependencyCollector::visit(InterfaceBlockReference &iface)
 void DependencyCollector::visit(FunctionCall &call)
 {
 	if(call.declaration)
+	{
 		dependencies.insert(call.declaration);
+		call.declaration->visit(*this);
+	}
 	TraversingVisitor::visit(call);
 }
 
@@ -295,6 +298,15 @@ void DependencyCollector::visit(VariableDeclaration &var)
 	TraversingVisitor::visit(var);
 }
 
+void DependencyCollector::visit(FunctionDeclaration &func)
+{
+	if(!visited_functions.count(&func))
+	{
+		visited_functions.insert(&func);
+		TraversingVisitor::visit(func);
+	}
+}
+
 } // namespace SL
 } // namespace GL
 } // namespace Msp
diff --git a/source/glsl/reflect.h b/source/glsl/reflect.h
index 56ead82c..c2d7aa04 100644
--- a/source/glsl/reflect.h
+++ b/source/glsl/reflect.h
@@ -85,6 +85,7 @@ class DependencyCollector: private TraversingVisitor
 private:
 	std::set<Node *> dependencies;
 	std::set<Node *> locals;
+	std::set<FunctionDeclaration *> visited_functions;
 
 public:
 	std::set<Node *> apply(FunctionDeclaration &);
@@ -94,6 +95,7 @@ private:
 	virtual void visit(InterfaceBlockReference &);
 	virtual void visit(FunctionCall &);
 	virtual void visit(VariableDeclaration &);
+	virtual void visit(FunctionDeclaration &);
 };
 
 } // namespace SL
-- 
2.43.0