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

Author: Leonardo de Moura
*/
#pragma once
#include <iostream>
#include <memory>
#include "expr.h"
#include "level.h"

namespace lean {
/**
   \brief Lean environment for defining constants, inductive
   datatypes, universe variables, et.c
*/
class environment {
    struct imp;
    std::shared_ptr<imp> m_imp;
    explicit environment(std::shared_ptr<imp> const & ptr);
    explicit environment(imp * new_ptr);
public:
    environment();
    ~environment();

    /**
       \brief Define a new universe variable with name \c n and constraint <tt>n >= l</tt>.
       Return the new variable.

       \remark An exception is thrown if a universe inconsistency is detected.
    */
    level define_uvar(name const & n, level const & l);
    level define_uvar(name const & n) { return define_uvar(n, level()); }
    level define_uvar(char const * n, level const & l) { return define_uvar(name(n), l); }
    level define_uvar(char const * n) { return define_uvar(name(n), level()); }

    /**
       \brief Return true iff the constraint l1 >= l2 is implied by the constraints
       in the environment.
    */
    bool is_ge(level const & l1, level const & l2) const;

    /** \brief Display universal variables, and their constraints */
    void display_uvars(std::ostream & out) const;

    /**
       \brief Return universal variable with the given name.
       Throw an exception if variable is not defined in this environment.
    */
    level get_uvar(name const & n) const;
    level get_uvar(char const * n) const { return get_uvar(name(n)); }

    /**
       \brief Create a child environment. This environment will only allow "read-only" operations until
       all children environments are deleted.
    */
    environment mk_child() const;

    /** \brief Return true iff this environment has children environments. */
    bool has_children() const;

    /** \brief Return true iff this environment has a parent environment. */
    bool has_parent() const;

    /**
        \brief Return parent environment of this environment.
        \pre has_parent()
    */
    environment parent() const;

    enum class object_kind { Definition, Fact };

    /**
        \brief Base class for environment objects
        It is just a place holder at this point.
    */
    class object {
    public:
        object() {}
        object(object const & o) = delete;
        object & operator=(object const & o) = delete;

        virtual ~object() {}
        virtual object_kind kind() const = 0;
        virtual void display(std::ostream & out) const = 0;
        virtual expr const & get_type() const = 0;
    };

    class definition : public object {
        name m_name;
        expr m_type;
        expr m_value;
        bool m_opaque;
    public:
        definition(name const & n, expr const & t, expr const & v, bool opaque);
        virtual ~definition();
        virtual object_kind kind() const;
        name const & get_name()  const { return m_name; }
        virtual expr const & get_type()  const { return m_type; }
        expr const & get_value() const { return m_value; }
        bool         is_opaque() const { return m_opaque; }
        virtual void display(std::ostream & out) const;
    };

    class fact : public object {
        name m_name;
        expr m_type;
    public:
        fact(name const & n, expr const & t);
        virtual ~fact();
        virtual object_kind kind() const;
        name const & get_name()  const { return m_name; }
        virtual expr const & get_type()  const { return m_type; }
        virtual void display(std::ostream & out) const;
    };

    friend bool is_definition(object const & o) { return o.kind() == object_kind::Definition; }
    friend bool is_fact(object const & o) { return o.kind() == object_kind::Fact; }

    friend definition const & to_definition(object const & o) { lean_assert(is_definition(o)); return static_cast<definition const &>(o); }
    friend fact const & to_fact(object const & o) { lean_assert(is_fact(o)); return static_cast<fact const &>(o); }

    /**
       \brief Add a new definition n : t := v.
       It throws an exception if v does not have type t.
       It throws an exception if there is already an object with the given name.
       If opaque == true, then definition is not used by normalizer.
    */
    void add_definition(name const & n, expr const & t, expr const & v, bool opaque = false);
    void add_definition(char const * n, expr const & t, expr const & v, bool opaque = false) { add_definition(name(n), t, v, opaque); }

    /**
       \brief Add a new definition n : infer_type(v) := v.
       It throws an exception if there is already an object with the given name.
       If opaque == true, then definition is not used by normalizer.
    */
    void add_definition(name const & n, expr const & v, bool opaque = false);
    void add_definition(char const * n, expr const & v, bool opaque = false) { add_definition(name(n), v, opaque); }

    /**
       \brief Add a new fact to the environment.
       It throws an exception if there is already an object with the given name.
    */
    void add_fact(name const & n, expr const & t);
    void add_fact(char const * n, expr const & t) { add_fact(name(n), t); }

    /**
       \brief Return the object with the given name.
       It throws an exception if the environment does not have an object with the given name.
    */
    object const & get_object(name const & n) const;

    /**
       \brief Return the object with the given name.
       Return nullptr if there is no object with the given name.
    */
    object const * get_object_ptr(name const & n) const;
};
}