diff options
Diffstat (limited to 'lib/collections.hpp')
| -rw-r--r-- | lib/collections.hpp | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/lib/collections.hpp b/lib/collections.hpp index 5d39e79..47967b2 100644 --- a/lib/collections.hpp +++ b/lib/collections.hpp @@ -1,7 +1,10 @@ #pragma once +#include <algorithm> #include <array> #include <span> +#include <utility> +#include <vector> template<typename T, typename E> concept SequentialCollection = requires(T c) { @@ -59,3 +62,43 @@ operator*=(std::span<T> & in, auto && f) } return in; } + +template<template<typename...> typename C, typename... T> +constexpr auto +operator*(const C<T...> & in, auto && f) +{ + C<decltype(f(in[0]))> out; + + std::transform(in.begin(), in.end(), std::inserter(out, out.end()), f); + return out; +} + +template<typename... T> +constexpr auto & +operator+=(std::vector<T...> & in, std::vector<T...> && src) +{ + in.reserve(in.size() + src.size()); + std::move(src.begin(), src.end(), std::back_inserter(in)); + return in; +} + +template<typename... T> +constexpr auto +operator+(std::vector<T...> && in, std::vector<T...> && src) +{ + in.reserve(in.size() + src.size()); + std::move(src.begin(), src.end(), std::back_inserter(in)); + return in; +} + +template<template<typename> typename Direction = std::plus> +static auto +vectorOfN(std::integral auto N, unsigned int start = {}, unsigned int step = 1) +{ + std::vector<unsigned int> v; + v.resize(N); + std::generate_n(v.begin(), N, [&start, step, adj = Direction {}]() { + return std::exchange(start, adj(start, step)); + }); + return v; +} |