about summary refs log tree commit diff stats
path: root/tools/validator
Commit message (Expand)AuthorAgeFilesLines
* Allow validator to ignore custom nodesStar Rauchenberger2025-10-231-0/+4
* Fixed typos in the_advanced locationsStar Rauchenberger2025-10-231-7/+13
* Mark some doors as "latched"Star Rauchenberger2025-10-201-1/+1
* Annotate "worldport entrances"Star Rauchenberger2025-10-193-2/+24
* [Tools] Typo in header guardStar Rauchenberger2025-09-221-1/+1
* [Data] Annotate shuffleable portsStar Rauchenberger2025-09-211-0/+15
* [Data] Allow WALL solution to the_entry!OPENStar Rauchenberger2025-09-111-1/+2
* [Data] Fixed connection target required door logic bugsStar Rauchenberger2025-09-113-4/+73
* Added door groupsStar Rauchenberger2025-09-073-0/+70
* [Data] Strip unnecessary AP IDsStar Rauchenberger2025-09-043-4/+147
* Added progressive doorsStar Rauchenberger2025-09-013-0/+52
* Handled cyan doorsStar Rauchenberger2025-08-311-0/+9
* Changed how door location names are formattedStar Rauchenberger2025-08-304-214/+315
* [Data] Made proxies with the same answer as the panel explicitStar Rauchenberger2025-08-301-1/+0
* Added control_centerStar Rauchenberger2025-08-273-1/+17
* Added daedalusStar Rauchenberger2025-08-241-0/+5
* Added "endings" object typeStar Rauchenberger2025-08-203-1/+39
* Added the_repetitiveStar Rauchenberger2025-08-181-10/+6
* Validate that nodes in game files are usedStar Rauchenberger2025-08-187-4/+122
* Validate that node paths aren't used multiple timesStar Rauchenberger2025-08-173-1/+40
* Minor changes for compiling on WindowsStar Rauchenberger2025-08-161-1/+1
* Started writing a data validatorStar Rauchenberger2025-08-167-0/+840
generated by cgit-pink 1.4.1 (git 2.36.1) at 2026-03-24 00:23:13 +0000
a id='n255' href='#n255'>255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 
#include <google/protobuf/message.h>
#include <google/protobuf/text_format.h>

#include <cstdint>
#include <filesystem>
#include <fstream>
#include <iostream>
#include <map>
#include <sstream>
#include <string>

#include "proto/human.pb.h"
#include "util/ids_yaml_format.h"
#include "util/naming.h"

namespace com::fourisland::lingo2_archipelago {
namespace {

template <typename T>
T ReadMessageFromFile(const std::string& path) {
  std::cout << "Processing " << path << std::endl;

  std::ifstream file(path);
  std::stringstream buffer;
  buffer << file.rdbuf();

  T message;
  google::protobuf::TextFormat::ParseFromString(buffer.str(), &message);

  return message;
}

class AssignIds {
 public:
  AssignIds(const std::string& mapdir) : mapdir_(mapdir) {}

  void Run() {
    std::filesystem::path datadir_path = mapdir_;
    std::filesystem::path ids_path = datadir_path / "ids.yaml";

    ReadIds(ids_path);

    ProcessMaps(datadir_path);
    ProcessSpecialIds();
    ProcessProgressivesFile(datadir_path / "progressives.txtpb");
    ProcessDoorGroupsFile(datadir_path / "door_groups.txtpb");
    ProcessGlobalMetadataFile(datadir_path / "metadata.txtpb");

    WriteIds(ids_path);

    std::cout << "Next ID: " << next_id_ << std::endl;
  }

  void ReadIds(std::filesystem::path path) {
    if (!std::filesystem::exists(path)) {
      return;
    }

    id_mappings_ = ReadIdsFromYaml(path.string());

    for (const auto& [_, map] : id_mappings_.maps()) {
      UpdateNextId(map.doors());

      for (const auto& [_, room] : map.rooms()) {
        UpdateNextId(room.panels());
        UpdateNextId(room.masteries());
        UpdateNextId(room.keyholders());
      }
    }

    UpdateNextId(id_mappings_.special());
    UpdateNextId(id_mappings_.letters());
    UpdateNextId(id_mappings_.endings());
    UpdateNextId(id_mappings_.progressives());
    UpdateNextId(id_mappings_.door_groups());

    next_id_++;
  }

