awayjs-display/lib/partition/SceneGraphPartition.ts

AwayJS displaylist classes

import IAbstractionPool				from "awayjs-core/lib/library/IAbstractionPool";

import DisplayObject				from "../display/DisplayObject";
import DisplayObjectContainer		from "../display/DisplayObjectContainer";
import ITraverser				from "../ITraverser";
import SceneGraphNode				from "../partition/SceneGraphNode";
import PartitionBase				from "../partition/PartitionBase";
import IContainerNode				from "../partition/IContainerNode";
import DisplayObjectNode			from "../partition/DisplayObjectNode";

/**
 * @class away.partition.Partition
 */
class SceneGraphPartition extends PartitionBase
{
	private _sceneGraphNodePool:SceneGraphNodePool;

	constructor()
	{
		super();

		this._sceneGraphNodePool = new SceneGraphNodePool();
	}

	public traverse(traverser:ITraverser)
	{
		super.traverse(traverser);
	}


	/**
	 *
	 * @param entity
	 * @returns {away.partition.NodeBase}
	 */
	public findParentForNode(node:DisplayObjectNode):IContainerNode
	{
		if (node.isSceneGraphNode && (node._displayObject.partition == this || node._displayObject._iIsRoot)) {
			this._rootNode = <SceneGraphNode> node;
			return null;
		}

		if (!node.isSceneGraphNode && node._displayObject.isContainer)
			return this._sceneGraphNodePool.getAbstraction(<DisplayObjectContainer> node._displayObject);

		return this._sceneGraphNodePool.getAbstraction(node._displayObject.parent);
	}

	/**
	 * @internal
	 */
	public _iRegisterEntity(displayObject:DisplayObject)
	{
		super._iRegisterEntity(displayObject);

		if (displayObject.isContainer)
			this.iMarkForUpdate(this._sceneGraphNodePool.getAbstraction(<DisplayObjectContainer> displayObject));
	}

	/**
	 * @internal
	 */
	public _iUnregisterEntity(displayObject:DisplayObject)
	{
		super._iUnregisterEntity(displayObject);

		if (displayObject.isContainer)
			this.iRemoveEntity(this._sceneGraphNodePool.getAbstraction(<DisplayObjectContainer> displayObject));
	}
}

export default SceneGraphPartition;


/**
 * @class away.pool.SceneGraphNodePool
 */
class SceneGraphNodePool implements IAbstractionPool
{
	private _abstractionPool:Object = new Object();

	/**
	 * //TODO
	 *
	 * @param entity
	 * @returns EntityNode
	 */
	public getAbstraction(displayObjectContainer:DisplayObjectContainer):SceneGraphNode
	{
		return (this._abstractionPool[displayObjectContainer.id] || (this._abstractionPool[displayObjectContainer.id] = new SceneGraphNode(displayObjectContainer, this)));
	}

	/**
	 * //TODO
	 *
	 * @param entity
	 */
	public clearAbstraction(displayObjectContainer:DisplayObjectContainer)
	{
		delete this._abstractionPool[displayObjectContainer.id];
	}
}