diff --git a/wpiutil/src/main/native/include/wpi/SpanExtras.h b/wpiutil/src/main/native/include/wpi/SpanExtras.h index aa7b9abcf5..790a66f0ea 100644 --- a/wpiutil/src/main/native/include/wpi/SpanExtras.h +++ b/wpiutil/src/main/native/include/wpi/SpanExtras.h @@ -10,19 +10,49 @@ namespace wpi { /// Drop the first \p N elements of the array. -template -constexpr std::span drop_front(std::span in, +template +constexpr std::span drop_front(std::span in, typename std::span::size_type n = 1) { assert(in.size() >= n && "Dropping more elements than exist"); return in.subspan(n, in.size() - n); } /// Drop the last \p N elements of the array. -template -constexpr std::span drop_back(std::span in, +template +constexpr std::span drop_back(std::span in, typename std::span::size_type n = 1) { assert(in.size() >= n && "Dropping more elements than exist"); return in.subspan(0, in.size() - n); } +/** + * Returns a span equal to @p in but with only the first @p n + * elements remaining. If @p n is greater than the length of the + * span, the entire span is returned. + */ +template +constexpr std::span take_front(std::span in, + typename std::span::size_type n = 1) { + auto length = in.size(); + if (n >= length) { + return in; + } + return drop_back(in, length - n); +} + +/** + * Returns a span equal to @p in but with only the last @p n + * elements remaining. If @p n is greater than the length of the + * span, the entire span is returned. + */ +template +constexpr std::span take_back(std::span in, + typename std::span::size_type n = 1) { + auto length = in.size(); + if (n >= length) { + return in; + } + return drop_front(in, length - n); +} + } // namespace wpi