dungeon_map_panel.h

Go to the documentation of this file.

#ifndef YAZE_APP_EDITOR_DUNGEON_PANELS_DUNGEON_MAP_PANEL_H_
#define YAZE_APP_EDITOR_DUNGEON_PANELS_DUNGEON_MAP_PANEL_H_
 
#include <algorithm>
#include <array>
#include <cmath>
#include <functional>
#include <map>
#include <memory>
#include <string>
#include <vector>
#include "util/i18n/tr.h"
 
#include "app/editor/agent/agent_ui_theme.h"
#include "app/editor/dungeon/dungeon_canvas_viewer.h"
#include "app/editor/dungeon/dungeon_room_composite.h"
#include "app/editor/dungeon/dungeon_room_selector.h"
#include "app/editor/dungeon/dungeon_room_store.h"
#include "app/editor/system/workspace/editor_panel.h"
#include "app/gfx/resource/arena.h"
#include "app/gfx/resource/bitmap_texture_queue.h"
#include "app/gui/core/icons.h"
#include "core/hack_manifest.h"
#include "imgui/imgui.h"
#include "zelda3/dungeon/room.h"
#include "zelda3/dungeon/room_census.h"
#include "zelda3/dungeon/room_layer_manager.h"
#include "zelda3/resource_labels.h"
 
namespace yaze {
namespace editor {
 
class DungeonMapPanel : public WindowContent {
 public:
  DungeonMapPanel(int* current_room_id, ImVector<int>* active_rooms,
                  std::function<void(int)> on_room_selected,
                  DungeonRoomStore* rooms = nullptr)
      : current_room_id_(current_room_id),
        active_rooms_(active_rooms),
        rooms_(rooms),
        on_room_selected_(std::move(on_room_selected)) {}
 
  // ==========================================================================
  // WindowContent Identity
  // ==========================================================================
 
  std::string GetId() const override { return "dungeon.dungeon_map"; }
  std::string GetDisplayName() const override { return "Dungeon Map"; }
  std::string GetIcon() const override { return ICON_MD_MAP; }
  std::string GetEditorCategory() const override { return "Dungeon"; }
  int GetPriority() const override { return 35; }
  std::string GetWorkflowGroup() const override { return "Editors"; }
 
  void SetRoomIntentCallback(
      std::function<void(int, RoomSelectionIntent)> callback) {
    on_room_intent_ = std::move(callback);
  }
 
  // ==========================================================================
  // Configuration
  // ==========================================================================
 
  void SetDungeonRooms(const std::vector<int>& room_ids) {
    room_composite_outputs_.clear();
    dungeon_room_ids_ = room_ids;
    AutoLayoutRooms();
  }
 
  void AddRoom(int room_id) {
    // Avoid duplicates
    for (int id : dungeon_room_ids_) {
      if (id == room_id)
        return;
    }
    dungeon_room_ids_.push_back(room_id);
    AutoLayoutRooms();
  }
 
  void ClearRooms() {
    room_composite_outputs_.clear();
    dungeon_room_ids_.clear();
    room_positions_.clear();
    room_types_.clear();
    stair_connections_.clear();
    holewarp_connections_.clear();
  }
 
  void SetRoomPosition(int room_id, int grid_x, int grid_y) {
    room_positions_[room_id] =
        ImVec2(static_cast<float>(grid_x), static_cast<float>(grid_y));
  }
 
  void SetRenderContextCallback(std::function<void(int)> callback) {
    prepare_render_context_ = std::move(callback);
  }
 
  void SetRooms(DungeonRoomStore* rooms) {
    if (rooms_ != rooms) {
      room_composite_outputs_.clear();
      rooms_ = rooms;
    }
  }
 
  void SetHackManifest(const core::HackManifest* manifest) {
    hack_manifest_ = manifest;
  }
 
