include/timer.hpp

   1 #ifndef TIMER_HPP
   2 #define TIMER_HPP
   3
   4 #include <functional>
   5 #include <inttypes.h>
   6
   7 #include "marshal.hpp"
   8
   9 #define USEC_PER_SEC 1000000
  10
  11 enum TimeUnit
  12 {
  13     SEC = 0,
  14     MSEC,
  15     USEC,
  16     NSEC,
  17     PC_FREQ0,  /* 1.193182Mhz */
  18     PC_FREQ1,  /* 32.768 Khz */
  19 };
  20
  21 class Timer
  22 {
  23 public:
  24     Timer(void);
  25
  26     void set(std::function<void (void)> func);
  27
  28     template <typename T>
  29     void set(T *obj, void (T::*method)(void)) {
  30         this->func = std::bind(method, obj);
  31     }
  32
  33     void update(uint64_t next_deadline, TimeUnit unit);
  34     void cancel(void);
  35
  36     uint64_t get_deadline(void);
  37     void fire(void);
  38
  39     void marshal(Marshaller *m, const char *name);
  40
  41 private:
  42     uint64_t deadline;
  43     std::function<void (void)> func;
  44 };
  45
  46 class TimeSource
  47 {
  48 public:
  49     static TimeSource *get_instance(void);
  50
  51     virtual uint64_t now(TimeUnit unit) = 0;
  52     virtual void arm(Timer *timer) = 0;
  53     virtual void cancel(Timer *timer) = 0;
  54 };
  55
  56 static inline uint64_t now(TimeUnit unit)
  57 {
  58     return TimeSource::get_instance()->now(unit);
  59 }
  60
  61 uint64_t time_to_ns(uint64_t when, TimeUnit unit);
  62 uint64_t ns_to_time(uint64_t when, TimeUnit unit);
  63
  64 void register_time_source(TimeSource *ts);
  65
  66 #endif