Craftimizer/Solver/Crafty/ArenaNode.cs

using System.Runtime.CompilerServices;

namespace Craftimizer.Solver.Crafty;

public sealed class ArenaNode<T> where T : struct
{
    public T State;
    public ArenaBuffer<T> Children;
    public NodeScoresBuffer ChildScores;
    public (int arrayIdx, int subIdx) ChildIdx;
    public readonly ArenaNode<T>? Parent;

    public NodeScoresBuffer? ParentScores => Parent?.ChildScores;

    public ArenaNode(T state, ArenaNode<T>? parent = null)
    {
        State = state;
        Children = new();
        ChildScores = new();
        Parent = parent;
    }

    public ArenaNode<T> ChildAt((int arrayIdx, int subIdx) at) =>
        Children.Data[at.arrayIdx][at.subIdx];

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    public ArenaNode<T> AddConcurrent(T state)
    {
        var node = new ArenaNode<T>(state, this);
        Children.AddConcurrent(node);
        ChildScores.AddConcurrent();
        return node;
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    public ArenaNode<T> Add(T state)
    {
        var node = new ArenaNode<T>(state, this);
        Children.Add(node);
        ChildScores.Add();
        return node;
    }
}