rj-action-library/Runtime/Structures/Spatial/QuadTree/QuadTreeWalker.cs

using System.Collections;
using System.Collections.Generic;
using Godot;
using System;



namespace Rokojori
{
  public class QuadTreeWalker<T>:TreeWalker<QuadTreeNode<T>>
  { 
    public override QuadTreeNode<T> Parent( QuadTreeNode<T> node )
    {
      if ( node is QuadTree<T> )
      {
        return null;
      }

      var cell = node as QuadTreeCell<T>;

      return cell.isRoot ? cell.tree : cell.parent;
    }

    public override int NumChildren( QuadTreeNode<T> node )
    {
      if ( node is QuadTree<T> tree )
      {
        return tree.rootCells.Count;
      }

      var cell = node as QuadTreeCell<T>;

      return cell.numCells;
    }

    public override QuadTreeNode<T> ChildAt( QuadTreeNode<T> node, int index )
    {
      if ( index < 0 || index >= NumChildren( node ) )
      {
        return null;
      }
      
      if ( node is QuadTree<T> tree )
      {
        return tree.rootCells[ index ];
      }

      var cell = node as QuadTreeCell<T>;

      return cell.cells[ index ];
    }
  }
}