mirror of
https://github.com/bolero-MURAKAMI/Sprout
synced 2024-11-12 21:09:01 +00:00
115 lines
4.3 KiB
C++
115 lines
4.3 KiB
C++
#ifndef SPROUT_ALGORITHM_FIXED_SHUFFLE_HPP
|
|
#define SPROUT_ALGORITHM_FIXED_SHUFFLE_HPP
|
|
|
|
#include <cstddef>
|
|
#include <sprout/config.hpp>
|
|
#include <sprout/index_tuple.hpp>
|
|
#include <sprout/array.hpp>
|
|
#include <sprout/pit.hpp>
|
|
#include <sprout/container/traits.hpp>
|
|
#include <sprout/container/functions.hpp>
|
|
#include <sprout/iterator/operation.hpp>
|
|
#include <sprout/utility/forward.hpp>
|
|
#include <sprout/algorithm/fixed/result_of.hpp>
|
|
#include <sprout/algorithm/fixed/swap_element.hpp>
|
|
#include <sprout/numeric/fixed/iota.hpp>
|
|
#ifdef SPROUT_WORKAROUND_NOT_TERMINATE_RECURSIVE_CONSTEXPR_FUNCTION_TEMPLATE
|
|
# include <sprout/random/uniform_smallint.hpp>
|
|
# define SPROUT_WORKAROUND_UNIFORM_INT_DISTRIBUTION sprout::random::uniform_smallint
|
|
#else
|
|
# include <sprout/random/uniform_int_distribution.hpp>
|
|
# define SPROUT_WORKAROUND_UNIFORM_INT_DISTRIBUTION sprout::random::uniform_int_distribution
|
|
#endif
|
|
|
|
namespace sprout {
|
|
namespace fixed {
|
|
namespace detail {
|
|
template<std::size_t N, typename Random>
|
|
inline SPROUT_CONSTEXPR sprout::array<std::ptrdiff_t, N>
|
|
make_shuffle_indexes_1(std::ptrdiff_t n, Random const& rnd, sprout::array<std::ptrdiff_t, N> const& arr, std::ptrdiff_t i) {
|
|
return i < n - 1
|
|
? sprout::fixed::detail::make_shuffle_indexes_1(
|
|
n,
|
|
rnd(),
|
|
sprout::fixed::swap_element(arr, arr.begin() + i, arr.begin() + rnd.result()),
|
|
i + 1
|
|
)
|
|
: sprout::fixed::swap_element(arr, arr.begin() + i, arr.begin() + rnd.result())
|
|
;
|
|
}
|
|
template<std::size_t N, typename UniformRandomNumberGenerator>
|
|
inline SPROUT_CONSTEXPR sprout::array<std::ptrdiff_t, N>
|
|
make_shuffle_indexes(std::ptrdiff_t n, UniformRandomNumberGenerator&& g) {
|
|
return n > 1
|
|
? sprout::fixed::detail::make_shuffle_indexes_1(
|
|
n,
|
|
SPROUT_WORKAROUND_UNIFORM_INT_DISTRIBUTION<std::ptrdiff_t>(0, n - 1)(sprout::forward<UniformRandomNumberGenerator>(g)),
|
|
sprout::fixed::iota(sprout::pit<sprout::array<std::ptrdiff_t, N> >(), 0),
|
|
0
|
|
)
|
|
: sprout::array<std::ptrdiff_t, N>{{}}
|
|
;
|
|
}
|
|
template<typename Container, typename Shuffled, sprout::index_t... Indexes>
|
|
inline SPROUT_CONSTEXPR typename sprout::fixed::result_of::algorithm<Container>::type
|
|
shuffle_impl_1(
|
|
Container const& cont,
|
|
sprout::index_tuple<Indexes...>,
|
|
Shuffled const& shuffled,
|
|
typename sprout::container_traits<Container>::difference_type offset,
|
|
typename sprout::container_traits<Container>::size_type size
|
|
)
|
|
{
|
|
return sprout::remake<Container>(
|
|
cont,
|
|
sprout::size(cont),
|
|
(Indexes >= offset && Indexes < offset + size
|
|
? *sprout::next(sprout::begin(cont), shuffled[Indexes - offset])
|
|
: *sprout::next(sprout::internal_begin(cont), Indexes)
|
|
)...
|
|
);
|
|
}
|
|
template<typename Container, typename UniformRandomNumberGenerator, sprout::index_t... Indexes>
|
|
inline SPROUT_CONSTEXPR typename sprout::fixed::result_of::algorithm<Container>::type
|
|
shuffle_impl(
|
|
Container const& cont,
|
|
sprout::index_tuple<Indexes...> indexes,
|
|
UniformRandomNumberGenerator&& g,
|
|
typename sprout::container_traits<Container>::difference_type offset,
|
|
typename sprout::container_traits<Container>::size_type size
|
|
)
|
|
{
|
|
return sprout::fixed::detail::shuffle_impl_1(
|
|
cont,
|
|
indexes,
|
|
sprout::fixed::detail::make_shuffle_indexes<sprout::container_traits<Container>::static_size>(
|
|
size,
|
|
sprout::forward<UniformRandomNumberGenerator>(g)
|
|
),
|
|
offset,
|
|
size
|
|
);
|
|
}
|
|
} // namespace detail
|
|
//
|
|
// shuffle
|
|
//
|
|
template<typename Container, typename UniformRandomNumberGenerator>
|
|
inline SPROUT_CONSTEXPR typename sprout::fixed::result_of::algorithm<Container>::type
|
|
shuffle(Container const& cont, UniformRandomNumberGenerator&& g) {
|
|
return sprout::fixed::detail::shuffle_impl(
|
|
cont,
|
|
sprout::index_range<0, sprout::container_traits<Container>::static_size>::make(),
|
|
sprout::forward<UniformRandomNumberGenerator>(g),
|
|
sprout::internal_begin_offset(cont),
|
|
sprout::size(cont)
|
|
);
|
|
}
|
|
} // namespace fixed
|
|
|
|
using sprout::fixed::shuffle;
|
|
} // namespace sprout
|
|
|
|
#undef SPROUT_WORKAROUND_UNIFORM_INT_DISTRIBUTION
|
|
|
|
#endif // #ifndef SPROUT_ALGORITHM_FIXED_SHUFFLE_HPP
|