Add OIDN Denoiser.

example/index.js

@@ -29,6 +29,8 @@
 import { getScaledSettings } from './utils/getScaledSettings.js';
 import { LoaderElement } from './utils/LoaderElement.js';
 import { ParallelMeshBVHWorker } from 'three-mesh-bvh/src/workers/ParallelMeshBVHWorker.js';
+import path from 'path';
+import { on } from 'events';
 
 const envMaps = {
 	'Royal Esplanade': 'https://raw.githubusercontent.com/mrdoob/three.js/dev/examples/textures/equirectangular/royal_esplanade_1k.hdr',
@@ -108,6 +110,16 @@
 	floorRoughness: 0.2,
 	floorMetalness: 0.2,
 
+	// Denoiser
+	maxSamples: 6,
+	limitlessSamples: false,
+	enableDenoiser: false,
+	useAux: true,
+	cleanAux: true,
+	renderAux: '',
+	denoiserQuality: 'fast',
+	denoiserDebugging: false,
+
 	...getScaledSettings(),
 
 };
@@ -269,11 +281,70 @@
 
 	}
 
+	// Denoiser
+	if ( params.limitlessSamples ) pathTracer.maxSamples = Number.POSITIVE_INFINITY;
+	else pathTracer.maxSamples = params.maxSamples;
+	const renderScaleController = findController( 'renderScale' );
+	const oldEnableState = pathTracer.enableDenoiser;
+	pathTracer.enableDenoiser = params.enableDenoiser;
+	pathTracer.renderAux = params.renderAux;
+	pathTracer.denoiserDebugging = params.denoiserDebugging;
+	if ( oldEnableState !== params.enableDenoiser ) {
+
+		// force the renderscale to be 1 when using the denoiser
+		pathTracer.renderScale = 1;
+		params.renderScale = 1;
+		// set and disable the renderScale gui
+		if ( renderScaleController ) {
+
+			renderScaleController.disable();
+			renderScaleController.setValue( 1 );
+
+		}
+
+	} else {
+
+		if ( renderScaleController ) renderScaleController.enable();
+
+	}
+
 	pathTracer.updateMaterials();
 	pathTracer.updateEnvironment();
 
 }
 
+function hardDenoiserParamsChange() {
+
+	// When these params change it causes the denoiser to have to rebuild. So I split them out
+	if ( params.enableDenoiser ) {
+
+		pathTracer.useAux = params.useAux;
+		pathTracer.denoiserQuality = params.denoiserQuality;
+		pathTracer.cleanAux = params.cleanAux;
+
+
+	}
+
+}
+
+function findController( property ) {
+
+	if ( ! gui ) return null;
+
+	let targetController = null;
+	// get the controller by property
+	// biome-ignore lint/complexity/noForEach: <explanation>
+	gui.controllersRecursive().forEach( controller => {
+
+		if ( controller.property === property ) targetController = controller;
+
+	} );
+	return targetController;
+
+}
+
+
+
 function onHashChange() {
 
 	let hashModel = '';
@@ -316,7 +387,6 @@
 	orthoCamera.top = orthoHeight / 2;
 	orthoCamera.bottom = orthoHeight / - 2;
 	orthoCamera.updateProjectionMatrix();
-
 	pathTracer.updateCamera();
 
 }
@@ -340,6 +410,8 @@
 	const pathTracingFolder = gui.addFolder( 'Path Tracer' );
 	pathTracingFolder.add( params, 'enable' );
 	pathTracingFolder.add( params, 'pause' );
+	pathTracingFolder.add( params, 'maxSamples', 1, 300, 1 ).onChange( onParamsChange ).name( 'Max Samples' );
+	pathTracingFolder.add( params, 'limitlessSamples' ).onChange( onParamsChange ).name( 'Limitless Samples' );
 	pathTracingFolder.add( params, 'multipleImportanceSampling' ).onChange( onParamsChange );
 	pathTracingFolder.add( params, 'acesToneMapping' ).onChange( v => {
 
@@ -365,6 +437,15 @@
 	} );
 	pathTracingFolder.open();
 
+	const denoisingFolder = gui.addFolder( 'Denoising' );
+	denoisingFolder.add( params, 'enableDenoiser' ).name( 'OIDN' ).onChange( onParamsChange );
+	denoisingFolder.add( params, 'denoiserQuality', [ 'fast', 'balanced' ] ).onChange( hardDenoiserParamsChange ).name( 'Quality' );
+	denoisingFolder.add( params, 'useAux' ).onChange( hardDenoiserParamsChange ).name( 'Use Aux Inputs' );
+	denoisingFolder.add( params, 'cleanAux' ).onChange( hardDenoiserParamsChange ).name( 'Aux Inputs 100% Clean' );
+	denoisingFolder.add( params, 'renderAux', { 'Denoised': '', 'Albedo': 'albedo', 'Normal': 'normal' } ).name( 'Render Output' ).onChange( onParamsChange );
+	denoisingFolder.add( params, 'denoiserDebugging' ).name( 'Debugging' ).onChange( onParamsChange );
+	denoisingFolder.open();
+
 	const environmentFolder = gui.addFolder( 'environment' );
 	environmentFolder.add( params, 'envMap', envMaps ).name( 'map' ).onChange( updateEnvMap );
 	environmentFolder.add( params, 'environmentIntensity', 0.0, 10.0 ).onChange( onParamsChange ).name( 'intensity' );
@@ -436,7 +517,6 @@
 
 	controls.object = activeCamera;
 	controls.update();
-
 	pathTracer.setCamera( activeCamera );
 
 }