thread.c

Go to the documentation of this file.

 
#include <arch/thread.h>
 
#ifdef __POSIX
#include <sched.h>
#include <signal.h>
#include <unistd.h>
 
#ifdef __MACOS
#include <mach/thread_act.h>
 
int thread_affinity_set(thr_id_t thr, unsigned core)
{
        thread_affinity_policy_data_t policy = {core};
        thread_port_t mach_thread = pthread_mach_thread_np(thr);
        kern_return_t ret = thread_policy_set(mach_thread, THREAD_AFFINITY_POLICY,
                                              (thread_policy_t) &policy, 1);
        return -(ret != KERN_SUCCESS);
}
 
#else
 
int thread_affinity_set(thr_id_t thr, unsigned core)
{
        cpu_set_t cpuset;
        CPU_ZERO(&cpuset);
        CPU_SET(core, &cpuset);
        return -(pthread_setaffinity_np(thr, sizeof(cpuset), &cpuset) != 0);
}
 
#endif
 
unsigned thread_cores_count(void)
{
        long ret = sysconf(_SC_NPROCESSORS_ONLN);
        return ret < 1 ? 1 : (unsigned)ret;
}
 
int thread_start(thr_id_t *thr_p, thr_run_fnc t_fnc, void *t_fnc_arg)
{
        return -(pthread_create(thr_p, NULL, t_fnc, t_fnc_arg) != 0);
}
 
int thread_wait(thr_id_t thr, thr_ret_t *ret)
{
        return -(pthread_join(thr, ret) != 0);
}
 
#endif
 
#ifdef __WINDOWS
#define WIN32_LEAN_AND_MEAN
#ifndef _WIN32_WINNT
#define _WIN32_WINNT _WIN32_WINNT_NT4
#endif
#include <windows.h>
 
unsigned thread_cores_count(void)
{
        SYSTEM_INFO sys_info;
        GetSystemInfo(&sys_info);
        return sys_info.dwNumberOfProcessors;
}
 
int thread_start(thr_id_t *thr_p, thr_run_fnc t_fnc, void *t_fnc_arg)
{
        *thr_p = CreateThread(NULL, 0, t_fnc, arg, 0, NULL);
        return -(*thr_p == NULL);
}
 
int thread_affinity_set(thr_id_t thr, unsigned core)
{
        return -(SetThreadAffinityMask(thr, 1 << core) != 0);
}
 
int thread_wait(thr_id_t thr, thr_ret_t *ret)
{
        if (WaitForSingleObject(thr, INFINITE) == WAIT_FAILED)
                return -1;
 
        if (ret)
                return -(GetExitCodeThread(thr, ret) == 0);
 
        return 0;
}
 
#endif