visit_variables.hh

#pragma once
 
#include <clingo/input/statement.hh>
 
#include <functional>
 
namespace CppClingo::Input {
 
 
enum class VariableContext : uint8_t {
    global, 
    all,    
};
 
using VarVisitFun = std::function<void(Location const &loc, String var)>;
 
void visit_variables(Term const &term, VarVisitFun fun);
void visit_variables(TheoryTerm const &term, VarVisitFun fun);
void visit_variables(TheoryElement const &term, VarVisitFun fun);
void visit_variables(Lit const &lit, VarVisitFun fun);
void visit_variables(CondLit const &lit, VarVisitFun fun);
void visit_variables(SetAggregateElement const &elem, VarVisitFun fun);
void visit_variables(HdLitAggregateElement const &elem, VarVisitFun fun);
void visit_variables(HdLit const &lit, VarVisitFun fun, VariableContext ctx);
void visit_variables(BdLitAggregateElement const &elem, VarVisitFun fun);
void visit_variables(BdLit const &lit, VarVisitFun fun, VariableContext ctx);
void visit_variables(OptimizeElement const &elem, VarVisitFun fun);
void visit_variables(Stm const &stm, VarVisitFun fun, VariableContext ctx);
 
template <class T> [[nodiscard]] inline auto select_variables(T const &x, size_t size_hint = 0) -> VariableSet {
    VariableSet vars;
    if (size_hint > 0) {
        vars.reserve(size_hint);
    }
    visit_variables(x, [&]([[maybe_unused]] Location const &loc, String const &var) { vars.emplace(var); });
    return vars;
}
 
template <class T>
[[nodiscard]] inline auto select_variables(T const &x, VariableContext context, size_t size_hint = 0) -> VariableSet {
    VariableSet vars;
    if (size_hint > 0) {
        vars.reserve(size_hint);
    }
    visit_variables(x, [&]([[maybe_unused]] Location const &loc, String const &var) { vars.emplace(var); }, context);
    return vars;
}
 
 
} // namespace CppClingo::Input