Add basic walking tree parameters.

1 files changed, 14 insertions, 5 deletions

diff --git a/src/analysis/analysis.cpp b/src/analysis/analysis.cpp
index fa1672c..3221ccf 100644
--- a/src/analysis/analysis.cpp
+++ b/src/analysis/analysis.cpp
@@ -22,6 +22,7 @@
 #include "command.h"
 
 #include "analysis/function.h"
+#include "analysis/walkitem.h"
 
 #include "nodes/decl/function_decl.h"
 
@@ -30,23 +31,31 @@
 namespace Analysis
 {
 
-void walkTree(Node *node)
+void startWalkTree(Node *node)
 {
-    analyseNode(node);
+    walkTree(node, WalkItem());
+}
+
+void walkTree(Node *node, WalkItem wi)
+{
+    wi = analyseNode(node, wi);
+    if (wi.stopWalking)
+        return;
 
     FOR_EACH (std::vector<Node*>::iterator, it, node->childs)
     {
-        walkTree(*it);
+        walkTree(*it, wi);
     }
 }
 
-void analyseNode(Node *node)
+WalkItem analyseNode(Node *node, WalkItem wi)
 {
     // searching function declaration
     if (node->nodeType == FUNCTION_DECL)
     {
-        analyseFunction(static_cast<FunctionDeclNode*>(node));
+        return analyseFunction(static_cast<FunctionDeclNode*>(node), wi);
     }
+    return wi;
 }
 
 }