diff --git a/src/util/extensible_object.h b/src/util/extensible_object.h new file mode 100644 index 000000000..9ceeb5762 --- /dev/null +++ b/src/util/extensible_object.h @@ -0,0 +1,83 @@ +/* +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 +#include +#include "util/thread.h" +#include "util/debug.h" +namespace lean { +/** + \brief Template for creating objects that can be attached to "extensions". +*/ +template +class extensible_object : public T { +public: + class extension { + friend class extensible_object; + extensible_object * m_owner; + public: + extension():m_owner(nullptr) {} + virtual ~extension() {} + extensible_object & get_owner() { return *m_owner; } + }; +private: + std::vector> m_extensions; +public: + template + extensible_object(Args &&... args):T(std::forward(args)...) {} + + typedef std::unique_ptr (*mk_extension)(); + + class extension_factory { + std::vector m_makers; + mutex m_makers_mutex; + public: + unsigned register_extension(mk_extension mk) { + lock_guard lock(m_makers_mutex); + unsigned r = m_makers.size(); + m_makers.push_back(mk); + return r; + } + + std::unique_ptr mk(unsigned extid) { + lock_guard lock(m_makers_mutex); + return m_makers[extid](); + } + }; + + static std::unique_ptr g_extension_factory; + + static extension_factory & get_extension_factory() { + if (!g_extension_factory) + g_extension_factory.reset(new extension_factory()); + return *g_extension_factory; + } + + static unsigned register_extension(mk_extension mk) { + return get_extension_factory().register_extension(mk); + } + + extension & get_extension_core(unsigned extid) { + if (extid >= m_extensions.size()) + m_extensions.resize(extid+1); + if (!m_extensions[extid]) { + std::unique_ptr ext = get_extension_factory().mk(extid); + ext->m_owner = this; + m_extensions[extid].swap(ext); + } + return *(m_extensions[extid].get()); + } + + template Ext & get_extension(unsigned extid) { + extension & ext = get_extension_core(extid); + lean_assert(dynamic_cast(&ext) != nullptr); + return static_cast(ext); + } +}; +template +std::unique_ptr::extension_factory> extensible_object::g_extension_factory; +}