forked from suyu/suyu
57 lines
1.6 KiB
C++
57 lines
1.6 KiB
C++
// 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
|