thread.hpp

// Copyright Take Vos 2020-2022.
// 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 "utility.hpp"
#include "architecture.hpp"
#if HI_OPERATING_SYSTEM == HI_OS_WINDOWS
#include <intrin.h>
#endif
#include <thread>
#include <string_view>
#include <functional>
#include <atomic>
#include <chrono>
#include <bit>
 
namespace hi::inline v1 {
 
#if HI_OPERATING_SYSTEM == HI_OS_WINDOWS
constexpr std::size_t maximum_num_cpus = 64;
#elif HI_OPERATING_SYSTEM == HI_OS_LINUX || HI_OPERATING_SYSTEM == HI_OS_MACOS
constexpr std::size_t maximum_num_cpus = CPU_SETSIZE;
#endif
 
#if HI_OPERATING_SYSTEM == HI_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 HI_OPERATING_SYSTEM == HI_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
}
 
void set_thread_name(std::string_view name) noexcept;
 
[[nodiscard]] std::string get_thread_name(thread_id id) noexcept;
 
[[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(std::size_t cpu_id);
 
std::size_t advance_thread_affinity(std::size_t &cpu) noexcept;
 
[[nodiscard]] std::size_t current_cpu_id() noexcept;
 
} // namespace hi::inline v1