Unity3D 8叉树

Octree.cs

using UnityEngine;
using System.Collections.Generic;

// 八叉树节点类
public struct OctreeNode<T> {
    // 节点包围盒的尺寸和信息
    public Bounds m_Bounds;

    // 节点当前深度
    public int m_CurrentDepth;

    // 节点包含对象列表
    public  List<GameObject> m_ObjectList;

    // 节点的子节点
    public OctreeNode<T>[] m_ChildNodes;
}

public class Octree<T>
{
    /// <summary>
    /// 根节点
    /// </summary>
    private OctreeNode<T> m_Root;
    public OctreeNode<T> Root
    {
        get
        {
            return m_Root;
        }

        set
        {
            m_Root = value;
        }
    }

    /// <summary>
    /// 最大深度
    /// </summary>
    private int m_MaxDepth;

    /// <summary>
    /// 构造函数
    /// </summary>
    /// <param name="center">树中心</param>
    /// <param name="size">树区域大小</param>
    /// <param name="maxDepth">树最大深度</param>
    public Octree(Vector3 center, Vector3 size, int maxDepth, List<GameObject> m_ObjectList)
    {
        // 初始化根节点
        this.m_MaxDepth = maxDepth;
        this.m_Root = new OctreeNode<T>();
        this.m_Root.m_Bounds = new Bounds(center, size);
        this.m_Root.m_CurrentDepth = 0;
        this.m_Root.m_ChildNodes = new OctreeNode<T>[8];
        this.m_Root.m_ObjectList = new List<GameObject>();
        this.m_Root.m_ObjectList = m_ObjectList;


        SplitBounds(ref this.m_Root, m_Root.m_ObjectList, this.m_Root.m_CurrentDepth);

    }

    /// <summary>
    /// 分八个区
    /// </summary>
    /// <param name="childNodes">八个子节点</param>
    /// <param name="oneBounds">当前节点的信息</param>
    private void SplitBounds(ref OctreeNode<T> currentNode, List<GameObject> objectList, int deep)
    {
        if (deep < m_MaxDepth)
        {
            currentNode.m_ObjectList = new List<GameObject>();
            // 1、判断是否相交 相交就添加到当前节点的列表里
            foreach (var gameObject in objectList)
            {
                if (gameObject.GetComponent<SphereCollider>() != null)
                {
                    var gameObjectBounds = gameObject.GetComponent<Collider>().bounds;
                    // 如果gameobject与分区相交 就添加到当前分区的列表里
                    if (currentNode.m_Bounds.Intersects(gameObjectBounds))
                    {
                        currentNode.m_ObjectList.Add(gameObject);
                    }
                }
            }

            // 2、分八个区域
            currentNode.m_ChildNodes = new OctreeNode<T>[8];
            currentNode.m_CurrentDepth = deep;

            var oneBounds = currentNode.m_Bounds;
            Vector3 half2Vector = oneBounds.size / 4;
            // 八块空间的位置
            currentNode.m_ChildNodes[0].m_Bounds.center = oneBounds.center - half2Vector;
            currentNode.m_ChildNodes[0].m_Bounds.size = oneBounds.size / 2;
            currentNode.m_ChildNodes[1].m_Bounds.center = currentNode.m_ChildNodes[0].m_Bounds.center + new Vector3(oneBounds.size.x / 2, 0, 0);
            currentNode.m_ChildNodes[1].m_Bounds.size = oneBounds.size / 2;
            currentNode.m_ChildNodes[2].m_Bounds.center = currentNode.m_ChildNodes[1].m_Bounds.center + new Vector3(0, 0, oneBounds.size.z / 2);
            currentNode.m_ChildNodes[2].m_Bounds.size = oneBounds.size / 2;
            currentNode.m_ChildNodes[3].m_Bounds.center = currentNode.m_ChildNodes[0].m_Bounds.center + new Vector3(0, 0, oneBounds.size.z / 2);
            currentNode.m_ChildNodes[3].m_Bounds.size = oneBounds.size / 2;
            currentNode.m_ChildNodes[4].m_Bounds.center = currentNode.m_ChildNodes[0].m_Bounds.center + new Vector3(0, oneBounds.size.y / 2, 0);
            currentNode.m_ChildNodes[4].m_Bounds.size = oneBounds.size / 2;
            currentNode.m_ChildNodes[5].m_Bounds.center = currentNode.m_ChildNodes[1].m_Bounds.center + new Vector3(0, oneBounds.size.y / 2, 0);
            currentNode.m_ChildNodes[5].m_Bounds.size = oneBounds.size / 2;
            currentNode.m_ChildNodes[6].m_Bounds.center = currentNode.m_ChildNodes[2].m_Bounds.center + new Vector3(0, oneBounds.size.y / 2, 0);
            currentNode.m_ChildNodes[6].m_Bounds.size = oneBounds.size / 2;
            currentNode.m_ChildNodes[7].m_Bounds.center = currentNode.m_ChildNodes[3].m_Bounds.center + new Vector3(0, oneBounds.size.y / 2, 0);
            currentNode.m_ChildNodes[7].m_Bounds.size = oneBounds.size / 2;

            // 3、遍历8个区域 递归调用
            for (int i = 0; i < currentNode.m_ChildNodes.Length; i++)
            {
                // 递归调用
                if (currentNode.m_ObjectList.Count != 0)
                {
                    // check if child node intersect
                    bool isIntersects = false;
                    currentNode.m_ChildNodes[i].m_ObjectList = new List<GameObject>();
                    foreach (var gameObject in currentNode.m_ObjectList)
                    {
                        if (gameObject.GetComponent<SphereCollider>() != null)
                        {
                            var gameObjectBounds = gameObject.GetComponent<Collider>().bounds;
                            if (currentNode.m_ChildNodes[i].m_Bounds.Intersects(gameObjectBounds))
                            {
                                currentNode.m_ChildNodes[i].m_ObjectList.Add(gameObject);
                                isIntersects = true;
                            }
                        }
                    }
                    if (isIntersects) {
                        if (currentNode.m_CurrentDepth != 0
                            && currentNode.m_ObjectList.Count == currentNode.m_ChildNodes[i].m_ObjectList.Count)
                        {
                            continue;
                        }
                        currentNode.m_CurrentDepth++;
                        SplitBounds(ref currentNode.m_ChildNodes[i], currentNode.m_ObjectList, currentNode.m_CurrentDepth);
                    }
                }
            }
        }
    }
}

Init.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class Init : MonoBehaviour {

    public GameObject root;
    Octree<GameObject> octree;
    // Use this for initialization
    void Start () {
        GameObject[] gameObjects = FindObjectsOfType(typeof(GameObject)) as GameObject[];

        List<GameObject> listGameObject = new List<GameObject>(gameObjects);

        Bounds rootBound = root.GetComponent<BoxCollider>().bounds;
        octree = new Octree<GameObject>(rootBound.center, rootBound.size, 10, listGameObject);
    }

    // Update is called once per frame
    void Update () {

    }

    // 画八叉树方框
#if UNITY_EDITOR
    private void OnDrawGizmos()
    {
        if (octree != null)
        {
            DrawCube(octree.Root.m_ChildNodes);
        }
    }

    private void DrawCube(OctreeNode<GameObject>[] octreeNode)
    {
        for (int i = 0; i < octreeNode.Length; i++)
        {
            if (octreeNode[i].m_ChildNodes != null)
            {
                DrawCube(octreeNode[i].m_ChildNodes);
            }
            octreeNode[i].m_Bounds.DrawBounds(new Color(0,0.7f,0));
        }
    }

#endif


}