sprite_editor_panel.h

Go to the documentation of this file.

#ifndef YAZE_APP_EDITOR_DUNGEON_PANELS_SPRITE_EDITOR_PANEL_H_
#define YAZE_APP_EDITOR_DUNGEON_PANELS_SPRITE_EDITOR_PANEL_H_
 
#include <algorithm>
#include <array>
#include <cstdint>
#include <functional>
#include <string>
#include "util/i18n/tr.h"
 
#include "absl/strings/str_format.h"
#include "app/editor/agent/agent_ui_theme.h"
#include "app/editor/dungeon/dungeon_canvas_viewer.h"
#include "app/editor/dungeon/dungeon_coordinates.h"
#include "app/editor/dungeon/dungeon_room_store.h"
#include "app/editor/system/workspace/editor_panel.h"
#include "app/gui/automation/widget_auto_register.h"
#include "app/gui/core/icons.h"
#include "app/gui/core/input.h"
#include "app/gui/core/style_guard.h"
#include "imgui/imgui.h"
#include "zelda3/dungeon/room.h"
#include "zelda3/sprite/sprite.h"
 
namespace yaze {
namespace editor {
 
class SpriteEditorPanel : public WindowContent {
 public:
  SpriteEditorPanel(int* current_room_id, DungeonRoomStore* rooms,
                    DungeonCanvasViewer* canvas_viewer = nullptr)
      : current_room_id_(current_room_id),
        rooms_(rooms),
        canvas_viewer_(canvas_viewer) {}
 
  // ==========================================================================
  // WindowContent Identity
  // ==========================================================================
 
  std::string GetId() const override { return "dungeon.sprite_editor"; }
  std::string GetDisplayName() const override { return "Sprite Editor"; }
  std::string GetIcon() const override { return ICON_MD_PERSON; }
  std::string GetEditorCategory() const override { return "Dungeon"; }
  int GetPriority() const override { return 65; }
 
  // ==========================================================================
  // WindowContent Drawing
  // ==========================================================================
 
  void Draw(bool* p_open) override {
    gui::AutoWidgetScope automation_scope("Dungeon/SpriteEditor");
    if (!current_room_id_ || !rooms_) {
      ImGui::TextDisabled(tr("No room data available"));
      return;
    }
 
    if (*current_room_id_ < 0 ||
        *current_room_id_ >= static_cast<int>(rooms_->size())) {
      ImGui::TextDisabled(tr("No room selected"));
      return;
    }
 
    DrawPlacementControls();
    ImGui::Separator();
    DrawSpriteSelector();
    ImGui::Separator();
    DrawRoomSprites();
  }
 
  // ==========================================================================
  // Panel-Specific Methods
  // ==========================================================================
 
  void SetCanvasViewer(DungeonCanvasViewer* viewer) { canvas_viewer_ = viewer; }
 
  void SetSpritePlacedCallback(
      std::function<void(const zelda3::Sprite&)> callback) {
    sprite_placed_callback_ = std::move(callback);
  }
  void SetOpenSelectionInspectorCallback(std::function<void()> callback) {
    open_selection_inspector_callback_ = std::move(callback);
  }
 
 private:
  void DrawPlacementControls() {
    const auto& theme = AgentUI::GetTheme();
    // Placement mode indicator
    if (placement_mode_) {
      ImGui::TextColored(
          theme.status_warning, ICON_MD_PLACE " Placing: %s (0x%02X)",
          zelda3::ResolveSpriteName(selected_sprite_id_), selected_sprite_id_);
      if (ImGui::SmallButton(ICON_MD_CANCEL " Cancel")) {
        placement_mode_ = false;
        if (canvas_viewer_) {
          canvas_viewer_->object_interaction().SetSpritePlacementMode(false, 0);
        }
      }
    } else {
      ImGui::TextColored(theme.text_secondary_gray,
                         ICON_MD_INFO " Select a sprite to place");
    }
  }
 
