blender/intern/cycles/util/thread.cpp

59 lines
1.1 KiB
C++

/* SPDX-License-Identifier: Apache-2.0
* Copyright 2011-2022 Blender Foundation */
#include "util/thread.h"
#include "util/system.h"
#include "util/windows.h"
#include <system_error>
CCL_NAMESPACE_BEGIN
thread::thread(function<void()> run_cb) : run_cb_(run_cb), joined_(false)
{
#ifdef __APPLE__
/* Set the stack size to 2MB to match Linux. The default 512KB on macOS is
* too small for Embree, and consistent stack size also makes things more
* predictable in general. */
pthread_attr_t attribute;
pthread_attr_init(&attribute);
pthread_attr_setstacksize(&attribute, 1024 * 1024 * 2);
pthread_create(&pthread_id, &attribute, run, (void *)this);
#else
std_thread = std::thread(&thread::run, this);
#endif
}
thread::~thread()
{
if (!joined_) {
join();
}
}
void *thread::run(void *arg)
{
thread *self = (thread *)(arg);
self->run_cb_();
return NULL;
}
bool thread::join()
{
joined_ = true;
#ifdef __APPLE__
return pthread_join(pthread_id, NULL) == 0;
#else
try {
std_thread.join();
return true;
}
catch (const std::system_error &) {
return false;
}
#endif
}
CCL_NAMESPACE_END