collect names util function

loda-lang · Dec 17, 2023 · 12c1f82 · 12c1f82
1 parent 31fa07b
commit 12c1f82
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/form/expression_util.cpp b/src/form/expression_util.cpp
@@ -304,3 +304,13 @@ bool ExpressionUtil::canBeNegative(const Expression& e) {
   }
   return false;
 }
+
+void ExpressionUtil::collectNames(const Expression& e, Expression::Type type,
+                                  std::set<std::string>& target) {
+  if (e.type == type) {
+    target.insert(e.name);
+  }
+  for (auto c : e.children) {
+    collectNames(*c, type, target);
+  }
+}
diff --git a/src/form/expression_util.hpp b/src/form/expression_util.hpp
@@ -1,5 +1,7 @@
 #pragma once
 
+#include <set>
+
 #include "form/expression.hpp"
 
 class ExpressionUtil {
@@ -15,4 +17,7 @@ class ExpressionUtil {
   static bool isSimpleFunction(const Expression& e, bool strict = true);
 
   static bool canBeNegative(const Expression& e);
+
+  static void collectNames(const Expression& e, Expression::Type type,
+                           std::set<std::string>& target);
 };