diff options
Diffstat (limited to 'AK/Function.h')
-rw-r--r-- | AK/Function.h | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/AK/Function.h b/AK/Function.h new file mode 100644 index 0000000000..0fef35abe8 --- /dev/null +++ b/AK/Function.h @@ -0,0 +1,111 @@ +/* + * Copyright (C) 2016 Apple Inc. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, + * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS + * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF + * THE POSSIBILITY OF SUCH DAMAGE. + */ + +#pragma once + +#include "OwnPtr.h" +#include "StdLib.h" + +namespace AK { + +template<typename> class Function; + +template <typename Out, typename... In> +class Function<Out(In...)> { +public: + Function() = default; + Function(std::nullptr_t) { } + + template<typename CallableType, class = typename EnableIf<!(IsPointer<CallableType>::value && IsFunction<typename RemovePointer<CallableType>::Type>::value) && IsRvalueReference<CallableType&&>::value>::Type> + Function(CallableType&& callable) + : m_callableWrapper(make<CallableWrapper<CallableType>>(move(callable))) + { + } + + template<typename FunctionType, class = typename EnableIf<IsPointer<FunctionType>::value && IsFunction<typename RemovePointer<FunctionType>::Type>::value>::Type> + Function(FunctionType f) + : m_callableWrapper(make<CallableWrapper<FunctionType>>(move(f))) + { + } + + Out operator()(In... in) + { + ASSERT(m_callableWrapper); + return m_callableWrapper->call(forward<In>(in)...); + } + + explicit operator bool() const { return !!m_callableWrapper; } + + template<typename CallableType, class = typename EnableIf<!(IsPointer<CallableType>::value && IsFunction<typename RemovePointer<CallableType>::Type>::value) && IsRvalueReference<CallableType&&>::value>::Type> + Function& operator=(CallableType&& callable) + { + m_callableWrapper = make<CallableWrapper<CallableType>>(move(callable)); + return *this; + } + + template<typename FunctionType, class = typename EnableIf<IsPointer<FunctionType>::value && IsFunction<typename RemovePointer<FunctionType>::Type>::value>::Type> + Function& operator=(FunctionType f) + { + m_callableWrapper = make<CallableWrapper<FunctionType>>(move(f)); + return *this; + } + + Function& operator=(std::nullptr_t) + { + m_callableWrapper = nullptr; + return *this; + } + +private: + class CallableWrapperBase { + public: + virtual ~CallableWrapperBase() { } + virtual Out call(In...) = 0; + }; + + template<typename CallableType> + class CallableWrapper : public CallableWrapperBase { + public: + explicit CallableWrapper(CallableType&& callable) + : m_callable(move(callable)) + { + } + + CallableWrapper(const CallableWrapper&) = delete; + CallableWrapper& operator=(const CallableWrapper&) = delete; + + Out call(In... in) final { return m_callable(forward<In>(in)...); } + + private: + CallableType m_callable; + }; + + OwnPtr<CallableWrapperBase> m_callableWrapper; +}; + +} + +using AK::Function; + |