  void LoadFromDungeonEntry(const core::DungeonEntry& dungeon) {
    ClearRooms();
    current_dungeon_name_ = dungeon.name;
    for (const auto& room : dungeon.rooms) {
      dungeon_room_ids_.push_back(room.id);
      room_positions_[room.id] = ImVec2(static_cast<float>(room.grid_col),
                                        static_cast<float>(room.grid_row));
      room_types_[room.id] = room.type;
    }
    stair_connections_ = dungeon.stairs;
    holewarp_connections_ = dungeon.holewarps;
  }
 
  // ==========================================================================
  // WindowContent Drawing
  // ==========================================================================
 
  void Draw(bool* p_open) override {
    if (!current_room_id_ || !active_rooms_)
      return;
 
    const auto& theme = AgentUI::GetTheme();
 
    // Show dungeon selection/quick presets
    DrawDungeonSelector();
 
    ImGui::Separator();
 
    // Room size in the map
    constexpr float kRoomWidth = 64.0f;
    constexpr float kRoomHeight = 64.0f;
    constexpr float kRoomSpacing = 8.0f;
 
    // Calculate canvas size based on room positions
    float max_x = 0, max_y = 0;
    for (const auto& [room_id, pos] : room_positions_) {
      max_x = std::max(max_x, pos.x);
      max_y = std::max(max_y, pos.y);
    }
    float canvas_width =
        (max_x + 1) * (kRoomWidth + kRoomSpacing) + kRoomSpacing;
    float canvas_height =
        (max_y + 1) * (kRoomHeight + kRoomSpacing) + kRoomSpacing;
 
    // Minimum size
    canvas_width = std::max(canvas_width, 200.0f);
    canvas_height = std::max(canvas_height, 200.0f);
 
    const ImVec2 available = ImGui::GetContentRegionAvail();
    const float footer = ImGui::GetTextLineHeightWithSpacing();
    if (!ImGui::BeginChild("##DungeonMapCanvas",
                           ImVec2(0, std::max(1.0f, available.y - footer)),
                           false, ImGuiWindowFlags_HorizontalScrollbar)) {
      ImGui::EndChild();
      return;
    }
    const ImVec2 canvas_size(canvas_width, canvas_height);
 
    // Begin canvas area
    ImVec2 canvas_pos = ImGui::GetCursorScreenPos();
    ImDrawList* draw_list = ImGui::GetWindowDrawList();
 
    // Background
    ImU32 bg_color = ImGui::ColorConvertFloat4ToU32(theme.panel_bg_darker);
    draw_list->AddRectFilled(
        canvas_pos,
        ImVec2(canvas_pos.x + canvas_size.x, canvas_pos.y + canvas_size.y),
        bg_color);
 
    // Helper lambda: compute the center pixel position for a room on the canvas
    auto RoomCenter = [&](int room_id) -> ImVec2 {
      auto it = room_positions_.find(room_id);
      if (it == room_positions_.end())
        return ImVec2(0, 0);
      ImVec2 pos = it->second;
      return ImVec2(canvas_pos.x + kRoomSpacing +
                        pos.x * (kRoomWidth + kRoomSpacing) + kRoomWidth * 0.5f,
                    canvas_pos.y + kRoomSpacing +
                        pos.y * (kRoomHeight + kRoomSpacing) +
                        kRoomHeight * 0.5f);
    };
 
    // Registry edges may be historical adjacency guesses or raw header
    // bytes. Draw only connections supported by current editable rooms.
    ImVec4 connection_color = theme.dungeon_room_border_dark;
    connection_color.w = 0.45f;
    for (const auto& link : CollectConnections()) {
      const ImVec2 from = RoomCenter(link.from_room_id);
      const ImVec2 to = RoomCenter(link.to_room_id);
      switch (link.type) {
        case DungeonConnectedLinkType::Door:
          draw_list->AddLine(
              from, to, ImGui::ColorConvertFloat4ToU32(connection_color), 1.5f);
          break;
        case DungeonConnectedLinkType::Staircase:
          DrawDashedLine(draw_list, from, to, IM_COL32(100, 149, 237, 200),
                         1.5f, 6.0f);
          break;
        case DungeonConnectedLinkType::Holewarp: {
          const ImU32 red = IM_COL32(220, 60, 60, 200);
          draw_list->AddLine(from, to, red, 2.0f);
          DrawArrowhead(draw_list, from, to, red, 6.0f);
          break;
        }
      }
    }
 
    // Draw each room
    for (int room_id : dungeon_room_ids_) {
      auto pos_it = room_positions_.find(room_id);
      if (pos_it == room_positions_.end())
        continue;
 
      ImVec2 grid_pos = pos_it->second;
      ImVec2 room_min(canvas_pos.x + kRoomSpacing +
                          grid_pos.x * (kRoomWidth + kRoomSpacing),
                      canvas_pos.y + kRoomSpacing +
                          grid_pos.y * (kRoomHeight + kRoomSpacing));
      ImVec2 room_max(room_min.x + kRoomWidth, room_min.y + kRoomHeight);
 
      // Check if room is valid
      if (room_id < 0 || room_id >= 0x128)
        continue;
 
      bool is_current = (*current_room_id_ == room_id);
      bool is_open = false;
      for (int i = 0; i < active_rooms_->Size; i++) {
        if ((*active_rooms_)[i] == room_id) {
          is_open = true;
          break;
        }
      }
 
      // Draw room thumbnail or placeholder
      if (rooms_) {
        auto* loaded_room = rooms_->GetIfLoaded(room_id);
        if (loaded_room != nullptr) {
          if (prepare_render_context_)
            prepare_render_context_(room_id);
          auto& output = room_composite_outputs_[room_id];
          if (!output) {
            output = std::make_unique<RoomCompositeOutput>();
          }
          auto& preview_bitmap =
              PrepareCanonicalRoomComposite(*loaded_room, *output);
          gfx::EnsureCompositeBitmapTextureQueued(preview_bitmap);
          gfx::Arena::Get().ProcessTextureQueue(nullptr);
          if (preview_bitmap.is_active() && preview_bitmap.texture() != 0) {
            // Draw room thumbnail
            draw_list->AddImage((ImTextureID)(intptr_t)preview_bitmap.texture(),
                                room_min, room_max, ImVec2(0, 0), ImVec2(1, 1),
                                ImGui::GetColorU32(ImVec4(1, 1, 1, 1)));
          } else {
            // Placeholder for loaded but no texture
            draw_list->AddRectFilled(
                room_min, room_max,
                ImGui::ColorConvertFloat4ToU32(theme.panel_bg_color));
          }
        } else {
          // Not loaded - gray placeholder
          draw_list->AddRectFilled(
              room_min, room_max,
              ImGui::ColorConvertFloat4ToU32(theme.panel_bg_darker));
 
          // Show room ID
          char label[8];
          snprintf(label, sizeof(label), "%02X", room_id);
          ImVec2 text_size = ImGui::CalcTextSize(label);
          ImVec2 text_pos(room_min.x + (kRoomWidth - text_size.x) * 0.5f,
                          room_min.y + (kRoomHeight - text_size.y) * 0.5f);
          draw_list->AddText(
              text_pos,
              ImGui::ColorConvertFloat4ToU32(theme.text_secondary_gray), label);
        }
      } else {
        // Not loaded - gray placeholder
        draw_list->AddRectFilled(
            room_min, room_max,
            ImGui::ColorConvertFloat4ToU32(theme.panel_bg_darker));
 
        // Show room ID
        char label[8];
        snprintf(label, sizeof(label), "%02X", room_id);
        ImVec2 text_size = ImGui::CalcTextSize(label);
        ImVec2 text_pos(room_min.x + (kRoomWidth - text_size.x) * 0.5f,
                        room_min.y + (kRoomHeight - text_size.y) * 0.5f);
        draw_list->AddText(
            text_pos, ImGui::ColorConvertFloat4ToU32(theme.text_secondary_gray),
            label);
      }
 
      // Draw border based on state
      if (is_current) {
        // Glow effect
        ImVec4 glow = theme.dungeon_selection_primary;
        glow.w = 0.4f;
        ImVec2 glow_min(room_min.x - 2, room_min.y - 2);
        ImVec2 glow_max(room_max.x + 2, room_max.y + 2);
        draw_list->AddRect(glow_min, glow_max,
                           ImGui::ColorConvertFloat4ToU32(glow), 0.0f, 4.0f);
        // Inner border
        draw_list->AddRect(
            room_min, room_max,
            ImGui::ColorConvertFloat4ToU32(theme.dungeon_selection_primary),
            0.0f, 2.0f);
      } else if (is_open) {
        draw_list->AddRect(
            room_min, room_max,
            ImGui::ColorConvertFloat4ToU32(theme.dungeon_grid_cell_selected),
            0.0f, 2.0f);
      } else {
        draw_list->AddRect(
            room_min, room_max,
            ImGui::ColorConvertFloat4ToU32(theme.dungeon_grid_cell_border),
            0.0f, 1.0f);
      }
 
      // Room type badge (small colored dot in top-left corner)
      auto type_it = room_types_.find(room_id);
      if (type_it != room_types_.end()) {
        ImU32 badge_color = 0;
        if (type_it->second == "entrance") {
          badge_color = IM_COL32(76, 175, 80, 220);  // Green
        } else if (type_it->second == "boss") {
          badge_color = IM_COL32(244, 67, 54, 220);  // Red
        } else if (type_it->second == "mini_boss") {
          badge_color = IM_COL32(255, 152, 0, 220);  // Orange
        }
        if (badge_color != 0) {
          ImVec2 badge_center(room_min.x + 6.0f, room_min.y + 6.0f);
          draw_list->AddCircleFilled(badge_center, 4.0f, badge_color);
        }
      }
 
      // Handle clicks
      ImGui::SetCursorScreenPos(room_min);
      char btn_id[32];
      snprintf(btn_id, sizeof(btn_id), "##map_room%d", room_id);
      ImGui::InvisibleButton(btn_id, ImVec2(kRoomWidth, kRoomHeight));
 
      if (ImGui::IsItemClicked()) {
        if (ImGui::IsMouseDoubleClicked(ImGuiMouseButton_Left)) {
          if (on_room_intent_) {
            on_room_intent_(room_id, RoomSelectionIntent::kOpenStandalone);
          } else if (on_room_selected_) {
            on_room_selected_(room_id);
          }
        } else if (on_room_selected_) {
          on_room_selected_(room_id);
        }
      }
 
      // Tooltip
      if (ImGui::IsItemHovered()) {
        ImGui::BeginTooltip();
        ImGui::Text("[%03X] %s", room_id,
                    zelda3::GetRoomLabel(room_id).c_str());
        if (rooms_) {
          if (auto* loaded_room = rooms_->GetIfLoaded(room_id)) {
            ImGui::TextDisabled(tr("Palette: %d"), loaded_room->palette());
          }
        }
        ImGui::TextDisabled(tr("Click to select"));
        ImGui::EndTooltip();
      }
    }
 
    // Invisible room buttons moved the cursor. Reserve the grid from its
    // original origin, rather than adding its extent after the last room.
    ImGui::SetCursorScreenPos(canvas_pos);
    ImGui::Dummy(canvas_size);
    ImGui::EndChild();
 
    // Status bar
    ImGui::TextDisabled(tr("%zu rooms in view"), dungeon_room_ids_.size());
  }
 
