this_thread.hpp

浏览该文件的文档.

#ifndef NEFORCE_CORE_ASYNC_THIS_THREAD_HPP__
#define NEFORCE_CORE_ASYNC_THIS_THREAD_HPP__

 
#include "NeForce/core/typeinfo/types.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 <unistd.h>
#endif
#ifdef NEFORCE_ARCH_RISCV
#    include <riscv_pause.h>
#endif
NEFORCE_BEGIN_NAMESPACE__
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_PLATFORM_WINDOWS
    ::_mm_pause();
#else
#    if defined(NEFORCE_ARCH_X86)
    __builtin_ia32_pause();
#    elif defined(NEFORCE_ARCH_ARM)
    asm volatile("yield" ::: "memory");
#    elif defined(NEFORCE_ARCH_RISCV)
    ::riscv_pause();
#    elif defined(NEFORCE_ARCH_LOONGARCH)
    asm volatile("dbar 0" ::: "memory");
#    else
    this_thread::yield();
#    endif
#endif
}

NEFORCE_ALWAYS_INLINE_INLINE void relax(const int count) noexcept {
    if (count < 4) {
        for (int i = 0; i < (1 << count); ++i) {
            this_thread::relax();
        }
    } else if (count < 8) {
        this_thread::relax();
    } else {
        this_thread::yield();
    }
}

NEFORCE_ALWAYS_INLINE_INLINE void sleep_for_ms(uint32_t milliseconds) noexcept {
#ifdef NEFORCE_PLATFORM_WINDOWS
    ::Sleep(milliseconds);
#else
    ::usleep(milliseconds * 1000);
#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 affinity(size_t cpu_mask) noexcept;

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