BAN: Add cool API for LinkedList

You can now move elements between LinkedLists without allocations or
deallocations. Same node moves from source to destination
This commit is contained in:
Bananymous 2024-02-01 14:19:02 +02:00
parent 9213dd13bc
commit c4bf1641bd
1 changed files with 51 additions and 25 deletions

View File

@ -42,6 +42,8 @@ namespace BAN
iterator remove(iterator); iterator remove(iterator);
void clear(); void clear();
iterator move_element_to_other_linked_list(LinkedList& dest_list, iterator dest_iter, iterator src_iter);
iterator begin() { return iterator(m_data, empty()); } iterator begin() { return iterator(m_data, empty()); }
const_iterator begin() const { return const_iterator(m_data, empty()); } const_iterator begin() const { return const_iterator(m_data, empty()); }
iterator end() { return iterator(m_last, true); } iterator end() { return iterator(m_last, true); }
@ -65,7 +67,11 @@ namespace BAN
Node* prev; Node* prev;
}; };
ErrorOr<Node*> allocate_node() const; template<typename... Args>
ErrorOr<Node*> allocate_node(Args&&...) const;
Node* remove_node(iterator);
void insert_node(iterator, Node*);
Node* m_data = nullptr; Node* m_data = nullptr;
Node* m_last = nullptr; Node* m_last = nullptr;
@ -137,6 +143,31 @@ namespace BAN
return *this; return *this;
} }
template<typename T>
LinkedList<T>::Node* LinkedList<T>::remove_node(iterator iter)
{
ASSERT(!empty() && iter);
Node* node = iter.m_current;
Node* prev = node->prev;
Node* next = node->next;
(prev ? prev->next : m_data) = next;
(next ? next->prev : m_last) = prev;
m_size--;
return node;
}
template<typename T>
void LinkedList<T>::insert_node(iterator iter, Node* node)
{
Node* next = iter.m_past_end ? nullptr : iter.m_current;
Node* prev = next ? next->prev : m_last;
node->next = next;
node->prev = prev;
(prev ? prev->next : m_data) = node;
(next ? next->prev : m_last) = node;
m_size++;
}
template<typename T> template<typename T>
ErrorOr<void> LinkedList<T>::push_back(const T& value) ErrorOr<void> LinkedList<T>::push_back(const T& value)
{ {
@ -158,15 +189,8 @@ namespace BAN
template<typename T> template<typename T>
ErrorOr<void> LinkedList<T>::insert(iterator iter, T&& value) ErrorOr<void> LinkedList<T>::insert(iterator iter, T&& value)
{ {
Node* next = iter.m_past_end ? nullptr : iter.m_current; Node* new_node = TRY(allocate_node(move(value)));
Node* prev = next ? next->prev : m_last; insert_node(iter, new_node);
Node* new_node = TRY(allocate_node());
new (&new_node->value) T(move(value));
new_node->next = next;
new_node->prev = prev;
(prev ? prev->next : m_data) = new_node;
(next ? next->prev : m_last) = new_node;
m_size++;
return {}; return {};
} }
@ -181,21 +205,15 @@ namespace BAN
template<typename... Args> template<typename... Args>
ErrorOr<void> LinkedList<T>::emplace(iterator iter, Args&&... args) ErrorOr<void> LinkedList<T>::emplace(iterator iter, Args&&... args)
{ {
Node* next = iter.m_past_end ? nullptr : iter.m_current; Node* new_node = TRY(allocate_node(forward<Args>(args)...));
Node* prev = next ? next->prev : m_last; insert_node(iter, new_node);
Node* new_node = TRY(allocate_node());
new (&new_node->value) T(forward<Args>(args)...);
new_node->next = next;
new_node->prev = prev;
(prev ? prev->next : m_data) = new_node;
(next ? next->prev : m_last) = new_node;
m_size++;
return {}; return {};
} }
template<typename T> template<typename T>
void LinkedList<T>::pop_back() void LinkedList<T>::pop_back()
{ {
ASSERT(!empty());
remove(iterator(m_last, false)); remove(iterator(m_last, false));
} }
@ -203,14 +221,10 @@ namespace BAN
LinkedList<T>::iterator LinkedList<T>::remove(iterator iter) LinkedList<T>::iterator LinkedList<T>::remove(iterator iter)
{ {
ASSERT(!empty() && iter); ASSERT(!empty() && iter);
Node* node = iter.m_current; Node* node = remove_node(iter);
Node* prev = node->prev;
Node* next = node->next; Node* next = node->next;
node->value.~T(); node->value.~T();
BAN::deallocator(node); BAN::deallocator(node);
(prev ? prev->next : m_data) = next;
(next ? next->prev : m_last) = prev;
m_size--;
return next ? iterator(next, false) : iterator(m_last, true); return next ? iterator(next, false) : iterator(m_last, true);
} }
@ -230,6 +244,16 @@ namespace BAN
m_size = 0; m_size = 0;
} }
template<typename T>
LinkedList<T>::iterator LinkedList<T>::move_element_to_other_linked_list(LinkedList& dest_list, iterator dest_iter, iterator src_iter)
{
ASSERT(!empty() && src_iter);
Node* node = remove_node(src_iter);
iterator ret = node->next ? iterator(node->next, false) : iterator(m_last, true);
dest_list.insert_node(dest_iter, node);
return ret;
}
template<typename T> template<typename T>
const T& LinkedList<T>::back() const const T& LinkedList<T>::back() const
{ {
@ -284,11 +308,13 @@ namespace BAN
} }
template<typename T> template<typename T>
ErrorOr<typename LinkedList<T>::Node*> LinkedList<T>::allocate_node() const template<typename... Args>
ErrorOr<typename LinkedList<T>::Node*> LinkedList<T>::allocate_node(Args&&... args) const
{ {
Node* node = (Node*)BAN::allocator(sizeof(Node)); Node* node = (Node*)BAN::allocator(sizeof(Node));
if (node == nullptr) if (node == nullptr)
return Error::from_errno(ENOMEM); return Error::from_errno(ENOMEM);
new (&node->value) T(forward<Args>(args)...);
return node; return node;
} }