 private:
  std::vector<DungeonConnectedRoomLink> CollectConnections() const {
    std::vector<DungeonConnectedRoomLink> links;
    if (!rooms_) {
      return links;
    }
    const auto warp_tags =
        hack_manifest_ ? zelda3::RoomCensusWarpTagsFromManifest(*hack_manifest_)
                       : std::set<uint8_t>{};
    for (int room_id : dungeon_room_ids_) {
      const auto* room = rooms_->GetIfMaterialized(room_id);
      if (!room || !room->IsLoaded()) {
        continue;
      }
      const auto outgoing = CollectDungeonConnectedRoomLinks(
          room_id, *room,
          [this](int neighbor_id, zelda3::DoorDirection direction) {
            const auto* neighbor = rooms_->GetIfMaterialized(neighbor_id);
            if (!neighbor || !neighbor->IsLoaded()) {
              return false;
            }
            return std::any_of(
                neighbor->GetDoors().begin(), neighbor->GetDoors().end(),
                [direction](const auto& door) {
                  return door.direction == direction &&
                         zelda3::IsRoomConnectionDoorType(door.type);
                });
          },
          nullptr, warp_tags);
      for (const auto& link : outgoing) {
        if (room_positions_.contains(link.to_room_id) &&
            (link.type != DungeonConnectedLinkType::Door ||
             link.from_room_id < link.to_room_id)) {
          links.push_back(link);
        }
      }
    }
    return links;
  }
 
