Babylon.js/src/Materials/babylon.imageProcessingConfiguration.ts

module BABYLON {
    /**
     * Interface to follow in your material defines to integrate easily the
     * Image proccessing functions.
     */
    export interface IImageProcessingConfigurationDefines {
        IMAGEPROCESSING: boolean;
        VIGNETTE: boolean;
        VIGNETTEBLENDMODEMULTIPLY: boolean;
        VIGNETTEBLENDMODEOPAQUE: boolean;
        TONEMAPPING: boolean;
        CONTRAST: boolean;
        EXPOSURE: boolean;
        COLORCURVES: boolean;
        COLORGRADING: boolean;
        COLORGRADING3D: boolean;
        SAMPLER3DGREENDEPTH: boolean;
        SAMPLER3DBGRMAP: boolean;
        IMAGEPROCESSINGPOSTPROCESS: boolean;
        /** 
         * If the grain should be performed in the image processing shader.
         */
        GRAIN: boolean;
    }

    /**
     * This groups together the common properties used for image processing either in direct forward pass
     * or through post processing effect depending on the use of the image processing pipeline in your scene 
     * or not.
     */
    export class ImageProcessingConfiguration {

        /**
         * Color curves setup used in the effect if colorCurvesEnabled is set to true 
         */
        @serializeAsColorCurves()
        public colorCurves: Nullable<ColorCurves> = new ColorCurves();

        @serialize()
        private _colorCurvesEnabled = false;
        /**
         * Gets wether the color curves effect is enabled.
         */
        public get colorCurvesEnabled(): boolean {
            return this._colorCurvesEnabled;
        }
        /**
         * Sets wether the color curves effect is enabled.
         */
        public set colorCurvesEnabled(value: boolean) {
            if (this._colorCurvesEnabled === value) {
                return;
            }

            this._colorCurvesEnabled = value;
            this._updateParameters();
        }

        /**
         * Color grading LUT texture used in the effect if colorGradingEnabled is set to true 
         */
        @serializeAsTexture()
        public colorGradingTexture: Nullable<BaseTexture>;

        @serialize()
        private _colorGradingEnabled = false;
        /**
         * Gets wether the color grading effect is enabled.
         */
        public get colorGradingEnabled(): boolean {
            return this._colorGradingEnabled;
        }
        /**
         * Sets wether the color grading effect is enabled.
         */
        public set colorGradingEnabled(value: boolean) {
            if (this._colorGradingEnabled === value) {
                return;
            }

            this._colorGradingEnabled = value;
            this._updateParameters();
        }

        @serialize()
        private _colorGradingWithGreenDepth = true;
        /**
         * Gets wether the color grading effect is using a green depth for the 3d Texture.
         */
        public get colorGradingWithGreenDepth(): boolean {
            return this._colorGradingWithGreenDepth;
        }
        /**
         * Sets wether the color grading effect is using a green depth for the 3d Texture.
         */
        public set colorGradingWithGreenDepth(value: boolean) {
            if (this._colorGradingWithGreenDepth === value) {
                return;
            }

            this._colorGradingWithGreenDepth = value;
            this._updateParameters();
        }

        @serialize()
        private _colorGradingBGR = true;
        /**
         * Gets wether the color grading texture contains BGR values.
         */
        public get colorGradingBGR(): boolean {
            return this._colorGradingBGR;
        }
        /**
         * Sets wether the color grading texture contains BGR values.
         */
        public set colorGradingBGR(value: boolean) {
            if (this._colorGradingBGR === value) {
                return;
            }

            this._colorGradingBGR = value;
            this._updateParameters();
        }

        @serialize()
        public _exposure = 1.0;
        /**
         * Gets the Exposure used in the effect.
         */
        public get exposure(): number {
            return this._exposure;
        }
        /**
         * Sets the Exposure used in the effect.
         */
        public set exposure(value: number) {
            if (this._exposure === value) {
                return;
            }

            this._exposure = value;
            this._updateParameters();
        }

