summaryrefslogtreecommitdiffstats
path: root/src/core/hle/kernel/k_light_condition_variable.h
blob: 362d0db28408ac7bb14283c27e8e9855e1e60853 (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
53
54
55
56
57
// Copyright 2020 yuzu Emulator Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

// This file references various implementation details from Atmosphere, an open-source firmware for
// the Nintendo Switch. Copyright 2018-2020 Atmosphere-NX.

#pragma once

#include "common/common_types.h"
#include "core/hle/kernel/k_scheduler.h"
#include "core/hle/kernel/k_scoped_scheduler_lock_and_sleep.h"
#include "core/hle/kernel/k_thread_queue.h"
#include "core/hle/kernel/time_manager.h"

namespace Kernel {
class KernelCore;

class KLightConditionVariable {
public:
    explicit KLightConditionVariable(KernelCore& kernel) : thread_queue(kernel), kernel(kernel) {}

    void Wait(KLightLock* lock, s64 timeout = -1) {
        WaitImpl(lock, timeout);
        lock->Lock();
    }

    void Broadcast() {
        KScopedSchedulerLock lk{kernel};
        while (thread_queue.WakeupFrontThread() != nullptr) {
            // We want to signal all threads, and so should continue waking up until there's nothing
            // to wake.
        }
    }

private:
    void WaitImpl(KLightLock* lock, s64 timeout) {
        KThread* owner = GetCurrentThreadPointer(kernel);

        // Sleep the thread.
        {
            KScopedSchedulerLockAndSleep lk(kernel, owner, timeout);
            lock->Unlock();

            if (!thread_queue.SleepThread(owner)) {
                lk.CancelSleep();
                return;
            }
        }

        // Cancel the task that the sleep setup.
        kernel.TimeManager().UnscheduleTimeEvent(owner);
    }
    KThreadQueue thread_queue;
    KernelCore& kernel;
};
} // namespace Kernel