Babylon.js/Babylon/Rendering/babylon.boundingBoxRenderer.js

"use strict";

var BABYLON = BABYLON || {};

(function () {
    BABYLON.BoundingBoxRenderer = function (scene) {
        this._scene = scene;
        this._colorShader = new BABYLON.ShaderMaterial("colorShader", scene, "color",
        {
            attributes: ["position"],
            uniforms: ["worldViewProjection", "color"]
        });
        this.frontColor = new BABYLON.Color3(1, 1, 1);
        this.backColor = new BABYLON.Color3(0.1, 0.1, 0.1);

        var engine = this._scene.getEngine();
        var boxdata = BABYLON.VertexData.CreateBox(1.0);
        this._vb = new BABYLON.VertexBuffer(null, boxdata.positions, BABYLON.VertexBuffer.PositionKind, false, engine);
        this._ib = engine.createIndexBuffer([0, 1, 1, 2, 2, 3, 3, 0, 4, 5, 5, 6, 6, 7, 7, 4, 0, 7, 1, 6, 2, 5, 3, 4]);

        this.renderList = new BABYLON.SmartArray(32);
    };

    // Members
    BABYLON.BoundingBoxRenderer.prototype.showBackLines = true;

    // Functions
    BABYLON.BoundingBoxRenderer.prototype.reset = function() {
        this.renderList.reset();
    };

    BABYLON.BoundingBoxRenderer.prototype.render = function () {
        if (this.renderList.length == 0 || !this._colorShader.isReady()) {
            return;
        }

        var engine = this._scene.getEngine();
        engine.setDepthWrite(false);
        this._colorShader._preBind();
        for (var boundingBoxIndex = 0; boundingBoxIndex < this.renderList.length; boundingBoxIndex++) {
            var mesh = this.renderList.data[boundingBoxIndex];
            var boundingBox = mesh.getBoundingInfo().boundingBox;
            var min = boundingBox.minimum;
            var max = boundingBox.maximum;
            var diff = max.subtract(min);
            var median = min.add(diff.scale(0.5));

            var worldMatrix = BABYLON.Matrix.Scaling(diff.x, diff.y, diff.z)
                .multiply(BABYLON.Matrix.Translation(median.x, median.y, median.z))
                .multiply(mesh.getWorldMatrix());

            // VBOs
            engine.bindBuffers(this._vb._buffer, this._ib, [3], 3 * 4, this._colorShader.getEffect());

            if (this.showBackLines) {
                // Back
                engine.setDepthFunctionToGreaterOrEqual();
                this._colorShader.setColor3("color", this.backColor);
                this._colorShader.bind(worldMatrix, mesh);

                // Draw order
                engine.draw(false, 0, 24);
            }

            // Front
            engine.setDepthFunctionToLess();
            this._colorShader.setColor3("color", this.frontColor);
            this._colorShader.bind(worldMatrix, mesh);

            // Draw order
            engine.draw(false, 0, 24);
        }
        this._colorShader.unbind();
        engine.setDepthFunctionToLessOrEqual();
        engine.setDepthWrite(true);
    };


    BABYLON.BoundingBoxRenderer.prototype.dispose = function () {
        this._colorShader.dispose();
        this._vb.dispose();
        this._scene.getEngine()._releaseBuffer(this._ib);
    };
})();