ibrcommon/ibrcommon/thread/Timer.cpp

Go to the documentation of this file.

/*
 * Timer.cpp
 *
 *  Created on: 29.09.2009
 *      Author: morgenro
 */

#include "ibrcommon/config.h"
#include "ibrcommon/thread/Timer.h"
#include "ibrcommon/thread/MutexLock.h"

#include <iostream>
#include <queue>
#include <utility>
#include <sstream>

namespace ibrcommon
{
        size_t Timer::get_current_time()
        {
                time_t rawtime = time(NULL);
                tm * ptm;

                ptm = gmtime ( &rawtime );

                return mktime(ptm);
        }

        Timer::Timer(TimerCallback &callback, size_t timeout)
         : _state(TIMER_UNSET), _callback(callback), _timeout(timeout * 1000)
        {
        }

        Timer::~Timer()
        {
                stop();
        }

        void Timer::stop()
        {
                {
                        MutexLock l(_state);
                        _state.setState(TIMER_STOPPED);
                }
                join();
        }

        void Timer::set(size_t timeout)
        {
                MutexLock l(_state);
                _timeout = timeout * 1000;
                _state.setState(TIMER_RESET);
        }

        void Timer::reset()
        {
                MutexLock l(_state);
                _state.setState(TIMER_RESET);
        }

        void Timer::run()
        {
                MutexLock l(_state);
                _state.setState(TIMER_RUNNING);

                while (_state.ifState(TIMER_RUNNING))
                {
                        if (_state.wait(_timeout))
                        {
                                // got signal, check for reset request
                                if (_state.ifState(TIMER_RESET))
                                {
                                        _state.setState(TIMER_RUNNING);
                                }
                                else
                                {
                                        // stop the timer
                                        _state.setState(TIMER_STOPPED);
                                }
                        }
                        else
                        {
                                if (_state.ifState(TIMER_STOPPED)) return;

                                // timeout exceeded, call callback method
                                if (_callback.timeout(this))
                                {
                                        // reset the timer
                                        _state.setState(TIMER_RUNNING);
                                }
                                else
                                {
                                        // stop the timer
                                        _state.setState(TIMER_STOPPED);
                                }
                        }
                }
        }

        SimpleTimer::SimpleTimer(SimpleTimerCallback &callback, size_t identifier)
         : _callback(callback), _identifier(identifier)
        {
        }

        SimpleTimer::~SimpleTimer()
        {
                remove();
        }

        void SimpleTimer::set(size_t timeout)
        {
                ibrcommon::MutexLock l(_timecond);
                _countdown = timeout;
        }

        void SimpleTimer::remove()
        {
                {
                        ibrcommon::MutexLock l(_timecond);
                        _running = false;
                        _timecond.signal(true);
                }
                join();
        }

        void SimpleTimer::run()
        {
                ibrcommon::MutexLock l(_timecond);
                _running = true;

                while (_running && _countdown > 0)
                {
                        if (_timecond.wait(1000))
                        {
                                // got stop signal
                                _countdown = 0;
                                _running = false;
                                return;
                        }
                        else
                        {
                                _countdown--;

                                if (_countdown == 0)
                                {
                                        _countdown = _callback.timeout(_identifier);
                                }
                        }
                }

        }

        MultiTimer::MultiTimer()
         : _running(true)
        {
        }

        MultiTimer::~MultiTimer()
        {
                {
                        ibrcommon::MutexLock l(_timecond);
                        _running = false;
                        _timecond.signal(true);
                }
                join();
        }

        void MultiTimer::set(SimpleTimerCallback &callback, size_t identifier, size_t timeout)
        {
                size_t tt = Timer::get_current_time() + timeout;
                Eggtimer eggtimer(callback, identifier, tt);

//              std::cout << "set timer " << eggtimer.toString() << std::endl;

                ibrcommon::MutexLock l(_listlock);

                for (std::multiset<Eggtimer>::iterator iter = _timers.begin(); iter != _timers.end(); iter++)
                {
                        if ((*iter)._identifier == identifier)
                        {
                                if (&((*iter)._callback) == &callback)
                                {
                                        _timers.erase(iter++);
                                }
                        }
                }

                _timers.insert( eggtimer );
        }