  void WriteIds(std::filesystem::path path) {
    WriteIdsAsYaml(output_, path.string());
  }

  void ProcessMaps(std::filesystem::path path) {
    std::filesystem::path maps_dir = path / "maps";
    for (auto const& dir_entry :
         std::filesystem::directory_iterator(maps_dir)) {
      ProcessMap(dir_entry.path());
    }
  }

  void ProcessMap(std::filesystem::path path) {
    std::string map_name = path.filename().string();

    ProcessDoorsFile(path / "doors.txtpb", map_name);
    ProcessRooms(path / "rooms", map_name);
  }

  void ProcessDoorsFile(std::filesystem::path path,
                        const std::string& current_map_name) {
    if (!std::filesystem::exists(path)) {
      return;
    }

    auto doors = ReadMessageFromFile<HumanDoors>(path.string());

    for (const HumanDoor& door : doors.doors()) {
      ProcessDoor(door, current_map_name);
    }
  }

  void ProcessDoor(const HumanDoor& h_door,
                   const std::string& current_map_name) {
    if (h_door.type() == DoorType::EVENT && !h_door.latch()) {
      return;
    }

    auto& maps = *output_.mutable_maps();
    auto& doors = *maps[current_map_name].mutable_doors();

    if (!id_mappings_.maps().contains(current_map_name) ||
        !id_mappings_.maps()
             .at(current_map_name)
             .doors()
             .contains(h_door.name())) {
      doors[h_door.name()] = next_id_++;
    } else {
      doors[h_door.name()] =
          id_mappings_.maps().at(current_map_name).doors().at(h_door.name());
    }
  }

  void ProcessRooms(std::filesystem::path path,
                    const std::string& current_map_name) {
    for (auto const& dir_entry : std::filesystem::directory_iterator(path)) {
      auto room = ReadMessageFromFile<HumanRoom>(dir_entry.path().string());
      ProcessRoom(room, current_map_name);
    }
  }

  void ProcessRoom(const HumanRoom& h_room,
                   const std::string& current_map_name) {
    for (const HumanPanel& h_panel : h_room.panels()) {
      auto& maps = *output_.mutable_maps();
      auto& rooms = *maps[current_map_name].mutable_rooms();
      auto& panels = *rooms[h_room.name()].mutable_panels();

      if (!id_mappings_.maps().contains(current_map_name) ||
          !id_mappings_.maps()
               .at(current_map_name)
               .rooms()
               .contains(h_room.name()) ||
          !id_mappings_.maps()
               .at(current_map_name)
               .rooms()
               .at(h_room.name())
               .panels()
               .contains(h_panel.name())) {
        panels[h_panel.name()] = next_id_++;
      } else {
        panels[h_panel.name()] = id_mappings_.maps()
                                     .at(current_map_name)
                                     .rooms()
                                     .at(h_room.name())
                                     .panels()
                                     .at(h_panel.name());
      }
    }

    for (const HumanLetter& h_letter : h_room.letters()) {
      std::string lettername = GetLetterName(h_letter.key(), h_letter.level2());

      auto& letters = *output_.mutable_letters();
      if (!id_mappings_.letters().contains(lettername)) {
        letters[lettername] = next_id_++;
      } else {
        letters[lettername] = id_mappings_.letters().at(lettername);
      }
    }

    for (const HumanMastery& h_mastery : h_room.masteries()) {
      auto& maps = *output_.mutable_maps();
      auto& rooms = *maps[current_map_name].mutable_rooms();
      auto& masteries = *rooms[h_room.name()].mutable_masteries();

      if (!id_mappings_.maps().contains(current_map_name) ||
          !id_mappings_.maps()
               .at(current_map_name)
               .rooms()
               .contains(h_room.name()) ||
          !id_mappings_.maps()
               .at(current_map_name)
               .rooms()
               .at(h_room.name())
               .masteries()
               .contains(h_mastery.name())) {
        masteries[h_mastery.name()] = next_id_++;
      } else {
        masteries[h_mastery.name()] = id_mappings_.maps()
                                          .at(current_map_name)
                                          .rooms()
                                          .at(h_room.name())
                                          .masteries()
                                          .at(h_mastery.name());
      }
    }

    for (const HumanEnding& h_ending : h_room.endings()) {
      auto& endings = *output_.mutable_endings();

      if (!id_mappings_.endings().contains(h_ending.name())) {
        endings[h_ending.name()] = next_id_++;
      } else {
        endings[h_ending.name()] = id_mappings_.endings().at(h_ending.name());
      }
    }

    for (const HumanKeyholder& h_keyholder : h_room.keyholders()) {
      if (!h_keyholder.has_key()) {
        continue;
      }

      auto& maps = *output_.mutable_maps();
      auto& rooms = *maps[current_map_name].mutable_rooms();
      auto& keyholders = *rooms[h_room.name()].mutable_keyholders();

      if (!id_mappings_.maps().contains(current_map_name) ||
          !id_mappings_.maps()
               .at(current_map_name)
               .rooms()
               .contains(h_room.name()) ||
          !id_mappings_.maps()
               .at(current_map_name)
               .rooms()
               .at(h_room.name())
               .keyholders()
               .contains(h_keyholder.name())) {
        keyholders[h_keyholder.name()] = next_id_++;
      } else {
        keyholders[h_keyholder.name()] = id_mappings_.maps()
                                             .at(current_map_name)
                                             .rooms()
                                             .at(h_room.name())
                                             .keyholders()
                                             .at(h_keyholder.name());
      }
    }
  }

