23988f528c
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
538 lines
19 KiB
C++
538 lines
19 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
|
|
Soonho Kong
|
|
*/
|
|
#include <vector>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <algorithm>
|
|
#include "util/list_fn.h"
|
|
#include "util/hash.h"
|
|
#include "util/buffer.h"
|
|
#include "util/object_serializer.h"
|
|
#include "kernel/expr.h"
|
|
#include "kernel/expr_eq_fn.h"
|
|
// #include "kernel/free_vars.h"
|
|
// #include "kernel/metavar.h"
|
|
// #include "kernel/max_sharing.h"
|
|
|
|
namespace lean {
|
|
static expr g_dummy(mk_var(0));
|
|
expr::expr():expr(g_dummy) {}
|
|
|
|
unsigned hash_levels(levels const & ls) {
|
|
unsigned r = 23;
|
|
for (auto const & l : ls)
|
|
r = hash(hash(l), r);
|
|
return r;
|
|
}
|
|
|
|
bool has_meta(levels const & ls) {
|
|
for (auto const & l : ls) {
|
|
if (has_meta(l))
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
expr_cell::expr_cell(expr_kind k, unsigned h, bool has_mv):
|
|
m_kind(static_cast<unsigned>(k)),
|
|
m_flags(has_mv ? 4 : 0),
|
|
m_hash(h),
|
|
m_rc(0) {
|
|
// m_hash_alloc does not need to be a unique identifier.
|
|
// We want diverse hash codes because given expr_cell * c1 and expr_cell * c2,
|
|
// if c1 != c2, then there is high probability c1->m_hash_alloc != c2->m_hash_alloc.
|
|
// Remark: using pointer address as a hash code is not a good idea.
|
|
// - each execution run will behave differently.
|
|
// - the hash is not diverse enough
|
|
static LEAN_THREAD_LOCAL unsigned g_hash_alloc_counter = 0;
|
|
m_hash_alloc = g_hash_alloc_counter;
|
|
g_hash_alloc_counter++;
|
|
}
|
|
|
|
void expr_cell::dec_ref(expr & e, buffer<expr_cell*> & todelete) {
|
|
if (e.m_ptr) {
|
|
expr_cell * c = e.steal_ptr();
|
|
lean_assert(!(e.m_ptr));
|
|
if (c->dec_ref_core())
|
|
todelete.push_back(c);
|
|
}
|
|
}
|
|
|
|
void expr_cell::dec_ref(optional<expr> & c, buffer<expr_cell*> & todelete) {
|
|
if (c)
|
|
dec_ref(*c, todelete);
|
|
}
|
|
|
|
optional<bool> expr_cell::is_arrow() const {
|
|
// it is stored in bits 3-4
|
|
unsigned r = (m_flags & (8+16)) >> 3;
|
|
if (r == 0) {
|
|
return optional<bool>();
|
|
} else if (r == 1) {
|
|
return optional<bool>(true);
|
|
} else {
|
|
lean_assert(r == 2);
|
|
return optional<bool>(false);
|
|
}
|
|
}
|
|
|
|
void expr_cell::set_is_arrow(bool flag) {
|
|
unsigned mask = flag ? 8 : 16;
|
|
m_flags |= mask;
|
|
lean_assert(is_arrow() && *is_arrow() == flag);
|
|
}
|
|
|
|
// Expr variables
|
|
expr_var::expr_var(unsigned idx):
|
|
expr_cell(expr_kind::Var, idx, false),
|
|
m_vidx(idx) {}
|
|
|
|
// Expr constants
|
|
expr_const::expr_const(name const & n, levels const & ls):
|
|
expr_cell(expr_kind::Constant, ::lean::hash(n.hash(), hash_levels(ls)), has_meta(ls)),
|
|
m_name(n),
|
|
m_levels(ls) {}
|
|
|
|
// Expr metavariables and local variables
|
|
expr_mlocal::expr_mlocal(bool is_meta, name const & n, expr const & t):
|
|
expr_cell(is_meta ? expr_kind::Meta : expr_kind::Local, n.hash(), t.has_metavar()),
|
|
m_name(n),
|
|
m_type(t) {}
|
|
void expr_mlocal::dealloc(buffer<expr_cell*> & todelete) {
|
|
dec_ref(m_type, todelete);
|
|
delete(this);
|
|
}
|
|
|
|
// Composite expressions
|
|
expr_composite::expr_composite(expr_kind k, unsigned h, bool has_mv, unsigned d):
|
|
expr_cell(k, h, has_mv),
|
|
m_depth(d) {}
|
|
|
|
// Expr dependent pairs
|
|
expr_dep_pair::expr_dep_pair(expr const & f, expr const & s, expr const & t):
|
|
expr_composite(expr_kind::Pair, ::lean::hash(f.hash(), s.hash()), f.has_metavar() || s.has_metavar() || t.has_metavar(),
|
|
std::max(get_depth(f), get_depth(s))+1),
|
|
m_first(f), m_second(s), m_type(t) {
|
|
}
|
|
void expr_dep_pair::dealloc(buffer<expr_cell*> & todelete) {
|
|
dec_ref(m_first, todelete);
|
|
dec_ref(m_second, todelete);
|
|
dec_ref(m_type, todelete);
|
|
delete(this);
|
|
}
|
|
|
|
// Expr pair projection
|
|
expr_proj::expr_proj(bool f, expr const & e):
|
|
expr_composite(f ? expr_kind::Fst : expr_kind::Snd, ::lean::hash(17, e.hash()), e.has_metavar(), get_depth(e)+1),
|
|
m_expr(e) {}
|
|
void expr_proj::dealloc(buffer<expr_cell*> & todelete) {
|
|
dec_ref(m_expr, todelete);
|
|
delete(this);
|
|
}
|
|
|
|
// Expr applications
|
|
expr_app::expr_app(expr const & fn, expr const & arg):
|
|
expr_composite(expr_kind::App, ::lean::hash(fn.hash(), arg.hash()), fn.has_metavar() || arg.has_metavar(),
|
|
std::max(get_depth(fn), get_depth(arg)) + 1),
|
|
m_fn(fn), m_arg(arg) {}
|
|
void expr_app::dealloc(buffer<expr_cell*> & todelete) {
|
|
dec_ref(m_fn, todelete);
|
|
dec_ref(m_arg, todelete);
|
|
delete(this);
|
|
}
|
|
|
|
// Expr binders (Lambda, Pi and Sigma)
|
|
expr_binder::expr_binder(expr_kind k, name const & n, expr const & t, expr const & b):
|
|
expr_composite(k, ::lean::hash(t.hash(), b.hash()), t.has_metavar() || b.has_metavar(),
|
|
std::max(get_depth(m_domain), get_depth(m_body)) + 1),
|
|
m_name(n),
|
|
m_domain(t),
|
|
m_body(b) {
|
|
lean_assert(k == expr_kind::Lambda || k == expr_kind::Pi || k == expr_kind::Sigma);
|
|
}
|
|
void expr_binder::dealloc(buffer<expr_cell*> & todelete) {
|
|
dec_ref(m_body, todelete);
|
|
dec_ref(m_domain, todelete);
|
|
delete(this);
|
|
}
|
|
|
|
// Expr Sort
|
|
expr_sort::expr_sort(level const & l):
|
|
expr_cell(expr_kind::Sort, ::lean::hash(l), has_meta(l)),
|
|
m_level(l) {
|
|
}
|
|
expr_sort::~expr_sort() {}
|
|
|
|
// Expr Let
|
|
expr_let::expr_let(name const & n, optional<expr> const & t, expr const & v, expr const & b):
|
|
expr_composite(expr_kind::Let, ::lean::hash(v.hash(), b.hash()), v.has_metavar() || b.has_metavar() || ::lean::has_metavar(t),
|
|
std::max({get_depth(t), get_depth(v), get_depth(b)}) + 1),
|
|
m_name(n),
|
|
m_type(t),
|
|
m_value(v),
|
|
m_body(b) {
|
|
}
|
|
void expr_let::dealloc(buffer<expr_cell*> & todelete) {
|
|
dec_ref(m_body, todelete);
|
|
dec_ref(m_value, todelete);
|
|
dec_ref(m_type, todelete);
|
|
delete(this);
|
|
}
|
|
expr_let::~expr_let() {}
|
|
|
|
// Macro attachment
|
|
int macro::push_lua(lua_State *) const { return 0; } // NOLINT
|
|
void macro::display(std::ostream & out) const { out << get_name(); }
|
|
bool macro::operator==(macro const & other) const { return typeid(*this) == typeid(other); }
|
|
bool macro::operator<(macro const & other) const {
|
|
if (get_name() == other.get_name())
|
|
return lt(other);
|
|
else
|
|
return get_name() < other.get_name();
|
|
}
|
|
format macro::pp(formatter const &, options const &) const { return format(get_name()); }
|
|
bool macro::is_atomic_pp(bool, bool) const { return true; }
|
|
unsigned macro::hash() const { return get_name().hash(); }
|
|
|
|
typedef std::unordered_map<std::string, macro::reader> macro_readers;
|
|
static std::unique_ptr<macro_readers> g_macro_readers;
|
|
macro_readers & get_macro_readers() {
|
|
if (!g_macro_readers)
|
|
g_macro_readers.reset(new macro_readers());
|
|
return *(g_macro_readers.get());
|
|
}
|
|
|
|
void macro::register_deserializer(std::string const & k, macro::reader rd) {
|
|
macro_readers & readers = get_macro_readers();
|
|
lean_assert(readers.find(k) == readers.end());
|
|
readers[k] = rd;
|
|
}
|
|
static expr read_macro(deserializer & d) {
|
|
auto k = d.read_string();
|
|
macro_readers & readers = get_macro_readers();
|
|
auto it = readers.find(k);
|
|
lean_assert(it != readers.end());
|
|
return it->second(d);
|
|
}
|
|
|
|
expr_macro::expr_macro(macro * m):
|
|
expr_cell(expr_kind::Macro, m->hash(), false),
|
|
m_macro(m) {
|
|
m_macro->inc_ref();
|
|
}
|
|
expr_macro::~expr_macro() {
|
|
m_macro->dec_ref();
|
|
}
|
|
|
|
void expr_cell::dealloc() {
|
|
try {
|
|
buffer<expr_cell*> todo;
|
|
todo.push_back(this);
|
|
while (!todo.empty()) {
|
|
expr_cell * it = todo.back();
|
|
todo.pop_back();
|
|
lean_assert(it->get_rc() == 0);
|
|
switch (it->kind()) {
|
|
case expr_kind::Var: delete static_cast<expr_var*>(it); break;
|
|
case expr_kind::Macro: delete static_cast<expr_macro*>(it); break;
|
|
case expr_kind::Meta:
|
|
case expr_kind::Local: static_cast<expr_mlocal*>(it)->dealloc(todo); break;
|
|
case expr_kind::Constant: delete static_cast<expr_const*>(it); break;
|
|
case expr_kind::Sort: delete static_cast<expr_sort*>(it); break;
|
|
case expr_kind::Pair: static_cast<expr_dep_pair*>(it)->dealloc(todo); break;
|
|
case expr_kind::Fst:
|
|
case expr_kind::Snd: static_cast<expr_proj*>(it)->dealloc(todo); break;
|
|
case expr_kind::App: static_cast<expr_app*>(it)->dealloc(todo); break;
|
|
case expr_kind::Lambda:
|
|
case expr_kind::Pi:
|
|
case expr_kind::Sigma: static_cast<expr_binder*>(it)->dealloc(todo); break;
|
|
case expr_kind::Let: static_cast<expr_let*>(it)->dealloc(todo); break;
|
|
}
|
|
}
|
|
} catch (std::bad_alloc&) {
|
|
// We need this catch, because push_back may fail when expanding the buffer.
|
|
// In this case, we avoid the crash, and "accept" the memory leak.
|
|
}
|
|
}
|
|
|
|
// Auxiliary constructors
|
|
expr mk_app(unsigned num_args, expr const * args) {
|
|
lean_assert(num_args >= 2);
|
|
expr r = mk_app(args[0], args[1]);
|
|
for (unsigned i = 2; i < num_args; i++)
|
|
r = mk_app(r, args[i]);
|
|
return r;
|
|
}
|
|
|
|
static name g_default_var_name("a");
|
|
bool is_default_var_name(name const & n) { return n == g_default_var_name; }
|
|
expr mk_arrow(expr const & t, expr const & e) { return mk_pi(g_default_var_name, t, e); }
|
|
expr mk_cartesian_product(expr const & t, expr const & e) { return mk_sigma(g_default_var_name, t, e); }
|
|
|
|
expr Bool = mk_sort(mk_level_zero());
|
|
expr Type = mk_sort(mk_level_one());
|
|
expr mk_Bool() { return Bool; }
|
|
expr mk_Type() { return Type; }
|
|
|
|
unsigned get_depth(expr const & e) {
|
|
switch (e.kind()) {
|
|
case expr_kind::Var: case expr_kind::Constant: case expr_kind::Sort:
|
|
case expr_kind::Meta: case expr_kind::Local: case expr_kind::Macro:
|
|
return 1;
|
|
case expr_kind::Lambda: case expr_kind::Pi: case expr_kind::Sigma:
|
|
case expr_kind::Pair: case expr_kind::Fst: case expr_kind::Snd:
|
|
case expr_kind::App: case expr_kind::Let:
|
|
return static_cast<expr_composite*>(e.raw())->m_depth;
|
|
}
|
|
lean_unreachable(); // LCOV_EXCL_LINE
|
|
}
|
|
|
|
bool operator==(expr const & a, expr const & b) { return expr_eq_fn()(a, b); }
|
|
|
|
#if 0
|
|
|
|
|
|
|
|
bool is_arrow(expr const & t) {
|
|
optional<bool> r = t.raw()->is_arrow();
|
|
if (r) {
|
|
return *r;
|
|
} else {
|
|
bool res = is_pi(t) && !has_free_var(abst_body(t), 0);
|
|
t.raw()->set_is_arrow(res);
|
|
return res;
|
|
}
|
|
}
|
|
|
|
bool is_cartesian(expr const & t) {
|
|
return is_sigma(t) && !has_free_var(abst_body(t), 0);
|
|
}
|
|
|
|
|
|
expr copy(expr const & a) {
|
|
switch (a.kind()) {
|
|
case expr_kind::Var: return mk_var(var_idx(a));
|
|
case expr_kind::Constant: return mk_constant(const_name(a), const_type(a));
|
|
case expr_kind::Type: return mk_type(ty_level(a));
|
|
case expr_kind::Value: return mk_value(static_cast<expr_value*>(a.raw())->m_val);
|
|
case expr_kind::Pair: return mk_pair(pair_first(a), pair_second(a), pair_type(a));
|
|
case expr_kind::Proj: return mk_proj(proj_first(a), proj_arg(a));
|
|
case expr_kind::App: return mk_app(num_args(a), begin_args(a));
|
|
case expr_kind::Lambda: return mk_lambda(abst_name(a), abst_domain(a), abst_body(a));
|
|
case expr_kind::Pi: return mk_pi(abst_name(a), abst_domain(a), abst_body(a));
|
|
case expr_kind::Sigma: return mk_sigma(abst_name(a), abst_domain(a), abst_body(a));
|
|
case expr_kind::Let: return mk_let(let_name(a), let_type(a), let_value(a), let_body(a));
|
|
case expr_kind::HEq: return mk_heq(heq_lhs(a), heq_rhs(a));
|
|
case expr_kind::MetaVar: return mk_metavar(metavar_name(a), metavar_lctx(a));
|
|
}
|
|
lean_unreachable(); // LCOV_EXCL_LINE
|
|
}
|
|
|
|
serializer & operator<<(serializer & s, local_context const & lctx) {
|
|
s << length(lctx);
|
|
for (auto const & e : lctx) {
|
|
if (e.is_lift()) {
|
|
s << true << e.s() << e.n();
|
|
} else {
|
|
s << false << e.s() << e.v();
|
|
}
|
|
}
|
|
return s;
|
|
}
|
|
|
|
local_context read_local_context(deserializer & d) {
|
|
unsigned num = d.read_unsigned();
|
|
buffer<local_entry> entries;
|
|
for (unsigned i = 0; i < num; i++) {
|
|
if (d.read_bool()) {
|
|
unsigned s, n;
|
|
d >> s >> n;
|
|
entries.push_back(mk_lift(s, n));
|
|
} else {
|
|
unsigned s; expr v;
|
|
d >> s >> v;
|
|
entries.push_back(mk_inst(s, v));
|
|
}
|
|
}
|
|
return to_list(entries.begin(), entries.end());
|
|
}
|
|
|
|
// To save space, we pack the number of arguments in small applications in the kind byte.
|
|
// The extra kinds start at g_first_app_size_kind. This value should be bigger than the
|
|
// real kinds. Moreover g_first_app_size_kind + g_small_app_num_args should fit in a byte.
|
|
constexpr char g_first_app_size_kind = 32;
|
|
constexpr char g_small_app_num_args = 32;
|
|
constexpr bool is_small(expr_kind k) { return 0 <= static_cast<char>(k) && static_cast<char>(k) < g_first_app_size_kind; }
|
|
static_assert(is_small(expr_kind::Var) && is_small(expr_kind::Constant) && is_small(expr_kind::Value) && is_small(expr_kind::App) &&
|
|
is_small(expr_kind::Pair) && is_small(expr_kind::Proj) &&
|
|
is_small(expr_kind::Lambda) && is_small(expr_kind::Pi) && is_small(expr_kind::Sigma) && is_small(expr_kind::Type) &&
|
|
is_small(expr_kind::Let) && is_small(expr_kind::HEq) && is_small(expr_kind::MetaVar), "expr_kind is too big");
|
|
|
|
class expr_serializer : public object_serializer<expr, expr_hash_alloc, expr_eqp> {
|
|
typedef object_serializer<expr, expr_hash_alloc, expr_eqp> super;
|
|
max_sharing_fn m_max_sharing_fn;
|
|
|
|
void write_core(optional<expr> const & a) {
|
|
serializer & s = get_owner();
|
|
if (a) {
|
|
s << true;
|
|
write_core(*a);
|
|
} else {
|
|
s << false;
|
|
}
|
|
}
|
|
|
|
void write_core(expr const & a) {
|
|
auto k = a.kind();
|
|
char kc;
|
|
if (k == expr_kind::App && num_args(a) < g_small_app_num_args) {
|
|
kc = static_cast<char>(g_first_app_size_kind + num_args(a));
|
|
} else {
|
|
kc = static_cast<char>(k);
|
|
}
|
|
super::write_core(a, kc, [&]() {
|
|
serializer & s = get_owner();
|
|
if (kc >= static_cast<char>(g_first_app_size_kind)) {
|
|
// compressed application
|
|
for (unsigned i = 0; i < num_args(a); i++)
|
|
write_core(arg(a, i));
|
|
return;
|
|
}
|
|
switch (k) {
|
|
case expr_kind::Var: s << var_idx(a); break;
|
|
case expr_kind::Constant: s << const_name(a); write_core(const_type(a)); break;
|
|
case expr_kind::Type: s << ty_level(a); break;
|
|
case expr_kind::Value: to_value(a).write(s); break;
|
|
case expr_kind::Pair: write_core(pair_first(a)); write_core(pair_second(a)); write_core(pair_type(a)); break;
|
|
case expr_kind::Proj: s << proj_first(a); write_core(proj_arg(a)); break;
|
|
case expr_kind::HEq: write_core(heq_lhs(a)); write_core(heq_rhs(a)); break;
|
|
case expr_kind::App:
|
|
s << num_args(a);
|
|
for (unsigned i = 0; i < num_args(a); i++)
|
|
write_core(arg(a, i));
|
|
break;
|
|
case expr_kind::Lambda:
|
|
case expr_kind::Pi:
|
|
case expr_kind::Sigma: s << abst_name(a); write_core(abst_domain(a)); write_core(abst_body(a)); break;
|
|
case expr_kind::Let: s << let_name(a); write_core(let_type(a)); write_core(let_value(a)); write_core(let_body(a)); break;
|
|
case expr_kind::MetaVar: s << metavar_name(a) << metavar_lctx(a); break;
|
|
}
|
|
});
|
|
}
|
|
public:
|
|
void write(expr const & a) {
|
|
write_core(m_max_sharing_fn(a));
|
|
}
|
|
};
|
|
|
|
class expr_deserializer : public object_deserializer<expr> {
|
|
typedef object_deserializer<expr> super;
|
|
public:
|
|
optional<expr> read_opt() {
|
|
deserializer & d = get_owner();
|
|
if (d.read_bool()) {
|
|
return some_expr(read());
|
|
} else {
|
|
return none_expr();
|
|
}
|
|
}
|
|
|
|
expr read() {
|
|
return super::read_core([&](char c) {
|
|
deserializer & d = get_owner();
|
|
if (c >= g_first_app_size_kind) {
|
|
// compressed application
|
|
unsigned num = c - g_first_app_size_kind;
|
|
buffer<expr> args;
|
|
for (unsigned i = 0; i < num; i++)
|
|
args.push_back(read());
|
|
return mk_app(args);
|
|
}
|
|
auto k = static_cast<expr_kind>(c);
|
|
switch (k) {
|
|
case expr_kind::Var:
|
|
return mk_var(d.read_unsigned());
|
|
case expr_kind::Constant: {
|
|
auto n = read_name(d);
|
|
return mk_constant(n, read_opt());
|
|
}
|
|
case expr_kind::Type:
|
|
return mk_type(read_level(d));
|
|
break;
|
|
case expr_kind::Value:
|
|
return read_value(d);
|
|
case expr_kind::Pair: {
|
|
expr f = read();
|
|
expr s = read();
|
|
return mk_pair(f, s, read());
|
|
}
|
|
case expr_kind::HEq: {
|
|
expr lhs = read();
|
|
return mk_heq(lhs, read());
|
|
}
|
|
case expr_kind::Proj: {
|
|
bool f = d.read_bool();
|
|
return mk_proj(f, read());
|
|
}
|
|
case expr_kind::App: {
|
|
buffer<expr> args;
|
|
unsigned num = d.read_unsigned();
|
|
for (unsigned i = 0; i < num; i++)
|
|
args.push_back(read());
|
|
return mk_app(args);
|
|
}
|
|
case expr_kind::Lambda: {
|
|
name n = read_name(d);
|
|
expr d = read();
|
|
return mk_lambda(n, d, read());
|
|
}
|
|
case expr_kind::Pi: {
|
|
name n = read_name(d);
|
|
expr d = read();
|
|
return mk_pi(n, d, read());
|
|
}
|
|
case expr_kind::Sigma: {
|
|
name n = read_name(d);
|
|
expr d = read();
|
|
return mk_sigma(n, d, read());
|
|
}
|
|
case expr_kind::Let: {
|
|
name n = read_name(d);
|
|
optional<expr> t = read_opt();
|
|
expr v = read();
|
|
return mk_let(n, t, v, read());
|
|
}
|
|
case expr_kind::MetaVar: {
|
|
name n = read_name(d);
|
|
return mk_metavar(n, read_local_context(d));
|
|
}}
|
|
throw_corrupted_file(); // LCOV_EXCL_LINE
|
|
});
|
|
}
|
|
};
|
|
|
|
struct expr_sd {
|
|
unsigned m_s_extid;
|
|
unsigned m_d_extid;
|
|
expr_sd() {
|
|
m_s_extid = serializer::register_extension([](){ return std::unique_ptr<serializer::extension>(new expr_serializer()); });
|
|
m_d_extid = deserializer::register_extension([](){ return std::unique_ptr<deserializer::extension>(new expr_deserializer()); });
|
|
}
|
|
};
|
|
static expr_sd g_expr_sd;
|
|
|
|
serializer & operator<<(serializer & s, expr const & n) {
|
|
s.get_extension<expr_serializer>(g_expr_sd.m_s_extid).write(n);
|
|
return s;
|
|
}
|
|
|
|
expr read_expr(deserializer & d) {
|
|
return d.get_extension<expr_deserializer>(g_expr_sd.m_d_extid).read();
|
|
}
|
|
#endif
|
|
}
|