  void DrawSpriteSelector() {
    const auto& theme = AgentUI::GetTheme();
    ImGui::Text(ICON_MD_PERSON " Select Sprite:");
 
    // Filter by category
    static const char* kCategories[] = {"All", "Enemies", "NPCs", "Bosses",
                                        "Items"};
    ImGui::SetNextItemWidth(100);
    ImGui::Combo("##Category", &selected_category_, kCategories,
                 IM_ARRAYSIZE(kCategories));
    ImGui::SameLine();
 
    // Search filter
    ImGui::SetNextItemWidth(120);
    ImGui::InputTextWithHint("##Search", "Search...", search_filter_,
                             sizeof(search_filter_));
 
    // Sprite grid with responsive sizing
    float available_height = ImGui::GetContentRegionAvail().y;
    // Reserve space for room sprites section
    float reserved_height = 120.0f;
    // Calculate grid height: at least 150px, responsive to available space
    float grid_height =
        std::max(150.0f, std::min(400.0f, available_height - reserved_height));
 
    // Responsive sprite size based on panel width
    float panel_width = ImGui::GetContentRegionAvail().x;
    float sprite_size =
        std::max(28.0f, std::min(40.0f, (panel_width - 40.0f) / 8.0f));
    int items_per_row =
        std::max(1, static_cast<int>(panel_width / (sprite_size + 6)));
 
    ImGui::BeginChild("##SpriteGrid", ImVec2(0, grid_height), true,
                      ImGuiWindowFlags_HorizontalScrollbar);
 
    int col = 0;
    for (int i = 0; i < 256; ++i) {
      // Apply filters
      if (!MatchesFilter(i))
        continue;
 
      bool is_selected = (selected_sprite_id_ == i);
 
      ImGui::PushID(i);
 
      // Color-coded button based on sprite type using theme colors
      ImVec4 button_color = GetSpriteTypeColor(i, theme);
      if (is_selected) {
        button_color.x = std::min(1.0f, button_color.x + 0.2f);
        button_color.y = std::min(1.0f, button_color.y + 0.2f);
        button_color.z = std::min(1.0f, button_color.z + 0.2f);
      }
 
      {
        gui::StyleColorGuard sprite_btn_guard(
            {{ImGuiCol_Button, button_color},
             {ImGuiCol_ButtonHovered,
              ImVec4(std::min(1.0f, button_color.x + 0.1f),
                     std::min(1.0f, button_color.y + 0.1f),
                     std::min(1.0f, button_color.z + 0.1f), 1.0f)},
             {ImGuiCol_ButtonActive,
              ImVec4(std::min(1.0f, button_color.x + 0.2f),
                     std::min(1.0f, button_color.y + 0.2f),
                     std::min(1.0f, button_color.z + 0.2f), 1.0f)}});
 
        // Get category icon based on sprite type
        const char* icon = GetSpriteTypeIcon(i);
        std::string label = absl::StrFormat("%s\n%02X", icon, i);
        if (ImGui::Button(label.c_str(), ImVec2(sprite_size, sprite_size))) {
          selected_sprite_id_ = i;
          placement_mode_ = true;
          if (canvas_viewer_) {
            canvas_viewer_->object_interaction().SetSpritePlacementMode(true,
                                                                        i);
          }
        }
      }
 
      if (ImGui::IsItemHovered()) {
        const char* category = GetSpriteCategoryName(i);
        ImGui::SetTooltip(
            tr("%s (0x%02X)\n[%s]\nClick to select for placement"),
            zelda3::ResolveSpriteName(i), i, category);
      }
 
      // Selection highlight using theme color
      if (is_selected) {
        ImVec2 min = ImGui::GetItemRectMin();
        ImVec2 max = ImGui::GetItemRectMax();
        ImU32 sel_color =
            ImGui::ColorConvertFloat4ToU32(theme.dungeon_selection_primary);
        ImGui::GetWindowDrawList()->AddRect(min, max, sel_color, 0.0f, 0, 2.0f);
      }
 
      ImGui::PopID();
 
      col++;
      if (col < items_per_row) {
        ImGui::SameLine();
      } else {
        col = 0;
      }
    }
 
    ImGui::EndChild();
  }
 
