#pragma once

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

class RAM;

uint32_t hti(char);
uint32_t hToI(const char *, uint32_t);
void loadHexImpl(const char *, RAM *);

class RAM {
public:
  uint8_t *mem[1 << 12];

  RAM() {
    for (uint32_t i = 0; i < (1 << 12); i++)
      mem[i] = NULL;
  }
  ~RAM() {
    for (uint32_t i = 0; i < (1 << 12); i++)
      if (mem[i])
        delete[] mem[i];
  }

  size_t size() const {
    return (1ull << 32);
  }

  void clear() {
    for (uint32_t i = 0; i < (1 << 12); i++) {
      if (mem[i]) {
        delete mem[i];
        mem[i] = NULL;
      }
    }
  }

  uint8_t *get(uint32_t address) {

    if (mem[address >> 20] == NULL) {
      uint8_t *ptr = new uint8_t[1024 * 1024];
      for (uint32_t i = 0; i < 1024 * 1024; i += 4) {
        ptr[i + 0] = 0x00;
        ptr[i + 1] = 0x00;
        ptr[i + 2] = 0x00;
        ptr[i + 3] = 0x00;
      }
      mem[address >> 20] = ptr;
    }
    return &mem[address >> 20][address & 0xFFFFF];
  }

  void read(uint32_t address, uint32_t length, uint8_t *data) {
    for (unsigned i = 0; i < length; i++) {
      data[i] = (*this)[address + i];
    }
  }

  void write(uint32_t address, uint32_t length, uint8_t *data) {
    for (unsigned i = 0; i < length; i++) {
      (*this)[address + i] = data[i];
    }
  }

  void getBlock(uint32_t address, uint8_t *data) {
    uint32_t block_number = address & 0xffffff00; // To zero out block offset
    uint32_t bytes_num = 256;

    this->read(block_number, bytes_num, data);
  }

  void getWord(uint32_t address, uint32_t *data) {
    data[0] = 0;

    uint8_t first = *get(address + 0);
    uint8_t second = *get(address + 1);
    uint8_t third = *get(address + 2);
    uint8_t fourth = *get(address + 3);

    data[0] = (data[0] << 0) | fourth;
    data[0] = (data[0] << 8) | third;
    data[0] = (data[0] << 8) | second;
    data[0] = (data[0] << 8) | first;
  }

  void writeWord(uint32_t address, uint32_t *data) {
    uint32_t data_to_write = *data;

    uint32_t byte_mask = 0xFF;

    for (int i = 0; i < 4; i++) {
      (*this)[address + i] = data_to_write & byte_mask;
      data_to_write = data_to_write >> 8;
    }
  }

  void writeHalf(uint32_t address, uint32_t *data) {
    uint32_t data_to_write = *data;

    uint32_t byte_mask = 0xFF;

    for (int i = 0; i < 2; i++) {
      (*this)[address + i] = data_to_write & byte_mask;
      data_to_write = data_to_write >> 8;
    }
  }

  void writeByte(uint32_t address, uint32_t *data) {
    uint32_t data_to_write = *data;

    uint32_t byte_mask = 0xFF;

    (*this)[address] = data_to_write & byte_mask;
    data_to_write = data_to_write >> 8;
  }

  uint8_t &operator[](uint32_t address) {
    return *get(address);
  }
};

// MEMORY UTILS

inline uint32_t hti(char c) {
  if (c >= 'A' && c <= 'F')
    return c - 'A' + 10;
  if (c >= 'a' && c <= 'f')
    return c - 'a' + 10;
  return c - '0';
}

inline uint32_t hToI(const char *c, uint32_t size) {
  uint32_t value = 0;
  for (uint32_t i = 0; i < size; i++) {
    value += hti(c[i]) << ((size - i - 1) * 4);
  }
  return value;
}

inline void loadHexImpl(const char *path, RAM *mem) {
  mem->clear();
  FILE *fp = fopen(path, "r");
  if (fp == 0) {
    printf("Path not found %s\n", path);
    return;
    // std::cout << path << " not found" << std::endl;
  }
  //Preload 0x0 <-> 0x80000000 jumps
  ((uint32_t *)mem->get(0))[1] = 0xf1401073;

  ((uint32_t *)mem->get(0))[2] = 0x30101073;

  ((uint32_t *)mem->get(0))[3] = 0x800000b7;
  ((uint32_t *)mem->get(0))[4] = 0x000080e7;

  ((uint32_t *)mem->get(0x80000000))[0] = 0x00000097;

  ((uint32_t *)mem->get(0xb0000000))[0] = 0x01C02023;
  // F00FFF10
  ((uint32_t *)mem->get(0xf00fff10))[0] = 0x12345678;

  fseek(fp, 0, SEEK_END);
  uint32_t size = ftell(fp);
  fseek(fp, 0, SEEK_SET);
  char *content = new char[size];
  fread(content, 1, size, fp);

  int offset = 0;
  char *line = content;
  // std::cout << "WHTA\n";
  while (1) {
    if (line[0] == ':') {
      uint32_t byteCount = hToI(line + 1, 2);
      uint32_t nextAddr = hToI(line + 3, 4) + offset;
      uint32_t key = hToI(line + 7, 2);
      switch (key) {
      case 0:
        for (uint32_t i = 0; i < byteCount; i++) {

          unsigned add = nextAddr + i;

          *(mem->get(add)) = hToI(line + 9 + i * 2, 2);
        }
        break;
      case 2:
        //              cout << offset << endl;
        offset = hToI(line + 9, 4) << 4;
        break;
      case 4:
        //              cout << offset << endl;
        offset = hToI(line + 9, 4) << 16;
        break;
      default:
        //              cout << "??? " << key << endl;
        break;
      }
    }

    while (*line != '\n' && size != 0) {
      line++;
      size--;
    }
    if (size <= 1)
      break;
    line++;
    size--;
  }

  if (content)
    delete[] content;
}