michael/lean2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(library/export): export the whole environment when using "--expor-all"

Browse source
This commit is contained in:
Leonardo de Moura 2015-07-30 14:59:57 -07:00
parent bed751a2d7
commit 9bf64c10fd
1 changed files with 39 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/library/export.cpp
View file

@ -187,8 +187,6 @@ class exporter {
for_each(e, [&](expr const & e, unsigned) { for_each(e, [&](expr const & e, unsigned) {
if (is_constant(e)) { if (is_constant(e)) {
name const & n = const_name(e); name const & n = const_name(e);
if (!inductive::is_intro_rule(m_env, n) &&
!inductive::is_elim_rule(m_env, n))
export_declaration(n); export_declaration(n);
} }
return true; return true;
@ -196,6 +194,8 @@ class exporter {
} }
void export_definition(declaration const & d) { void export_definition(declaration const & d) {
if (inductive::is_intro_rule(m_env, d.get_name()) || inductive::is_elim_rule(m_env, d.get_name()))
return;
if (already_exported(d.get_name())) if (already_exported(d.get_name()))
return; return;
mark(d.get_name()); mark(d.get_name());
@ -284,6 +284,11 @@ class exporter {
} }
void export_declarations() { void export_declarations() {
if (m_all) {
m_env.for_each_declaration([&](declaration const & d) {
export_declaration(d.get_name());
});
} else {
buffer<name> ns; buffer<name> ns;
to_buffer(get_curr_module_decl_names(m_env), ns); to_buffer(get_curr_module_decl_names(m_env), ns);
std::reverse(ns.begin(), ns.end()); std::reverse(ns.begin(), ns.end());
@ -291,8 +296,10 @@ class exporter {
export_declaration(n); export_declaration(n);
} }
} }
}
void export_direct_imports() { void export_direct_imports() {
if (!m_all) {
buffer<module_name> imports; buffer<module_name> imports;
to_buffer(get_curr_module_imports(m_env), imports); to_buffer(get_curr_module_imports(m_env), imports);
std::reverse(imports.begin(), imports.end()); std::reverse(imports.begin(), imports.end());
@ -305,8 +312,15 @@ class exporter {
} }
} }
} }
}
void export_global_universes() { void export_global_universes() {
if (m_all) {
m_env.for_each_universe([&](name const & u) {
unsigned n = export_name(u);
m_out << "#UNI " << n << "\n";
});
} else {
buffer<name> ns; buffer<name> ns;
to_buffer(get_curr_module_univ_names(m_env), ns); to_buffer(get_curr_module_univ_names(m_env), ns);
std::reverse(ns.begin(), ns.end()); std::reverse(ns.begin(), ns.end());
@ -315,6 +329,7 @@ class exporter {
m_out << "#UNI " << n << "\n"; m_out << "#UNI " << n << "\n";
} }
} }
}
public: public:
exporter(std::ostream & out, environment const & env, bool all):m_out(out), m_env(env), m_all(all) {} exporter(std::ostream & out, environment const & env, bool all):m_out(out), m_env(env), m_all(all) {}
@ -322,7 +337,6 @@ public:
void operator()() { void operator()() {
m_name2idx.insert(mk_pair(name(), 0)); m_name2idx.insert(mk_pair(name(), 0));
m_level2idx.insert(mk_pair(level(), 0)); m_level2idx.insert(mk_pair(level(), 0));
if (!m_all)
export_direct_imports(); export_direct_imports();
export_global_universes(); export_global_universes();
export_declarations(); export_declarations();