18#include <initializer_list>
27#include "lib/stdcompat/utility.h"
28#include "pw_assert/assert.h"
29#include "pw_containers/internal/deque_iterator.h"
30#include "pw_containers/internal/traits.h"
31#include "pw_containers/internal/wrap.h"
32#include "pw_numeric/checked_arithmetic.h"
33#include "pw_span/span.h"
35namespace pw::containers::internal {
38using EnableIfIterable =
39 std::enable_if_t<
true,
decltype(T().begin(), T().end())>;
53template <
typename SizeType>
56 using size_type = SizeType;
58 static_assert(std::is_unsigned_v<SizeType>,
"SizeType must be unsigned.");
62 [[nodiscard]]
constexpr bool empty()
const noexcept {
return size() == 0; }
64 [[nodiscard]]
constexpr bool full()
const noexcept {
69 constexpr size_type
size() const noexcept {
return count_; }
72 constexpr size_type
capacity() const noexcept {
return capacity_; }
78 capacity_ = std::exchange(other.capacity_, 0);
79 count_ = std::exchange(other.count_, 0);
80 head_ = cpp20::exchange(other.head_, 0);
81 tail_ = cpp20::exchange(other.tail_, 0);
84 void SwapIndices(GenericDequeBase& other)
noexcept {
85 std::swap(capacity_, other.capacity_);
86 std::swap(count_, other.count_);
87 std::swap(head_, other.head_);
88 std::swap(tail_, other.tail_);
93 bool CanExtendBuffer()
const {
return tail_ > head_ || tail_ == 0; }
96 bool CanShrinkBuffer()
const {
return head_ == 0; }
98 void HandleNewBuffer(size_type new_capacity) {
99 capacity_ = new_capacity;
101 tail_ =
size() == new_capacity ? 0 : count_;
104 void HandleExtendedBuffer(size_type new_capacity) {
105 capacity_ = new_capacity;
107 tail_ = head_ + count_;
111 void HandleShrunkBuffer(size_type new_capacity) {
112 capacity_ = new_capacity;
113 if (tail_ == new_capacity) {
120 template <
typename Derived,
typename ValueType,
typename S>
121 friend class GenericDeque;
123 explicit constexpr GenericDequeBase(size_type initial_capacity) noexcept
124 : capacity_(initial_capacity), count_(0), head_(0), tail_(0) {}
126 constexpr void ClearIndices() {
135 constexpr size_type AbsoluteIndex(
const size_type relative_index)
const {
136 const size_type absolute_index = head_ + relative_index;
138 return absolute_index;
144 constexpr size_type AbsoluteIndexChecked(
145 const size_type relative_index)
const {
146 PW_ASSERT(relative_index <
size());
147 return AbsoluteIndex(relative_index);
150 constexpr void PushBack(size_type count = 1) {
151 IncrementWithWrap(tail_, count,
capacity());
154 constexpr void PushFront(size_type count = 1) {
155 DecrementWithWrap(head_, count,
capacity());
158 constexpr void PopFront() {
159 IncrementWithWrap(head_, size_type(1),
capacity());
162 constexpr void PopBack() {
163 DecrementWithWrap(tail_, size_type(1),
capacity());
177template <
typename Derived,
typename ValueType,
typename SizeType>
183 using value_type = ValueType;
184 using size_type = SizeType;
185 using difference_type = ptrdiff_t;
186 using reference = value_type&;
187 using const_reference =
const value_type&;
188 using pointer = value_type*;
189 using const_pointer =
const value_type*;
190 using iterator = containers::internal::DequeIterator<Derived>;
191 using const_iterator = containers::internal::DequeIterator<const Derived>;
209 void assign(size_type count,
const value_type& value) {
215 template <
typename It,
217 typename = containers::internal::EnableIfInputIterator<It>>
221 void assign(
const std::initializer_list<value_type>& list) {
222 assign(list.begin(), list.end());
227 constexpr reference at(size_type index) {
228 return data()[Base::AbsoluteIndexChecked(index)];
230 constexpr const_reference at(size_type index)
const {
231 return data()[Base::AbsoluteIndexChecked(index)];
234 constexpr reference operator[](size_type index) {
235 PW_DASSERT(index <
size());
236 return data()[Base::AbsoluteIndex(index)];
238 constexpr const_reference operator[](size_type index)
const {
239 PW_DASSERT(index <
size());
240 return data()[Base::AbsoluteIndex(index)];
243 constexpr reference front() {
244 PW_DASSERT(!empty());
245 return data()[head()];
247 constexpr const_reference front()
const {
248 PW_DASSERT(!empty());
249 return data()[head()];
252 constexpr reference back() {
253 PW_DASSERT(!empty());
254 return data()[Base::AbsoluteIndex(
size() - 1)];
256 constexpr const_reference back()
const {
257 PW_DASSERT(!empty());
258 return data()[Base::AbsoluteIndex(
size() - 1)];
262 constexpr std::pair<span<const value_type>, span<const value_type>>
265 auto [first, second] = std::as_const(*this).contiguous_data();
266 return {{
const_cast<pointer
>(first.data()), first.size()},
267 {
const_cast<pointer
>(second.data()), second.size()}};
272 constexpr iterator begin() noexcept {
277 return iterator(derived(), 0);
279 constexpr const_iterator begin() const noexcept {
return cbegin(); }
280 constexpr const_iterator cbegin() const noexcept {
284 return const_iterator(derived(), 0);
287 constexpr iterator end() noexcept {
return iterator(derived(),
size()); }
288 constexpr const_iterator end() const noexcept {
return cend(); }
289 constexpr const_iterator cend() const noexcept {
290 return const_iterator(derived(),
size());
295 constexpr void clear() {
297 Base::ClearIndices();
301 iterator
erase(const_iterator pos) {
302 PW_DASSERT(pos.pos_ <
size());
303 return erase(pos, pos + 1);
308 iterator
erase(const_iterator first, const_iterator last);
310 void push_back(
const value_type& value) { PW_ASSERT(try_push_back(value)); }
312 void push_back(value_type&& value) {
313 PW_ASSERT(try_push_back(std::move(value)));
316 template <
typename... Args>
317 void emplace_back(Args&&... args) {
318 PW_ASSERT(try_emplace_back(std::forward<Args>(args)...));
323 void push_front(
const value_type& value) { PW_ASSERT(try_push_front(value)); }
325 void push_front(value_type&& value) {
326 PW_ASSERT(try_push_front(std::move(value)));
329 template <
typename... Args>
330 void emplace_front(Args&&... args) {
331 PW_ASSERT(try_emplace_front(std::forward<Args>(args)...));
340 template <
typename... Args>
341 iterator
emplace(const_iterator pos, Args&&... args) {
342 std::optional<iterator> result =
344 PW_ASSERT(result.has_value());
351 iterator
insert(const_iterator pos,
const value_type& value) {
352 std::optional<iterator> result =
try_insert(pos, value);
353 PW_ASSERT(result.has_value());
361 iterator
insert(const_iterator pos, value_type&& value) {
362 std::optional<iterator> result =
try_insert(pos, std::move(value));
363 PW_ASSERT(result.has_value());
373 const value_type& value) {
374 std::optional<iterator> result =
try_insert(pos, count, value);
375 PW_ASSERT(result.has_value());
383 template <
typename InputIt,
384 typename = containers::internal::EnableIfInputIterator<InputIt>>
385 iterator
insert(const_iterator pos, InputIt first, InputIt last);
390 iterator
insert(const_iterator pos, std::initializer_list<value_type> ilist) {
391 std::optional<iterator> result =
try_insert(pos, ilist);
392 PW_ASSERT(result.has_value());
396 void resize(size_type new_size) { resize(new_size, value_type()); }
398 void resize(size_type new_size,
const value_type& value) {
399 PW_ASSERT(try_resize(new_size, value));
403 explicit constexpr GenericDeque(size_type initial_capacity) noexcept
404 : GenericDequeBase<SizeType>(initial_capacity) {}
406 constexpr void DestroyAll() {
407 if constexpr (!std::is_trivially_destructible_v<value_type>) {
409 std::destroy(first.begin(), first.end());
410 std::destroy(second.begin(), second.end());
417 Derived& operator=(
const std::initializer_list<value_type>& list) {
422 template <
typename T,
typename = containers::
internal::EnableIfIterable<T>>
423 Derived& operator=(
const T& other) {
424 assign(other.begin(), other.end());
434 [[nodiscard]]
bool try_assign(size_type count,
const value_type& value);
442 template <
typename It,
444 typename = containers::internal::EnableIfForwardIterator<It>>
449 [[nodiscard]]
bool try_assign(
const std::initializer_list<value_type>& list) {
460 template <
typename... Args>
461 [[nodiscard]] std::optional<iterator>
try_emplace(const_iterator pos,
469 [[nodiscard]] std::optional<iterator>
try_insert(const_iterator pos,
470 const value_type& value) {
479 [[nodiscard]] std::optional<iterator>
try_insert(const_iterator pos,
480 value_type&& value) {
489 [[nodiscard]] std::optional<iterator>
try_insert(const_iterator pos,
491 const value_type& value);
498 template <
typename ForwardIt,
499 typename = containers::internal::EnableIfForwardIterator<ForwardIt>>
500 [[nodiscard]] std::optional<iterator>
try_insert(const_iterator pos,
510 const_iterator pos, std::initializer_list<value_type> ilist) {
511 return try_insert(pos, ilist.begin(), ilist.end());
514 [[nodiscard]]
bool try_push_back(
const value_type& value) {
515 return try_emplace_back(value);
518 [[nodiscard]]
bool try_push_back(value_type&& value) {
519 return try_emplace_back(std::move(value));
522 template <
typename... Args>
523 [[nodiscard]]
bool try_emplace_back(Args&&... args);
525 [[nodiscard]]
bool try_push_front(
const value_type& value) {
526 return try_emplace_front(value);
529 [[nodiscard]]
bool try_push_front(value_type&& value) {
530 return try_emplace_front(std::move(value));
533 template <
typename... Args>
534 [[nodiscard]]
bool try_emplace_front(Args&&... args);
536 [[nodiscard]]
bool try_resize(size_type new_size) {
537 return try_resize(new_size, value_type());
540 [[nodiscard]]
bool try_resize(size_type new_size,
const value_type& value);
542 template <
typename... Args>
543 [[nodiscard]] std::optional<iterator> try_emplace_shift_right(
544 const_iterator pos, Args&&... args);
546 [[nodiscard]] std::optional<iterator> try_insert_shift_right(
547 const_iterator pos, size_type count,
const value_type& value);
549 template <
typename ForwardIt,
550 typename = containers::internal::EnableIfForwardIterator<ForwardIt>>
551 [[nodiscard]] std::optional<iterator> try_insert_shift_right(
552 const_iterator pos, ForwardIt first, ForwardIt last);
555 constexpr Derived& derived() {
return static_cast<Derived&
>(*this); }
556 constexpr const Derived& derived()
const {
557 return static_cast<const Derived&
>(*this);
560 constexpr size_type head()
const {
return Base::head_; }
561 constexpr size_type tail()
const {
return Base::tail_; }
564 constexpr pointer data() {
return derived().data(); }
565 constexpr const_pointer data()
const {
return derived().data(); }
568 bool ShiftForInsert(size_type insert_index, size_type new_items);
571 void ShiftLeft(size_type insert_index, size_type new_items);
572 void ShiftRight(size_type insert_index, size_type new_items);
575 bool CheckCapacityAdd(size_type count) {
577 return CheckedAdd(
size(), count, new_size) && CheckCapacity(new_size);
581 constexpr bool CheckCapacity(size_type new_size) {
582 if constexpr (Derived::kFixedCapacity) {
585 return derived().try_reserve(new_size);
590 template <
typename... Args>
591 void EmplaceBackUnchecked(Args&&... args) {
592 new (&data()[tail()]) value_type(std::forward<Args>(args)...);
599template <
typename Derived,
typename ValueType,
typename SizeType>
600template <
typename It,
int&...,
typename>
603 if constexpr (Derived::kFixedCapacity ||
605 typename std::iterator_traits<It>::iterator_category,
606 std::input_iterator_tag>) {
608 while (start != finish) {
613 PW_ASSERT(try_assign(start, finish));
617template <
typename Derived,
typename ValueType,
typename SizeType>
619 size_type count,
const value_type& value) {
620 if (!CheckCapacity(count)) {
624 Base::PushBack(count);
625 std::uninitialized_fill_n(data(), count, value);
629template <
typename Derived,
typename ValueType,
typename SizeType>
630template <
typename It,
int&...,
typename>
634 static_assert(std::is_convertible_v<
635 typename std::iterator_traits<It>::iterator_category,
636 std::forward_iterator_tag>);
637 const auto items = std::distance(start, finish);
638 PW_DASSERT(items >= 0);
639 if (
static_cast<std::make_unsigned_t<decltype(items)
>>(items) >
640 std::numeric_limits<size_type>::max()) {
643 const size_type count =
static_cast<size_type
>(items);
644 if (!CheckCapacity(count)) {
649 Base::PushBack(count);
650 std::uninitialized_move(start, finish, data());
654template <
typename Derived,
typename ValueType,
typename SizeType>
660 if (tail() > head()) {
674template <
typename Derived,
typename ValueType,
typename SizeType>
675template <
typename... Args>
678 if (!CheckCapacityAdd(1)) {
681 EmplaceBackUnchecked(std::forward<Args>(args)...);
685template <
typename Derived,
typename ValueType,
typename SizeType>
686void GenericDeque<Derived, ValueType, SizeType>::pop_back() {
688 if constexpr (!std::is_trivially_destructible_v<value_type>) {
689 std::destroy_at(&back());
694template <
typename Derived,
typename ValueType,
typename SizeType>
695template <
typename... Args>
696bool GenericDeque<Derived, ValueType, SizeType>::try_emplace_front(
698 if (!CheckCapacityAdd(1)) {
702 new (&data()[head()]) value_type(std::forward<Args>(args)...);
706template <
typename Derived,
typename ValueType,
typename SizeType>
707void GenericDeque<Derived, ValueType, SizeType>::pop_front() {
709 if constexpr (!std::is_trivially_destructible_v<value_type>) {
710 std::destroy_at(&front());
715template <
typename Derived,
typename ValueType,
typename SizeType>
716bool GenericDeque<Derived, ValueType, SizeType>::try_resize(
717 size_type new_size,
const value_type& value) {
718 if (size() < new_size) {
719 if (!CheckCapacity(new_size)) {
722 const size_type new_items = new_size - size();
723 for (size_type i = 0; i < new_items; ++i) {
724 EmplaceBackUnchecked(value);
727 while (size() > new_size) {
734template <
typename Derived,
typename ValueType,
typename SizeType>
735typename GenericDeque<Derived, ValueType, SizeType>::iterator
737 const_iterator last) {
738 PW_DASSERT(first <= last);
739 const iterator first_it(derived(), first.pos_);
740 const iterator last_it(derived(), last.pos_);
742 const size_type items_to_erase =
static_cast<size_type
>(last - first);
743 if (items_to_erase == 0) {
747 const size_type items_after =
static_cast<size_type
>(size() - last.pos_);
749 if (first.pos_ < items_after) {
750 std::move_backward(begin(), first_it, last_it);
751 if constexpr (!std::is_trivially_destructible_v<value_type>) {
752 std::destroy(begin(), begin() + items_to_erase);
754 Base::head_ = Base::AbsoluteIndex(items_to_erase);
756 std::move(last_it, end(), first_it);
757 if constexpr (!std::is_trivially_destructible_v<value_type>) {
758 std::destroy(first_it + items_after, end());
760 Base::tail_ = Base::AbsoluteIndex(first.pos_ + items_after);
762 Base::count_ = size() - items_to_erase;
766template <
typename Derived,
typename ValueType,
typename SizeType>
767template <
typename... Args>
768std::optional<typename GenericDeque<Derived, ValueType, SizeType>::iterator>
772 if (!ShiftForInsert(pos.pos_, 1)) {
775 iterator it(derived(), pos.pos_);
776 new (std::addressof(*it)) value_type(std::forward<Args>(args)...);
780template <
typename Derived,
typename ValueType,
typename SizeType>
781bool GenericDeque<Derived, ValueType, SizeType>::ShiftForInsert(
782 const size_type insert_index, size_type new_items) {
783 if (!CheckCapacityAdd(new_items)) {
787 if (insert_index < size() / 2) {
788 ShiftLeft(insert_index, new_items);
790 ShiftRight(insert_index, new_items);
795template <
typename Derived,
typename ValueType,
typename SizeType>
796void GenericDeque<Derived, ValueType, SizeType>::ShiftLeft(
797 const size_type insert_index, size_type new_items) {
798 Base::PushFront(new_items);
799 iterator original_begin(derived(), new_items);
801 const size_type move_to_new_slots = std::min(new_items, insert_index);
802 auto [next_src, next_dst] =
803 std::uninitialized_move_n(original_begin, move_to_new_slots, begin());
805 const size_type move_to_existing_slots = insert_index - move_to_new_slots;
806 std::move(next_src, next_src + move_to_existing_slots, next_dst);
811 if constexpr (!std::is_trivially_destructible_v<value_type>) {
813 iterator(derived(), std::max(insert_index, original_begin.pos_)),
814 iterator(derived(), insert_index + new_items));
818template <
typename Derived,
typename ValueType,
typename SizeType>
819void GenericDeque<Derived, ValueType, SizeType>::ShiftRight(
820 const size_type insert_index, size_type new_items) {
821 const size_type items_after = size() - insert_index;
822 iterator original_end = end();
823 Base::PushBack(new_items);
825 const size_type move_to_new_slots = std::min(new_items, items_after);
826 std::uninitialized_move(original_end - move_to_new_slots,
828 end() - move_to_new_slots);
830 const size_type move_to_existing_slots = items_after - move_to_new_slots;
831 iterator pos(derived(), insert_index);
832 std::move_backward(pos, pos + move_to_existing_slots, original_end);
837 if constexpr (!std::is_trivially_destructible_v<value_type>) {
839 iterator(derived(), insert_index),
841 std::min(
static_cast<size_type
>(insert_index + new_items),
842 original_end.pos_)));
846template <
typename Derived,
typename ValueType,
typename SizeType>
847std::optional<typename GenericDeque<Derived, ValueType, SizeType>::iterator>
849 const_iterator pos, size_type count,
const value_type& value) {
851 return iterator(derived(), pos.pos_);
853 if (!ShiftForInsert(pos.pos_, count)) {
857 iterator it(derived(), pos.pos_);
858 std::uninitialized_fill_n(it, count, value);
862template <
typename Derived,
typename ValueType,
typename SizeType>
863template <
typename InputIt,
typename>
864typename GenericDeque<Derived, ValueType, SizeType>::iterator
869 if constexpr (std::is_same_v<
870 typename std::iterator_traits<InputIt>::iterator_category,
871 std::input_iterator_tag>) {
874 iterator insert_pos = iterator(derived(), pos.pos_);
875 while (first != last) {
876 insert_pos = emplace(insert_pos, *first);
880 return iterator(derived(), pos.pos_);
882 std::optional<iterator> result = try_insert(pos, first, last);
883 PW_ASSERT(result.has_value());
888template <
typename Derived,
typename ValueType,
typename SizeType>
889template <
typename ForwardIt,
typename>
890std::optional<typename GenericDeque<Derived, ValueType, SizeType>::iterator>
894 static_assert(std::is_convertible_v<
895 typename std::iterator_traits<ForwardIt>::iterator_category,
896 std::forward_iterator_tag>);
897 const auto distance = std::distance(first, last);
898 PW_DASSERT(distance >= 0);
899 const size_type count =
static_cast<size_type
>(distance);
901 const iterator it(derived(), pos.pos_);
905 if (!ShiftForInsert(pos.pos_, count)) {
908 std::uninitialized_move(first, last, it);
912template <
typename Derived,
typename ValueType,
typename SizeType>
913template <
typename... Args>
914std::optional<typename GenericDeque<Derived, ValueType, SizeType>::iterator>
915GenericDeque<Derived, ValueType, SizeType>::try_emplace_shift_right(
916 const_iterator pos, Args&&... args) {
917 if (!CheckCapacityAdd(1)) {
920 ShiftRight(pos.pos_, 1);
921 iterator it(derived(), pos.pos_);
922 new (std::addressof(*it)) value_type(std::forward<Args>(args)...);
926template <
typename Derived,
typename ValueType,
typename SizeType>
927std::optional<typename GenericDeque<Derived, ValueType, SizeType>::iterator>
928GenericDeque<Derived, ValueType, SizeType>::try_insert_shift_right(
929 const_iterator pos, size_type count,
const value_type& value) {
931 return iterator(derived(), pos.pos_);
933 if (!CheckCapacityAdd(count)) {
937 ShiftRight(pos.pos_, count);
938 iterator it(derived(), pos.pos_);
939 std::uninitialized_fill_n(it, count, value);
943template <
typename Derived,
typename ValueType,
typename SizeType>
944template <
typename ForwardIt,
typename>
945std::optional<typename GenericDeque<Derived, ValueType, SizeType>::iterator>
946GenericDeque<Derived, ValueType, SizeType>::try_insert_shift_right(
947 const_iterator pos, ForwardIt first, ForwardIt last) {
948 static_assert(std::is_convertible_v<
949 typename std::iterator_traits<ForwardIt>::iterator_category,
950 std::forward_iterator_tag>);
951 const auto distance = std::distance(first, last);
952 PW_DASSERT(distance >= 0);
953 const size_type count =
static_cast<size_type
>(distance);
955 const iterator it(derived(), pos.pos_);
959 if (!CheckCapacityAdd(count)) {
962 ShiftRight(pos.pos_, count);
963 std::uninitialized_move(first, last, it);
Definition: generic_deque.h:54
Definition: generic_deque.h:178
constexpr size_type capacity() const noexcept
Returns the maximum number of elements in the deque.
Definition: generic_deque.h:72
constexpr size_type size() const noexcept
Returns the number of elements in the deque.
Definition: generic_deque.h:69
Definition: span_impl.h:235
iterator insert(const_iterator pos, const value_type &value)
Definition: generic_deque.h:351
iterator insert(const_iterator pos, size_type count, const value_type &value)
Definition: generic_deque.h:371
iterator insert(const_iterator pos, InputIt first, InputIt last)
Definition: generic_deque.h:865
std::optional< iterator > try_insert(const_iterator pos, const value_type &value)
Definition: generic_deque.h:469
bool try_assign(size_type count, const value_type &value)
Definition: generic_deque.h:618
constexpr std::pair< span< const value_type >, span< const value_type > > contiguous_data() const
Provides access to the valid data in a contiguous form.
Definition: generic_deque.h:656
std::optional< iterator > try_emplace(const_iterator pos, Args &&... args)
iterator erase(const_iterator pos)
Erases the item at pos, which must be a dereferenceable iterator.
Definition: generic_deque.h:301
std::optional< iterator > try_insert(const_iterator pos, std::initializer_list< value_type > ilist)
Definition: generic_deque.h:509
iterator insert(const_iterator pos, value_type &&value)
Definition: generic_deque.h:361
void assign(It start, It finish)
Definition: generic_deque.h:601
void assign(size_type count, const value_type &value)
Sets the contents to count copies of value. Crashes if cannot fit.
Definition: generic_deque.h:209
iterator insert(const_iterator pos, std::initializer_list< value_type > ilist)
Definition: generic_deque.h:390
constexpr size_type capacity() const noexcept
Returns the maximum number of elements in the deque.
Definition: generic_deque.h:72
constexpr size_type size() const noexcept
Returns the number of elements in the deque.
Definition: generic_deque.h:69
bool try_assign(const std::initializer_list< value_type > &list)
Definition: generic_deque.h:449
std::optional< iterator > try_insert(const_iterator pos, value_type &&value)
Definition: generic_deque.h:479
void assign(const std::initializer_list< value_type > &list)
Sets contents to copies of the items from the list. Crashes if cannot fit.
Definition: generic_deque.h:221
iterator emplace(const_iterator pos, Args &&... args)
Definition: generic_deque.h:341
std::optional< iterator > try_insert(const_iterator pos, ForwardIt first, ForwardIt last)
iterator erase(const_iterator first, const_iterator last)
Definition: generic_deque.h:736
constexpr bool CheckedAdd(A a, B b, T &result)
Definition: checked_arithmetic.h:70