93b17e2ec1
Now, any exception that requires pretty printing support should be a subclass of ext_exception
48 lines
1.4 KiB
C++
48 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 "library/head_map.h"
|
|
#include "library/explicit.h"
|
|
|
|
namespace lean {
|
|
head_index::head_index(expr const & e) {
|
|
expr f = get_app_fn(e);
|
|
while (true) {
|
|
if (is_as_atomic(f))
|
|
f = get_app_fn(get_as_atomic_arg(f));
|
|
else if (is_explicit(f))
|
|
f = get_explicit_arg(f);
|
|
else if (is_consume_args(f))
|
|
f = get_consume_args_arg(f);
|
|
else
|
|
break;
|
|
}
|
|
m_kind = f.kind();
|
|
if (is_constant(f))
|
|
m_name = const_name(f);
|
|
else if (is_local(f))
|
|
m_name = mlocal_name(f);
|
|
}
|
|
|
|
int head_index::cmp::operator()(head_index const & i1, head_index const & i2) const {
|
|
if (i1.m_kind != i2.m_kind || (i1.m_kind != expr_kind::Constant && i1.m_kind != expr_kind::Local))
|
|
return static_cast<int>(i1.m_kind) - static_cast<int>(i2.m_kind);
|
|
else
|
|
return quick_cmp(i1.m_name, i2.m_name);
|
|
}
|
|
|
|
std::ostream & operator<<(std::ostream & out, head_index const & head_idx) {
|
|
if (head_idx.m_kind == expr_kind::Constant || head_idx.m_kind == expr_kind::Local)
|
|
out << head_idx.m_name;
|
|
else
|
|
out << head_idx.m_kind;
|
|
return out;
|
|
}
|
|
|
|
io_state_stream const & operator<<(io_state_stream const & out, head_index const & head_idx) {
|
|
return display(out, head_idx);
|
|
}
|
|
}
|