60 lines
1.8 KiB
C
60 lines
1.8 KiB
C
|
/*
|
||
|
|
Copyright (c) 2013 Microsoft Corporation. All rights reserved.
|
||
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
||
|
|
|
||
|
|
Author: Leonardo de Moura
|
||
|
|
*/
|
||
|
|
#pragma once
|
||
|
|
#include "util/scoped_map.h"
|
||
|
|
#include "kernel/replace.h"
|
||
|
|
#include "kernel/context.h"
|
||
|
|
|
||
|
|
namespace lean {
|
||
|
|
/**
|
||
|
|
\brief Base class for implementing operations that apply modifications
|
||
|
|
an expressions.
|
||
|
|
The default behavior is a NOOP, users must create subclasses and
|
||
|
|
redefine the visit_* methods.
|
||
|
|
|
||
|
|
This is a more expensive (and flexible) version of replace_fn in
|
||
|
|
the file kernel/replace.h
|
||
|
|
*/
|
||
|
|
class replace_visitor {
|
||
|
|
protected:
|
||
|
|
typedef scoped_map<expr, expr, expr_hash, expr_eqp> cache;
|
||
|
|
cache m_cache;
|
||
|
|
context m_ctx;
|
||
|
|
|
||
|
|
virtual expr visit_type(expr const &, context const &);
|
||
|
|
virtual expr visit_value(expr const &, context const &);
|
||
|
|
virtual expr visit_constant(expr const &, context const &);
|
||
|
|
virtual expr visit_var(expr const &, context const &);
|
||
|
|
virtual expr visit_metavar(expr const &, context const &);
|
||
|
|
virtual expr visit_app(expr const &, context const &);
|
||
|
|
virtual expr visit_eq(expr const &, context const &);
|
||
|
|
virtual expr visit_abst(expr const &, context const &);
|
||
|
|
virtual expr visit_lambda(expr const &, context const &);
|
||
|
|
virtual expr visit_pi(expr const &, context const &);
|
||
|
|
virtual expr visit_let(expr const &, context const &);
|
||
|
|
virtual expr visit(expr const &, context const &);
|
||
|
|
|
||
|
|
void set_ctx(context const & ctx) {
|
||
|
|
if (!is_eqp(m_ctx, ctx)) {
|
||
|
|
m_ctx = ctx;
|
||
|
|
m_cache.clear();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public:
|
||
|
|
expr operator()(expr const & e, context const & ctx = context()) {
|
||
|
|
set_ctx(ctx);
|
||
|
|
return visit(e, ctx);
|
||
|
|
}
|
||
|
|
|
||
|
|
void clear() {
|
||
|
|
m_ctx = context();
|
||
|
|
m_cache.clear();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|