Examples: postprocessing material ao: show diffuse pass on both sides

Author: Rabbid76

examples/jsm/materials/MeshPostProcessingMaterial.js

@@ -35,8 +35,10 @@ class MeshPostProcessingMaterial extends MeshPhysicalMaterial {
 		super( parameters );
 
 		this.onBeforeCompile = this._onBeforeCompile;
+		this.customProgramCacheKey = this._customProgramCacheKey;
 		this._aoPassMap = aoPassMap;
 		this.aoPassMapScale = aoPassMapScale;
+		this._shader = null;
 
 	}
 
@@ -50,12 +52,21 @@ class MeshPostProcessingMaterial extends MeshPhysicalMaterial {
 
 		this._aoPassMap = aoPassMap;
 		this.needsUpdate = true;
+		this._setUniforms();
+
+	}
+
+	_customProgramCacheKey() {
+
+		return this._aoPassMap !== undefined && this._aoPassMap !== null ? 'aoPassMap' : '';
 
 	}
 
 	_onBeforeCompile( shader ) {
 
-		if ( this._aoPassMap !== undefined ) {
+		this._shader = shader;
+
+		if ( this._aoPassMap !== undefined && this._aoPassMap !== null ) {
 
 			shader.fragmentShader = shader.fragmentShader.replace(
 				'#include <aomap_pars_fragment>',
@@ -65,8 +76,19 @@ class MeshPostProcessingMaterial extends MeshPhysicalMaterial {
 				'#include <aomap_fragment>',
 				aomap_fragment_replacement
 			);
-			shader.uniforms.tAoPassMap = { value: this._aoPassMap };
-			shader.uniforms.aoPassMapScale = { value: this.aoPassMapScale };
+
+		}
+
+		this._setUniforms();
+
+	}
+
+	_setUniforms() {
+
+		if ( this._shader ) {
+
+			this._shader.uniforms.tAoPassMap = { value: this._aoPassMap };
+			this._shader.uniforms.aoPassMapScale = { value: this.aoPassMapScale };
 
 		}
 

examples/webgl_postprocessing_material_ao.html

@@ -215,14 +215,16 @@
 				const gui = new GUI();
 				gui.add( gtaoPasses[ 0 ], 'output', {
 					'Default': GTAOPass.OUTPUT.Default,
-					'Diffuse (right side)': GTAOPass.OUTPUT.Diffuse,
+					'Diffuse': GTAOPass.OUTPUT.Diffuse,
 					'AO': GTAOPass.OUTPUT.Denoise,
 
 				} ).onChange( function ( value ) {
 
 					gtaoPasses[ 0 ].output = value;
 					gtaoPasses[ 1 ].output = value === GTAOPass.OUTPUT.Denoise ? value : GTAOPass.OUTPUT.Off;
-					renderPasses[ 1 ].enabled = value !== GTAOPass.OUTPUT.Denoise;
+					renderPasses.forEach( pass => pass.enabled = value !== GTAOPass.OUTPUT.Denoise );
+					groundMaterials[ 1 ].aoPassMap = value === GTAOPass.OUTPUT.Default ? gtaoPasses[ 1 ].gtaoMap : null;
+					objectMaterials[ 1 ].aoPassMap = value === GTAOPass.OUTPUT.Default ? gtaoPasses[ 1 ].gtaoMap : null;
 
 				} );
 				gui.add( sceneParameters, 'envMapIntensity' ).min( 0 ).max( 1 ).step( 0.01 ).onChange( () => {