roland/xo-alloc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

xo-expression: + Expression::get_free_variables()

Browse source
This commit is contained in:
Roland Conybeare 2024-07-01 16:50:17 -04:00
commit b7db7c5454
7 changed files with 59 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
include/xo/expression/Apply.hpp
View file

@ -36,6 +36,20 @@ namespace xo {
const ref::rp<Expression> & fn() const { return fn_; }
const std::vector<ref::rp<Expression>> & argv() const { return argv_; }
virtual std::set<std::string> get_free_variables() const override {
std::set<std::string> retval = fn_->get_free_variables();
for (const auto & arg : argv_) {
std::set<std::string> arg_free_set
= arg->get_free_variables();
for (const auto & name : arg_free_set)
retval.insert(name);
}
return retval;
}
virtual std::size_t visit_preorder(VisitFn visitor_fn) override {
std::size_t n = 1;