SimpleTimer.h

Go to the documentation of this file.

#pragma once
 
#include <boost/function.hpp>
#include <boost/system/error_code.hpp>
#ifdef _WIN32
#include <windows.h>
#else
// cannot get this to work on windows
#include <boost/asio.hpp>
#endif
 
class CallbackInfo;
 
struct TimerCallback
{
    virtual void timer_callback() = 0;
};
 
typedef void(*TimerCallbackInfo)(CallbackInfo *);
 
class SimpleTimer
{
public:
    typedef boost::function<void()> Handler;
    typedef boost::function<void(CallbackInfo *info)> HandlerCBI;
 
#ifdef _WIN32
    SimpleTimer(const std::size_t& tid = 10, const std::size_t& interval = 500, Handler _handler = NULL) 
        : _tid(tid), _interval(interval), _tickCount(0)
    {
        _callbackInfo = NULL;
        _caller = NULL;
        _hTimer = NULL;
        _isStopped = true;
        _tid = 0;
        _interval = 0;
        _tickCount = 0;
    };
#else
    SimpleTimer(boost::asio::io_service& io_service,
        const std::size_t& tid = 10,
        const std::size_t& interval = 500,
        Handler _handler = NULL)
        : _strand(io_service),
        _timer(io_service),
        _tid(tid),
        _interval(interval),
        _tickCount(0)//,
        // io_work(io_service_)
    {
        caller = NULL;
    };
#endif
    virtual ~SimpleTimer(void);
    void Start();
    void Stop();
 
    void SetInterval(long i);
    void SetHandler(Handler handler);
 
    void SetHandler(HandlerCBI hcb, void * c);
 
    void SetCaller(TimerCallback *caller)
    {
        _caller = caller;
    }
 
    void InitiateTimer();
 
    void TimerHandler();
 
private:
    void    Print           (const boost::system::error_code& error);
    void    HandleTimerEvent(const boost::system::error_code &error);
 
    Handler         _handler;
    HandlerCBI      _handlerCallbackInfo;
    CallbackInfo*   _callbackInfo;
 
    TimerCallback*  _caller;
 
    bool            _isStopped;
    std::size_t     _tid;
    std::size_t     _interval;
    std::size_t     _tickCount;
 
#ifdef _WIN32
    HANDLE          _hTimer;
#else
    boost::asio::strand _strand;
    // boost::asio::io_service::work& _io_work;
    boost::asio::deadline_timer _timer;
    // boost::posix_time::ptime _before;
    // boost::posix_time::ptime _after;
#endif
};