  friend class DungeonMapPanelTestPeer;
 
  void AutoLayoutRooms() {
    room_positions_.clear();
 
    int cols = static_cast<int>(
        std::ceil(std::sqrt(static_cast<double>(dungeon_room_ids_.size()))));
    cols = std::max(1, cols);
 
    for (size_t i = 0; i < dungeon_room_ids_.size(); i++) {
      int room_id = dungeon_room_ids_[i];
      int grid_x = static_cast<int>(i % cols);
      int grid_y = static_cast<int>(i / cols);
      room_positions_[room_id] =
          ImVec2(static_cast<float>(grid_x), static_cast<float>(grid_y));
    }
  }
 
  void DrawDungeonSelector() {
    bool has_registry = hack_manifest_ && hack_manifest_->HasProjectRegistry();
 
    if (has_registry) {
      DrawRegistrySelector();
    } else {
      DrawVanillaPresetSelector();
    }
 
    ImGui::SameLine();
    if (ImGui::Button(ICON_MD_ADD " Add Current")) {
      if (current_room_id_ && *current_room_id_ >= 0) {
        AddRoom(*current_room_id_);
      }
    }
    if (ImGui::IsItemHovered()) {
      ImGui::SetTooltip(tr("Add currently selected room to the map"));
    }
 
    ImGui::SameLine();
    if (ImGui::Button(ICON_MD_CLEAR " Clear")) {
      ClearRooms();
      current_dungeon_name_ = "Select Dungeon...";
      selected_preset_ = -1;
    }
  }
 
