this_thread.hpp

浏览该文件的文档.

#ifndef NEFORCE_CORE_ASYNC_THIS_THREAD_HPP__
#define NEFORCE_CORE_ASYNC_THIS_THREAD_HPP__

 
#include "NeForce/core/typeinfo/type_traits.hpp"
#ifdef NEFORCE_PLATFORM_WINDOWS
#    include "NeForce/core/config/windef.hpp"
#    include <processthreadsapi.h>
#    include <synchapi.h>
#    ifdef max
#        undef max
#    endif
#    ifdef min
#        undef min
#    endif
#endif
#ifdef NEFORCE_PLATFORM_LINUX
#    include <sched.h>
#    include <ctime>
#endif
NEFORCE_BEGIN_NAMESPACE__

struct cpu_times {
    uint64_t user;   
    uint64_t kernel; 
    uint64_t idle;   
    NEFORCE_NODISCARD uint64_t total() const noexcept { return user + kernel + idle; }
};
 
NEFORCE_BEGIN_THIS_THREAD__



NEFORCE_ALWAYS_INLINE_INLINE int64_t current_cpu() noexcept {
#ifdef NEFORCE_PLATFORM_WINDOWS
    return static_cast<int64_t>(::GetCurrentProcessorNumber());
#else
    return static_cast<int64_t>(::sched_getcpu());
#endif
}

NEFORCE_ALWAYS_INLINE_INLINE void yield() noexcept {
#ifdef NEFORCE_PLATFORM_WINDOWS
    ::SwitchToThread();
#else
    ::sched_yield();
#endif
}

NEFORCE_ALWAYS_INLINE_INLINE void relax() noexcept {
#ifdef NEFORCE_COMPILER_MSVC
#    ifdef NEFORCE_ARCH_X86
    ::_mm_pause();
#    else
    ::__yield();
#    endif
#else
#    if defined(NEFORCE_ARCH_X86)
    asm volatile("pause" ::: "memory");
#    elif defined(NEFORCE_ARCH_ARM)
    asm volatile("yield" ::: "memory");
#    elif defined(NEFORCE_ARCH_RISCV)
#        ifdef __riscv_zihintpause
    asm volatile("pause" ::: "memory");
#        else
    asm volatile("fence" ::: "memory");
#        endif
#    elif defined(NEFORCE_ARCH_LOONGARCH)
    asm volatile("dbar 0" ::: "memory");
#    else
    this_thread::yield();
#    endif
#endif
}

NEFORCE_ALWAYS_INLINE_INLINE void sleep_for_ms(uint32_t milliseconds) noexcept {
#ifdef NEFORCE_PLATFORM_WINDOWS
    ::Sleep(milliseconds);
#else
    ::timespec ts;
    ts.tv_sec = milliseconds / 1000;
    ts.tv_nsec = static_cast<decay_t<decltype(ts.tv_nsec)>>(milliseconds % 1000) * 1000000;
    ::nanosleep(&ts, nullptr);
#endif
}

void NEFORCE_API sleep_for_ms(uint32_t ms, bool busy_wait) noexcept;

void NEFORCE_API sleep_for_us(uint64_t us) noexcept;

void NEFORCE_API sleep_for_ns(uint64_t ns) noexcept;

bool NEFORCE_API set_affinity(size_t cpu_mask) noexcept;

bool NEFORCE_API affinity(uint64_t& affi) noexcept;

bool NEFORCE_API cpu_time(cpu_times& times) noexcept;

bool NEFORCE_API set_priority(int priority) noexcept;

int NEFORCE_API priority() noexcept;
 // Thread
 // AsyncComponents
 
NEFORCE_END_THIS_THREAD__
NEFORCE_END_NAMESPACE__
#endif // NEFORCE_CORE_ASYNC_THIS_THREAD_HPP__