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

mrdoob · Dec 31, 2023 · 20e5e00 · 20e5e00
1 parent 7819ad8
commit 20e5e00
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/examples/jsm/materials/MeshPostProcessingMaterial.js b/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._mapVersion = 0;
 
 	}
 
@@ -50,12 +52,19 @@ class MeshPostProcessingMaterial extends MeshPhysicalMaterial {
 
 		this._aoPassMap = aoPassMap;
 		this.needsUpdate = true;
+		if ( aoPassMap !== undefined && aoPassMap !== null ) this._mapVersion ++;
+
+	}
+
+	_customProgramCacheKey() {
+
+		return this._aoPassMap !== undefined && this._aoPassMap !== null ? this._aoPassMap.uuid + this._mapVersion : '';
 
 	}
 
 	_onBeforeCompile( shader ) {
 
-		if ( this._aoPassMap !== undefined ) {
+		if ( this._aoPassMap !== undefined && this._aoPassMap !== null ) {
 
 			shader.fragmentShader = shader.fragmentShader.replace(
 				'#include <aomap_pars_fragment>',

diff --git a/examples/webgl_postprocessing_material_ao.html b/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( () => {