/*
Copyright (c) 2015 Daniel Selsam. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Author: Daniel Selsam
*/
#include "library/abstract_expr_manager.h"
#include "util/safe_arith.h"
#include "util/list_fn.h"

namespace lean {

unsigned abstract_expr_manager::hash(expr const & e) {
    switch (e.kind()) {
    case expr_kind::Constant:
    case expr_kind::Local:
    case expr_kind::Meta:
    case expr_kind::Sort:
    case expr_kind::Var:
    case expr_kind::Macro:
        return e.hash();
    case expr_kind::Lambda:
    case expr_kind::Pi:
        return ::lean::hash(hash(binding_domain(e)), hash(binding_body(e)));
    case expr_kind::App:
        buffer<expr> args;
        expr f = get_app_args(e, args);
        fun_info f_info = m_fun_info_manager.get(f, args.size());
        unsigned h = hash(f);
        int i = -1;
        for_each(f_info.get_params_info(), [&](param_info const & p_info) {
                i++;
                if (p_info.is_subsingleton()) return;
                h = ::lean::hash(h, hash(args[i]));
            });
        return h;
    }
    lean_unreachable();
}

bool abstract_expr_manager::is_equal(expr const & a, expr const & b) {
        if (is_eqp(a, b))          return true;
        if (hash(a) != hash(b))    return false;
        if (a.kind() != b.kind())  return false;
        if (is_var(a))             return var_idx(a) == var_idx(b);
        switch (a.kind()) {
        case expr_kind::Var:
            lean_unreachable(); // LCOV_EXCL_LINE
        case expr_kind::Constant: case expr_kind::Sort:
            return a == b;
        case expr_kind::Meta: case expr_kind::Local:
            return mlocal_name(a) == mlocal_name(b) && is_equal(mlocal_type(a), mlocal_type(b));
        case expr_kind::Lambda: case expr_kind::Pi:
            return is_equal(binding_domain(a), binding_domain(b)) && is_equal(binding_body(a), binding_body(b));
        case expr_kind::Macro:
            if (macro_def(a) != macro_def(b) || macro_num_args(a) != macro_num_args(b))
                return false;
            for (unsigned i = 0; i < macro_num_args(a); i++) {
                if (!is_equal(macro_arg(a, i), macro_arg(b, i)))
                    return false;
            }
            return true;
        case expr_kind::App:
            buffer<expr> a_args, b_args;
            expr f_a = get_app_args(a, a_args);
            expr f_b = get_app_args(b, b_args);
            if (!is_equal(f_a, f_b)) return false;
            if (a_args.size() != b_args.size()) return false;
            fun_info f_info = m_fun_info_manager.get(f_a, a_args.size());
            int i = -1;
            bool not_equal = false;
            for_each(f_info.get_params_info(), [&](param_info const & p_info) {
                    i++;
                    if (p_info.is_subsingleton()) return;
                    if (!is_equal(a_args[i], b_args[i])) not_equal = true;
                });
            return !not_equal;
        }
        lean_unreachable(); // LCOV_EXCL_LINE
}

}