Browse Source
- new intrusive linked list container (c++ template) that sorts on insertion - primarily for convenience inspecting things in the system like uORB or WorkQueues - uorb status or top sorted alphabetically - work_queue status threads sorted by priority, then items sorted alphabetically within eachsbg
11 changed files with 502 additions and 13 deletions
@ -0,0 +1,187 @@ |
|||||||
|
/****************************************************************************
|
||||||
|
* |
||||||
|
* Copyright (C) 2020 PX4 Development Team. All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions |
||||||
|
* are met: |
||||||
|
* |
||||||
|
* 1. Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* 2. Redistributions in binary form must reproduce the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer in |
||||||
|
* the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* 3. Neither the name PX4 nor the names of its contributors may be |
||||||
|
* used to endorse or promote products derived from this software |
||||||
|
* without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS |
||||||
|
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE |
||||||
|
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, |
||||||
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, |
||||||
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS |
||||||
|
* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED |
||||||
|
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
||||||
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN |
||||||
|
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
||||||
|
* POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
****************************************************************************/ |
||||||
|
|
||||||
|
/**
|
||||||
|
* @file IntrusiveSortedList.hpp |
||||||
|
* |
||||||
|
* An intrusive linked list where nodes are sorted on insertion. |
||||||
|
*/ |
||||||
|
|
||||||
|
#pragma once |
||||||
|
|
||||||
|
#include <stdlib.h> |
||||||
|
|
||||||
|
template<class T> |
||||||
|
class IntrusiveSortedListNode |
||||||
|
{ |
||||||
|
public: |
||||||
|
void setSortedSibling(T sibling) { _sorted_list_node_sibling = sibling; } |
||||||
|
const T getSortedSibling() const { return _sorted_list_node_sibling; } |
||||||
|
protected: |
||||||
|
T _sorted_list_node_sibling{nullptr}; |
||||||
|
}; |
||||||
|
|
||||||
|
template<class T> |
||||||
|
class IntrusiveSortedList |
||||||
|
{ |
||||||
|
public: |
||||||
|
|
||||||
|
void add(T newNode) |
||||||
|
{ |
||||||
|
if (_head == nullptr) { |
||||||
|
// list is empty, add as head
|
||||||
|
_head = newNode; |
||||||
|
return; |
||||||
|
|
||||||
|
} else { |
||||||
|
if (*newNode <= *_head) { |
||||||
|
newNode->setSortedSibling(_head); |
||||||
|
_head = newNode; |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// find last node and add to end
|
||||||
|
T node = _head; |
||||||
|
|
||||||
|
while (node != nullptr && node->getSortedSibling() != nullptr) { |
||||||
|
|
||||||
|
if (*newNode <= *node->getSortedSibling()) { |
||||||
|
// insert newNode
|
||||||
|
newNode->setSortedSibling(node->getSortedSibling()); |
||||||
|
node->setSortedSibling(newNode); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
node = node->getSortedSibling(); |
||||||
|
} |
||||||
|
|
||||||
|
// reached the end, add
|
||||||
|
node->setSortedSibling(newNode); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
bool remove(T removeNode) |
||||||
|
{ |
||||||
|
if (removeNode == nullptr) { |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
// base case
|
||||||
|
if (removeNode == _head) { |
||||||
|
if (_head != nullptr) { |
||||||
|
_head = _head->getSortedSibling(); |
||||||
|
} |
||||||
|
|
||||||
|
removeNode->setSortedSibling(nullptr); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
for (T node = _head; node != nullptr; node = node->getSortedSibling()) { |
||||||
|
// is sibling the node to remove?
|
||||||
|
if (node->getSortedSibling() == removeNode) { |
||||||
|
// replace sibling
|
||||||
|
if (node->getSortedSibling() != nullptr) { |
||||||
|
node->setSortedSibling(node->getSortedSibling()->getSortedSibling()); |
||||||
|
|
||||||
|
} else { |
||||||
|
node->setSortedSibling(nullptr); |
||||||
|
} |
||||||
|
|
||||||
|
removeNode->setSortedSibling(nullptr); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
struct Iterator { |
||||||
|
T node; |
||||||
|
explicit Iterator(T v) : node(v) {} |
||||||
|
|
||||||
|
operator T() const { return node; } |
||||||
|
operator T &() { return node; } |
||||||
|
T operator* () const { return node; } |
||||||
|
Iterator &operator++ () |
||||||
|
{ |
||||||
|
if (node) { |
||||||
|
node = node->getSortedSibling(); |
||||||
|
}; |
||||||
|
|
||||||
|
return *this; |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
Iterator begin() { return Iterator(_head); } |
||||||
|
Iterator end() { return Iterator(nullptr); } |
||||||
|
|
||||||
|
bool empty() const { return _head == nullptr; } |
||||||
|
|
||||||
|
size_t size() const |
||||||
|
{ |
||||||
|
size_t sz = 0; |
||||||
|
|
||||||
|
for (T node = _head; node != nullptr; node = node->getSortedSibling()) { |
||||||
|
sz++; |
||||||
|
} |
||||||
|
|
||||||
|
return sz; |
||||||
|
} |
||||||
|
|
||||||
|
void deleteNode(T node) |
||||||
|
{ |
||||||
|
if (remove(node)) { |
||||||
|
// only delete if node was successfully removed
|
||||||
|
delete node; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void clear() |
||||||
|
{ |
||||||
|
T node = _head; |
||||||
|
|
||||||
|
while (node != nullptr) { |
||||||
|
T next = node->getSortedSibling(); |
||||||
|
delete node; |
||||||
|
node = next; |
||||||
|
} |
||||||
|
|
||||||
|
_head = nullptr; |
||||||
|
} |
||||||
|
|
||||||
|
protected: |
||||||
|
|
||||||
|
T _head{nullptr}; |
||||||
|
}; |
@ -0,0 +1,285 @@ |
|||||||
|
/****************************************************************************
|
||||||
|
* |
||||||
|
* Copyright (C) 2020 PX4 Development Team. All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions |
||||||
|
* are met: |
||||||
|
* |
||||||
|
* 1. Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* 2. Redistributions in binary form must reproduce the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer in |
||||||
|
* the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* 3. Neither the name PX4 nor the names of its contributors may be |
||||||
|
* used to endorse or promote products derived from this software |
||||||
|
* without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS |
||||||
|
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE |
||||||
|
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, |
||||||
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, |
||||||
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS |
||||||
|
* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED |
||||||
|
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
||||||
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN |
||||||
|
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
||||||
|
* POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
****************************************************************************/ |
||||||
|
|
||||||
|
/**
|
||||||
|
* @file test_IntrusiveSortedList.cpp |
||||||
|
* Tests the IntrusiveSortedList container. |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <unit_test.h> |
||||||
|
#include <containers/IntrusiveSortedList.hpp> |
||||||
|
#include <float.h> |
||||||
|
#include <math.h> |
||||||
|
|
||||||
|
class testContainer : public IntrusiveSortedListNode<testContainer *> |
||||||
|
{ |
||||||
|
public: |
||||||
|
int i{0}; |
||||||
|
|
||||||
|
// sorted numerically
|
||||||
|
bool operator<=(const testContainer &rhs) const { return i <= rhs.i; } |
||||||
|
}; |
||||||
|
|
||||||
|
class IntrusiveSortedListTest : public UnitTest |
||||||
|
{ |
||||||
|
public: |
||||||
|
virtual bool run_tests(); |
||||||
|
|
||||||
|
bool test_add(); |
||||||
|
bool test_remove(); |
||||||
|
bool test_range_based_for(); |
||||||
|
bool test_reinsert(); |
||||||
|
|
||||||
|
}; |
||||||
|
|
||||||
|
bool IntrusiveSortedListTest::run_tests() |
||||||
|
{ |
||||||
|
ut_run_test(test_add); |
||||||
|
ut_run_test(test_remove); |
||||||
|
ut_run_test(test_range_based_for); |
||||||
|
ut_run_test(test_reinsert); |
||||||
|
|
||||||
|
return (_tests_failed == 0); |
||||||
|
} |
||||||
|
|
||||||
|
bool IntrusiveSortedListTest::test_add() |
||||||
|
{ |
||||||
|
IntrusiveSortedList<testContainer *> list1; |
||||||
|
|
||||||
|
// size should be 0 initially
|
||||||
|
ut_compare("size initially 0", list1.size(), 0); |
||||||
|
ut_assert_true(list1.empty()); |
||||||
|
|
||||||
|
// insert 100
|
||||||
|
for (int i = 0; i < 100; i++) { |
||||||
|
testContainer *t = new testContainer(); |
||||||
|
t->i = i; |
||||||
|
list1.add(t); |
||||||
|
|
||||||
|
ut_compare("size increasing with i", list1.size(), i + 1); |
||||||
|
ut_assert_true(!list1.empty()); |
||||||
|
} |
||||||
|
|
||||||
|
// verify full size (100)
|
||||||
|
ut_assert_true(list1.size() == 100); |
||||||
|
|
||||||
|
int i = 0; |
||||||
|
|
||||||
|
for (auto t : list1) { |
||||||
|
// verify all elements were inserted in order
|
||||||
|
ut_compare("stored i", i, t->i); |
||||||
|
i++; |
||||||
|
} |
||||||
|
|
||||||
|
// delete all elements
|
||||||
|
list1.clear(); |
||||||
|
|
||||||
|
// verify list has been cleared
|
||||||
|
ut_assert_true(list1.empty()); |
||||||
|
ut_compare("size 0", list1.size(), 0); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
bool IntrusiveSortedListTest::test_remove() |
||||||
|
{ |
||||||
|
IntrusiveSortedList<testContainer *> list1; |
||||||
|
|
||||||
|
// size should be 0 initially
|
||||||
|
ut_compare("size initially 0", list1.size(), 0); |
||||||
|
ut_assert_true(list1.empty()); |
||||||
|
|
||||||
|
// insert 100
|
||||||
|
for (int i = 0; i < 100; i++) { |
||||||
|
testContainer *t = new testContainer(); |
||||||
|
t->i = i; |
||||||
|
list1.add(t); |
||||||
|
|
||||||
|
ut_compare("size increasing with i", list1.size(), i + 1); |
||||||
|
ut_assert_true(!list1.empty()); |
||||||
|
} |
||||||
|
|
||||||
|
// verify full size (100)
|
||||||
|
ut_assert_true(list1.size() == 100); |
||||||
|
|
||||||
|
// test removing elements
|
||||||
|
for (int remove_i = 0; remove_i < 100; remove_i++) { |
||||||
|
|
||||||
|
// find node with i == remove_i
|
||||||
|
testContainer *removed = nullptr; |
||||||
|
|
||||||
|
for (auto t : list1) { |
||||||
|
if (t->i == remove_i) { |
||||||
|
ut_assert_true(list1.remove(t)); |
||||||
|
removed = t; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
delete removed; |
||||||
|
|
||||||
|
// iterate list again to verify removal
|
||||||
|
for (auto t : list1) { |
||||||
|
ut_assert_true(t->i != remove_i); |
||||||
|
} |
||||||
|
|
||||||
|
ut_assert_true(list1.size() == 100 - remove_i - 1); |
||||||
|
} |
||||||
|
|
||||||
|
// list should now be empty
|
||||||
|
ut_assert_true(list1.empty()); |
||||||
|
ut_compare("size 0", list1.size(), 0); |
||||||
|
|
||||||
|
// delete all elements (should be safe on empty list)
|
||||||
|
list1.clear(); |
||||||
|
|
||||||
|
// verify list has been cleared
|
||||||
|
ut_assert_true(list1.empty()); |
||||||
|
ut_compare("size 0", list1.size(), 0); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
bool IntrusiveSortedListTest::test_range_based_for() |
||||||
|
{ |
||||||
|
IntrusiveSortedList<testContainer *> list1; |
||||||
|
|
||||||
|
// size should be 0 initially
|
||||||
|
ut_compare("size initially 0", list1.size(), 0); |
||||||
|
ut_assert_true(list1.empty()); |
||||||
|
|
||||||
|
// insert 100 elements in order
|
||||||
|
for (int i = 99; i >= 0; i--) { |
||||||
|
testContainer *t = new testContainer(); |
||||||
|
t->i = i; |
||||||
|
|
||||||
|
list1.add(t); |
||||||
|
|
||||||
|
ut_assert_true(!list1.empty()); |
||||||
|
} |
||||||
|
|
||||||
|
// verify all elements are in order
|
||||||
|
int i = 0; |
||||||
|
|
||||||
|
for (auto t1 : list1) { |
||||||
|
ut_compare("check count", i, t1->i); |
||||||
|
i++; |
||||||
|
} |
||||||
|
|
||||||
|
// verify full size (100)
|
||||||
|
ut_compare("size check", list1.size(), 100); |
||||||
|
|
||||||
|
// delete all elements
|
||||||
|
list1.clear(); |
||||||
|
|
||||||
|
// verify list has been cleared
|
||||||
|
ut_assert_true(list1.empty()); |
||||||
|
ut_compare("size check", list1.size(), 0); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
bool IntrusiveSortedListTest::test_reinsert() |
||||||
|
{ |
||||||
|
IntrusiveSortedList<testContainer *> l1; |
||||||
|
|
||||||
|
// size should be 0 initially
|
||||||
|
ut_compare("size initially 0", l1.size(), 0); |
||||||
|
ut_assert_true(l1.empty()); |
||||||
|
|
||||||
|
// insert 100
|
||||||
|
for (int i = 0; i < 100; i++) { |
||||||
|
testContainer *t = new testContainer(); |
||||||
|
t->i = i; |
||||||
|
l1.add(t); |
||||||
|
|
||||||
|
ut_compare("size increasing with i", l1.size(), i + 1); |
||||||
|
ut_assert_true(!l1.empty()); |
||||||
|
} |
||||||
|
|
||||||
|
// verify full size (100)
|
||||||
|
ut_assert_true(l1.size() == 100); |
||||||
|
ut_assert_false(l1.empty()); |
||||||
|
|
||||||
|
// test removing elements
|
||||||
|
for (int remove_i = 0; remove_i < 100; remove_i++) { |
||||||
|
|
||||||
|
ut_assert_false(l1.empty()); |
||||||
|
|
||||||
|
// find node with i == remove_i
|
||||||
|
testContainer *removed = nullptr; |
||||||
|
|
||||||
|
for (auto t : l1) { |
||||||
|
if (t->i == remove_i) { |
||||||
|
ut_assert_true(l1.remove(t)); |
||||||
|
removed = t; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// l1 shouldn't be empty until the very last iteration
|
||||||
|
ut_assert_false(l1.empty()); |
||||||
|
|
||||||
|
// iterate list again to verify removal
|
||||||
|
for (auto t : l1) { |
||||||
|
ut_assert_true(t->i != remove_i); |
||||||
|
} |
||||||
|
|
||||||
|
// size temporarily reduced by 1
|
||||||
|
ut_assert_true(l1.size() == 100 - 1); |
||||||
|
|
||||||
|
// now re-insert the removed node
|
||||||
|
const size_t sz1 = l1.size(); |
||||||
|
l1.add(removed); |
||||||
|
const size_t sz2 = l1.size(); |
||||||
|
|
||||||
|
// verify the size increase
|
||||||
|
ut_assert_true(sz2 > sz1); |
||||||
|
|
||||||
|
// size restored to 100
|
||||||
|
ut_assert_true(l1.size() == 100); |
||||||
|
} |
||||||
|
|
||||||
|
// queue shouldn't be empty
|
||||||
|
ut_assert_false(l1.empty()); |
||||||
|
ut_compare("size still 100", l1.size(), 100); |
||||||
|
|
||||||
|
// delete all elements
|
||||||
|
l1.clear(); |
||||||
|
|
||||||
|
// verify list has been cleared
|
||||||
|
ut_assert_true(l1.empty()); |
||||||
|
ut_compare("size 0", l1.size(), 0); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
ut_declare_test_c(test_IntrusiveSortedList, IntrusiveSortedListTest) |
Loading…
Reference in new issue