You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

125 lines
3.2 KiB
C++

/*
* Copyright 2013-2019 Software Radio Systems Limited
*
* This file is part of srsLTE.
*
* srsLTE is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of
* the License, or (at your option) any later version.
*
* srsLTE is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* A copy of the GNU Affero General Public License can be found in
* the LICENSE file in the top-level directory of this distribution
* and at http://www.gnu.org/licenses/.
*
*/
/******************************************************************************
* File: timeout.h
* Description: Millisecond resolution timeouts. Uses a dedicated thread to
* call an optional callback function upon timeout expiry.
* Reference:
*****************************************************************************/
#ifndef SRSLTE_TIMEOUT_H
#define SRSLTE_TIMEOUT_H
#include "srslte/srslte.h"
#include <pthread.h>
#include <stdint.h>
#include <sys/time.h>
#include <unistd.h>
namespace srslte {
class timeout_callback
{
public:
virtual void timeout_expired(uint32_t timeout_id) = 0;
};
class timeout
{
public:
timeout() : running(false), callback(NULL), thread(0), timeout_id(0) {}
~timeout()
{
if (running && callback)
pthread_join(thread, NULL);
}
void start(int duration_msec_, uint32_t timeout_id_ = 0, timeout_callback* callback_ = NULL)
{
if (duration_msec_ < 0)
return;
reset();
gettimeofday(&start_time[1], NULL);
duration_msec = duration_msec_;
running = true;
timeout_id = timeout_id_;
callback = callback_;
if (callback)
pthread_create(&thread, NULL, &thread_start, this);
}
void reset()
{
if (callback)
pthread_cancel(thread);
running = false;
}
static void* thread_start(void* t_)
{
timeout* t = (timeout*)t_;
t->thread_func();
return NULL;
}
void thread_func()
{
// substract time elapsed until now from timer duration
gettimeofday(&start_time[2], NULL);
get_time_interval(start_time);
int32_t usec = duration_msec * 1000 - start_time[0].tv_usec;
if (usec > 0)
usleep(usec);
if (callback && running)
callback->timeout_expired(timeout_id);
}
bool expired()
{
if (running) {
gettimeofday(&start_time[2], NULL);
get_time_interval(start_time);
return start_time[0].tv_usec > duration_msec * 1000;
} else {
return false;
}
}
int32_t get_msec_to_expire()
{
if (running) {
gettimeofday(&start_time[2], NULL);
get_time_interval(start_time);
return (duration_msec * 1000 - start_time[0].tv_usec) / 1000;
}
return 0;
}
bool is_running() { return running; }
private:
struct timeval start_time[3];
pthread_t thread;
uint32_t timeout_id;
timeout_callback* callback;
bool running;
int duration_msec;
};
} // namespace srslte
#endif // SRSLTE_TIMEOUT_H