lib-during/thread-pool-base.cpp

Go to the documentation of this file.

/********************************************************************
 * Copyright (C) 2012 by Verimag                                    *
 * Initial author: Matthieu Moy                                     *
 * This file must not be distributed outside Verimag                *
 ********************************************************************/

#include "thread-pool-base.h"
#include "utils/io-lock.h"
#include "task-manager.h"
#include "utils/cpu-affinity.h"

using namespace std;

atomic_variable<int> thread_pool_base::m_index(0);

void thread_pool_base::process_tasks(thread_pool_base *p,
                                     sc_core::sc_process_b *sc_process) {
#ifdef SC_DURING_USE_AFFINITY
        int affinity = m_index++ % boost::thread::hardware_concurrency();
        L_COUT << "// TEST-IGNORE: affinity = " << affinity << endl;
        cpu_bind(affinity);
#endif
        task_manager::get_instance()->alloc_current_task();
        while (true) {
                sync_task *r = p->pop_task(sc_process);
                task_manager::get_instance()->set_current_task(r);
                if (r == NULL) {
                        return;
                }
                //L_COUT << "// TEST-IGNORE: Processing next task" << boost::this_thread::get_id() << endl;
                r->process();
        }
}

thread_pool_base::thread_ptr
thread_pool_base::add_thread(sc_core::sc_process_b *sc_process) {
        thread_ptr t = new boost::thread(process_tasks, this, sc_process);
        m_thread_array.push_back(t);
        //L_COUT << "// TEST-IGNORE: add_thread(" << sc_process << ") = " << t->get_id() << endl;
        return t;
}

void thread_pool_base::stop_all_threads() {
        L_COUT << "// TEST-IGNORE: End of simulation, sending stop tasks" << endl;
        for (unsigned i = 0; i < m_thread_array.size(); i++)
                m_thread_array[i]->interrupt();
        L_COUT << "// TEST-IGNORE: All threads interrupted, now joining" << endl;
        for (unsigned i = 0; i < m_thread_array.size(); i++)
                m_thread_array[i]->join();
        L_COUT << "// TEST-IGNORE: all joins completed" << endl;
}

thread_pool_base::thread_pool_base() {
        // Create the instance before going multithread
        task_manager::get_instance();                                    
}