mirror of
https://github.com/sockspls/badfish
synced 2025-07-11 11:39:15 +00:00
Try to mimic std::thread API
No functional change. Signed-off-by: Marco Costalba <mcostalba@gmail.com>
This commit is contained in:
parent
41561c9bb8
commit
c483ffc773
3 changed files with 23 additions and 29 deletions
|
@ -53,7 +53,7 @@ inline uint64_t time_to_msec(const sys_time_t& t) { return t.tv_sec * 1000LL + t
|
||||||
# include <pthread.h>
|
# include <pthread.h>
|
||||||
typedef pthread_mutex_t Lock;
|
typedef pthread_mutex_t Lock;
|
||||||
typedef pthread_cond_t WaitCondition;
|
typedef pthread_cond_t WaitCondition;
|
||||||
typedef pthread_t ThreadHandle;
|
typedef pthread_t NativeHandle;
|
||||||
typedef void*(*start_fn)(void*);
|
typedef void*(*start_fn)(void*);
|
||||||
|
|
||||||
# define lock_init(x) pthread_mutex_init(&(x), NULL)
|
# define lock_init(x) pthread_mutex_init(&(x), NULL)
|
||||||
|
@ -90,7 +90,7 @@ inline uint64_t time_to_msec(const sys_time_t& t) { return t.time * 1000LL + t.m
|
||||||
// but apart from this they have the same speed performance of SRW locks.
|
// but apart from this they have the same speed performance of SRW locks.
|
||||||
typedef CRITICAL_SECTION Lock;
|
typedef CRITICAL_SECTION Lock;
|
||||||
typedef HANDLE WaitCondition;
|
typedef HANDLE WaitCondition;
|
||||||
typedef HANDLE ThreadHandle;
|
typedef HANDLE NativeHandle;
|
||||||
|
|
||||||
# define lock_init(x) InitializeCriticalSection(&(x))
|
# define lock_init(x) InitializeCriticalSection(&(x))
|
||||||
# define lock_grab(x) EnterCriticalSection(&(x))
|
# define lock_grab(x) EnterCriticalSection(&(x))
|
||||||
|
|
|
@ -32,37 +32,24 @@ ThreadsManager Threads; // Global object
|
||||||
namespace { extern "C" {
|
namespace { extern "C" {
|
||||||
|
|
||||||
// start_routine() is the C function which is called when a new thread
|
// start_routine() is the C function which is called when a new thread
|
||||||
// is launched. It simply calls idle_loop() of the supplied thread. The first
|
// is launched. It is a wrapper to member function pointed by start_fn
|
||||||
// and last thread are special. First one is the main search thread while the
|
|
||||||
// last one mimics a timer, they run in main_loop() and timer_loop().
|
|
||||||
|
|
||||||
long start_routine(Thread* th) {
|
long start_routine(Thread* th) { (th->*(th->start_fn))(); return 0; }
|
||||||
|
|
||||||
if (th->threadID == 0)
|
|
||||||
th->main_loop();
|
|
||||||
|
|
||||||
else if (th->threadID == MAX_THREADS)
|
|
||||||
th->timer_loop();
|
|
||||||
|
|
||||||
else
|
|
||||||
th->idle_loop(NULL);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
} }
|
} }
|
||||||
|
|
||||||
|
|
||||||
// Thread c'tor creates and launches the OS thread, that will go immediately to
|
// Thread c'tor starts a newly-created thread of execution that will call
|
||||||
// sleep.
|
// the idle loop function pointed by start_fn going immediately to sleep.
|
||||||
|
|
||||||
Thread::Thread(int id) {
|
Thread::Thread(Fn fn) {
|
||||||
|
|
||||||
is_searching = do_exit = false;
|
is_searching = do_exit = false;
|
||||||
maxPly = splitPointsCnt = 0;
|
maxPly = splitPointsCnt = 0;
|
||||||
curSplitPoint = NULL;
|
curSplitPoint = NULL;
|
||||||
threadID = id;
|
start_fn = fn;
|
||||||
do_sleep = (id != 0); // Avoid a race with start_thinking()
|
threadID = Threads.size();
|
||||||
|
do_sleep = (threadID != 0); // Avoid a race with start_thinking()
|
||||||
|
|
||||||
lock_init(sleepLock);
|
lock_init(sleepLock);
|
||||||
cond_init(sleepCond);
|
cond_init(sleepCond);
|
||||||
|
@ -72,7 +59,7 @@ Thread::Thread(int id) {
|
||||||
|
|
||||||
if (!thread_create(handle, start_routine, this))
|
if (!thread_create(handle, start_routine, this))
|
||||||
{
|
{
|
||||||
std::cerr << "Failed to create thread number " << id << std::endl;
|
std::cerr << "Failed to create thread number " << threadID << std::endl;
|
||||||
::exit(EXIT_FAILURE);
|
::exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -221,8 +208,10 @@ void ThreadsManager::read_uci_options() {
|
||||||
useSleepingThreads = Options["Use Sleeping Threads"];
|
useSleepingThreads = Options["Use Sleeping Threads"];
|
||||||
int requested = Options["Threads"];
|
int requested = Options["Threads"];
|
||||||
|
|
||||||
|
assert(requested > 0);
|
||||||
|
|
||||||
while (size() < requested)
|
while (size() < requested)
|
||||||
threads.push_back(new Thread(size()));
|
threads.push_back(new Thread(&Thread::idle_loop));
|
||||||
|
|
||||||
while (size() > requested)
|
while (size() > requested)
|
||||||
{
|
{
|
||||||
|
@ -264,8 +253,9 @@ void ThreadsManager::init() {
|
||||||
|
|
||||||
cond_init(sleepCond);
|
cond_init(sleepCond);
|
||||||
lock_init(splitLock);
|
lock_init(splitLock);
|
||||||
timer = new Thread(MAX_THREADS);
|
timer = new Thread(&Thread::timer_loop);
|
||||||
read_uci_options(); // Creates at least the main thread
|
threads.push_back(new Thread(&Thread::main_loop));
|
||||||
|
read_uci_options();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -70,14 +70,17 @@ class Thread {
|
||||||
Thread(const Thread&); // Only declared to disable the default ones
|
Thread(const Thread&); // Only declared to disable the default ones
|
||||||
Thread& operator=(const Thread&); // that are not suitable in this case.
|
Thread& operator=(const Thread&); // that are not suitable in this case.
|
||||||
|
|
||||||
|
typedef void (Thread::* Fn) ();
|
||||||
|
|
||||||
public:
|
public:
|
||||||
Thread(int id);
|
Thread(Fn fn);
|
||||||
~Thread();
|
~Thread();
|
||||||
|
|
||||||
void wake_up();
|
void wake_up();
|
||||||
bool cutoff_occurred() const;
|
bool cutoff_occurred() const;
|
||||||
bool is_available_to(int master) const;
|
bool is_available_to(int master) const;
|
||||||
void idle_loop(SplitPoint* sp_master);
|
void idle_loop(SplitPoint* sp_master);
|
||||||
|
void idle_loop() { idle_loop(NULL); } // Hack to allow storing in start_fn
|
||||||
void main_loop();
|
void main_loop();
|
||||||
void timer_loop();
|
void timer_loop();
|
||||||
void wait_for_stop_or_ponderhit();
|
void wait_for_stop_or_ponderhit();
|
||||||
|
@ -89,7 +92,8 @@ public:
|
||||||
int maxPly;
|
int maxPly;
|
||||||
Lock sleepLock;
|
Lock sleepLock;
|
||||||
WaitCondition sleepCond;
|
WaitCondition sleepCond;
|
||||||
ThreadHandle handle;
|
NativeHandle handle;
|
||||||
|
Fn start_fn;
|
||||||
SplitPoint* volatile curSplitPoint;
|
SplitPoint* volatile curSplitPoint;
|
||||||
volatile int splitPointsCnt;
|
volatile int splitPointsCnt;
|
||||||
volatile bool is_searching;
|
volatile bool is_searching;
|
||||||
|
|
Loading…
Add table
Reference in a new issue