  void DrawRegistrySelector() {
    const auto& dungeons = hack_manifest_->project_registry().dungeons;
 
    ImGui::SetNextItemWidth(
        std::max(100.0f, ImGui::GetContentRegionAvail().x - 220.0f));
    if (ImGui::BeginCombo("##DungeonRegistry", current_dungeon_name_.c_str())) {
      for (size_t i = 0; i < dungeons.size(); i++) {
        const auto& dungeon = dungeons[i];
        char label[128];
        if (!dungeon.vanilla_name.empty()) {
          snprintf(label, sizeof(label), "%s: %s (%s)", dungeon.id.c_str(),
                   dungeon.name.c_str(), dungeon.vanilla_name.c_str());
        } else {
          snprintf(label, sizeof(label), "%s: %s", dungeon.id.c_str(),
                   dungeon.name.c_str());
        }
        bool selected = (current_dungeon_name_ == dungeon.name);
        if (ImGui::Selectable(label, selected)) {
          LoadFromDungeonEntry(dungeon);
          selected_preset_ = static_cast<int>(i);
        }
      }
      ImGui::EndCombo();
    }
  }
 
  void DrawVanillaPresetSelector() {
    struct DungeonPreset {
      const char* name;
      int start_room;
      int count;
    };
 
    static const DungeonPreset kPresets[] = {
        {"Eastern Palace", 0xC8, 8}, {"Desert Palace", 0x33, 8},
        {"Tower of Hera", 0x07, 8},  {"Palace of Darkness", 0x09, 12},
        {"Swamp Palace", 0x28, 10},  {"Skull Woods", 0x29, 10},
        {"Thieves' Town", 0x44, 8},  {"Ice Palace", 0x0E, 12},
        {"Misery Mire", 0x61, 10},   {"Turtle Rock", 0x04, 12},
        {"Ganon's Tower", 0x0C, 16}, {"Hyrule Castle", 0x01, 12},
    };
 
    ImGui::SetNextItemWidth(
        std::max(100.0f, ImGui::GetContentRegionAvail().x - 220.0f));
    if (ImGui::BeginCombo(
            "##DungeonPreset",
            selected_preset_ >= 0 && selected_preset_ < IM_ARRAYSIZE(kPresets)
                ? kPresets[selected_preset_].name
                : "Select Dungeon...")) {
      for (int i = 0; i < IM_ARRAYSIZE(kPresets); i++) {
        if (ImGui::Selectable(kPresets[i].name, selected_preset_ == i)) {
          selected_preset_ = i;
          ClearRooms();
          for (int j = 0; j < kPresets[i].count; j++) {
            int room_id = kPresets[i].start_room + j;
            if (room_id < 0x128) {
              dungeon_room_ids_.push_back(room_id);
            }
          }
          AutoLayoutRooms();
        }
      }
      ImGui::EndCombo();
    }
  }
 
