AS3四叉树 QuadTree

转载http://dengshiheng.blog.163.com/blog/static/1348387192011102232918581/

 

在一些游戏中，当地图上有很多元素时，或者有一张很大的地图需要渲染。通常的做法是遍历所有的显示对象，但是这样做，可能会占用大量的CPU，所以为了处理这种情况。我们可以使用四叉树，这里的四叉树，就是把场景划分为四等分，然后每一小份再分成四等分，这下分下去，通常情况分四五次就行了。具体分几层，可以在实践中去测试，以获得最高的效率。
下面是四叉树中的公开的类，其实很简单，就是先建树，然后往里面添加对象，移除对象，清空所有的对象。最重要的就是搜索显示对象。

package bing.utils
{
 import flash.display.DisplayObject;
 import flash.geom.Point;
 import flash.geom.Rectangle;

 /**
  * 四叉树 ，主要用于渲染动态地表和建筑
  * @author zhouzhanglin
  * @date 2010/9/11
  */ 
 public class QuadTree
 {
  //四叉树的层数
  private var _layerNum:int = 0;
  //最大的范围
  private var _maxRect:Rectangle  = null ;
  //主节点
  private var _mainNode:Node = null ;
  //节点集合
  private var _nodeList:Vector.<Node> = null ;
  
  /**
   *  四叉树构造函数
   * @param layerNum 四叉树的层数
   * @param maxRect 最大的范围
   */  
  public function QuadTree( layerNum:int , maxRect:Rectangle )
  {
   this._layerNum = layerNum+1 ; //四叉树的层数
   _maxRect = maxRect ;
   _nodeList = new Vector.<Node>();
   //初始化树的根节点
   _mainNode = new Node();
   _mainNode.hasSon = true ;
   _mainNode.rect = this._maxRect ;
   initTree( _mainNode );
  }
  /**
   * 初始化树
   * @node 树节点
   */
  private function initTree( node:Node):void{
   if(node==null || node.rect.width<=this._maxRect.width/Math.pow(2,_layerNum ) || node.rect.height<=this._maxRect.height/Math.pow(2,_layerNum ) ){
    node.hasSon = false ;
    return ;
   }
   _nodeList.push( node );
   //设置子节点
   for( var i:int=0; i<node.sonNodeList.length; ++i){
    node.sonNodeList[i]= new Node();
    node.sonNodeList[i].parentNode = node; 
    node.sonNodeList[i].rect = new Rectangle( node.rect.x + (i % 2) * node.rect.width*0.5, node.rect.y + int( i > 1) * node.rect.height*0.5, node.rect.width*0.5, node.rect.height*0.5);
    initTree(node.sonNodeList[i]); 
   }
  }
  
  /**
   * 添加可视对象到树中 
   * @param obj 类型为DisplayObjects
   */  
  public function insertObj( obj:DisplayObject ):void{
   var sonNode:Node = searchNodeByPoint(new Point(obj.x,obj.y) , _mainNode );
   sonNode.objVec.push( obj );
  }
  
  /**
   * 从树中删除对象 
   * @param obj
   */  
  public function deleteObj (obj:DisplayObject):void{
   var sonNode:Node = searchNodeByPoint(new Point(obj.x,obj.y) , _mainNode );
   for(var i:int = 0;i<sonNode.objVec.length ; i++){
    if(sonNode.objVec[i]==obj){
     sonNode.objVec.splice(i,1);
     break;
    }
   }
  }
  
  /**
   * 通过矩形区域，查询显示对象 
   * @param rect 矩形区域
   * @param exact true表示精确查询
   * @return 该区域的显示对象集合
   */  
  public function searchByRect( rect:Rectangle , exact:Boolean ):Vector.<DisplayObject>{
   var objVec:Vector.<DisplayObject> = new Vector.<DisplayObject>();
   if(_mainNode!=null){ 
    queryAndAdd(objVec , rect , _mainNode ,exact ) ;
   }
   return objVec ;
  }
  
  /**
   * 遍历节点和儿子节点，查找最终的对象 
   * @param objVec 查询结果
   * @param rect 范围
   * @param tempNode
   */  
  private function queryAndAdd(objVec:Vector.<DisplayObject>,rect:Rectangle ,  tempNode:Node ,exact:Boolean ):void{
   //如果没有交集，则返回
   if(!rect.intersects(tempNode.rect)){
    return;
   }
   //判断是否有儿子节点，递归找儿子
   if(tempNode.hasSon){
    //遍历儿子节点
    for each(var son:Node in tempNode.sonNodeList){
     if(son.rect.intersects(rect)){ 
      queryAndAdd(objVec,rect, son ,exact);
     }
    }
   }else{
    //如果是最后的节点，则把里面的对象加入数组中
    for each(var obj:DisplayObject in tempNode.objVec){
     if(exact){
      var sonRect:Rectangle = new Rectangle(obj.x,obj.y,obj.width,obj.height) ;
      if(sonRect.intersects(rect)){
       objVec.push( obj );
      }
     }else{
      objVec.push( obj );
     }
    }
   }
  }
  
  /**
   * 通过坐标来找节点 
   * @param point
   * @return 
   */  
  private function searchNodeByPoint( point:Point ,node:Node ):Node{
   if(node.hasSon){
    if(node.checkPointIsIn(point)){
     //遍历儿子节点
     for each(var son:Node in node.sonNodeList){
      if(son.checkPointIsIn(point )){ 
       node = searchNodeByPoint( point , son );
      }
     }
    }
   }
   return node ;
  }
  
  /**
   * 从四叉树中移除所有 
   */  
  public function removeAll():void 
  {
   for each( var node:Node in _nodeList)
   {
    node.dispose() ;
   }
  }
 }
}

