Add snapshow facility, record scene pass as a RenderBundle for reuse,

and reuse if present.
mrdoob · Oct 16, 2023 · 309f9db · 309f9db
1 parent f8ab8ac
commit 309f9db
Show file tree

Hide file tree

Showing 3 changed files with 74 additions and 0 deletions.
diff --git a/examples/jsm/renderers/common/Backend.js b/examples/jsm/renderers/common/Backend.js
@@ -80,6 +80,12 @@ class Backend {
 
 	destroyAttribute( attribute ) { }
 
+	// snapshot
+
+	createSnapshot() {}
+
+	destroySnapshot() {}
+
 	// canvas
 
 	updateSize() { }

diff --git a/examples/jsm/renderers/common/Renderer.js b/examples/jsm/renderers/common/Renderer.js
@@ -610,6 +610,18 @@ class Renderer {
 
 	}
 
+	createSnapshot() {
+
+		this.backend.createSnapshot();
+
+	}
+
+	destroySnapshot() {
+
+		this.backend.destroySnapshot();
+
+	}
+
 	dispose() {
 
 		this.info.dispose();

diff --git a/examples/jsm/renderers/webgpu/WebGPUBackend.js b/examples/jsm/renderers/webgpu/WebGPUBackend.js
@@ -66,6 +66,9 @@ class WebGPUBackend extends Backend {
 		this.textureUtils = new WebGPUTextureUtils( this );
 		this.occludedResolveCache = new Map();
 
+		this.createSnapshotBundle = false;
+		this.snapshotRenderBundles = null;
+
 	}
 
 	async init( renderer ) {
@@ -135,6 +138,12 @@ class WebGPUBackend extends Backend {
 
 	beginRender( renderContext ) {
 
+		if ( this.createSnapshotBundle ) {
+
+			return this._beginCreateSnapshot( renderContext );
+
+		}
+
 		const renderContextData = this.get( renderContext );
 
 		const device = this.device;
@@ -369,6 +378,25 @@ class WebGPUBackend extends Backend {
 
 		}
 
+		if ( this.createSnapshotBundle ) {
+
+			this.createSnapshotBundle = false;
+
+			this.snapshotRenderBundles = [ renderContextData.currentPass.finish() ];
+
+			// start real pass
+
+			this.beginRender( renderContext );
+
+
+		}
+
+		if ( this.snapshotRenderBundles !== null ) {
+
+			renderContextData.currentPass.executeBundles( this.snapshotRenderBundles );
+
+		}
+
 		renderContextData.currentPass.end();
 
 		if ( occlusionQueryCount > 0 ) {
@@ -606,6 +634,8 @@ class WebGPUBackend extends Backend {
 
 	draw( renderObject, info ) {
 
+		if ( this.snapshotRenderBundles !== null ) return;
+
 		const { object, geometry, context, pipeline } = renderObject;
 
 		const bindingsData = this.get( renderObject.getBindings() );
@@ -923,6 +953,18 @@ class WebGPUBackend extends Backend {
 
 	}
 
+	createSnapshot() {
+
+		this.createSnapshotBundle = true;
+
+	}
+
+	destroySnapshot() {
+
+		this.snapshotRenderBundles = null;
+
+	}
+
 	// canvas
 
 	updateSize() {
@@ -1080,6 +1122,20 @@ class WebGPUBackend extends Backend {
 
 	}
 
+	_beginCreateSnapshot( renderContext ) {
+
+		const renderContextData = this.get( renderContext );
+
+		renderContextData.currentPass = this.device.createRenderBundleEncoder( {
+			colorFormats: [ GPUTextureFormat.BGRA8Unorm ],
+			depthStencilFormat: this._getDepthBufferGPU( renderContext ).format,
+			sampleCount: this.parameters.sampleCount
+		} );
+
+		renderContextData.currentSets = { attributes: {} };
+
+	}
+
 }
 
 export default WebGPUBackend;