From e7d946aadebfdf49cc96e6a518c1af1ef7f5c71c Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Mon, 10 Apr 2023 14:34:56 +0100
Subject: Add operator* helper specialised for std::span

---
 lib/collections.hpp | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

(limited to 'lib')

diff --git a/lib/collections.hpp b/lib/collections.hpp
index b51d18e..8f5a43d 100644
--- a/lib/collections.hpp
+++ b/lib/collections.hpp
@@ -78,6 +78,27 @@ operator*(const C<T...> & in, auto && f)
 	return out;
 }
 
+template<typename T, std::size_t N>
+[[nodiscard]] constexpr auto
+operator*(const std::span<T, N> in, auto && f)
+{
+	std::array<decltype(f(std::declval<T>())), N> out;
+
+	std::transform(in.begin(), in.end(), out.begin(), f);
+	return out;
+}
+
+template<typename T>
+[[nodiscard]] constexpr auto
+operator*(const std::span<T, std::dynamic_extent> in, auto && f)
+{
+	std::vector<decltype(f(std::declval<T>()))> out;
+
+	out.reserve(in.size());
+	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)
-- 
cgit v1.2.3