It just adds extra complexity and is in conflict for our plans for Lean4. Moreover, in our experiments it impacts negatively on performance: master and lean4 branches. The negative impact has been confirmed by @kha too.
2682 lines
93 KiB
C++
2682 lines
93 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 <utility>
|
||
#include <string>
|
||
#include <limits>
|
||
#include <vector>
|
||
#include "library/profiling.h"
|
||
#include "library/library_task_builder.h"
|
||
#include "util/utf8.h"
|
||
#include "util/interrupt.h"
|
||
#include "util/sstream.h"
|
||
#include "util/flet.h"
|
||
#include "util/sexpr/option_declarations.h"
|
||
#include "kernel/for_each_fn.h"
|
||
#include "kernel/replace_fn.h"
|
||
#include "kernel/find_fn.h"
|
||
#include "kernel/abstract.h"
|
||
#include "kernel/instantiate.h"
|
||
#include "kernel/error_msgs.h"
|
||
#include "library/st_task_queue.h"
|
||
#include "library/module_mgr.h"
|
||
#include "library/export_decl.h"
|
||
#include "library/trace.h"
|
||
#include "library/quote.h"
|
||
#include "library/class.h"
|
||
#include "library/exception.h"
|
||
#include "library/aliases.h"
|
||
#include "library/constants.h"
|
||
#include "library/annotation.h"
|
||
#include "library/private.h"
|
||
#include "library/locals.h"
|
||
#include "library/local_context.h"
|
||
#include "library/protected.h"
|
||
#include "library/choice.h"
|
||
#include "library/placeholder.h"
|
||
#include "library/deep_copy.h"
|
||
#include "library/module.h"
|
||
#include "library/scoped_ext.h"
|
||
#include "library/explicit.h"
|
||
#include "library/num.h"
|
||
#include "library/string.h"
|
||
#include "library/sorry.h"
|
||
#include "library/documentation.h"
|
||
#include "library/pp_options.h"
|
||
#include "library/noncomputable.h"
|
||
#include "library/replace_visitor.h"
|
||
#include "kernel/scope_pos_info_provider.h"
|
||
#include "library/type_context.h"
|
||
#include "library/pattern_attribute.h"
|
||
#include "library/equations_compiler/equations.h"
|
||
#include "frontends/lean/tokens.h"
|
||
#include "frontends/lean/builtin_exprs.h"
|
||
#include "frontends/lean/parser.h"
|
||
#include "frontends/lean/util.h"
|
||
#include "frontends/lean/notation_cmd.h"
|
||
#include "frontends/lean/parser_pos_provider.h"
|
||
#include "frontends/lean/builtin_cmds.h"
|
||
#include "frontends/lean/prenum.h"
|
||
#include "frontends/lean/elaborator.h"
|
||
#include "frontends/lean/local_context_adapter.h"
|
||
#include "frontends/lean/structure_instance.h"
|
||
#include "frontends/lean/typed_expr.h"
|
||
|
||
#ifndef LEAN_DEFAULT_PARSER_SHOW_ERRORS
|
||
#define LEAN_DEFAULT_PARSER_SHOW_ERRORS true
|
||
#endif
|
||
|
||
namespace lean {
|
||
static name * g_frontend_fresh = nullptr;
|
||
|
||
void break_at_pos_exception::report_goal_pos(pos_info goal_pos) {
|
||
if (!m_goal_pos)
|
||
m_goal_pos = goal_pos;
|
||
}
|
||
|
||
// ==========================================
|
||
// Parser configuration options
|
||
static name * g_parser_show_errors;
|
||
|
||
bool get_parser_show_errors(options const & opts) {
|
||
return opts.get_bool(*g_parser_show_errors, LEAN_DEFAULT_PARSER_SHOW_ERRORS);
|
||
}
|
||
|
||
// ==========================================
|
||
|
||
parser::local_scope::local_scope(parser & p, bool save_options):
|
||
m_p(p), m_env(p.env()) {
|
||
m_p.push_local_scope(save_options);
|
||
}
|
||
parser::local_scope::local_scope(parser & p, environment const & env):
|
||
m_p(p), m_env(p.env()) {
|
||
m_p.m_env = env;
|
||
m_p.push_local_scope();
|
||
}
|
||
parser::local_scope::local_scope(parser & p, optional<environment> const & env):
|
||
m_p(p), m_env(p.env()) {
|
||
if (env)
|
||
m_p.m_env = *env;
|
||
m_p.push_local_scope();
|
||
}
|
||
parser::local_scope::~local_scope() {
|
||
m_p.pop_local_scope();
|
||
m_p.m_env = m_env;
|
||
}
|
||
|
||
LEAN_THREAD_VALUE(id_behavior, g_outer_id_behavior, id_behavior::ErrorIfUndef);
|
||
|
||
parser::quote_scope::quote_scope(parser & p, bool q, id_behavior i):
|
||
m_p(p), m_id_behavior(m_p.m_id_behavior), m_old_in_quote(m_p.m_in_quote), m_in_quote(q),
|
||
m_saved_in_pattern(p.m_in_pattern) {
|
||
// m_p.m_in_pattern = false;
|
||
if (m_in_quote && !m_old_in_quote) {
|
||
g_outer_id_behavior = m_p.m_id_behavior;
|
||
m_p.m_id_behavior = i;
|
||
m_p.m_in_quote = true;
|
||
m_p.push_local_scope(false);
|
||
m_p.m_quote_stack = cons(m_p.mk_parser_scope(), m_p.m_quote_stack);
|
||
if (i != id_behavior::ErrorIfUndef)
|
||
m_p.clear_expr_locals();
|
||
} else if (!m_in_quote && m_old_in_quote) {
|
||
lean_assert(m_p.m_quote_stack);
|
||
m_p.m_id_behavior = g_outer_id_behavior;
|
||
m_p.push_local_scope(false);
|
||
m_p.m_in_quote = false;
|
||
m_p.restore_parser_scope(head(m_p.m_quote_stack));
|
||
}
|
||
}
|
||
|
||
parser::quote_scope::~quote_scope() {
|
||
m_p.m_in_pattern = m_saved_in_pattern;
|
||
if (m_in_quote && !m_old_in_quote) {
|
||
lean_assert(m_p.m_in_quote);
|
||
m_p.m_in_quote = false;
|
||
m_p.pop_local_scope();
|
||
m_p.m_quote_stack = tail(m_p.m_quote_stack);
|
||
} else if (!m_in_quote && m_old_in_quote) {
|
||
lean_assert(!m_p.m_in_quote);
|
||
m_p.m_in_quote = true;
|
||
m_p.pop_local_scope();
|
||
}
|
||
m_p.m_id_behavior = m_id_behavior;
|
||
}
|
||
|
||
parser::undef_id_to_local_scope::undef_id_to_local_scope(parser & p):
|
||
flet<id_behavior>(p.m_id_behavior, id_behavior::AssumeLocalIfUndef) {}
|
||
parser::error_if_undef_scope::error_if_undef_scope(parser & p):
|
||
flet<id_behavior>(p.m_id_behavior, id_behavior::ErrorIfUndef) {}
|
||
parser::all_id_local_scope::all_id_local_scope(parser & p):
|
||
flet<id_behavior>(p.m_id_behavior, id_behavior::AllLocal) {}
|
||
|
||
parser::parser(environment const & env, io_state const & ios,
|
||
module_loader const & import_fn,
|
||
std::istream & strm, std::string const & file_name,
|
||
bool use_exceptions) :
|
||
m_env(env), m_ngen(*g_frontend_fresh), m_ios(ios),
|
||
m_use_exceptions(use_exceptions),
|
||
m_import_fn(import_fn),
|
||
m_file_name(file_name),
|
||
m_scanner(strm, m_file_name.c_str()),
|
||
m_imports_parsed(false) {
|
||
m_next_inst_idx = 1;
|
||
m_ignore_noncomputable = false;
|
||
m_in_quote = false;
|
||
m_in_pattern = false;
|
||
m_has_params = false;
|
||
m_id_behavior = id_behavior::ErrorIfUndef;
|
||
updt_options();
|
||
m_next_tag_idx = 0;
|
||
m_curr = token_kind::Identifier;
|
||
}
|
||
|
||
parser::~parser() {
|
||
}
|
||
|
||
void parser::check_break_at_pos(break_at_pos_exception::token_context ctxt) {
|
||
auto p = pos();
|
||
if (m_break_at_pos && p.first == m_break_at_pos->first && p.second <= m_break_at_pos->second) {
|
||
name tk;
|
||
if (curr_is_identifier() || curr() == token_kind::FieldName) {
|
||
tk = get_name_val();
|
||
} else if (curr_is_command() || curr_is_keyword()) {
|
||
tk = get_token_info().token();
|
||
// When completing at the end of a token that cannot be extended into an identifier,
|
||
// start an empty completion instead (in the next call to `check_break_before/at_pos`, using the correct
|
||
// context).
|
||
if (m_complete && m_break_at_pos->second == p.second + tk.utf8_size() - 1 &&
|
||
!curr_is_token(get_period_tk())) {
|
||
auto s = tk.to_string();
|
||
if (!is_id_rest(get_utf8_last_char(s.c_str()), s.c_str() + s.size()))
|
||
return;
|
||
}
|
||
} else {
|
||
return;
|
||
}
|
||
if (m_break_at_pos->second < p.second + tk.utf8_size())
|
||
throw break_at_pos_exception(p, tk, ctxt);
|
||
}
|
||
}
|
||
|
||
void parser::check_break_before(break_at_pos_exception::token_context ctxt) {
|
||
if (!get_complete())
|
||
ctxt = break_at_pos_exception::token_context::none;
|
||
if (m_break_at_pos && *m_break_at_pos < pos())
|
||
throw break_at_pos_exception(*m_break_at_pos, "", ctxt);
|
||
}
|
||
|
||
void parser::scan() {
|
||
check_break_before();
|
||
check_break_at_pos();
|
||
pos_info curr_pos = pos();
|
||
if (m_error_since_last_cmd && (curr_is_command() && !curr_is_token(get_end_tk()))) {
|
||
// If we're during error recovery, do not read past command tokens.
|
||
// `end` is not treated as a command token since it occurs in begin-end blocks.
|
||
return;
|
||
}
|
||
if (m_break_at_pos && m_break_at_pos->first == curr_pos.first && curr_is_identifier()) {
|
||
name curr_ident = get_name_val();
|
||
m_curr = m_scanner.scan(m_env);
|
||
// when breaking on a '.' token trailing an identifier, report them as a single, concatenated token
|
||
if (*m_break_at_pos == pos() && curr_is_token(get_period_tk()))
|
||
throw break_at_pos_exception(curr_pos, name(curr_ident.to_string() + get_period_tk()));
|
||
return;
|
||
}
|
||
m_curr = m_scanner.scan(m_env);
|
||
}
|
||
|
||
expr parser::mk_sorry(pos_info const & p, bool synthetic) {
|
||
return save_pos(::lean::mk_sorry(mk_Prop(), synthetic), p);
|
||
}
|
||
|
||
void parser::updt_options() {
|
||
m_profile = get_profiler(m_ios.get_options());
|
||
m_show_errors = get_parser_show_errors(m_ios.get_options());
|
||
}
|
||
|
||
void parser::sync_command() {
|
||
// Keep consuming tokens until we find a Command or End-of-file
|
||
while (curr() != token_kind::CommandKeyword && curr() != token_kind::Eof)
|
||
next();
|
||
}
|
||
|
||
tag parser::get_tag(expr e) {
|
||
tag t = e.get_tag();
|
||
if (t == nulltag) {
|
||
t = m_next_tag_idx;
|
||
e.set_tag(t);
|
||
m_next_tag_idx++;
|
||
}
|
||
return t;
|
||
}
|
||
|
||
name parser::mk_anonymous_inst_name() {
|
||
name n = ::lean::mk_anonymous_inst_name(m_next_inst_idx);
|
||
m_next_inst_idx++;
|
||
return n;
|
||
}
|
||
|
||
expr parser::save_pos(expr const & e, pos_info p) {
|
||
auto t = get_tag(e);
|
||
if (!m_pos_table.contains(t))
|
||
m_pos_table.insert(t, p);
|
||
return e;
|
||
}
|
||
|
||
void parser::erase_pos(expr const & e) {
|
||
auto t = get_tag(e);
|
||
m_pos_table.erase(t);
|
||
}
|
||
|
||
expr parser::update_pos(expr e, pos_info p) {
|
||
auto t = get_tag(e);
|
||
m_pos_table.insert(t, p);
|
||
return e;
|
||
}
|
||
|
||
expr parser::rec_save_pos(expr const & e, pos_info p) {
|
||
unsigned m = std::numeric_limits<unsigned>::max();
|
||
pos_info dummy(m, 0);
|
||
for_each(e, [&](expr const & e, unsigned) {
|
||
if (pos_of(e, dummy).first == m) {
|
||
save_pos(e, p);
|
||
return true;
|
||
} else {
|
||
return false;
|
||
}
|
||
});
|
||
return e;
|
||
}
|
||
|
||
/** \brief Create a copy of \c e, and the position of new expression with p */
|
||
expr parser::copy_with_new_pos(expr const & e, pos_info p) {
|
||
switch (e.kind()) {
|
||
case expr_kind::Sort: case expr_kind::Constant: case expr_kind::Meta:
|
||
case expr_kind::Var: case expr_kind::Local:
|
||
return save_pos(copy(e), p);
|
||
case expr_kind::App:
|
||
return save_pos(::lean::mk_app(copy_with_new_pos(app_fn(e), p),
|
||
copy_with_new_pos(app_arg(e), p)),
|
||
p);
|
||
case expr_kind::Lambda:
|
||
return save_pos(::lean::mk_lambda(binding_name(e),
|
||
copy_with_new_pos(binding_domain(e), p),
|
||
copy_with_new_pos(binding_body(e), p),
|
||
binding_info(e)),
|
||
|
||
p);
|
||
case expr_kind::Pi:
|
||
return save_pos(::lean::mk_pi(binding_name(e),
|
||
copy_with_new_pos(binding_domain(e), p),
|
||
copy_with_new_pos(binding_body(e), p),
|
||
binding_info(e)),
|
||
p);
|
||
case expr_kind::Let:
|
||
return save_pos(::lean::mk_let(let_name(e),
|
||
copy_with_new_pos(let_type(e), p),
|
||
copy_with_new_pos(let_value(e), p),
|
||
copy_with_new_pos(let_body(e), p)),
|
||
p);
|
||
case expr_kind::Macro:
|
||
if (is_pexpr_quote(e)) {
|
||
return save_pos(mk_pexpr_quote(copy_with_new_pos(get_pexpr_quote_value(e), p)), p);
|
||
} else {
|
||
buffer<expr> args;
|
||
for (unsigned i = 0; i < macro_num_args(e); i++)
|
||
args.push_back(copy_with_new_pos(macro_arg(e, i), p));
|
||
return save_pos(::lean::mk_macro(macro_def(e), args.size(), args.data()), p);
|
||
}
|
||
}
|
||
lean_unreachable(); // LCOV_EXCL_LINE
|
||
}
|
||
|
||
pos_info parser::pos_of(expr const & e, pos_info default_pos) const {
|
||
tag t = e.get_tag();
|
||
if (t == nulltag)
|
||
return default_pos;
|
||
if (auto it = m_pos_table.find(t))
|
||
return *it;
|
||
else
|
||
return default_pos;
|
||
}
|
||
|
||
bool parser::curr_is_token(name const & tk) const {
|
||
return
|
||
(curr() == token_kind::Keyword || curr() == token_kind::CommandKeyword) &&
|
||
get_token_info().value() == tk;
|
||
}
|
||
|
||
bool parser::curr_is_token_or_id(name const & tk) const {
|
||
if (curr() == token_kind::Keyword || curr() == token_kind::CommandKeyword)
|
||
return get_token_info().value() == tk;
|
||
else if (curr() == token_kind::Identifier)
|
||
return get_name_val() == tk;
|
||
else
|
||
return false;
|
||
}
|
||
|
||
bool parser::check_token_next(name const & tk, char const * msg) {
|
||
if (!curr_is_token(tk)) {
|
||
maybe_throw_error({msg, pos()});
|
||
return false;
|
||
}
|
||
next();
|
||
return true;
|
||
}
|
||
|
||
void parser::check_token_or_id_next(name const & tk, char const * msg) {
|
||
if (!curr_is_token_or_id(tk))
|
||
return maybe_throw_error({msg, pos()});
|
||
next();
|
||
}
|
||
|
||
name parser::check_id_next(char const * msg, break_at_pos_exception::token_context ctxt) {
|
||
// initiate empty completion even if following token is not an identifier
|
||
if (get_complete())
|
||
check_break_before(ctxt);
|
||
name r;
|
||
if (!curr_is_identifier()) {
|
||
auto _ = no_error_recovery_scope_if(curr_is_command());
|
||
maybe_throw_error({msg, pos()});
|
||
return "_";
|
||
} else {
|
||
r = get_name_val();
|
||
}
|
||
try {
|
||
next();
|
||
} catch (break_at_pos_exception & e) {
|
||
e.m_token_info.m_context = ctxt;
|
||
throw;
|
||
}
|
||
return r;
|
||
}
|
||
|
||
void parser::check_not_internal(name const & id, pos_info const & p) {
|
||
if (is_internal_name(id))
|
||
maybe_throw_error({
|
||
sstream() << "invalid declaration name '" << id
|
||
<< "', identifiers starting with '_' are reserved to the system",
|
||
p});
|
||
}
|
||
|
||
name parser::check_decl_id_next(char const * msg, break_at_pos_exception::token_context ctxt) {
|
||
auto p = pos();
|
||
name id = check_id_next(msg, ctxt);
|
||
check_not_internal(id, p);
|
||
return id;
|
||
}
|
||
|
||
name parser::check_atomic_id_next(char const * msg) {
|
||
auto p = pos();
|
||
name id = check_id_next(msg);
|
||
if (!id.is_atomic())
|
||
maybe_throw_error({msg, p});
|
||
return id;
|
||
}
|
||
|
||
name parser::check_atomic_decl_id_next(char const * msg) {
|
||
auto p = pos();
|
||
name id = check_atomic_id_next(msg);
|
||
check_not_internal(id, p);
|
||
return id;
|
||
}
|
||
|
||
expr parser::mk_app(expr fn, expr arg, pos_info const & p) {
|
||
return save_pos(::lean::mk_app(fn, arg), p);
|
||
}
|
||
|
||
expr parser::mk_app(expr fn, buffer<expr> const & args, pos_info const & p) {
|
||
expr r = fn;
|
||
for (expr const & arg : args) {
|
||
r = mk_app(r, arg, p);
|
||
}
|
||
return r;
|
||
}
|
||
|
||
expr parser::mk_app(std::initializer_list<expr> const & args, pos_info const & p) {
|
||
lean_assert(args.size() >= 2);
|
||
auto it = args.begin();
|
||
expr r = *it;
|
||
it++;
|
||
for (; it != args.end(); it++)
|
||
r = mk_app(r, *it, p);
|
||
return r;
|
||
}
|
||
|
||
parser_scope parser::mk_parser_scope(optional<options> const & opts) {
|
||
return parser_scope(opts, m_level_variables, m_variables, m_include_vars,
|
||
m_next_inst_idx, m_has_params,
|
||
m_local_level_decls, m_local_decls);
|
||
}
|
||
|
||
void parser::restore_parser_scope(parser_scope const & s) {
|
||
if (s.m_options) {
|
||
m_ios.set_options(*s.m_options);
|
||
updt_options();
|
||
}
|
||
m_local_level_decls = s.m_local_level_decls;
|
||
m_local_decls = s.m_local_decls;
|
||
m_level_variables = s.m_level_variables;
|
||
m_variables = s.m_variables;
|
||
m_include_vars = s.m_include_vars;
|
||
m_has_params = s.m_has_params;
|
||
m_next_inst_idx = s.m_next_inst_idx;
|
||
}
|
||
|
||
void parser::push_local_scope(bool save_options) {
|
||
optional<options> opts;
|
||
if (save_options)
|
||
opts = m_ios.get_options();
|
||
m_parser_scope_stack = cons(mk_parser_scope(opts), m_parser_scope_stack);
|
||
}
|
||
|
||
void parser::pop_local_scope() {
|
||
lean_assert(m_parser_scope_stack);
|
||
auto s = head(m_parser_scope_stack);
|
||
restore_parser_scope(s);
|
||
m_parser_scope_stack = tail(m_parser_scope_stack);
|
||
}
|
||
|
||
void parser::clear_expr_locals() {
|
||
m_local_decls = local_expr_decls();
|
||
}
|
||
|
||
void parser::add_local_level(name const & n, level const & l, bool is_variable) {
|
||
if (m_local_level_decls.contains(n))
|
||
maybe_throw_error({sstream() << "invalid universe declaration, '" << n << "' shadows a local universe", pos()});
|
||
m_local_level_decls.insert(n, l);
|
||
if (is_variable) {
|
||
lean_assert(is_param(l));
|
||
m_level_variables.insert(n);
|
||
}
|
||
}
|
||
|
||
void parser::add_local_expr(name const & n, expr const & p, bool is_variable) {
|
||
if (!m_in_quote) {
|
||
// HACK: Certainly not in a pattern. We need this information early in `builtin_exprs::parse_quoted_expr`.
|
||
// Without it, the quotation would be elaborated only in `patexpr_to_expr`, with the local not being
|
||
// in the context any more. For example, in
|
||
//
|
||
// do foo (fun x, ```(%%x))
|
||
//
|
||
// the quotation would be elaborated only after parsing the full application of `foo`.
|
||
m_in_pattern = false;
|
||
}
|
||
m_local_decls.insert(n, p);
|
||
if (is_variable) {
|
||
lean_assert(is_local(p));
|
||
m_variables.insert(mlocal_name(p));
|
||
}
|
||
}
|
||
|
||
environment parser::add_local_ref(environment const & env, name const & n, expr const & ref) {
|
||
add_local_expr(n, ref, false);
|
||
if (is_as_atomic(ref)) {
|
||
buffer<expr> args;
|
||
expr f = get_app_args(get_as_atomic_arg(ref), args);
|
||
if (is_explicit(f))
|
||
f = get_explicit_arg(f);
|
||
if (is_constant(f)) {
|
||
return ::lean::add_local_ref(env, const_name(f), ref);
|
||
} else {
|
||
return env;
|
||
}
|
||
} else if (is_constant(ref) && const_levels(ref)) {
|
||
return ::lean::add_local_ref(env, const_name(ref), ref);
|
||
} else {
|
||
return env;
|
||
}
|
||
}
|
||
|
||
static void check_no_metavars(name const & n, expr const & e) {
|
||
lean_assert(is_local(e));
|
||
if (has_metavar(e)) {
|
||
throw generic_exception(none_expr(), [=](formatter const & fmt) {
|
||
format r("failed to add declaration '");
|
||
r += format(n);
|
||
r += format("' to local context, type has metavariables");
|
||
r += pp_until_meta_visible(fmt, mlocal_type(e));
|
||
return r;
|
||
});
|
||
}
|
||
}
|
||
|
||
void parser::add_variable(name const & n, expr const & v) {
|
||
lean_assert(is_local(v));
|
||
check_no_metavars(n, v);
|
||
add_local_expr(n, v, true);
|
||
}
|
||
|
||
void parser::add_parameter(name const & n, expr const & p) {
|
||
lean_assert(is_local(p));
|
||
check_no_metavars(n, p);
|
||
add_local_expr(n, p, false);
|
||
m_has_params = true;
|
||
}
|
||
|
||
bool parser::is_local_decl(expr const & l) {
|
||
lean_assert(is_local(l));
|
||
// TODO(Leo): add a name_set with internal ids if this is a bottleneck
|
||
for (pair<name, expr> const & p : m_local_decls.get_entries()) {
|
||
if (is_local(p.second) && mlocal_name(p.second) == mlocal_name(l))
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
bool parser::update_local_binder_info(name const & n, binder_info const & bi) {
|
||
auto it = get_local(n);
|
||
if (!it || !is_local(*it)) return false;
|
||
|
||
buffer<pair<name, expr>> entries;
|
||
to_buffer(m_local_decls.get_entries(), entries);
|
||
std::reverse(entries.begin(), entries.end());
|
||
unsigned idx = m_local_decls.find_idx(n);
|
||
lean_assert(idx > 0);
|
||
lean_assert_eq(entries[idx-1].second, *it);
|
||
|
||
buffer<expr> old_locals;
|
||
buffer<expr> new_locals;
|
||
old_locals.push_back(*it);
|
||
expr new_l = update_local(*it, bi);
|
||
entries[idx-1].second = new_l;
|
||
new_locals.push_back(new_l);
|
||
|
||
for (unsigned i = idx; i < entries.size(); i++) {
|
||
expr const & curr_e = entries[i].second;
|
||
expr r = is_local(curr_e) ? mlocal_type(curr_e) : curr_e;
|
||
if (std::any_of(old_locals.begin(), old_locals.end(), [&](expr const & l) { return depends_on(r, l); })) {
|
||
r = replace_locals(r, old_locals, new_locals);
|
||
if (is_local(curr_e)) {
|
||
expr new_e = update_mlocal(curr_e, r);
|
||
entries[i].second = new_e;
|
||
old_locals.push_back(curr_e);
|
||
new_locals.push_back(new_e);
|
||
} else {
|
||
entries[i].second = r;
|
||
}
|
||
}
|
||
}
|
||
auto new_entries = m_local_decls.get_entries();
|
||
unsigned sz_to_updt = entries.size() - idx + 1;
|
||
for (unsigned i = 0; i < sz_to_updt; i++)
|
||
new_entries = tail(new_entries); // remove entries that will be updated
|
||
for (unsigned i = idx-1; i < entries.size(); i++)
|
||
new_entries = cons(entries[i], new_entries);
|
||
m_local_decls.update_entries(new_entries);
|
||
return true;
|
||
}
|
||
|
||
unsigned parser::get_local_index(name const & n) const {
|
||
return m_local_decls.find_idx(n);
|
||
}
|
||
|
||
void parser::get_include_variables(buffer<expr> & vars) const {
|
||
m_include_vars.for_each([&](name const & n) {
|
||
vars.push_back(*get_local(n));
|
||
});
|
||
}
|
||
|
||
list<expr> parser::locals_to_context() const {
|
||
return map_filter<expr>(m_local_decls.get_entries(),
|
||
[](pair<name, expr> const & p, expr & out) {
|
||
out = p.second;
|
||
return is_local(p.second);
|
||
});
|
||
}
|
||
|
||
static unsigned g_level_add_prec = 10;
|
||
|
||
unsigned parser::get_small_nat() {
|
||
mpz val = get_num_val().get_numerator();
|
||
lean_assert(val >= 0);
|
||
if (!val.is_unsigned_int()) {
|
||
maybe_throw_error({"invalid numeral, value does not fit in a machine integer", pos()});
|
||
return 0;
|
||
}
|
||
return val.get_unsigned_int();
|
||
}
|
||
|
||
std::string parser::parse_string_lit() {
|
||
std::string v = get_str_val();
|
||
next();
|
||
return v;
|
||
}
|
||
|
||
unsigned parser::parse_small_nat() {
|
||
unsigned r = 0;
|
||
if (!curr_is_numeral()) {
|
||
auto _ = no_error_recovery_scope_if(curr_is_command());
|
||
maybe_throw_error({"(small) natural number expected", pos()});
|
||
} else {
|
||
r = get_small_nat();
|
||
}
|
||
next();
|
||
return r;
|
||
}
|
||
|
||
double parser::parse_double() {
|
||
double r = 0;
|
||
if (curr() != token_kind::Decimal) {
|
||
auto _ = no_error_recovery_scope_if(curr_is_command());
|
||
maybe_throw_error({"decimal value expected", pos()});
|
||
} else {
|
||
r = get_num_val().get_double();
|
||
}
|
||
next();
|
||
return r;
|
||
}
|
||
|
||
static level lift(level l, unsigned k) {
|
||
while (k > 0) {
|
||
k--;
|
||
l = mk_succ(l);
|
||
}
|
||
return l;
|
||
}
|
||
|
||
unsigned parser::curr_level_lbp() const {
|
||
if (curr_is_token(get_add_tk()))
|
||
return g_level_add_prec;
|
||
else
|
||
return 0;
|
||
}
|
||
|
||
level parser::parse_max_imax(bool is_max) {
|
||
auto p = pos();
|
||
next();
|
||
buffer<level> lvls;
|
||
while (curr_is_identifier() || curr_is_numeral() || curr_is_token(get_lparen_tk())) {
|
||
lvls.push_back(parse_level(get_max_prec()));
|
||
}
|
||
if (lvls.size() < 2) {
|
||
return parser_error_or_level(
|
||
{"invalid level expression, max must have at least two arguments", p});
|
||
}
|
||
unsigned i = lvls.size() - 1;
|
||
level r = lvls[i];
|
||
while (i > 0) {
|
||
--i;
|
||
if (is_max)
|
||
r = mk_max(lvls[i], r);
|
||
else
|
||
r = mk_imax(lvls[i], r);
|
||
}
|
||
return r;
|
||
}
|
||
|
||
level parser::parse_level_id() {
|
||
auto p = pos();
|
||
name id = get_name_val();
|
||
next();
|
||
if (auto it = m_local_level_decls.find(id))
|
||
return *it;
|
||
|
||
return parser_error_or_level({sstream() << "unknown universe '" << id << "'", p});
|
||
}
|
||
|
||
level parser::parse_level_nud() {
|
||
if (curr_is_token_or_id(get_max_tk())) {
|
||
return parse_max_imax(true);
|
||
} else if (curr_is_token_or_id(get_imax_tk())) {
|
||
return parse_max_imax(false);
|
||
} else if (curr_is_token_or_id(get_placeholder_tk())) {
|
||
next();
|
||
return mk_level_placeholder();
|
||
} else if (curr_is_token(get_lparen_tk())) {
|
||
next();
|
||
level l = parse_level();
|
||
check_token_next(get_rparen_tk(), "invalid level expression, ')' expected");
|
||
return l;
|
||
} else if (curr_is_numeral()) {
|
||
unsigned k = parse_small_nat();
|
||
return lift(level(), k);
|
||
} else if (curr_is_identifier()) {
|
||
return parse_level_id();
|
||
} else {
|
||
return parser_error_or_level({"invalid level expression", pos()});
|
||
}
|
||
}
|
||
|
||
level parser::parse_level_led(level left) {
|
||
auto p = pos();
|
||
if (curr_is_token(get_add_tk())) {
|
||
next();
|
||
if (curr_is_numeral()) {
|
||
unsigned k = parse_small_nat();
|
||
return lift(left, k);
|
||
} else {
|
||
return parser_error_or_level(
|
||
{"invalid level expression, right hand side of '+' "
|
||
"(aka universe lift operator) must be a numeral", p});
|
||
}
|
||
} else {
|
||
return parser_error_or_level({"invalid level expression", p});
|
||
}
|
||
}
|
||
|
||
level parser::parse_level(unsigned rbp) {
|
||
level left = parse_level_nud();
|
||
while (rbp < curr_level_lbp()) {
|
||
left = parse_level_led(left);
|
||
}
|
||
return left;
|
||
}
|
||
|
||
pair<expr, level_param_names> parser::elaborate(name const & decl_name,
|
||
metavar_context & mctx, local_context_adapter const & adapter,
|
||
expr const & e, bool check_unassigned) {
|
||
expr tmp_e = adapter.translate_to(e);
|
||
pair<expr, level_param_names> r =
|
||
::lean::elaborate(m_env, get_options(), decl_name, mctx, adapter.lctx(), tmp_e, check_unassigned, m_error_recovery);
|
||
expr new_e = r.first;
|
||
new_e = adapter.translate_from(new_e);
|
||
return mk_pair(new_e, r.second);
|
||
}
|
||
|
||
pair<expr, level_param_names> parser::elaborate(name const & decl_name, metavar_context & mctx, list<expr> const & lctx, expr const & e, bool check_unassigned) {
|
||
local_context_adapter adapter(lctx);
|
||
return elaborate(decl_name, mctx, adapter, e, check_unassigned);
|
||
}
|
||
|
||
pair<expr, level_param_names> parser::elaborate(name const & decl_name, metavar_context & mctx, expr const & e, bool check_unassigned) {
|
||
local_context_adapter adapter(m_local_decls);
|
||
return elaborate(decl_name, mctx, adapter, e, check_unassigned);
|
||
}
|
||
|
||
pair<expr, level_param_names> parser::elaborate(name const & decl_name, list<expr> const & ctx, expr const & e) {
|
||
metavar_context mctx;
|
||
return elaborate(decl_name, mctx, ctx, e, true);
|
||
}
|
||
|
||
pair<expr, level_param_names> parser::elaborate_type(name const & decl_name, list<expr> const & ctx, expr const & e) {
|
||
metavar_context mctx;
|
||
expr Type = copy_tag(e, mk_sort(mk_level_placeholder()));
|
||
expr new_e = copy_tag(e, mk_typed_expr(Type, e));
|
||
return elaborate(decl_name, mctx, ctx, new_e, true);
|
||
}
|
||
|
||
pair<expr, level_param_names> parser::elaborate_type(name const & decl_name, metavar_context & mctx, expr const & e) {
|
||
expr Type = copy_tag(e, mk_sort(mk_level_placeholder()));
|
||
expr new_e = copy_tag(e, mk_typed_expr(Type, e));
|
||
return elaborate(decl_name, mctx, new_e, true);
|
||
}
|
||
|
||
void parser::throw_invalid_open_binder(pos_info const & pos) {
|
||
maybe_throw_error({"invalid binder, '(', '{', '[', '{{', '⦃' or identifier expected", pos});
|
||
}
|
||
|
||
/**
|
||
\brief Return an optional binder_info object based on the current token
|
||
- '(' : default
|
||
- '{' : implicit
|
||
- '{{' or '⦃' : strict implicit
|
||
- '[' : inst_implicit (i.e., implicit argument that should be
|
||
synthesized using type class resolution)
|
||
|
||
If simple_only, then only `(` is considered
|
||
*/
|
||
optional<binder_info> parser::parse_optional_binder_info(bool simple_only) {
|
||
if (curr_is_token(get_lparen_tk())) {
|
||
next();
|
||
return some(binder_info());
|
||
} else if (simple_only) {
|
||
return optional<binder_info>();
|
||
} else if (curr_is_token(get_lcurly_tk())) {
|
||
next();
|
||
if (curr_is_token(get_lcurly_tk())) {
|
||
next();
|
||
return some(mk_strict_implicit_binder_info());
|
||
} else {
|
||
return some(mk_implicit_binder_info());
|
||
}
|
||
} else if (curr_is_token(get_lbracket_tk())) {
|
||
next();
|
||
return some(mk_inst_implicit_binder_info());
|
||
} else if (curr_is_token(get_ldcurly_tk())) {
|
||
next();
|
||
return some(mk_strict_implicit_binder_info());
|
||
} else {
|
||
return optional<binder_info>();
|
||
}
|
||
}
|
||
|
||
/**
|
||
\brief Return binder_info object based on the current token, it fails if the current token
|
||
is not '(', '{', '{{', '⦃', or '['
|
||
|
||
\see parse_optional_binder_info
|
||
*/
|
||
binder_info parser::parse_binder_info(bool simple_only) {
|
||
auto p = pos();
|
||
if (auto bi = parse_optional_binder_info(simple_only)) {
|
||
return *bi;
|
||
} else {
|
||
throw_invalid_open_binder(p);
|
||
return binder_info();
|
||
}
|
||
}
|
||
|
||
/**
|
||
\brief Consume the next token based on the value of \c bi
|
||
- none : do not consume anything
|
||
- default : consume ')'
|
||
- implicit : consume '}'
|
||
- strict implicit : consume '}}' or '⦄'
|
||
- inst implicit : consume ']'
|
||
*/
|
||
void parser::parse_close_binder_info(optional<binder_info> const & bi) {
|
||
if (!bi) {
|
||
return;
|
||
} else if (bi->is_implicit()) {
|
||
check_token_next(get_rcurly_tk(), "invalid declaration, '}' expected");
|
||
} else if (bi->is_inst_implicit()) {
|
||
check_token_next(get_rbracket_tk(), "invalid declaration, ']' expected");
|
||
} else if (bi->is_strict_implicit()) {
|
||
if (curr_is_token(get_rcurly_tk())) {
|
||
next();
|
||
check_token_next(get_rcurly_tk(), "invalid declaration, '}' expected");
|
||
} else {
|
||
check_token_next(get_rdcurly_tk(), "invalid declaration, '⦄' expected");
|
||
}
|
||
} else {
|
||
check_token_next(get_rparen_tk(), "invalid declaration, ')' expected");
|
||
}
|
||
}
|
||
|
||
/** \brief Parse <tt>ID ':' expr</tt>, where the expression represents the type of the identifier. */
|
||
expr parser::parse_binder_core(binder_info const & bi, unsigned rbp) {
|
||
auto p = pos();
|
||
name id;
|
||
if (curr_is_token(get_placeholder_tk())) {
|
||
id = "_x";
|
||
next();
|
||
} else {
|
||
id = check_atomic_id_next("invalid binder, atomic identifier expected");
|
||
}
|
||
expr type;
|
||
if (curr_is_token(get_colon_tk())) {
|
||
next();
|
||
type = parse_expr(rbp);
|
||
} else {
|
||
type = save_pos(mk_expr_placeholder(), p);
|
||
}
|
||
return save_pos(mk_local(id, type, bi), p);
|
||
}
|
||
|
||
expr parser::parse_binder(unsigned rbp) {
|
||
if (curr_is_identifier()) {
|
||
return parse_binder_core(binder_info(), rbp);
|
||
} else {
|
||
bool simple_only = false;
|
||
binder_info bi = parse_binder_info(simple_only);
|
||
rbp = 0;
|
||
auto r = parse_binder_core(bi, rbp);
|
||
parse_close_binder_info(bi);
|
||
return r;
|
||
}
|
||
}
|
||
|
||
/* Lean allow binders of the form <tt>ID_1 ... ID_n 'op' S</tt>
|
||
Where 'op' is an infix operator, and s an expression (i.e., "collection").
|
||
This notation expands to:
|
||
(ID_1 ... ID_n : _) (H_1 : ID_1 'op' S) ... (H_n : ID_n 'op' S)
|
||
|
||
This method return true if the next token is an infix operator,
|
||
and populates r with the locals above.
|
||
*/
|
||
bool parser::parse_binder_collection(buffer<pair<pos_info, name>> const & names, binder_info const & bi, buffer<expr> & r) {
|
||
if (!curr_is_keyword())
|
||
return false;
|
||
name tk = get_token_info().value();
|
||
list<pair<notation::transition, parse_table>> trans_list = led().find(tk);
|
||
if (length(trans_list) != 1)
|
||
return false;
|
||
pair<notation::transition, parse_table> const & p = head(trans_list);
|
||
list<notation::accepting> const & acc_lst = p.second.is_accepting();
|
||
if (length(acc_lst) != 1)
|
||
return false; // no overloading
|
||
notation::accepting const & acc = head(acc_lst);
|
||
lean_assert(!acc.get_postponed());
|
||
expr pred = acc.get_expr();
|
||
auto k = p.first.get_action().kind();
|
||
if (k == notation::action_kind::Skip ||
|
||
k == notation::action_kind::Ext)
|
||
return false;
|
||
unsigned rbp = p.first.get_action().rbp();
|
||
next(); // consume tk
|
||
expr S = parse_expr(rbp);
|
||
unsigned old_sz = r.size();
|
||
/* Add (ID_1 ... ID_n : _) to r */
|
||
for (auto p : names) {
|
||
expr arg_type = save_pos(mk_expr_placeholder(), p.first);
|
||
expr local = save_pos(mk_local(p.second, arg_type, bi), p.first);
|
||
add_local(local);
|
||
r.push_back(local);
|
||
}
|
||
/* Add (H_1 : ID_1 'op' S) ... (H_n : ID_n 'op' S) */
|
||
unsigned i = old_sz;
|
||
for (auto p : names) {
|
||
expr ID = r[i];
|
||
expr args[2] = {ID, S};
|
||
expr ID_op_S = instantiate_rev(pred, 2, args);
|
||
expr local = save_pos(mk_local("H", ID_op_S, bi), p.first);
|
||
add_local(local);
|
||
r.push_back(local);
|
||
i++;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
\brief Parse <tt>ID ... ID ':' expr</tt>, where the expression
|
||
represents the type of the identifiers.
|
||
*/
|
||
void parser::parse_binder_block(buffer<expr> & r, binder_info const & bi, unsigned rbp, bool allow_default) {
|
||
buffer<pair<pos_info, name>> names;
|
||
while (curr_is_identifier() || curr_is_token(get_placeholder_tk())) {
|
||
auto p = pos();
|
||
if (curr_is_identifier()) {
|
||
names.emplace_back(p, check_atomic_id_next("invalid binder, atomic identifier expected"));
|
||
} else {
|
||
names.emplace_back(p, "_x");
|
||
next();
|
||
}
|
||
}
|
||
if (names.empty())
|
||
return maybe_throw_error({"invalid binder, identifier expected", pos()});
|
||
optional<expr> type;
|
||
if (curr_is_token(get_colon_tk())) {
|
||
next();
|
||
type = parse_expr(rbp);
|
||
if (allow_default && curr_is_token(get_assign_tk())) {
|
||
next();
|
||
expr val = parse_expr(rbp);
|
||
type = mk_opt_param(*type, val);
|
||
} else if (allow_default && curr_is_token(get_period_tk())) {
|
||
type = parse_auto_param(*this, *type);
|
||
}
|
||
} else if (allow_default && curr_is_token(get_assign_tk())) {
|
||
next();
|
||
expr val = parse_expr(rbp);
|
||
type = mk_opt_param(copy_tag(val, mk_expr_placeholder()), val);
|
||
} else if (parse_binder_collection(names, bi, r)) {
|
||
return;
|
||
}
|
||
for (auto p : names) {
|
||
expr arg_type = type ? *type : save_pos(mk_expr_placeholder(), p.first);
|
||
expr local = save_pos(mk_local(p.second, arg_type, bi), p.first);
|
||
add_local(local);
|
||
r.push_back(local);
|
||
}
|
||
}
|
||
|
||
expr parser::parse_inst_implicit_decl() {
|
||
binder_info bi = mk_inst_implicit_binder_info();
|
||
auto id_pos = pos();
|
||
name id;
|
||
expr type;
|
||
if (curr_is_identifier()) {
|
||
id = get_name_val();
|
||
next();
|
||
if (curr_is_token(get_colon_tk())) {
|
||
next();
|
||
type = parse_expr();
|
||
} else {
|
||
expr left = id_to_expr(id, id_pos);
|
||
id = mk_anonymous_inst_name();
|
||
unsigned rbp = 0;
|
||
while (rbp < curr_lbp()) {
|
||
left = parse_led(left);
|
||
}
|
||
type = left;
|
||
}
|
||
} else {
|
||
id = mk_anonymous_inst_name();
|
||
type = parse_expr();
|
||
}
|
||
expr local = save_pos(mk_local(id, type, bi), id_pos);
|
||
add_local(local);
|
||
return local;
|
||
}
|
||
|
||
|
||
void parser::parse_inst_implicit_decl(buffer<expr> & r) {
|
||
expr local = parse_inst_implicit_decl();
|
||
r.push_back(local);
|
||
}
|
||
|
||
void parser::parse_binders_core(buffer<expr> & r, parse_binders_config & cfg) {
|
||
bool first = true;
|
||
while (true) {
|
||
if (curr_is_identifier() || curr_is_token(get_placeholder_tk())) {
|
||
if (cfg.m_explicit_delimiters) {
|
||
throw parser_error("invalid binder declaration, delimiter/bracket expected (i.e., '(', '{', '[', '{{')",
|
||
pos());
|
||
}
|
||
/* We only allow the default parameter value syntax for declarations with
|
||
surrounded by () */
|
||
bool new_allow_default = false;
|
||
parse_binder_block(r, binder_info(), cfg.m_rbp, new_allow_default);
|
||
cfg.m_last_block_delimited = false;
|
||
} else {
|
||
/* We only allow the default parameter value syntax for declarations with
|
||
surrounded by () */
|
||
bool new_allow_default = cfg.m_allow_default && curr_is_token(get_lparen_tk());
|
||
optional<binder_info> bi = parse_optional_binder_info(cfg.m_simple_only);
|
||
if (bi) {
|
||
if (first && cfg.m_infer_kind != nullptr) {
|
||
/* Parse {} or () prefix */
|
||
if (bi->is_implicit() && curr_is_token(get_rcurly_tk())) {
|
||
next();
|
||
*cfg.m_infer_kind = implicit_infer_kind::RelaxedImplicit;
|
||
first = false;
|
||
continue;
|
||
} else if (is_explicit(*bi) && curr_is_token(get_rparen_tk())) {
|
||
next();
|
||
*cfg.m_infer_kind = implicit_infer_kind::None;
|
||
first = false;
|
||
continue;
|
||
} else {
|
||
*cfg.m_infer_kind = implicit_infer_kind::Implicit;
|
||
}
|
||
}
|
||
unsigned rbp = 0;
|
||
cfg.m_last_block_delimited = true;
|
||
if (bi->is_inst_implicit()) {
|
||
parse_inst_implicit_decl(r);
|
||
} else {
|
||
if (cfg.m_simple_only || !parse_local_notation_decl(cfg.m_nentries))
|
||
parse_binder_block(r, *bi, rbp, new_allow_default);
|
||
}
|
||
parse_close_binder_info(bi);
|
||
} else {
|
||
return;
|
||
}
|
||
}
|
||
first = false;
|
||
}
|
||
}
|
||
|
||
local_environment parser::parse_binders(buffer<expr> & r, parse_binders_config & cfg) {
|
||
flet<environment> save1(m_env, m_env); // save environment
|
||
flet<local_expr_decls> save2(m_local_decls, m_local_decls);
|
||
unsigned old_sz = r.size();
|
||
parse_binders_core(r, cfg);
|
||
if (!cfg.m_allow_empty && old_sz == r.size())
|
||
throw_invalid_open_binder(pos());
|
||
return local_environment(m_env);
|
||
}
|
||
|
||
bool parser::parse_local_notation_decl(buffer<notation_entry> * nentries) {
|
||
if (curr_is_notation_decl(*this)) {
|
||
parser::in_notation_ctx ctx(*this);
|
||
buffer<token_entry> new_tokens;
|
||
bool overload = false;
|
||
bool allow_local = true;
|
||
auto ne = ::lean::parse_notation(*this, overload, new_tokens, allow_local);
|
||
for (auto const & te : new_tokens)
|
||
m_env = add_token(m_env, te);
|
||
if (nentries) nentries->push_back(ne);
|
||
m_env = add_notation(m_env, ne);
|
||
return true;
|
||
} else {
|
||
return false;
|
||
}
|
||
}
|
||
|
||
void parser::process_postponed(buffer<expr> const & args, bool is_left,
|
||
buffer<notation::action_kind> const & kinds,
|
||
buffer<list<expr>> const & nargs,
|
||
buffer<expr> const & ps,
|
||
buffer<pair<unsigned, pos_info>> const & scoped_info,
|
||
list<notation::action> const & postponed,
|
||
pos_info const & p,
|
||
buffer<expr> & new_args) {
|
||
unsigned args_idx = 0;
|
||
if (is_left) {
|
||
new_args.push_back(args[0]);
|
||
args_idx = 1;
|
||
}
|
||
unsigned kinds_idx = 0;
|
||
unsigned nargs_idx = 0;
|
||
unsigned scoped_idx = 0;
|
||
list<notation::action> it = postponed;
|
||
for (; kinds_idx < kinds.size(); kinds_idx++, args_idx++) {
|
||
auto k = kinds[kinds_idx];
|
||
switch (k) {
|
||
case notation::action_kind::Exprs: {
|
||
if (!it || head(it).kind() != k || nargs_idx >= nargs.size())
|
||
throw exception("ill-formed parsing tables");
|
||
notation::action const & a = head(it);
|
||
buffer<expr> r_args;
|
||
to_buffer(nargs[nargs_idx], r_args);
|
||
nargs_idx++;
|
||
expr rec = copy_with_new_pos(a.get_rec(), p);
|
||
expr r;
|
||
if (a.is_fold_right()) {
|
||
if (a.get_initial()) {
|
||
r = instantiate_rev(copy_with_new_pos(*a.get_initial(), p), new_args.size(), new_args.data());
|
||
} else {
|
||
r = r_args.back();
|
||
r_args.pop_back();
|
||
}
|
||
unsigned i = r_args.size();
|
||
while (i > 0) {
|
||
--i;
|
||
new_args.push_back(r_args[i]);
|
||
new_args.push_back(r);
|
||
r = instantiate_rev(rec, new_args.size(), new_args.data());
|
||
new_args.pop_back(); new_args.pop_back();
|
||
}
|
||
} else {
|
||
unsigned fidx = 0;
|
||
if (a.get_initial()) {
|
||
r = instantiate_rev(copy_with_new_pos(*a.get_initial(), p), new_args.size(), new_args.data());
|
||
} else {
|
||
r = r_args[0];
|
||
fidx++;
|
||
}
|
||
for (unsigned i = fidx; i < r_args.size(); i++) {
|
||
new_args.push_back(r_args[i]);
|
||
new_args.push_back(r);
|
||
r = instantiate_rev(rec, new_args.size(), new_args.data());
|
||
new_args.pop_back(); new_args.pop_back();
|
||
}
|
||
}
|
||
new_args.push_back(r);
|
||
it = tail(it);
|
||
break;
|
||
}
|
||
case notation::action_kind::ScopedExpr: {
|
||
if (!it || head(it).kind() != k || scoped_idx >= scoped_info.size())
|
||
throw exception("ill-formed parsing tables");
|
||
expr r = args[args_idx];
|
||
notation::action const & a = head(it);
|
||
bool no_cache = false;
|
||
unsigned ps_sz = scoped_info[scoped_idx].first;
|
||
pos_info binder_pos = scoped_info[scoped_idx].second;
|
||
scoped_idx++;
|
||
if (is_var(a.get_rec(), 0)) {
|
||
if (a.use_lambda_abstraction())
|
||
r = Fun(ps_sz, ps.data(), r, no_cache);
|
||
else
|
||
r = Pi(ps_sz, ps.data(), r, no_cache);
|
||
r = rec_save_pos(r, binder_pos);
|
||
} else {
|
||
expr rec = copy_with_new_pos(a.get_rec(), p);
|
||
unsigned i = ps_sz;
|
||
while (i > 0) {
|
||
--i;
|
||
expr const & l = ps[i];
|
||
if (a.use_lambda_abstraction())
|
||
r = Fun(l, r, no_cache);
|
||
else
|
||
r = Pi(l, r, no_cache);
|
||
r = save_pos(r, binder_pos);
|
||
new_args.push_back(r);
|
||
r = instantiate_rev(rec, new_args.size(), new_args.data());
|
||
new_args.pop_back();
|
||
}
|
||
}
|
||
new_args.push_back(r);
|
||
it = tail(it);
|
||
break;
|
||
}
|
||
default:
|
||
new_args.push_back(args[args_idx]);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
// Return true iff the current token is the terminator of some Exprs action, and store the matching pair in r
|
||
static bool curr_is_terminator_of_exprs_action(parser const & p, list<pair<notation::transition, parse_table>> const & lst, pair<notation::transition, parse_table> const * & r) {
|
||
for (auto const & pr : lst) {
|
||
notation::action const & a = pr.first.get_action();
|
||
if (a.kind() == notation::action_kind::Exprs &&
|
||
a.get_terminator() &&
|
||
p.curr_is_token(name(utf8_trim(a.get_terminator()->to_string())))) {
|
||
r = ≺
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
// Return true iff \c lst contains a Skip action, and store the matching pair in r.
|
||
static bool has_skip(list<pair<notation::transition, parse_table>> const & lst, pair<notation::transition, parse_table> const * & r) {
|
||
for (auto const & p : lst) {
|
||
notation::action const & a = p.first.get_action();
|
||
if (a.kind() == notation::action_kind::Skip) {
|
||
r = &p;
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
static pair<notation::transition, parse_table> const * get_non_skip(list<pair<notation::transition, parse_table>> const & lst) {
|
||
for (auto const & p : lst) {
|
||
notation::action const & a = p.first.get_action();
|
||
if (a.kind() != notation::action_kind::Skip)
|
||
return &p;
|
||
}
|
||
return nullptr;
|
||
}
|
||
|
||
expr parser::parse_notation(parse_table t, expr * left) {
|
||
check_system("parse_notation");
|
||
lean_assert(curr() == token_kind::Keyword);
|
||
auto p = pos();
|
||
auto first_token = get_token_info().value();
|
||
auto check_break = [&]() {
|
||
try {
|
||
check_break_at_pos(break_at_pos_exception::token_context::notation);
|
||
} catch (break_at_pos_exception & e) {
|
||
// info is stored at position of first notation token
|
||
e.m_token_info.m_pos = p;
|
||
throw;
|
||
}
|
||
};
|
||
buffer<expr> args;
|
||
buffer<notation::action_kind> kinds;
|
||
buffer<list<expr>> nargs; // nary args
|
||
buffer<expr> ps;
|
||
buffer<pair<unsigned, pos_info>> scoped_info; // size of ps and binder_pos for scoped_exprs
|
||
// Invariants:
|
||
// args.size() == kinds.size() if not left
|
||
// args.size() == kinds.size()+1 if left
|
||
// nargs.size() == number of Exprs in kinds
|
||
// scoped_info.size() == number of Scoped_Exprs in kinds
|
||
bool has_Exprs = false;
|
||
local_environment lenv(m_env);
|
||
pos_info binder_pos;
|
||
if (left)
|
||
args.push_back(*left);
|
||
while (true) {
|
||
if (curr() != token_kind::Keyword)
|
||
break;
|
||
auto r = t.find(get_token_info().value());
|
||
if (!r)
|
||
break;
|
||
check_break();
|
||
pair<notation::transition, parse_table> const * curr_pair = nullptr;
|
||
if (tail(r)) {
|
||
// There is more than one possible actions.
|
||
// In the current implementation, we support the following possible cases (Skip, Expr), (Skip, Exprs) amd (Skip, ScopedExpr)
|
||
next();
|
||
if (curr_is_terminator_of_exprs_action(*this, r, curr_pair)) {
|
||
lean_assert(curr_pair->first.get_action().kind() == notation::action_kind::Exprs);
|
||
} else if (has_skip(r, curr_pair) && !curr_starts_expr()) {
|
||
lean_assert(curr_pair->first.get_action().kind() == notation::action_kind::Skip);
|
||
} else {
|
||
curr_pair = get_non_skip(r);
|
||
}
|
||
} else {
|
||
// there is only one possible action
|
||
curr_pair = &head(r);
|
||
if (curr_pair->first.get_action().kind() != notation::action_kind::Ext)
|
||
next();
|
||
}
|
||
lean_assert(curr_pair);
|
||
notation::action const & a = curr_pair->first.get_action();
|
||
switch (a.kind()) {
|
||
case notation::action_kind::Skip:
|
||
break;
|
||
case notation::action_kind::Expr:
|
||
args.push_back(parse_expr(a.rbp()));
|
||
kinds.push_back(a.kind());
|
||
break;
|
||
case notation::action_kind::Exprs: {
|
||
buffer<expr> r_args;
|
||
auto terminator = a.get_terminator();
|
||
if (terminator)
|
||
terminator = some(name(utf8_trim(terminator->to_string()))); // remove padding
|
||
if (!terminator || !curr_is_token(*terminator)) {
|
||
r_args.push_back(parse_expr(a.rbp()));
|
||
name sep = utf8_trim(a.get_sep().to_string()); // remove padding
|
||
while (curr_is_token(sep)) {
|
||
check_break();
|
||
next();
|
||
r_args.push_back(parse_expr(a.rbp()));
|
||
}
|
||
}
|
||
if (terminator) {
|
||
check_break();
|
||
if (curr_is_token(*terminator)) {
|
||
next();
|
||
} else {
|
||
maybe_throw_error({sstream() << "invalid composite expression, '"
|
||
<< *terminator << "' expected" , pos()});
|
||
}
|
||
}
|
||
has_Exprs = true;
|
||
args.push_back(expr()); // placeholder
|
||
kinds.push_back(a.kind());
|
||
nargs.push_back(to_list(r_args));
|
||
break;
|
||
}
|
||
case notation::action_kind::Binder:
|
||
binder_pos = pos();
|
||
ps.push_back(parse_binder(a.rbp()));
|
||
break;
|
||
case notation::action_kind::Binders:
|
||
binder_pos = pos();
|
||
lenv = parse_binders(ps, a.rbp());
|
||
break;
|
||
case notation::action_kind::ScopedExpr: {
|
||
expr r = parse_scoped_expr(ps, lenv, a.rbp());
|
||
args.push_back(r);
|
||
kinds.push_back(a.kind());
|
||
scoped_info.push_back(mk_pair(ps.size(), binder_pos));
|
||
break;
|
||
}
|
||
case notation::action_kind::Ext:
|
||
args.push_back(a.get_parse_fn()(*this, args.size(), args.data(), p));
|
||
kinds.push_back(a.kind());
|
||
break;
|
||
}
|
||
t = curr_pair->second;
|
||
}
|
||
list<notation::accepting> const & as = t.is_accepting();
|
||
if (is_nil(as)) {
|
||
if (m_backtracking_pos && !consumed_input()) throw backtracking_exception();
|
||
// TODO(gabriel): search children of t for accepting states
|
||
sstream msg;
|
||
msg << "invalid expression";
|
||
if (p != pos()) {
|
||
msg << " starting at " << p.first << ":" << p.second;
|
||
}
|
||
return parser_error_or_expr({msg, pos()});
|
||
}
|
||
lean_assert(left || args.size() == kinds.size());
|
||
lean_assert(!left || args.size() == kinds.size() + 1);
|
||
/*
|
||
IF there are multiple choices and Exprs was not used, we create a lambda for each choice.
|
||
In this case, we copy args to actual_args and store local constants in args. */
|
||
buffer<expr> actual_args;
|
||
buffer<expr> cs;
|
||
bool create_lambdas = length(as) > 1 && !has_Exprs;
|
||
if (create_lambdas) {
|
||
name x("x");
|
||
unsigned idx = 1;
|
||
for (expr & arg : args) {
|
||
actual_args.push_back(arg);
|
||
arg = mk_local(next_name(), x.append_after(idx), mk_expr_placeholder(), binder_info());
|
||
idx++;
|
||
}
|
||
}
|
||
for (notation::accepting const & a : as) {
|
||
expr a_expr = a.get_expr();
|
||
if (m_in_quote)
|
||
a_expr = freeze_names(a_expr);
|
||
expr r = copy_with_new_pos(a_expr, p);
|
||
list<notation::action> const & postponed = a.get_postponed();
|
||
if (postponed) {
|
||
buffer<expr> new_args;
|
||
process_postponed(args, left, kinds, nargs, ps, scoped_info, postponed, p, new_args);
|
||
lean_assert(!args.empty());
|
||
r = instantiate_rev(r, new_args.size(), new_args.data());
|
||
} else {
|
||
lean_assert(nargs.empty() && scoped_info.empty());
|
||
r = instantiate_rev(r, args.size(), args.data());
|
||
}
|
||
if (create_lambdas) {
|
||
bool no_cache = false;
|
||
r = rec_save_pos(eta_reduce(Fun(args, r, no_cache)), p);
|
||
}
|
||
cs.push_back(r);
|
||
}
|
||
expr r = save_pos(mk_choice(cs.size(), cs.data()), p);
|
||
if (create_lambdas) {
|
||
r = rec_save_pos(::lean::mk_app(r, actual_args), p);
|
||
}
|
||
return r;
|
||
}
|
||
|
||
expr parser::parse_nud_notation() {
|
||
return parse_notation(nud(), nullptr);
|
||
}
|
||
|
||
expr parser::parse_inaccessible() {
|
||
auto p = pos();
|
||
next();
|
||
expr t = parse_expr(get_max_prec());
|
||
return save_pos(mk_inaccessible(t), p);
|
||
}
|
||
|
||
expr parser::parse_placeholder() {
|
||
auto p = pos();
|
||
next();
|
||
return save_pos(mk_explicit_expr_placeholder(), p);
|
||
}
|
||
|
||
expr parser::parse_anonymous_var_pattern() {
|
||
auto p = pos();
|
||
next();
|
||
expr t = mk_local(next_name(), "_x", mk_expr_placeholder(), binder_info());
|
||
return save_pos(t, p);
|
||
}
|
||
|
||
expr parser::parse_led_notation(expr left) {
|
||
if (led().find(get_token_info().value())) {
|
||
return parse_notation(led(), &left);
|
||
} else {
|
||
return mk_app(left, parse_expr(get_max_prec()), pos_of(left));
|
||
}
|
||
}
|
||
|
||
/**
|
||
\brief Auxiliary object for converting pattern_or_expr into a pattern.
|
||
The main points are:
|
||
|
||
1- Collect all pattern variables. Each pattern variable can only be
|
||
"declared" once. That is, the following equation is not valid
|
||
|
||
definition f : nat -> nat -> nat
|
||
| a a := a
|
||
|
||
2- We do not collect pattern variables inside inaccessible term such as:
|
||
.(f a)
|
||
|
||
|
||
Remark: An inaccessible term may contain a reference to a variable defined
|
||
later. Here is an example:
|
||
|
||
inductive imf {A B : Type} (f : A → B) : B → Type
|
||
| mk : ∀ (a : A), imf (f a)
|
||
|
||
definition inv {A B : Type} (f : A → B) : ∀ (b : B), imf f b → A
|
||
| .(f a) (imf.mk .f a) := a
|
||
|
||
The 'a' in .(f a) is a reference to the variable 'a' being declared at
|
||
(imf.mk .f a)
|
||
|
||
3- The type in type ascriptions is implicitly marked as inaccessible.
|
||
|
||
4- An inaccessible term cannot be the function in an application.
|
||
Example: (.f a) is not allowed.
|
||
|
||
5- In a pattern (f a), f must be a constructor or a constant tagged with the
|
||
[pattern] attribute */
|
||
struct to_pattern_fn {
|
||
parser & m_parser;
|
||
buffer<expr> & m_new_locals;
|
||
name_map<expr> m_locals_map; // local variable name --> its interpretation
|
||
expr_map<expr> m_anonymous_vars; // for _
|
||
|
||
|
||
to_pattern_fn(parser & p, buffer<expr> & new_locals):
|
||
m_parser(p), m_new_locals(new_locals) {}
|
||
|
||
environment const & env() { return m_parser.env(); }
|
||
|
||
/* Return true iff the constant n may occur in a pattern */
|
||
bool is_pattern_constant(name const & n) {
|
||
if (inductive::is_intro_rule(env(), n))
|
||
return true;
|
||
if (has_pattern_attribute(env(), n))
|
||
return true;
|
||
return false;
|
||
}
|
||
|
||
/* (Try to) process a choice-expression.
|
||
Return "some" if all alternatives are constants. The result will contain only the alternatives
|
||
that can occur in patterns.
|
||
*/
|
||
optional<expr> process_choice(expr const & e, expr const & ref) {
|
||
lean_assert(is_choice(e));
|
||
bool all_constant = true;
|
||
buffer<expr> pattern_constants;
|
||
for (unsigned i = 0; i < get_num_choices(e); i++) {
|
||
expr const & c = get_choice(e, i);
|
||
if (is_constant(c)) {
|
||
if (is_pattern_constant(const_name(c)))
|
||
pattern_constants.push_back(c);
|
||
} else {
|
||
all_constant = false;
|
||
}
|
||
}
|
||
if (pattern_constants.size() == get_num_choices(e)) {
|
||
return some_expr(e);
|
||
} else if (!pattern_constants.empty()) {
|
||
if (all_constant) {
|
||
/* Filter overloads that cannot occur in patterns. */
|
||
return some_expr(copy_tag(e, mk_choice(pattern_constants.size(), pattern_constants.data())));
|
||
} else {
|
||
m_parser.maybe_throw_error({
|
||
sstream() << "invalid pattern, '" << ref << "' is overloaded, "
|
||
<< "and some interpretations may occur in patterns and others not "
|
||
<< "(solution: use fully qualified names)",
|
||
m_parser.pos_of(ref)});
|
||
return none_expr();
|
||
}
|
||
} else {
|
||
return none_expr();
|
||
}
|
||
}
|
||
|
||
void add_new_local(expr const & l) {
|
||
name const & n = mlocal_pp_name(l);
|
||
if (!n.is_atomic()) {
|
||
return m_parser.maybe_throw_error({
|
||
"invalid pattern: variable, constructor or constant tagged as pattern expected",
|
||
m_parser.pos_of(l)});
|
||
}
|
||
if (m_locals_map.contains(n)) {
|
||
return m_parser.maybe_throw_error({
|
||
sstream() << "invalid pattern, '" << n << "' already appeared in this pattern",
|
||
m_parser.pos_of(l)});
|
||
}
|
||
m_locals_map.insert(n, l);
|
||
m_new_locals.push_back(l);
|
||
}
|
||
|
||
void collect_new_local(expr const & e) {
|
||
name const & n = mlocal_pp_name(e);
|
||
bool resolve_only = true;
|
||
expr new_e = m_parser.id_to_expr(n, m_parser.pos_of(e), resolve_only);
|
||
if (is_as_atomic(new_e)) {
|
||
new_e = get_app_fn(get_as_atomic_arg(new_e));
|
||
if (is_explicit(new_e))
|
||
new_e = get_explicit_arg(new_e);
|
||
}
|
||
|
||
if (is_constant(new_e) && is_pattern_constant(const_name(new_e))) {
|
||
m_locals_map.insert(n, new_e);
|
||
return;
|
||
} else if (is_choice(new_e)) {
|
||
if (auto r = process_choice(new_e, e)) {
|
||
m_locals_map.insert(n, *r);
|
||
return;
|
||
} else {
|
||
// assume e is a variable shadowing overloaded constants
|
||
}
|
||
}
|
||
add_new_local(e);
|
||
}
|
||
|
||
void collect_new_locals(expr const & e, bool skip_main_fn) {
|
||
if (is_typed_expr(e)) {
|
||
collect_new_locals(get_typed_expr_expr(e), false);
|
||
} else if (is_prenum(e) || is_string_macro(e)) {
|
||
// do nothing
|
||
} else if (is_inaccessible(e)) {
|
||
// do nothing
|
||
} else if (is_placeholder(e)) {
|
||
expr r = copy_tag(e, mk_local(m_parser.next_name(), "_x", copy_tag(e, mk_expr_placeholder()), binder_info()));
|
||
m_new_locals.push_back(r);
|
||
m_anonymous_vars.insert(mk_pair(e, r));
|
||
} else if (is_as_pattern(e)) {
|
||
// skip `collect_new_local`: we assume the lhs to always be a new local
|
||
add_new_local(get_as_pattern_lhs(e));
|
||
collect_new_locals(get_as_pattern_rhs(e), false);
|
||
} else if (is_app(e)) {
|
||
collect_new_locals(app_fn(e), skip_main_fn);
|
||
collect_new_locals(app_arg(e), false);
|
||
} else if (is_choice(e)) {
|
||
if (!process_choice(e, e)) {
|
||
return m_parser.maybe_throw_error({
|
||
sstream() << "invalid pattern, '" << e << "' is overloaded, "
|
||
<< "and this kind of overloading is not currently supported in patterns",
|
||
m_parser.pos_of(e)});
|
||
}
|
||
} else if (is_local(e)) {
|
||
if (skip_main_fn) {
|
||
// do nothing
|
||
} else {
|
||
collect_new_local(e);
|
||
}
|
||
} else if (is_anonymous_constructor(e)) {
|
||
buffer<expr> args;
|
||
get_app_args(get_annotation_arg(e), args);
|
||
for (expr const & arg : args)
|
||
collect_new_locals(arg, skip_main_fn);
|
||
} else if (is_structure_instance(e)) {
|
||
auto info = get_structure_instance_info(e);
|
||
if (info.m_sources.size()) {
|
||
throw parser_error("invalid occurrence of structure notation source in pattern",
|
||
*get_pos_info(info.m_sources[0]));
|
||
}
|
||
for (expr const & val : info.m_field_values)
|
||
collect_new_locals(val, false);
|
||
} else if (is_annotation(e)) {
|
||
collect_new_locals(get_annotation_arg(e), skip_main_fn);
|
||
} else if (is_constant(e) && is_pattern_constant(const_name(e))) {
|
||
// do nothing
|
||
} else {
|
||
return m_parser.maybe_throw_error({
|
||
"invalid pattern, must be an application, "
|
||
"constant, variable, type ascription, aliasing pattern or inaccessible term",
|
||
m_parser.pos_of(e)});
|
||
}
|
||
}
|
||
|
||
expr to_expr(expr const & e) {
|
||
return replace(e, [&](expr const & e, unsigned) {
|
||
if (is_local(e)) {
|
||
if (auto r = m_locals_map.find(mlocal_pp_name(e)))
|
||
return some_expr(*r);
|
||
else
|
||
return some_expr(m_parser.patexpr_to_expr(e));
|
||
}
|
||
return none_expr();
|
||
});
|
||
}
|
||
|
||
expr visit(expr const & e) {
|
||
if (is_typed_expr(e)) {
|
||
expr new_v = visit(get_typed_expr_expr(e));
|
||
expr new_t = to_expr(get_typed_expr_type(e));
|
||
return copy_tag(e, mk_typed_expr(new_t, new_v));
|
||
} else if (is_prenum(e) || is_string_macro(e)) {
|
||
return e;
|
||
} else if (is_inaccessible(e)) {
|
||
return to_expr(e);
|
||
} else if (is_placeholder(e)) {
|
||
return m_anonymous_vars.find(e)->second;
|
||
} else if (is_as_pattern(e)) {
|
||
auto new_rhs = visit(get_as_pattern_rhs(e));
|
||
return copy_tag(e, mk_as_pattern(get_as_pattern_lhs(e), new_rhs));
|
||
} else if (is_app(e)) {
|
||
if (is_inaccessible(app_fn(e))) {
|
||
return m_parser.parser_error_or_expr({
|
||
"invalid inaccessible annotation, it cannot be used around functions in applications",
|
||
m_parser.pos_of(e)});
|
||
}
|
||
expr new_f = visit(app_fn(e));
|
||
expr new_a = visit(app_arg(e));
|
||
return update_app(e, new_f, new_a);
|
||
} else if (is_choice(e)) {
|
||
auto new_e = process_choice(e, e);
|
||
if (!new_e) {
|
||
return m_parser.mk_sorry(m_parser.pos_of(e), true);
|
||
} else if (!is_eqp(*new_e, e)) {
|
||
return visit(*new_e);
|
||
} else {
|
||
buffer<expr> new_args;
|
||
for (unsigned i = 0; i < macro_num_args(e); i++)
|
||
new_args.push_back(visit(macro_arg(e, i)));
|
||
return update_macro(e, new_args.size(), new_args.data());
|
||
}
|
||
} else if (is_local(e)) {
|
||
if (auto r = m_locals_map.find(mlocal_pp_name(e)))
|
||
return *r;
|
||
else
|
||
return e;
|
||
} else if (is_anonymous_constructor(e)) {
|
||
buffer<expr> args;
|
||
expr a = get_annotation_arg(e);
|
||
expr fn = get_app_args(a, args);
|
||
lean_assert(is_placeholder(fn));
|
||
for (expr & arg : args)
|
||
arg = visit(arg);
|
||
expr r = copy_tag(a, mk_app(fn, args));
|
||
return copy_tag(e, mk_anonymous_constructor(r));
|
||
} else if (is_structure_instance(e)) {
|
||
auto info = get_structure_instance_info(e);
|
||
lean_assert(info.m_sources.empty());
|
||
for (expr & val : info.m_field_values)
|
||
val = visit(val);
|
||
return copy_tag(e, mk_structure_instance(info));
|
||
} else if (is_annotation(e)) {
|
||
return copy_tag(e, mk_annotation(get_annotation_kind(e), visit(get_annotation_arg(e))));
|
||
} else if (is_constant(e) && is_pattern_constant(const_name(e))) {
|
||
return e;
|
||
} else {
|
||
return m_parser.parser_error_or_expr({
|
||
"invalid pattern, must be an application, "
|
||
"constant, variable, type ascription or inaccessible term",
|
||
m_parser.pos_of(e)});
|
||
}
|
||
}
|
||
|
||
expr operator()(expr const & e, bool skip_main_fn) {
|
||
collect_new_locals(e, skip_main_fn);
|
||
expr r = visit(e);
|
||
return r;
|
||
}
|
||
};
|
||
|
||
static expr quote(expr const & e) {
|
||
switch (e.kind()) {
|
||
case expr_kind::Var:
|
||
return mk_app(mk_constant({"expr", "var"}), quote(var_idx(e)));
|
||
case expr_kind::Sort:
|
||
return mk_app(mk_constant({"expr", "sort"}), mk_expr_placeholder());
|
||
case expr_kind::Constant:
|
||
return mk_app(mk_constant({"expr", "const"}), quote(const_name(e)), mk_expr_placeholder());
|
||
case expr_kind::Meta:
|
||
return mk_expr_placeholder();
|
||
case expr_kind::Local:
|
||
throw elaborator_exception(e, sstream() << "invalid quotation, unexpected local constant '"
|
||
<< mlocal_pp_name(e) << "'");
|
||
case expr_kind::App:
|
||
return mk_app(mk_constant({"expr", "app"}), quote(app_fn(e)), quote(app_arg(e)));
|
||
case expr_kind::Lambda:
|
||
return mk_app(mk_constant({"expr", "lam"}), mk_expr_placeholder(), mk_expr_placeholder(),
|
||
quote(binding_domain(e)), quote(binding_body(e)));
|
||
case expr_kind::Pi:
|
||
return mk_app(mk_constant({"expr", "pi"}), mk_expr_placeholder(), mk_expr_placeholder(),
|
||
quote(binding_domain(e)), quote(binding_body(e)));
|
||
case expr_kind::Let:
|
||
return mk_app(mk_constant({"expr", "elet"}), mk_expr_placeholder(), quote(let_type(e)),
|
||
quote(let_value(e)), quote(let_body(e)));
|
||
case expr_kind::Macro:
|
||
if (is_antiquote(e))
|
||
return get_antiquote_expr(e);
|
||
if (is_typed_expr(e))
|
||
return mk_typed_expr(quote(get_typed_expr_expr(e)), quote(get_typed_expr_type(e)));
|
||
if (is_inaccessible(e))
|
||
return mk_expr_placeholder();
|
||
throw elaborator_exception(e, sstream() << "invalid quotation, unsupported macro '"
|
||
<< macro_def(e).get_name() << "'");
|
||
}
|
||
lean_unreachable();
|
||
}
|
||
|
||
/** \brief Elaborate quote in an empty local context. We need to elaborate expr patterns in the parser to find out
|
||
their pattern variables. */
|
||
static expr elaborate_quote(parser & p, expr e) {
|
||
lean_assert(is_expr_quote(e));
|
||
environment const & env = p.env();
|
||
options const & opts = p.get_options();
|
||
e = get_expr_quote_value(e);
|
||
|
||
name x("_x");
|
||
buffer<expr> locals;
|
||
buffer<expr> aqs;
|
||
e = replace(e, [&](expr const & t, unsigned) {
|
||
if (is_antiquote(t)) {
|
||
expr local = mk_local(p.next_name(), x.append_after(locals.size() + 1),
|
||
mk_expr_placeholder(), binder_info());
|
||
locals.push_back(local);
|
||
aqs.push_back(t);
|
||
return some_expr(local);
|
||
}
|
||
return none_expr();
|
||
});
|
||
e = copy_tag(e, Fun(locals, e));
|
||
|
||
metavar_context ctx;
|
||
local_context lctx;
|
||
elaborator elab(env, opts, "_elab_quote", ctx, lctx, /* recover_from_errors */ false,
|
||
/* in_pattern */ true, /* in_quote */ true);
|
||
e = elab.elaborate(e);
|
||
e = elab.finalize(e, /* check_unassigned */ false, /* to_simple_metavar */ true).first;
|
||
|
||
expr body = e;
|
||
for (unsigned i = 0; i < aqs.size(); i++)
|
||
body = binding_body(body);
|
||
|
||
e = instantiate_rev(body, aqs.size(), aqs.data());
|
||
e = quote(e);
|
||
return mk_typed_expr(mk_app(mk_constant(get_expr_name()), mk_bool_tt()), e);
|
||
}
|
||
|
||
expr parser::patexpr_to_pattern(expr const & pat_or_expr, bool skip_main_fn, buffer<expr> & new_locals) {
|
||
undef_id_to_local_scope scope(*this);
|
||
auto e = replace(pat_or_expr, [&](expr const & e) {
|
||
if (is_expr_quote(e)) {
|
||
return some_expr(elaborate_quote(*this, e));
|
||
} else {
|
||
return none_expr();
|
||
}
|
||
});
|
||
return to_pattern_fn(*this, new_locals)(e, skip_main_fn);
|
||
}
|
||
|
||
expr parser::parse_pattern_or_expr(unsigned rbp) {
|
||
flet<bool> set_in_pattern(m_in_pattern, true);
|
||
if (m_id_behavior != id_behavior::AssumeLocalIfNotLocal) {
|
||
all_id_local_scope scope(*this);
|
||
return parse_expr(rbp);
|
||
} else {
|
||
// keep AssumeLocalIfNotLocal in quotes
|
||
return parse_expr(rbp);
|
||
}
|
||
}
|
||
|
||
expr parser::parse_pattern(std::function<expr(parser &)> const & fn, buffer<expr> & new_locals) {
|
||
all_id_local_scope scope(*this);
|
||
flet<bool> set_in_pattern(m_in_pattern, true);
|
||
expr r = fn(*this);
|
||
return patexpr_to_pattern(r, false, new_locals);
|
||
}
|
||
|
||
/* Auxiliary functional object for patexpr_to_expr.
|
||
We cannot simply use replace anymore because of the field notation.
|
||
To fix the local declarations, we need to track the local bindings when we
|
||
go inside lambda/pi/let. */
|
||
class patexpr_to_expr_fn : public replace_visitor {
|
||
parser & m_p;
|
||
list<name> m_locals;
|
||
|
||
virtual expr visit_binding(expr const & e) override {
|
||
expr new_d = visit(binding_domain(e));
|
||
flet<list<name>> set(m_locals, cons(binding_name(e), m_locals));
|
||
expr new_b = visit(binding_body(e));
|
||
return update_binding(e, new_d, new_b);
|
||
}
|
||
|
||
virtual expr visit_let(expr const & e) override {
|
||
expr new_type = visit(let_type(e));
|
||
expr new_val = visit(let_value(e));
|
||
flet<list<name>> set(m_locals, cons(let_name(e), m_locals));
|
||
expr new_body = visit(let_body(e));
|
||
return update_let(e, new_type, new_val, new_body);
|
||
}
|
||
|
||
virtual expr visit_macro(expr const & e) override {
|
||
if (is_inaccessible(e) && is_placeholder(get_annotation_arg(e))) {
|
||
return get_annotation_arg(e);
|
||
} else {
|
||
return replace_visitor::visit_macro(e);
|
||
}
|
||
}
|
||
|
||
virtual expr visit_local(expr const & e) override {
|
||
return m_p.id_to_expr(mlocal_pp_name(e), m_p.pos_of(e), true, true, m_locals);
|
||
}
|
||
|
||
public:
|
||
patexpr_to_expr_fn(parser & p):m_p(p) {}
|
||
};
|
||
|
||
expr parser::patexpr_to_expr(expr const & pat_or_expr) {
|
||
error_if_undef_scope scope(*this);
|
||
return patexpr_to_expr_fn(*this)(pat_or_expr);
|
||
}
|
||
|
||
optional<expr> parser::resolve_local(name const & id, pos_info const & p, list<name> const & extra_locals,
|
||
bool allow_field_notation) {
|
||
/* Remark: (auxiliary) local constants many not be atomic.
|
||
Example: when elaborating
|
||
|
||
protected def list.sizeof {α : Type u} [has_sizeof α] : list α → nat
|
||
| list.nil := 1
|
||
| (list.cons a l) := 1 + sizeof a + list.sizeof l
|
||
|
||
the local context will contain the auxiliary local constant `list.size_of`
|
||
*/
|
||
|
||
// extra locals
|
||
unsigned vidx = 0;
|
||
for (name const & extra : extra_locals) {
|
||
if (id == extra)
|
||
return some_expr(save_pos(mk_var(vidx), p));
|
||
vidx++;
|
||
}
|
||
|
||
// locals
|
||
if (auto it1 = m_local_decls.find(id)) {
|
||
return some_expr(copy_with_new_pos(*it1, p));
|
||
}
|
||
|
||
if (allow_field_notation && !id.is_atomic() && id.is_string()) {
|
||
if (auto r = resolve_local(id.get_prefix(), p, extra_locals)) {
|
||
auto field_pos = p;
|
||
field_pos.second += id.get_prefix().utf8_size();
|
||
return some_expr(save_pos(mk_field_notation_compact(*r, id.get_string()), field_pos));
|
||
} else {
|
||
return none_expr();
|
||
}
|
||
} else {
|
||
return none_expr();
|
||
}
|
||
}
|
||
|
||
expr parser::id_to_expr(name const & id, pos_info const & p, bool resolve_only, bool allow_field_notation, list<name> const & extra_locals) {
|
||
buffer<level> lvl_buffer;
|
||
levels ls;
|
||
bool explicit_levels = false;
|
||
if (!resolve_only && curr_is_token(get_llevel_curly_tk())) {
|
||
next();
|
||
explicit_levels = true;
|
||
while (!curr_is_token(get_rcurly_tk())) {
|
||
auto _ = backtracking_scope();
|
||
try {
|
||
lvl_buffer.push_back(parse_level());
|
||
} catch (backtracking_exception) {}
|
||
if (!consumed_input()) break;
|
||
}
|
||
next();
|
||
ls = to_list(lvl_buffer.begin(), lvl_buffer.end());
|
||
}
|
||
|
||
auto check_no_levels = [&] (levels const & ls, pos_info const & p) {
|
||
if (ls) {
|
||
maybe_throw_error({
|
||
"invalid use of explicit universe parameter, identifier is a variable, "
|
||
"parameter or a constant bound to parameters in a section", p});
|
||
}
|
||
};
|
||
|
||
if (!explicit_levels && m_id_behavior == id_behavior::AllLocal) {
|
||
return save_pos(mk_local(id, save_pos(mk_expr_placeholder(), p)), p);
|
||
}
|
||
|
||
if (auto r = resolve_local(id, p, extra_locals, allow_field_notation)) {
|
||
check_no_levels(ls, p);
|
||
return *r;
|
||
}
|
||
|
||
if (!explicit_levels && m_id_behavior == id_behavior::AssumeLocalIfNotLocal) {
|
||
return save_pos(mk_local(id, save_pos(mk_expr_placeholder(), p)), p);
|
||
}
|
||
|
||
if (auto ref = get_local_ref(m_env, id)) {
|
||
check_no_levels(ls, p);
|
||
return copy_with_new_pos(*ref, p);
|
||
}
|
||
|
||
for (name const & ns : get_namespaces(m_env)) {
|
||
auto new_id = ns + id;
|
||
if (!ns.is_anonymous() && m_env.find(new_id) &&
|
||
(!id.is_atomic() || !is_protected(m_env, new_id))) {
|
||
return save_pos(mk_constant(new_id, ls), p);
|
||
}
|
||
}
|
||
|
||
if (!id.is_atomic()) {
|
||
name new_id = id;
|
||
new_id = remove_root_prefix(new_id);
|
||
if (m_env.find(new_id)) {
|
||
return save_pos(mk_constant(new_id, ls), p);
|
||
}
|
||
}
|
||
|
||
optional<expr> r;
|
||
// globals
|
||
|
||
if (m_env.find(id))
|
||
r = save_pos(mk_constant(id, ls), p);
|
||
// aliases
|
||
auto as = get_expr_aliases(m_env, id);
|
||
if (!is_nil(as)) {
|
||
buffer<expr> new_as;
|
||
if (r)
|
||
new_as.push_back(*r);
|
||
for (auto const & e : as) {
|
||
new_as.push_back(copy_with_new_pos(mk_constant(e, ls), p));
|
||
}
|
||
r = save_pos(mk_choice(new_as.size(), new_as.data()), p);
|
||
}
|
||
if (!r) {
|
||
if (m_id_behavior == id_behavior::AssumeLocalIfUndef) {
|
||
expr local = mk_local(id, save_pos(mk_expr_placeholder(), p));
|
||
r = save_pos(local, p);
|
||
}
|
||
}
|
||
if (!r && allow_field_notation && !id.is_atomic() && id.is_string()) {
|
||
try {
|
||
auto _ = no_error_recovery_scope();
|
||
expr s = id_to_expr(id.get_prefix(), p, resolve_only, allow_field_notation, extra_locals);
|
||
auto field_pos = p;
|
||
field_pos.second += id.get_prefix().utf8_size();
|
||
r = save_pos(mk_field_notation_compact(s, id.get_string()), field_pos);
|
||
} catch (exception &) {}
|
||
}
|
||
if (!r) {
|
||
return parser_error_or_expr({sstream() << "unknown identifier '" << id.escape() << "'", p});
|
||
}
|
||
return *r;
|
||
}
|
||
|
||
list<name> parser::to_constants(name const & id, char const * msg, pos_info const & p) const {
|
||
buffer<name> rs;
|
||
|
||
std::function<void(expr const & e)> extract_names = [&](expr const & e) {
|
||
if (in_section(m_env) && is_as_atomic(e)) {
|
||
extract_names(get_app_fn(get_as_atomic_arg(e)));
|
||
} else if (is_explicit(e)) {
|
||
extract_names(get_explicit_arg(e));
|
||
} else if (is_choice(e)) {
|
||
for (unsigned i = 0; i < get_num_choices(e); i++)
|
||
extract_names(get_choice(e, i));
|
||
} else if (is_constant(e)) {
|
||
rs.push_back(const_name(e));
|
||
} else {
|
||
throw parser_error(msg, p);
|
||
}
|
||
};
|
||
|
||
// locals
|
||
if (auto it1 = m_local_decls.find(id)) {
|
||
extract_names(*it1);
|
||
return to_list(rs);
|
||
}
|
||
|
||
for (name const & ns : get_namespaces(m_env)) {
|
||
auto new_id = ns + id;
|
||
if (!ns.is_anonymous() && m_env.find(new_id) &&
|
||
(!id.is_atomic() || !is_protected(m_env, new_id))) {
|
||
return to_list(new_id);
|
||
}
|
||
}
|
||
|
||
if (!id.is_atomic()) {
|
||
name new_id = id;
|
||
new_id = remove_root_prefix(new_id);
|
||
if (m_env.find(new_id))
|
||
return to_list(new_id);
|
||
}
|
||
|
||
buffer<expr> alts;
|
||
// globals
|
||
if (m_env.find(id))
|
||
rs.push_back(id);
|
||
// aliases
|
||
auto as = get_expr_aliases(m_env, id);
|
||
for (name const & n : as) {
|
||
rs.push_back(n);
|
||
}
|
||
|
||
if (rs.empty()) {
|
||
throw parser_error(sstream() << "unknown identifier '" << id.escape() << "'", p);
|
||
}
|
||
|
||
return to_list(rs);
|
||
}
|
||
|
||
name parser::to_constant(name const & id, char const * msg, pos_info const & p) {
|
||
return head(to_constants(id, msg, p));
|
||
}
|
||
|
||
name parser::check_constant_next(char const * msg) {
|
||
auto p = pos();
|
||
name id = check_id_next(msg);
|
||
try {
|
||
return to_constant(id, msg, p);
|
||
} catch (parser_error e) {
|
||
maybe_throw_error(std::move(e));
|
||
return id;
|
||
}
|
||
}
|
||
|
||
expr parser::parse_id(bool allow_field_notation) {
|
||
auto p = pos();
|
||
lean_assert(curr_is_identifier());
|
||
name id = check_id_next("", break_at_pos_exception::token_context::expr);
|
||
expr e = id_to_expr(id, p, /* resolve_only */ false, allow_field_notation);
|
||
if (is_constant(e) && get_global_info_manager()) {
|
||
get_global_info_manager()->add_const_info(m_env, p, const_name(e));
|
||
}
|
||
return e;
|
||
}
|
||
|
||
expr parser::parse_numeral_expr(bool user_notation) {
|
||
auto p = pos();
|
||
mpz n = get_num_val().get_numerator();
|
||
next();
|
||
list<expr> vals;
|
||
if (user_notation)
|
||
vals = get_mpz_notation(m_env, n);
|
||
if (!vals) {
|
||
return save_pos(mk_prenum(n), p);
|
||
} else {
|
||
buffer<expr> cs;
|
||
cs.push_back(save_pos(mk_prenum(n), p));
|
||
for (expr const & c : vals)
|
||
cs.push_back(copy_with_new_pos(c, p));
|
||
if (cs.size() == 1)
|
||
return cs[0];
|
||
else
|
||
return save_pos(mk_choice(cs.size(), cs.data()), p);
|
||
}
|
||
}
|
||
|
||
expr parser::parse_decimal_expr() {
|
||
auto p = pos();
|
||
mpq val = get_num_val();
|
||
next();
|
||
expr num = save_pos(mk_prenum(val.get_numerator()), p);
|
||
if (val.get_denominator() == 1) {
|
||
return num;
|
||
} else {
|
||
expr den = save_pos(mk_prenum(val.get_denominator()), p);
|
||
expr div = save_pos(mk_constant(get_has_div_div_name()), p);
|
||
return save_pos(lean::mk_app(div, num, den), p);
|
||
}
|
||
}
|
||
|
||
expr parser::parse_string_expr() {
|
||
std::string v = get_str_val();
|
||
next();
|
||
return from_string(v);
|
||
}
|
||
|
||
expr parser::parse_char_expr() {
|
||
auto p = pos();
|
||
std::string v = get_str_val();
|
||
buffer<unsigned> tmp;
|
||
utf8_decode(v, tmp);
|
||
lean_assert(tmp.size() == 1);
|
||
next();
|
||
return mk_app(save_pos(mk_constant(get_char_of_nat_name()), p),
|
||
save_pos(mk_prenum(mpz(tmp[0])), p),
|
||
p);
|
||
}
|
||
|
||
expr parser::parse_nud() {
|
||
switch (curr()) {
|
||
case token_kind::Keyword:
|
||
if (curr_is_token(get_placeholder_tk()))
|
||
return parse_placeholder();
|
||
else
|
||
return parse_nud_notation();
|
||
case token_kind::Identifier: {
|
||
auto id_pos = pos();
|
||
auto id = get_name_val();
|
||
expr e = parse_id();
|
||
// if `id` is immediately followed by `@`, parse an as pattern `id@pat`
|
||
if (in_pattern() && id.is_atomic() && curr_is_token(get_explicit_tk())) {
|
||
lean_assert(is_local(e));
|
||
// note: This number is not accurate for an escaped identifier. We should be able to do a better job
|
||
// in the new backtracking parser.
|
||
auto id_len = utf8_strlen(id.to_string().c_str());
|
||
auto p = pos();
|
||
if (p.first == id_pos.first && p.second == id_pos.second + id_len) {
|
||
next();
|
||
auto pat = parse_expr(get_max_prec());
|
||
return save_pos(mk_as_pattern(e, pat), id_pos);
|
||
}
|
||
}
|
||
return e;
|
||
}
|
||
case token_kind::Numeral: return parse_numeral_expr();
|
||
case token_kind::Decimal: return parse_decimal_expr();
|
||
case token_kind::String: return parse_string_expr();
|
||
case token_kind::Char: return parse_char_expr();
|
||
default: return parser_error_or_expr({"invalid expression, unexpected token", pos()});
|
||
}
|
||
}
|
||
|
||
// Return true if the current token can be the beginning of an expression
|
||
bool parser::curr_starts_expr() {
|
||
switch (curr()) {
|
||
case token_kind::Keyword:
|
||
return !is_nil(nud().find(get_token_info().value()));
|
||
case token_kind::Identifier:
|
||
case token_kind::Numeral:
|
||
case token_kind::Decimal:
|
||
case token_kind::String:
|
||
default:
|
||
return false;
|
||
}
|
||
}
|
||
|
||
expr parser::parse_led(expr left) {
|
||
if (is_sort_wo_universe(left) &&
|
||
(curr_is_numeral() || curr_is_identifier() || curr_is_token(get_lparen_tk()) || curr_is_token(get_placeholder_tk()))) {
|
||
left = get_annotation_arg(left);
|
||
level l = parse_level(get_max_prec());
|
||
lean_assert(sort_level(left) == mk_level_one() || sort_level(left) == mk_level_zero());
|
||
if (sort_level(left) == mk_level_one())
|
||
l = mk_succ(l);
|
||
return copy_tag(left, update_sort(left, l));
|
||
} else {
|
||
switch (curr()) {
|
||
case token_kind::Keyword:
|
||
return parse_led_notation(left);
|
||
case token_kind::FieldName: {
|
||
expr r = save_pos(mk_field_notation(left, get_name_val()), pos());
|
||
next();
|
||
return r;
|
||
}
|
||
case token_kind::FieldNum: {
|
||
expr r = save_pos(mk_field_notation(left, get_small_nat()), pos());
|
||
next();
|
||
return r;
|
||
}
|
||
default:
|
||
return mk_app(left, parse_expr(get_max_prec()), pos_of(left));
|
||
}
|
||
}
|
||
}
|
||
|
||
unsigned parser::curr_lbp() const {
|
||
switch (curr()) {
|
||
case token_kind::Keyword:
|
||
return get_token_info().expr_precedence();
|
||
case token_kind::CommandKeyword: case token_kind::Eof:
|
||
case token_kind::QuotedSymbol: case token_kind::DocBlock:
|
||
case token_kind::ModDocBlock:
|
||
return 0;
|
||
case token_kind::Identifier: case token_kind::Numeral:
|
||
case token_kind::Decimal: case token_kind::String:
|
||
case token_kind::Char:
|
||
return get_max_prec();
|
||
case token_kind::FieldNum: case token_kind::FieldName:
|
||
return get_max_prec()+1;
|
||
}
|
||
lean_unreachable(); // LCOV_EXCL_LINE
|
||
}
|
||
|
||
expr parser::parse_led_loop(expr left, unsigned rbp) {
|
||
while (rbp < curr_lbp()) {
|
||
auto _ = backtracking_scope();
|
||
try { left = parse_led(left); } catch (backtracking_exception) {}
|
||
|
||
// We did not consume any input, this can happen if we fail inside parse_notation.
|
||
if (!consumed_input()) break;
|
||
}
|
||
return left;
|
||
}
|
||
|
||
optional<expr> parser::maybe_parse_expr(unsigned rbp) {
|
||
auto _ = backtracking_scope();
|
||
try {
|
||
auto res = parse_expr(rbp);
|
||
if (consumed_input()) return some_expr(res);
|
||
} catch (backtracking_exception) {}
|
||
return none_expr();
|
||
}
|
||
|
||
expr parser::parse_expr(unsigned rbp) {
|
||
expr left = parse_nud();
|
||
return parse_led_loop(left, rbp);
|
||
}
|
||
|
||
pair<optional<name>, expr> parser::parse_id_tk_expr(name const & tk, unsigned rbp) {
|
||
if (curr_is_identifier()) {
|
||
auto id_pos = pos();
|
||
name id = get_name_val();
|
||
next();
|
||
if (curr_is_token(tk)) {
|
||
next();
|
||
return mk_pair(optional<name>(id), parse_expr(rbp));
|
||
} else {
|
||
expr left = id_to_expr(id, id_pos);
|
||
while (rbp < curr_lbp()) {
|
||
left = parse_led(left);
|
||
}
|
||
return mk_pair(optional<name>(), left);
|
||
}
|
||
} else {
|
||
return mk_pair(optional<name>(), parse_expr(rbp));
|
||
}
|
||
}
|
||
|
||
pair<optional<name>, expr> parser::parse_qualified_expr(unsigned rbp) {
|
||
return parse_id_tk_expr(get_colon_tk(), rbp);
|
||
}
|
||
|
||
pair<optional<name>, expr> parser::parse_optional_assignment(unsigned rbp) {
|
||
return parse_id_tk_expr(get_assign_tk(), rbp);
|
||
}
|
||
|
||
expr parser::parse_scoped_expr(unsigned num_ps, expr const * ps, local_environment const & lenv, unsigned rbp) {
|
||
local_scope scope(*this);
|
||
m_env = lenv;
|
||
for (unsigned i = 0; i < num_ps; i++)
|
||
add_local(ps[i]);
|
||
return parse_expr(rbp);
|
||
}
|
||
|
||
expr parser::parse_expr_with_env(local_environment const & lenv, unsigned rbp) {
|
||
flet<environment> set_env(m_env, lenv);
|
||
return parse_expr(rbp);
|
||
}
|
||
|
||
/** \brief Helper class for creating type context only if needed */
|
||
class lazy_type_context : public abstract_type_context {
|
||
environment m_env;
|
||
options m_opts;
|
||
std::unique_ptr<type_context_old> m_ctx;
|
||
type_context_old & ctx() {
|
||
if (!m_ctx)
|
||
m_ctx.reset(new type_context_old(m_env, m_opts));
|
||
return *m_ctx;
|
||
}
|
||
public:
|
||
lazy_type_context(environment const & env, options const & opts):m_env(env), m_opts(opts) {}
|
||
virtual ~lazy_type_context() {}
|
||
virtual environment const & env() const override { return const_cast<lazy_type_context*>(this)->ctx().env(); }
|
||
virtual expr whnf(expr const & e) override { return ctx().whnf(e); }
|
||
virtual bool is_def_eq(expr const & e1, expr const & e2) override { return ctx().is_def_eq(e1, e2); }
|
||
virtual expr infer(expr const & e) override { return ctx().infer(e); }
|
||
virtual expr check(expr const & e) override { return ctx().check(e); }
|
||
virtual optional<expr> is_stuck(expr const & e) override { return ctx().is_stuck(e); }
|
||
virtual name next_name() override { return ctx().next_name(); }
|
||
};
|
||
|
||
void parser::parse_command(cmd_meta const & meta) {
|
||
if (curr() != token_kind::CommandKeyword) {
|
||
auto p = pos();
|
||
maybe_throw_error({"expected command", p});
|
||
return;
|
||
}
|
||
reset_thread_local();
|
||
m_last_cmd_pos = pos();
|
||
name cmd_name = get_token_info().value();
|
||
m_cmd_token = get_token_info().token();
|
||
if (auto it = cmds().find(cmd_name)) {
|
||
lazy_type_context tc(m_env, get_options());
|
||
scope_global_ios scope1(m_ios);
|
||
scope_trace_env scope2(m_env, m_ios.get_options(), tc);
|
||
scope_traces_as_messages traces_as_messages(get_stream_name(), pos());
|
||
if (is_notation_cmd(cmd_name)) {
|
||
in_notation_ctx ctx(*this);
|
||
if (it->get_skip_token())
|
||
next();
|
||
m_env = it->get_fn()(*this, meta);
|
||
} else {
|
||
if (it->get_skip_token())
|
||
next();
|
||
m_env = it->get_fn()(*this, meta);
|
||
}
|
||
} else {
|
||
auto p = pos();
|
||
next();
|
||
maybe_throw_error({sstream() << "unknown command '" << cmd_name << "'", p});
|
||
}
|
||
}
|
||
|
||
std::string parser::parse_doc_block() {
|
||
auto val = m_scanner.get_str_val();
|
||
next();
|
||
return val;
|
||
}
|
||
|
||
void parser::parse_mod_doc_block() {
|
||
m_env = add_module_doc_string(m_env, m_scanner.get_str_val());
|
||
next();
|
||
}
|
||
|
||
#if defined(__GNUC__) && !defined(__CLANG__)
|
||
#pragma GCC diagnostic ignored "-Wmaybe-uninitialized"
|
||
#endif
|
||
|
||
void parser::parse_imports(unsigned & fingerprint, std::vector<module_name> & imports) {
|
||
init_scanner();
|
||
scanner::field_notation_scope scope(m_scanner, false);
|
||
m_last_cmd_pos = pos();
|
||
bool prelude = false;
|
||
if (curr_is_token(get_prelude_tk())) {
|
||
next();
|
||
prelude = true;
|
||
}
|
||
if (!prelude) {
|
||
module_name m("init");
|
||
imports.push_back(m);
|
||
}
|
||
while (curr_is_token(get_import_tk())) {
|
||
m_last_cmd_pos = pos();
|
||
next();
|
||
while (true) {
|
||
pos_info p = pos();
|
||
bool k_init = false;
|
||
unsigned k = 0;
|
||
try {
|
||
unsigned h = 0;
|
||
while (true) {
|
||
if (curr_is_token(get_period_tk()) || curr_is_token(get_dotdot_tk()) ||
|
||
curr_is_token(get_ellipsis_tk())) {
|
||
unsigned d = get_token_info().token().size();
|
||
if (!k_init) {
|
||
k = d - 1;
|
||
k_init = true;
|
||
h = d - 1;
|
||
} else {
|
||
k = d;
|
||
h = d;
|
||
}
|
||
next();
|
||
} else {
|
||
check_break_before();
|
||
break;
|
||
}
|
||
}
|
||
if (!curr_is_identifier())
|
||
break;
|
||
name f = get_name_val();
|
||
fingerprint = hash(fingerprint, f.hash());
|
||
if (k_init) {
|
||
fingerprint = hash(fingerprint, h);
|
||
}
|
||
if (k_init) {
|
||
module_name m(f, k);
|
||
imports.push_back(m);
|
||
} else {
|
||
module_name m(f);
|
||
imports.push_back(m);
|
||
}
|
||
next();
|
||
} catch (break_at_pos_exception & e) {
|
||
if (k_init)
|
||
e.m_token_info.m_token = std::string(k + 1, '.') + e.m_token_info.m_token.to_string();
|
||
e.m_token_info.m_context = break_at_pos_exception::token_context::import;
|
||
e.m_token_info.m_pos = p;
|
||
throw;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
void parser::process_imports() {
|
||
unsigned fingerprint = 0;
|
||
std::vector<module_name> imports;
|
||
|
||
std::exception_ptr exception_during_scanning;
|
||
auto begin_pos = pos();
|
||
try {
|
||
parse_imports(fingerprint, imports);
|
||
} catch (parser_exception) {
|
||
exception_during_scanning = std::current_exception();
|
||
}
|
||
|
||
scope_log_tree lt("importing", {begin_pos, pos()});
|
||
|
||
buffer<import_error> import_errors;
|
||
std::unordered_set<std::string> already_checked;
|
||
module_loader sorry_checking_import_fn =
|
||
[&] (std::string const & mod_id, module_name const & import) {
|
||
auto mod = m_import_fn(mod_id, import);
|
||
|
||
auto pos = m_last_cmd_pos;
|
||
auto mod_name = mod->m_module_name;
|
||
auto fn = m_file_name;
|
||
if (!already_checked.count(mod_name)) {
|
||
add_library_task(map<unit>(mod->m_uses_sorry, [pos, mod_name, fn](bool uses_sorry) {
|
||
if (uses_sorry)
|
||
report_message(message(fn, pos, WARNING,
|
||
(sstream() << "imported file '" << mod_name
|
||
<< "' uses sorry").str()));
|
||
return unit {};
|
||
}), "checking import for sorry", true, log_tree::CrossModuleLintLevel);
|
||
already_checked.insert(mod_name);
|
||
}
|
||
|
||
return mod;
|
||
};
|
||
m_env = import_modules(m_env, m_file_name, imports, sorry_checking_import_fn, import_errors);
|
||
|
||
if (!import_errors.empty()) {
|
||
for (auto & e : import_errors) {
|
||
try {
|
||
std::rethrow_exception(e.m_ex);
|
||
} catch (throwable & t) {
|
||
parser_exception error((sstream() << "invalid import: " << e.m_import.m_name << "\n" << t.what()).str(),
|
||
m_file_name.c_str(), m_last_cmd_pos);
|
||
if (!m_use_exceptions && m_show_errors)
|
||
report_message(error);
|
||
if (m_use_exceptions)
|
||
throw error;
|
||
}
|
||
}
|
||
}
|
||
|
||
m_env = update_fingerprint(m_env, fingerprint);
|
||
m_env = activate_export_decls(m_env, {}); // explicitly activate exports in root namespace
|
||
m_env = replay_export_decls_core(m_env, m_ios);
|
||
m_imports_parsed = true;
|
||
|
||
if (exception_during_scanning) std::rethrow_exception(exception_during_scanning);
|
||
}
|
||
|
||
void parser::get_imports(std::vector<module_name> & imports) {
|
||
scope_pos_info_provider scope1(*this);
|
||
unsigned fingerprint;
|
||
parse_imports(fingerprint, imports);
|
||
}
|
||
|
||
bool parser::parse_command_like() {
|
||
init_scanner();
|
||
m_error_since_last_cmd = false;
|
||
|
||
scope_pos_info_provider scope1(*this);
|
||
|
||
check_interrupted();
|
||
|
||
if (!m_imports_parsed) {
|
||
process_imports();
|
||
return false;
|
||
}
|
||
|
||
switch (curr()) {
|
||
case token_kind::CommandKeyword:
|
||
parse_command(cmd_meta());
|
||
updt_options();
|
||
break;
|
||
case token_kind::DocBlock:
|
||
parse_command(cmd_meta({}, {}, some(parse_doc_block())));
|
||
break;
|
||
case token_kind::ModDocBlock:
|
||
parse_mod_doc_block();
|
||
break;
|
||
case token_kind::Eof:
|
||
if (has_open_scopes(m_env)) {
|
||
maybe_throw_error({"invalid end of module, expecting 'end'", pos()});
|
||
}
|
||
return true;
|
||
break;
|
||
case token_kind::Keyword:
|
||
if (curr_is_token(get_period_tk())) {
|
||
next();
|
||
break;
|
||
}
|
||
/* fall-thru */
|
||
default:
|
||
throw parser_error("command expected", pos());
|
||
}
|
||
return false;
|
||
}
|
||
|
||
bool parser::curr_is_command_like() const {
|
||
switch (curr()) {
|
||
case token_kind::CommandKeyword:
|
||
case token_kind::Eof:
|
||
case token_kind::DocBlock:
|
||
case token_kind::ModDocBlock:
|
||
return true;
|
||
case token_kind::Keyword:
|
||
return curr_is_token(get_period_tk());
|
||
default:
|
||
return false;
|
||
}
|
||
}
|
||
|
||
std::shared_ptr<snapshot> parser::mk_snapshot() {
|
||
return std::make_shared<snapshot>(
|
||
m_env, m_ngen, m_local_level_decls, m_local_decls,
|
||
m_level_variables, m_variables, m_include_vars,
|
||
m_ios.get_options(), m_imports_parsed, m_ignore_noncomputable, m_parser_scope_stack, m_next_inst_idx, pos());
|
||
}
|
||
|
||
optional<pos_info> parser::get_pos_info(expr const & e) const {
|
||
tag t = e.get_tag();
|
||
if (t == nulltag)
|
||
return optional<pos_info>();
|
||
if (auto it = m_pos_table.find(t))
|
||
return optional<pos_info>(*it);
|
||
else
|
||
return optional<pos_info>();
|
||
}
|
||
|
||
pos_info parser::get_some_pos() const {
|
||
return m_last_cmd_pos;
|
||
}
|
||
|
||
char const * parser::get_file_name() const {
|
||
return get_stream_name().c_str();
|
||
}
|
||
|
||
message_builder parser::mk_message(pos_info const &p, message_severity severity) const {
|
||
std::shared_ptr<abstract_type_context> tc = std::make_shared<type_context_old>(env(), get_options());
|
||
return message_builder(tc, env(), ios(), get_file_name(), p, severity);
|
||
}
|
||
|
||
message_builder parser::mk_message(pos_info const & start_pos, pos_info const & end_pos, message_severity severity) const {
|
||
std::shared_ptr<abstract_type_context> tc = std::make_shared<type_context_old>(env(), get_options());
|
||
message_builder b(tc, env(), ios(), get_file_name(), start_pos, severity);
|
||
b.set_end_pos(end_pos);
|
||
return b;
|
||
}
|
||
|
||
message_builder parser::mk_message(message_severity severity) const {
|
||
return mk_message(pos(), severity);
|
||
}
|
||
|
||
void parser::init_scanner() {
|
||
if (!m_scanner_inited) {
|
||
m_curr = m_scanner.scan(m_env); // same code as scan(), but without break-at-pos checking
|
||
m_scanner_inited = true;
|
||
}
|
||
}
|
||
|
||
void parser::maybe_throw_error(parser_error && err) {
|
||
if (m_error_recovery) {
|
||
auto err_pos = err.get_pos() ? *err.get_pos() : pos();
|
||
if (err_pos > m_last_recovered_error_pos) {
|
||
check_system("parser error recovery");
|
||
mk_message(ERROR).set_exception(err).report();
|
||
m_last_recovered_error_pos = err_pos;
|
||
m_error_since_last_cmd = true;
|
||
}
|
||
} else {
|
||
throw err;
|
||
}
|
||
}
|
||
|
||
expr parser::parser_error_or_expr(parser_error && err) {
|
||
auto err_pos = err.get_pos() ? *err.get_pos() : pos();
|
||
maybe_throw_error(std::move(err));
|
||
return mk_sorry(err_pos, true);
|
||
}
|
||
|
||
level parser::parser_error_or_level(parser_error && err) {
|
||
maybe_throw_error(std::move(err));
|
||
return mk_level_placeholder();
|
||
}
|
||
|
||
bool parse_commands(environment & env, io_state & ios, char const * fname) {
|
||
// st_task_queue tq;
|
||
// scope_global_task_queue scope(&tq);
|
||
fs_module_vfs vfs;
|
||
vfs.m_modules_to_load_from_source.insert(std::string(fname));
|
||
log_tree lt;
|
||
module_mgr mod_mgr(&vfs, lt.get_root(), standard_search_path().get_path(), env, ios);
|
||
|
||
auto mod = mod_mgr.get_module(fname);
|
||
env = mod->get_produced_env();
|
||
return get(has_errors(lt.get_root()));
|
||
}
|
||
|
||
void initialize_parser() {
|
||
g_frontend_fresh = new name("_ffresh");
|
||
register_name_generator_prefix(*g_frontend_fresh);
|
||
g_parser_show_errors = new name{"parser", "show_errors"};
|
||
register_bool_option(*g_parser_show_errors, LEAN_DEFAULT_PARSER_SHOW_ERRORS,
|
||
"(lean parser) display error messages in the regular output channel");
|
||
}
|
||
|
||
void finalize_parser() {
|
||
delete g_frontend_fresh;
|
||
delete g_parser_show_errors;
|
||
}
|
||
}
|