145 lines
3.3 KiB
JavaScript
145 lines
3.3 KiB
JavaScript
import { GPULoadOp, GPUStoreOp } from './constants.js';
|
|
import { Color, Mesh, BoxGeometry, BackSide } from 'three';
|
|
import { context, transformDirection, positionWorld, modelWorldMatrix, MeshBasicNodeMaterial } from 'three/nodes';
|
|
|
|
let _clearAlpha;
|
|
const _clearColor = new Color();
|
|
|
|
class WebGPUBackground {
|
|
|
|
constructor( renderer ) {
|
|
|
|
this.renderer = renderer;
|
|
|
|
this.boxMesh = null;
|
|
|
|
this.forceClear = false;
|
|
|
|
}
|
|
|
|
clear() {
|
|
|
|
this.forceClear = true;
|
|
|
|
}
|
|
|
|
update( renderList, scene ) {
|
|
|
|
const renderer = this.renderer;
|
|
const background = ( scene.isScene === true ) ? scene.backgroundNode || scene.background : null;
|
|
|
|
let forceClear = this.forceClear;
|
|
|
|
if ( background === null ) {
|
|
|
|
// no background settings, use clear color configuration from the renderer
|
|
|
|
_clearColor.copy( renderer._clearColor );
|
|
_clearAlpha = renderer._clearAlpha;
|
|
|
|
} else if ( background.isColor === true ) {
|
|
|
|
// background is an opaque color
|
|
|
|
_clearColor.copy( background );
|
|
_clearAlpha = 1;
|
|
forceClear = true;
|
|
|
|
} else if ( background.isNode === true ) {
|
|
|
|
_clearColor.copy( renderer._clearColor );
|
|
_clearAlpha = renderer._clearAlpha;
|
|
|
|
let boxMesh = this.boxMesh;
|
|
|
|
if ( boxMesh === null ) {
|
|
|
|
const colorNode = context( background, {
|
|
uvNode: transformDirection( positionWorld, modelWorldMatrix )
|
|
} );
|
|
|
|
const nodeMaterial = new MeshBasicNodeMaterial();
|
|
nodeMaterial.colorNode = colorNode;
|
|
nodeMaterial.side = BackSide;
|
|
nodeMaterial.depthTest = false;
|
|
nodeMaterial.depthWrite = false;
|
|
nodeMaterial.fog = false;
|
|
|
|
this.boxMesh = boxMesh = new Mesh( new BoxGeometry( 1, 1, 1 ), nodeMaterial );
|
|
|
|
boxMesh.onBeforeRender = function ( renderer, scene, camera ) {
|
|
|
|
this.matrixWorld.copyPosition( camera.matrixWorld );
|
|
|
|
};
|
|
|
|
}
|
|
|
|
renderList.unshift( boxMesh, boxMesh.geometry, boxMesh.material, 0, 0, null );
|
|
|
|
} else {
|
|
|
|
console.error( 'THREE.WebGPURenderer: Unsupported background configuration.', background );
|
|
|
|
}
|
|
|
|
// configure render pass descriptor
|
|
|
|
const renderPassDescriptor = renderer._renderPassDescriptor;
|
|
const colorAttachment = renderPassDescriptor.colorAttachments[ 0 ];
|
|
const depthStencilAttachment = renderPassDescriptor.depthStencilAttachment;
|
|
|
|
if ( renderer.autoClear === true || forceClear === true ) {
|
|
|
|
if ( renderer.autoClearColor === true ) {
|
|
|
|
_clearColor.multiplyScalar( _clearAlpha );
|
|
|
|
colorAttachment.clearValue = { r: _clearColor.r, g: _clearColor.g, b: _clearColor.b, a: _clearAlpha };
|
|
colorAttachment.loadOp = GPULoadOp.Clear;
|
|
colorAttachment.storeOp = GPUStoreOp.Store;
|
|
|
|
} else {
|
|
|
|
colorAttachment.loadOp = GPULoadOp.Load;
|
|
|
|
}
|
|
|
|
if ( renderer.autoClearDepth === true ) {
|
|
|
|
depthStencilAttachment.depthClearValue = renderer._clearDepth;
|
|
depthStencilAttachment.depthLoadOp = GPULoadOp.Clear;
|
|
|
|
} else {
|
|
|
|
depthStencilAttachment.depthLoadOp = GPULoadOp.Load;
|
|
|
|
}
|
|
|
|
if ( renderer.autoClearStencil === true ) {
|
|
|
|
depthStencilAttachment.stencilClearValue = renderer._clearStencil;
|
|
depthStencilAttachment.stencilLoadOp = GPULoadOp.Clear;
|
|
|
|
} else {
|
|
|
|
depthStencilAttachment.stencilLoadOp = GPULoadOp.Load;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
colorAttachment.loadOp = GPULoadOp.Load;
|
|
depthStencilAttachment.depthLoadOp = GPULoadOp.Load;
|
|
depthStencilAttachment.stencilLoadOp = GPULoadOp.Load;
|
|
|
|
}
|
|
|
|
this.forceClear = false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export default WebGPUBackground;
|