From 5e025a3257069c4307ea2614d1ba409063aa9fe6 Mon Sep 17 00:00:00 2001
From: scawful <justin.scofield@outlook.com>
Date: Sun, 14 Jan 2024 11:15:03 -0500
Subject: [PATCH] Add Usage Statistics to DungeonEditor

---
 src/app/editor/dungeon_editor.cc | 114 ++++++++++++++++++++++++++++++-
 src/app/editor/dungeon_editor.h  |  10 +++
 2 files changed, 123 insertions(+), 1 deletion(-)

diff --git a/src/app/editor/dungeon_editor.cc b/src/app/editor/dungeon_editor.cc
index 0560297e..b873c8d3 100644
--- a/src/app/editor/dungeon_editor.cc
+++ b/src/app/editor/dungeon_editor.cc
@@ -60,6 +60,26 @@ absl::Status DungeonEditor::Update() {
     refresh_graphics_ = false;
   }
 
+  TAB_BAR("##DungeonEditorTabBar")
+  TAB_ITEM("Dungeon Room Editor")
+  UpdateDungeonRoomView();
+  END_TAB_ITEM()
+  TAB_ITEM("Usage Statistics")
+  if (is_loaded_) {
+    static bool calc_stats = false;
+    if (!calc_stats) {
+      CalculateUsageStats();
+      calc_stats = true;
+    }
+    DrawUsageStats();
+  }
+  END_TAB_ITEM()
+  END_TAB_BAR()
+
+  return absl::OkStatus();
+}
+
+void DungeonEditor::UpdateDungeonRoomView() {
   DrawToolset();
 
   if (palette_showing_) {
@@ -90,7 +110,6 @@ absl::Status DungeonEditor::Update() {
     DrawTileSelector();
     ImGui::EndTable();
   }
-  return absl::OkStatus();
 }
 
 void DungeonEditor::DrawToolset() {
@@ -384,6 +403,99 @@ void DungeonEditor::DrawObjectRenderer() {
   // }
 }
 
+void DungeonEditor::CalculateUsageStats() {
+  // Create a hash map of the usage for elements of each Dungeon Room such as
+  // the blockset, spriteset, palette, etc. This is so we can keep track of
+  // which graphics sets and palette sets are in use and which are not.
+
+  for (const auto& room : rooms_) {
+    // Blockset
+    if (blockset_usage_.find(room.blockset) == blockset_usage_.end()) {
+      blockset_usage_[room.blockset] = 1;
+    } else {
+      blockset_usage_[room.blockset] += 1;
+    }
+
+    // Spriteset
+    if (spriteset_usage_.find(room.spriteset) == spriteset_usage_.end()) {
+      spriteset_usage_[room.spriteset] = 1;
+    } else {
+      spriteset_usage_[room.spriteset] += 1;
+    }
+
+    // Palette
+    if (palette_usage_.find(room.palette) == palette_usage_.end()) {
+      palette_usage_[room.palette] = 1;
+    } else {
+      palette_usage_[room.palette] += 1;
+    }
+  }
+}
+
+namespace {
+template <typename T>
+void RenderSetUsage(const absl::flat_hash_map<T, int>& usage_map) {
+  // Sort the usage map by set number
+  std::vector<std::pair<T, int>> sorted_usage(usage_map.begin(),
+                                              usage_map.end());
+  std::sort(sorted_usage.begin(), sorted_usage.end(),
+            [](const auto& a, const auto& b) { return a.first < b.first; });
+  for (const auto& [set, count] : sorted_usage) {
+    ImGui::Text("%#02x: %d uses", set, count);
+  }
+}
+
+// Calculate the unused sets in a usage map
+// Range for blocksets 0-0x24
+// Range for spritesets 0-0x8F
+// Range for palettes 0-0x47
+template <typename T>
+void RenderUnusedSets(const absl::flat_hash_map<T, int>& usage_map,
+                      int max_set) {
+  std::vector<int> unused_sets;
+  for (int i = 0; i < max_set; i++) {
+    if (usage_map.find(i) == usage_map.end()) {
+      unused_sets.push_back(i);
+    }
+  }
+  for (const auto& set : unused_sets) {
+    ImGui::Text("%#02x", set);
+  }
+}
+}  // namespace
+
+void DungeonEditor::DrawUsageStats() {
+  if (ImGui::BeginTable("DungeonUsageStatsTable", 6, kDungeonTableFlags,
+                        ImVec2(0, 0))) {
+    TableSetupColumn("Blockset Usage");
+    TableSetupColumn("Unused Blockset");
+    TableSetupColumn("Palette Usage");
+    TableSetupColumn("Unused Palette");
+    TableSetupColumn("Spriteset Usage");
+    TableSetupColumn("Unused Spriteset");
+    TableHeadersRow();
+
+    TableNextColumn();
+    RenderSetUsage(blockset_usage_);
+
+    TableNextColumn();
+    RenderUnusedSets(blockset_usage_, 0x25);
+
+    TableNextColumn();
+    RenderSetUsage(palette_usage_);
+
+    TableNextColumn();
+    RenderUnusedSets(palette_usage_, 0x48);
+
+    TableNextColumn();
+    RenderSetUsage(spriteset_usage_);
+
+    TableNextColumn();
+    RenderUnusedSets(spriteset_usage_, 0x90);
+  }
+  ImGui::EndTable();
+}
+
 }  // namespace editor
 }  // namespace app
 }  // namespace yaze
\ No newline at end of file
diff --git a/src/app/editor/dungeon_editor.h b/src/app/editor/dungeon_editor.h
index cb158303..c1d6f4cf 100644
--- a/src/app/editor/dungeon_editor.h
+++ b/src/app/editor/dungeon_editor.h
@@ -41,6 +41,9 @@ class DungeonEditor : public Editor,
   absl::Status Redo() override { return absl::OkStatus(); }
 
  private:
+
+  void UpdateDungeonRoomView();
+
   void DrawToolset();
   void DrawRoomSelector();
 
@@ -51,6 +54,9 @@ class DungeonEditor : public Editor,
   void DrawTileSelector();
   void DrawObjectRenderer();
 
+  void CalculateUsageStats();
+  void DrawUsageStats();
+
   enum BackgroundType {
     kNoBackground,
     kBackground1,
@@ -92,6 +98,10 @@ class DungeonEditor : public Editor,
   std::vector<zelda3::dungeon::Room> rooms_;
   std::vector<gfx::BitmapManager> room_graphics_;
   zelda3::dungeon::DungeonObjectRenderer object_renderer_;
+
+  absl::flat_hash_map<uint16_t, int> spriteset_usage_;
+  absl::flat_hash_map<uint16_t, int> blockset_usage_;
+  absl::flat_hash_map<uint16_t, int> palette_usage_;
 };
 
 }  // namespace editor