#include "Runtime/CGameAllocator.hpp"

#include <logvisor/logvisor.hpp>

namespace metaforce {
static logvisor::Module Log("metaforce::CGameAllocator");

#pragma GCC diagnostic ignored "-Wclass-memaccess"

std::vector<CGameAllocator::SAllocationDescription> CGameAllocator::m_allocations;

u8* CGameAllocator::Alloc(size_t len) {
  size_t roundedLen = ROUND_UP_64(len + sizeof(SChunkDescription));
  for (SAllocationDescription& alloc : m_allocations) {
    /* We need to supply enough room for allocation information */
    if (alloc.freeOffset + roundedLen < alloc.allocSize) {
      u8* ptr = alloc.memptr.get() + alloc.freeOffset;
      SChunkDescription* chunkInfo = reinterpret_cast<SChunkDescription*>(ptr);
      *chunkInfo = SChunkDescription();
      chunkInfo->parent = &alloc;
      chunkInfo->len = len;
      alloc.freeOffset += roundedLen;
      return ptr + sizeof(SChunkDescription);
    }
  }

  /* 1MiB minimum allocation to prevent constantly allocating small amounts of memory */
  size_t allocSz = len;
  if (allocSz < (1 * 1024 * 1024 * 1024))
    allocSz = 1 * 1024 * 1024 * 1024;

  /* Pad size to allow for allocation information */
  allocSz = ROUND_UP_64(allocSz + sizeof(SChunkDescription));
  auto& alloc = m_allocations.emplace_back();
  alloc.memptr.reset(new u8[allocSz]);
  u8* ptr = alloc.memptr.get();
  alloc.allocSize = allocSz;
  alloc.freeOffset += roundedLen;
  SChunkDescription* chunkInfo = reinterpret_cast<SChunkDescription*>(ptr);
  *chunkInfo = SChunkDescription();
  chunkInfo->parent = &alloc;
  chunkInfo->len = len;
  return ptr + sizeof(SChunkDescription);
}

void CGameAllocator::Free(u8* ptr) {
  SChunkDescription* info = reinterpret_cast<SChunkDescription*>(ptr - sizeof(SChunkDescription));
  if (info->magic != 0xE8E8E8E8 || info->sentinal != 0xEFEFEFEF) {
    Log.report(logvisor::Fatal, FMT_STRING("Invalid chunk description, memory corruption!"));
    return;
  }

  SAllocationDescription& alloc = *info->parent;
  size_t roundedLen = ROUND_UP_32(info->len + sizeof(SChunkDescription));
  alloc.freeOffset -= roundedLen;
  /* Invalidate chunk allocation descriptor */
  memset(info, 0, ROUND_UP_64(info->len + sizeof(SChunkDescription)));
}
} // namespace metaforce