[libvmake]add gourp()

This commit is contained in:
方而静 2023-07-19 17:08:59 +08:00
parent 708f9ca9ec
commit 650c363f03
2 changed files with 67 additions and 0 deletions

View File

@ -0,0 +1,9 @@
#include "../vmake.hpp"
using namespace std;
int main() {
vmake::outputln_n(cout, 5, " ", vmake::transform(vmake::group<2>(vmake::iota(1)), [](auto x) {
return get<0>(x) * get<1>(x);
}));
return 0;
}

View File

@ -401,6 +401,64 @@ inline filteror<typename std::decay<Gen>::type, typename std::decay<Pred>::type>
return {std::forward<Gen>(g), std::forward<Pred>(p)};
}
namespace details {
template<typename Tp, int n>
struct repeat_tuple {
static_assert(n > 0, "");
using type = decltype(std::tuple_cat(std::declval<std::tuple<Tp>>()
, std::declval<typename repeat_tuple<Tp, n - 1>::type>()));
};
template<typename Tp>
struct repeat_tuple<Tp, 1> {
using type = std::tuple<Tp>;
};
template<typename Gen, int n>
struct group_helper {
static auto load(Gen &g) {
auto x = g();
auto y = group_helper<Gen, n - 1>::load(g);
return std::tuple_cat(std::make_tuple<typename Gen::result>(std::move(x)), std::move(y));
}
};
template<typename Gen>
struct group_helper<Gen, 1> {
static auto load(Gen &g) {
return std::make_tuple<typename Gen::result>(g());
}
};
template<typename Gen, int n>
struct grouper {
static_assert(n > 0, "");
using result = typename repeat_tuple<typename Gen::result, n>::type;
using core = Gen;
Gen g;
grouper(grouper<Gen, n>&& g) = default;
grouper(const grouper<Gen, n>& g) = default;
grouper(Gen &&g) : g(std::forward<Gen>(g)) {}
bool is_terminated() noexcept {
return g.is_terminated();
}
result operator()() {
return group_helper<Gen, n>::load(g);
}
};
}
template<int n, typename Gen>
inline auto group(Gen &&g) {
return details::grouper<typename std::decay<Gen>::type, n>(std::forward<Gen>(g));
}
namespace rng {
inline auto make_seed() {