michael/lean2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(library/relation_manager): add helper procedures

Browse source
This commit is contained in:
Leonardo de Moura 2015-11-14 14:11:52 -08:00
parent 0803db21d3
commit 2ba9979822
2 changed files with 49 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
src/library/relation_manager.cpp
View file

@ -285,6 +285,46 @@ relation_info const * get_relation_info(environment const & env, name const & ro
return rel_ext::get_state(env).m_rop_table.find(rop);
}
relation_info_getter mk_relation_info_getter(environment const & env) {
auto table = rel_ext::get_state(env).m_rop_table;
return [=](name const & rop) {
if (auto r = table.find(rop))
return optional<relation_info>(*r);
else
optional<relation_info>();
};
}
bool is_relation(name_map<relation_info> const & table, expr const & e, name & rop, expr & lhs, expr & rhs) {
if (!is_app(e))
return false;
expr const & f = get_app_fn(e);
if (!is_constant(f))
return false;
auto r = table.find(const_name(f));
if (!r)
return false;
buffer<expr> args;
get_app_args(e, args);
if (r->get_arity() != args.size())
return false;
rop = const_name(f);
lhs = args[r->get_lhs_pos()];
rhs = args[r->get_rhs_pos()];
return true;
}
bool is_relation(environment const & env, expr const & e, name & rop, expr & lhs, expr & rhs) {
return is_relation(rel_ext::get_state(env).m_rop_table, e, rop, lhs, rhs);
}
is_relation_pred mk_is_relation_pred(environment const & env) {
name_map<relation_info> table = rel_ext::get_state(env).m_rop_table;
return [=](expr const & e, name & rop, expr & lhs, expr & rhs) {
return is_relation(table, e, rop, lhs, rhs);
};
}
void initialize_relation_manager() {
g_rel_name = new name("rel");
g_key = new std::string("rel");

9
src/library/relation_manager.h
View file

@ -34,6 +34,15 @@ bool is_equivalence(environment const & env, name const & rop);
relation_info const * get_relation_info(environment const & env, name const & rop);
inline bool is_relation(environment const & env, name const & rop) { return get_relation_info(env, rop) != nullptr; }
typedef std::function<optional<relation_info>(name const &)> relation_info_getter;
relation_info_getter mk_relation_info_getter(environment const & env);
/** \brief Return true iff \c e is of the form (lhs rop rhs) where rop is a registered relation. */
bool is_relation(environment const & env, expr const & e, name & rop, expr & lhs, expr & rhs);
typedef std::function<bool(expr const &, name &, expr &, expr &)> is_relation_pred;
/** \brief Construct an \c is_relation predicate for the given environment. */
is_relation_pred mk_is_relation_pred(environment const & env);
/** \brief Declare a new binary relation named \c n */
environment add_relation(environment const & env, name const & n, bool persistent = true);