game_data.cc

Go to the documentation of this file.

#include "zelda3/game_data.h"
 
#include "absl/strings/str_format.h"
#include "app/gfx/util/compression.h"
#include "core/rom_settings.h"
#include "util/log.h"
#include "util/macro.h"
#include "zelda3/dungeon/dungeon_rom_addresses.h"
 
#include <algorithm>
 
#ifdef __EMSCRIPTEN__
#include "app/platform/wasm/wasm_loading_manager.h"
#endif
 
namespace yaze {
namespace zelda3 {
 
namespace {
 
constexpr uint32_t kUncompressedSheetSize = 0x0800;
// constexpr uint32_t kTile16Ptr = 0x78000;
 
// Helper to get address from bytes
uint32_t AddressFromBytes(uint8_t bank, uint8_t high, uint8_t low) {
  return (bank << 16) | (high << 8) | low;
}
 
// Helper to convert SNES to PC address
uint32_t SnesToPc(uint32_t snes_addr) {
  return (snes_addr & 0x7FFF) | ((snes_addr & 0x7F0000) >> 1);
}
 
struct PaletteSlice {
  size_t offset = 0;
  int length = 0;
  bool valid = false;
};
 
PaletteSlice GetDefaultPaletteSlice(const gfx::SnesPalette& palette) {
  const int palette_size = static_cast<int>(palette.size());
  if (palette_size <= 0) {
    return {};
  }
 
  const bool has_explicit_transparent =
      palette_size >= 16 && (palette_size % 16 == 0);
  PaletteSlice slice;
  slice.offset = has_explicit_transparent ? 1 : 0;
  slice.length = has_explicit_transparent ? 15 : std::min(palette_size, 15);
  slice.valid =
      slice.length > 0 &&
      (slice.offset + static_cast<size_t>(slice.length) <= palette.size());
 
  if (!slice.valid) {
    slice.offset = 0;
    slice.length = std::min(palette_size, 15);
    slice.valid =
        slice.length > 0 &&
        (slice.offset + static_cast<size_t>(slice.length) <= palette.size());
  }
 
  return slice;
}
 
void ApplyDefaultPalette(gfx::Bitmap& bitmap, const gfx::SnesPalette& palette) {
  auto slice = GetDefaultPaletteSlice(palette);
  if (!slice.valid) {
    return;
  }
  bitmap.SetPaletteWithTransparent(palette, slice.offset, slice.length);
}
 
// Helper to convert PC to SNES address
// uint32_t PcToSnes(uint32_t pc_addr) {
//   return ((pc_addr & 0x7FFF) | 0x8000) | ((pc_addr & 0x3F8000) << 1);
// }
 
// ============================================================================
// Graphics Address Resolution
// ============================================================================
 
}  // namespace
 
uint32_t GetGraphicsAddress(const uint8_t* data, uint8_t addr, uint32_t ptr1,
                            uint32_t ptr2, uint32_t ptr3, size_t rom_size) {
  if (ptr1 > UINT32_MAX - addr || ptr1 + addr >= rom_size ||
      ptr2 > UINT32_MAX - addr || ptr2 + addr >= rom_size ||
      ptr3 > UINT32_MAX - addr || ptr3 + addr >= rom_size) {
    return static_cast<uint32_t>(rom_size);
  }
  return SnesToPc(AddressFromBytes(data[ptr1 + addr], data[ptr2 + addr],
                                   data[ptr3 + addr]));
}
 
absl::Status LoadGameData(Rom& rom, GameData& data,
                          const LoadOptions& options) {
  data.Clear();
  data.set_rom(&rom);
 
  if (options.populate_metadata) {
    RETURN_IF_ERROR(LoadMetadata(rom, data));
  }
 
  if (options.load_palettes) {
    RETURN_IF_ERROR(LoadPalettes(rom, data));
  }
 
  if (options.load_gfx_groups) {
    RETURN_IF_ERROR(LoadGfxGroups(rom, data));
  }
 
  if (options.load_graphics) {
    RETURN_IF_ERROR(LoadGraphics(rom, data));
  }
 
  if (options.expand_rom) {
    if (rom.size() < 1048576 * 2) {
      rom.Expand(1048576 * 2);
    }
  }
 
  return absl::OkStatus();
}
 
absl::Status SaveGameData(Rom& rom, GameData& data) {
  if (core::FeatureFlags::get().kSaveAllPalettes) {
    // TODO: Implement SaveAllPalettes logic using Rom::WriteColor
    // This was previously in Rom::SaveAllPalettes
    return data.palette_groups.for_each(
        [&](gfx::PaletteGroup& group) -> absl::Status {
          for (size_t i = 0; i < group.size(); ++i) {
            auto* palette = group.mutable_palette(i);
            for (size_t j = 0; j < palette->size(); ++j) {
              gfx::SnesColor color = (*palette)[j];
              if (color.is_modified()) {
                RETURN_IF_ERROR(rom.WriteColor(
                    gfx::GetPaletteAddress(group.name(), i, j), color));
                color.set_modified(false);
              }
            }
          }
          return absl::OkStatus();
        });
  }
 
  if (core::FeatureFlags::get().kSaveGfxGroups) {
    RETURN_IF_ERROR(SaveGfxGroups(rom, data));
  }
 
  // TODO: Implement SaveAllGraphicsData logic
  return absl::OkStatus();
}
 
absl::Status LoadMetadata(const Rom& rom, GameData& data) {
  constexpr uint32_t kTitleStringOffset = 0x7FC0;
  constexpr uint32_t kTitleStringLength = 20;
 
  if (rom.size() < kTitleStringOffset + kTitleStringLength) {
    return absl::OutOfRangeError("ROM too small for metadata");
  }
 
  // Check version byte at offset + 0x19 (0x7FD9)
  if (kTitleStringOffset + 0x19 < rom.size()) {
    // Access directly via data() since ReadByte is non-const
    uint8_t version_byte = rom.data()[kTitleStringOffset + 0x19];
    data.version =
        (version_byte == 0) ? zelda3_version::JP : zelda3_version::US;
  }
 
  auto title_bytes = rom.ReadByteVector(kTitleStringOffset, kTitleStringLength);
  if (title_bytes.ok()) {
    data.title.assign(title_bytes->begin(), title_bytes->end());
  }
 
  return absl::OkStatus();
}
 
absl::Status LoadPalettes(const Rom& rom, GameData& data) {
  // Create a vector from rom data for palette loading
  const std::vector<uint8_t>& rom_vec = rom.vector();
  return gfx::LoadAllPalettes(rom_vec, data.palette_groups);
}
 
absl::Status LoadGfxGroups(Rom& rom, GameData& data) {
  if (kVersionConstantsMap.find(data.version) == kVersionConstantsMap.end()) {
    return absl::FailedPreconditionError("Unsupported ROM version");
  }
 
  auto version_constants = kVersionConstantsMap.at(data.version);
 
  // Load Main Blocksets
  auto main_ptr_res = rom.ReadWord(kGfxGroupsPointer);
  if (main_ptr_res.ok()) {
    uint32_t main_ptr = SnesToPc(*main_ptr_res);
    for (uint32_t i = 0; i < kNumMainBlocksets; i++) {
      for (int j = 0; j < 8; j++) {
        auto val = rom.ReadByte(main_ptr + (i * 8) + j);
        if (val.ok())
          data.main_blockset_ids[i][j] = *val;
      }
    }
  }
 
  // Load Room Blocksets
  for (uint32_t i = 0; i < kNumRoomBlocksets; i++) {
    for (int j = 0; j < 4; j++) {
      auto val = rom.ReadByte(kEntranceGfxGroup + (i * 4) + j);
      if (val.ok())
        data.room_blockset_ids[i][j] = *val;
    }
  }
 
  // Load Sprite Blocksets
  for (uint32_t i = 0; i < kNumSpritesets; i++) {
    for (int j = 0; j < 4; j++) {
      auto val =
          rom.ReadByte(version_constants.kSpriteBlocksetPointer + (i * 4) + j);
      if (val.ok())
        data.spriteset_ids[i][j] = *val;
    }
  }
 
  // Load Palette Sets
  for (uint32_t i = 0; i < kNumPalettesets; i++) {
    for (int j = 0; j < 4; j++) {
      auto val =
          rom.ReadByte(version_constants.kDungeonPalettesGroups + (i * 4) + j);
      if (val.ok())
        data.paletteset_ids[i][j] = *val;
    }
  }
 
  return absl::OkStatus();
}
 
absl::Status SaveGfxGroups(Rom& rom, const GameData& data) {
  auto version_constants = kVersionConstantsMap.at(data.version);
 
  ASSIGN_OR_RETURN(auto main_ptr, rom.ReadWord(kGfxGroupsPointer));
  main_ptr = SnesToPc(main_ptr);
 
  // Save Main Blocksets
  for (uint32_t i = 0; i < kNumMainBlocksets; i++) {
    for (int j = 0; j < 8; j++) {
      RETURN_IF_ERROR(
          rom.WriteByte(main_ptr + (i * 8) + j, data.main_blockset_ids[i][j]));
    }
  }
 
  // Save Room Blocksets
  for (uint32_t i = 0; i < kNumRoomBlocksets; i++) {
    for (int j = 0; j < 4; j++) {
      RETURN_IF_ERROR(rom.WriteByte(kEntranceGfxGroup + (i * 4) + j,
                                    data.room_blockset_ids[i][j]));
    }
  }
 
  // Save Sprite Blocksets
  for (uint32_t i = 0; i < kNumSpritesets; i++) {
    for (int j = 0; j < 4; j++) {
      RETURN_IF_ERROR(
          rom.WriteByte(version_constants.kSpriteBlocksetPointer + (i * 4) + j,
                        data.spriteset_ids[i][j]));
    }
  }
 
  // Save Palette Sets
  for (uint32_t i = 0; i < kNumPalettesets; i++) {
    for (int j = 0; j < 4; j++) {
      RETURN_IF_ERROR(
          rom.WriteByte(version_constants.kDungeonPalettesGroups + (i * 4) + j,
                        data.paletteset_ids[i][j]));
    }
  }
 
  return absl::OkStatus();
}
 
// ============================================================================
// Main Graphics Loading
// ============================================================================
 
struct SheetLoadResult {
  std::vector<uint8_t> data;
  bool is_compressed = false;
  bool decompression_succeeded = false;
  bool is_bpp3 = false; // true if 3BPP, false if skipped/2BPP
  uint32_t pc_offset = 0;
};
 
SheetLoadResult LoadSheetRaw(const Rom& rom, uint32_t i, uint32_t ptr1,
                             uint32_t ptr2, uint32_t ptr3) {
  SheetLoadResult result;
  result.data.assign(zelda3::kUncompressedSheetSize, 0); // Default empty
 
  // Uncompressed 3BPP (115-126)
  if (i >= 115 && i <= 126) {
    result.is_compressed = false;
    result.pc_offset =
        GetGraphicsAddress(rom.data(), i, ptr1, ptr2, ptr3, rom.size());
 
    auto read_res =
        rom.ReadByteVector(result.pc_offset, zelda3::kUncompressedSheetSize);
    if (read_res.ok()) {
      result.data = *read_res;
      result.decompression_succeeded = true;
      result.is_bpp3 = true;
    } else {
      result.decompression_succeeded = false;
    }
  }
  // 2BPP (113-114, 218+) - Skipped in main loop
  else if (i == 113 || i == 114 || i >= 218) {
    result.is_compressed = true;
    result.is_bpp3 = false;
  }
  // Compressed 3BPP (Standard)
  else {
    result.is_compressed = true;
    result.pc_offset =
        GetGraphicsAddress(rom.data(), i, ptr1, ptr2, ptr3, rom.size());
 
    if (result.pc_offset < rom.size()) {
      auto decomp_res =
          gfx::lc_lz2::DecompressV2(rom.data(), result.pc_offset, 0x800, 1, rom.size());
      if (decomp_res.ok()) {
        result.data = *decomp_res;
        result.decompression_succeeded = true;
        result.is_bpp3 = true;
      } else {
        result.decompression_succeeded = false;
      }
    }
  }
  return result;
}
 
void ProcessSheetBitmap(GameData& data, uint32_t i, const SheetLoadResult& result) {
  if (result.is_bpp3) {
    auto converted_sheet = gfx::SnesTo8bppSheet(result.data, 3);
    if (converted_sheet.size() != 4096)
      converted_sheet.resize(4096, 0);
 
    data.raw_gfx_sheets[i] = converted_sheet;
    data.gfx_bitmaps[i].Create(gfx::kTilesheetWidth, gfx::kTilesheetHeight,
                               gfx::kTilesheetDepth, converted_sheet);
 
    // Apply default palettes
    if (!data.palette_groups.empty()) {
      gfx::SnesPalette default_palette;
      if (i < 113 && data.palette_groups.dungeon_main.size() > 0) {
        default_palette = data.palette_groups.dungeon_main[0];
      } else if (i < 128 && data.palette_groups.sprites_aux1.size() > 0) {
        default_palette = data.palette_groups.sprites_aux1[0];
      } else if (data.palette_groups.hud.size() > 0) {
        default_palette = data.palette_groups.hud.palette(0);
      }
 
      if (!default_palette.empty()) {
        ApplyDefaultPalette(data.gfx_bitmaps[i], default_palette);
      } else {
        // Fallback to grayscale if no palette found
        std::vector<gfx::SnesColor> grayscale;
        for (int color_idx = 0; color_idx < 16; ++color_idx) {
          float val = color_idx / 15.0f;
          grayscale.emplace_back(ImVec4(val, val, val, 1.0f));
        }
        if (!grayscale.empty()) {
          grayscale[0].set_transparent(true);
        }
        data.gfx_bitmaps[i].SetPalette(gfx::SnesPalette(grayscale));
      }
    } else {
      // Fallback to grayscale if no palette groups loaded
      std::vector<gfx::SnesColor> grayscale;
      for (int color_idx = 0; color_idx < 16; ++color_idx) {
        float val = color_idx / 15.0f;
        grayscale.emplace_back(ImVec4(val, val, val, 1.0f));
      }
      if (!grayscale.empty()) {
        grayscale[0].set_transparent(true);
      }
      data.gfx_bitmaps[i].SetPalette(gfx::SnesPalette(grayscale));
    }
 
    data.graphics_buffer.insert(
        data.graphics_buffer.end(), data.gfx_bitmaps[i].data(),
        data.gfx_bitmaps[i].data() + data.gfx_bitmaps[i].size());
  } else {
    // Placeholder - Fill with 0 (transparent) instead of 0xFF (white)
    std::vector<uint8_t> placeholder(4096, 0);
    data.raw_gfx_sheets[i] = placeholder;
    data.gfx_bitmaps[i].Create(gfx::kTilesheetWidth, gfx::kTilesheetHeight,
                               gfx::kTilesheetDepth, placeholder);
    data.graphics_buffer.resize(data.graphics_buffer.size() + 4096, 0);
  }
}
 
absl::Status LoadGraphics(Rom& rom, GameData& data) {
  if (kVersionConstantsMap.find(data.version) == kVersionConstantsMap.end()) {
    return absl::FailedPreconditionError(
        "Unsupported ROM version for graphics");
  }
  auto version_constants = kVersionConstantsMap.at(data.version);
  const uint32_t gfx_ptr1 = core::RomSettings::Get().GetAddressOr(
      core::RomAddressKey::kOverworldGfxPtr1,
      version_constants.kOverworldGfxPtr1);
  const uint32_t gfx_ptr2 = core::RomSettings::Get().GetAddressOr(
      core::RomAddressKey::kOverworldGfxPtr2,
      version_constants.kOverworldGfxPtr2);
  const uint32_t gfx_ptr3 = core::RomSettings::Get().GetAddressOr(
      core::RomAddressKey::kOverworldGfxPtr3,
      version_constants.kOverworldGfxPtr3);
 
  data.graphics_buffer.clear();
 
#ifdef __EMSCRIPTEN__
  auto loading_handle =
      app::platform::WasmLoadingManager::BeginLoading("Loading Graphics");
#endif
 
  // Initialize Diagnostics
  auto& diag = data.diagnostics;
  diag.rom_size = rom.size();
  diag.ptr1_loc = gfx_ptr1;
  diag.ptr2_loc = gfx_ptr2;
  diag.ptr3_loc = gfx_ptr3;
 
  LOG_INFO("Graphics", "Loading %d graphics sheets...", kNumGfxSheets);
 
  for (uint32_t i = 0; i < kNumGfxSheets; i++) {
#ifdef __EMSCRIPTEN__
    app::platform::WasmLoadingManager::UpdateProgress(
        loading_handle, static_cast<float>(i) / kNumGfxSheets);
#endif
 
    // Inside LoadGraphics loop:
    auto result = LoadSheetRaw(rom, i, gfx_ptr1, gfx_ptr2, gfx_ptr3);
    
    // Update Diagnostics
    auto& sd = diag.sheets[i];
    sd.index = i;
    sd.is_compressed = result.is_compressed;
    sd.pc_offset = result.pc_offset;
    sd.decompression_succeeded = result.decompression_succeeded;
    sd.actual_decomp_size = result.data.size();
    if (!result.data.empty()) {
        size_t count = std::min<size_t>(result.data.size(), 8);
        sd.first_bytes.assign(result.data.begin(), result.data.begin() + count);
    }
    if (result.is_compressed && !result.is_bpp3) {
        sd.decomp_size_param = 0x800; // Expected for LC-LZ2
    }
 
    ProcessSheetBitmap(data, i, result);
    
    if (i % 50 == 0 || i == kNumGfxSheets - 1) {
        LOG_DEBUG("Graphics", "Sheet %d: offset=0x%06X, size=%zu, %s", 
                  i, result.pc_offset, result.data.size(), 
                  result.decompression_succeeded ? "OK" : "FAILED");
    }
  }
 
  diag.Analyze();
  LOG_INFO("Graphics", "Graphics loading complete. Sheets processed: %d", kNumGfxSheets);
 
#ifdef __EMSCRIPTEN__
  app::platform::WasmLoadingManager::EndLoading(loading_handle);
#endif
 
  return absl::OkStatus();
}
 
// ============================================================================
// Link Graphics Loading
// ============================================================================
 
absl::StatusOr<std::array<gfx::Bitmap, kNumLinkSheets>> LoadLinkGraphics(
    const Rom& rom) {
  std::array<gfx::Bitmap, kNumLinkSheets> link_graphics;
  for (uint32_t i = 0; i < kNumLinkSheets; i++) {
    auto link_sheet_data_result =
        rom.ReadByteVector(/*offset=*/kLinkGfxOffset + (i * kLinkGfxLength),
                           /*length=*/kLinkGfxLength);
    if (!link_sheet_data_result.ok()) {
      return link_sheet_data_result.status();
    }
    auto link_sheet_8bpp =
        gfx::SnesTo8bppSheet(*link_sheet_data_result, /*bpp=*/4);
    if (link_sheet_8bpp.size() != 4096)
      link_sheet_8bpp.resize(4096, 0);
 
    link_graphics[i].Create(gfx::kTilesheetWidth, gfx::kTilesheetHeight,
                            gfx::kTilesheetDepth, link_sheet_8bpp);
    // Palette is applied by the caller since GameData may not be available here
  }
  return link_graphics;
}
 
// ============================================================================
// 2BPP Graphics Loading
// ============================================================================
 
absl::StatusOr<std::vector<uint8_t>> Load2BppGraphics(const Rom& rom) {
  std::vector<uint8_t> sheet;
  const uint8_t sheets[] = {0x71, 0x72, 0xDA, 0xDB, 0xDC, 0xDD, 0xDE};
 
  // Get version constants - default to US if we don't know
  auto version_constants = kVersionConstantsMap.at(zelda3_version::US);
  const uint32_t gfx_ptr1 = core::RomSettings::Get().GetAddressOr(
      core::RomAddressKey::kOverworldGfxPtr1,
      version_constants.kOverworldGfxPtr1);
  const uint32_t gfx_ptr2 = core::RomSettings::Get().GetAddressOr(
      core::RomAddressKey::kOverworldGfxPtr2,
      version_constants.kOverworldGfxPtr2);
  const uint32_t gfx_ptr3 = core::RomSettings::Get().GetAddressOr(
      core::RomAddressKey::kOverworldGfxPtr3,
      version_constants.kOverworldGfxPtr3);
 
  for (const auto& sheet_id : sheets) {
    auto offset = GetGraphicsAddress(rom.data(), sheet_id, gfx_ptr1, gfx_ptr2,
                                     gfx_ptr3, rom.size());
 
    if (offset >= rom.size()) {
      return absl::OutOfRangeError(absl::StrFormat(
          "2BPP graphics sheet %u offset %u exceeds ROM size %zu", sheet_id,
          offset, rom.size()));
    }
 
    // Decompress using LC-LZ2 algorithm with 0x800 byte output buffer.
    auto decomp_result =
        gfx::lc_lz2::DecompressV2(rom.data(), offset, 0x800, 1, rom.size());
    if (!decomp_result.ok()) {
      return decomp_result.status();
    }
    auto converted_sheet = gfx::SnesTo8bppSheet(*decomp_result, 2);
    for (const auto& each_pixel : converted_sheet) {
      sheet.push_back(each_pixel);
    }
  }
  return sheet;
}
 
// ============================================================================
// Font Graphics Loading
// ============================================================================
 
absl::StatusOr<gfx::Bitmap> LoadFontGraphics(const Rom& rom) {
  // Font sprites are located at 0x70000, 2BPP format
  constexpr uint32_t kFontDataSize = 0x4000;  // 16KB of font data
 
  auto font_data_result =
      rom.ReadByteVector(kFontSpriteLocation, kFontDataSize);
  if (!font_data_result.ok()) {
    return font_data_result.status();
  }
 
  // Convert from 2BPP SNES format to 8BPP
  auto font_8bpp = gfx::SnesTo8bppSheet(*font_data_result, /*bpp=*/2);
 
  gfx::Bitmap font_bitmap;
  font_bitmap.Create(gfx::kTilesheetWidth * 2, gfx::kTilesheetHeight * 4,
                     gfx::kTilesheetDepth, font_8bpp);
 
  return font_bitmap;
}
 
// ============================================================================
// Graphics Saving
// ============================================================================
 
absl::Status SaveAllGraphicsData(
    [[maybe_unused]] Rom& rom,
    [[maybe_unused]] const std::array<gfx::Bitmap, kNumGfxSheets>& sheets) {
  // For now, return OK status - full implementation would write sheets back
  // to ROM at their respective addresses with proper compression
  LOG_INFO("SaveAllGraphicsData", "Graphics save not yet fully implemented");
  return absl::OkStatus();
}
 
}  // namespace zelda3
}  // namespace yaze