// MIT License, Copyright (c) 2020 Marvin Borner

#include <cpu.h>
#include <def.h>
#include <interrupts.h>
#include <proc.h>
#include <timer.h>

static u32 timer_ticks = 0;
static u8 call_scheduler = 0;

static void timer_phase(int hz)
{
	int divisor = 3579545 / 3 / hz;
	outb(0x43, 0x36); // 01 10 11 0b // CTR, RW, MODE, BCD
	outb(0x40, (u8)(divisor & 0xFF));
	outb(0x40, (u8)(divisor >> 8));
}

u32 timer_get(void)
{
	return timer_ticks;
}

void timer_handler(struct regs *r)
{
	if (timer_ticks >= U32_MAX)
		timer_ticks = 0;
	else
		timer_ticks++;

	if (call_scheduler)
		scheduler(r);
}

// "Delay" function with CPU sleep
void timer_wait(u32 ticks)
{
	u32 eticks = timer_ticks + ticks;
	while (timer_ticks < eticks) {
		__asm__ volatile("sti//hlt//cli");
	}
}

// Install timer handler into IRQ0
void timer_install(void)
{
	/* hpet_install(10000); // TODO: Find optimal femtosecond period */
	/* if (!hpet) */
	timer_phase(1000);
	irq_install_handler(0, timer_handler);
}

void scheduler_enable(void)
{
	call_scheduler = 1;
	irq_install_handler(0, timer_handler);
}

void scheduler_disable(void)
{
	call_scheduler = 0;
	irq_install_handler(0, timer_handler);
}