project_tileset_artifact_reader.cpp

Go to the documentation of this file.

#include "porytiles/infra/repos/project_tileset_artifact_reader.hpp"
 
#include <expected>
#include <filesystem>
#include <format>
#include <fstream>
#include <functional>
#include <iterator>
#include <optional>
 
#include "porytiles/domain/models/animation.hpp"
#include "porytiles/domain/models/metatile_attribute.hpp"
#include "porytiles/domain/models/porytiles_tileset_component.hpp"
#include "porytiles/domain/models/tilemap_entry.hpp"
#include "porytiles/domain/models/tileset.hpp"
#include "porytiles/domain/repos/artifact_key.hpp"
#include "porytiles/infra/algorithms/anim_frame_loader.hpp"
#include "porytiles/infra/algorithms/porymap_artifact_parsers.hpp"
#include "porytiles/utilities/dynamic_cased_name.hpp"
#include "porytiles/utilities/panic/panic.hpp"
#include "porytiles/utilities/string_utils.hpp"
 
#include <iostream>
 
namespace {
 
using namespace porytiles;
 
ChainableResult<void> import_layer_png(
    Tileset &dest,
    const ArtifactKey &src_key,
    const std::filesystem::path &project_root,
    const PngRgbaImageLoader &loader,
    const std::function<void(PorytilesTilesetComponent &, const Image<Rgba32> &)> &layer_img_setter)
{
    // Keys are relative to project_root, so prepend for file I/O
    auto image_result = loader.load_from_file((project_root / src_key.key()).string());
    if (!image_result.has_value()) {
        switch (image_result.error().type()) {
        case ImageLoadError::Type::file_not_found:
        case ImageLoadError::Type::unsupported_channel_count:
        case ImageLoadError::Type::other_load_error:
            return ChainableResult<void>{
                FormattableError{"Failed to load layer image '{}'.", FormatParam{src_key.key(), Style::bold}},
                image_result};
        default:
            panic("unhandled ImageLoadError type");
        }
    }
    layer_img_setter(dest.porytiles_component(), *image_result.value());
    return {};
}
 
ChainableResult<void> import_porytiles_palette(
    Tileset &dest,
    const ArtifactKey &src_key,
    std::size_t index,
    const std::filesystem::path &project_root,
    const FilePaletteLoader &loader)
{
    if (index >= palette::num_palettes) {
        panic(std::format("invalid palette index {}: out of range", index));
    }
 
    // Keys are relative to project_root, so prepend for file I/O
    PT_TRY_ASSIGN_CHAIN_ERR(
        palette,
        loader.load_with_wildcards((project_root / src_key.key()).string()),
        void,
        "Failed to load palette file.");
    dest.porytiles_component().set_palette(index, palette);
 
    return {};
}
 
template <SupportsTransparency PixelType, typename LoaderType, typename ComponentGetter>
ChainableResult<void> import_anim_frame_impl(
    Tileset &dest,
    const ArtifactKey &src_key,
    const std::filesystem::path &project_root,
    const std::string &anim_name,
    const std::string &frame_name,
    const LoaderType &loader,
    ComponentGetter component_getter,
    std::string_view error_context)
{
    // Use shared helper to load PNG and extract tiles
    const auto png_path = project_root / src_key.key();
    PT_TRY_ASSIGN_CHAIN_ERR(
        load_result,
        load_animation_frame_from_png<PixelType>(png_path, frame_name, loader),
        void,
        "{}: failed to load {}",
        FormatParam(src_key.key(), Style::bold),
        FormatParam(std::string(error_context)));
 
    // Get or create the animation in the component
    auto &component = component_getter(dest);
    if (!component.has_anim(anim_name)) {
        Animation<PixelType> anim{anim_name};
        component.add_anim(std::move(anim));
    }
 
    auto &anim = component.anims().at(anim_name);
 
    if (frame_name == "key") {
        // Key frame
        anim.key_frame(std::move(load_result.frame));
    }
    else {
        // Regular frame - use string-based map storage
        anim.put_frame(frame_name, std::move(load_result.frame));
    }
 
    // Update dimensions in params if not already set
    auto params = anim.params();
    if (params.width_tiles() == 0 && params.height_tiles() == 0) {
        params.width_tiles(load_result.width_tiles);
        params.height_tiles(load_result.height_tiles);
        anim.params(std::move(params));
    }
 
    return {};
}
 
} // namespace
 
