lean2/src/library/metavar_closure.h

39 lines
1.6 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
*/
#pragma once
#include "util/rb_tree.h"
#include "kernel/metavar.h"
#include "library/expr_lt.h"
namespace lean {
/** \brief Helper object for collecting metavariables occurring in expressions.
It also provides a method that given a substitution \c s, produces a constraint
?m =?= s(?m)
for each collected metavariable ?m that is assigned in \c s.
\remark Metavariables occuring in the type of other metavariables are ignored.
*/
class metavar_closure {
rb_tree<expr, expr_quick_cmp> m_expr_mvars;
rb_tree<level, level_quick_cmp> m_lvl_mvars;
public:
metavar_closure() {}
metavar_closure(expr const & e) { add(e); }
/** \brief Collect metavariables occurring in \c e */
void add(expr const & e);
void add(level const & l);
/** \brief Execute \c fn for each collected metavariable */
void for_each_expr_mvar(std::function<void(expr const &)> const & fn) const;
void for_each_level_mvar(std::function<void(level const &)> const & fn) const;
/** \brief For each collected metavariable ?m, store in \c r constraints of the form ?m =?= s(?m)
if ?m is assigned by \c s. The constraints are justified by \c j and \c relax
\see mk_eq_cnstr
*/
void mk_constraints(substitution s, justification const & j, bool relax, buffer<constraint> & r) const;
constraints mk_constraints(substitution s, justification const & j, bool relax) const;
};
}