allow for pre-reservation of timers vector and priority queue

This commit is contained in:
Francisco Paisana 2019-09-17 11:29:03 +01:00
parent 50a1c9d557
commit 551192705e
2 changed files with 92 additions and 70 deletions

View File

@ -29,6 +29,7 @@
#ifndef SRSLTE_TIMERS_H #ifndef SRSLTE_TIMERS_H
#define SRSLTE_TIMERS_H #define SRSLTE_TIMERS_H
#include <algorithm>
#include <functional> #include <functional>
#include <limits> #include <limits>
#include <queue> #include <queue>
@ -284,6 +285,16 @@ public:
uint32_t timer_id; uint32_t timer_id;
}; };
explicit timers2(uint32_t capacity = 64)
{
timer_list.reserve(capacity);
// reserve a priority queue using a vector
std::vector<timer_run> v;
v.reserve(capacity);
std::priority_queue<timer_run> q(std::less<timer_run>(), std::move(v));
running_timers = std::move(q);
}
void step_all() void step_all()
{ {
cur_time++; cur_time++;
@ -325,6 +336,10 @@ public:
} }
uint32_t get_cur_time() const { return cur_time; } uint32_t get_cur_time() const { return cur_time; }
uint32_t nof_timers() const
{
return std::count_if(timer_list.begin(), timer_list.end(), [](const timer_impl& t) { return t.active; });
}
private: private:
struct timer_run { struct timer_run {

View File

@ -37,16 +37,19 @@ int timers2_test()
timers2 timers; timers2 timers;
uint32_t dur = 5; uint32_t dur = 5;
{
timers2::unique_timer t = timers.get_unique_timer(); timers2::unique_timer t = timers.get_unique_timer();
TESTASSERT(not t.is_running() and not t.is_expired()) TESTASSERT(not t.is_running() and not t.is_expired())
TESTASSERT(t.id() == 0) TESTASSERT(t.id() == 0)
timers2::unique_timer t2 = timers.get_unique_timer(); timers2::unique_timer t2 = timers.get_unique_timer();
TESTASSERT(not t2.is_running() and not t2.is_expired()) TESTASSERT(not t2.is_running() and not t2.is_expired())
TESTASSERT(t2.id() == 1) TESTASSERT(t2.id() == 1)
TESTASSERT(timers.nof_timers() == 2)
// TEST: Run multiple times with the same duration // TEST: Run multiple times with the same duration
bool callback_called = false; bool callback_called = false;
t.set(dur, [&callback_called](int) { callback_called = true; }); t.set(dur, [&callback_called](int) { callback_called = true; });
TESTASSERT(timers.get_cur_time() == 0)
for (uint32_t runs = 0; runs < 3; ++runs) { for (uint32_t runs = 0; runs < 3; ++runs) {
callback_called = false; callback_called = false;
TESTASSERT(not t.is_running()) TESTASSERT(not t.is_running())
@ -60,6 +63,7 @@ int timers2_test()
TESTASSERT(not t.is_running() and t.is_expired()) TESTASSERT(not t.is_running() and t.is_expired())
TESTASSERT(callback_called) TESTASSERT(callback_called)
} }
TESTASSERT(timers.get_cur_time() == 3 * (1 + dur))
// TEST: interrupt a timer. check if callback was called // TEST: interrupt a timer. check if callback was called
callback_called = false; callback_called = false;
@ -116,6 +120,9 @@ int timers2_test()
TESTASSERT(first_id == 1) TESTASSERT(first_id == 1)
printf("Last timer id=%d\n", last_id); printf("Last timer id=%d\n", last_id);
TESTASSERT(last_id == 2) TESTASSERT(last_id == 2)
}
// TEST: timer dtor is called and removes "timer" from "timers"
TESTASSERT(timers.nof_timers() == 0)
printf("Success\n"); printf("Success\n");