Babylon.js/Babylon/Culling/Octrees/babylon.octreeBlock.ts

module BABYLON {
    export class OctreeBlock {
        public meshes = new Array<Mesh>();
        public subMeshes = new Array <Array<SubMesh>>();
        public blocks: Array<OctreeBlock>;

        private _capacity: number;
        private _minPoint: Vector3;
        private _maxPoint: Vector3;
        private _boundingVectors = new Array<Vector3>();

        constructor(minPoint: Vector3, maxPoint: Vector3, capacity: number) {
            this._capacity = capacity;

            this._minPoint = minPoint;
            this._maxPoint = maxPoint;

            this._boundingVectors.push(minPoint.clone());
            this._boundingVectors.push(maxPoint.clone());

            this._boundingVectors.push(minPoint.clone());
            this._boundingVectors[2].x = maxPoint.x;

            this._boundingVectors.push(minPoint.clone());
            this._boundingVectors[3].y = maxPoint.y;

            this._boundingVectors.push(minPoint.clone());
            this._boundingVectors[4].z = maxPoint.z;

            this._boundingVectors.push(maxPoint.clone());
            this._boundingVectors[5].z = minPoint.z;

            this._boundingVectors.push(maxPoint.clone());
            this._boundingVectors[6].x = minPoint.x;

            this._boundingVectors.push(maxPoint.clone());
            this._boundingVectors[7].y = minPoint.y;
        }

        // Methods
        public addMesh(mesh: Mesh): void {
            if (!mesh.subMeshes) {
                return;
            }

            if (this.blocks) {
                for (var index = 0; index < this.blocks.length; index++) {
                    var block = this.blocks[index];
                    block.addMesh(mesh);
                }
                return;
            }

            if (mesh.getBoundingInfo().boundingBox.intersectsMinMax(this._minPoint, this._maxPoint)) {
                var localMeshIndex = this.meshes.length;
                this.meshes.push(mesh);

                this.subMeshes[localMeshIndex] = [];
                for (var subIndex = 0; subIndex < mesh.subMeshes.length; subIndex++) {
                    var subMesh = mesh.subMeshes[subIndex];
                    if (mesh.subMeshes.length === 1 || subMesh.getBoundingInfo().boundingBox.intersectsMinMax(this._minPoint, this._maxPoint)) {
                        this.subMeshes[localMeshIndex].push(subMesh);
                    }
                }
            }

            if (this.subMeshes.length > this._capacity) {
                Octree._CreateBlocks(this._minPoint, this._maxPoint, this.meshes, this._capacity, this);
            }
        }

        public addEntries(meshes: Mesh[]): void {
            for (var index = 0; index < meshes.length; index++) {
                var mesh = meshes[index];
                this.addMesh(mesh);
            }
        }

        public select(frustumPlanes: Plane[], selection: OctreeBlock[]): void {
            if (this.blocks) {
                for (var index = 0; index < this.blocks.length; index++) {
                    var block = this.blocks[index];
                    block.select(frustumPlanes, selection);
                }
                return;
            }
            if (BABYLON.BoundingBox.IsInFrustum(this._boundingVectors, frustumPlanes)) {
                selection.push(this);
            }
        }
    }
}