From c0428f8e5feba9573e572a1a1b3c66cfa1f29108 Mon Sep 17 00:00:00 2001
From: Amlal El Mahrouss <amlal@el-mahrouss-logic.com>
Date: Fri, 17 May 2024 22:25:04 +0200
Subject: MHR-23: Improve scheduler code.

Signed-off-by: Amlal El Mahrouss <amlal@el-mahrouss-logic.com>
---
 Kernel/KernelKit/ProcessScheduler.hpp | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

(limited to 'Kernel/KernelKit')

diff --git a/Kernel/KernelKit/ProcessScheduler.hpp b/Kernel/KernelKit/ProcessScheduler.hpp
index 243857ae..9768d56a 100644
--- a/Kernel/KernelKit/ProcessScheduler.hpp
+++ b/Kernel/KernelKit/ProcessScheduler.hpp
@@ -252,23 +252,25 @@ namespace NewOS
 		{
 			return mTeam.AsArray().Count() > 0;
 		}
+
 		bool operator!()
 		{
 			return mTeam.AsArray().Count() == 0;
 		}
 
-		ProcessTeam& CurrentTeam()
-		{
-			return mTeam;
-		}
+	public:
+		ProcessTeam& CurrentTeam();
 
+	public:
 		SizeT Add(Ref<ProcessHeader>& headerRef);
 		bool  Remove(SizeT headerIndex);
 
+	public:
 		Ref<ProcessHeader>& GetCurrent();
 		SizeT				Run() noexcept;
 
-		static Ref<ProcessScheduler> Shared();
+	public:
+		static Ref<ProcessScheduler&> Shared();
 
 	private:
 		ProcessTeam mTeam;
-- 
cgit v1.2.3