        @serialize()
        private _toneMappingEnabled = false;
        /**
         * Gets wether the tone mapping effect is enabled.
         */
        public get toneMappingEnabled(): boolean {
            return this._toneMappingEnabled;
        }
        /**
         * Sets wether the tone mapping effect is enabled.
         */
        public set toneMappingEnabled(value: boolean) {
            if (this._toneMappingEnabled === value) {
                return;
            }

            this._toneMappingEnabled = value;
            this._updateParameters();
        }

        @serialize()
        protected _contrast = 1.0;
        /**
         * Gets the contrast used in the effect.
         */
        public get contrast(): number {
            return this._contrast;
        }
        /**
         * Sets the contrast used in the effect.
         */
        public set contrast(value: number) {
            if (this._contrast === value) {
                return;
            }

            this._contrast = value;
            this._updateParameters();
        }

        /**
         * Vignette stretch size.
         */
        @serialize()
        public vignetteStretch = 0;

        /**
         * Vignette centre X Offset.
         */
        @serialize()
        public vignetteCentreX = 0;

        /**
         * Vignette centre Y Offset.
         */
        @serialize()
        public vignetteCentreY = 0;

        /**
         * Vignette weight or intensity of the vignette effect.
         */
        @serialize()
        public vignetteWeight = 1.5;

        /**
         * Color of the vignette applied on the screen through the chosen blend mode (vignetteBlendMode)
         * if vignetteEnabled is set to true.
         */
        @serializeAsColor4()
        public vignetteColor: Color4 = new Color4(0, 0, 0, 0);

        /**
         * Camera field of view used by the Vignette effect.
         */
        @serialize()
        public vignetteCameraFov = 0.5;

        @serialize()
        private _grainVarianceAmount = 0;
        /**
         * Amount of grain to be applied by the grain effect.
         */
        public get grainVarianceAmount(): number {
            return this._grainVarianceAmount;
        }
        public set grainVarianceAmount(value: number) {
            if (this._grainVarianceAmount === value) {
                return;
            }

            this._grainVarianceAmount = value;
            this._updateParameters();
        }

        @serialize()
        private _grainAnimated = false;

        /**
         * If the grain effect should be animated.
         */
        public get grainAnimated(): boolean {
            return this._grainAnimated;
        }
        public set grainAnimated(value: boolean) {
            if (this._grainAnimated === value) {
                return;
            }

            this._grainAnimated = value;
            this._updateParameters();
        }

        @serialize()
        private _vignetteBlendMode = ImageProcessingConfiguration.VIGNETTEMODE_MULTIPLY;
        /**
         * Gets the vignette blend mode allowing different kind of effect.
         */
        public get vignetteBlendMode(): number {
            return this._vignetteBlendMode;
        }
        /**
         * Sets the vignette blend mode allowing different kind of effect.
         */
        public set vignetteBlendMode(value: number) {
            if (this._vignetteBlendMode === value) {
                return;
            }

            this._vignetteBlendMode = value;
            this._updateParameters();
        }

        @serialize()
        private _vignetteEnabled = false;
        /**
         * Gets wether the vignette effect is enabled.
         */
        public get vignetteEnabled(): boolean {
            return this._vignetteEnabled;
        }
        /**
         * Sets wether the vignette effect is enabled.
         */
        public set vignetteEnabled(value: boolean) {
            if (this._vignetteEnabled === value) {
                return;
            }

            this._vignetteEnabled = value;
            this._updateParameters();
        }

        @serialize()
        private _applyByPostProcess = false;
        /**
         * Gets wether the image processing is applied through a post process or not.
         */
        public get applyByPostProcess(): boolean {
            return this._applyByPostProcess;
        }
        /**
         * Sets wether the image processing is applied through a post process or not.
         */
        public set applyByPostProcess(value: boolean) {
            if (this._applyByPostProcess === value) {
                return;
            }

            this._applyByPostProcess = value;
            this._updateParameters();
        }

        @serialize()
        private _isEnabled = true;
        /**
         * Gets wether the image processing is enabled or not.
         */
        public get isEnabled(): boolean {
            return this._isEnabled;
        }
        /**
         * Sets wether the image processing is enabled or not.
         */
        public set isEnabled(value: boolean) {
            if (this._isEnabled === value) {
                return;
            }

            this._isEnabled = value;
            this._updateParameters();
        }

