lean4-htt/src/frontends/lean/print_cmd.cpp
2018-04-10 16:29:04 -07:00

618 lines
23 KiB
C++

/*
Copyright (c) 2015 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Author: Leonardo de Moura
*/
#include <algorithm>
#include <string>
#include "library/trace.h"
#include "library/sorry.h"
#include "util/sstream.h"
#include "util/sexpr/option_declarations.h"
#include "kernel/for_each_fn.h"
#include "kernel/inductive/inductive.h"
#include "kernel/quotient/quotient.h"
#include "library/util.h"
#include "library/class.h"
#include "library/aliases.h"
#include "library/pp_options.h"
#include "library/private.h"
#include "library/protected.h"
#include "library/attribute_manager.h"
#include "library/user_recursors.h"
#include "library/noncomputable.h"
#include "library/type_context.h"
#include "library/reducible.h"
#include "library/tactic/simp_lemmas.h"
#include "library/tactic/kabstract.h"
#include "frontends/lean/parser.h"
#include "frontends/lean/util.h"
#include "frontends/lean/tokens.h"
#include "frontends/lean/structure_cmd.h"
#include "frontends/lean/info_manager.h"
// TODO(gabriel): make print command async
namespace lean {
struct print_axioms_deps {
environment m_env;
io_state_stream m_ios;
name_set m_visited;
bool m_use_axioms;
bool m_used_sorry;
print_axioms_deps(environment const & env, io_state_stream const & ios):
m_env(env), m_ios(ios), m_use_axioms(false), m_used_sorry(false) {}
void visit(name const & n) {
if (m_visited.contains(n))
return;
m_visited.insert(n);
declaration const & d = m_env.get(n);
if (!d.is_definition() && !m_env.is_builtin(n)) {
m_use_axioms = true;
m_ios << d.get_name() << "\n";
}
visit(d.get_type());
if (d.is_definition())
visit(d.get_value());
}
void visit(expr const & e) {
for_each(e, [&](expr const & e, unsigned) {
if (is_sorry(e) && !m_used_sorry) {
m_used_sorry = true;
m_ios << "[sorry]" << "\n";
}
if (is_constant(e))
visit(const_name(e));
return true;
});
}
void operator()(name const & n) {
visit(n);
if (!m_use_axioms)
m_ios << "no axioms" << endl;
}
};
static void print_axioms(parser & p, message_builder & out) {
if (p.curr_is_identifier()) {
name c = p.check_constant_next("invalid '#print axioms', constant expected");
auto env = p.env();
type_context_old tc(env, p.get_options());
auto new_out = io_state_stream(env, p.ios(), tc, out.get_text_stream().get_channel());
print_axioms_deps(env, new_out)(c);
} else {
bool has_axioms = false;
p.env().for_each_declaration([&](declaration const & d) {
name const & n = d.get_name();
if (!d.is_definition() && !p.env().is_builtin(n) && d.is_trusted()) {
out << n << " : " << d.get_type() << endl;
has_axioms = true;
}
});
if (!has_axioms)
out << "no axioms" << endl;
}
}
static void print_prefix(parser & p, message_builder & out) {
name prefix = p.check_id_next("invalid '#print prefix' command, identifier expected");
buffer<declaration> to_print;
p.env().for_each_declaration([&](declaration const & d) {
if (is_prefix_of(prefix, d.get_name())) {
to_print.push_back(d);
}
});
std::sort(to_print.begin(), to_print.end(), [](declaration const & d1, declaration const & d2) { return d1.get_name() < d2.get_name(); });
for (declaration const & d : to_print) {
out << d.get_name() << " : " << d.get_type() << "\n";
}
if (to_print.empty())
out << "no declaration starting with prefix '" << prefix << "'\n";
}
static void print_fields(parser const & p, message_builder & out, name const & S, pos_info const & pos) {
environment const & env = p.env();
if (!is_structure(env, S))
throw parser_error(sstream() << "invalid '#print fields' command, '" << S << "' is not a structure", pos);
for (name const & field_name : get_structure_fields(env, S)) {
declaration d = env.get(S + field_name);
out << d.get_name() << " : " << d.get_type() << endl;
}
}
static bool uses_token(unsigned num, notation::transition const * ts, name const & token) {
for (unsigned i = 0; i < num; i++) {
if (ts[i].get_token() == token)
return true;
}
return false;
}
static bool uses_some_token(unsigned num, notation::transition const * ts, buffer<name> const & tokens) {
return
tokens.empty() ||
std::any_of(tokens.begin(), tokens.end(), [&](name const & token) { return uses_token(num, ts, token); });
}
static bool print_parse_table(parser const & p, message_builder & rep, parse_table const & t, bool nud, buffer<name> const & tokens, bool tactic_table = false) {
bool found = false;
options os = rep.get_text_stream().get_options();
os = os.update_if_undef(get_pp_full_names_name(), true);
os = os.update(get_pp_notation_name(), false);
os = os.update(get_pp_preterm_name(), true);
auto out = rep.get_text_stream().update_options(os);
optional<token_table> tt(get_token_table(p.env()));
t.for_each([&](unsigned num, notation::transition const * ts, list<notation::accepting> const & overloads) {
if (uses_some_token(num, ts, tokens)) {
if (tactic_table)
out << "tactic notation ";
found = true;
notation::display(out, num, ts, overloads, nud, tt);
}
});
return found;
}
static void print_notation(parser & p, message_builder & out) {
buffer<name> tokens;
while (p.curr_is_keyword()) {
tokens.push_back(p.get_token_info().token());
p.next();
}
bool found = false;
if (print_parse_table(p, out, get_nud_table(p.env()), true, tokens))
found = true;
if (print_parse_table(p, out, get_led_table(p.env()), false, tokens))
found = true;
if (!found)
out << "no notation";
}
#if 0
static void print_patterns(parser & p, name const & n) {
if (is_forward_lemma(p.env(), n)) {
// we regenerate the patterns to make sure they reflect the current set of reducible constants
try {
blast::scope_debug scope(p.env(), p.ios());
auto hi = blast::mk_hi_lemma(n, LEAN_DEFAULT_PRIORITY);
if (hi.m_multi_patterns) {
options opts = p.get_options();
opts = opts.update_if_undef(get_pp_metavar_args_name(), true);
io_state new_ios(p.ios(), opts);
type_context_old tc(p.env(), opts);
io_state_stream out = regular(p.env(), new_ios, tc);
out << "(multi-)patterns:\n";
if (!is_nil(hi.m_mvars)) {
expr m = head(hi.m_mvars);
out << m << " : " << mlocal_type(m);
for (expr const & m : tail(hi.m_mvars)) {
out << ", " << m << " : " << mlocal_type(m);
}
}
out << "\n";
for (multi_pattern const & mp : hi.m_multi_patterns) {
out << "{";
bool first = true;
for (expr const & p : mp) {
if (first) first = false; else out << ", ";
out << p;
}
out << "}\n";
}
}
} catch (exception & ex) {
p.display_error(ex);
}
}
}
#endif
static name to_user_name(environment const & env, name const & n) {
if (auto r = hidden_to_user_name(env, n))
return *r;
else
return n;
}
static void print_definition(environment const & env, message_builder & out, name const & n, pos_info const & pos) {
declaration d = env.get(n);
if (!d.is_definition())
throw parser_error(sstream() << "invalid '#print definition', '" << to_user_name(env, n) << "' is not a definition", pos);
options opts = out.get_text_stream().get_options();
opts = opts.update_if_undef(get_pp_beta_name(), false);
out.get_text_stream().update_options(opts) << d.get_value() << endl;
}
static void print_attributes(parser const & p, message_builder & out, name const & n) {
environment const & env = p.env();
buffer<attribute const *> attrs;
get_attributes(p.env(), attrs);
std::sort(attrs.begin(), attrs.end(), [](attribute const * a1, attribute const * a2) {
return a1->get_name() < a2->get_name();
});
bool first = true;
for (auto attr : attrs) {
if (attr->get_name() == "reducibility")
continue;
if (auto data = attr->get_untyped(env, n)) {
if (first) {
out << "@[";
first = false;
} else {
out << ", ";
}
out << attr->get_name();
data->print(out.get_text_stream().get_stream());
unsigned prio = attr->get_prio(env, n);
if (prio != LEAN_DEFAULT_PRIORITY)
out << ", priority " << prio;
}
}
if (!first)
out << "]\n";
}
static void print_inductive(parser const & p, message_builder & out, name const & n, pos_info const & pos) {
environment const & env = p.env();
if (auto idecl = inductive::is_inductive_decl(env, n)) {
level_param_names ls = idecl->m_level_params;
print_attributes(p, out, n);
if (is_structure(env, n))
out << "structure";
else
out << "inductive";
out << " " << n;
out << " : " << env.get(n).get_type() << "\n";
if (is_structure(env, n)) {
out << "fields:\n";
print_fields(p, out, n, pos);
} else {
out << "constructors:\n";
buffer<name> constructors;
get_intro_rule_names(env, n, constructors);
for (name const & c : constructors) {
out << c << " : " << env.get(c).get_type() << "\n";
}
}
} else {
throw parser_error(sstream() << "invalid '#print inductive', '" << n << "' is not an inductive declaration", pos);
}
}
static void print_recursor_info(parser & p, message_builder & out) {
name c = p.check_constant_next("invalid '#print [recursor]', constant expected");
recursor_info info = get_recursor_info(p.env(), c);
out << "recursor information\n"
<< " num. parameters: " << info.get_num_params() << "\n"
<< " num. indices: " << info.get_num_indices() << "\n"
<< " num. minors: " << info.get_num_minors() << "\n"
<< " recursive: " << info.is_recursive() << "\n"
<< " universe param pos.: ";
for (unsigned idx : info.get_universe_pos()) {
if (idx == recursor_info::get_motive_univ_idx()) {
out << " [motive univ]";
} else {
out << " " << idx;
}
}
out << "\n";
out << " motive pos.: " << info.get_motive_pos() + 1 << "\n"
<< " major premise pos.: " << info.get_major_pos() + 1 << "\n"
<< " dep. elimination: " << info.has_dep_elim() << "\n";
if (info.get_num_params() > 0) {
out << " parameters pos. at major:";
for (optional<unsigned> const & p : info.get_params_pos()) {
if (p)
out << " " << *p+1;
else
out << " [instance]";
}
out << "\n";
}
if (info.get_num_indices() > 0) {
out << " indices pos. at major: ";
for (unsigned p : info.get_indices_pos())
out << " " << p+1;
out << "\n";
}
}
static bool print_constant(parser const & p, message_builder & out, char const * kind, declaration const & d, bool is_def = false) {
type_checker tc(p.env());
print_attributes(p, out, d.get_name());
if (is_protected(p.env(), d.get_name()))
out << "protected ";
if (d.is_definition() && is_marked_noncomputable(p.env(), d.get_name()))
out << "noncomputable ";
if (!d.is_trusted())
out << "meta ";
out << kind << " " << to_user_name(p.env(), d.get_name());
out.get_text_stream().update_options(out.get_text_stream().get_options().update((name {"pp", "binder_types"}), true))
<< " : " << d.get_type();
if (is_def)
out << " :=";
out << "\n";
return true;
}
void print_id_info(parser & p, message_builder & out, name const & id, bool show_value, pos_info const & pos) {
environment const & env = p.env();
bool found = false;
// declarations
list<name> cs;
try {
cs = p.to_constants(id, "", pos);
found = true;
} catch (parser_error) {}
bool first = true;
for (name const & c : cs) {
if (first) first = false; else out << "\n";
declaration const & d = env.get(c);
if (d.is_theorem()) {
print_constant(p, out, "theorem", d, show_value);
try {
if (show_value)
print_definition(env, out, c, pos);
} catch (std::exception & ex) {
out << "[incorrect proof]\n";
bool use_pos = false;
out.set_exception(ex, use_pos);
}
} else if (d.is_axiom() || d.is_constant_assumption()) {
if (inductive::is_inductive_decl(env, c)) {
print_inductive(p, out, c, pos);
} else if (inductive::is_intro_rule(env, c)) {
print_constant(p, out, "constructor", d);
} else if (inductive::is_elim_rule(env, c)) {
print_constant(p, out, "eliminator", d);
} else if (is_quotient_decl(env, c)) {
print_constant(p, out, "builtin-quotient-type-constant", d);
} else if (d.is_axiom()) {
print_constant(p, out, "axiom", d);
} else {
print_constant(p, out, "constant", d);
}
} else if (d.is_definition()) {
print_constant(p, out, "def", d, show_value);
if (show_value)
print_definition(env, out, c, pos);
}
// print_patterns(p, c);
}
// add to info_manager when not overloaded
if (auto c = head_opt(cs))
if (!tail(cs))
if (auto infom = get_global_info_manager()) {
infom->add_const_info(p.env(), pos, *c);
}
if (found) return;
// variables and parameters
if (expr const * type = p.get_local(id)) {
if (is_local(*type)) {
if (p.is_local_variable(*type)) {
out << "variable " << id << " : " << mlocal_type(*type) << "\n";
} else {
out << "parameter " << id << " : " << mlocal_type(*type) << "\n";
}
return;
}
}
// options
get_option_declarations().for_each([&](name const &, option_declaration const & opt) {
if (found) return;
if (opt.get_name() == id || opt.get_name() == name("lean") + id) {
out << "option " << opt.get_name() << " (" << opt.kind() << ") "
<< opt.get_description() << " (default: " << opt.get_default_value() << ")" << endl;
found = true;
}
});
if (!found) throw parser_error(sstream() << "unknown identifier " << id, p.pos());
}
bool print_token_info(parser const & p, message_builder & out, name const & tk) {
buffer<name> tokens;
tokens.push_back(tk);
bool found = false;
if (print_parse_table(p, out, get_nud_table(p.env()), true, tokens)) {
found = true;
}
if (print_parse_table(p, out, get_led_table(p.env()), false, tokens)) {
found = true;
}
return found;
}
void print_polymorphic(parser & p, message_builder & out) {
auto pos = p.pos();
// notation
if (p.curr_is_keyword()) {
name tk = p.get_token_info().token();
if (print_token_info(p, out, tk)) {
p.next();
return;
}
}
name id = p.check_id_next("invalid #print command", break_at_pos_exception::token_context::expr);
bool show_value = true;
print_id_info(p, out, id, show_value, pos);
}
static void print_simp_rules(parser & p, message_builder & out) {
name attr = p.check_id_next("invalid '#print [simp]' command, identifier expected");
simp_lemmas slss = get_simp_lemmas(p.env(), attr);
out << slss.pp_simp(out.get_formatter());
}
static void print_congr_rules(parser & p, message_builder & out) {
name attr = p.check_id_next("invalid '#print [congr]' command, identifier expected");
simp_lemmas slss = get_simp_lemmas(p.env(), attr);
out << slss.pp_congr(out.get_formatter());
}
static void print_aliases(parser const & p, message_builder & out) {
for_each_expr_alias(p.env(), [&](name const & n, list<name> const & as) {
out << n << " -> {";
bool first = true;
for (name const & a : as) {
if (first) first = false; else out << ", ";
out << a;
}
out << "}\n";
});
}
static void print_key_equivalences(parser & p, message_builder & out) {
for_each_key_equivalence(p.env(), [&](buffer<name> const & ns) {
out << "[";
for (unsigned i = 0; i < ns.size(); i++) {
if (i > 0) out << ", ";
out << ns[i];
}
out << "]\n";
});
}
static void print_attribute(parser & p, message_builder & out, attribute const & attr) {
buffer<name> instances;
attr.get_instances(p.env(), instances);
// oldest first
unsigned i = instances.size();
while (i > 0) {
--i;
out << instances[i] << "\n";
}
}
environment print_cmd(parser & p) {
// Fallbacks are handled via exceptions.
auto _ = p.no_error_recovery_scope();
auto out = p.mk_message(p.cmd_pos(), INFORMATION);
out.set_caption("print result");
auto env = p.env();
if (p.curr() == token_kind::String) {
out << p.get_str_val() << endl;
p.next();
} else if (p.curr_is_token_or_id(get_raw_tk())) {
p.next();
auto _ = p.error_recovery_scope(true);
expr e = p.parse_expr();
options opts = out.get_text_stream().get_options();
opts = opts.update(get_pp_notation_name(), false);
out.get_text_stream().update_options(opts) << e << endl;
} else if (p.curr_is_token_or_id(get_options_tk())) {
p.next();
out << p.ios().get_options() << endl;
} else if (p.curr_is_token_or_id(get_trust_tk())) {
p.next();
out << "trust level: " << p.env().trust_lvl() << endl;
} else if (p.curr_is_token_or_id(get_key_equivalences_tk())) {
p.next();
print_key_equivalences(p, out);
} else if (p.curr_is_token_or_id(get_definition_tk())) {
p.next();
auto pos = p.pos();
name id = p.check_id_next("invalid '#print definition', constant expected");
list<name> cs = p.to_constants(id, "invalid '#print definition', constant expected", pos);
bool first = true;
for (name const & c : cs) {
if (first)
first = false;
else
out << "\n";
declaration const & d = p.env().get(c);
if (d.is_theorem()) {
print_constant(p, out, "theorem", d);
print_definition(env, out, c, pos);
} else if (d.is_definition()) {
print_constant(p, out, "definition", d);
print_definition(env, out, c, pos);
} else {
throw parser_error(sstream() << "invalid '#print definition', '" << to_user_name(p.env(), c) << "' is not a definition", pos);
}
}
} else if (p.curr_is_token_or_id(get_instances_tk())) {
p.next();
name c = p.check_constant_next("invalid '#print instances', constant expected");
for (name const & i : get_class_instances(env, c)) {
out << i << " : " << env.get(i).get_type() << endl;
}
} else if (p.curr_is_token_or_id(get_classes_tk())) {
p.next();
buffer<name> classes;
get_classes(env, classes);
std::sort(classes.begin(), classes.end());
for (name const & c : classes) {
out << c << " : " << env.get(c).get_type() << endl;
}
} else if (p.curr_is_token_or_id(get_attributes_tk())) {
p.next();
buffer<attribute const *> attrs;
get_attributes(p.env(), attrs);
std::sort(attrs.begin(), attrs.end(), [](attribute const * a1, attribute const * a2) {
return a1->get_name() < a2->get_name();
});
for (auto attr : attrs) {
out << "[" << attr->get_name() << "] " << attr->get_description() << endl;
}
} else if (p.curr_is_token_or_id(get_prefix_tk())) {
p.next();
print_prefix(p, out);
} else if (p.curr_is_token_or_id(get_aliases_tk())) {
p.next();
print_aliases(p, out);
} else if (p.curr_is_token_or_id(get_axioms_tk())) {
p.next();
print_axioms(p, out);
} else if (p.curr_is_token_or_id(get_fields_tk())) {
p.next();
auto pos = p.pos();
name S = p.check_constant_next("invalid '#print fields' command, constant expected");
print_fields(p, out, S, pos);
} else if (p.curr_is_token_or_id(get_notation_tk())) {
p.next();
print_notation(p, out);
} else if (p.curr_is_token_or_id(get_inductive_tk())) {
p.next();
auto pos = p.pos();
name c = p.check_constant_next("invalid '#print inductive', constant expected");
print_inductive(p, out, c, pos);
} else if (p.curr_is_token(get_lbracket_tk())) {
p.next();
auto pos = p.pos();
auto name = p.check_id_next("invalid attribute declaration, identifier expected");
p.check_token_next(get_rbracket_tk(), "invalid '#print [<attr>]', ']' expected");
if (name == "recursor") {
print_recursor_info(p, out);
} else if (name == "simp") {
print_simp_rules(p, out);
} else if (name == "congr") {
print_congr_rules(p, out);
} else {
if (!is_attribute(p.env(), name))
throw parser_error(sstream() << "unknown attribute [" << name << "]", pos);
auto const & attr = get_attribute(p.env(), name);
print_attribute(p, out, attr);
}
} else {
print_polymorphic(p, out);
}
out.set_end_pos(p.pos());
out.report();
return p.env();
}
}