#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <math.h>
#include <time.h>

#ifdef _WIN32
#include <winsock.h>
#else 
#include <arpa/inet.h>
#include <unistd.h>
#endif

#include "util.h"
#include "chip8.h"

chip8_vm vm;

static u8 font[] = {
  0xF0, 0x90, 0x90, 0x90, 0xF0, // 0
  0x20, 0x60, 0x20, 0x20, 0x70, // 1
  0xF0, 0x10, 0xF0, 0x80, 0xF0, // 2
  0xF0, 0x10, 0xF0, 0x10, 0xF0, // 3
  0x90, 0x90, 0xF0, 0x10, 0x10, // 4
  0xF0, 0x80, 0xF0, 0x10, 0xF0, // 5
  0xF0, 0x80, 0xF0, 0x90, 0xF0, // 6
  0xF0, 0x10, 0x20, 0x40, 0x40, // 7
  0xF0, 0x90, 0xF0, 0x90, 0xF0, // 8
  0xF0, 0x90, 0xF0, 0x10, 0xF0, // 9
  0xF0, 0x90, 0xF0, 0x90, 0x90, // A
  0xE0, 0x90, 0xE0, 0x90, 0xE0, // B
  0xF0, 0x80, 0x80, 0x80, 0xF0, // C
  0xE0, 0x80, 0x80, 0x80, 0xE0, // D
  0xF0, 0x80, 0xF0, 0x80, 0xF0, // E
  0xF0, 0x80, 0xF0, 0x80, 0x80  // F
};

void init_vm() {
  vm.PC = 0x200; // start of chip8 programs
}

void load_rom(char *file_path) {
  memset(&vm, 0, sizeof(chip8_vm));
  int fd = open(file_path, O_RDONLY);
  FILE* f = fdopen(fd, "r");
  fseek(f, 0, SEEK_END);
  off_t fsize = ftell(f);
  rewind(f);

  vm.rom = (u8*)malloc(0xFFF);

  memset(vm.rom, 0, 0xFFF);
  
#ifdef _WIN32
  size_t res = read(fd, &vm.rom[0x200], fsize);
#else
  size_t res = fread(&vm.rom[0x200], 1, fsize, f);
#endif

  if (res != fsize) {
    printf("err: failed to read rom fully\n");
    exit(1);
  }

  fclose(f); close(fd);
}

