source: trash/InterruptableSleeper.cpp@ 5953

#include "InterruptableSleeper.h"

namespace spovnet {
namespace common {

InterruptableSleeper::InterruptableSleeper() {
        boost::mutex::scoped_lock scopedLock(m_mutex);

#ifdef WIN32
        m_handle = CreateEvent (NULL, TRUE, FALSE, NULL);
        assert (m_handle != NULL);
#else
        m_handle = new MUTEX_COND();
        int ret = 0;

        ret = pthread_mutex_init(&(m_handle->mutex), NULL);
        assert (ret == 0);

        ret = pthread_cond_init(&(m_handle->cond), NULL);
        assert (ret == 0);
#endif
}

InterruptableSleeper::~InterruptableSleeper() {
        boost::mutex::scoped_lock scopedLock(m_mutex);

#ifdef WIN32
        BOOL ret = CloseHandle (m_handle);
        assert (ret == TRUE);
#else
        int ret = 0;

        ret = pthread_cond_destroy(&(m_handle->cond));
        assert (ret == 0);

        ret = pthread_mutex_destroy(&(m_handle->mutex));
        assert (ret == 0);

        delete m_handle;
#endif
}

void InterruptableSleeper::interrupt() {
        boost::mutex::scoped_lock scopedLock(m_mutex);

#ifdef WIN32
        BOOL ret = SetEvent (m_handle);
        assert (ret == TRUE);
#else
        int ret = pthread_cond_signal(&(m_handle)->cond);
        assert (ret == 0);
#endif
}

void InterruptableSleeper::sleep(unsigned long millis) {
        //
        // NO mutex, otherwise the ::interrupt
        // function can not be called at all
        //

#ifdef WIN32

        DWORD dret = WaitForSingleObject (m_handle, millis);
        assert (dret != WAIT_FAILED);

#else

        int ret;
        struct timespec tm;
        struct timeb tp;

        ret = pthread_mutex_lock(&(m_handle->mutex));
        assert (ret == 0);

        long sec = millis / 1000;
        long millisec = millis % 1000;

        ftime(&tp);
        tp.time += sec;
        tp.millitm += millisec;

        if (tp.millitm > 999) {
                tp.millitm -= 1000;
                tp.time++;
        }

        tm.tv_sec = tp.time;
        tm.tv_nsec = tp.millitm * 1000000;

        ret = pthread_cond_timedwait(&(m_handle->cond), &(m_handle->mutex), &tm);
        assert (ret == 0 || ret == ETIMEDOUT);

        ret = pthread_mutex_unlock(&(m_handle->mutex));
        assert (ret == 0);

#endif

        //
        // under Windows we have to reset the event again
        //

#ifdef WIN32
        {
                boost::mutex::scoped_lock scopedLock (m_mutex);
                BOOL bret;
                bret = ResetEvent (m_handle);
                assert (bret == TRUE);
        }
#endif
}
}
} // namespace spovnet, common