cpuaffinity.cpp

Go to the documentation of this file.

/**
 * @file
 * @ingroup numapp
 * @copyright ESO 2024 - European Southern Observatory
 *
 * @brief Definition of CPU affinity classes.
 */
#include <numapp/cpuaffinity.hpp>
 
#include <ostream>
 
#include <numapp/lowlevel.hpp>
#include <numapp/thread.hpp>
 
namespace numapp {
namespace thisThread {
[[nodiscard]] std::error_code Apply(CpuAffinity const& affinity) noexcept {
    return Apply(GetThreadId(), affinity);
}
}  // namespace thisThread
 
[[nodiscard]] std::error_code Apply(pid_t thread, CpuAffinity const& affinity) noexcept {
    if (auto err = ll::SetCpuAffinity(thread, affinity.GetMask()); err) {
        return err;
    }
    return {};
}
 
CpuAffinity::CpuAffinity(Cpumask&& mask) noexcept : m_mask(std::move(mask)) {
}
 
CpuAffinity::CpuAffinity(Cpumask const& mask) noexcept : m_mask(mask) {
}
 
[[nodiscard]] CpuAffinity CpuAffinity::MakeFromActive() {
    auto mask = Cpumask();
    if (auto err = ll::GetCpuAffinity(thisThread::GetThreadId(), mask); err) {
        throw std::system_error(err);
    }
    return CpuAffinity(std::move(mask));
}
 
[[nodiscard]] CpuAffinity CpuAffinity::MakeFromCpuString(char const* cpustring) {
    return CpuAffinity(Cpumask::MakeFromCpuString(cpustring));
}
 
[[nodiscard]] CpuAffinity CpuAffinity::MakeFromCpuStringAll(char const* cpustring) {
    return CpuAffinity(Cpumask::MakeFromCpuStringAll(cpustring));
}
 
[[nodiscard]] CpuAffinity CpuAffinity::MakeBindNode(int node) {
    auto mask = Cpumask();
    if (auto err = ll::NumaNodeToCpumask(node, mask); err) {
        throw std::system_error(err);
    }
    return CpuAffinity(std::move(mask));
}
 
[[nodiscard]] CpuAffinity CpuAffinity::MakeBindCpu(int cpu) {
    auto mask = Cpumask();
    mask.Set(cpu);
    return CpuAffinity(std::move(mask));
}
 
Cpumask const& CpuAffinity::GetMask() const noexcept {
    return m_mask;
}
 
std::ostream& operator<<(std::ostream& os, CpuAffinity const& affinity) {
    os << "CpuAffinity {mask=" << affinity.GetMask() << "}";
    return os;
}
 
}  // namespace numapp