#pragma once

#include "types.hpp"
#include "asm.hpp"

/**
 * Current number of page (from the beginning of the ram) used by the kernel that
 * should not be used by the paging allocation mechanism and should not be granted
 * for allocation
 */
/// @brief New number of page reachable at the end of the paging_enable() call
#define MAX_TABLES 280
#define PAGING_MAX_PAGE (20*512)
#define PAGE_ALLOCATE() paging_allocate_contiguous(1)

/// @brief Options
#define PAGING_OPT_P 1
#define PAGING_OPT_RW (1<<1)
#define PAGING_OPT_PCD (1<<3)

/// @brief Get page address that contain addr
#define PAGE(addr) (addr&(~(0xFFF)))
#define VIRT(addr) ((u64*)(((u64)addr)+kvar_kernel_vma))

/// @brief Mapping facilities 
#define PAGE_ID_MAP(addr) paging_allocate_addr(kpages[0],(u64)(addr),(u64)(addr),PAGING_OPT_P|PAGING_OPT_RW,1)
#define PAGE_ID_RMAP(addr, n) { \
    for(u64 i=0;i<(n);i+=4096){ \
        paging_allocate_addr(kpages[0],((u64)(addr))+i,((u64)(addr))+i,PAGING_OPT_P|PAGING_OPT_RW,1); \
}}
#define PAGE_MAP(addr) paging_allocate_addr(kpages[0],(u64)((addr)+kvar_kernel_vma),(u64)(addr),PAGING_OPT_P|PAGING_OPT_RW,1)
#define PAGE_RMAP(addr, n) { \
    for(u64 i=0;i<(n);i+=4096){ \
        paging_allocate_addr(kpages[0],(((u64)(addr))+kvar_kernel_vma)+i,((u64)(addr))+i,PAGING_OPT_P|PAGING_OPT_RW,1); \
}}
#define PAGE_MAP_PHY(addr,phy) paging_allocate_addr(kpages[0],(u64)((addr)+kvar_kernel_vma),(u64)(phy),PAGING_OPT_P|PAGING_OPT_RW,1)
#define PAGE_RMAP_PHY(addr,phy, n) { \
    for(u64 i=0;i<(n);i+=4096){ \
        paging_allocate_addr(kpages[0],(((u64)(addr))+kvar_kernel_vma)+i,((u64)(phy))+i,PAGING_OPT_P|PAGING_OPT_RW,1); \
}}

/// @brief All PAE table structures are allocated here
extern u64 kpages[MAX_TABLES][512];

/// CF boucane.hpp
extern u64 kvar_kernel_vma,kvar_stack_pma,kvar_userspace_pma;
extern void (*printk)(char *str,...);

/**
 * Setup and enable PAE paging
 */
void paging_enable();

/**
 * Allocate the next available page
 * and return its physical address
 */
u64* paging_allocate_contiguous(int npages);

/**
 * Deallocate a page located at addr
 */
void paging_deallocate(u64 addr);

/**
 * Dump a specific range of bytes in the paging_status
 */
void paging_dump(int min, int max);

/**
 * Deallocate all the pages linked to a pml4
 */
void paging_deallocate_pml4(u64* pml4);

/**
 * Deallocate all the pages related to a pml4 structure
 */
void paging_deallocate_table(u64* table);

/**
 * Map virtual page associated to virt
 * to the physical page associated with phy
 */
void paging_allocate_addr(u64* pml4_table,u64 virt, u64 phy, u16 options, char useKernelTables);


u64* paging_create_task(int npages);