threading.cpp

Go to the documentation of this file.

/*
 * Copyright (C) <2012-2014> <EDF-R&D> <FRANCE>
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * This program 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 General Public License for more details.
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
 
#include <QThread>
 
#include "threading.h"
 
OSlaveThread::OSlaveThread(OThreadPool* pool) : _pool(pool)
{
    _bToEnd = false;
}
 
OSlaveThread::~OSlaveThread()
{
    this->terminate();
}
 
void OSlaveThread::run()
{
    while (isRunning() && !_bToEnd)
    {
        // Wait for available task.
        TY_LOCK_SHARED_MUTEX(_pool)
        if (isRunning() && !_bToEnd && !_pool->_tasks.empty())
        {
            // Dequeue next task.
            LPOTask task = _pool->_tasks.front();
            _pool->_tasks.pop();
 
            TY_UNLOCK_SHARED_MUTEX(_pool)
 
            // Signal all that task is running.
            TY_LOCK_SHARED_MUTEX(task)
            task->_running = true;
 
            // Run task.
            task->main();
 
            // Signal all that task is completed.
            task->_running = false;
            task->_completed = true;
            task->wakeAll();
            TY_UNLOCK_SHARED_MUTEX(task)
 
            // Increment pool counter
            TY_LOCK_SHARED_MUTEX(_pool)
            _pool->_counter++;
            TY_UNLOCK_SHARED_MUTEX(_pool)
        }
        else
            TY_UNLOCK_SHARED_MUTEX(_pool)
    }
}
 
OTask::OTask() : _running(false), _completed(false), _canceled(false) {}
 
OTask::~OTask()
{
    while (!isCompleted() && !isCanceled())
    {
        wait(this);
    }
}
 
bool OTask::isRunning() const
{
    TY_OMUTEXLOCKER_SHARED_MUTEX(this)
    return _running;
}
 
bool OTask::isCompleted() const
{
    TY_OMUTEXLOCKER_SHARED_MUTEX(this)
    return _completed;
}
 
bool OTask::isCanceled() const
{
    TY_OMUTEXLOCKER_SHARED_MUTEX(this)
    return _canceled;
}
 
void OTask::reset()
{
    TY_OMUTEXLOCKER_SHARED_MUTEX(this)
    _running = _completed = _canceled = false;
}
 
OThreadPool::OThreadPool(unsigned int slaves) : _totalCount(0), _counter(0)
{
    // Allocate slave threads.
    for (unsigned int i = 0; i < slaves; ++i)
    {
        OSlaveThread* thread = new OSlaveThread(this);
        push_back(thread);
    }
}
 
OThreadPool::~OThreadPool()
{
    TY_LOCK_SHARED_MUTEX(this)
 
    // Wait for queue to become empty.
    while (!_tasks.empty())
    {
        // Wait for last task in queue to at least start running.
        LPOTask task = _tasks.back();
        task->_completed = true;
        task->_canceled = true;
    }
 
    // Now terminate all threads.
    unsigned int i = 0;
    for (i = 0; i < size(); ++i)
    {
        (*this)[i]->_bToEnd = true;
        (*this)[i]->terminate();
    }
 
    // Signal them to wake up.
    TY_UNLOCK_SHARED_MUTEX(this)
 
    // Then delete them (the thread destructor will wait for thread completion).
    for (i = 0; i < size(); ++i)
    {
        delete (*this)[i];
    }
}
 
void OThreadPool::push(OTask* task)
{
    // Reset task flags.
    task->reset();
 
    // Push task onto queue and signal availability.
    TY_LOCK_SHARED_MUTEX(this)
    _tasks.push(task);
    TY_UNLOCK_SHARED_MUTEX(this)
}
 
unsigned int OThreadPool::getTotalCount() const
{
    TY_OMUTEXLOCKER_SHARED_MUTEX(this);
    return _totalCount;
}
 
unsigned int OThreadPool::getCount() const
{
    TY_OMUTEXLOCKER_SHARED_MUTEX(this);
    return _counter;
}
 
void OThreadPool::begin(unsigned int count)
{
    TY_LOCK_SHARED_MUTEX(this);
    _totalCount = count;
    _counter = 0;
    TY_UNLOCK_SHARED_MUTEX(this);
}
 
void OThreadPool::startPool()
{
 
    for (unsigned int i = 0; i < size(); ++i)
    {
        (*this)[i]->_bToEnd = false;
        (*this)[i]->start();
    }
}
 
bool OThreadPool::end()
{
    unsigned int totalCount = getTotalCount();
 
    unsigned int last = 0;
    while (last < totalCount)
    {
        unsigned int current = getCount();
        last = current;
        OSleeper::msleep(5);
    }
    stop();
    return true;
}
 
void OThreadPool::stop()
{
    TY_LOCK_SHARED_MUTEX(this);
 
    // For each task
    for (size_t i = 0; i < _tasks.size(); ++i)
    {
        // Dequeue next task
        LPOTask task = _tasks.front();
        _tasks.pop();
 
        // Cancel task
        TY_LOCK_SHARED_MUTEX(task)
        task->_canceled = true;
        TY_UNLOCK_SHARED_MUTEX(task)
 
        // Increment counter
        _counter++;
    }
 
    for (unsigned int i = 0; i < size(); ++i)
    {
        (*this)[i]->_bToEnd = true;
    }
 
    TY_UNLOCK_SHARED_MUTEX(this);
 
    // Waiting for thread termination
    while (getCount() < getTotalCount())
    {
        OSleeper::msleep(1);
    }
}