  void ProcessSpecialIds() {
    auto& specials = *output_.mutable_special();

    for (const auto& [special_name, ap_id] : id_mappings_.special()) {
      specials[special_name] = ap_id;
    }
  }

  void ProcessProgressivesFile(std::filesystem::path path) {
    if (!std::filesystem::exists(path)) {
      return;
    }

    auto h_progs = ReadMessageFromFile<HumanProgressives>(path.string());
    auto& progs = *output_.mutable_progressives();

    for (const HumanProgressive& h_prog : h_progs.progressives()) {
      if (!id_mappings_.progressives().contains(h_prog.name())) {
        progs[h_prog.name()] = next_id_++;
      } else {
        progs[h_prog.name()] = id_mappings_.progressives().at(h_prog.name());
      }
    }
  }

  void ProcessDoorGroupsFile(std::filesystem::path path) {
    if (!std::filesystem::exists(path)) {
      return;
    }

    auto h_groups = ReadMessageFromFile<HumanDoorGroups>(path.string());
    auto& groups = *output_.mutable_door_groups();

    for (const HumanDoorGroup& h_group : h_groups.door_groups()) {
      if (!id_mappings_.door_groups().contains(h_group.name())) {
        groups[h_group.name()] = next_id_++;
      } else {
        groups[h_group.name()] = id_mappings_.door_groups().at(h_group.name());
      }
    }
  }

  void ProcessGlobalMetadataFile(std::filesystem::path path) {
    if (!std::filesystem::exists(path)) {
      return;
    }

    auto h_metadata = ReadMessageFromFile<HumanGlobalMetadata>(path.string());
    auto& specials = *output_.mutable_special();

    for (const std::string& h_special : h_metadata.special_names()) {
      if (!id_mappings_.special().contains(h_special)) {
        specials[h_special] = next_id_++;
      } else {
        specials[h_special] = id_mappings_.special().at(h_special);
      }
    }
  }

 private:
  void UpdateNextId(const google::protobuf::Map<std::string, uint64_t>& ids) {
    for (const auto& [_, id] : ids) {
      if (id > next_id_) {
        next_id_ = id;
      }
    }
  }

  std::string mapdir_;

  uint64_t next_id_ = 1;

  IdMappings id_mappings_;
  IdMappings output_;
};

}  // namespace
}  // namespace com::fourisland::lingo2_archipelago

int main(int argc, char** argv) {
  if (argc != 2) {
    std::cout << "Incorrect argument count." << std::endl;
    std::cout << "Usage: assign_ids [path to map directory]" << std::endl;
    return 1;
  }

  std::string mapdir = argv[1];

  com::fourisland::lingo2_archipelago::AssignIds assign_ids(mapdir);
  assign_ids.Run();

  return 0;
}
generated by cgit-pink 1.4.1 (git 2.36.1) at 2026-03-24 00:23:13 +0000