lean2/src/kernel/replace_fn.h

27 lines
980 B
C++

/*
Copyright (c) 2013-2014 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Author: Leonardo de Moura
*/
#pragma once
#include <tuple>
#include "util/buffer.h"
#include "util/interrupt.h"
#include "kernel/expr.h"
#include "kernel/expr_maps.h"
namespace lean {
/**
\brief Apply <tt>f</tt> to the subexpressions of a given expression.
f is invoked for each subexpression \c s of the input expression e.
In a call <tt>f(s, n)</tt>, n is the scope level, i.e., the number of
bindings operators that enclosing \c s. The replaces only visits children of \c e
if f return none_expr.
*/
expr replace(expr const & e, std::function<optional<expr>(expr const &, unsigned)> const & f, bool use_cache = true);
inline expr replace(expr const & e, std::function<optional<expr>(expr const &)> const & f, bool use_cache = true) {
return replace(e, [&](expr const & e, unsigned) { return f(e); }, use_cache);
}
}