Move delete later to platform::gui.

1 files changed, 16 insertions, 2 deletions

diff --git a/src/platform/gui/UiApplication.cpp b/src/platform/gui/UiApplication.cpp
index f035224f..56570125 100644
--- a/src/platform/gui/UiApplication.cpp
+++ b/src/platform/gui/UiApplication.cpp
@@ -1,9 +1,23 @@
 #include "cru/platform/gui/UiApplication.h"
-
 #include "cru/base/Base.h"
 
+#include <unordered_set>
+
 namespace cru::platform::gui {
 
+void DeleteLaterPool::Add(Object* object) { pool_.push_back(object); }
+
+void DeleteLaterPool::Clean() {
+  std::unordered_set<Object*> deleted;
+  for (auto object : pool_) {
+    if (!deleted.contains(object)) {
+      deleted.insert(object);
+      delete object;
+    }
+  }
+  pool_.clear();
+}
+
 namespace {
 IUiApplication* instance = nullptr;
 }
@@ -12,7 +26,7 @@ IUiApplication* IUiApplication::GetInstance() { return instance; }
 
 IUiApplication::IUiApplication() {
   if (instance) {
-    throw std::runtime_error("An ui application has already been created.");
+    throw Exception("A ui application has already been created.");
   }
 
   instance = this;