1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79
|
// SPDX-License-Identifier: BSD-2-Clause
/* Copyright (C) 2015 - 2021 Intel Corporation. */
#pragma once
#include <assert.h>
#include <pthread.h>
#include "Runnable.hpp"
class Thread
{
public:
Thread(Runnable *runnable) : runnable_task(runnable)
{}
void start()
{
int err = pthread_create(&thread_handle, NULL, execute_thread,
static_cast<void *>(runnable_task));
assert(!err);
};
void wait()
{
pthread_join(thread_handle, NULL);
};
Runnable *get_runnable_task()
{
return runnable_task;
}
private:
static void *execute_thread(void *ptr)
{
Runnable *runnable = static_cast<Runnable *>(ptr);
assert(runnable);
runnable->run();
pthread_exit(NULL);
}
pthread_t thread_handle;
Runnable *runnable_task;
};
class ThreadsManager
{
public:
ThreadsManager(std::vector<Thread *> &threads_vec) : threads(threads_vec)
{}
~ThreadsManager()
{
release();
}
void start()
{
for (int i = 0; i < threads.size(); i++) {
threads[i]->start();
}
}
void barrier()
{
for (int i = 0; i < threads.size(); i++) {
threads[i]->wait();
}
}
void release()
{
for (int i = 0; i < threads.size(); i++) {
delete threads[i];
}
threads.clear();
}
private:
std::vector<Thread *> &threads;
};
|