lean2/src/kernel/normalizer_extension.cpp

52 lines
2 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 "kernel/normalizer_extension.h"
namespace lean {
class id_normalizer_extension : public normalizer_extension {
public:
virtual optional<pair<expr, constraint_seq>> operator()(expr const &, extension_context &) const {
return optional<pair<expr, constraint_seq>>();
}
virtual optional<expr> may_reduce_later(expr const &, extension_context &) const { return none_expr(); }
virtual bool supports(name const &) const { return false; }
};
std::unique_ptr<normalizer_extension> mk_id_normalizer_extension() {
return std::unique_ptr<normalizer_extension>(new id_normalizer_extension());
}
class comp_normalizer_extension : public normalizer_extension {
std::unique_ptr<normalizer_extension> m_ext1;
std::unique_ptr<normalizer_extension> m_ext2;
public:
comp_normalizer_extension(std::unique_ptr<normalizer_extension> && ext1, std::unique_ptr<normalizer_extension> && ext2):
m_ext1(std::move(ext1)), m_ext2(std::move(ext2)) {}
virtual optional<pair<expr, constraint_seq>> operator()(expr const & e, extension_context & ctx) const {
if (auto r = (*m_ext1)(e, ctx))
return r;
else
return (*m_ext2)(e, ctx);
}
virtual optional<expr> may_reduce_later(expr const & e, extension_context & ctx) const {
if (auto r = m_ext1->may_reduce_later(e, ctx))
return r;
else
return m_ext2->may_reduce_later(e, ctx);
}
virtual bool supports(name const & feature) const {
return m_ext1->supports(feature) || m_ext2->supports(feature);
}
};
std::unique_ptr<normalizer_extension> compose(std::unique_ptr<normalizer_extension> && ext1, std::unique_ptr<normalizer_extension> && ext2) {
return std::unique_ptr<normalizer_extension>(new comp_normalizer_extension(std::move(ext1), std::move(ext2)));
}
}