  void DrawRoomSprites() {
    const auto& theme = AgentUI::GetTheme();
    auto& room = (*rooms_)[*current_room_id_];
    auto& sprites = room.GetSprites();
    int selected_sprite_list_index = -1;
    if (canvas_viewer_) {
      const auto& selected_entities = canvas_viewer_->object_interaction()
                                          .entity_coordinator()
                                          .GetSelectedEntities();
      if (selected_entities.size() == 1 &&
          selected_entities.front().type == EntityType::Sprite) {
        selected_sprite_list_index =
            static_cast<int>(selected_entities.front().index);
      }
    }
 
    // Sprite count with limit warning
    int sprite_count = static_cast<int>(sprites.size());
    ImVec4 count_color =
        sprite_count > 16 ? theme.text_error_red : theme.text_primary;
    ImGui::TextColored(count_color, ICON_MD_LIST " Room Sprites: %d/16",
                       sprite_count);
    if (selected_sprite_list_index >= 0 && open_selection_inspector_callback_) {
      ImGui::SameLine();
      if (ImGui::SmallButton(ICON_MD_OPEN_IN_NEW " Inspect Selected")) {
        open_selection_inspector_callback_();
      }
    }
 
    if (sprite_count > 16) {
      ImGui::SameLine();
      ImGui::TextColored(theme.text_warning_yellow, ICON_MD_WARNING);
      if (ImGui::IsItemHovered()) {
        ImGui::SetTooltip(
            tr("Room exceeds sprite limit (16 max)!\n"
               "This may cause game crashes."));
      }
    }
 
    if (sprites.empty()) {
      ImGui::TextColored(theme.text_secondary_gray,
                         ICON_MD_INFO " No sprites in this room");
      return;
    }
 
    if (selected_sprite_list_index >= 0 &&
        selected_sprite_list_index < static_cast<int>(sprites.size()) &&
        canvas_viewer_) {
      const auto& sprite = sprites[selected_sprite_list_index];
      int sprite_x = sprite.x();
      int sprite_y = sprite.y();
      ImGui::TextUnformatted(tr("Selected position"));
      ImGui::SameLine();
      ImGui::SetNextItemWidth(64.0f);
      const bool x_changed = gui::InputScalarDeferred(
          "##SelectedSpriteX", ImGuiDataType_S32, &sprite_x, "%d", 0,
          {reinterpret_cast<uintptr_t>(rooms_),
           static_cast<uint64_t>(*current_room_id_),
           static_cast<uint64_t>(selected_sprite_list_index)});
      gui::AutoRegisterLastItem("input_int", "selected_sprite_x",
                                "Selected sprite X coordinate");
      ImGui::SameLine();
      ImGui::SetNextItemWidth(64.0f);
      const bool y_changed = gui::InputScalarDeferred(
          "##SelectedSpriteY", ImGuiDataType_S32, &sprite_y, "%d", 0,
          {reinterpret_cast<uintptr_t>(rooms_),
           static_cast<uint64_t>(*current_room_id_),
           static_cast<uint64_t>(selected_sprite_list_index)});
      gui::AutoRegisterLastItem("input_int", "selected_sprite_y",
                                "Selected sprite Y coordinate");
      if (x_changed || y_changed) {
        const int clamped_x =
            std::clamp(sprite_x, 0, dungeon_coords::kSpriteGridMax);
        const int clamped_y =
            std::clamp(sprite_y, 0, dungeon_coords::kSpriteGridMax);
        auto& handler = canvas_viewer_->object_interaction()
                            .entity_coordinator()
                            .sprite_handler();
        handler.NudgeSelected(clamped_x - static_cast<int>(sprite.x()),
                              clamped_y - static_cast<int>(sprite.y()));
      }
    }
 
    // Split view: list on top, properties below
    float available = ImGui::GetContentRegionAvail().y;
    float list_height = std::max(100.0f, available * 0.4f);
 
    ImGui::BeginChild("##SpriteList", ImVec2(0, list_height), true);
    for (size_t i = 0; i < sprites.size(); ++i) {
      const auto& sprite = sprites[i];
      bool is_selected = (selected_sprite_list_index == static_cast<int>(i));
 
      ImGui::PushID(static_cast<int>(i));
 
      // Build display string with indicators
      std::string label = absl::StrFormat(
          "[%02X] %s", sprite.id(), zelda3::ResolveSpriteName(sprite.id()));
 
      // Add key drop indicator
      if (sprite.key_drop() == 1) {
        label += " " ICON_MD_KEY;  // Small key
      } else if (sprite.key_drop() == 2) {
        label += " " ICON_MD_VPN_KEY;  // Big key
      }
 
      // Add overlord indicator
      if (sprite.IsOverlord()) {
        label += " " ICON_MD_STAR;
      }
 
      const bool row_selected = ImGui::Selectable(label.c_str(), is_selected);
      gui::AutoRegisterLastItem("selectable",
                                absl::StrFormat("sprite_row_%zu", i),
                                "Select a sprite in the current dungeon room");
      if (row_selected) {
        if (canvas_viewer_) {
          canvas_viewer_->object_interaction().SelectEntity(EntityType::Sprite,
                                                            i);
        }
      }
 
      // Show position on same line
      ImGui::SameLine(ImGui::GetContentRegionAvail().x - 80);
      ImGui::TextColored(theme.text_secondary_gray, tr("(%d,%d) L%d"),
                         sprite.x(), sprite.y(), sprite.layer());
 
      ImGui::PopID();
    }
    ImGui::EndChild();
  }
 
