function_ref.h 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. // Copyright 2019 The Abseil Authors.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // https://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #ifndef ABSL_FUNCTIONAL_INTERNAL_FUNCTION_REF_H_
  15. #define ABSL_FUNCTIONAL_INTERNAL_FUNCTION_REF_H_
  16. #include <cassert>
  17. #include <functional>
  18. #include <type_traits>
  19. #include "absl/base/internal/invoke.h"
  20. #include "absl/meta/type_traits.h"
  21. namespace absl {
  22. ABSL_NAMESPACE_BEGIN
  23. namespace functional_internal {
  24. // Like a void* that can handle function pointers as well. The standard does not
  25. // allow function pointers to round-trip through void*, but void(*)() is fine.
  26. //
  27. // Note: It's important that this class remains trivial and is the same size as
  28. // a pointer, since this allows the compiler to perform tail-call optimizations
  29. // when the underlying function is a callable object with a matching signature.
  30. union VoidPtr {
  31. const void* obj;
  32. void (*fun)();
  33. };
  34. // Chooses the best type for passing T as an argument.
  35. // Attempt to be close to SystemV AMD64 ABI. Objects with trivial copy ctor are
  36. // passed by value.
  37. template <typename T>
  38. constexpr bool PassByValue() {
  39. return !std::is_lvalue_reference<T>::value &&
  40. absl::is_trivially_copy_constructible<T>::value &&
  41. absl::is_trivially_copy_assignable<
  42. typename std::remove_cv<T>::type>::value &&
  43. std::is_trivially_destructible<T>::value &&
  44. sizeof(T) <= 2 * sizeof(void*);
  45. }
  46. template <typename T>
  47. struct ForwardT : std::conditional<PassByValue<T>(), T, T&&> {};
  48. // An Invoker takes a pointer to the type-erased invokable object, followed by
  49. // the arguments that the invokable object expects.
  50. //
  51. // Note: The order of arguments here is an optimization, since member functions
  52. // have an implicit "this" pointer as their first argument, putting VoidPtr
  53. // first allows the compiler to perform tail-call optimization in many cases.
  54. template <typename R, typename... Args>
  55. using Invoker = R (*)(VoidPtr, typename ForwardT<Args>::type...);
  56. //
  57. // InvokeObject and InvokeFunction provide static "Invoke" functions that can be
  58. // used as Invokers for objects or functions respectively.
  59. //
  60. // static_cast<R> handles the case the return type is void.
  61. template <typename Obj, typename R, typename... Args>
  62. R InvokeObject(VoidPtr ptr, typename ForwardT<Args>::type... args) {
  63. auto o = static_cast<const Obj*>(ptr.obj);
  64. return static_cast<R>(
  65. absl::base_internal::invoke(*o, std::forward<Args>(args)...));
  66. }
  67. template <typename Fun, typename R, typename... Args>
  68. R InvokeFunction(VoidPtr ptr, typename ForwardT<Args>::type... args) {
  69. auto f = reinterpret_cast<Fun>(ptr.fun);
  70. return static_cast<R>(
  71. absl::base_internal::invoke(f, std::forward<Args>(args)...));
  72. }
  73. template <typename Sig>
  74. void AssertNonNull(const std::function<Sig>& f) {
  75. assert(f != nullptr);
  76. (void)f;
  77. }
  78. template <typename F>
  79. void AssertNonNull(const F&) {}
  80. template <typename F, typename C>
  81. void AssertNonNull(F C::*f) {
  82. assert(f != nullptr);
  83. (void)f;
  84. }
  85. template <bool C>
  86. using EnableIf = typename ::std::enable_if<C, int>::type;
  87. } // namespace functional_internal
  88. ABSL_NAMESPACE_END
  89. } // namespace absl
  90. #endif // ABSL_FUNCTIONAL_INTERNAL_FUNCTION_REF_H_