void vm_step() {
  // read 2 bytes and convert to big-endian
  u16 in = htons(vm.rom[vm.PC + 0x1]<<8 | vm.rom[vm.PC]);

  // in>>12&0xF  0xFFFF
  //               ^
  // in>>8&0xF   0xFFFF
  //                ^
  // in>>4&0xF   0xFFFF
  //                 ^
  // in&0xF      0xFFFF
  //                  ^
  // in&0xFFF    0xFFFF
  //                ^^^
  // in&0xFF     0xFFFF
  //                 ^^
  // in&0xFF     0xFFFF
  //                 ^^

  switch ((in>>12&0xF)) {
    case 0x0: 
      switch (in&0xFF) {
        case 0xE0: // CLS
          memset(&vm.screen, 0, sizeof(vm.screen));
          break;
        case 0xEE: // RET
          vm.PC = vm.stack[vm.SP--];
          break;
      } 
      break;
    case 0x1: // JP addr
      vm.PC = (in&0xFFF) - 0x2;
      break;
    case 0x2: // CALL addr
      vm.stack[++vm.SP] = vm.PC;
      vm.PC = (in&0xFFF) - 0x2;
      break;
    case 0x3: // SE Vx, byte
      if (vm.Vx[in>>8&0xF] == (in&0xFF)) {
        vm.PC += 0x2;
      }
      break;
    case 0x4: // SNE Vx, byte
      if (vm.Vx[in>>8&0xF] != (in&0xFF)) {
        vm.PC += 0x2;
      }
      break;
    case 0x5: // SE Vx, Vy
      if (vm.Vx[in>>8&0xF] == vm.Vx[in>>4&0xF]) {
        vm.PC += 0x2;
      }
      break;
    case 0x6: // LD Vx, byte
      vm.Vx[in>>8&0xF] = in&0xFF;
      break;
    case 0x7: // ADD Vx, byte
      vm.Vx[in>>8&0xF] += in&0xFF;
      break;
    case 0x8: 
      switch(in&0xF) {
        case 0x0: // LD Vx, Vy 
          vm.Vx[in>>8&0xF] = vm.Vx[in>>4&0xF];
          break;
        case 0x1: // OR Vx, Vy 
          vm.Vx[in>>8&0xF] |= vm.Vx[in>>4&0xF];
          break;
        case 0x2: // AND Vx, Vy
          vm.Vx[in>>8&0xF] &= vm.Vx[in>>4&0xF];
          break;
        case 0x3: // XOR Vx, Vy 
          vm.Vx[in>>8&0xF] ^= vm.Vx[in>>4&0xF];
          break;
        case 0x4: // ADD Vx, Vy 
          if ((vm.Vx[in>>8&0xF] + vm.Vx[in>>4&0xF]) > 0xFF) {
            vm.Vx[0xF] = 1;
          } else {
            vm.Vx[0xF] = 0;
          }
          vm.Vx[in>>8&0xF] = (vm.Vx[in>>8&0xF] + vm.Vx[in>>4&0xF])&0xFF;
          break;
        case 0x5: // SUB Vx, Vy 
          if (vm.Vx[in>>8&0xF] > vm.Vx[in>>4&0xF]) {
            vm.Vx[0xF] = 1;
          } else {
            vm.Vx[0xF] = 0;
          }
          vm.Vx[in>>8&0xF] -= vm.Vx[in>>4&0xF];
          break;
        case 0x6: // SHR Vx {, Vy}
          vm.Vx[0xF] = (vm.Vx[in>>8&0xF] & 1); 
          // TODO: toggle for Vx /= 2
          vm.Vx[in>>8&0xF] = vm.Vx[in>>8&0xF]>>1;
          break;
        case 0x7: // SUBN Vx, Vy
          if (vm.Vx[in>>8&0xF] < vm.Vx[in>>4&0xF]) {
            vm.Vx[0xF] = 1;
          } else {
            vm.Vx[0xF] = 0;
          }
          vm.Vx[in>>8&0xF] = vm.Vx[in>>4&0xF] - vm.Vx[in>>8&0xF];
          break;
        case 0xE: // SHL Vx {, Vy} 
          vm.Vx[0xF] = ((vm.Vx[in>>8&0xF]>>7) & 1);
          // TODO: toggle for Vx *= 2;
          vm.Vx[in>>8&0xF] = vm.Vx[in>>8&0xF]<<1;
          break;
      }
      break;
    case 0x9: // SNE Vx, Vy 
      if (vm.Vx[(in>>8&0xF)] != vm.Vx[(in>>4&0xF)]) {
        vm.PC += 0x2;
      }
      break;
    case 0xA: // LD I, addr
      vm.I = in&0xFFF;
      break;
    case 0xB: // JP V0, addr
      vm.PC = ((in&0xFFF) + vm.Vx[0x0]) - 0x2;
      break;
    case 0xC: // RND Vx, byte
      vm.Vx[in>>8&0xF] = (rand() % 255) & in&0xFF;
      break;
    case 0xD: // DRW Vx, Vy, nibble
      vm.Vx[0xF] = 0;
      for (u32 i = 0; i < (in&0xF); i++) {
        for (u32 s = 0; s < 8; s++) {
          u8 pix = (vm.rom[vm.I + i] & (0x80 >> s)) != 0;

          if (pix) {
            u8 y = (vm.Vx[in>>4&0xF] + i)&0x1F;
            u8 x = (vm.Vx[in>>8&0xF] + s)&0x3F;
 
            u32 byte = y * 64 + x;
            u32 bit = 1 << (byte & 0x07);

            if (vm.screen[byte>>3] & bit) {
              vm.Vx[0xF] = 1;
            }

            vm.screen[byte>>3] ^= bit;
          }
        }
      }
      break;
    case 0xE: switch (in&0xFF) {
        case 0x9E: // SKP Vx
          if (vm.keyboard[vm.Vx[in>>8&0xF]]) {
            vm.PC += 0x2;
          }
          break;
        case 0xA1: // SKNP Vx
          if (!vm.keyboard[vm.Vx[in>>8&0xF]]) {
            vm.PC += 0x2;
          }
          break;
      }
      break;
    case 0xF:
      switch (in&0xFF) {
        case 0x07: // LD Vx, DT 
          vm.Vx[in>>8&0xF] = vm.DT;
          break;
        case 0x0A: // LD Vx, K
          //vm.Vx[in>>8&0xF] = 0xA;//ui_get_key(true);
          break;
        case 0x15: // LD DT, Vx
          vm.DT = vm.Vx[in>>8&0xF];
          break;
        case 0x18: // LD ST, Vx
          vm.ST = vm.Vx[in>>8&0xF];
          break;
        case 0x1E: // ADD I, Vx
          vm.I += vm.Vx[in>>8&0xF];
          break;
        case 0x29: // LD F, Vx
          // characters
          break;
        case 0x33: // LD B, Vx
          vm.rom[vm.I] = (vm.Vx[in>>8&0xF] / 100);
          vm.rom[vm.I+0x1] = ((vm.Vx[in>>8&0xF] % 100)/10);
          vm.rom[vm.I+0x2] = (vm.Vx[in>>8&0xF] % 10);
          break;
        case 0x55: // LD [I], Vx
          for (u8 i = 0; i <= (in>>8&0xF); i++) {
            vm.rom[(vm.I) + i] = vm.Vx[i];
          }
          break;
        case 0x65: // LD Vx, [I]
          for (u8 i = 0; i <= (in>>8&0xF); i++) {
            vm.Vx[i]  = vm.rom[(vm.I) + i];
          }
          break;
      }
      break;
  }

  vm.PC += 0x2;
}