summaryrefslogtreecommitdiffstats
path: root/src/core/hle/kernel/k_affinity_mask.h
blob: b58716e90ed3c8615dc05270245ee7de291e8158 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
// SPDX-FileCopyrightText: Copyright 2020 yuzu Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#pragma once

#include "common/assert.h"
#include "common/common_types.h"
#include "core/hardware_properties.h"

namespace Kernel {

class KAffinityMask {
public:
    constexpr KAffinityMask() = default;

    [[nodiscard]] constexpr u64 GetAffinityMask() const {
        return this->mask;
    }

    constexpr void SetAffinityMask(u64 new_mask) {
        ASSERT((new_mask & ~AllowedAffinityMask) == 0);
        this->mask = new_mask;
    }

    [[nodiscard]] constexpr bool GetAffinity(s32 core) const {
        return (this->mask & GetCoreBit(core)) != 0;
    }

    constexpr void SetAffinity(s32 core, bool set) {
        if (set) {
            this->mask |= GetCoreBit(core);
        } else {
            this->mask &= ~GetCoreBit(core);
        }
    }

    constexpr void SetAll() {
        this->mask = AllowedAffinityMask;
    }

private:
    [[nodiscard]] static constexpr u64 GetCoreBit(s32 core) {
        ASSERT(0 <= core && core < static_cast<s32>(Core::Hardware::NUM_CPU_CORES));
        return (1ULL << core);
    }

    static constexpr u64 AllowedAffinityMask = (1ULL << Core::Hardware::NUM_CPU_CORES) - 1;

    u64 mask{};
};

} // namespace Kernel