  static void DrawDashedLine(ImDrawList* dl, ImVec2 from, ImVec2 to,
                             ImU32 color, float thickness, float dash_len) {
    float dx = to.x - from.x;
    float dy = to.y - from.y;
    float length = std::sqrt(dx * dx + dy * dy);
    if (length < 1.0f)
      return;
    float nx = dx / length;
    float ny = dy / length;
 
    float drawn = 0.0f;
    bool visible = true;
    while (drawn < length) {
      float seg = std::min(dash_len, length - drawn);
      ImVec2 seg_start(from.x + nx * drawn, from.y + ny * drawn);
      ImVec2 seg_end(from.x + nx * (drawn + seg), from.y + ny * (drawn + seg));
      if (visible) {
        dl->AddLine(seg_start, seg_end, color, thickness);
      }
      drawn += seg;
      visible = !visible;
    }
  }
 
  static void DrawArrowhead(ImDrawList* dl, ImVec2 from, ImVec2 to, ImU32 color,
                            float size) {
    float dx = to.x - from.x;
    float dy = to.y - from.y;
    float length = std::sqrt(dx * dx + dy * dy);
    if (length < 1.0f)
      return;
    float nx = dx / length;
    float ny = dy / length;
    // Perpendicular
    float px = -ny;
    float py = nx;
 
    ImVec2 tip = to;
    ImVec2 left(to.x - nx * size + px * size * 0.5f,
                to.y - ny * size + py * size * 0.5f);
    ImVec2 right(to.x - nx * size - px * size * 0.5f,
                 to.y - ny * size - py * size * 0.5f);
    dl->AddTriangleFilled(tip, left, right, color);
  }
 
  int* current_room_id_ = nullptr;
  ImVector<int>* active_rooms_ = nullptr;
  DungeonRoomStore* rooms_ = nullptr;
  std::function<void(int)> prepare_render_context_;
  std::function<void(int)> on_room_selected_;
  std::function<void(int, RoomSelectionIntent)> on_room_intent_;
 
  // Room data
  std::vector<int> dungeon_room_ids_;
  std::map<int, ImVec2> room_positions_;
  std::map<int, std::string> room_types_;
  // Each configured room needs stable texture identity through ImGui's
  // deferred frame render. Clear the set when switching dungeon contexts.
  std::map<int, std::unique_ptr<RoomCompositeOutput>> room_composite_outputs_;
  int selected_preset_ = -1;
 
  // Project registry integration
  const core::HackManifest* hack_manifest_ = nullptr;
  std::vector<core::DungeonConnection> stair_connections_;
  std::vector<core::DungeonConnection> holewarp_connections_;
  std::string current_dungeon_name_ = "Select Dungeon...";
};
 
}  // namespace editor
}  // namespace yaze
 
#endif  // YAZE_APP_EDITOR_DUNGEON_PANELS_DUNGEON_MAP_PANEL_H_