#include <folly/CppAttributes.h>
#include <folly/Portability.h>
#include <folly/Traits.h>
+#include <folly/functional/Invoke.h>
namespace folly {
std::aligned_storage<6 * sizeof(void*)>::type tiny;
};
-template <typename Fun, typename FunT = typename std::decay<Fun>::type>
+template <typename Fun, typename = Fun*>
using IsSmall = Conjunction<
- std::integral_constant<bool, (sizeof(FunT) <= sizeof(Data::tiny))>,
- std::is_nothrow_move_constructible<FunT>>;
+ std::integral_constant<bool, (sizeof(Fun) <= sizeof(Data::tiny))>,
+ std::is_nothrow_move_constructible<Fun>>;
using SmallTag = std::true_type;
using HeapTag = std::false_type;
-template <class T>
+template <typename T>
struct NotFunction : std::true_type {};
-template <class T>
+template <typename T>
struct NotFunction<Function<T>> : std::false_type {};
-template <typename Fun, typename FunT = typename std::decay<Fun>::type>
-using DecayIfConstructible = typename std::enable_if<
- Conjunction<NotFunction<FunT>, std::is_constructible<FunT, Fun>>::value,
- FunT>::type;
+template <typename T>
+using EnableIfNotFunction =
+ typename std::enable_if<NotFunction<T>::value>::type;
struct CoerceTag {};
}
ReturnType operator()(Args... args) {
- auto& fn = *static_cast<Function<ReturnType(Args...)>*>(this);
+ auto& fn = *static_cast<Function<NonConstSignature>*>(this);
return fn.call_(fn.data_, static_cast<Args&&>(args)...);
}
class SharedProxy {
- std::shared_ptr<Function<ReturnType(Args...)>> sp_;
+ std::shared_ptr<Function<NonConstSignature>> sp_;
public:
- explicit SharedProxy(Function<ReturnType(Args...)>&& func)
- : sp_(std::make_shared<Function<ReturnType(Args...)>>(
- std::move(func))) {}
+ explicit SharedProxy(Function<NonConstSignature>&& func)
+ : sp_(std::make_shared<Function<NonConstSignature>>(std::move(func))) {}
ReturnType operator()(Args&&... args) const {
return (*sp_)(static_cast<Args&&>(args)...);
}
}
ReturnType operator()(Args... args) const {
- auto& fn = *static_cast<const Function<ReturnType(Args...) const>*>(this);
+ auto& fn = *static_cast<const Function<ConstSignature>*>(this);
return fn.call_(fn.data_, static_cast<Args&&>(args)...);
}
class SharedProxy {
- std::shared_ptr<Function<ReturnType(Args...) const>> sp_;
+ std::shared_ptr<Function<ConstSignature>> sp_;
public:
- explicit SharedProxy(Function<ReturnType(Args...) const>&& func)
- : sp_(std::make_shared<Function<ReturnType(Args...) const>>(
- std::move(func))) {}
+ explicit SharedProxy(Function<ConstSignature>&& func)
+ : sp_(std::make_shared<Function<ConstSignature>>(std::move(func))) {}
ReturnType operator()(Args&&... args) const {
return (*sp_)(static_cast<Args&&>(args)...);
}
return true;
}
-// Invoke helper
-template <typename F, typename... Args>
-inline constexpr auto invoke(F&& f, Args&&... args)
- -> decltype(std::forward<F>(f)(std::forward<Args>(args)...)) {
- return std::forward<F>(f)(std::forward<Args>(args)...);
-}
-
-template <typename M, typename C, typename... Args>
-inline constexpr auto invoke(M(C::*d), Args&&... args)
- -> decltype(std::mem_fn(d)(std::forward<Args>(args)...)) {
- return std::mem_fn(d)(std::forward<Args>(args)...);
-}
-
} // namespace function
} // namespace detail
// not copyable
Function(const Function&) = delete;
+#if __OBJC__
+ // Make sure Objective C blocks are copied
+ template <class ReturnType, class... Args>
+ /*implicit*/ Function(ReturnType (^objCBlock)(Args... args))
+ : Function([blockCopy = (ReturnType (^)(Args...))[objCBlock copy]](
+ Args... args) { return blockCopy(args...); }){};
+#endif
+
/**
* Move constructor
*/
*/
template <
typename Fun,
- typename FunT = detail::function::DecayIfConstructible<Fun>,
+ typename = detail::function::EnableIfNotFunction<Fun>,
typename = typename Traits::template ResultOf<Fun>>
- /* implicit */ Function(Fun&& fun) noexcept(
- IsSmall<Fun>::value && noexcept(FunT(std::declval<Fun>())))
+ /* implicit */ Function(Fun fun) noexcept(
+ IsSmall<Fun>::value && noexcept(Fun(std::declval<Fun>())))
: Function(static_cast<Fun&&>(fun), IsSmall<Fun>{}) {}
/**
Function& operator=(const Function&) = delete;
+#if __OBJC__
+ // Make sure Objective C blocks are copied
+ template <class ReturnType, class... Args>
+ /* implicit */ Function &operator=(ReturnType (^objCBlock)(Args... args)) {
+ (*this) = [blockCopy = (ReturnType (^)(Args...))[objCBlock copy]](
+ Args... args) { return blockCopy(args...); };
+ return *this;
+ }
+#endif
+
/**
* Move assignment operator
*
* selected by overload resolution when `fun` is not a compatible function.
*/
template <typename Fun, typename = decltype(Function(std::declval<Fun>()))>
- Function& operator=(Fun&& fun) noexcept(
+ Function& operator=(Fun fun) noexcept(
noexcept(/* implicit */ Function(std::declval<Fun>()))) {
// Doing this in place is more efficient when we can do so safely.
if (noexcept(/* implicit */ Function(std::declval<Fun>()))) {
template <typename Fun, typename FunctionType>
struct IsCallableAs : IsCallableAsImpl<Fun, FunctionType> {};
-}
-}
+} // namespace function
+} // namespace detail
/**
* @class FunctionRef
template <typename Fun>
static ReturnType call(void* object, Args&&... args) {
using Pointer = _t<std::add_pointer<Fun>>;
- return static_cast<ReturnType>(detail::function::invoke(
+ return static_cast<ReturnType>(invoke(
static_cast<Fun&&>(*static_cast<Pointer>(object)),
static_cast<Args&&>(args)...));
}