/*
 * Copyright (c) Atmosphère-NX
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#pragma once
#include <mesosphere/kern_common.hpp>
#include <mesosphere/kern_k_typed_address.hpp>
#include <mesosphere/kern_select_cpu.hpp>
#include <mesosphere/kern_k_memory_layout.hpp>
#include <mesosphere/kern_k_memory_manager.hpp>
#include <mesosphere/kern_k_scheduler.hpp>
#include <mesosphere/kern_k_interrupt_task_manager.hpp>
#include <mesosphere/kern_select_interrupt_manager.hpp>
#include <mesosphere/kern_select_hardware_timer.hpp>
#include <mesosphere/kern_k_worker_task_manager.hpp>

namespace ams::kern {

    class KThread;
    class KHardwareTimer;
    class KResourceLimit;
    class KInterruptManager;
    class KInterruptTaskManager;
    class KScheduler;
    class KMemoryManager;
    class KPageTableManager;
    class KMemoryBlockSlabManager;
    class KBlockInfoManager;
    class KUnsafeMemory;

#if defined(ATMOSPHERE_ARCH_ARM64)

    namespace arch::arm64 {
        class KSupervisorPageTable;
    }
    using ams::kern::arch::arm64::KSupervisorPageTable;

#else

    #error "Unknown architecture for KSupervisorPageTable forward declare"

#endif

    class Kernel {
        public:
            enum class State : u8 {
                Invalid = 0,
                Initializing = 1,
                Initialized  = 2,
            };

            static constexpr size_t ApplicationMemoryBlockSlabHeapSize = 20000;
            static constexpr size_t SystemMemoryBlockSlabHeapSize      = 10000;
            static constexpr size_t BlockInfoSlabHeapSize              = 4000;
            static constexpr size_t ReservedDynamicPageCount           = 70;
        private:
            static State s_state;
            static KResourceLimit s_system_resource_limit;
            static KMemoryManager s_memory_manager;
            static KPageTableSlabHeap s_page_table_heap;
            static KMemoryBlockSlabHeap s_app_memory_block_heap;
            static KMemoryBlockSlabHeap s_sys_memory_block_heap;
            static KBlockInfoSlabHeap s_block_info_heap;
            static KPageTableManager s_app_page_table_manager;
            static KPageTableManager s_sys_page_table_manager;
            static KMemoryBlockSlabManager s_app_memory_block_manager;
            static KMemoryBlockSlabManager s_sys_memory_block_manager;
            static KBlockInfoManager s_app_block_info_manager;
            static KBlockInfoManager s_sys_block_info_manager;
            static KSupervisorPageTable s_supervisor_page_table;
            static KUnsafeMemory s_unsafe_memory;
            static KWorkerTaskManager s_worker_task_managers[KWorkerTaskManager::WorkerType_Count];
            static KInterruptManager s_interrupt_manager;
            static KScheduler s_schedulers[cpu::NumCores];
            static KInterruptTaskManager s_interrupt_task_managers[cpu::NumCores];
            static KHardwareTimer s_hardware_timers[cpu::NumCores];
        public:
            static NOINLINE void InitializeCoreLocalRegion(s32 core_id);
            static NOINLINE void InitializeMainAndIdleThreads(s32 core_id);
            static NOINLINE void InitializeResourceManagers(KVirtualAddress address, size_t size);
            static NOINLINE void PrintLayout();

            static ALWAYS_INLINE State GetState() { return s_state; }
            static ALWAYS_INLINE void SetState(State state) { s_state = state; }

            static KThread &GetMainThread(s32 core_id);
            static KThread &GetIdleThread(s32 core_id);

            static ALWAYS_INLINE KScheduler &GetScheduler() {
                return s_schedulers[GetCurrentCoreId()];
            }

            static ALWAYS_INLINE KScheduler &GetScheduler(s32 core_id) {
                return s_schedulers[core_id];
            }

            static ALWAYS_INLINE KInterruptTaskManager &GetInterruptTaskManager() {
                return s_interrupt_task_managers[GetCurrentCoreId()];
            }

            static ALWAYS_INLINE KInterruptManager &GetInterruptManager() {
                return s_interrupt_manager;
            }

            static ALWAYS_INLINE KHardwareTimer &GetHardwareTimer() {
                return s_hardware_timers[GetCurrentCoreId()];
            }

            static ALWAYS_INLINE KHardwareTimer &GetHardwareTimer(s32 core_id) {
                return s_hardware_timers[core_id];
            }

            static ALWAYS_INLINE KResourceLimit &GetSystemResourceLimit() {
                return s_system_resource_limit;
            }

            static ALWAYS_INLINE KMemoryManager &GetMemoryManager() {
                return s_memory_manager;
            }

            static ALWAYS_INLINE KMemoryBlockSlabManager &GetApplicationMemoryBlockManager() {
                return s_app_memory_block_manager;
            }

            static ALWAYS_INLINE KMemoryBlockSlabManager &GetSystemMemoryBlockManager() {
                return s_sys_memory_block_manager;
            }

            static ALWAYS_INLINE KBlockInfoManager &GetApplicationBlockInfoManager() {
                return s_app_block_info_manager;
            }

            static ALWAYS_INLINE KBlockInfoManager &GetSystemBlockInfoManager() {
                return s_sys_block_info_manager;
            }

            static ALWAYS_INLINE KPageTableManager &GetApplicationPageTableManager() {
                return s_app_page_table_manager;
            }

            static ALWAYS_INLINE KPageTableManager &GetSystemPageTableManager() {
                return s_sys_page_table_manager;
            }

            static ALWAYS_INLINE KSupervisorPageTable &GetKernelPageTable() {
                return s_supervisor_page_table;
            }

            static ALWAYS_INLINE KUnsafeMemory &GetUnsafeMemory() {
                return s_unsafe_memory;
            }

            static ALWAYS_INLINE KWorkerTaskManager &GetWorkerTaskManager(KWorkerTaskManager::WorkerType type) {
                MESOSPHERE_ASSERT(type <= KWorkerTaskManager::WorkerType_Count);
                return s_worker_task_managers[type];
            }
    };

    ALWAYS_INLINE KScheduler &GetCurrentScheduler() {
        return Kernel::GetScheduler();
    }

}