mirror of
https://github.com/Ryujinx/Ryujinx.git
synced 2024-10-01 12:30:00 +02:00
380 lines
No EOL
10 KiB
C#
380 lines
No EOL
10 KiB
C#
using Ryujinx.HLE.Logging;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Threading;
|
|
|
|
namespace Ryujinx.HLE.OsHle.Handles
|
|
{
|
|
class KProcessScheduler : IDisposable
|
|
{
|
|
private ConcurrentDictionary<KThread, SchedulerThread> AllThreads;
|
|
|
|
private ThreadQueue WaitingToRun;
|
|
|
|
private KThread[] CoreThreads;
|
|
|
|
private bool[] CoreReschedule;
|
|
|
|
private object SchedLock;
|
|
|
|
private Logger Log;
|
|
|
|
public KProcessScheduler(Logger Log)
|
|
{
|
|
this.Log = Log;
|
|
|
|
AllThreads = new ConcurrentDictionary<KThread, SchedulerThread>();
|
|
|
|
WaitingToRun = new ThreadQueue();
|
|
|
|
CoreThreads = new KThread[4];
|
|
|
|
CoreReschedule = new bool[4];
|
|
|
|
SchedLock = new object();
|
|
}
|
|
|
|
public void StartThread(KThread Thread)
|
|
{
|
|
lock (SchedLock)
|
|
{
|
|
SchedulerThread SchedThread = new SchedulerThread(Thread);
|
|
|
|
if (!AllThreads.TryAdd(Thread, SchedThread))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (TryAddToCore(Thread))
|
|
{
|
|
SchedThread.IsRunning = true;
|
|
|
|
Thread.Thread.Execute();
|
|
|
|
PrintDbgThreadInfo(Thread, "running.");
|
|
}
|
|
else
|
|
{
|
|
WaitingToRun.Push(SchedThread);
|
|
|
|
PrintDbgThreadInfo(Thread, "waiting to run.");
|
|
}
|
|
}
|
|
}
|
|
|
|
public void RemoveThread(KThread Thread)
|
|
{
|
|
PrintDbgThreadInfo(Thread, "exited.");
|
|
|
|
lock (SchedLock)
|
|
{
|
|
if (AllThreads.TryRemove(Thread, out SchedulerThread SchedThread))
|
|
{
|
|
WaitingToRun.Remove(SchedThread);
|
|
|
|
SchedThread.Dispose();
|
|
}
|
|
|
|
int ActualCore = Thread.ActualCore;
|
|
|
|
SchedulerThread NewThread = WaitingToRun.Pop(ActualCore);
|
|
|
|
if (NewThread == null)
|
|
{
|
|
Log.PrintDebug(LogClass.KernelScheduler, $"Nothing to run on core {ActualCore}!");
|
|
|
|
CoreThreads[ActualCore] = null;
|
|
|
|
return;
|
|
}
|
|
|
|
NewThread.Thread.ActualCore = ActualCore;
|
|
|
|
RunThread(NewThread);
|
|
}
|
|
}
|
|
|
|
public void SetThreadActivity(KThread Thread, bool Active)
|
|
{
|
|
SchedulerThread SchedThread = AllThreads[Thread];
|
|
|
|
SchedThread.IsActive = Active;
|
|
|
|
if (Active)
|
|
{
|
|
SchedThread.WaitActivity.Set();
|
|
}
|
|
else
|
|
{
|
|
SchedThread.WaitActivity.Reset();
|
|
}
|
|
}
|
|
|
|
public bool IsThreadRunning(KThread Thread)
|
|
{
|
|
if (!AllThreads.TryGetValue(Thread, out SchedulerThread SchedThread))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return SchedThread.IsRunning;
|
|
}
|
|
|
|
public void EnterWait(KThread Thread, int TimeoutMs = Timeout.Infinite)
|
|
{
|
|
SchedulerThread SchedThread = AllThreads[Thread];
|
|
|
|
Suspend(Thread);
|
|
|
|
SchedThread.WaitSync.WaitOne(TimeoutMs);
|
|
|
|
TryResumingExecution(SchedThread);
|
|
}
|
|
|
|
public void WakeUp(KThread Thread)
|
|
{
|
|
AllThreads[Thread].WaitSync.Set();
|
|
}
|
|
|
|
public void ChangeCore(KThread Thread, int IdealCore, int CoreMask)
|
|
{
|
|
lock (SchedLock)
|
|
{
|
|
if (IdealCore != -3)
|
|
{
|
|
Thread.IdealCore = IdealCore;
|
|
}
|
|
|
|
Thread.CoreMask = CoreMask;
|
|
|
|
if (AllThreads.ContainsKey(Thread))
|
|
{
|
|
SetReschedule(Thread.ActualCore);
|
|
|
|
SchedulerThread SchedThread = AllThreads[Thread];
|
|
|
|
//Note: Aways if the thread is on the queue first, and try
|
|
//adding to a new core later, to ensure that a thread that
|
|
//is already running won't be added to another core.
|
|
if (WaitingToRun.HasThread(SchedThread) && TryAddToCore(Thread))
|
|
{
|
|
WaitingToRun.Remove(SchedThread);
|
|
|
|
RunThread(SchedThread);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public void Suspend(KThread Thread)
|
|
{
|
|
lock (SchedLock)
|
|
{
|
|
AllThreads[Thread].IsRunning = false;
|
|
|
|
PrintDbgThreadInfo(Thread, "suspended.");
|
|
|
|
int ActualCore = Thread.ActualCore;
|
|
|
|
CoreReschedule[ActualCore] = false;
|
|
|
|
SchedulerThread SchedThread = WaitingToRun.Pop(ActualCore);
|
|
|
|
if (SchedThread != null)
|
|
{
|
|
SchedThread.Thread.ActualCore = ActualCore;
|
|
|
|
CoreThreads[ActualCore] = SchedThread.Thread;
|
|
|
|
RunThread(SchedThread);
|
|
}
|
|
else
|
|
{
|
|
Log.PrintDebug(LogClass.KernelScheduler, $"Nothing to run on core {Thread.ActualCore}!");
|
|
|
|
CoreThreads[ActualCore] = null;
|
|
}
|
|
}
|
|
}
|
|
|
|
public void SetReschedule(int Core)
|
|
{
|
|
lock (SchedLock)
|
|
{
|
|
CoreReschedule[Core] = true;
|
|
}
|
|
}
|
|
|
|
public void Reschedule(KThread Thread)
|
|
{
|
|
bool NeedsReschedule;
|
|
|
|
lock (SchedLock)
|
|
{
|
|
int ActualCore = Thread.ActualCore;
|
|
|
|
NeedsReschedule = CoreReschedule[ActualCore];
|
|
|
|
CoreReschedule[ActualCore] = false;
|
|
}
|
|
|
|
if (NeedsReschedule)
|
|
{
|
|
Yield(Thread, Thread.ActualPriority - 1);
|
|
}
|
|
}
|
|
|
|
public void Yield(KThread Thread)
|
|
{
|
|
Yield(Thread, Thread.ActualPriority);
|
|
}
|
|
|
|
private void Yield(KThread Thread, int MinPriority)
|
|
{
|
|
PrintDbgThreadInfo(Thread, "yielded execution.");
|
|
|
|
lock (SchedLock)
|
|
{
|
|
int ActualCore = Thread.ActualCore;
|
|
|
|
SchedulerThread NewThread = WaitingToRun.Pop(ActualCore, MinPriority);
|
|
|
|
if (NewThread != null)
|
|
{
|
|
NewThread.Thread.ActualCore = ActualCore;
|
|
|
|
CoreThreads[ActualCore] = NewThread.Thread;
|
|
|
|
RunThread(NewThread);
|
|
}
|
|
else
|
|
{
|
|
CoreThreads[ActualCore] = null;
|
|
}
|
|
}
|
|
|
|
Resume(Thread);
|
|
}
|
|
|
|
public void Resume(KThread Thread)
|
|
{
|
|
TryResumingExecution(AllThreads[Thread]);
|
|
}
|
|
|
|
private void TryResumingExecution(SchedulerThread SchedThread)
|
|
{
|
|
SchedThread.IsRunning = false;
|
|
|
|
KThread Thread = SchedThread.Thread;
|
|
|
|
PrintDbgThreadInfo(Thread, "trying to resume...");
|
|
|
|
SchedThread.WaitActivity.WaitOne();
|
|
|
|
lock (SchedLock)
|
|
{
|
|
if (TryAddToCore(Thread))
|
|
{
|
|
SchedThread.IsRunning = true;
|
|
|
|
PrintDbgThreadInfo(Thread, "resuming execution...");
|
|
|
|
return;
|
|
}
|
|
|
|
WaitingToRun.Push(SchedThread);
|
|
|
|
SetReschedule(Thread.ProcessorId);
|
|
|
|
PrintDbgThreadInfo(Thread, "entering wait state...");
|
|
}
|
|
|
|
SchedThread.WaitSched.WaitOne();
|
|
|
|
PrintDbgThreadInfo(Thread, "resuming execution...");
|
|
}
|
|
|
|
private void RunThread(SchedulerThread SchedThread)
|
|
{
|
|
if (!SchedThread.Thread.Thread.Execute())
|
|
{
|
|
PrintDbgThreadInfo(SchedThread.Thread, "waked.");
|
|
|
|
SchedThread.WaitSched.Set();
|
|
}
|
|
else
|
|
{
|
|
PrintDbgThreadInfo(SchedThread.Thread, "running.");
|
|
}
|
|
|
|
SchedThread.IsRunning = true;
|
|
}
|
|
|
|
public void Resort(KThread Thread)
|
|
{
|
|
if (AllThreads.TryGetValue(Thread, out SchedulerThread SchedThread))
|
|
{
|
|
WaitingToRun.Resort(SchedThread);
|
|
}
|
|
}
|
|
|
|
private bool TryAddToCore(KThread Thread)
|
|
{
|
|
//First, try running it on Ideal Core.
|
|
int IdealCore = Thread.IdealCore;
|
|
|
|
if (IdealCore != -1 && CoreThreads[IdealCore] == null)
|
|
{
|
|
Thread.ActualCore = IdealCore;
|
|
|
|
CoreThreads[IdealCore] = Thread;
|
|
|
|
return true;
|
|
}
|
|
|
|
//If that fails, then try running on any core allowed by Core Mask.
|
|
int CoreMask = Thread.CoreMask;
|
|
|
|
for (int Core = 0; Core < CoreThreads.Length; Core++, CoreMask >>= 1)
|
|
{
|
|
if ((CoreMask & 1) != 0 && CoreThreads[Core] == null)
|
|
{
|
|
Thread.ActualCore = Core;
|
|
|
|
CoreThreads[Core] = Thread;
|
|
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
private void PrintDbgThreadInfo(KThread Thread, string Message)
|
|
{
|
|
Log.PrintDebug(LogClass.KernelScheduler, "(" +
|
|
"ThreadId = " + Thread.ThreadId + ", " +
|
|
"CoreMask = 0x" + Thread.CoreMask.ToString("x1") + ", " +
|
|
"ActualCore = " + Thread.ActualCore + ", " +
|
|
"IdealCore = " + Thread.IdealCore + ", " +
|
|
"ActualPriority = " + Thread.ActualPriority + ", " +
|
|
"WantedPriority = " + Thread.WantedPriority + ") " + Message);
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
}
|
|
|
|
protected virtual void Dispose(bool Disposing)
|
|
{
|
|
if (Disposing)
|
|
{
|
|
foreach (SchedulerThread SchedThread in AllThreads.Values)
|
|
{
|
|
SchedThread.Dispose();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} |