48 lines
1.8 KiB
C++
48 lines
1.8 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<expr> operator()(expr const &, extension_context &) const { return none_expr(); }
|
||
|
virtual bool may_reduce_later(expr const &, extension_context &) const { return false; }
|
||
|
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<expr> operator()(expr const & e, extension_context & ctx) const {
|
||
|
if (auto r = (*m_ext1)(e, ctx))
|
||
|
return r;
|
||
|
else
|
||
|
return (*m_ext2)(e, ctx);
|
||
|
}
|
||
|
|
||
|
virtual bool may_reduce_later(expr const & e, extension_context & ctx) const {
|
||
|
return m_ext1->may_reduce_later(e, ctx) || 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)));
|
||
|
}
|
||
|
}
|