  bool MatchesFilter(int sprite_id) {
    // Category filter
    if (selected_category_ > 0) {
      // Simplified category matching - in real implementation, use proper categorization
      bool is_enemy = (sprite_id >= 0x09 && sprite_id <= 0x7F);
      bool is_npc = (sprite_id >= 0x80 && sprite_id <= 0xBF);
      bool is_boss = (sprite_id >= 0xC0 && sprite_id <= 0xD8);
      bool is_item = (sprite_id >= 0xD9 && sprite_id <= 0xFF);
 
      if (selected_category_ == 1 && !is_enemy)
        return false;
      if (selected_category_ == 2 && !is_npc)
        return false;
      if (selected_category_ == 3 && !is_boss)
        return false;
      if (selected_category_ == 4 && !is_item)
        return false;
    }
 
    // Text search filter
    if (search_filter_[0] != '\0') {
      const char* name = zelda3::ResolveSpriteName(sprite_id);
      // Simple case-insensitive substring search
      std::string name_lower = name;
      std::string filter_lower = search_filter_;
      for (auto& c : name_lower)
        c = static_cast<char>(tolower(c));
      for (auto& c : filter_lower)
        c = static_cast<char>(tolower(c));
      if (name_lower.find(filter_lower) == std::string::npos) {
        return false;
      }
    }
 
    return true;
  }
 
  ImVec4 GetSpriteTypeColor(int sprite_id, const AgentUITheme& theme) {
    // Color-code based on sprite type using theme colors
    if (sprite_id >= 0xC0 && sprite_id <= 0xD8) {
      return theme.status_error;  // Red for bosses
    } else if (sprite_id >= 0x80 && sprite_id <= 0xBF) {
      return theme.dungeon_sprite_layer0;  // Green for NPCs
    } else if (sprite_id >= 0xD9) {
      return theme.dungeon_object_chest;  // Gold for items
    }
    return theme.dungeon_sprite_layer1;  // Blue for enemies
  }
 
  const char* GetSpriteTypeIcon(int sprite_id) {
    // Return category-appropriate icons
    if (sprite_id >= 0xC0 && sprite_id <= 0xD8) {
      return ICON_MD_DANGEROUS;  // Skull for bosses
    } else if (sprite_id >= 0x80 && sprite_id <= 0xBF) {
      return ICON_MD_PERSON;  // Person for NPCs
    } else if (sprite_id >= 0xD9) {
      return ICON_MD_STAR;  // Star for items
    }
    return ICON_MD_PEST_CONTROL;  // Bug for enemies
  }
 
  const char* GetSpriteCategoryName(int sprite_id) {
    if (sprite_id >= 0xC0 && sprite_id <= 0xD8) {
      return "Boss";
    } else if (sprite_id >= 0x80 && sprite_id <= 0xBF) {
      return "NPC";
    } else if (sprite_id >= 0xD9) {
      return "Item";
    }
    return "Enemy";
  }
 
  int* current_room_id_ = nullptr;
  DungeonRoomStore* rooms_ = nullptr;
  DungeonCanvasViewer* canvas_viewer_ = nullptr;
 
  // Selection state
  int selected_sprite_id_ = 0;
  int selected_category_ = 0;
  char search_filter_[64] = {0};
  bool placement_mode_ = false;
 
  std::function<void(const zelda3::Sprite&)> sprite_placed_callback_;
  std::function<void()> open_selection_inspector_callback_;
};
 
}  // namespace editor
}  // namespace yaze
 
#endif  // YAZE_APP_EDITOR_DUNGEON_PANELS_SPRITE_EDITOR_PANEL_H_