1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175
|
/// \file
// Range v3 library
//
// Copyright Eric Niebler 2013-present
//
// Use, modification and distribution is subject to the
// Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
//
// Project home: https://github.com/ericniebler/range-v3
//
#ifndef RANGES_V3_FUNCTIONAL_OVERLOAD_HPP
#define RANGES_V3_FUNCTIONAL_OVERLOAD_HPP
#include <meta/meta.hpp>
#include <concepts/concepts.hpp>
#include <range/v3/range_fwd.hpp>
#include <range/v3/functional/concepts.hpp>
#include <range/v3/functional/invoke.hpp>
#include <range/v3/utility/static_const.hpp>
#include <range/v3/detail/prologue.hpp>
namespace ranges
{
/// \addtogroup group-functional
/// @{
/// \cond
namespace detail
{
struct _id
{
template<typename T>
using invoke = T;
};
struct _ref
{
template<typename T>
using invoke = T &;
};
struct _cref
{
template<typename T>
using invoke = T const &;
};
template<typename T>
struct _bind_front
{
template<typename... Args>
using invoke = invoke_result_t<T, Args...>;
};
} // namespace detail
/// \endcond
template<typename... Ts>
struct overloaded;
template<>
struct overloaded<>
{
private:
template<typename...>
friend struct overloaded;
template<typename, typename...>
using _result_t = void;
};
template<typename First, typename... Rest>
struct overloaded<First, Rest...>
{
private:
template<typename...>
friend struct overloaded;
RANGES_NO_UNIQUE_ADDRESS
First first_;
RANGES_NO_UNIQUE_ADDRESS
overloaded<Rest...> second_;
template<typename Qual>
using _result_first = detail::_bind_front<meta::invoke<Qual, First>>;
template<typename Qual>
struct _result_second
{
template<typename... Args>
using invoke = typename overloaded<Rest...>
::template _result_t<Qual, Args...>;
};
template<typename Qual, typename... Args>
using _result_t =
meta::invoke<
meta::conditional_t<
(bool) invocable<meta::invoke<Qual, First>, Args...>,
_result_first<Qual>,
_result_second<Qual>>,
Args...>;
public:
overloaded() = default;
constexpr overloaded(First first, Rest... rest)
: first_(static_cast<First &&>(first))
, second_{static_cast<Rest &&>(rest)...}
{}
template(typename... Args)(
requires invocable<First, Args...>)
constexpr _result_t<detail::_id, Args...> operator()(Args &&... args) &&
{
return invoke((First &&) first_, (Args &&) args...);
}
template(typename... Args)(
requires (!invocable<First, Args...>) AND
invocable<overloaded<Rest...>, Args...>)
constexpr _result_t<detail::_id, Args...> operator()(Args &&... args) &&
{
return invoke((overloaded<Rest...> &&) second_, (Args &&) args...);
}
template(typename... Args)(
requires invocable<First &, Args...>)
constexpr _result_t<detail::_ref, Args...> operator()(Args &&... args) &
{
return invoke(first_, (Args &&) args...);
}
template(typename... Args)(
requires (!invocable<First &, Args...>) AND
invocable<overloaded<Rest...> &, Args...>)
constexpr _result_t<detail::_ref, Args...> operator()(Args &&... args) &
{
return invoke(second_, (Args &&) args...);
}
template(typename... Args)(
requires invocable<First const &, Args...>)
constexpr _result_t<detail::_cref, Args...> operator()(Args &&... args) const &
{
return invoke(first_, (Args &&) args...);
}
template(typename... Args)(
requires (!invocable<First const &, Args...>) AND
invocable<overloaded<Rest...> const &, Args...>)
constexpr _result_t<detail::_cref, Args...> operator()(Args &&... args) const &
{
return invoke(second_, (Args &&) args...);
}
};
struct overload_fn
{
template<typename Fn>
constexpr Fn operator()(Fn fn) const
{
return fn;
}
template<typename... Fns>
constexpr overloaded<Fns...> operator()(Fns... fns) const
{
return overloaded<Fns...>{static_cast<Fns &&>(fns)...};
}
};
/// \ingroup group-functional
/// \sa `overload_fn`
RANGES_INLINE_VARIABLE(overload_fn, overload)
/// @}
} // namespace ranges
#include <range/v3/detail/epilogue.hpp>
#endif
|