namespace porytiles {
 
// Porymap artifacts
ChainableResult<void> ProjectTilesetArtifactReader::read_metatiles_bin(Tileset &dest, const ArtifactKey &src_key) const
{
    // Keys are relative to project_root_, so prepend for file I/O
    PT_TRY_ASSIGN_PASS_ERR(entries, parse_metatiles_bin(project_root_ / src_key.key()), void);
    for (auto &entry : entries) {
        dest.porymap_component().push_back_tilemap_entry(std::move(entry));
    }
    return {};
}
 
ChainableResult<void>
ProjectTilesetArtifactReader::read_metatile_attributes_bin(Tileset &dest, const ArtifactKey &src_key) const
{
    // Keys are relative to project_root_, so prepend for file I/O
    const auto path = project_root_ / src_key.key();
    PT_TRY_ASSIGN_CHAIN_ERR(
        attributes, parse_metatile_attributes(path, *schema_), void, "Failed to read metatile_attributes.bin.");
    for (auto &attribute : attributes) {
        dest.porymap_component().push_back_attribute(std::move(attribute));
    }
    return {};
}
 
ChainableResult<void> ProjectTilesetArtifactReader::read_tiles_png(Tileset &dest, const ArtifactKey &src_key) const
{
    // Keys are relative to project_root_, so prepend for file I/O
    PT_TRY_ASSIGN_PASS_ERR(image, load_indexed_png(project_root_ / src_key.key(), *png_indexed_loader_), void);
    dest.porymap_component().tiles_png(*image);
    return {};
}
 
ChainableResult<void>
ProjectTilesetArtifactReader::read_porymap_palette_n(Tileset &dest, const ArtifactKey &src_key, std::size_t index) const
{
    if (index >= palette::num_palettes) {
        panic(std::format("invalid palette index {}: out of range", index));
    }
    // Keys are relative to project_root_, so prepend for file I/O
    PT_TRY_ASSIGN_PASS_ERR(palette, load_porymap_palette(project_root_ / src_key.key(), *palette_loader_), void);
    dest.porymap_component().set_palette(index, std::move(palette));
    return {};
}
 
[[nodiscard]] ChainableResult<void> ProjectTilesetArtifactReader::read_porymap_anim(
    Tileset &dest,
    const std::string &anim_name,
    const ArtifactKey &params_key,
    const std::vector<std::pair<std::string, ArtifactKey>> &frame_keys) const
{
    // Load each frame using the unified template helper
    // The first call creates the animation in the component, subsequent calls add frames to it
    for (const auto &[frame_name, frame_key] : frame_keys) {
        PT_TRY_CALL_PASS_ERR(
            (import_anim_frame_impl<IndexPixel>(
                dest,
                frame_key,
                project_root_,
                anim_name,
                frame_name,
                *png_indexed_loader_,
                [](Tileset &t) -> PorymapTilesetComponent & { return t.porymap_component(); },
                "Porymap animation frame")),
            void);
    }
 
    // Skip param loading if there is no generated header file present
    std::filesystem::path params_path = project_root_ / params_key.key();
    if (!std::filesystem::exists(params_path)) {
        return {};
    }
 
    // Setup callback info, use Porytiles-managed callback regardless of metadata
    const auto tileset_cased = extract_tileset_cased_name(dest.name());
    const std::string callback_func = anim::managed_callback_name(dest.name());
 
    // Parse C code for animation params
    PT_TRY_ASSIGN_CHAIN_ERR(
        parsed_params,
        anim_code_parser_->parse_from_callback(params_path, callback_func, tileset_cased, true),
        void,
        "{}: Failed to parse animation parameters.",
        FormatParam(params_path, Style::bold));
 
    // Apply params to the specific animation if found
    if (parsed_params.contains(DynamicCasedName{anim_name})) {
        auto &anim = dest.porymap_component().anims().at(anim_name);
        auto existing_params = anim.params();
        auto new_params = parsed_params.at(DynamicCasedName{anim_name});
 
        // Preserve dimensions from frame import (C code doesn't have this info)
        new_params.width_tiles(existing_params.width_tiles());
        new_params.height_tiles(existing_params.height_tiles());
 
        anim.params(std::move(new_params));
    }
 
    return {};
}
 
// Porytiles artifacts
ChainableResult<void> ProjectTilesetArtifactReader::read_bottom_png(Tileset &dest, const ArtifactKey &src_key) const
{
    PT_TRY_CALL_PASS_ERR(
        import_layer_png(
            dest,
            src_key,
            project_root_,
            *png_rgba_loader_,
            [](PorytilesTilesetComponent &comp, const Image<Rgba32> &img) { comp.bottom(img); }),
        void);
    return {};
}
 
ChainableResult<void> ProjectTilesetArtifactReader::read_middle_png(Tileset &dest, const ArtifactKey &src_key) const
{
    PT_TRY_CALL_PASS_ERR(
        import_layer_png(
            dest,
            src_key,
            project_root_,
            *png_rgba_loader_,
            [](PorytilesTilesetComponent &comp, const Image<Rgba32> &img) { comp.middle(img); }),
        void);
    return {};
}
 
ChainableResult<void> ProjectTilesetArtifactReader::read_top_png(Tileset &dest, const ArtifactKey &src_key) const
{
    PT_TRY_CALL_PASS_ERR(
        import_layer_png(
            dest,
            src_key,
            project_root_,
            *png_rgba_loader_,
            [](PorytilesTilesetComponent &comp, const Image<Rgba32> &img) { comp.top(img); }),
        void);
    return {};
}
 
ChainableResult<void> ProjectTilesetArtifactReader::read_attributes_csv(Tileset &dest, const ArtifactKey &src_key) const
{
    // The CSV still loads using the owning tileset's name. The schema is supposed to be project-global, but per-tileset
    // formatting config like role_pins resolves at that tileset's config scope.
    PT_TRY_ASSIGN_PASS_ERR(
        load_result,
        attributes_csv_loader_->load((project_root_ / src_key.key()).string(), *schema_, *provider_map_, dest.name()),
        void);
    for (const auto &[metatile_id, attribute] : load_result.attributes) {
        dest.porytiles_component().insert_attribute(metatile_id, attribute);
    }
    // Record, per role, whether the active pin column was present. The decompiler's round-trip merge reads this to
    // decide between preserving prior pin state (column present) and pinning every row (column absent).
    for (const auto &[role, present] : load_result.active_pin_column_present) {
        dest.porytiles_component().prior_pin_column_state(
            role, present ? PriorPinColumnState::column_present : PriorPinColumnState::column_absent);
    }
    return {};
}
 
ChainableResult<void> ProjectTilesetArtifactReader::read_porytiles_palette_n(
    Tileset &dest, const ArtifactKey &src_key, std::size_t index) const
{
    PT_TRY_CALL_PASS_ERR(import_porytiles_palette(dest, src_key, index, project_root_, *palette_loader_), void);
    return {};
}
 
[[nodiscard]] ChainableResult<void> ProjectTilesetArtifactReader::read_porytiles_anim(
    Tileset &dest,
    const std::string &anim_name,
    const ArtifactKey &params_key,
    const std::optional<ArtifactKey> &key_frame_key,
    const std::vector<std::pair<std::string, ArtifactKey>> &frame_keys) const
{
    // Parse anim.json to get params for this animation
    // Keys are relative to project_root_, so prepend for file I/O
    PT_TRY_ASSIGN_CHAIN_ERR(
        parsed_anim_params,
        anim_json_parser_->parse((project_root_ / params_key.key()).string()),
        void,
        "{}: Failed to parse animation parameters.",
        FormatParam(project_root_ / params_key.key(), Style::bold));
 
    // Find params for this specific animation
    auto it = parsed_anim_params.find(DynamicCasedName{anim_name});
    if (it == parsed_anim_params.end()) {
        return ChainableResult<void>{FormattableError{
            "{}: Animation '{}' not found in animation parameters file.",
            FormatParam{project_root_ / params_key.key(), Style::bold},
            FormatParam{anim_name, Style::bold}}};
    }
 
    const auto &params = it->second;
 
    // Create the animation with params FIRST, before loading frames
    // This ensures YAML-specified width_tiles/height_tiles take precedence over auto-detected dimensions
    Animation<Rgba32> anim{anim_name, params};
    dest.porytiles_component().add_anim(std::move(anim));
 
    // Load key frame using the unified template helper (only present for automatic/hybrid frame linking)
    if (key_frame_key.has_value()) {
        PT_TRY_CALL_PASS_ERR(
            (import_anim_frame_impl<Rgba32>(
                dest,
                key_frame_key.value(),
                project_root_,
                anim_name,
                "key",
                *png_rgba_loader_,
                [](Tileset &t) -> PorytilesTilesetComponent & { return t.porytiles_component(); },
                "Porytiles animation frame")),
            void);
    }
 
    // Load remaining frames using the unified template helper
    for (const auto &[frame_name, frame_key] : frame_keys) {
        PT_TRY_CALL_PASS_ERR(
            (import_anim_frame_impl<Rgba32>(
                dest,
                frame_key,
                project_root_,
                anim_name,
                frame_name,
                *png_rgba_loader_,
                [](Tileset &t) -> PorytilesTilesetComponent & { return t.porytiles_component(); },
                "Porytiles animation frame")),
            void);
    }
 
    return {};
}
 
[[nodiscard]] ChainableResult<void>
ProjectTilesetArtifactReader::read_porytiles_primary_anim_references(Tileset &dest, const ArtifactKey &params_key) const
{
    const auto json_path = project_root_ / params_key.key();
 
    PT_TRY_ASSIGN_CHAIN_ERR(
        parsed_refs,
        anim_json_parser_->parse_primary_references(json_path),
        void,
        "{}: Failed to parse primary animation references.",
        FormatParam(json_path, Style::bold));
 
    if (parsed_refs.empty()) {
        return {};
    }
 
    std::map<std::string, std::vector<AnimOverrideEntry>> converted;
    for (auto &[cased_name, entries] : parsed_refs) {
        converted[cased_name.to_snake_case()] = std::move(entries);
    }
 
    dest.porytiles_component().primary_anim_overrides(std::move(converted));
    return {};
}
 
} // namespace porytiles