Add Memory debugger

source/amethyst.cpp

@@ -1,6 +1,10 @@
 #include <amethyst.hpp>
 #include <exception>
 #include <iostream>
+#include <map>
+#include <vector>
+
+#define TRACK_NEW_DELETE
 
 #ifdef AMY_3DS
 #include <3ds.h>
@@ -11,7 +15,7 @@ void NpiD7CxxExceptionHandler() {
   std::cout << "[C++ Error Handler]" << std::endl;
   try {
     std::rethrow_exception(std::current_exception());
-  } catch (const std::exception& e) {
+  } catch (const std::exception &e) {
     std::cout << "\n\n" << e.what() << std::endl;
   }
   aptSetHomeAllowed(false);
@@ -26,6 +30,60 @@ void NpiD7CxxExceptionHandler() {
 }
 #endif
 
+static Amy::Memory::MemMetrics metrics;
+
+#ifdef TRACK_NEW_DELETE
+
+void *operator new(size_t size) {
+  void *ptr = malloc(size + sizeof(size_t));
+  if (!ptr) return ptr;
+  *((size_t *)ptr) = size;
+  metrics.Allocated.fetch_add(size, std::memory_order_relaxed);
+  metrics.Current.fetch_add(size, std::memory_order_relaxed);
+  metrics.Allocations.fetch_add(1, std::memory_order_relaxed);
+  return ptr + sizeof(size_t);
+}
+
+void operator delete(void *ptr, size_t size) noexcept {
+  void *raw = ptr - sizeof(size_t);
+  size_t szs = *((size_t *)raw);
+  metrics.Deleted.fetch_add(szs, std::memory_order_relaxed);
+  metrics.Current.fetch_sub(szs, std::memory_order_relaxed);
+  metrics.Allocations.fetch_sub(1, std::memory_order_relaxed);
+  free(raw);
+}
+
+void operator delete(void *ptr) noexcept {
+  void *raw = ptr - sizeof(size_t);
+  size_t szs = *((size_t *)raw);
+  metrics.Deleted.fetch_add(szs, std::memory_order_relaxed);
+  metrics.Current.fetch_sub(szs, std::memory_order_relaxed);
+  metrics.Allocations.fetch_sub(1, std::memory_order_relaxed);
+  free(raw);
+}
+
+void *operator new[](size_t size) {
+  void *ptr = malloc(size + sizeof(size_t));
+  if (!ptr) return ptr;
+
+  *((size_t *)ptr) = size;
+  metrics.Allocated.fetch_add(size, std::memory_order_relaxed);
+  metrics.Current.fetch_add(size, std::memory_order_relaxed);
+  metrics.Allocations.fetch_add(1, std::memory_order_relaxed);
+
+  return ptr + sizeof(size_t);
+}
+
+void operator delete[](void *ptr) noexcept {
+  void *raw = ptr - sizeof(size_t);
+  size_t szs = *((size_t *)raw);
+  metrics.Deleted.fetch_add(szs, std::memory_order_relaxed);
+  metrics.Current.fetch_sub(szs, std::memory_order_relaxed);
+  metrics.Allocations.fetch_sub(1, std::memory_order_relaxed);
+  free(raw);
+}
+#endif
+
 namespace Amy {
 
 void RegisterCxxExceptionHandler() {
@@ -33,4 +91,35 @@ void RegisterCxxExceptionHandler() {
   std::set_terminate(NpiD7CxxExceptionHandler);
 #endif
 }
-}  // namespace Amy
+void *Malloc(size_t size) {
+#ifdef TRACK_NEW_DELETE
+  void *ptr = malloc(size + sizeof(size_t));
+  if (!ptr) return nullptr;
+
+  *((size_t *)ptr) = size;
+  metrics.Allocated.fetch_add(size, std::memory_order_relaxed);
+  metrics.Current.fetch_add(size, std::memory_order_relaxed);
+  metrics.Allocations.fetch_add(1, std::memory_order_relaxed);
+
+  return ptr + sizeof(size_t);
+#else
+  return malloc(size);
+#endif
+}
+void Free(void *ptr) {
+#ifdef TRACK_NEW_DELETE
+  void *raw = ptr - sizeof(size_t);
+  size_t szs = *((size_t *)raw);
+  metrics.Deleted.fetch_add(szs, std::memory_order_relaxed);
+  metrics.Current.fetch_sub(szs, std::memory_order_relaxed);
+  metrics.Allocations.fetch_sub(1, std::memory_order_relaxed);
+  free(raw);
+#else
+  free(ptr);
+#endif
+}
+ull Memory::GetTotalAllocated() { return metrics.Allocated; }
+ull Memory::GetTotalFreed() { return metrics.Deleted; }
+ull Memory::GetCurrent() { return metrics.CurrentlyAllocated(); }
+ull Memory::GetAllocationCount() { return metrics.Allocations; }
+}  // namespace Amy

source/image.cpp

@@ -4,7 +4,7 @@
 
 // Only include stb Code if we need it
 // Otherwise we just build the header
-#if AMY_STB_IMAGE == 1
+#ifdef AMY_STB_IMAGE
 #define STB_IMAGE_IMPLEMENTATION
 #endif
 #include <stb_image.h>

source/iron/font.cpp

@@ -3,9 +3,9 @@
 #include <filesystem>
 #include <fstream>
 
-#if AMY_STB_TT == 1
+//#ifdef AMY_STB_TT
 #define STB_TRUETYPE_IMPLEMENTATION
-#endif
+//#endif
 #include <stb_truetype.h>
 
 namespace Amy {