track_collision_generator.cc

Go to the documentation of this file.

#include "zelda3/dungeon/track_collision_generator.h"
 
#include <algorithm>
#include <array>
#include <cstdint>
#include <sstream>
#include <string>
#include <utility>
 
#include "absl/status/status.h"
#include "absl/strings/str_format.h"
#include "rom/snes.h"
#include "rom/write_fence.h"
#include "util/macro.h"
#include "zelda3/dungeon/dimension_service.h"
#include "zelda3/dungeon/dungeon_rom_addresses.h"
#include "zelda3/dungeon/room_object.h"
 
namespace yaze {
namespace zelda3 {
 
namespace {
 
constexpr int kGridSize = 64;
constexpr uint16_t kCollisionSingleTileMarker = 0xF0F0;
constexpr uint16_t kCollisionEndMarker = 0xFFFF;
constexpr int kFallbackTrackFootprintWidthTiles = 2;
constexpr int kFallbackTrackFootprintHeightTiles = 2;
 
// Map corner type to its switch equivalent for promotion.
TrackTileType PromoteCornerToSwitch(TrackTileType corner) {
  switch (corner) {
    case TrackTileType::CornerTL:
      return TrackTileType::SwitchTL;
    case TrackTileType::CornerBL:
      return TrackTileType::SwitchBL;
    case TrackTileType::CornerTR:
      return TrackTileType::SwitchTR;
    case TrackTileType::CornerBR:
      return TrackTileType::SwitchBR;
    default:
      return corner;
  }
}
 
bool IsCornerTile(uint8_t tile) {
  return tile >= 0xB2 && tile <= 0xB5;
}
 
// Classify a tile based on its 4-neighbor connectivity.
//
// The algorithm: for each occupied tile in the grid, check which of the
// 4 cardinal neighbors are also occupied. The pattern of neighbors uniquely
// determines the tile type:
//   - 1 neighbor (endpoint) → stop tile, direction based on which neighbor
//   - 2 neighbors (line or corner) → straight or corner
//   - 3 neighbors → T-junction
//   - 4 neighbors → intersection
uint8_t ClassifyTile(bool up, bool down, bool left, bool right) {
  int count = (up ? 1 : 0) + (down ? 1 : 0) + (left ? 1 : 0) + (right ? 1 : 0);
 
  if (count == 0) {
    // Isolated tile — treat as intersection (shouldn't happen in practice)
    return static_cast<uint8_t>(TrackTileType::Intersection);
  }
 
  if (count == 1) {
    // Endpoint → stop tile. Direction is where the neighbor IS, because
    // the cart arrives from that direction and will depart back that way.
    if (down)
      return static_cast<uint8_t>(TrackTileType::StopNorth);
    if (up)
      return static_cast<uint8_t>(TrackTileType::StopSouth);
    if (right)
      return static_cast<uint8_t>(TrackTileType::StopWest);
    if (left)
      return static_cast<uint8_t>(TrackTileType::StopEast);
  }
 
  if (count == 2) {
    // Two neighbors — either a straight line or a corner
    if (left && right)
      return static_cast<uint8_t>(TrackTileType::HorizStraight);
    if (up && down)
      return static_cast<uint8_t>(TrackTileType::VertStraight);
    if (down && right)
      return static_cast<uint8_t>(TrackTileType::CornerTL);
    if (up && right)
      return static_cast<uint8_t>(TrackTileType::CornerBL);
    if (down && left)
      return static_cast<uint8_t>(TrackTileType::CornerTR);
    if (up && left)
      return static_cast<uint8_t>(TrackTileType::CornerBR);
  }
 
  if (count == 3) {
    // T-junction — named for the direction WITHOUT a neighbor
    if (!up)
      return static_cast<uint8_t>(TrackTileType::TJuncSouth);
    if (!down)
      return static_cast<uint8_t>(TrackTileType::TJuncNorth);
    if (!left)
      return static_cast<uint8_t>(TrackTileType::TJuncEast);
    if (!right)
      return static_cast<uint8_t>(TrackTileType::TJuncWest);
  }
 
  // count == 4: full intersection
  return static_cast<uint8_t>(TrackTileType::Intersection);
}
 
// Get the tile label character for ASCII visualization.
char TileToChar(uint8_t tile) {
  switch (tile) {
    case 0xB0:
      return '-';  // horiz straight
    case 0xB1:
      return '|';  // vert straight
    case 0xB2:
      return '/';  // corner TL (down+right)
    case 0xB3:
      return '\\';  // corner BL (up+right)
    case 0xB4:
      return '\\';  // corner TR (down+left)
    case 0xB5:
      return '/';  // corner BR (up+left)
    case 0xB6:
      return '+';  // intersection
    case 0xB7:
      return 'N';  // stop north
    case 0xB8:
      return 'S';  // stop south
    case 0xB9:
      return 'W';  // stop west
    case 0xBA:
      return 'E';  // stop east
    case 0xBB:
      return 'T';  // T-junc north
    case 0xBC:
      return 'T';  // T-junc south
    case 0xBD:
      return 'T';  // T-junc east
    case 0xBE:
      return 'T';  // T-junc west
    case 0xD0:
      return '@';  // switch TL
    case 0xD1:
      return '@';  // switch BL
    case 0xD2:
      return '@';  // switch TR
    case 0xD3:
      return '@';  // switch BR
    default:
      return '.';
  }
}
 
DimensionService::DimensionResult ResolveTrackObjectDimensions(
    const RoomObject& obj, const GeneratorOptions& options,
    const DimensionService& dimension_service) {
  auto dims = dimension_service.GetDimensions(obj);
 
  // In CLI-only contexts, custom object feature flags are often disabled.
  // Object 0x31 may then map through DrawNothing and collapse to a 1x1
  // footprint. Track assets are authored as 2x2 pieces, so recover a stable
  // fallback here to avoid sparse/gapped collision generation.
  if (obj.id_ == static_cast<int16_t>(options.track_object_id) &&
      dims.offset_x_tiles == 0 && dims.offset_y_tiles == 0 &&
      dims.width_tiles == 1 && dims.height_tiles == 1) {
    dims.width_tiles = kFallbackTrackFootprintWidthTiles;
    dims.height_tiles = kFallbackTrackFootprintHeightTiles;
  }
 
  return dims;
}
 
}  // namespace
 
absl::StatusOr<TrackCollisionResult> GenerateTrackCollision(
    Room* room, const GeneratorOptions& options) {
  if (!room) {
    return absl::InvalidArgumentError("Room pointer is null");
  }
 
  // Ensure objects are loaded
  if (room->GetTileObjects().empty()) {
    room->LoadObjects();
  }
 
  TrackCollisionResult result;
  result.room_id = room->id();
  result.collision_map.tiles.fill(0);
 
  // Step 1: Build occupancy grid from rail objects.
  // Rail objects (ID 0x31) have coordinates in the room's tile space.
  // RoomObject x_ and y_ are in tile coordinates (each tile = 8 pixels).
  // The collision grid is 64x64 (covering 512x512 pixels = full room).
  std::array<bool, kGridSize * kGridSize> occupied{};
  auto& dimension_service = DimensionService::Get();
 
  for (const auto& obj : room->GetTileObjects()) {
    if (obj.id_ != static_cast<int16_t>(options.track_object_id)) {
      continue;
    }
 
    const auto dims =
        ResolveTrackObjectDimensions(obj, options, dimension_service);
    int base_x = obj.x_ + dims.offset_x_tiles;
    int base_y = obj.y_ + dims.offset_y_tiles;
    int w = std::max(1, dims.width_tiles);
    int h = std::max(1, dims.height_tiles);
 
    for (int dy = 0; dy < h; ++dy) {
      for (int dx = 0; dx < w; ++dx) {
        int gx = base_x + dx;
        int gy = base_y + dy;
        if (gx >= 0 && gx < kGridSize && gy >= 0 && gy < kGridSize) {
          occupied[gy * kGridSize + gx] = true;
        }
      }
    }
  }
 
  // Step 2: Classify each occupied tile by neighbor connectivity.
  for (int y = 0; y < kGridSize; ++y) {
    for (int x = 0; x < kGridSize; ++x) {
      if (!occupied[y * kGridSize + x])
        continue;
 
      bool up = (y > 0) && occupied[(y - 1) * kGridSize + x];
      bool down = (y < kGridSize - 1) && occupied[(y + 1) * kGridSize + x];
      bool left = (x > 0) && occupied[y * kGridSize + (x - 1)];
      bool right = (x < kGridSize - 1) && occupied[y * kGridSize + (x + 1)];
 
      uint8_t tile = ClassifyTile(up, down, left, right);
      result.collision_map.tiles[y * kGridSize + x] = tile;
      result.tiles_generated++;
 
      if (tile >= 0xB7 && tile <= 0xBA)
        result.stop_count++;
      if (tile >= 0xB2 && tile <= 0xB5)
        result.corner_count++;
    }
  }
 
  // Step 3: Apply switch promotions.
  for (const auto& [sx, sy] : options.switch_promotions) {
    if (sx < 0 || sx >= kGridSize || sy < 0 || sy >= kGridSize)
      continue;
    size_t idx = sy * kGridSize + sx;
    uint8_t tile = result.collision_map.tiles[idx];
    if (IsCornerTile(tile)) {
      result.collision_map.tiles[idx] = static_cast<uint8_t>(
          PromoteCornerToSwitch(static_cast<TrackTileType>(tile)));
      result.corner_count--;
      result.switch_count++;
    }
  }
 
  // Step 4: Apply manual stop overrides.
  for (const auto& [ox, oy, otype] : options.stop_overrides) {
    if (ox < 0 || ox >= kGridSize || oy < 0 || oy >= kGridSize)
      continue;
    size_t idx = oy * kGridSize + ox;
    if (result.collision_map.tiles[idx] != 0) {
      result.collision_map.tiles[idx] = static_cast<uint8_t>(otype);
    }
  }
 
  result.collision_map.has_data = (result.tiles_generated > 0);
  result.ascii_visualization = VisualizeCollisionMap(result.collision_map);
 
  return result;
}
 
absl::Status WriteTrackCollision(Rom* rom, int room_id,
                                 const CustomCollisionMap& map) {
  if (!rom || !rom->is_loaded()) {
    return absl::InvalidArgumentError("ROM not loaded");
  }
  if (room_id < 0 || room_id >= kNumberOfRooms) {
    return absl::OutOfRangeError("Room ID out of range");
  }
 
  const auto& data = rom->vector();
  if (data.empty()) {
    return absl::FailedPreconditionError("ROM vector is empty");
  }
 
  const int ptrs_size = kNumberOfRooms * 3;
  if (kCustomCollisionRoomPointers + ptrs_size >
      static_cast<int>(data.size())) {
    return absl::FailedPreconditionError(
        "Custom collision pointer table not present in this ROM");
  }
  if (kCustomCollisionDataPosition >= static_cast<int>(data.size())) {
    return absl::FailedPreconditionError(
        "Custom collision data region not present in this ROM");
  }
  if (kCustomCollisionDataSoftEnd > static_cast<int>(data.size())) {
    return absl::FailedPreconditionError(
        "Custom collision data region truncated (ROM too small)");
  }
 
  // Save-time guardrails: only allow writes to the collision pointer table and
  // collision data bank (excluding the reserved WaterFill tail region).
  yaze::rom::WriteFence fence;
  RETURN_IF_ERROR(fence.Allow(
      static_cast<uint32_t>(kCustomCollisionRoomPointers),
      static_cast<uint32_t>(kCustomCollisionRoomPointers + ptrs_size),
      "CustomCollisionPointers"));
  RETURN_IF_ERROR(
      fence.Allow(static_cast<uint32_t>(kCustomCollisionDataPosition),
                  static_cast<uint32_t>(kCustomCollisionDataSoftEnd),
                  "CustomCollisionData"));
  yaze::rom::ScopedWriteFence scope(rom, &fence);
 
  // Encode collision data in single-tile format.
  // Format: [F0 F0] [offset_lo offset_hi tile] ... [FF FF]
  std::vector<uint8_t> encoded;
  encoded.push_back(0xF0);
  encoded.push_back(0xF0);
 
  for (int y = 0; y < kGridSize; ++y) {
    for (int x = 0; x < kGridSize; ++x) {
      uint8_t tile = map.tiles[y * kGridSize + x];
      if (tile == 0)
        continue;
      uint16_t offset = static_cast<uint16_t>(y * kGridSize + x);
      encoded.push_back(offset & 0xFF);
      encoded.push_back(offset >> 8);
      encoded.push_back(tile);
    }
  }
  encoded.push_back(0xFF);
  encoded.push_back(0xFF);
 
  // Find the end of existing collision data by scanning all room pointers
  // to determine the highest used offset.
  const size_t safe_end =
      std::min(static_cast<size_t>(data.size()),
               static_cast<size_t>(kCustomCollisionDataSoftEnd));
  uint32_t max_used_pc = static_cast<uint32_t>(kCustomCollisionDataPosition);
  for (int r = 0; r < kNumberOfRooms; ++r) {
    int ptr_offset = kCustomCollisionRoomPointers + (r * 3);
    if (ptr_offset + 2 >= static_cast<int>(data.size()))
      continue;
 
    uint32_t snes_ptr = data[ptr_offset] | (data[ptr_offset + 1] << 8) |
                        (data[ptr_offset + 2] << 16);
    if (snes_ptr == 0)
      continue;
 
    uint32_t pc = SnesToPc(snes_ptr);
    if (pc < static_cast<uint32_t>(kCustomCollisionDataPosition)) {
      return absl::FailedPreconditionError(
          absl::StrFormat("Custom collision pointer for room 0x%02X points "
                          "before data region (pc=0x%06X)",
                          r, pc));
    }
    if (pc >= static_cast<uint32_t>(kCustomCollisionDataSoftEnd)) {
      return absl::FailedPreconditionError(
          absl::StrFormat("Custom collision pointer for room 0x%02X overlaps "
                          "WaterFill reserved region (pc=0x%06X)",
                          r, pc));
    }
    if (pc >= data.size()) {
      return absl::OutOfRangeError("Custom collision pointer out of ROM range");
    }
    // Walk past this room's data to find its end
    size_t cursor = pc;
    bool single_mode = false;
    bool found_end_marker = false;
    while (cursor + 1 < safe_end) {
      uint16_t val = data[cursor] | (data[cursor + 1] << 8);
      cursor += 2;
      if (val == kCollisionEndMarker) {
        found_end_marker = true;
        break;
      }
      if (val == kCollisionSingleTileMarker) {
        single_mode = true;
        continue;
      }
      if (!single_mode) {
        // Rectangle mode: skip width, height, then width*height bytes
        if (cursor + 1 >= safe_end)
          break;
        uint8_t w = data[cursor];
        uint8_t h = data[cursor + 1];
        cursor += 2;
        cursor += w * h;
      } else {
        // Single tile mode: skip 1 byte (tile value)
        cursor += 1;
      }
    }
    // If we hit the reserved region without an end marker, treat as corruption.
    if (!found_end_marker && cursor + 1 >= safe_end) {
      return absl::FailedPreconditionError(
          absl::StrFormat("Custom collision data for room 0x%02X is "
                          "unterminated before WaterFill reserved region",
                          r));
    }
    if (cursor > max_used_pc) {
      max_used_pc = static_cast<uint32_t>(cursor);
    }
  }
 
  // Check if there's enough space
  uint32_t write_pos = max_used_pc;
  if (write_pos + encoded.size() > kCustomCollisionDataSoftEnd) {
    return absl::ResourceExhaustedError(absl::StrFormat(
        "Not enough collision data space. Need %d bytes at 0x%06X, "
        "region ends at 0x%06X",
        encoded.size(), write_pos, kCustomCollisionDataSoftEnd));
  }
 
  if (write_pos + encoded.size() > data.size()) {
    return absl::OutOfRangeError(
        absl::StrFormat("ROM too small for custom collision write (need "
                        "end=0x%06X, size=0x%06X)",
                        write_pos + encoded.size(), data.size()));
  }
  RETURN_IF_ERROR(
      rom->WriteVector(static_cast<int>(write_pos), std::move(encoded)));
 
  // Update pointer table: 3-byte SNES address
  uint32_t snes_addr = PcToSnes(write_pos);
  int ptr_offset = kCustomCollisionRoomPointers + (room_id * 3);
  RETURN_IF_ERROR(rom->WriteByte(ptr_offset, snes_addr & 0xFF));
  RETURN_IF_ERROR(rom->WriteByte(ptr_offset + 1, (snes_addr >> 8) & 0xFF));
  RETURN_IF_ERROR(rom->WriteByte(ptr_offset + 2, (snes_addr >> 16) & 0xFF));
 
  return absl::OkStatus();
}
 
std::string VisualizeCollisionMap(const CustomCollisionMap& map) {
  // Find bounding box of non-zero tiles to avoid printing the entire 64x64.
  int min_x = kGridSize, max_x = 0, min_y = kGridSize, max_y = 0;
  for (int y = 0; y < kGridSize; ++y) {
    for (int x = 0; x < kGridSize; ++x) {
      if (map.tiles[y * kGridSize + x] != 0) {
        min_x = std::min(min_x, x);
        max_x = std::max(max_x, x);
        min_y = std::min(min_y, y);
        max_y = std::max(max_y, y);
      }
    }
  }
 
  if (min_x > max_x)
    return "(empty)\n";
 
  // Add 1-tile padding
  min_x = std::max(0, min_x - 1);
  min_y = std::max(0, min_y - 1);
  max_x = std::min(kGridSize - 1, max_x + 1);
  max_y = std::min(kGridSize - 1, max_y + 1);
 
  std::stringstream ss;
  // Column header
  ss << "    ";
  for (int x = min_x; x <= max_x; ++x) {
    ss << absl::StrFormat("%X", x % 16);
  }
  ss << "\n";
 
  for (int y = min_y; y <= max_y; ++y) {
    ss << absl::StrFormat("%02X: ", y);
    for (int x = min_x; x <= max_x; ++x) {
      uint8_t tile = map.tiles[y * kGridSize + x];
      ss << TileToChar(tile);
    }
    ss << "\n";
  }
 
  return ss.str();
}
 
}  // namespace zelda3
}  // namespace yaze