#include <lib.h>
#include <memory.h>
#include <stdint.h>
#include <panic.h>
#include <pci.h>

#include "bindings.h"

#define INDEX           0x1CE
#define DATA            0x1CF

#define INDEX_ID            0
#define INDEX_XRES          1
#define INDEX_YRES          2
#define INDEX_BPP           3
#define INDEX_ENABLE        4
#define INDEX_BANK          5
#define INDEX_VIRT_WIDTH    6
#define INDEX_VIRT_HEIGHT   7
#define INDEX_X_OFFSET      8
#define INDEX_Y_OFFSET      9

#define DATA_DISP_DISABLE   0x00
#define DATA_DISP_ENABLE    0x01
#define DATA_LFB_ENABLE     0x40
#define DATA_NO_CLEAR_MEM   0x80

#define BOCHS_PCI_VENDOR    0x1234
#define BOCHS_PCI_DEVICE    0x1111

static void write(uint16_t index, uint16_t data) {
    outw(INDEX, index);
    outw(DATA, data);
}

static uint16_t read(uint16_t value) {
    outw(INDEX, value);
    return inw(DATA);
}

static int is_available(void) {
    return (read(INDEX_ID) == 0xB0C5);
}

static void set_mode(uint16_t width, uint16_t height, uint16_t bit_depth, int lfb, int clear) {
    write(INDEX_ENABLE, DATA_DISP_DISABLE);
    write(INDEX_XRES, width);
    write(INDEX_YRES, height);
    write(INDEX_BPP, bit_depth);
    write(INDEX_ENABLE, DATA_DISP_ENABLE |
        (lfb ? DATA_LFB_ENABLE : 0) |
        (clear ? 0 : DATA_NO_CLEAR_MEM));
}

//static void set_bank(uint16_t bank) {
//    write(INDEX_BANK, bank);
//}

int fb_init(uint16_t width, uint16_t height) {

	set_mode(width, height, 32, true, true);

	if (!is_available())
		panic("bochs framebuffer not avaliable");

    struct pci_device bochs = {0};
    bool found = pci_findby_id(&bochs, BOCHS_PCI_DEVICE, BOCHS_PCI_VENDOR, NULL);
    if (!found)
        panic("bochs pci device not avaliable");

    uint32_t bar0 = pci_rcfg_d(bochs, PCI_BAR0_D);
    uint32_t *fb = (uint32_t*) (uintptr_t) bar0;
    fb = mmap(fb, width * height * 4);

	for (uint16_t y = 0; y < height; y++) {
		for (uint16_t x = 0; x < width; x++) {
            double dx = x / (double) width,
                   dy = y / (double) height,
                   dz = (2 - dx - dy) / 2;
            uint32_t r = (uint32_t)(dx * 255),
                     g = (uint32_t)(dy * 255),
                     b = (uint32_t)(dz * 255);
			fb[x + y * width] = (b << 0) | (g << 8) | (r << 16);
		}
	}


    return 0;
}