2021-01-30 10:40:49 +01:00
|
|
|
// 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/time_manager.h"
|
|
|
|
|
|
|
|
namespace Kernel {
|
|
|
|
class KernelCore;
|
|
|
|
|
|
|
|
class KLightConditionVariable {
|
|
|
|
public:
|
2021-06-10 19:56:35 +02:00
|
|
|
explicit KLightConditionVariable(KernelCore& kernel_) : kernel{kernel_} {}
|
2021-01-30 10:40:49 +01:00
|
|
|
|
2021-06-10 19:56:35 +02:00
|
|
|
void Wait(KLightLock* lock, s64 timeout = -1, bool allow_terminating_thread = true) {
|
|
|
|
WaitImpl(lock, timeout, allow_terminating_thread);
|
2021-01-30 10:40:49 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void Broadcast() {
|
|
|
|
KScopedSchedulerLock lk{kernel};
|
2021-06-10 19:56:35 +02:00
|
|
|
|
|
|
|
// Signal all threads.
|
|
|
|
for (auto& thread : wait_list) {
|
|
|
|
thread.SetState(ThreadState::Runnable);
|
2021-01-30 10:40:49 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
2021-06-10 19:56:35 +02:00
|
|
|
void WaitImpl(KLightLock* lock, s64 timeout, bool allow_terminating_thread) {
|
2021-01-30 10:40:49 +01:00
|
|
|
KThread* owner = GetCurrentThreadPointer(kernel);
|
|
|
|
|
2021-02-02 03:23:00 +01:00
|
|
|
// Sleep the thread.
|
2021-01-30 10:40:49 +01:00
|
|
|
{
|
2021-06-10 19:56:35 +02:00
|
|
|
KScopedSchedulerLockAndSleep lk{kernel, owner, timeout};
|
2021-01-30 10:40:49 +01:00
|
|
|
|
2021-06-10 19:56:35 +02:00
|
|
|
if (!allow_terminating_thread && owner->IsTerminationRequested()) {
|
2021-01-30 10:40:49 +01:00
|
|
|
lk.CancelSleep();
|
|
|
|
return;
|
|
|
|
}
|
2021-06-10 19:56:35 +02:00
|
|
|
|
|
|
|
lock->Unlock();
|
|
|
|
|
|
|
|
// Set the thread as waiting.
|
|
|
|
GetCurrentThread(kernel).SetState(ThreadState::Waiting);
|
|
|
|
|
|
|
|
// Add the thread to the queue.
|
|
|
|
wait_list.push_back(GetCurrentThread(kernel));
|
|
|
|
}
|
|
|
|
|
|
|
|
// Remove the thread from the wait list.
|
|
|
|
{
|
|
|
|
KScopedSchedulerLock sl{kernel};
|
|
|
|
|
|
|
|
wait_list.erase(wait_list.iterator_to(GetCurrentThread(kernel)));
|
2021-01-30 10:40:49 +01:00
|
|
|
}
|
|
|
|
|
2021-02-02 03:23:00 +01:00
|
|
|
// Cancel the task that the sleep setup.
|
2021-01-30 10:40:49 +01:00
|
|
|
kernel.TimeManager().UnscheduleTimeEvent(owner);
|
2021-06-10 19:56:35 +02:00
|
|
|
|
|
|
|
// Re-acquire the lock.
|
|
|
|
lock->Lock();
|
2021-01-30 10:40:49 +01:00
|
|
|
}
|
2021-06-10 19:56:35 +02:00
|
|
|
|
2021-02-02 03:23:34 +01:00
|
|
|
KernelCore& kernel;
|
2021-06-10 19:56:35 +02:00
|
|
|
KThread::WaiterList wait_list{};
|
2021-01-30 10:40:49 +01:00
|
|
|
};
|
|
|
|
} // namespace Kernel
|