thread.hpp

// Copyright Take Vos 2019-2020.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
 
#pragma once
 
#include "required.hpp"
#include "architecture.hpp"
#if TT_OPERATING_SYSTEM == TT_OS_WINDOWS
#include <intrin.h>
#endif
#include <thread>
#include <string_view>
#include <functional>
#include <atomic>
#include <chrono>
#include <bit>
 
namespace tt {
 
void set_thread_name(std::string_view name);
 
#if TT_OPERATING_SYSTEM == TT_OS_WINDOWS
constexpr size_t maximum_num_cpus = 64;
#elif TT_OPERATING_SYSTEM == TT_OS_LINUX || TT_OPERATING_SYSTEM == TT_OS_MACOS
constexpr size_t maximum_num_cpus = CPU_SETSIZE;
#endif
 
#if TT_OPERATING_SYSTEM == TT_OS_WINDOWS
using thread_id = uint32_t;
#else
using thread_id = uint64_t;
inline thread_local thread_id current_thread_id_dummy = 0;
#endif
 
[[nodiscard]] inline thread_id current_thread_id() noexcept
{
#if TT_OPERATING_SYSTEM == TT_OS_WINDOWS
    // Thread IDs on Win32 are guaranteed to be not zero.
    constexpr uint64_t NT_TIB_CurrentThreadID = 0x48;
    return __readgsdword(NT_TIB_CurrentThreadID);
#else
    // Addresses can not be zero.
    return reinterpret_cast<uint64_t>(&current_thread_id_dummy);
#endif
}
 
[[nodiscard]] std::vector<bool> process_affinity_mask() noexcept;
 
std::vector<bool> set_thread_affinity_mask(std::vector<bool> const &mask);
 
std::vector<bool> set_thread_affinity(size_t cpu_id);
 
size_t advance_thread_affinity(size_t &cpu) noexcept;
 
[[nodiscard]] size_t current_cpu_id() noexcept;
 
 
}