feat(kernel): improved scheduler and kernel main code.

Signed-off-by: Amlal El Mahrouss <amlal@nekernel.org>

1 files changed, 2 insertions, 3 deletions

diff --git a/dev/kernel/src/UserProcessScheduler.cc b/dev/kernel/src/UserProcessScheduler.cc
index 54f8ca77..1a208bae 100644
--- a/dev/kernel/src/UserProcessScheduler.cc
+++ b/dev/kernel/src/UserProcessScheduler.cc
@@ -495,9 +495,6 @@ SizeT UserProcessScheduler::Run() noexcept {
 
       this->CurrentProcess() = process;
 
-      process.PTime = static_cast<Int32>(process.Affinity);
-      process.RTime = 0UL;
-
       // tell helper to find a core to schedule on, otherwise run on this core directly.
       if (!UserProcessHelper::Switch(process.StackFrame, process.ProcessId)) {
         if (process.ProcessId == this->CurrentProcess().Leak().ProcessId &&
@@ -510,7 +507,9 @@ SizeT UserProcessScheduler::Run() noexcept {
             else if (process.RTime < (Int32) AffinityKind::kStandard)
               process.PTime = (Int32) AffinityKind::kHigh;
 
+            process.PTime = static_cast<Int32>(process.Affinity);
             process.RTime = 0UL;
+
           } else {
             ++process.RTime;
           }