/*
 * Copyright (c) 2014-2015 Travis Geiselbrecht
 *
 * Use of this source code is governed by a MIT-style
 * license that can be found in the LICENSE file or at
 * https://opensource.org/licenses/MIT
 */
#include "include/dev/uart/pl011.h"

#include <assert.h>
#include <lk/trace.h>
#include <lib/cbuf.h>
#include <lib/io.h>
#include <dev/uart.h>
#include <kernel/thread.h>
#include <platform/interrupts.h>

// PL011 UART driver

// PL011 registers
enum pl011_regs {
    UART_DR    = 0x00,
    UART_RSR   = 0x04,
    UART_TFR   = 0x18,
    UART_ILPR  = 0x20,
    UART_IBRD  = 0x24,
    UART_FBRD  = 0x28,
    UART_LCRH  = 0x2c,
    UART_CR    = 0x30,
    UART_IFLS  = 0x34,
    UART_IMSC  = 0x38,
    UART_TRIS  = 0x3c,
    UART_TMIS  = 0x40,
    UART_ICR   = 0x44,
    UART_DMACR = 0x48
};

// Control register bits that are used in the driver
#define UART_TFR_RXFE (1<<4)
#define UART_TFR_TXFF (1<<5)
#define UART_TFR_RXFF (1<<6)
#define UART_TFR_TXFE (1<<7)

#define UART_IMSC_RXIM (1<<4)

#define UART_TMIS_RXMIS (1<<4)

#define UART_CR_UARTEN (1<<0)
#define UART_CR_TXEN (1<<8)
#define UART_CR_RXEN (1<<9)

// TODO: have these be configurable
#define RXBUF_SIZE 32
#define NUM_UART 1

struct pl011_struct {
    struct pl011_config config;

    cbuf_t uart_rx_buf;
};

static struct pl011_struct uart[NUM_UART];

static inline void write_uart_reg(uintptr_t base, size_t offset, uint32_t val) {
    mmio_write32((uint32_t *)(base + offset), val);
}

static inline uint32_t read_uart_reg(uintptr_t base, size_t offset) {
    return mmio_read32((uint32_t *)(base + offset));
}

static inline void set_uart_reg_bits(uintptr_t base, size_t offset, uint32_t bits) {
    write_uart_reg(base, offset, read_uart_reg(base, offset) | bits);
}

static inline void clear_uart_reg_bits(uintptr_t base, size_t offset, uint32_t bits) {
    write_uart_reg(base, offset, read_uart_reg(base, offset) & ~bits);
}

static inline uintptr_t uart_to_ptr(unsigned int n) {
    DEBUG_ASSERT(n < NUM_UART);
    return uart[n].config.base;
}

static enum handler_return uart_irq(void *arg) {
    struct pl011_struct *u = (struct pl011_struct *)arg;

    /* read interrupt status and mask */
    uint32_t isr = read_uart_reg(u->config.base, UART_TMIS);

    bool resched = false;
    if (isr & UART_TMIS_RXMIS) { // rxmis
        cbuf_t *rxbuf = &uart->uart_rx_buf;

        /* while fifo is not empty, read chars out of it */
        while ((read_uart_reg(u->config.base, UART_TFR) & UART_TFR_RXFE) == 0) {
#if CONSOLE_HAS_INPUT_BUFFER
            if (u->config.flag & PL011_FLAG_DEBUG_UART) {
                char c = read_uart_reg(u->config.base, UART_DR);
                cbuf_write_char(&console_input_cbuf, c, false);
            } else
#endif
            {
                /* if we're out of rx buffer, mask the irq instead of handling it */
                if (cbuf_space_avail(rxbuf) == 0) {
                    clear_uart_reg_bits(u->config.base, UART_IMSC, UART_IMSC_RXIM); // !rxim
                    break;
                }

                char c = read_uart_reg(u->config.base, UART_DR);
                cbuf_write_char(rxbuf, c, false);
            }

            resched = true;
        }
    }

    return resched ? INT_RESCHEDULE : INT_NO_RESCHEDULE;
}

void pl011_init(int port) {
    uintptr_t base = uart_to_ptr(port);

    // create circular buffer to hold received data
    cbuf_initialize(&uart[port].uart_rx_buf, RXBUF_SIZE);

    // assumes interrupts are contiguous
    register_int_handler(uart[port].config.irq, &uart_irq, (void *)&uart[port]);

    // clear all irqs
    write_uart_reg(base, UART_ICR, 0x3ff);

    // set fifo trigger level
    write_uart_reg(base, UART_IFLS, 0); // 1/8 rxfifo, 1/8 txfifo

    // enable rx interrupt
    write_uart_reg(base, UART_IMSC, UART_IMSC_RXIM); // rxim

    // enable receive
    set_uart_reg_bits(base, UART_CR, UART_CR_RXEN); // rxen

    // enable interrupt
    unmask_interrupt(uart[port].config.irq);
}

void pl011_init_early(int port, const struct pl011_config *config) {
    if (port >= NUM_UART) {
        return;
    }

    // TODO: validate config
    uart[port].config = *config;

    write_uart_reg(uart[port].config.base, UART_CR, UART_CR_TXEN | UART_CR_UARTEN); // tx_enable, uarten
}

int uart_putc(int port, char c) {
    uintptr_t base = uart_to_ptr(port);

    /* spin while fifo is full */
    while (read_uart_reg(base, UART_TFR) & UART_TFR_TXFF)
        ;
    write_uart_reg(base, UART_DR, c);

    return 1;
}

int uart_getc(int port, bool wait) {
    cbuf_t *rxbuf = &uart[port].uart_rx_buf;

    char c;
    if (cbuf_read_char(rxbuf, &c, wait) == 1) {
        set_uart_reg_bits(uart[port].config.base, UART_IMSC, UART_IMSC_RXIM); // rxim
        return c;
    }

    return -1;
}

/* panic-time getc/putc */
int uart_pputc(int port, char c) {
    uintptr_t base = uart_to_ptr(port);

    /* spin while fifo is full */
    while (read_uart_reg(base, UART_TFR) & UART_TFR_TXFF)
        ;
    write_uart_reg(base, UART_DR, c);

    return 1;
}

int uart_pgetc(int port) {
    uintptr_t base = uart_to_ptr(port);

    if ((read_uart_reg(base, UART_TFR) & UART_TFR_RXFE) == 0) {
        unsigned char c = read_uart_reg(base, UART_DR);
        return c;
    } else {
        return -1;
    }
}

void uart_flush_tx(int port) {
}

void uart_flush_rx(int port) {
}

// TODO collapse this into the early/regular init routines
void uart_init_port(int port, uint baud) {
}