link_sprite_view.cc

Go to the documentation of this file.

#include "app/editor/graphics/ui/sprite/link_sprite_view.h"
 
#include "absl/strings/str_format.h"
#include "app/gfx/resource/arena.h"
#include "app/gui/core/icons.h"
#include "app/gui/core/layout_helpers.h"
#include "app/gui/core/style.h"
#include "app/gui/core/style_guard.h"
#include "app/gui/core/ui_helpers.h"
#include "imgui/imgui.h"
#include "rom/rom.h"
#include "util/file_util.h"
#include "util/log.h"
 
namespace yaze {
namespace editor {
 
LinkSpriteView::LinkSpriteView(GraphicsEditorState* state, Rom* rom)
    : state_(state), rom_(rom) {}
 
void LinkSpriteView::Initialize() {
  preview_canvas_.SetCanvasSize(
      ImVec2(128 * preview_zoom_, 32 * preview_zoom_));
}
 
void LinkSpriteView::Draw(bool* p_open) {
  // WindowContent interface - delegate to existing Update() logic
  // Lazy-load Link sheets on first update
  if (!sheets_loaded_ && rom_ && rom_->is_loaded()) {
    auto status = LoadLinkSheets();
    if (!status.ok()) {
      ImGui::TextColored(gui::GetErrorColor(), "Failed to load Link sheets: %s",
                         status.message().data());
      return;
    }
  }
 
  DrawToolbar();
  ImGui::Separator();
 
  // Split layout: left side grid, right side preview
  float panel_width = ImGui::GetContentRegionAvail().x;
  float grid_width = std::min(300.0f, panel_width * 0.4f);
 
  // Left column: Sheet grid
  ImGui::BeginChild("##LinkSheetGrid", ImVec2(grid_width, 0), true);
  DrawSheetGrid();
  ImGui::EndChild();
 
  ImGui::SameLine();
 
  // Right column: Preview and controls
  ImGui::BeginChild("##LinkPreviewArea", ImVec2(0, 0), true);
  DrawPreviewCanvas();
  ImGui::Separator();
  DrawPaletteSelector();
  ImGui::Separator();
  DrawInfoPanel();
  ImGui::EndChild();
}
 
absl::Status LinkSpriteView::Update() {
  // Lazy-load Link sheets on first update
  if (!sheets_loaded_ && rom_ && rom_->is_loaded()) {
    auto status = LoadLinkSheets();
    if (!status.ok()) {
      ImGui::TextColored(gui::GetErrorColor(), "Failed to load Link sheets: %s",
                         status.message().data());
      return status;
    }
  }
 
  DrawToolbar();
  ImGui::Separator();
 
  // Split layout: left side grid, right side preview
  float panel_width = ImGui::GetContentRegionAvail().x;
  float grid_width = std::min(300.0f, panel_width * 0.4f);
 
  // Left column: Sheet grid
  ImGui::BeginChild("##LinkSheetGrid", ImVec2(grid_width, 0), true);
  DrawSheetGrid();
  ImGui::EndChild();
 
  ImGui::SameLine();
 
  // Right column: Preview and controls
  ImGui::BeginChild("##LinkPreviewArea", ImVec2(0, 0), true);
  DrawPreviewCanvas();
  ImGui::Separator();
  DrawPaletteSelector();
  ImGui::Separator();
  DrawInfoPanel();
  ImGui::EndChild();
 
  return absl::OkStatus();
}
 
void LinkSpriteView::DrawToolbar() {
  if (ImGui::Button(ICON_MD_FILE_UPLOAD " Import ZSPR")) {
    ImportZspr();
  }
  HOVER_HINT("Import a .zspr Link sprite file");
 
  ImGui::SameLine();
  if (ImGui::Button(ICON_MD_RESTORE " Reset to Vanilla")) {
    ResetToVanilla();
  }
  HOVER_HINT("Reset Link graphics to vanilla ROM data");
 
  // Show loaded ZSPR info
  if (loaded_zspr_.has_value()) {
    ImGui::SameLine();
    ImGui::TextColored(gui::GetSuccessColor(),
                       ICON_MD_CHECK_CIRCLE " Loaded: %s",
                       loaded_zspr_->metadata.display_name.c_str());
  }
 
  // Unsaved changes indicator
  if (has_unsaved_changes_) {
    ImGui::SameLine();
    ImGui::TextColored(gui::GetWarningColor(), ICON_MD_EDIT " [Unsaved]");
  }
}
 
void LinkSpriteView::DrawSheetGrid() {
  ImGui::Text("Link Sheets (14)");
  ImGui::Separator();
 
  // 4x4 grid (14 sheets + 2 empty slots)
  const float cell_size = kThumbnailSize + kThumbnailPadding * 2;
  int col = 0;
 
  for (int i = 0; i < kNumLinkSheets; i++) {
    if (col > 0) {
      ImGui::SameLine();
    }
 
    ImGui::PushID(i);
    DrawSheetThumbnail(i);
    ImGui::PopID();
 
    col++;
    if (col >= 4) {
      col = 0;
    }
  }
}
 
void LinkSpriteView::DrawSheetThumbnail(int sheet_index) {
  bool is_selected = (selected_sheet_ == sheet_index);
 
  // Selection highlight
  std::optional<gui::StyleColorGuard> sel_bg_guard;
  if (is_selected) {
    sel_bg_guard.emplace(ImGuiCol_ChildBg, ImVec4(0.3f, 0.5f, 0.8f, 0.4f));
  }
 
  ImGui::BeginChild(absl::StrFormat("##LinkSheet%d", sheet_index).c_str(),
                    ImVec2(kThumbnailSize + kThumbnailPadding,
                           kThumbnailSize + 16 + kThumbnailPadding),
                    true, ImGuiWindowFlags_NoScrollbar);
 
  // Draw thumbnail
  auto& sheet = link_sheets_[sheet_index];
  if (sheet.is_active()) {
    // Ensure texture exists
    if (!sheet.texture() && sheet.surface()) {
      gfx::Arena::Get().QueueTextureCommand(
          gfx::Arena::TextureCommandType::CREATE,
          const_cast<gfx::Bitmap*>(&sheet));
    }
 
    if (sheet.texture()) {
      ImVec2 cursor_pos = ImGui::GetCursorScreenPos();
      ImGui::GetWindowDrawList()->AddImage(
          (ImTextureID)(intptr_t)sheet.texture(), cursor_pos,
          ImVec2(cursor_pos.x + kThumbnailSize,
                 cursor_pos.y + kThumbnailSize / 4));  // 128x32 aspect
    }
  }
 
  // Click handling
  if (ImGui::IsWindowHovered() &&
      ImGui::IsMouseClicked(ImGuiMouseButton_Left)) {
    selected_sheet_ = sheet_index;
  }
 
  // Double-click to open in pixel editor
  if (ImGui::IsWindowHovered() &&
      ImGui::IsMouseDoubleClicked(ImGuiMouseButton_Left)) {
    OpenSheetInPixelEditor();
  }
 
  // Sheet label
  ImGui::SetCursorPosY(ImGui::GetCursorPosY() + kThumbnailSize / 4 + 2);
  ImGui::Text("%d", sheet_index);
 
  ImGui::EndChild();
 
  sel_bg_guard.reset();
 
  // Tooltip
  if (ImGui::IsItemHovered()) {
    ImGui::BeginTooltip();
    ImGui::Text("Link Sheet %d", sheet_index);
    ImGui::Text("Double-click to edit");
    ImGui::EndTooltip();
  }
}
 
void LinkSpriteView::DrawPreviewCanvas() {
  ImGui::Text("Sheet %d Preview", selected_sheet_);
 
  // Preview canvas
  float canvas_width = ImGui::GetContentRegionAvail().x - 16;
  float canvas_height = canvas_width / 4;  // 4:1 aspect ratio (128x32)
 
  preview_canvas_.SetCanvasSize(ImVec2(canvas_width, canvas_height));
  const float grid_step = 8.0f * (canvas_width / 128.0f);
  {
    gui::CanvasFrameOptions frame_opts;
    frame_opts.canvas_size = ImVec2(canvas_width, canvas_height);
    frame_opts.draw_context_menu = false;
    frame_opts.draw_grid = true;
    frame_opts.grid_step = grid_step;
 
    auto rt = gui::BeginCanvas(preview_canvas_, frame_opts);
 
    auto& sheet = link_sheets_[selected_sheet_];
    if (sheet.is_active() && sheet.texture()) {
      gui::BitmapDrawOpts draw_opts;
      draw_opts.dest_pos = ImVec2(0, 0);
      draw_opts.dest_size = ImVec2(canvas_width, canvas_height);
      draw_opts.ensure_texture = false;
      gui::DrawBitmap(rt, sheet, draw_opts);
    }
 
    gui::EndCanvas(preview_canvas_, rt, frame_opts);
  }
 
  ImGui::Spacing();
 
  // Open in editor button
  if (ImGui::Button(ICON_MD_EDIT " Open in Pixel Editor")) {
    OpenSheetInPixelEditor();
  }
  HOVER_HINT("Open this sheet in the main pixel editor");
 
  // Zoom slider
  ImGui::SameLine();
  ImGui::SetNextItemWidth(gui::LayoutHelpers::GetSliderWidth());
  ImGui::SliderFloat("Zoom", &preview_zoom_, 1.0f, 8.0f, "%.1fx");
}
 
void LinkSpriteView::DrawPaletteSelector() {
  ImGui::Text("Display Palette:");
  ImGui::SameLine();
 
  const char* palette_names[] = {"Green Mail", "Blue Mail", "Red Mail",
                                 "Bunny"};
  int current = static_cast<int>(selected_palette_);
 
  ImGui::SetNextItemWidth(gui::LayoutHelpers::GetComboWidth());
  if (ImGui::Combo("##PaletteSelect", &current, palette_names, 4)) {
    selected_palette_ = static_cast<PaletteType>(current);
    ApplySelectedPalette();
  }
  HOVER_HINT("Change the display palette for preview");
}
 
void LinkSpriteView::DrawInfoPanel() {
  ImGui::Text("Info:");
  ImGui::BulletText("896 total tiles (8x8 each)");
  ImGui::BulletText("14 graphics sheets");
  ImGui::BulletText("4BPP format");
 
  if (loaded_zspr_.has_value()) {
    ImGui::Separator();
    ImGui::Text("Loaded ZSPR:");
    ImGui::BulletText("Name: %s", loaded_zspr_->metadata.display_name.c_str());
    ImGui::BulletText("Author: %s", loaded_zspr_->metadata.author.c_str());
    ImGui::BulletText("Tiles: %zu", loaded_zspr_->tile_count());
  }
}
 
void LinkSpriteView::ImportZspr() {
  // Open file dialog for .zspr files
  auto file_path = util::FileDialogWrapper::ShowOpenFileDialog();
  if (file_path.empty()) {
    return;
  }
 
  LOG_INFO("LinkSpriteView", "Importing ZSPR: %s", file_path.c_str());
 
  // Load ZSPR file
  auto zspr_result = gfx::ZsprLoader::LoadFromFile(file_path);
  if (!zspr_result.ok()) {
    LOG_ERROR("LinkSpriteView", "Failed to load ZSPR: %s",
              zspr_result.status().message().data());
    return;
  }
 
  loaded_zspr_ = std::move(zspr_result.value());
 
  // Verify it's a Link sprite
  if (!loaded_zspr_->is_link_sprite()) {
    LOG_ERROR("LinkSpriteView", "ZSPR is not a Link sprite (type=%d)",
              loaded_zspr_->metadata.sprite_type);
    loaded_zspr_.reset();
    return;
  }
 
  // Apply to ROM
  if (rom_ && rom_->is_loaded()) {
    auto status = gfx::ZsprLoader::ApplyToRom(*rom_, *loaded_zspr_);
    if (!status.ok()) {
      LOG_ERROR("LinkSpriteView", "Failed to apply ZSPR to ROM: %s",
                status.message().data());
      return;
    }
 
    // Also apply palette
    status = gfx::ZsprLoader::ApplyPaletteToRom(*rom_, *loaded_zspr_);
    if (!status.ok()) {
      LOG_WARN("LinkSpriteView", "Failed to apply ZSPR palette: %s",
               status.message().data());
    }
 
    // Reload Link sheets to reflect changes
    sheets_loaded_ = false;
    has_unsaved_changes_ = true;
 
    LOG_INFO("LinkSpriteView", "ZSPR '%s' imported successfully",
             loaded_zspr_->metadata.display_name.c_str());
  }
}
 
void LinkSpriteView::ResetToVanilla() {
  // TODO: Implement reset to vanilla
  // This would require keeping a backup of the original Link graphics
  // or reloading from a vanilla ROM file
  LOG_WARN("LinkSpriteView", "Reset to vanilla not yet implemented");
  loaded_zspr_.reset();
}
 
void LinkSpriteView::OpenSheetInPixelEditor() {
  // Signal to open the selected Link sheet in the main pixel editor
  // Link sheets are separate from the main 223 sheets, so we need
  // a special handling mechanism
 
  // For now, log the intent - full integration requires additional state
  LOG_INFO("LinkSpriteView", "Request to open Link sheet %d in pixel editor",
           selected_sheet_);
 
  // TODO: Add Link sheet to open_sheets with a special identifier
  // or add a link_sheets_to_edit set to GraphicsEditorState
}
 
absl::Status LinkSpriteView::LoadLinkSheets() {
  if (!rom_ || !rom_->is_loaded()) {
    return absl::FailedPreconditionError("ROM not loaded");
  }
 
  // Use the existing LoadLinkGraphics function
  auto result = zelda3::LoadLinkGraphics(*rom_);
  if (!result.ok()) {
    return result.status();
  }
 
  link_sheets_ = std::move(result.value());
  sheets_loaded_ = true;
 
  LOG_INFO("LinkSpriteView", "Loaded %d Link graphics sheets",
           zelda3::kNumLinkSheets);
 
  // Apply default palette for display
  ApplySelectedPalette();
 
  return absl::OkStatus();
}
 
void LinkSpriteView::ApplySelectedPalette() {
  if (!rom_ || !rom_->is_loaded())
    return;
 
  // Get the appropriate palette based on selection
  // Link palettes are in Group 4 (Sprites Aux1) and Group 5 (Sprites Aux2)
  // Green Mail: Group 4, Index 0 (Standard Link)
  // Blue Mail: Group 4, Index 0 (Standard Link) - but with different colors in game
  // Red Mail: Group 4, Index 0 (Standard Link) - but with different colors in game
  // Bunny: Group 4, Index 1 (Bunny Link)
 
  // For now, we'll use the standard sprite palettes from GameData if available
  // In a full implementation, we would load the specific mail palettes
 
  // Default to Green Mail (Standard Link palette)
  const gfx::SnesPalette* palette = nullptr;
 
  // We need access to GameData to get the palettes
  // Since we don't have direct access to GameData here (only Rom), we'll try to find it
  // or use a hardcoded fallback if necessary.
  // Ideally, LinkSpriteView should have access to GameData.
  // For this fix, we will assume the standard sprite palette location in ROM if GameData isn't available,
  // or use a simplified approach.
 
  // Actually, we can get GameData from the main Editor instance if we had access,
  // but we only have Rom. Let's try to read the palette directly from ROM for now
  // to ensure it works without refactoring the whole dependency injection.
 
  // Standard Link Palette (Green Mail) is usually at 0x1BD318 (PC) / 0x37D318 (SNES) in vanilla
  // But we should use the loaded palette data if possible.
 
  // Let's use a safe fallback: Create a default Link palette
  static gfx::SnesPalette default_palette;
  if (default_palette.empty()) {
    // Basic Green Mail colors (approximate)
    default_palette.Resize(16);
    default_palette[0] = gfx::SnesColor(0, 0, 0);        // Transparent
    default_palette[1] = gfx::SnesColor(24, 24, 24);     // Tunic Dark
    default_palette[2] = gfx::SnesColor(0, 19, 0);       // Tunic Green
    default_palette[3] = gfx::SnesColor(255, 255, 255);  // White
    default_palette[4] = gfx::SnesColor(255, 165, 66);   // Skin
    default_palette[5] = gfx::SnesColor(255, 100, 50);   // Skin Dark
    default_palette[6] = gfx::SnesColor(255, 0, 0);      // Red
    default_palette[7] = gfx::SnesColor(255, 255, 0);    // Yellow
    // ... fill others as needed
  }
 
  // If we can't get the real palette, use default
  palette = &default_palette;
 
  // Apply to all Link sheets
  for (auto& sheet : link_sheets_) {
    if (sheet.is_active() && sheet.surface()) {
      // Use the palette
      sheet.SetPaletteWithTransparent(*palette, 0);
 
      // Force texture update
      gfx::Arena::Get().QueueTextureCommand(
          gfx::Arena::TextureCommandType::UPDATE, &sheet);
    }
  }
 
  LOG_INFO("LinkSpriteView", "Applied palette %s to %zu sheets",
           GetPaletteName(selected_palette_), link_sheets_.size());
}
 
const char* LinkSpriteView::GetPaletteName(PaletteType type) {
  switch (type) {
    case PaletteType::kGreenMail:
      return "Green Mail";
    case PaletteType::kBlueMail:
      return "Blue Mail";
    case PaletteType::kRedMail:
      return "Red Mail";
    case PaletteType::kBunny:
      return "Bunny";
    default:
      return "Unknown";
  }
}
 
}  // namespace editor
}  // namespace yaze