        /**
        * An event triggered when the configuration changes and requires Shader to Update some parameters.
        * @type {BABYLON.Observable}
        */
        public onUpdateParameters = new Observable<ImageProcessingConfiguration>();

        /**
         * Method called each time the image processing information changes requires to recompile the effect.
         */
        protected _updateParameters(): void {
            this.onUpdateParameters.notifyObservers(this);
        }

        public getClassName(): string {
            return "ImageProcessingConfiguration";
        }

        /**
         * Prepare the list of uniforms associated with the Image Processing effects.
         * @param uniformsList The list of uniforms used in the effect
         * @param defines the list of defines currently in use
         */
        public static PrepareUniforms(uniforms: string[], defines: IImageProcessingConfigurationDefines): void {
            if (defines.EXPOSURE) {
                uniforms.push("exposureLinear");
            }
            if (defines.CONTRAST) {
                uniforms.push("contrast");
            }
            if (defines.COLORGRADING) {
                uniforms.push("colorTransformSettings");
            }
            if (defines.VIGNETTE) {
                uniforms.push("vInverseScreenSize");
                uniforms.push("vignetteSettings1");
                uniforms.push("vignetteSettings2");
            }
            if (defines.COLORCURVES) {
                ColorCurves.PrepareUniforms(uniforms);
            }
            if (defines.GRAIN){
                uniforms.push("grainVarianceAmount");
                uniforms.push("grainAnimatedSeed");
            }
        }

        /**
         * Prepare the list of samplers associated with the Image Processing effects.
         * @param uniformsList The list of uniforms used in the effect
         * @param defines the list of defines currently in use
         */
        public static PrepareSamplers(samplersList: string[], defines: IImageProcessingConfigurationDefines): void {
            if (defines.COLORGRADING) {
                samplersList.push("txColorTransform");
            }
        }

        /**
         * Prepare the list of defines associated to the shader.
         * @param defines the list of defines to complete
         */
        public prepareDefines(defines: IImageProcessingConfigurationDefines, forPostProcess: boolean = false): void {
            if (forPostProcess !== this.applyByPostProcess || !this._isEnabled) {
                defines.VIGNETTE = false;
                defines.TONEMAPPING = false;
                defines.CONTRAST = false;
                defines.EXPOSURE = false;
                defines.COLORCURVES = false;
                defines.COLORGRADING = false;
                defines.COLORGRADING3D = false;
                defines.IMAGEPROCESSING = false;
                defines.IMAGEPROCESSINGPOSTPROCESS = this.applyByPostProcess && this._isEnabled;
                return;
            }

            defines.VIGNETTE = this.vignetteEnabled;
            defines.VIGNETTEBLENDMODEMULTIPLY = (this.vignetteBlendMode === ImageProcessingConfiguration._VIGNETTEMODE_MULTIPLY);
            defines.VIGNETTEBLENDMODEOPAQUE = !defines.VIGNETTEBLENDMODEMULTIPLY;
            defines.TONEMAPPING = this.toneMappingEnabled;
            defines.CONTRAST = (this.contrast !== 1.0);
            defines.EXPOSURE = (this.exposure !== 1.0);
            defines.COLORCURVES = (this.colorCurvesEnabled && !!this.colorCurves);
            defines.COLORGRADING = (this.colorGradingEnabled && !!this.colorGradingTexture);
            if (defines.COLORGRADING) {
                defines.COLORGRADING3D = this.colorGradingTexture!.is3D;
            } else {
                defines.COLORGRADING3D = false;
            }
            defines.SAMPLER3DGREENDEPTH = this.colorGradingWithGreenDepth;
            defines.SAMPLER3DBGRMAP = this.colorGradingBGR;
            defines.IMAGEPROCESSINGPOSTPROCESS = this.applyByPostProcess;
            defines.IMAGEPROCESSING = defines.VIGNETTE || defines.TONEMAPPING || defines.CONTRAST || defines.EXPOSURE || defines.COLORCURVES || defines.COLORGRADING;
            defines.GRAIN = (this.grainVarianceAmount != 0.0);
        }

