nanogl-pbr/lighting/SpotLightModel.js

pbr material for nanogl

import { ShadowMappedLightModel } from "./AbstractLightModel";
import LightType from "./LightType";
export default class SpotLightModel extends ShadowMappedLightModel {
    constructor(code, preCode) {
        super(code, preCode);
        this.type = LightType.SPOT;
        this._positions = null;
        this._directions = null;
        this._colors = null;
        this._attenuation = null;
    }
    genCodePerLights(light, index, shadowIndex) {
        var o = {
            index: index,
            shadowIndex: shadowIndex,
            infinite: light.radius <= 0,
        };
        return this.codeTemplate(o);
    }
    allocate(n) {
        if (this._colors === null || this._colors.length / 4 !== n) {
            this._positions = new Float32Array(n * 3);
            this._directions = new Float32Array(n * 3);
            this._colors = new Float32Array(n * 4);
            this._attenuation = new Float32Array(n * 4);
        }
    }
    prepare(gl, model) {
        const lights = this.lights;
        this.allocate(lights.length);
        for (var i = 0; i < lights.length; i++) {
            var l = lights[i];
            this._positions.set(l._wposition, i * 3);
            this._directions.set(l._wdir, i * 3);
            this._colors.set(l._color, i * 4);
            this._attenuation.set(l._attenuationData, i * 4);
            this._colors[i * 4 + 3] = l.iblShadowing;
            if (l.castShadows) {
                l.initShadowmap(gl);
                var shIndex = model.shadowChunk.addLight(l);
                if (this.shadowIndices[i] !== shIndex) {
                    this.invalidateCode();
                }
                this.shadowIndices[i] = shIndex;
            }
            else {
                this.shadowIndices[i] = -1;
            }
        }
        this._invalid = true;
    }
    setup(prg) {
        if (this.lights.length > 0) {
            super.setup(prg);
            prg.uLSpotPositions(this._positions);
            prg.uLSpotDirections(this._directions);
            prg.uLSpotColors(this._colors);
            prg.uLSpotAttenuation(this._attenuation);
            this._invalid = false;
        }
    }
}