lean2/src/kernel/environment.cpp

/*
Copyright (c) 2013-2014 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.

Author: Leonardo de Moura
*/
#include <limits>
#include "kernel/environment.h"
#include "kernel/kernel_exception.h"

namespace lean {
/**
   \brief "Do nothing" normalizer extension.
*/
class noop_normalizer_extension : public normalizer_extension {
public:
    virtual optional<std::pair<expr, constraints>> operator()(expr const &, environment const &, type_checker &) const {
        return optional<std::pair<expr, constraints>>();
    }
};

environment_header::environment_header(unsigned trust_lvl, bool proof_irrel, bool eta, std::unique_ptr<normalizer_extension const> ext):
    m_trust_lvl(trust_lvl), m_proof_irrel(proof_irrel), m_eta(eta), m_norm_ext(std::move(ext)) {}

environment_extension::~environment_extension() {}

environment::environment(header const & h, definitions const & d, extensions const & exts):
    m_header(h), m_definitions(d), m_extensions(exts) {}

environment::environment(unsigned trust_lvl, bool proof_irrel, bool eta):
    environment(trust_lvl, proof_irrel, eta, std::unique_ptr<normalizer_extension>(new noop_normalizer_extension()))
{}

environment::environment(unsigned trust_lvl, bool proof_irrel, bool eta, std::unique_ptr<normalizer_extension> ext):
    m_header(std::make_shared<environment_header>(trust_lvl, proof_irrel, eta, std::move(ext))),
    m_extensions(std::make_shared<environment_extensions const>())
{}

optional<definition> environment::find(name const & n) const {
    definition const * r = m_definitions.find(n);
    return r ? some_definition(*r) : none_definition();
}

definition environment::get(name const & n) const {
   definition const * r = m_definitions.find(n);
   if (!r)
       throw_unknown_declaration(*this, n);
   return *r;
}

[[ noreturn ]] void throw_incompatible_environment(environment const & env) {
    throw_kernel_exception(env, "invalid declaration, it was checked/certified in an incompatible environment");
}

environment environment::add(certified_definition const & d) const {
    if (d.get_header().get() != m_header.get())
        throw_incompatible_environment(*this);
    name const & n = d.get_definition().get_name();
    if (find(n))
        throw_already_declared(*this, n);
    return environment(m_header, insert(m_definitions, n, d.get_definition()), m_extensions);
}

environment environment::replace(certified_definition const & t) const {
    if (t.get_header().get() != m_header.get())
        throw_incompatible_environment(*this);
    name const & n = t.get_definition().get_name();
    auto ax = find(n);
    if (!ax)
        throw_kernel_exception(*this, "invalid replacement of axiom with theorem, the environment does not have an axiom with the given name");
    if (!ax->is_axiom())
        throw_kernel_exception(*this, "invalid replacement of axiom with theorem, the current declaration in the environment is not an axiom");
    if (!t.get_definition().is_theorem())
        throw_kernel_exception(*this, "invalid replacement of axiom with theorem, the new declaration is not a theorem");
    if (ax->get_type() != t.get_definition().get_type())
        throw_kernel_exception(*this, "invalid replacement of axiom with theorem, the 'replace' operation can only be used when the axiom and theorem have the same type");
    return environment(m_header, insert(m_definitions, n, t.get_definition()), m_extensions);
}

class extension_manager {
    std::vector<std::shared_ptr<environment_extension const>> m_exts;
    mutex                                                     m_mutex;
public:
    unsigned register_extension(std::shared_ptr<environment_extension const> const & ext) {
        lock_guard<mutex> lock(m_mutex);
        unsigned r = m_exts.size();
        m_exts.push_back(ext);
        return r;
    }

    bool has_ext(unsigned extid) const { return extid < m_exts.size(); }

    environment_extension const & get_initial(unsigned extid) {
        lock_guard<mutex> lock(m_mutex);
        return *(m_exts[extid].get());
    }
};

static std::unique_ptr<extension_manager> g_extension_manager;

static extension_manager & get_extension_manager() {
    if (!g_extension_manager)
        g_extension_manager.reset(new extension_manager());
    return *g_extension_manager;
}

unsigned environment::register_extension(std::shared_ptr<environment_extension const> const & initial) {
    return get_extension_manager().register_extension(initial);
}

[[ noreturn ]] void throw_invalid_extension(environment const & env) {
    throw_kernel_exception(env, "invalid environment extension identifier");
}

environment_extension const & environment::get_extension(unsigned id) const {
    if (id >= get_extension_manager().has_ext(id))
        throw_invalid_extension(*this);
    if (id < m_extensions->size() || !(*m_extensions)[id])
        return get_extension_manager().get_initial(id);
    return *((*m_extensions)[id].get());
}

environment environment::update(unsigned id, std::shared_ptr<environment_extension const> const & ext) const {
    if (id >= get_extension_manager().has_ext(id))
        throw_invalid_extension(*this);
    auto new_exts = std::make_shared<environment_extensions>(*m_extensions);
    if (id >= new_exts->size())
        new_exts->resize(id+1);
    (*new_exts)[id] = ext;
    return environment(m_header, m_definitions, new_exts);
}
}