lean4-htt/src/library/tactic/intro_tactic.cpp
Leonardo de Moura 53667dd602 fix(library): change API and make sure we don't crash when searching for a non existing local decl
Fixes #1363

After error recovery has been implemented in the elaborator, a few
assumptions made in the type context are not valid anymore since we may
be recovering from errors, and the local and metavariable contexts may
be invalid.

I used the approach used in the class environment.

- find* methods return optional<...>
- get* methods throw exception for unknown elements

Remarks:

I preserved code patterns such as

     optional<local_decl> d = lctx.find_local_decl(...)
     lean_assert(d)

and did not convert them into

     local_decl d = lctx.get_local_decl(...)

Reason: the intention is clear that the local must be defined there.
If it is not we should analyze the problem and decide whether we should
throw an exception or not.

However, I converted code patterns such as

    local_decl d = *lctx.find_local_decl(...)

into

    local_decl d = lctx.get_local_decl(...)

Disclaimer: this change fixes issue #1363, but it may obfuscate other bugs.
2017-02-07 09:38:19 -08:00

171 lines
7 KiB
C++

/*
Copyright (c) 2016 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Author: Leonardo de Moura
*/
#include "kernel/instantiate.h"
#include "kernel/abstract.h"
#include "library/util.h"
#include "library/delayed_abstraction.h"
#include "library/vm/vm_name.h"
#include "library/vm/vm_nat.h"
#include "library/vm/vm_expr.h"
#include "library/tactic/tactic_state.h"
namespace lean {
static name mk_aux_name(list<name> & given_names, name const & default_name) {
if (given_names) {
name r = head(given_names);
given_names = tail(given_names);
return r;
} else {
return default_name;
}
}
optional<expr> intron(environment const & env, options const & opts, metavar_context & mctx,
expr const & mvar, unsigned n, list<name> & given_names, buffer<name> & new_Hns) {
lean_assert(is_metavar(mvar));
optional<metavar_decl> g = mctx.find_metavar_decl(mvar);
if (!g) return none_expr();
type_context ctx = mk_type_context_for(env, opts, mctx, g->get_context());
expr type = g->get_type();
type_context::tmp_locals new_locals(ctx);
buffer<expr> new_Hs;
for (unsigned i = 0; i < n; i++) {
if (!is_pi(type) && !is_let(type)) {
type = ctx.whnf(type);
if (!is_pi(type))
return none_expr();
}
lean_assert(is_pi(type) || is_let(type));
if (is_pi(type)) {
expr H = new_locals.push_local(mk_aux_name(given_names, binding_name(type)), annotated_head_beta_reduce(binding_domain(type)),
binding_info(type));
type = instantiate(binding_body(type), H);
new_Hs.push_back(H);
new_Hns.push_back(mlocal_name(H));
} else {
lean_assert(is_let(type));
expr H = new_locals.push_let(mk_aux_name(given_names, let_name(type)), annotated_head_beta_reduce(let_type(type)), let_value(type));
type = instantiate(let_body(type), H);
new_Hs.push_back(H);
new_Hns.push_back(mlocal_name(H));
}
}
expr new_M = ctx.mk_metavar_decl(ctx.lctx(), type);
expr new_val = abstract_locals(mk_delayed_abstraction_with_locals(new_M, new_Hs), new_Hs.size(), new_Hs.data());
unsigned i = new_Hs.size();
while (i > 0) {
--i;
local_decl d = ctx.lctx().get_local_decl(new_Hs[i]);
expr type = d.get_type();
type = abstract_locals(type, i, new_Hs.data());
if (auto letval = d.get_value()) {
letval = abstract_locals(*letval, i, new_Hs.data());
new_val = mk_let(d.get_pp_name(), type, *letval, new_val);
} else {
new_val = mk_lambda(d.get_pp_name(), type, new_val, d.get_info());
}
}
lean_assert(!ctx.mctx().is_assigned(new_M));
mctx = ctx.mctx();
mctx.assign(mvar, new_val);
return some_expr(new_M);
}
optional<expr> intron(environment const & env, options const & opts, metavar_context & mctx,
expr const & mvar, unsigned n, list<name> & given_names) {
buffer<name> tmp;
return intron(env, opts, mctx, mvar, n, given_names, tmp);
}
optional<expr> intron(environment const & env, options const & opts, metavar_context & mctx,
expr const & mvar, unsigned n) {
list<name> empty;
return intron(env, opts, mctx, mvar, n, empty);
}
optional<expr> intron(environment const & env, options const & opts, metavar_context & mctx,
expr const & mvar, unsigned n, buffer<name> & new_Hns) {
list<name> empty;
return intron(env, opts, mctx, mvar, n, empty, new_Hns);
}
optional<tactic_state> intron(unsigned n, tactic_state const & s, buffer<name> & new_Hns) {
if (n == 0) return some_tactic_state(s);
optional<expr> mvar = s.get_main_goal();
if (!mvar) return none_tactic_state();
list<name> new_names;
metavar_context mctx = s.mctx();
if (optional<expr> new_M = intron(s.env(), s.get_options(), mctx, *mvar, n, new_names, new_Hns)) {
list<expr> new_gs(*new_M, tail(s.goals()));
return some_tactic_state(set_mctx_goals(s, mctx, new_gs));
} else {
return none_tactic_state();
}
}
optional<tactic_state> intron(unsigned n, tactic_state const & s) {
buffer<name> tmp;
return intron(n, s, tmp);
}
vm_obj tactic_intron(vm_obj const & num, vm_obj const & s) {
optional<metavar_decl> g = to_tactic_state(s).get_main_goal_decl();
if (!g) return mk_no_goals_exception(to_tactic_state(s));
buffer<name> new_Hs;
if (auto new_s = intron(force_to_unsigned(num, 0), to_tactic_state(s), new_Hs))
return mk_tactic_success(*new_s);
else
return mk_tactic_exception("intron tactic failed, insufficient binders", to_tactic_state(s));
}
vm_obj intro(name const & n, tactic_state const & s) {
optional<metavar_decl> g = s.get_main_goal_decl();
if (!g) return mk_no_goals_exception(s);
type_context ctx = mk_type_context_for(s);
expr type = g->get_type();
if (!is_pi(type) && !is_let(type)) {
type = ctx.whnf(type);
if (!is_pi(type))
return mk_tactic_exception("intro tactic failed, Pi/let expression expected", s);
}
local_context lctx = g->get_context();
metavar_context mctx = ctx.mctx();
if (is_pi(type)) {
name n1 = n == "_" ? lctx.get_unused_name(binding_name(type)) : n;
expr H = lctx.mk_local_decl(n1, annotated_head_beta_reduce(binding_domain(type)), binding_info(type));
expr new_type = instantiate(binding_body(type), H);
expr new_M = mctx.mk_metavar_decl(lctx, new_type);
expr new_val = mk_lambda(n1, binding_domain(type), mk_delayed_abstraction(new_M, mlocal_name(H)));
mctx.assign(head(s.goals()), new_val);
list<expr> new_gs(new_M, tail(s.goals()));
return mk_tactic_success(to_obj(H), set_mctx_goals(s, mctx, new_gs));
} else {
lean_assert(is_let(type));
name n1 = n == "_" ? lctx.get_unused_name(let_name(type)) : n;
expr H = lctx.mk_local_decl(n1, annotated_head_beta_reduce(let_type(type)), let_value(type));
expr new_type = instantiate(let_body(type), H);
expr new_M = mctx.mk_metavar_decl(lctx, new_type);
expr new_val = mk_let(n1, let_type(type), let_value(type), mk_delayed_abstraction(new_M, mlocal_name(H)));
mctx.assign(head(s.goals()), new_val);
list<expr> new_gs(new_M, tail(s.goals()));
return mk_tactic_success(to_obj(H), set_mctx_goals(s, mctx, new_gs));
}
}
vm_obj tactic_intro(vm_obj const & n, vm_obj const & s) {
return intro(to_name(n), to_tactic_state(s));
}
void initialize_intro_tactic() {
DECLARE_VM_BUILTIN(name({"tactic", "intro_core"}), tactic_intro);
DECLARE_VM_BUILTIN(name({"tactic", "intron"}), tactic_intron);
}
void finalize_intro_tactic() {
}
}