lean4-htt/src/kernel/declaration.cpp
2020-01-16 17:20:36 -08:00

289 lines
12 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/declaration.h"
#include "kernel/environment.h"
#include "kernel/for_each_fn.h"
namespace lean {
int compare(reducibility_hints const & h1, reducibility_hints const & h2) {
if (h1.kind() == h2.kind()) {
if (h1.kind() == reducibility_hints_kind::Regular) {
if (h1.get_height() == h2.get_height())
return 0; /* unfold both */
else if (h1.get_height() > h2.get_height())
return -1; /* unfold f1 */
else
return 1; /* unfold f2 */
return h1.get_height() > h2.get_height() ? -1 : 1;
} else {
return 0; /* reduce both */
}
} else {
if (h1.kind() == reducibility_hints_kind::Opaque) {
return 1; /* reduce f2 */
} else if (h2.kind() == reducibility_hints_kind::Opaque) {
return -1; /* reduce f1 */
} else if (h1.kind() == reducibility_hints_kind::Abbreviation) {
return -1; /* reduce f1 */
} else if (h2.kind() == reducibility_hints_kind::Abbreviation) {
return 1; /* reduce f2 */
} else {
lean_unreachable();
}
}
}
constant_val::constant_val(name const & n, names const & lparams, expr const & type):
object_ref(mk_cnstr(0, n, lparams, type)) {
}
axiom_val::axiom_val(name const & n, names const & lparams, expr const & type, bool is_unsafe):
object_ref(mk_cnstr(0, constant_val(n, lparams, type), 1)) {
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*), static_cast<unsigned char>(is_unsafe));
}
bool axiom_val::is_unsafe() const { return cnstr_get_scalar<unsigned char>(raw(), sizeof(object*)) != 0; }
definition_val::definition_val(name const & n, names const & lparams, expr const & type, expr const & val, reducibility_hints const & hints, bool is_unsafe):
object_ref(mk_cnstr(0, constant_val(n, lparams, type), val, hints, 1)) {
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*)*3, static_cast<unsigned char>(is_unsafe));
}
bool definition_val::is_unsafe() const { return cnstr_get_scalar<unsigned char>(raw(), sizeof(object*)*3) != 0; }
theorem_val::theorem_val(name const & n, names const & lparams, expr const & type, expr const & val):
object_ref(mk_cnstr(0, constant_val(n, lparams, type), object_ref(lean_task_pure(val.to_obj_arg())))) {
}
expr theorem_val::get_value() const {
object * v_task = lean_ctor_get(raw(), 1);
object * v = lean_task_get(v_task);
return expr(v, true);
}
opaque_val::opaque_val(name const & n, names const & lparams, expr const & type, expr const & val, bool is_unsafe):
object_ref(mk_cnstr(0, constant_val(n, lparams, type), val, 1)) {
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*)*2, static_cast<unsigned char>(is_unsafe));
}
bool opaque_val::is_unsafe() const { return cnstr_get_scalar<unsigned char>(raw(), sizeof(object*)*2) != 0; }
quot_val::quot_val(name const & n, names const & lparams, expr const & type, quot_kind k):
object_ref(mk_cnstr(0, constant_val(n, lparams, type), 1)) {
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*), static_cast<unsigned char>(k));
}
quot_kind quot_val::get_quot_kind() const { return static_cast<quot_kind>(cnstr_get_scalar<unsigned char>(raw(), sizeof(object*))); }
recursor_rule::recursor_rule(name const & cnstr, unsigned nfields, expr const & rhs):
object_ref(mk_cnstr(0, cnstr, nat(nfields), rhs)) {
}
inductive_val::inductive_val(name const & n, names const & lparams, expr const & type, unsigned nparams,
unsigned nindices, names const & all, names const & cnstrs, bool rec, bool unsafe, bool is_refl):
object_ref(mk_cnstr(0, constant_val(n, lparams, type), nat(nparams), nat(nindices), all, cnstrs, 3)) {
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*)*5, static_cast<unsigned char>(rec));
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*)*5 + 1, static_cast<unsigned char>(unsafe));
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*)*5 + 2, static_cast<unsigned char>(is_refl));
lean_assert(is_unsafe() == unsafe);
lean_assert(is_rec() == rec);
lean_assert(is_reflexive() == is_refl);
}
constructor_val::constructor_val(name const & n, names const & lparams, expr const & type, name const & induct, unsigned cidx, unsigned nparams, unsigned nfields, bool is_unsafe):
object_ref(mk_cnstr(0, constant_val(n, lparams, type), induct, nat(cidx), nat(nparams), nat(nfields), 1)) {
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*)*5, static_cast<unsigned char>(is_unsafe));
}
recursor_val::recursor_val(name const & n, names const & lparams, expr const & type,
names const & all, unsigned nparams, unsigned nindices, unsigned nmotives,
unsigned nminors, recursor_rules const & rules, bool k, bool is_unsafe):
object_ref(mk_cnstr(0, constant_val(n, lparams, type), all, nat(nparams), nat(nindices), nat(nmotives),
nat(nminors), rules, 2)) {
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*)*7, static_cast<unsigned char>(k));
cnstr_set_scalar<unsigned char>(raw(), sizeof(object*)*7 + 1, static_cast<unsigned char>(is_unsafe));
}
bool declaration::is_unsafe() const {
switch (kind()) {
case declaration_kind::Definition: return to_definition_val().is_unsafe();
case declaration_kind::Axiom: return to_axiom_val().is_unsafe();
case declaration_kind::Theorem: return false;
case declaration_kind::Opaque: return to_opaque_val().is_unsafe();
case declaration_kind::Inductive: return inductive_decl(*this).is_unsafe();
case declaration_kind::Quot: return false;
case declaration_kind::MutualDefinition: return true;
}
lean_unreachable();
}
bool use_unsafe(environment const & env, expr const & e) {
bool found = false;
for_each(e, [&](expr const & e, unsigned) {
if (found) return false;
if (is_constant(e)) {
if (auto info = env.find(const_name(e))) {
if (info->is_unsafe()) {
found = true;
return false;
}
}
}
return true;
});
return found;
}
static declaration * g_dummy = nullptr;
declaration::declaration():declaration(*g_dummy) {}
static unsigned get_max_height(environment const & env, expr const & v) {
unsigned h = 0;
for_each(v, [&](expr const & e, unsigned) {
if (is_constant(e)) {
auto d = env.find(const_name(e));
if (d && d->get_hints().get_height() > h)
h = d->get_hints().get_height();
}
return true;
});
return h;
}
definition_val mk_definition_val(environment const & env, name const & n, names const & params, expr const & t, expr const & v, bool unsafe) {
unsigned h = get_max_height(env, v);
return definition_val(n, params, t, v, reducibility_hints::mk_regular(h+1), unsafe);
}
declaration mk_definition(name const & n, names const & params, expr const & t, expr const & v,
reducibility_hints const & h, bool unsafe) {
return declaration(mk_cnstr(static_cast<unsigned>(declaration_kind::Definition), definition_val(n, params, t, v, h, unsafe)));
}
declaration mk_definition(environment const & env, name const & n, names const & params, expr const & t,
expr const & v, bool unsafe) {
return declaration(mk_cnstr(static_cast<unsigned>(declaration_kind::Definition), mk_definition_val(env, n, params, t, v, unsafe)));
}
declaration mk_theorem(name const & n, names const & params, expr const & t, expr const & v) {
return declaration(mk_cnstr(static_cast<unsigned>(declaration_kind::Theorem), theorem_val(n, params, t, v)));
}
declaration mk_opaque(name const & n, names const & params, expr const & t, expr const & v, bool is_unsafe) {
return declaration(mk_cnstr(static_cast<unsigned>(declaration_kind::Opaque), opaque_val(n, params, t, v, is_unsafe)));
}
declaration mk_axiom(name const & n, names const & params, expr const & t, bool unsafe) {
return declaration(mk_cnstr(static_cast<unsigned>(declaration_kind::Axiom), axiom_val(n, params, t, unsafe)));
}
declaration mk_definition_inferring_unsafe(environment const & env, name const & n, names const & params,
expr const & t, expr const & v, reducibility_hints const & hints) {
bool unsafe = use_unsafe(env, t) || use_unsafe(env, v);
return mk_definition(n, params, t, v, hints, unsafe);
}
declaration mk_definition_inferring_unsafe(environment const & env, name const & n, names const & params,
expr const & t, expr const & v) {
bool unsafe = use_unsafe(env, t) && use_unsafe(env, v);
unsigned h = get_max_height(env, v);
return mk_definition(n, params, t, v, reducibility_hints::mk_regular(h+1), unsafe);
}
declaration mk_axiom_inferring_unsafe(environment const & env, name const & n,
names const & params, expr const & t) {
return mk_axiom(n, params, t, use_unsafe(env, t));
}
declaration mk_mutual_definitions(definition_vals const & ds) {
if (length(ds) == 1)
return declaration(mk_cnstr(static_cast<unsigned>(declaration_kind::Definition), head(ds)));
else
return declaration(mk_cnstr(static_cast<unsigned>(declaration_kind::MutualDefinition), ds));
}
declaration mk_quot_decl() {
return declaration(box(static_cast<unsigned>(declaration_kind::Quot)));
}
inductive_type::inductive_type(name const & id, expr const & type, constructors const & cnstrs):
object_ref(mk_cnstr(0, id, type, cnstrs)) {
}
static unsigned inductive_decl_scalar_offset() { return sizeof(object*)*3; }
declaration mk_inductive_decl(names const & lparams, nat const & nparams, inductive_types const & types, bool is_unsafe) {
declaration r(mk_cnstr(static_cast<unsigned>(declaration_kind::Inductive), lparams, nparams, types, 1));
cnstr_set_scalar<unsigned char>(r.raw(), inductive_decl_scalar_offset(), static_cast<unsigned char>(is_unsafe));
return r;
}
bool inductive_decl::is_unsafe() const { return cnstr_get_scalar<unsigned char>(raw(), inductive_decl_scalar_offset()) != 0; }
// =======================================
// Constant info
constant_info::constant_info():constant_info(*g_dummy) {}
constant_info::constant_info(declaration const & d):object_ref(d.raw()) {
lean_assert(d.is_definition() || d.is_theorem() || d.is_axiom() || d.is_opaque());
inc_ref(d.raw());
}
constant_info::constant_info(definition_val const & v):
object_ref(mk_cnstr(static_cast<unsigned>(constant_info_kind::Definition), v)) {
}
constant_info::constant_info(quot_val const & v):
object_ref(mk_cnstr(static_cast<unsigned>(constant_info_kind::Quot), v)) {
}
constant_info::constant_info(inductive_val const & v):
object_ref(mk_cnstr(static_cast<unsigned>(constant_info_kind::Inductive), v)) {
}
constant_info::constant_info(constructor_val const & v):
object_ref(mk_cnstr(static_cast<unsigned>(constant_info_kind::Constructor), v)) {
}
constant_info::constant_info(recursor_val const & v):
object_ref(mk_cnstr(static_cast<unsigned>(constant_info_kind::Recursor), v)) {
}
static reducibility_hints * g_opaque = nullptr;
reducibility_hints const & constant_info::get_hints() const {
if (is_definition())
return static_cast<reducibility_hints const &>(cnstr_get_ref(to_val(), 2));
else
return *g_opaque;
}
bool constant_info::is_unsafe() const {
switch (kind()) {
case constant_info_kind::Axiom: return to_axiom_val().is_unsafe();
case constant_info_kind::Definition: return to_definition_val().is_unsafe();
case constant_info_kind::Theorem: return false;
case constant_info_kind::Opaque: return to_opaque_val().is_unsafe();
case constant_info_kind::Quot: return false;
case constant_info_kind::Inductive: return to_inductive_val().is_unsafe();
case constant_info_kind::Constructor: return to_constructor_val().is_unsafe();
case constant_info_kind::Recursor: return to_recursor_val().is_unsafe();
}
lean_unreachable();
}
void initialize_declaration() {
g_opaque = new reducibility_hints(reducibility_hints::mk_opaque());
g_dummy = new declaration(mk_axiom(name(), names(), expr()));
}
void finalize_declaration() {
delete g_dummy;
delete g_opaque;
}
}