animation_graph_ops.h

#pragma once
 
#include "pixelbullet/scene/animation_graph_types.h"
 
#include <optional>
#include <string_view>
 
namespace pixelbullet
{
inline void NormalizeAnimationGraphAsset(AnimationGraphAsset& asset)
{
    if (asset.editor_graph && asset.editor_graph->Empty())
    {
        asset.editor_graph.reset();
    }
}
 
[[nodiscard]] inline std::optional<std::size_t> FindAnimationGraphStateIndex(const AnimationGraphAsset& asset,
                                                                             const std::string_view state_id)
{
    for (std::size_t index = 0; index < asset.states.size(); ++index)
    {
        if (asset.states[index].id == state_id)
        {
            return index;
        }
    }
    return std::nullopt;
}
 
[[nodiscard]] inline const AnimationGraphState* FindAnimationGraphState(const AnimationGraphAsset& asset,
                                                                        const std::string_view state_id)
{
    const auto index = FindAnimationGraphStateIndex(asset, state_id);
    return index ? &asset.states[*index] : nullptr;
}
 
[[nodiscard]] inline AnimationGraphState* FindAnimationGraphState(AnimationGraphAsset& asset, const std::string_view state_id)
{
    const auto index = FindAnimationGraphStateIndex(asset, state_id);
    return index ? &asset.states[*index] : nullptr;
}
 
[[nodiscard]] inline AnimationGraphEditorGraphNodeLayout* FindAnimationGraphEditorGraphNodeLayout(
    AnimationGraphAsset& asset, const std::string_view state_id)
{
    if (!asset.editor_graph)
    {
        return nullptr;
    }
 
    for (AnimationGraphEditorGraphNodeLayout& node : asset.editor_graph->nodes)
    {
        if (node.state_id == state_id)
        {
            return &node;
        }
    }
 
    return nullptr;
}
 
[[nodiscard]] inline const AnimationGraphEditorGraphNodeLayout* FindAnimationGraphEditorGraphNodeLayout(
    const AnimationGraphAsset& asset, const std::string_view state_id)
{
    if (!asset.editor_graph)
    {
        return nullptr;
    }
 
    for (const AnimationGraphEditorGraphNodeLayout& node : asset.editor_graph->nodes)
    {
        if (node.state_id == state_id)
        {
            return &node;
        }
    }
 
    return nullptr;
}
 
inline AnimationGraphEditorGraphNodeLayout& EnsureAnimationGraphEditorGraphNodeLayout(AnimationGraphAsset& asset,
                                                                                      const std::string_view state_id)
{
    if (!asset.editor_graph)
    {
        asset.editor_graph = AnimationGraphEditorGraphLayout{};
    }
 
    if (AnimationGraphEditorGraphNodeLayout* existing = FindAnimationGraphEditorGraphNodeLayout(asset, state_id))
    {
        return *existing;
    }
 
    asset.editor_graph->nodes.push_back(AnimationGraphEditorGraphNodeLayout{
        .state_id = std::string(state_id),
    });
    return asset.editor_graph->nodes.back();
}
} // namespace pixelbullet