53 lines
1.0 KiB
C#
53 lines
1.0 KiB
C#
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 ];
|
|
}
|
|
}
|
|
} |