        /**
         * Returns true if all the image processing information are ready.
         */
        public isReady() {
            // Color Grading texure can not be none blocking.
            return !this.colorGradingEnabled || !this.colorGradingTexture || this.colorGradingTexture.isReady();
        }

        /**
         * Binds the image processing to the shader.
         * @param effect The effect to bind to
         */
        public bind(effect: Effect, aspectRatio = 1): void {
            // Color Curves
            if (this._colorCurvesEnabled && this.colorCurves) {
                ColorCurves.Bind(this.colorCurves, effect);
            }

            // Vignette
            if (this._vignetteEnabled) {
                var inverseWidth = 1 / effect.getEngine().getRenderWidth();
                var inverseHeight = 1 / effect.getEngine().getRenderHeight();
                effect.setFloat2("vInverseScreenSize", inverseWidth, inverseHeight);

                let vignetteScaleY = Math.tan(this.vignetteCameraFov * 0.5);
                let vignetteScaleX = vignetteScaleY * aspectRatio;

                let vignetteScaleGeometricMean = Math.sqrt(vignetteScaleX * vignetteScaleY);
                vignetteScaleX = Tools.Mix(vignetteScaleX, vignetteScaleGeometricMean, this.vignetteStretch);
                vignetteScaleY = Tools.Mix(vignetteScaleY, vignetteScaleGeometricMean, this.vignetteStretch);

                effect.setFloat4("vignetteSettings1", vignetteScaleX, vignetteScaleY, -vignetteScaleX * this.vignetteCentreX, -vignetteScaleY * this.vignetteCentreY);

                let vignettePower = -2.0 * this.vignetteWeight;
                effect.setFloat4("vignetteSettings2", this.vignetteColor.r, this.vignetteColor.g, this.vignetteColor.b, vignettePower);
            }

            // Exposure
            effect.setFloat("exposureLinear", this.exposure);

            // Contrast
            effect.setFloat("contrast", this.contrast);

            // Color transform settings
            if (this.colorGradingTexture) {
                effect.setTexture("txColorTransform", this.colorGradingTexture);
                let textureSize = this.colorGradingTexture.getSize().height;

                effect.setFloat4("colorTransformSettings",
                    (textureSize - 1) / textureSize, // textureScale
                    0.5 / textureSize, // textureOffset
                    textureSize, // textureSize
                    this.colorGradingTexture.level // weight
                );
            }

            effect.setFloat("grainVarianceAmount", this.grainVarianceAmount);
            effect.setFloat("grainAnimatedSeed", this.grainAnimated ? Math.random() : 1);
        }

        /**
         * Clones the current image processing instance.
         * @return The cloned image processing
         */
        public clone(): ImageProcessingConfiguration {
            return SerializationHelper.Clone(() => new ImageProcessingConfiguration(), this);
        }

        /**
         * Serializes the current image processing instance to a json representation.
         * @return a JSON representation
         */
        public serialize(): any {
            return SerializationHelper.Serialize(this);
        }

        /**
         * Parses the image processing from a json representation.
         * @param source the JSON source to parse
         * @return The parsed image processing
         */
        public static Parse(source: any): ImageProcessingConfiguration {
            return SerializationHelper.Parse(() => new ImageProcessingConfiguration(), source, null, null);
        }

        // Static constants associated to the image processing.
        private static _VIGNETTEMODE_MULTIPLY = 0;
        private static _VIGNETTEMODE_OPAQUE = 1;

        /**
         * Used to apply the vignette as a mix with the pixel color.
         */
        public static get VIGNETTEMODE_MULTIPLY(): number {
            return this._VIGNETTEMODE_MULTIPLY;
        }

        /**
         * Used to apply the vignette as a replacement of the pixel color.
         */
        public static get VIGNETTEMODE_OPAQUE(): number {
            return this._VIGNETTEMODE_OPAQUE;
        }
    }
}