forked from suyu/suyu
80 lines
2.2 KiB
C++
80 lines
2.2 KiB
C++
|
// Copyright 2020 yuzu Emulator Project
|
||
|
// Licensed under GPLv2 or any later version
|
||
|
// Refer to the license.txt file included.
|
||
|
|
||
|
#pragma once
|
||
|
|
||
|
#include <atomic>
|
||
|
#include <vector>
|
||
|
|
||
|
#include "common/common_types.h"
|
||
|
#include "common/spin_lock.h"
|
||
|
#include "core/hardware_properties.h"
|
||
|
#include "core/hle/kernel/k_priority_queue.h"
|
||
|
#include "core/hle/kernel/k_scheduler_lock.h"
|
||
|
#include "core/hle/kernel/thread.h"
|
||
|
|
||
|
namespace Kernel {
|
||
|
|
||
|
class KernelCore;
|
||
|
class SchedulerLock;
|
||
|
|
||
|
using KSchedulerPriorityQueue =
|
||
|
KPriorityQueue<Thread, Core::Hardware::NUM_CPU_CORES, THREADPRIO_LOWEST, THREADPRIO_HIGHEST>;
|
||
|
static constexpr s32 HighestCoreMigrationAllowedPriority = 2;
|
||
|
|
||
|
class GlobalSchedulerContext final {
|
||
|
friend class KScheduler;
|
||
|
|
||
|
public:
|
||
|
explicit GlobalSchedulerContext(KernelCore& kernel);
|
||
|
~GlobalSchedulerContext();
|
||
|
|
||
|
/// Adds a new thread to the scheduler
|
||
|
void AddThread(std::shared_ptr<Thread> thread);
|
||
|
|
||
|
/// Removes a thread from the scheduler
|
||
|
void RemoveThread(std::shared_ptr<Thread> thread);
|
||
|
|
||
|
/// Returns a list of all threads managed by the scheduler
|
||
|
const std::vector<std::shared_ptr<Thread>>& GetThreadList() const {
|
||
|
return thread_list;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Rotates the scheduling queues of threads at a preemption priority and then does
|
||
|
* some core rebalancing. Preemption priorities can be found in the array
|
||
|
* 'preemption_priorities'.
|
||
|
*
|
||
|
* @note This operation happens every 10ms.
|
||
|
*/
|
||
|
void PreemptThreads();
|
||
|
|
||
|
/// Returns true if the global scheduler lock is acquired
|
||
|
bool IsLocked() const;
|
||
|
|
||
|
private:
|
||
|
friend class SchedulerLock;
|
||
|
|
||
|
/// Lock the scheduler to the current thread.
|
||
|
void Lock();
|
||
|
|
||
|
/// Unlocks the scheduler, reselects threads, interrupts cores for rescheduling
|
||
|
/// and reschedules current core if needed.
|
||
|
void Unlock();
|
||
|
|
||
|
using LockType = KAbstractSchedulerLock<KScheduler>;
|
||
|
|
||
|
KernelCore& kernel;
|
||
|
|
||
|
std::atomic_bool scheduler_update_needed{};
|
||
|
KSchedulerPriorityQueue priority_queue;
|
||
|
LockType scheduler_lock;
|
||
|
|
||
|
/// Lists all thread ids that aren't deleted/etc.
|
||
|
std::vector<std::shared_ptr<Thread>> thread_list;
|
||
|
Common::SpinLock global_list_guard{};
|
||
|
};
|
||
|
|
||
|
} // namespace Kernel
|