#include "acpi.hpp"

#include <memory.hpp>
#include <debug.h>

#include "../../kernel.h"

namespace ACPI
{
    MADT::MADT(ACPI::MADTHeader *madt)
    {
        trace("Initializing MADT");
        CPUCores = 0;
        LAPICAddress = (LAPIC *)(uintptr_t)madt->LocalControllerAddress;
        for (uint8_t *ptr = (uint8_t *)(madt->Entries);
             (uintptr_t)(ptr) < (uintptr_t)(madt) + madt->Header.Length;
             ptr += *(ptr + 1))
        {
            switch (*(ptr))
            {
            case 0:
            {
                if (ptr[4] & 1)
                {
                    lapic.push_back((LocalAPIC *)ptr);
                    KPrint("Local APIC \e8888FF%d\eCCCCCC (APIC \e8888FF%d\eCCCCCC) found.", lapic.back()->ACPIProcessorId, lapic.back()->APICId);
                    CPUCores++;
                }
                break;
            }
            case 1:
            {
                ioapic.push_back((MADTIOApic *)ptr);
                KPrint("I/O APIC \e8888FF%d\eCCCCCC (Address \e8888FF%#lx\eCCCCCC) found.", ioapic.back()->APICID, ioapic.back()->Address);
                Memory::Virtual().Map((void *)(uintptr_t)ioapic.back()->Address, (void *)(uintptr_t)ioapic.back()->Address, Memory::PTFlag::RW | Memory::PTFlag::PCD); // Make sure that the address is mapped.
                break;
            }
            case 2:
            {
                iso.push_back((MADTIso *)ptr);
                KPrint("ISO (IRQ:\e8888FF%#lx\eCCCCCC, BUS:\e8888FF%#lx\eCCCCCC, GSI:\e8888FF%#lx\eCCCCCC, %s\eCCCCCC/%s\eCCCCCC) found.",
                       iso.back()->IRQSource, iso.back()->BuSSource, iso.back()->GSI,
                       iso.back()->Flags & 0x00000004 ? "\e1770FFActive High" : "\e475EFFActive Low",
                       iso.back()->Flags & 0x00000100 ? "\e00962DEdge Triggered" : "\e008F58Level Triggered");
                break;
            }
            case 4:
            {
                nmi.push_back((MADTNmi *)ptr);
                KPrint("NMI \e8888FF%#lx\eCCCCCC (lint:\e8888FF%#lx\eCCCCCC) found.", nmi.back()->processor, nmi.back()->lint);
                break;
            }
            case 5:
            {
                LAPICAddress = (LAPIC *)ptr;
                KPrint("APIC found at \e8888FF%#lx\eCCCCCC", LAPICAddress);
                break;
            }
            }
            Memory::Virtual().Map((void *)LAPICAddress, (void *)LAPICAddress, Memory::PTFlag::RW | Memory::PTFlag::PCD); // I should map more than one page?
        }
        CPUCores--; // We start at 0 (BSP) and end at 11 (APs), so we have 12 cores.
        KPrint("Total CPU cores: %d", CPUCores + 1);
    }

    MADT::~MADT()
    {
    }
}