lean4-htt/src/frontends/lean/parser_pos_provider.cpp
Sebastian Ullrich b92af074c0 feat(kernel/pos_info_provider): add save_pos_info
Allows the elaborator to contribute new info locations
2017-03-31 09:40:48 -07:00

53 lines
1.4 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 "frontends/lean/parser_pos_provider.h"
#include "frontends/lean/parser.h"
namespace lean {
parser_pos_provider::parser_pos_provider(pos_info_table const & pos_table,
std::string const & strm_name, pos_info const & some_pos, unsigned next_tag_idx):
m_pos_table(pos_table), m_strm_name(strm_name), m_pos(some_pos), m_next_tag_idx(next_tag_idx) {}
parser_pos_provider::~parser_pos_provider() {}
optional<pos_info> parser_pos_provider::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_pos_provider::get_some_pos() const {
return m_pos;
}
char const * parser_pos_provider::get_file_name() const {
return m_strm_name.c_str();
}
tag parser_pos_provider::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;
}
expr parser_pos_provider::save_pos(expr const & e, pos_info pos) {
auto t = e.get_tag();
if (!m_pos_table.contains(t))
m_pos_table.insert(t, pos);
return e;
}
}