//===============================

import flash.display.DisplayObject;
import flash.geom.Point;
import flash.geom.Rectangle;

/**
 *  四叉树的节点
 * @author zhouzhanglin
 * @date 2010/9/11
 */
class Node{
 //四个子节点
 private var oneNode:Node = null;
 private var twoNode:Node = null ;
 private var threeNode:Node = null ;
 private var fourNode:Node = null ;
 //此节点的范围
 public var rect:Rectangle  = null ;
 //此节点的父亲节点
 public var parentNode:Node = null ;
 //是否有子节点
 public var hasSon:Boolean = true ;
 //此节点下所有的对象集合
 public var objVec:Vector.<DisplayObject> = new Vector.<DisplayObject>();
 //此节点的儿子节点集合
 public var sonNodeList:Vector.<Node> = new Vector.<Node>();
 
 public function Node(){
  sonNodeList.push(oneNode);
  sonNodeList.push(twoNode);
  sonNodeList.push(threeNode);
  sonNodeList.push(fourNode);
 }
 
 /**
  * 判断点是否在此节点中 
  * @param point
  * @return 
  */ 
 public function checkPointIsIn(point:Point):Boolean{
  if(point.x>=this.rect.x&&point.y>=this.rect.y&&point.x<this.rect.x+this.rect.width&&point.y<this.rect.y+this.rect.height){
   return true;
  }
  return false;
 }
 
 /** 
  * 判断是否是叶子节点
  * @param node
  * @return 
  */ 
 public function isLeaf(node:Node):Boolean{
  if(this.parentNode!=null&&node.parentNode!=null&&this.parentNode==node.parentNode){
   return true;
  }
  return false;
 }
 
 public function dispose():void
 {
  if(sonNodeList)
  {
   for each( var node:Node in sonNodeList)
   {
    if(node) node.dispose() ;
   }
  }
  
  if(objVec)
  {
   for each( var mc:DisplayObject in objVec)
   {
    mc = null ;
   }
   objVec = new Vector.<DisplayObject>(); ;
  }
 }
}

 

下面是元素类

package test
{
 import flash.display.Sprite;

 public class Model extends Sprite 
 {
  public function Model()
  {
   this.graphics.beginFill(0x333333 );
   this.graphics.drawRect(0,0,20,20);
   this.graphics.endFill();
  }
 }
}

 下面是测试四叉树，现在场景上有一千个元素，我们可以用鼠标快速地选择你想要的

 注：flex 3.x中没有Vector数组用Array替代。flex 4.X才有Vector数组。

源码示例下载：http://163.fm/ZHYQXAO   提取码：rJ255VV9

本文转载于：http://www.swfappstore.com/?p=18 《示例也可到原文中下载》

本实例存储在邮箱附件里。路径：网易博客附件存放文件夹>0.0.1>