#include "pit.h"
#include "io.h"
#include "irq.h"
#include "klibc.h"
#include "kthread.h"
#include "time.h"

int pitSetup(unsigned int freq)
{
    unsigned int divisor = PIT_FREQ / freq;
    if (divisor > 65535)
        divisor = 0;     // Used to represent 35536
    outb(PIT_CMD, 0x34); // chan 0; low then high; mode 2
    outb(PIT_CHAN_0, divisor & 0xFF);
    outb(PIT_CHAN_0, divisor >> 8u);

    return 0;
}

struct cpu_state *pitIrqHandler(struct cpu_state *prevCpu)
{
    __atomic_add_fetch(&jiffies, 1, __ATOMIC_RELAXED);
    kthreadOnJieffiesTick();
    // Uncomment for non-preemptible kernel
    // return prevCpu;
    return kthreadSwitch(prevCpu);
}