#include <stdlib.h>
#include <stdint.h>
#include <unistd.h>
#include <stdio.h>

#include "encoding.h"
#include "platform.h"
#include "plic_driver.h"

#include "eos.h"
#include "interrupt.h"

// Global Instance data for the PLIC
// for use by the PLIC Driver.
static plic_instance_t plic;

static eos_intr_handler_t ext_interrupt_handler[PLIC_NUM_INTERRUPTS];

uintptr_t eos_intr_handle(uintptr_t int_num) {
    if ((int_num >=1) && (int_num <= PLIC_NUM_INTERRUPTS) && (ext_interrupt_handler[int_num-1])) {
        ext_interrupt_handler[int_num-1]();
    } else {
        printf("INTR ERROR:%d\n", int_num);
        exit(int_num);
    }
    return int_num;
}

int eos_intr_init(uint8_t wakeup_cause) {
    for (int i = 0; i < PLIC_NUM_INTERRUPTS; i++){
        ext_interrupt_handler[i] = NULL;
    }

    /**************************************************************************
    * Set up the PLIC
    **************************************************************************/
    PLIC_init(&plic,
 	    PLIC_CTRL_ADDR,
 	    PLIC_NUM_INTERRUPTS,
 	    PLIC_NUM_PRIORITIES);

    // Enable Global (PLIC) interrupts.
    set_csr(mie, MIP_MEIP);

    // Enable all interrupts
    set_csr(mstatus, MSTATUS_MIE);

    return EOS_OK;
}

void eos_intr_set(uint8_t int_num, uint8_t priority, eos_intr_handler_t handler) {
    ext_interrupt_handler[int_num-1] = handler;
    PLIC_set_priority(&plic, int_num, priority);
    PLIC_enable_interrupt(&plic, int_num);
}

void eos_intr_set_handler(uint8_t int_num, eos_intr_handler_t handler) {
    ext_interrupt_handler[int_num-1] = handler;
}

void eos_intr_set_priority(uint8_t int_num, uint8_t priority) {
    PLIC_set_priority(&plic, int_num, priority);
}

void eos_intr_enable(uint8_t int_num) {
    PLIC_enable_interrupt(&plic, int_num);
}

void eos_intr_disable(uint8_t int_num) {
    PLIC_disable_interrupt(&plic, int_num);
}

void eos_intr_mask(uint8_t priority) {
    PLIC_set_threshold(&plic, priority);
}