palette_controls_panel.cc

Go to the documentation of this file.

#include "app/editor/graphics/palette_controls_panel.h"
 
#include <algorithm>
#include <string>
#include <string_view>
 
#include "absl/strings/str_format.h"
#include "app/gfx/resource/arena.h"
#include "app/gfx/types/snes_palette.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"
 
namespace yaze {
namespace editor {
 
using gfx::kPaletteGroupAddressesKeys;
 
namespace {
struct PaletteRowLayout {
  int colors_per_row;
  bool has_explicit_transparent;
};
 
PaletteRowLayout GetPaletteRowLayout(std::string_view group_name,
                                     size_t palette_size) {
  if (group_name == "ow_main" || group_name == "ow_aux" ||
      group_name == "ow_animated" || group_name == "sprites_aux1" ||
      group_name == "sprites_aux2" || group_name == "sprites_aux3") {
    return {7, false};
  }
  if (group_name == "global_sprites" || group_name == "armors" ||
      group_name == "dungeon_main") {
    return {15, false};
  }
  if (group_name == "hud" || group_name == "ow_mini_map") {
    return {16, true};
  }
  if (group_name == "swords") {
    return {3, false};
  }
  if (group_name == "shields") {
    return {4, false};
  }
  if (group_name == "grass") {
    return {3, false};
  }
  if (group_name == "3d_object") {
    return {8, false};
  }
 
  if (palette_size % 16 == 0) {
    return {16, true};
  }
  if (palette_size % 15 == 0) {
    return {15, false};
  }
  if (palette_size % 7 == 0) {
    return {7, false};
  }
 
  int fallback = palette_size > 0 ? static_cast<int>(palette_size) : 1;
  return {fallback, false};
}
 
int GetPaletteRowCount(size_t palette_size, int colors_per_row) {
  if (colors_per_row <= 0) {
    return 1;
  }
  return static_cast<int>((palette_size + colors_per_row - 1) / colors_per_row);
}
 
bool ComputePaletteSlice(std::string_view group_name,
                         const gfx::SnesPalette& palette, int row_index,
                         size_t& out_offset, int& out_length) {
  if (palette.empty()) {
    return false;
  }
 
  const auto layout = GetPaletteRowLayout(group_name, palette.size());
  const int max_rows =
      GetPaletteRowCount(palette.size(), layout.colors_per_row);
  const int clamped_row = std::clamp(row_index, 0, std::max(0, max_rows - 1));
  const int row_offset = clamped_row * layout.colors_per_row;
  const size_t offset = static_cast<size_t>(
      row_offset + (layout.has_explicit_transparent ? 1 : 0));
  int length =
      layout.colors_per_row - (layout.has_explicit_transparent ? 1 : 0);
  length = std::clamp(length, 1, 15);
 
  if (offset >= palette.size()) {
    return false;
  }
 
  if (offset + length > palette.size()) {
    length = static_cast<int>(palette.size() - offset);
  }
 
  out_offset = offset;
  out_length = length;
  return out_length > 0;
}
}  // namespace
 
void PaletteControlsPanel::Initialize() {
  // Initialize with default palette group
  state_->palette_group_index = 0;
  state_->palette_index = 0;
  state_->sub_palette_index = 0;
}
 
void PaletteControlsPanel::Draw(bool* p_open) {
  // EditorPanel interface - delegate to existing Update() logic
  if (!rom_ || !rom_->is_loaded()) {
    ImGui::TextDisabled("Load a ROM to manage palettes");
    return;
  }
 
  DrawPresets();
  ImGui::Separator();
  DrawPaletteGroupSelector();
  ImGui::Separator();
  DrawPaletteDisplay();
  ImGui::Separator();
  DrawApplyButtons();
}
 
absl::Status PaletteControlsPanel::Update() {
  if (!rom_ || !rom_->is_loaded()) {
    ImGui::TextDisabled("Load a ROM to manage palettes");
    return absl::OkStatus();
  }
 
  DrawPresets();
  ImGui::Separator();
  DrawPaletteGroupSelector();
  ImGui::Separator();
  DrawPaletteDisplay();
  ImGui::Separator();
  DrawApplyButtons();
 
  return absl::OkStatus();
}
 
void PaletteControlsPanel::DrawPresets() {
  gui::TextWithSeparators("Quick Presets");
 
  if (ImGui::Button(ICON_MD_LANDSCAPE " Overworld")) {
    state_->palette_group_index = 0;  // Dungeon Main (used for overworld too)
    state_->palette_index = 0;
    state_->refresh_graphics = true;
  }
  HOVER_HINT("Standard overworld palette");
 
  ImGui::SameLine();
 
  if (ImGui::Button(ICON_MD_CASTLE " Dungeon")) {
    state_->palette_group_index = 0;  // Dungeon Main
    state_->palette_index = 1;
    state_->refresh_graphics = true;
  }
  HOVER_HINT("Standard dungeon palette");
 
  ImGui::SameLine();
 
  if (ImGui::Button(ICON_MD_PERSON " Sprites")) {
    state_->palette_group_index = 4;  // Sprites Aux1
    state_->palette_index = 0;
    state_->refresh_graphics = true;
  }
  HOVER_HINT("Sprite/enemy palette");
 
  if (ImGui::Button(ICON_MD_ACCOUNT_BOX " Link")) {
    state_->palette_group_index = 3;  // Sprite Aux3 (Link's palettes)
    state_->palette_index = 0;
    state_->refresh_graphics = true;
  }
  HOVER_HINT("Link's palette");
 
  ImGui::SameLine();
 
  if (ImGui::Button(ICON_MD_MENU " HUD")) {
    state_->palette_group_index = 6;  // HUD palettes
    state_->palette_index = 0;
    state_->refresh_graphics = true;
  }
  HOVER_HINT("HUD/menu palette");
}
 
void PaletteControlsPanel::DrawPaletteGroupSelector() {
  gui::TextWithSeparators("Palette Selection");
 
  // Palette group combo
  ImGui::SetNextItemWidth(gui::LayoutHelpers::GetComboWidth());
  if (ImGui::Combo("Group",
                   reinterpret_cast<int*>(&state_->palette_group_index),
                   kPaletteGroupAddressesKeys,
                   IM_ARRAYSIZE(kPaletteGroupAddressesKeys))) {
    state_->refresh_graphics = true;
  }
 
  // Palette index within group
  ImGui::SetNextItemWidth(
      gui::LayoutHelpers::GetStandardInputWidth() * 0.75f);
  int palette_idx = static_cast<int>(state_->palette_index);
  if (ImGui::InputInt("Palette", &palette_idx)) {
    state_->palette_index = static_cast<uint64_t>(std::max(0, palette_idx));
    state_->refresh_graphics = true;
  }
  HOVER_HINT("Palette index within the group");
 
  // Sub-palette index (for multi-row palettes)
  ImGui::SetNextItemWidth(
      gui::LayoutHelpers::GetStandardInputWidth() * 0.75f);
  int sub_idx = static_cast<int>(state_->sub_palette_index);
  if (ImGui::InputInt("Sub-Palette", &sub_idx)) {
    state_->sub_palette_index = static_cast<uint64_t>(std::max(0, sub_idx));
    state_->refresh_graphics = true;
  }
  HOVER_HINT("Sub-palette row (0-7 for SNES 128-color palettes)");
}
 
void PaletteControlsPanel::DrawPaletteDisplay() {
  gui::TextWithSeparators("Current Palette");
 
  // Get the current palette from GameData
  if (!game_data_)
    return;
  auto palette_group_result = game_data_->palette_groups.get_group(
      kPaletteGroupAddressesKeys[state_->palette_group_index]);
  if (!palette_group_result) {
    ImGui::TextDisabled("Invalid palette group");
    return;
  }
 
  auto palette_group = *palette_group_result;
  if (state_->palette_index >= palette_group.size()) {
    ImGui::TextDisabled("Invalid palette index");
    return;
  }
 
  auto palette = palette_group.palette(state_->palette_index);
 
  auto palette_group_name =
      std::string_view(kPaletteGroupAddressesKeys[state_->palette_group_index]);
  auto layout = GetPaletteRowLayout(palette_group_name, palette.size());
 
  int colors_per_row = layout.colors_per_row;
  int total_colors = static_cast<int>(palette.size());
  int num_rows = GetPaletteRowCount(palette.size(), colors_per_row);
  if (state_->sub_palette_index >= static_cast<uint64_t>(num_rows)) {
    state_->sub_palette_index = 0;
  }
 
  for (int row = 0; row < num_rows; row++) {
    for (int col = 0; col < colors_per_row; col++) {
      int idx = row * colors_per_row + col;
      if (idx >= total_colors)
        break;
 
      if (col > 0)
        ImGui::SameLine();
 
      auto& color = palette[idx];
      ImVec4 im_color(color.rgb().x / 255.0f, color.rgb().y / 255.0f,
                      color.rgb().z / 255.0f, 1.0f);
 
      // Highlight current sub-palette row
      bool in_sub_palette =
          (row == static_cast<int>(state_->sub_palette_index));
      std::optional<gui::StyleVarGuard> pal_border_var;
      std::optional<gui::StyleColorGuard> pal_border_color;
      if (in_sub_palette) {
        pal_border_var.emplace(ImGuiStyleVar_FrameBorderSize, 2.0f);
        pal_border_color.emplace(ImGuiCol_Border, gui::GetWarningColor());
      }
 
      std::string id = absl::StrFormat("##PalColor%d", idx);
      if (ImGui::ColorButton(id.c_str(), im_color,
                             ImGuiColorEditFlags_NoTooltip, ImVec2(18, 18))) {
        // Clicking a color in a row selects that sub-palette
        state_->sub_palette_index = static_cast<uint64_t>(row);
        state_->refresh_graphics = true;
      }
 
      pal_border_color.reset();
      pal_border_var.reset();
 
      if (ImGui::IsItemHovered()) {
        ImGui::BeginTooltip();
        ImGui::Text("Index: %d (Row %d, Col %d)", idx, row, col);
        ImGui::Text("SNES: $%04X", color.snes());
        ImGui::Text("RGB: %d, %d, %d", static_cast<int>(color.rgb().x),
                    static_cast<int>(color.rgb().y),
                    static_cast<int>(color.rgb().z));
        ImGui::EndTooltip();
      }
    }
  }
 
  // Row selection buttons
  ImGui::Text("Sub-palette Row:");
  for (int i = 0; i < std::min(8, num_rows); i++) {
    if (i > 0)
      ImGui::SameLine();
    bool selected = (state_->sub_palette_index == static_cast<uint64_t>(i));
    {
      std::optional<gui::StyleColorGuard> sel_guard;
      if (selected) {
        sel_guard.emplace(ImGuiCol_Button, gui::GetSelectedColor());
      }
      if (ImGui::SmallButton(absl::StrFormat("%d", i).c_str())) {
        state_->sub_palette_index = static_cast<uint64_t>(i);
        state_->refresh_graphics = true;
      }
    }
  }
}
 
void PaletteControlsPanel::DrawApplyButtons() {
  gui::TextWithSeparators("Apply Palette");
 
  // Apply to current sheet
  bool no_sheets = state_->open_sheets.empty();
  ImGui::BeginDisabled(no_sheets);
  if (ImGui::Button(ICON_MD_BRUSH " Apply to Current Sheet")) {
    ApplyPaletteToSheet(state_->current_sheet_id);
  }
  ImGui::EndDisabled();
  if (no_sheets && ImGui::IsItemHovered(ImGuiHoveredFlags_AllowWhenDisabled)) {
    ImGui::SetTooltip("Open a graphics sheet first");
  } else {
    HOVER_HINT("Apply palette to the currently selected sheet");
  }
 
  ImGui::SameLine();
 
  // Apply to all sheets
  if (ImGui::Button(ICON_MD_FORMAT_PAINT " Apply to All Sheets")) {
    ApplyPaletteToAllSheets();
  }
  HOVER_HINT("Apply palette to all active graphics sheets");
 
  // Apply to selected sheets (multi-select)
  if (!state_->selected_sheets.empty()) {
    if (ImGui::Button(absl::StrFormat(ICON_MD_CHECKLIST
                                      " Apply to %zu Selected",
                                      state_->selected_sheets.size())
                          .c_str())) {
      for (uint16_t sheet_id : state_->selected_sheets) {
        ApplyPaletteToSheet(sheet_id);
      }
    }
    HOVER_HINT("Apply palette to all selected sheets in browser");
  }
 
  // Refresh button
  ImGui::Separator();
  if (ImGui::Button(ICON_MD_REFRESH " Refresh Graphics")) {
    state_->refresh_graphics = true;
    if (!state_->open_sheets.empty()) {
      ApplyPaletteToSheet(state_->current_sheet_id);
    }
  }
  HOVER_HINT("Force refresh of current sheet graphics");
}
 
void PaletteControlsPanel::ApplyPaletteToSheet(uint16_t sheet_id) {
  if (!rom_ || !rom_->is_loaded() || !game_data_)
    return;
 
  auto palette_group_name =
      std::string_view(kPaletteGroupAddressesKeys[state_->palette_group_index]);
  auto palette_group_result =
      game_data_->palette_groups.get_group(std::string(palette_group_name));
  if (!palette_group_result)
    return;
 
  auto palette_group = *palette_group_result;
  if (state_->palette_index >= palette_group.size())
    return;
 
  auto palette = palette_group.palette(state_->palette_index);
  if (palette.empty()) {
    return;
  }
 
  auto& sheet = gfx::Arena::Get().mutable_gfx_sheets()->at(sheet_id);
  if (sheet.is_active() && sheet.surface()) {
    size_t palette_offset = 0;
    int palette_length = 0;
    if (ComputePaletteSlice(palette_group_name, palette,
                            static_cast<int>(state_->sub_palette_index),
                            palette_offset, palette_length)) {
      sheet.SetPaletteWithTransparent(palette, palette_offset, palette_length);
    } else {
      sheet.SetPaletteWithTransparent(
          palette, 0, std::min(7, static_cast<int>(palette.size())));
    }
    gfx::Arena::Get().NotifySheetModified(sheet_id);
  }
}
 
void PaletteControlsPanel::ApplyPaletteToAllSheets() {
  if (!rom_ || !rom_->is_loaded() || !game_data_)
    return;
 
  auto palette_group_name =
      std::string_view(kPaletteGroupAddressesKeys[state_->palette_group_index]);
  auto palette_group_result =
      game_data_->palette_groups.get_group(std::string(palette_group_name));
  if (!palette_group_result)
    return;
 
  auto palette_group = *palette_group_result;
  if (state_->palette_index >= palette_group.size())
    return;
 
  auto palette = palette_group.palette(state_->palette_index);
  if (palette.empty()) {
    return;
  }
  size_t palette_offset = 0;
  int palette_length = 0;
  const bool has_slice = ComputePaletteSlice(
      palette_group_name, palette, static_cast<int>(state_->sub_palette_index),
      palette_offset, palette_length);
 
  for (int i = 0; i < zelda3::kNumGfxSheets; i++) {
    auto& sheet = gfx::Arena::Get().mutable_gfx_sheets()->data()[i];
    if (sheet.is_active() && sheet.surface()) {
      if (has_slice) {
        sheet.SetPaletteWithTransparent(palette, palette_offset,
                                        palette_length);
      } else {
        sheet.SetPaletteWithTransparent(
            palette, 0, std::min(7, static_cast<int>(palette.size())));
      }
      gfx::Arena::Get().NotifySheetModified(i);
    }
  }
}
 
}  // namespace editor
}  // namespace yaze