gfx_sheet_png_commands.cc

Go to the documentation of this file.

#include "cli/handlers/graphics/gfx_sheet_png_commands.h"
 
#include <algorithm>
#include <cctype>
#include <filesystem>
#include <optional>
#include <string>
#include <vector>
 
#include "absl/status/status.h"
#include "absl/strings/match.h"
#include "absl/strings/numbers.h"
#include "absl/strings/str_format.h"
#include "absl/strings/str_join.h"
#include "absl/strings/str_split.h"
#include "absl/strings/strip.h"
#include "core/gfx_sheet_policy_adapter.h"
#include "core/project.h"
#include "nlohmann/json.hpp"
#include "rom/rom.h"
#include "rom/rom_diff.h"
#include "util/hex.h"
#include "util/indexed_png.h"
#include "util/macro.h"
#include "zelda3/game_data.h"
#include "zelda3/gfx_sheet_inventory.h"
#include "zelda3/gfx_sheet_png.h"
#include "zelda3/gfx_sheet_storage.h"
 
namespace yaze {
namespace cli {
namespace {
 
absl::StatusOr<int> ParseNumber(absl::string_view text, const char* what) {
  text = absl::StripAsciiWhitespace(text);
  // yaze::util::ParseHexString instead of absl::SimpleHexAtoi: Ubuntu 22.04's
  // system Abseil (20210324) predates SimpleHexAtoi.
  const bool hex = absl::StartsWith(text, "0x") ||
                   absl::StartsWith(text, "0X") || absl::StartsWith(text, "$");
  int value = 0;
  const bool ok = hex ? yaze::util::ParseHexString(text, &value)
                      : absl::SimpleAtoi(text, &value);
  if (!ok) {
    return absl::InvalidArgumentError(
        absl::StrFormat("%s '%s' is not a number", what, std::string(text)));
  }
  return value;
}
 
absl::StatusOr<uint16_t> ParseSheet(const resources::ArgumentParser& parser) {
  auto text = parser.GetString("sheet");
  if (!text.has_value()) {
    return absl::InvalidArgumentError("--sheet is required");
  }
  ASSIGN_OR_RETURN(int sheet, ParseNumber(*text, "--sheet"));
  if (sheet < 0 || sheet >= static_cast<int>(zelda3::kGfxSheetCount)) {
    return absl::InvalidArgumentError("--sheet must be 0-222");
  }
  if (zelda3::GetGfxSheetStorageKind(static_cast<uint16_t>(sheet)) ==
      zelda3::GfxSheetStorageKind::kCompressed2bpp) {
    return absl::UnimplementedError(
        "2bpp sheets (113, 114, 218+) are not supported");
  }
  return static_cast<uint16_t>(sheet);
}
 
absl::StatusOr<int> ParseIntArg(const resources::ArgumentParser& parser,
                                const char* name, int fallback) {
  auto text = parser.GetString(name);
  if (!text.has_value()) {
    return fallback;
  }
  return ParseNumber(*text, name);
}
 
absl::StatusOr<zelda3::SheetPalette> ResolvePalette(
    Rom& rom, const resources::ArgumentParser& parser, int sheet = -1,
    std::optional<zelda3::RoomSheetPaletteContext>* context = nullptr) {
  const std::string spec = parser.GetString("palette").value_or("gray");
  if (spec == "gray") {
    return zelda3::GrayscaleSheetPalette();
  }
  if (absl::StartsWith(spec, "rgb:")) {
    std::vector<std::string> parts = absl::StrSplit(spec.substr(4), ',');
    if (parts.size() != 8) {
      return absl::InvalidArgumentError("rgb: palettes need 8 colors");
    }
    zelda3::SheetPalette palette;
    for (int i = 0; i < 8; ++i) {
      absl::string_view hex = absl::StripAsciiWhitespace(parts[i]);
      absl::ConsumePrefix(&hex, "#");
      uint32_t rgb = 0;
      if (hex.size() != 6 ||
          !std::all_of(hex.begin(), hex.end(),
                       [](unsigned char c) { return std::isxdigit(c) != 0; }) ||
          !yaze::util::ParseHexString(hex, &rgb)) {
        return absl::InvalidArgumentError("rgb: colors are #RRGGBB");
      }
      palette[i] = {static_cast<uint8_t>(rgb >> 16),
                    static_cast<uint8_t>(rgb >> 8), static_cast<uint8_t>(rgb)};
    }
    return palette;
  }
  if (absl::StartsWith(spec, "room-slot:")) {
    const std::vector<std::string> parts = absl::StrSplit(spec, ':');
    if (parts.size() != 4 || sheet < 0) {
      return absl::InvalidArgumentError(
          "room-slot:<room>:<slot>:<row> requires a single --sheet; use "
          "gfx-room-export --out-dir for per-slot room palettes");
    }
    ASSIGN_OR_RETURN(const int room, ParseNumber(parts[1], "room"));
    ASSIGN_OR_RETURN(const int slot, ParseNumber(parts[2], "slot"));
    ASSIGN_OR_RETURN(const int row, ParseNumber(parts[3], "row"));
    zelda3::GameData data;
    zelda3::LoadOptions options;
    options.load_graphics = false;
    options.expand_rom = false;
    RETURN_IF_ERROR(zelda3::LoadGameData(rom, data, options));
    ASSIGN_OR_RETURN(auto resolved, zelda3::ResolveRoomSheetPalette(
                                        rom, data, room, slot, row));
    if (resolved.background.sheets[slot] != sheet) {
      return absl::InvalidArgumentError(absl::StrFormat(
          "Room 0x%03X slot %d loads sheet 0x%02X, not --sheet 0x%02X", room,
          slot, resolved.background.sheets[slot], sheet));
    }
    if (context != nullptr)
      *context = resolved;
    return resolved.colors;
  }
  const bool background = absl::StartsWith(spec, "room-bg:");
  if (background || absl::StartsWith(spec, "room-sprite:")) {
    std::vector<std::string> parts = absl::StrSplit(spec, ':');
    if (parts.size() != 3) {
      return absl::InvalidArgumentError(
          "Room palettes are room-bg:<room>:<row> or room-sprite:<room>:<row>");
    }
    ASSIGN_OR_RETURN(int room, ParseNumber(parts[1], "room"));
    ASSIGN_OR_RETURN(int row, ParseNumber(parts[2], "row"));
    zelda3::GameData data;
    zelda3::LoadOptions options;
    options.load_graphics = false;
    options.expand_rom = false;
    RETURN_IF_ERROR(zelda3::LoadGameData(rom, data, options));
    return background ? zelda3::RoomBackgroundSheetPalette(rom, data, room, row)
                      : zelda3::RoomSpriteSheetPalette(rom, data, room, row);
  }
  return absl::InvalidArgumentError("Unknown --palette " + spec);
}
 
std::string PaletteText(const zelda3::SheetPalette& palette) {
  return absl::StrJoin(palette, ",", [](std::string* out, const auto& color) {
    out->append(absl::StrFormat("#%02X%02X%02X", color[0], color[1], color[2]));
  });
}
 
nlohmann::json PaletteContextJson(
    const zelda3::RoomSheetPaletteContext& context) {
  using Source = zelda3::RoomMainBlocksetSource;
  const char* main_source = "zero_fallback";
  switch (context.main.source) {
    case Source::kExplicitPreview:
      main_source = "explicit_preview";
      break;
    case Source::kDirectDefault:
      main_source = "direct_default";
      break;
    case Source::kInferredDefault:
      main_source = "inferred_default";
      break;
    case Source::kHeaderFallback:
      main_source = "header_fallback";
      break;
    case Source::kZeroFallback:
      break;
  }
  using EntranceSource = zelda3::RoomEntranceSource;
  const char* entrance_source = "none";
  switch (context.entrance.source) {
    case EntranceSource::kDirect:
      entrance_source = "direct";
      break;
    case EntranceSource::kDungeonMap:
      entrance_source = "dungeon_map";
      break;
    case EntranceSource::kNeighbour:
      entrance_source = "neighbour";
      break;
    case EntranceSource::kNone:
      break;
  }
  return {{"room", context.background.room},
          {"slot", context.slot},
          {"sheet", context.background.sheets[context.slot]},
          {"main_blockset", context.main.main_blockset},
          {"main_source", main_source},
          {"room_blockset", context.background.header_blockset},
          {"sheet_source", context.background.from_room_blockset[context.slot]
                               ? "room_blockset"
                               : "main_blockset"},
          {"entrance_id", context.entrance.entrance_id},
          {"entrance_source", entrance_source},
          {"entrance_via_room", context.entrance.via_room},
          {"palette_row", context.row},
          {"palette_half", context.palette_offset == 8 ? "right" : "left"},
          {"raw_index_to_cgram", context.cgram_indices},
          {"colors_rgb", context.colors},
          {"transparent_index", 0},
          {"animation_overlay_included", false}};
}
 
// Check all output aliases before the first write, including symlinks and hard
// links. PNG/JSON destinations must not destroy each other or the source ROM.
absl::Status ValidateExportPaths(const Rom& rom,
                                 const std::vector<std::string>& outputs) {
  std::vector<std::string> seen;
  if (!rom.filename().empty())
    seen.push_back(rom.filename());
  for (const auto& path : outputs) {
    if (path.empty())
      return absl::InvalidArgumentError("Empty export path");
    std::error_code ec;
    const auto canonical = std::filesystem::weakly_canonical(path, ec);
    if (ec)
      return absl::InvalidArgumentError("Invalid export path: " + path);
    for (const auto& other : seen) {
      const bool equivalent = std::filesystem::equivalent(path, other, ec);
      ec.clear();
      const auto other_canonical = std::filesystem::weakly_canonical(other, ec);
      if (equivalent || (!ec && canonical == other_canonical)) {
        return absl::InvalidArgumentError(
            "Export destinations must differ from each other and the input "
            "ROM");
      }
    }
    seen.push_back(path);
  }
  return absl::OkStatus();
}
 
absl::Status WriteJson(const std::string& path, const nlohmann::json& value) {
  const std::string text = value.dump(2) + "\n";
  return util::WriteBinaryFile(path, {text.begin(), text.end()});
}
 
std::string RangeText(const std::vector<std::pair<int, int>>& ranges) {
  return absl::StrJoin(ranges, " ", [](std::string* out, const auto& range) {
    out->append(
        absl::StrFormat("0x%03X-0x%03X", range.first, range.second - 1));
  });
}
 
struct RoomContext {
  zelda3::GameData data;
  zelda3::RoomBackgroundSet set;
  int row = 2;
  zelda3::SheetPalette palette;
};
 
absl::StatusOr<RoomContext> LoadRoomContext(
    Rom& rom, const resources::ArgumentParser& parser) {
  auto room_text = parser.GetString("room");
  if (!room_text.has_value()) {
    return absl::InvalidArgumentError("--room is required");
  }
  ASSIGN_OR_RETURN(const int room, ParseNumber(*room_text, "--room"));
  if (room < 0 || room >= zelda3::kNumDungeonRooms) {
    return absl::InvalidArgumentError("--room must be 0x000-0x127");
  }
  RoomContext context;
  ASSIGN_OR_RETURN(context.row, ParseIntArg(parser, "row", 2));
  zelda3::LoadOptions options;
  options.load_graphics = false;
  options.expand_rom = false;
  RETURN_IF_ERROR(zelda3::LoadGameData(rom, context.data, options));
  const auto rooms = zelda3::CollectRoomGfx(rom);
  const auto header = rooms.find(room);
  if (header == rooms.end()) {
    return absl::DataLossError(
        absl::StrFormat("Room 0x%03X header is unreadable", room));
  }
  context.set = zelda3::ResolveRoomBackgroundSet(context.data, room,
                                                 header->second.blockset);
  if (parser.GetString("palette").has_value()) {
    ASSIGN_OR_RETURN(context.palette, ResolvePalette(rom, parser));
  } else {
    ASSIGN_OR_RETURN(context.palette,
                     zelda3::RoomBackgroundSheetPalette(rom, context.data, room,
                                                        context.row));
  }
  return context;
}
 
absl::Status ExportRoomBundle(Rom& rom, RoomContext& context,
                              const std::string& directory,
                              resources::OutputFormatter& formatter) {
  if (directory.empty())
    return absl::InvalidArgumentError("Empty --out-dir");
  // Resolve and encode every supported slot before creating any output. Each
  // PNG keeps raw indices 0-7 while carrying that slot's display palette.
  std::vector<std::pair<std::string, std::vector<uint8_t>>> files;
  nlohmann::json manifest = {{"schema_version", 1},
                             {"room", context.set.room},
                             {"palette_row", context.row},
                             {"source_rom", rom.filename()},
                             {"slots", nlohmann::json::array()}};
  for (int slot = 0; slot < 8; ++slot) {
    ASSIGN_OR_RETURN(
        const auto palette_context,
        zelda3::ResolveRoomSheetPalette(rom, context.data, context.set.room,
                                        slot, context.row));
    auto entry = PaletteContextJson(palette_context);
    const uint16_t sheet = context.set.sheets[slot];
    if (sheet >= zelda3::kGfxSheetCount ||
        zelda3::GetGfxSheetStorageKind(sheet) ==
            zelda3::GfxSheetStorageKind::kCompressed2bpp) {
      entry["status"] = "unsupported_sheet_format";
    } else {
      ASSIGN_OR_RETURN(const auto raw, zelda3::ReadGfxSheetData(rom, sheet));
      ASSIGN_OR_RETURN(
          auto png, zelda3::ExportSheetBlocksPng(raw, palette_context.colors));
      const std::string name =
          absl::StrFormat("slot-%d-sheet-%02X.png", slot, sheet);
      entry["status"] = "exported";
      entry["png"] = name;
      entry["palette_spec"] = absl::StrFormat(
          "room-slot:0x%03X:%d:%d", context.set.room, slot, context.row);
      files.emplace_back(name, std::move(png));
    }
    manifest["slots"].push_back(std::move(entry));
  }
  std::error_code ec;
  if (!std::filesystem::create_directory(directory, ec)) {
    return absl::FailedPreconditionError(
        "--out-dir must be a new directory in an existing parent: " +
        directory + (ec ? " (" + ec.message() + ")" : ""));
  }
  auto write = [&]() -> absl::Status {
    for (const auto& [name, png] : files) {
      RETURN_IF_ERROR(util::WriteBinaryFile(
          (std::filesystem::path(directory) / name).string(), png));
    }
    return WriteJson(
        (std::filesystem::path(directory) / "manifest.json").string(),
        manifest);
  };
  const absl::Status status = write();
  if (!status.ok()) {
    // This invocation created the directory; never clean up a preexisting one.
    std::filesystem::remove_all(directory, ec);
    return status;
  }
  formatter.AddField("out_dir", directory);
  formatter.AddRawJsonField("manifest", manifest.dump());
  return absl::OkStatus();
}
 
std::string HexIds(const std::vector<int>& ids) {
  return absl::StrJoin(ids, " ", [](std::string* out, int id) {
    out->append(absl::StrFormat("%03X", id));
  });
}
 
// These commands use the US graphics pointer tables; on a Japanese ROM they
// would read and write the wrong sheets.
absl::Status RequireUsGfxTables(const Rom& rom) {
  if (zelda3::UsesJapaneseGfxSheetTables(rom)) {
    return absl::FailedPreconditionError(
        "This ROM uses the Japanese graphics pointer tables (header byte "
        "$7FD9 = 0); the gfx commands support US-table ROMs only");
  }
  return absl::OkStatus();
}
 
}  // namespace
 
absl::Status GfxExportCommandHandler::ValidateArgs(
    const resources::ArgumentParser& parser) {
  return parser.RequireArgs({"sheet", "png"});
}
 
absl::Status GfxExportCommandHandler::Execute(
    Rom* rom, const resources::ArgumentParser& parser,
    resources::OutputFormatter& formatter) {
  RETURN_IF_ERROR(RequireUsGfxTables(*rom));
  ASSIGN_OR_RETURN(const uint16_t sheet, ParseSheet(parser));
  ASSIGN_OR_RETURN(const int block, ParseIntArg(parser, "block", 0));
  ASSIGN_OR_RETURN(const int count, ParseIntArg(parser, "count", 16 - block));
  std::optional<zelda3::RoomSheetPaletteContext> context;
  ASSIGN_OR_RETURN(const auto palette,
                   ResolvePalette(*rom, parser, sheet, &context));
  ASSIGN_OR_RETURN(const auto data, zelda3::ReadGfxSheetData(*rom, sheet));
  ASSIGN_OR_RETURN(const auto png,
                   zelda3::ExportSheetBlocksPng(data, palette, block, count));
  const std::string path = *parser.GetString("png");
  const auto json_path = parser.GetString("json");
  std::vector<std::string> outputs = {path};
  if (json_path.has_value())
    outputs.push_back(*json_path);
  RETURN_IF_ERROR(ValidateExportPaths(*rom, outputs));
  nlohmann::json metadata = {{"sheet", sheet},        {"png", path},
                             {"first_block", block},  {"block_count", count},
                             {"colors_rgb", palette}, {"transparent_index", 0}};
  if (context.has_value())
    metadata["room_context"] = PaletteContextJson(*context);
  RETURN_IF_ERROR(util::WriteBinaryFile(path, png));
  if (json_path.has_value())
    RETURN_IF_ERROR(WriteJson(*json_path, metadata));
 
  formatter.AddField("sheet", absl::StrFormat("0x%02X", sheet));
  formatter.AddField("png", path);
  formatter.AddField("blocks",
                     absl::StrFormat("%d-%d", block, block + count - 1));
  formatter.AddField("width", std::min(count, 8) * 16);
  formatter.AddField("height", (count + 7) / 8 * 16);
  formatter.AddField("palette", PaletteText(palette));
  if (context.has_value()) {
    formatter.AddRawJsonField("room_context", metadata["room_context"].dump());
  }
  return absl::OkStatus();
}
 
absl::Status GfxImportCommandHandler::ValidateArgs(
    const resources::ArgumentParser& parser) {
  RETURN_IF_ERROR(parser.RequireArgs({"sheet", "png"}));
  const bool dry_run = parser.HasFlag("dry-run");
  const bool write = parser.HasFlag("write");
  if (dry_run == write) {
    return absl::InvalidArgumentError("Pass exactly one of --dry-run, --write");
  }
  if (write && !parser.GetString("out").has_value()) {
    return absl::InvalidArgumentError(
        "--write needs --out <rom file>; the input ROM is never overwritten");
  }
  return absl::OkStatus();
}
 
absl::Status GfxImportCommandHandler::Execute(
    Rom* rom, const resources::ArgumentParser& parser,
    resources::OutputFormatter& formatter) {
  RETURN_IF_ERROR(RequireUsGfxTables(*rom));
  ASSIGN_OR_RETURN(const uint16_t sheet, ParseSheet(parser));
  ASSIGN_OR_RETURN(const int block, ParseIntArg(parser, "block", 0));
  std::optional<zelda3::RoomSheetPaletteContext> context;
  ASSIGN_OR_RETURN(const auto palette,
                   ResolvePalette(*rom, parser, sheet, &context));
  if (context.has_value()) {
    formatter.AddRawJsonField("room_context",
                              PaletteContextJson(*context).dump());
  }
  ASSIGN_OR_RETURN(const auto current, zelda3::ReadGfxSheetData(*rom, sheet));
  const std::string png_path = *parser.GetString("png");
  ASSIGN_OR_RETURN(const auto png_bytes, util::ReadBinaryFile(png_path));
  ASSIGN_OR_RETURN(const auto png, util::DecodePng(png_bytes));
  ASSIGN_OR_RETURN(const auto imported,
                   zelda3::ImportSheetBlocksPng(current, png, block, palette));
 
  // Run the real writer on a copy of the ROM buffer; the diff is exactly
  // what --write would change.
  zelda3::GfxSheetWritePolicy policy;
  if (project_ != nullptr) {
    ASSIGN_OR_RETURN(policy, core::BuildGfxSheetWritePolicy(*project_));
  }
  Rom simulated = *rom;
  const auto written =
      zelda3::WriteGfxSheet(simulated, sheet, imported.snes_3bpp, policy);
 
  formatter.AddField("sheet", absl::StrFormat("0x%02X", sheet));
  formatter.AddField("png", png_path);
  formatter.AddField("changed_blocks",
                     absl::StrJoin(imported.changed_blocks, " "));
  formatter.AddField("changed_tiles",
                     static_cast<int>(imported.changed_tiles.size()));
  formatter.AddField("sheet_byte_ranges", RangeText(imported.changed_ranges));
  if (!written.ok()) {
    // The preview still stands; say why the write would be refused.
    formatter.AddField("write_status", std::string(written.status().message()));
    if (parser.HasFlag("dry-run")) {
      formatter.AddField("mode", "dry-run");
      return absl::OkStatus();
    }
    return written.status();
  }
  formatter.AddField("write_status", "ok");
  const auto rom_diff =
      rom::ComputeDiffRanges(rom->vector(), simulated.vector());
  formatter.AddField("placement",
                     written->placement == zelda3::GfxSheetPlacement::kRelocated
                         ? "relocated"
                         : "in place");
  formatter.AddField("rom_pc", absl::StrFormat("0x%06X", written->new_pc));
  formatter.AddField("stored_bytes",
                     absl::StrFormat("%zu -> %zu", written->old_stored_size,
                                     written->new_stored_size));
  formatter.AddField("rom_bytes_changed",
                     static_cast<int>(rom_diff.total_bytes_changed));
  formatter.AddField(
      "rom_ranges",
      absl::StrJoin(rom_diff.ranges, " ",
                    [](std::string* out, const auto& range) {
                      out->append(absl::StrFormat("0x%06X-0x%06X", range.first,
                                                  range.second - 1));
                    }));
 
  if (parser.HasFlag("dry-run")) {
    formatter.AddField("mode", "dry-run");
    return absl::OkStatus();
  }
  const std::string out = *parser.GetString("out");
  std::error_code ec;
  if (!rom->filename().empty() &&
      std::filesystem::equivalent(out, rom->filename(), ec)) {
    return absl::InvalidArgumentError(
        "--out is the input ROM; write to a new file");
  }
  RETURN_IF_ERROR(util::WriteBinaryFile(out, simulated.vector()));
  formatter.AddField("mode", "write");
  formatter.AddField("out", out);
  return absl::OkStatus();
}
 
absl::Status GfxRoomExportCommandHandler::ValidateArgs(
    const resources::ArgumentParser& parser) {
  RETURN_IF_ERROR(parser.RequireArgs({"room"}));
  const bool bundle = parser.GetString("out-dir").has_value();
  if (bundle == parser.GetString("png").has_value()) {
    return absl::InvalidArgumentError("Pass exactly one of --png or --out-dir");
  }
  if (bundle && (parser.GetString("palette").has_value() ||
                 parser.GetString("json").has_value())) {
    return absl::InvalidArgumentError(
        "--out-dir resolves each slot's palette and writes manifest.json; "
        "do not pass --palette or --json");
  }
  return absl::OkStatus();
}
 
absl::Status GfxRoomExportCommandHandler::Execute(
    Rom* rom, const resources::ArgumentParser& parser,
    resources::OutputFormatter& formatter) {
  RETURN_IF_ERROR(RequireUsGfxTables(*rom));
  ASSIGN_OR_RETURN(auto context, LoadRoomContext(*rom, parser));
  if (const auto directory = parser.GetString("out-dir");
      directory.has_value()) {
    return ExportRoomBundle(*rom, context, *directory, formatter);
  }
  const auto& set = context.set;
  ASSIGN_OR_RETURN(const auto png,
                   zelda3::ExportRoomBackgroundPng(*rom, set, context.palette));
  const std::string png_path = *parser.GetString("png");
  std::vector<std::string> outputs = {png_path};
  if (const auto json_path = parser.GetString("json"); json_path.has_value()) {
    outputs.push_back(*json_path);
  }
  RETURN_IF_ERROR(ValidateExportPaths(*rom, outputs));
  RETURN_IF_ERROR(util::WriteBinaryFile(png_path, png));
 
  // Who else sees these sheets: an edit changes every one of these rooms.
  const auto all_sets =
      zelda3::ResolveAllRoomBackgroundSets(*rom, context.data);
  std::vector<int> same_main, same_room_blockset;
  for (const auto& other : all_sets) {
    if (other.main_blockset == set.main_blockset) {
      same_main.push_back(other.room);
    }
    if (other.header_blockset == set.header_blockset) {
      same_room_blockset.push_back(other.room);
    }
  }
  nlohmann::json slots = nlohmann::json::array();
  for (int slot = 0; slot < 8; ++slot) {
    std::vector<int> rooms;
    for (const auto& other : all_sets) {
      if (std::find(other.sheets.begin(), other.sheets.end(),
                    set.sheets[slot]) != other.sheets.end()) {
        rooms.push_back(other.room);
      }
    }
    slots.push_back({{"slot", slot},
                     {"sheet", absl::StrFormat("0x%02X", set.sheets[slot])},
                     {"source", set.from_room_blockset[slot] ? "room_blockset"
                                                             : "main_blockset"},
                     {"png_y", slot * 32},
                     {"rooms_using_sheet", rooms}});
  }
  const std::string rom_path = rom->filename();
  const std::string import_command = absl::StrFormat(
      "z3ed gfx-room-import --rom %s --room 0x%03X --png %s --row %d --dry-run",
      rom_path, set.room, png_path, context.row);
  nlohmann::json sidecar = {
      {"room", absl::StrFormat("0x%03X", set.room)},
      {"main_blockset", set.main_blockset},
      {"room_blockset", set.header_blockset},
      {"palette_row", context.row},
      {"palette", PaletteText(context.palette)},
      {"palette_mode", "uniform_palette_all_slots"},
      {"png", png_path},
      {"slots", slots},
      {"rooms_with_same_main_blockset", same_main},
      {"rooms_with_same_room_blockset", same_room_blockset},
      {"reimport", import_command},
  };
  if (auto json_path = parser.GetString("json"); json_path.has_value()) {
    const std::string text = sidecar.dump(1) + "\n";
    RETURN_IF_ERROR(util::WriteBinaryFile(
        *json_path, std::vector<uint8_t>(text.begin(), text.end())));
  }
  for (const auto& [key, value] : sidecar.items()) {
    formatter.AddRawJsonField(key, value.dump());
  }
  return absl::OkStatus();
}
 
absl::Status GfxRoomImportCommandHandler::ValidateArgs(
    const resources::ArgumentParser& parser) {
  RETURN_IF_ERROR(parser.RequireArgs({"room", "png"}));
  const bool dry_run = parser.HasFlag("dry-run");
  const bool write = parser.HasFlag("write");
  if (dry_run == write) {
    return absl::InvalidArgumentError("Pass exactly one of --dry-run, --write");
  }
  if (write && !parser.GetString("out").has_value()) {
    return absl::InvalidArgumentError(
        "--write needs --out <rom file>; the input ROM is never overwritten");
  }
  return absl::OkStatus();
}
 
absl::Status GfxRoomImportCommandHandler::Execute(
    Rom* rom, const resources::ArgumentParser& parser,
    resources::OutputFormatter& formatter) {
  RETURN_IF_ERROR(RequireUsGfxTables(*rom));
  ASSIGN_OR_RETURN(auto context, LoadRoomContext(*rom, parser));
  const std::string png_path = *parser.GetString("png");
  ASSIGN_OR_RETURN(const auto png_bytes, util::ReadBinaryFile(png_path));
  ASSIGN_OR_RETURN(const auto png, util::DecodePng(png_bytes));
  ASSIGN_OR_RETURN(
      const auto imports,
      zelda3::ImportRoomBackgroundPng(*rom, context.set, png, context.palette));
 
  zelda3::GfxSheetWritePolicy policy;
  if (project_ != nullptr) {
    ASSIGN_OR_RETURN(policy, core::BuildGfxSheetWritePolicy(*project_));
  }
  Rom simulated = *rom;
  const auto all_sets =
      zelda3::ResolveAllRoomBackgroundSets(*rom, context.data);
  nlohmann::json sheets = nlohmann::json::array();
  std::string refusal;
  std::vector<int> affected_rooms;
  for (const auto& import : imports) {
    auto written = zelda3::WriteGfxSheet(simulated, import.sheet,
                                         import.result.snes_3bpp, policy);
    nlohmann::json entry = {
        {"sheet", absl::StrFormat("0x%02X", import.sheet)},
        {"slots", import.slots},
        {"changed_blocks", import.result.changed_blocks},
        {"changed_tiles", import.result.changed_tiles.size()},
    };
    if (written.ok()) {
      entry["placement"] =
          written->placement == zelda3::GfxSheetPlacement::kRelocated
              ? "relocated"
              : "in place";
      entry["stored_bytes"] = absl::StrFormat(
          "%zu -> %zu", written->old_stored_size, written->new_stored_size);
    } else {
      entry["write_status"] = std::string(written.status().message());
      if (refusal.empty()) {
        refusal = entry["write_status"];
      }
    }
    for (const auto& other : all_sets) {
      if (std::find(other.sheets.begin(), other.sheets.end(), import.sheet) !=
          other.sheets.end()) {
        affected_rooms.push_back(other.room);
      }
    }
    sheets.push_back(std::move(entry));
  }
  std::sort(affected_rooms.begin(), affected_rooms.end());
  affected_rooms.erase(
      std::unique(affected_rooms.begin(), affected_rooms.end()),
      affected_rooms.end());
 
  formatter.AddField("room", absl::StrFormat("0x%03X", context.set.room));
  formatter.AddRawJsonField("sheets", sheets.dump());
  formatter.AddField("rooms_affected", HexIds(affected_rooms));
  formatter.AddField("write_status", refusal.empty() ? "ok" : refusal);
  const auto rom_diff =
      rom::ComputeDiffRanges(rom->vector(), simulated.vector());
  formatter.AddField("rom_bytes_changed",
                     static_cast<int>(rom_diff.total_bytes_changed));
  if (parser.HasFlag("dry-run")) {
    formatter.AddField("mode", "dry-run");
    return absl::OkStatus();
  }
  if (!refusal.empty()) {
    return absl::FailedPreconditionError(refusal);
  }
  const std::string out = *parser.GetString("out");
  std::error_code ec;
  if (!rom->filename().empty() &&
      std::filesystem::equivalent(out, rom->filename(), ec)) {
    return absl::InvalidArgumentError(
        "--out is the input ROM; write to a new file");
  }
  RETURN_IF_ERROR(util::WriteBinaryFile(out, simulated.vector()));
  formatter.AddField("mode", "write");
  formatter.AddField("out", out);
  return absl::OkStatus();
}
 
}  // namespace cli
}  // namespace yaze