64 lines
2 KiB
C++
64 lines
2 KiB
C++
/*
|
|
Copyright (c) 2014 Microsoft Corporation. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
|
|
Author: Leonardo de Moura
|
|
*/
|
|
#include "kernel/for_each_fn.h"
|
|
#include "library/metavar_closure.h"
|
|
|
|
namespace lean {
|
|
void metavar_closure::add(level const & l) {
|
|
for_each(l, [&](level const & l) {
|
|
if (is_meta(l)) {
|
|
m_lvl_mvars.insert(l);
|
|
return false;
|
|
} else {
|
|
return has_meta(l);
|
|
}
|
|
});
|
|
}
|
|
|
|
void metavar_closure::add(expr const & e) {
|
|
for_each(e, [&](expr const & e, unsigned) {
|
|
if (is_metavar(e)) {
|
|
m_expr_mvars.insert(e);
|
|
return false; /* do not visit its type */
|
|
} else if (is_sort(e)) {
|
|
add(sort_level(e));
|
|
return false;
|
|
} else if (is_constant(e)) {
|
|
for (level const & l : const_levels(e))
|
|
add(l);
|
|
return false;
|
|
} else {
|
|
return has_metavar(e);
|
|
}
|
|
});
|
|
}
|
|
|
|
void metavar_closure::for_each_expr_mvar(std::function<void(expr const &)> const & fn) const {
|
|
m_expr_mvars.for_each(fn);
|
|
}
|
|
|
|
void metavar_closure::for_each_level_mvar(std::function<void(level const &)> const & fn) const {
|
|
m_lvl_mvars.for_each(fn);
|
|
}
|
|
|
|
void metavar_closure::mk_constraints(substitution s, justification const & j, buffer<constraint> & r) const {
|
|
m_expr_mvars.for_each([&](expr const & m) {
|
|
if (s.is_expr_assigned(mlocal_name(m)))
|
|
r.push_back(mk_eq_cnstr(m, s.instantiate(m), j));
|
|
});
|
|
m_lvl_mvars.for_each([&](level const & m) {
|
|
if (s.is_level_assigned(meta_id(m)))
|
|
r.push_back(mk_level_eq_cnstr(m, s.instantiate(m), j));
|
|
});
|
|
}
|
|
|
|
constraints metavar_closure::mk_constraints(substitution s, justification const & j) const {
|
|
buffer<constraint> cs;
|
|
mk_constraints(s, j, cs);
|
|
return to_list(cs.begin(), cs.end());
|
|
}
|
|
}
|