        void MultiTimer::remove(SimpleTimerCallback &callback, size_t identifier)
        {
//              Eggtimer eggtimer(callback, identifier, 0);
//              std::cout << "remove timer " << eggtimer.toString() << std::endl;

                ibrcommon::MutexLock l(_listlock);

                for (std::multiset<Eggtimer>::iterator iter = _timers.begin(); iter != _timers.end(); iter++)
                {
                        if ((*iter)._identifier == identifier)
                        {
                                if (&((*iter)._callback) == &callback)
                                {
                                        _timers.erase(iter++);
                                }
                        }
                }
        }

        void MultiTimer::removeAll()
        {
                {
                        ibrcommon::MutexLock ll(_listlock);
                        _timers.clear();
                }

                {
                        ibrcommon::MutexLock l(_timecond);
                        _running = false;
                        _timecond.signal();
                }
                join();
        }

        void MultiTimer::run()
        {
                ibrcommon::MutexLock l(_timecond);

                while (_running)
                {
                        std::queue<Eggtimer> _timeouts;

                        if (_timecond.wait(100))
                        {
                                // got stop signal
                                _running = false;
                                return;
                        }
                        else
                        {
                                ibrcommon::MutexLock l(_listlock);
                                size_t current_time = Timer::get_current_time();

//                              std::cout << "List of all timers" << std::endl;
//                              for (std::multiset<Eggtimer>::const_iterator iter = _timers.begin(); iter != _timers.end(); iter++)
//                              {
//                                      const Eggtimer &t = (*iter);
//                                      std::cout << t.toString() << std::endl;
//                              }

                                for (std::multiset<Eggtimer>::const_iterator iter = _timers.begin(); iter != _timers.end(); iter++)
                                {
                                        const Eggtimer &t = (*iter);

                                         if (t < current_time)
                                         {
                                                _timeouts.push(t);
                                                _timers.erase(iter++);
                                         }
                                         else
                                         {
                                                 break;
                                         }
                                }
                        }

                        while (!_timeouts.empty())
                        {
                                Eggtimer &t = _timeouts.front();
                                size_t newvalue = t._callback.timeout(t._identifier);
                                if (newvalue > 0)
                                {
                                        set(t._callback, t._identifier, newvalue);
                                }
                                _timeouts.pop();
                        }
                }
        }

        MultiTimer::Eggtimer::Eggtimer(SimpleTimerCallback &callback, size_t identifier, size_t timeout)
         : _callback(callback), _timeout(timeout), _identifier(identifier)
        {

        }

        MultiTimer::Eggtimer::~Eggtimer()
        {

        }

        bool MultiTimer::Eggtimer::operator>(const size_t &obj) const
        {
                return (_timeout > obj);
        }

        bool MultiTimer::Eggtimer::operator<(const size_t &obj) const
        {
                return (_timeout < obj);
        }

        bool MultiTimer::Eggtimer::operator>(const MultiTimer::Eggtimer &obj) const
        {
                return (_timeout > obj._timeout);
        }

        bool MultiTimer::Eggtimer::operator<(const MultiTimer::Eggtimer &obj) const
        {
                return (_timeout < obj._timeout);
        }

        bool MultiTimer::Eggtimer::operator==(const MultiTimer::Eggtimer &obj) const
        {
                return (_timeout == obj._timeout);
        }

        const std::string MultiTimer::Eggtimer::toString() const
        {
                std::stringstream ss;
                size_t current_time = Timer::get_current_time();

                ss << "Eggtimer: " << (_timeout - current_time) << " seconds, identifier: " << _identifier;

                return ss.str();
        }
}

---