| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903 |
- // debugger tools
- // import 'https://greggman.github.io/webgpu-avoid-redundant-state-setting/webgpu-check-redundant-state-setting.js';
- import { GPUFeatureName, GPULoadOp, GPUStoreOp, GPUIndexFormat, GPUTextureViewDimension, GPUFeatureMap, GPUShaderStage } from './utils/WebGPUConstants.js';
- import WGSLNodeBuilder from './nodes/WGSLNodeBuilder.js';
- import Backend from '../common/Backend.js';
- import WebGPUUtils, { submit } from './utils/WebGPUUtils.js';
- import WebGPUAttributeUtils from './utils/WebGPUAttributeUtils.js';
- import WebGPUBindingUtils from './utils/WebGPUBindingUtils.js';
- import WebGPUCapabilities from './utils/WebGPUCapabilities.js';
- import WebGPUPipelineUtils from './utils/WebGPUPipelineUtils.js';
- import WebGPUTextureUtils from './utils/WebGPUTextureUtils.js';
- import { WebGPUCoordinateSystem, TimestampQuery, REVISION, HalfFloatType, Compatibility } from '../../constants.js';
- import WebGPUTimestampQueryPool from './utils/WebGPUTimestampQueryPool.js';
- import { error } from '../../utils.js';
- import GPUBufferDescriptor from './descriptors/GPUBufferDescriptor.js';
- import GPUCommandEncoderDescriptor from './descriptors/GPUCommandEncoderDescriptor.js';
- import GPUComputePassDescriptor from './descriptors/GPUComputePassDescriptor.js';
- import GPUQuerySetDescriptor from './descriptors/GPUQuerySetDescriptor.js';
- import GPUShaderModuleDescriptor from './descriptors/GPUShaderModuleDescriptor.js';
- import GPURenderPassColorAttachment from './descriptors/GPURenderPassColorAttachment.js';
- import GPURenderPassDepthStencilAttachment from './descriptors/GPURenderPassDepthStencilAttachment.js';
- import GPURenderPassDescriptor from './descriptors/GPURenderPassDescriptor.js';
- import GPURenderPassTimestampWrites from './descriptors/GPURenderPassTimestampWrites.js';
- import GPUTexelCopyTextureInfo from './descriptors/GPUTexelCopyTextureInfo.js';
- import GPUTextureViewDescriptor from './descriptors/GPUTextureViewDescriptor.js';
- import GPUExtent3D from './descriptors/GPUExtent3D.js';
- const _clearValue = { r: 0, g: 0, b: 0, a: 1 };
- const _bufferDescriptor = new GPUBufferDescriptor();
- const _commandEncoderDescriptor = new GPUCommandEncoderDescriptor();
- const _computePassDescriptor = new GPUComputePassDescriptor();
- const _querySetDescriptor = new GPUQuerySetDescriptor();
- const _shaderModuleDescriptor = new GPUShaderModuleDescriptor();
- const _renderPassTimestampWrites = new GPURenderPassTimestampWrites();
- const _texelCopyTextureInfoSrc = new GPUTexelCopyTextureInfo();
- const _texelCopyTextureInfoDst = new GPUTexelCopyTextureInfo();
- const _viewDescriptor = new GPUTextureViewDescriptor();
- const _extent3D = new GPUExtent3D();
- /**
- * A backend implementation targeting WebGPU.
- *
- * @private
- * @augments Backend
- */
- class WebGPUBackend extends Backend {
- /**
- * WebGPUBackend options.
- *
- * @typedef {Object} WebGPUBackend~Options
- * @property {boolean} [logarithmicDepthBuffer=false] - Whether logarithmic depth buffer is enabled or not.
- * @property {boolean} [reversedDepthBuffer=false] - Whether reversed depth buffer is enabled or not.
- * @property {boolean} [alpha=true] - Whether the default framebuffer (which represents the final contents of the canvas) should be transparent or opaque.
- * @property {boolean} [depth=true] - Whether the default framebuffer should have a depth buffer or not.
- * @property {boolean} [stencil=false] - Whether the default framebuffer should have a stencil buffer or not.
- * @property {boolean} [antialias=false] - Whether MSAA as the default anti-aliasing should be enabled or not.
- * @property {number} [samples=0] - When `antialias` is `true`, `4` samples are used by default. Set this parameter to any other integer value than 0 to overwrite the default.
- * @property {boolean} [forceWebGL=false] - If set to `true`, the renderer uses a WebGL 2 backend no matter if WebGPU is supported or not.
- * @property {boolean} [trackTimestamp=false] - Whether to track timestamps with a Timestamp Query API or not.
- * @property {string} [powerPreference=undefined] - The power preference.
- * @property {Object} [requiredLimits=undefined] - Specifies the limits that are required by the device request. The request will fail if the adapter cannot provide these limits.
- * @property {GPUDevice} [device=undefined] - If there is an existing GPU device on app level, it can be passed to the renderer as a parameter.
- * @property {number} [outputType=undefined] - Texture type for output to canvas. By default, device's preferred format is used; other formats may incur overhead.
- */
- /**
- * Constructs a new WebGPU backend.
- *
- * @param {WebGPUBackend~Options} [parameters] - The configuration parameter.
- */
- constructor( parameters = {} ) {
- super( parameters );
- /**
- * This flag can be used for type testing.
- *
- * @type {boolean}
- * @readonly
- * @default true
- */
- this.isWebGPUBackend = true;
- // some parameters require default values other than "undefined"
- this.parameters.alpha = ( parameters.alpha === undefined ) ? true : parameters.alpha;
- this.parameters.requiredLimits = ( parameters.requiredLimits === undefined ) ? {} : parameters.requiredLimits;
- /**
- * Indicates whether the backend is in WebGPU compatibility mode or not.
- * The backend must be initialized before the property can be evaluated.
- *
- * @type {?boolean}
- * @readonly
- * @default null
- */
- this.compatibilityMode = null;
- /**
- * A reference to the device.
- *
- * @type {?GPUDevice}
- * @default null
- */
- this.device = null;
- /**
- * A reference to the default render pass descriptor.
- *
- * @type {?Object}
- * @default null
- */
- this.defaultRenderPassdescriptor = null;
- /**
- * A reference to a backend module holding common utility functions.
- *
- * @type {WebGPUUtils}
- */
- this.utils = new WebGPUUtils( this );
- /**
- * A reference to a backend module holding shader attribute-related
- * utility functions.
- *
- * @type {WebGPUAttributeUtils}
- */
- this.attributeUtils = new WebGPUAttributeUtils( this );
- /**
- * A reference to a backend module holding shader binding-related
- * utility functions.
- *
- * @type {WebGPUBindingUtils}
- */
- this.bindingUtils = new WebGPUBindingUtils( this );
- /**
- * A reference to a backend module holding device capability related
- * utility functions.
- *
- * @type {WebGPUCapabilities}
- */
- this.capabilities = new WebGPUCapabilities( this );
- /**
- * A reference to a backend module holding shader pipeline-related
- * utility functions.
- *
- * @type {WebGPUPipelineUtils}
- */
- this.pipelineUtils = new WebGPUPipelineUtils( this );
- /**
- * A reference to a backend module holding shader texture-related
- * utility functions.
- *
- * @type {WebGPUTextureUtils}
- */
- this.textureUtils = new WebGPUTextureUtils( this );
- /**
- * A map that manages the resolve buffers for occlusion queries.
- *
- * @type {Map<number,GPUBuffer>}
- */
- this.occludedResolveCache = new Map();
- // compatibility checks
- const compatibilityTextureCompare = typeof navigator === 'undefined' ? true : /Android/.test( navigator.userAgent ) === false;
- /**
- * A map of compatibility checks.
- *
- * @type {Object}
- */
- this._compatibility = {
- [ Compatibility.TEXTURE_COMPARE ]: compatibilityTextureCompare
- };
- }
- /**
- * Initializes the backend so it is ready for usage.
- *
- * @async
- * @param {Renderer} renderer - The renderer.
- * @return {Promise} A Promise that resolves when the backend has been initialized.
- */
- async init( renderer ) {
- await super.init( renderer );
- //
- const parameters = this.parameters;
- // create the device if it is not passed with parameters
- let device;
- if ( parameters.device === undefined ) {
- const adapterOptions = {
- powerPreference: parameters.powerPreference,
- featureLevel: 'compatibility',
- xrCompatible: renderer.xr.enabled
- };
- const adapter = ( typeof navigator !== 'undefined' ) ? await navigator.gpu.requestAdapter( adapterOptions ) : null;
- if ( adapter === null ) {
- throw new Error( 'THREE.WebGPUBackend: Unable to create WebGPU adapter.' );
- }
- // feature support
- const features = Object.values( GPUFeatureName );
- const supportedFeatures = [];
- for ( const name of features ) {
- if ( adapter.features.has( name ) ) {
- supportedFeatures.push( name );
- }
- }
- const deviceDescriptor = {
- requiredFeatures: supportedFeatures,
- requiredLimits: parameters.requiredLimits
- };
- device = await adapter.requestDevice( deviceDescriptor );
- } else {
- device = parameters.device;
- }
- this.compatibilityMode = ! device.features.has( 'core-features-and-limits' );
- if ( this.compatibilityMode ) {
- renderer._samples = 0;
- }
- device.lost.then( ( info ) => {
- if ( info.reason === 'destroyed' ) return;
- const deviceLossInfo = {
- api: 'WebGPU',
- message: info.message || 'Unknown reason',
- reason: info.reason || null,
- originalEvent: info
- };
- renderer.onDeviceLost( deviceLossInfo );
- } );
- device.onuncapturederror = ( event ) => {
- const gpuError = event.error;
- const type = gpuError && gpuError.constructor ? gpuError.constructor.name : 'GPUError';
- const message = ( gpuError && gpuError.message ) || 'Unknown uncaptured GPU error';
- renderer.onError( {
- api: 'WebGPU',
- type,
- message,
- originalEvent: event
- } );
- };
- this.device = device;
- this.trackTimestamp = this.trackTimestamp && this.hasFeature( GPUFeatureName.TimestampQuery );
- this.updateSize();
- }
- /**
- * Registers external GPU textures from `XRGPUBinding` for use in rendering.
- *
- * @param {RenderTarget} renderTarget - The render target to register the textures for.
- * @param {GPUTexture} colorTexture - The shared XR color GPUTexture.
- * @param {?Array<Object>} [viewDescriptors=null] - Optional view descriptors, one per XR view.
- */
- setXRRenderTargetTextures( renderTarget, colorTexture, viewDescriptors = null ) {
- this.set( renderTarget.texture, {
- texture: colorTexture,
- format: colorTexture.format,
- externalTexture: true,
- xrViewDescriptors: viewDescriptors,
- initialized: true
- } );
- }
- /**
- * A reference to the context.
- *
- * @type {?GPUCanvasContext}
- * @default null
- */
- get context() {
- const canvasTarget = this.renderer.getCanvasTarget();
- const canvasData = this.get( canvasTarget );
- let context = canvasData.context;
- if ( context === undefined ) {
- const parameters = this.parameters;
- if ( canvasTarget.isDefaultCanvasTarget === true && parameters.context !== undefined ) {
- context = parameters.context;
- } else {
- context = canvasTarget.domElement.getContext( 'webgpu' );
- }
- // OffscreenCanvas does not have setAttribute, see #22811
- if ( 'setAttribute' in canvasTarget.domElement ) canvasTarget.domElement.setAttribute( 'data-engine', `three.js r${ REVISION } webgpu` );
- const alphaMode = parameters.alpha ? 'premultiplied' : 'opaque';
- const toneMappingMode = parameters.outputType === HalfFloatType ? 'extended' : 'standard';
- context.configure( {
- device: this.device,
- format: this.utils.getPreferredCanvasFormat(),
- usage: GPUTextureUsage.RENDER_ATTACHMENT | GPUTextureUsage.COPY_SRC,
- alphaMode: alphaMode,
- toneMapping: {
- mode: toneMappingMode
- }
- } );
- canvasData.context = context;
- }
- return context;
- }
- /**
- * The coordinate system of the backend.
- *
- * @type {number}
- * @readonly
- */
- get coordinateSystem() {
- return WebGPUCoordinateSystem;
- }
- /**
- * Whether the backend supports query timestamps or not.
- *
- * @type {boolean}
- * @readonly
- */
- get hasTimestamp() {
- return true;
- }
- /**
- * This method performs a readback operation by moving buffer data from
- * a storage buffer attribute from the GPU to the CPU. ReadbackBuffer can
- * be used to retain and reuse handles to the intermediate buffers and prevent
- * new allocation.
- *
- * @async
- * @param {BufferAttribute} attribute - The storage buffer attribute to read frm.
- * @param {number} count - The offset from which to start reading the
- * @param {number} offset - The storage buffer attribute.
- * @param {ReadbackBuffer|ArrayBuffer} target - The storage buffer attribute.
- * @return {Promise<ArrayBuffer|ReadbackBuffer>} A promise that resolves with the buffer data when the data are ready.
- */
- async getArrayBufferAsync( attribute, target = null, offset = 0, count = - 1 ) {
- return await this.attributeUtils.getArrayBufferAsync( attribute, target, offset, count );
- }
- /**
- * Returns the backend's rendering context.
- *
- * @return {GPUCanvasContext} The rendering context.
- */
- getContext() {
- return this.context;
- }
- /**
- * Returns the default render pass descriptor.
- *
- * In WebGPU, the default framebuffer must be configured
- * like custom framebuffers so the backend needs a render
- * pass descriptor even when rendering directly to screen.
- *
- * @private
- * @return {Object} The render pass descriptor.
- */
- _getDefaultRenderPassDescriptor() {
- const renderer = this.renderer;
- const canvasTarget = renderer.getCanvasTarget();
- const canvasData = this.get( canvasTarget );
- const samples = renderer.currentSamples;
- let descriptor = canvasData.descriptor;
- if ( descriptor === undefined || canvasData.samples !== samples ) {
- descriptor = new GPURenderPassDescriptor();
- descriptor.colorAttachments.push( new GPURenderPassColorAttachment() );
- if ( renderer.depth === true || renderer.stencil === true ) {
- const depthStencilAttachment = new GPURenderPassDepthStencilAttachment();
- depthStencilAttachment.view = this.textureUtils.getDepthBuffer( renderer.depth, renderer.stencil ).createView();
- descriptor.depthStencilAttachment = depthStencilAttachment;
- }
- const colorAttachment = descriptor.colorAttachments[ 0 ];
- if ( samples > 0 ) {
- colorAttachment.view = this.textureUtils.getColorBuffer().createView();
- } else {
- colorAttachment.resolveTarget = undefined;
- }
- canvasData.descriptor = descriptor;
- canvasData.samples = samples;
- }
- const colorAttachment = descriptor.colorAttachments[ 0 ];
- if ( samples > 0 ) {
- colorAttachment.resolveTarget = this.context.getCurrentTexture().createView();
- } else {
- colorAttachment.view = this.context.getCurrentTexture().createView();
- }
- return descriptor;
- }
- /**
- * Returns whether the render target is a render target array with depth 2D array texture.
- *
- * @param {RenderContext} renderContext - The render context.
- * @return {boolean} Whether the render target is a render target array with depth 2D array texture.
- *
- * @private
- */
- _isRenderCameraDepthArray( renderContext ) {
- const camera = renderContext.camera;
- return renderContext.depthTexture && renderContext.depthTexture.isArrayTexture === true && camera !== null && camera.isArrayCamera === true;
- }
- /**
- * Returns whether the current render context references external textures.
- *
- * External textures can change every frame, so their descriptors must not be cached.
- *
- * @private
- * @param {RenderContext} renderContext - The render context.
- * @return {boolean} Whether the render context uses external textures.
- */
- _hasExternalTexture( renderContext ) {
- const textures = renderContext.textures;
- if ( textures === null ) return false;
- for ( let i = 0; i < textures.length; i ++ ) {
- if ( this.get( textures[ i ] ).externalTexture === true ) return true;
- }
- return false;
- }
- /**
- * Creates attachment views for an external texture render target.
- *
- * @private
- * @param {RenderContext} renderContext - The render context.
- * @param {Object} textureData - The backend data for the texture.
- * @return {Array<Object>} The attachment view descriptors.
- */
- _createExternalTextureViews( renderContext, textureData ) {
- const textureViews = [];
- const camera = renderContext.camera;
- if ( textureData.xrViewDescriptors && camera !== null && camera.isArrayCamera === true ) {
- for ( let i = 0; i < textureData.xrViewDescriptors.length; i ++ ) {
- textureViews.push( {
- view: textureData.texture.createView( textureData.xrViewDescriptors[ i ] ),
- resolveTarget: undefined,
- depthSlice: undefined
- } );
- }
- } else {
- textureViews.push( {
- view: textureData.texture.createView( {
- dimension: GPUTextureViewDimension.TwoD,
- baseArrayLayer: renderContext.activeCubeFace,
- arrayLayerCount: 1
- } ),
- resolveTarget: undefined,
- depthSlice: undefined
- } );
- }
- return textureViews;
- }
- /**
- * Returns the render pass descriptor for the given render context.
- *
- * @private
- * @param {RenderContext} renderContext - The render context.
- * @param {Object} colorAttachmentsConfig - Configuration object for the color attachments.
- * @return {Object} The render pass descriptor.
- */
- _getRenderPassDescriptor( renderContext, colorAttachmentsConfig = {} ) {
- const renderTarget = renderContext.renderTarget;
- const renderTargetData = this.get( renderTarget );
- const hasExternalTexture = this._hasExternalTexture( renderContext );
- let descriptors = renderTargetData.descriptors;
- if ( descriptors === undefined ||
- renderTargetData.width !== renderTarget.width ||
- renderTargetData.height !== renderTarget.height ||
- renderTargetData.samples !== renderTarget.samples ||
- hasExternalTexture
- ) {
- descriptors = {};
- renderTargetData.descriptors = descriptors;
- }
- const cacheKey = renderContext.getCacheKey();
- let descriptorBase = descriptors[ cacheKey ];
- if ( descriptorBase === undefined || hasExternalTexture ) {
- const textures = renderContext.textures;
- const textureViews = [];
- let sliceIndex;
- const isRenderCameraDepthArray = this._isRenderCameraDepthArray( renderContext );
- for ( let i = 0; i < textures.length; i ++ ) {
- const textureData = this.get( textures[ i ] );
- if ( textureData.externalTexture === true ) {
- textureViews.push( ...this._createExternalTextureViews( renderContext, textureData ) );
- continue;
- }
- _viewDescriptor.label = `colorAttachment_${ i }`;
- _viewDescriptor.baseMipLevel = renderContext.activeMipmapLevel;
- _viewDescriptor.mipLevelCount = 1;
- _viewDescriptor.baseArrayLayer = renderContext.activeCubeFace;
- _viewDescriptor.arrayLayerCount = 1;
- _viewDescriptor.dimension = GPUTextureViewDimension.TwoD;
- if ( renderTarget.isRenderTarget3D ) {
- sliceIndex = renderContext.activeCubeFace;
- _viewDescriptor.baseArrayLayer = 0;
- _viewDescriptor.dimension = GPUTextureViewDimension.ThreeD;
- } else if ( renderTarget.isRenderTarget && textures[ i ].image.depth > 1 ) {
- if ( isRenderCameraDepthArray === true ) {
- const cameras = renderContext.camera.cameras;
- for ( let layer = 0; layer < cameras.length; layer ++ ) {
- _viewDescriptor.baseArrayLayer = layer;
- _viewDescriptor.arrayLayerCount = 1;
- _viewDescriptor.dimension = GPUTextureViewDimension.TwoD;
- const textureView = textureData.texture.createView( _viewDescriptor );
- textureViews.push( {
- view: textureView,
- resolveTarget: undefined,
- depthSlice: undefined
- } );
- }
- } else {
- _viewDescriptor.dimension = GPUTextureViewDimension.TwoDArray;
- }
- }
- if ( isRenderCameraDepthArray !== true ) {
- const textureView = textureData.texture.createView( _viewDescriptor );
- let view, resolveTarget;
- if ( textureData.msaaTexture !== undefined ) {
- view = textureData.msaaTexture.createView();
- resolveTarget = textureView;
- } else {
- view = textureView;
- resolveTarget = undefined;
- }
- textureViews.push( {
- view,
- resolveTarget,
- depthSlice: sliceIndex
- } );
- }
- _viewDescriptor.reset();
- }
- const colorAttachments = [];
- for ( let i = 0; i < textureViews.length; i ++ ) {
- const viewInfo = textureViews[ i ];
- const attachment = new GPURenderPassColorAttachment();
- attachment.view = viewInfo.view;
- attachment.depthSlice = viewInfo.depthSlice;
- attachment.resolveTarget = viewInfo.resolveTarget;
- colorAttachments.push( attachment );
- }
- descriptorBase = {
- textureViews,
- colorAttachments,
- descriptor: new GPURenderPassDescriptor()
- };
- if ( renderContext.depth ) {
- const depthTextureData = this.get( renderContext.depthTexture );
- if ( renderContext.depthTexture.isArrayTexture || renderContext.depthTexture.isCubeTexture ) {
- _viewDescriptor.dimension = GPUTextureViewDimension.TwoD;
- _viewDescriptor.arrayLayerCount = 1;
- _viewDescriptor.baseArrayLayer = renderContext.activeCubeFace;
- }
- const depthStencilAttachment = new GPURenderPassDepthStencilAttachment();
- depthStencilAttachment.view = depthTextureData.texture.createView( _viewDescriptor );
- descriptorBase.depthStencilAttachment = depthStencilAttachment;
- _viewDescriptor.reset();
- }
- descriptors[ cacheKey ] = descriptorBase;
- renderTargetData.width = renderTarget.width;
- renderTargetData.height = renderTarget.height;
- renderTargetData.samples = renderTarget.samples;
- renderTargetData.activeMipmapLevel = renderContext.activeMipmapLevel;
- renderTargetData.activeCubeFace = renderContext.activeCubeFace;
- }
- const descriptor = descriptorBase.descriptor;
- descriptor.reset();
- // Apply dynamic properties to cached attachments
- for ( let i = 0; i < descriptorBase.colorAttachments.length; i ++ ) {
- const attachment = descriptorBase.colorAttachments[ i ];
- let clearValue = { r: 0, g: 0, b: 0, a: 1 };
- if ( i === 0 && colorAttachmentsConfig.clearValue ) {
- clearValue = colorAttachmentsConfig.clearValue;
- }
- attachment.loadOp = colorAttachmentsConfig.loadOp || GPULoadOp.Load;
- attachment.storeOp = colorAttachmentsConfig.storeOp || GPUStoreOp.Store;
- attachment.clearValue = clearValue;
- descriptor.colorAttachments.push( attachment );
- }
- if ( descriptorBase.depthStencilAttachment ) {
- descriptor.depthStencilAttachment = descriptorBase.depthStencilAttachment;
- }
- return descriptor;
- }
- /**
- * This method is executed at the beginning of a render call and prepares
- * the WebGPU state for upcoming render calls
- *
- * @param {RenderContext} renderContext - The render context.
- */
- beginRender( renderContext ) {
- const renderContextData = this.get( renderContext );
- //
- const device = this.device;
- const occlusionQueryCount = renderContext.occlusionQueryCount;
- let occlusionQuerySet;
- if ( occlusionQueryCount > 0 ) {
- if ( renderContextData.currentOcclusionQuerySet ) renderContextData.currentOcclusionQuerySet.destroy();
- if ( renderContextData.currentOcclusionQueryBuffer ) renderContextData.currentOcclusionQueryBuffer.destroy();
- // Get a reference to the array of objects with queries. The renderContextData property
- // can be changed by another render pass before the buffer.mapAsyc() completes.
- renderContextData.currentOcclusionQuerySet = renderContextData.occlusionQuerySet;
- renderContextData.currentOcclusionQueryBuffer = renderContextData.occlusionQueryBuffer;
- renderContextData.currentOcclusionQueryObjects = renderContextData.occlusionQueryObjects;
- //
- _querySetDescriptor.label = `occlusionQuerySet_${ renderContext.id }`;
- _querySetDescriptor.type = 'occlusion';
- _querySetDescriptor.count = occlusionQueryCount;
- occlusionQuerySet = device.createQuerySet( _querySetDescriptor );
- _querySetDescriptor.reset();
- renderContextData.occlusionQuerySet = occlusionQuerySet;
- renderContextData.occlusionQueryIndex = 0;
- renderContextData.occlusionQueryObjects = new Array( occlusionQueryCount );
- renderContextData.lastOcclusionObject = null;
- }
- let descriptor;
- if ( renderContext.textures === null ) {
- descriptor = this._getDefaultRenderPassDescriptor();
- } else {
- descriptor = this._getRenderPassDescriptor( renderContext, { loadOp: GPULoadOp.Load } );
- }
- this.initTimestampQuery( TimestampQuery.RENDER, this.getTimestampUID( renderContext ), descriptor );
- descriptor.occlusionQuerySet = occlusionQuerySet;
- const depthStencilAttachment = descriptor.depthStencilAttachment;
- if ( renderContext.textures !== null ) {
- const colorAttachments = descriptor.colorAttachments;
- for ( let i = 0; i < colorAttachments.length; i ++ ) {
- const colorAttachment = colorAttachments[ i ];
- if ( renderContext.clearColor ) {
- if ( i === 0 ) {
- colorAttachment.clearValue = renderContext.clearColorValue;
- } else {
- _clearValue.r = 0;
- _clearValue.g = 0;
- _clearValue.b = 0;
- _clearValue.a = 1;
- colorAttachment.clearValue = _clearValue;
- }
- colorAttachment.loadOp = GPULoadOp.Clear;
- } else {
- colorAttachment.loadOp = GPULoadOp.Load;
- }
- colorAttachment.storeOp = GPUStoreOp.Store;
- }
- } else {
- const colorAttachment = descriptor.colorAttachments[ 0 ];
- if ( renderContext.clearColor ) {
- colorAttachment.clearValue = renderContext.clearColorValue;
- colorAttachment.loadOp = GPULoadOp.Clear;
- } else {
- colorAttachment.loadOp = GPULoadOp.Load;
- }
- colorAttachment.storeOp = GPUStoreOp.Store;
- }
- //
- if ( renderContext.depth ) {
- if ( renderContext.clearDepth ) {
- depthStencilAttachment.depthClearValue = renderContext.clearDepthValue;
- depthStencilAttachment.depthLoadOp = GPULoadOp.Clear;
- } else {
- depthStencilAttachment.depthLoadOp = GPULoadOp.Load;
- }
- depthStencilAttachment.depthStoreOp = GPUStoreOp.Store;
- }
- if ( renderContext.stencil ) {
- if ( renderContext.clearStencil ) {
- depthStencilAttachment.stencilClearValue = renderContext.clearStencilValue;
- depthStencilAttachment.stencilLoadOp = GPULoadOp.Clear;
- } else {
- depthStencilAttachment.stencilLoadOp = GPULoadOp.Load;
- }
- depthStencilAttachment.stencilStoreOp = GPUStoreOp.Store;
- }
- //
- _commandEncoderDescriptor.label = 'renderContext_' + renderContext.id;
- const encoder = device.createCommandEncoder( _commandEncoderDescriptor );
- _commandEncoderDescriptor.reset();
- // Layered render targets: prepare bundle encoders for each camera in the array camera.
- if ( this._isRenderCameraDepthArray( renderContext ) === true ) {
- const cameras = renderContext.camera.cameras;
- if ( ! renderContextData.layerDescriptors || renderContextData.layerDescriptors.length !== cameras.length ) {
- this._createArrayCameraLayerDescriptors( renderContext, renderContextData, descriptor, cameras );
- } else {
- this._updateArrayCameraLayerDescriptors( renderContext, renderContextData, cameras );
- }
- // Create bundle encoders for each layer
- renderContextData.bundleEncoders = [];
- renderContextData.bundleSets = [];
- // Create separate bundle encoders for each camera in the array
- for ( let i = 0; i < cameras.length; i ++ ) {
- const bundleEncoder = this.pipelineUtils.createBundleEncoder(
- renderContext,
- 'renderBundleArrayCamera_' + i
- );
- // Initialize state tracking for this bundle
- const bundleSets = {
- attributes: {},
- bindingGroups: [],
- pipeline: null,
- index: null
- };
- renderContextData.bundleEncoders.push( bundleEncoder );
- renderContextData.bundleSets.push( bundleSets );
- }
- // We'll complete the bundles in finishRender
- renderContextData.currentPass = null;
- } else {
- const currentPass = encoder.beginRenderPass( descriptor );
- renderContextData.currentPass = currentPass;
- if ( renderContext.viewport ) {
- this.updateViewport( renderContext );
- }
- if ( renderContext.scissor ) {
- this.updateScissor( renderContext );
- }
- }
- //
- renderContextData.descriptor = descriptor;
- renderContextData.encoder = encoder;
- renderContextData.currentSets = { attributes: {}, bindingGroups: [], pipeline: null, index: null };
- renderContextData.renderBundles = [];
- }
- /**
- * Creates render pass descriptors for each camera in an array camera.
- *
- * @param {RenderContext} renderContext - The render context.
- * @param {Object} renderContextData - The render context data.
- * @param {Object} descriptor - The render pass descriptor.
- * @param {ArrayCamera} cameras - The array camera.
- *
- * @private
- */
- _createArrayCameraLayerDescriptors( renderContext, renderContextData, descriptor, cameras ) {
- const depthStencilAttachment = descriptor.depthStencilAttachment;
- renderContextData.layerDescriptors = [];
- const depthTextureData = this.get( renderContext.depthTexture );
- if ( ! depthTextureData.viewCache ) {
- depthTextureData.viewCache = [];
- }
- for ( let i = 0; i < cameras.length; i ++ ) {
- const sourceAttachment = descriptor.colorAttachments[ 0 ];
- const layerColorAttachment = new GPURenderPassColorAttachment();
- layerColorAttachment.view = descriptor.colorAttachments[ i ].view;
- layerColorAttachment.depthSlice = sourceAttachment.depthSlice;
- layerColorAttachment.resolveTarget = sourceAttachment.resolveTarget;
- layerColorAttachment.loadOp = sourceAttachment.loadOp;
- layerColorAttachment.storeOp = sourceAttachment.storeOp;
- layerColorAttachment.clearValue = sourceAttachment.clearValue;
- const layerDescriptor = new GPURenderPassDescriptor();
- layerDescriptor.label = descriptor.label;
- layerDescriptor.occlusionQuerySet = descriptor.occlusionQuerySet;
- layerDescriptor.timestampWrites = descriptor.timestampWrites;
- layerDescriptor.colorAttachments.push( layerColorAttachment );
- if ( descriptor.depthStencilAttachment ) {
- const layerIndex = i;
- if ( ! depthTextureData.viewCache[ layerIndex ] ) {
- _viewDescriptor.dimension = GPUTextureViewDimension.TwoD;
- _viewDescriptor.baseArrayLayer = i;
- _viewDescriptor.arrayLayerCount = 1;
- depthTextureData.viewCache[ layerIndex ] = depthTextureData.texture.createView( _viewDescriptor );
- _viewDescriptor.reset();
- }
- const layerDepthStencilAttachment = new GPURenderPassDepthStencilAttachment();
- layerDepthStencilAttachment.view = depthTextureData.viewCache[ layerIndex ];
- layerDepthStencilAttachment.depthLoadOp = depthStencilAttachment.depthLoadOp || GPULoadOp.Clear;
- layerDepthStencilAttachment.depthStoreOp = depthStencilAttachment.depthStoreOp || GPUStoreOp.Store;
- layerDepthStencilAttachment.depthClearValue = depthStencilAttachment.depthClearValue || 1.0;
- if ( renderContext.stencil ) {
- layerDepthStencilAttachment.stencilLoadOp = depthStencilAttachment.stencilLoadOp;
- layerDepthStencilAttachment.stencilStoreOp = depthStencilAttachment.stencilStoreOp;
- layerDepthStencilAttachment.stencilClearValue = depthStencilAttachment.stencilClearValue;
- }
- layerDescriptor.depthStencilAttachment = layerDepthStencilAttachment;
- } else {
- const layerDepthStencilAttachment = new GPURenderPassDepthStencilAttachment();
- layerDepthStencilAttachment.view = depthStencilAttachment.view;
- layerDepthStencilAttachment.depthLoadOp = depthStencilAttachment.depthLoadOp;
- layerDepthStencilAttachment.depthStoreOp = depthStencilAttachment.depthStoreOp;
- layerDepthStencilAttachment.depthClearValue = depthStencilAttachment.depthClearValue;
- layerDepthStencilAttachment.depthReadOnly = depthStencilAttachment.depthReadOnly;
- layerDepthStencilAttachment.stencilLoadOp = depthStencilAttachment.stencilLoadOp;
- layerDepthStencilAttachment.stencilStoreOp = depthStencilAttachment.stencilStoreOp;
- layerDepthStencilAttachment.stencilClearValue = depthStencilAttachment.stencilClearValue;
- layerDepthStencilAttachment.stencilReadOnly = depthStencilAttachment.stencilReadOnly;
- layerDescriptor.depthStencilAttachment = layerDepthStencilAttachment;
- }
- renderContextData.layerDescriptors.push( layerDescriptor );
- }
- }
- /**
- * Updates render pass descriptors for each camera in an array camera.
- *
- * @param {RenderContext} renderContext - The render context.
- * @param {Object} renderContextData - The render context data.
- * @param {ArrayCamera} cameras - The array camera.
- *
- */
- _updateArrayCameraLayerDescriptors( renderContext, renderContextData, cameras ) {
- for ( let i = 0; i < cameras.length; i ++ ) {
- const layerDescriptor = renderContextData.layerDescriptors[ i ];
- if ( layerDescriptor.depthStencilAttachment ) {
- const depthAttachment = layerDescriptor.depthStencilAttachment;
- if ( renderContext.depth ) {
- if ( renderContext.clearDepth ) {
- depthAttachment.depthClearValue = renderContext.clearDepthValue;
- depthAttachment.depthLoadOp = GPULoadOp.Clear;
- } else {
- depthAttachment.depthLoadOp = GPULoadOp.Load;
- }
- }
- if ( renderContext.stencil ) {
- if ( renderContext.clearStencil ) {
- depthAttachment.stencilClearValue = renderContext.clearStencilValue;
- depthAttachment.stencilLoadOp = GPULoadOp.Clear;
- } else {
- depthAttachment.stencilLoadOp = GPULoadOp.Load;
- }
- }
- }
- }
- }
- /**
- * This method is executed at the end of a render call and finalizes work
- * after draw calls.
- *
- * @param {RenderContext} renderContext - The render context.
- */
- finishRender( renderContext ) {
- const renderContextData = this.get( renderContext );
- const occlusionQueryCount = renderContext.occlusionQueryCount;
- if ( renderContextData.renderBundles.length > 0 ) {
- renderContextData.currentPass.executeBundles( renderContextData.renderBundles );
- }
- if ( occlusionQueryCount > renderContextData.occlusionQueryIndex ) {
- renderContextData.currentPass.endOcclusionQuery();
- }
- // Layered render targets: execute the bundle for each layer.
- const encoder = renderContextData.encoder;
- if ( this._isRenderCameraDepthArray( renderContext ) === true ) {
- const bundles = [];
- for ( let i = 0; i < renderContextData.bundleEncoders.length; i ++ ) {
- const bundleEncoder = renderContextData.bundleEncoders[ i ];
- bundles.push( bundleEncoder.finish() );
- }
- for ( let i = 0; i < renderContextData.layerDescriptors.length; i ++ ) {
- if ( i < bundles.length ) {
- const layerDescriptor = renderContextData.layerDescriptors[ i ];
- const renderPass = encoder.beginRenderPass( layerDescriptor );
- if ( renderContext.viewport ) {
- const { x, y, width, height, minDepth, maxDepth } = renderContext.viewportValue;
- renderPass.setViewport( x, y, width, height, minDepth, maxDepth );
- }
- if ( renderContext.scissor ) {
- const { x, y, width, height } = renderContext.scissorValue;
- renderPass.setScissorRect( x, y, width, height );
- }
- renderPass.executeBundles( [ bundles[ i ] ] );
- renderPass.end();
- }
- }
- } else if ( renderContextData.currentPass ) {
- renderContextData.currentPass.end();
- }
- if ( occlusionQueryCount > 0 ) {
- const bufferSize = occlusionQueryCount * 8; // 8 byte entries for query results
- //
- let queryResolveBuffer = this.occludedResolveCache.get( bufferSize );
- if ( queryResolveBuffer === undefined ) {
- _bufferDescriptor.size = bufferSize;
- _bufferDescriptor.usage = GPUBufferUsage.QUERY_RESOLVE | GPUBufferUsage.COPY_SRC;
- queryResolveBuffer = this.device.createBuffer( _bufferDescriptor );
- _bufferDescriptor.reset();
- this.occludedResolveCache.set( bufferSize, queryResolveBuffer );
- }
- //
- _bufferDescriptor.size = bufferSize;
- _bufferDescriptor.usage = GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ;
- const readBuffer = this.device.createBuffer( _bufferDescriptor );
- _bufferDescriptor.reset();
- // two buffers required here - WebGPU doesn't allow usage of QUERY_RESOLVE & MAP_READ to be combined
- renderContextData.encoder.resolveQuerySet( renderContextData.occlusionQuerySet, 0, occlusionQueryCount, queryResolveBuffer, 0 );
- renderContextData.encoder.copyBufferToBuffer( queryResolveBuffer, 0, readBuffer, 0, bufferSize );
- renderContextData.occlusionQueryBuffer = readBuffer;
- //
- this.resolveOccludedAsync( renderContext );
- }
- submit( this.device, renderContextData.encoder.finish() );
- //
- if ( renderContext.textures !== null ) {
- const textures = renderContext.textures;
- for ( let i = 0; i < textures.length; i ++ ) {
- const texture = textures[ i ];
- if ( texture.generateMipmaps === true ) {
- this.textureUtils.generateMipmaps( texture );
- }
- }
- }
- }
- /**
- * Returns `true` if the given 3D object is fully occluded by other
- * 3D objects in the scene.
- *
- * @param {RenderContext} renderContext - The render context.
- * @param {Object3D} object - The 3D object to test.
- * @return {boolean} Whether the 3D object is fully occluded or not.
- */
- isOccluded( renderContext, object ) {
- const renderContextData = this.get( renderContext );
- return renderContextData.occluded && renderContextData.occluded.has( object );
- }
- /**
- * This method processes the result of occlusion queries and writes it
- * into render context data.
- *
- * @async
- * @param {RenderContext} renderContext - The render context.
- * @return {Promise} A Promise that resolves when the occlusion query results have been processed.
- */
- async resolveOccludedAsync( renderContext ) {
- const renderContextData = this.get( renderContext );
- // handle occlusion query results
- const { currentOcclusionQueryBuffer, currentOcclusionQueryObjects } = renderContextData;
- if ( currentOcclusionQueryBuffer && currentOcclusionQueryObjects ) {
- const occluded = new WeakSet();
- renderContextData.currentOcclusionQueryObjects = null;
- renderContextData.currentOcclusionQueryBuffer = null;
- await currentOcclusionQueryBuffer.mapAsync( GPUMapMode.READ );
- const buffer = currentOcclusionQueryBuffer.getMappedRange();
- const results = new BigUint64Array( buffer );
- for ( let i = 0; i < currentOcclusionQueryObjects.length; i ++ ) {
- if ( results[ i ] === BigInt( 0 ) ) {
- occluded.add( currentOcclusionQueryObjects[ i ] );
- }
- }
- currentOcclusionQueryBuffer.destroy();
- renderContextData.occluded = occluded;
- }
- }
- /**
- * Updates the viewport with the values from the given render context.
- *
- * @param {RenderContext} renderContext - The render context.
- */
- updateViewport( renderContext ) {
- const { currentPass } = this.get( renderContext );
- const { x, y, width, height, minDepth, maxDepth } = renderContext.viewportValue;
- currentPass.setViewport( x, y, width, height, minDepth, maxDepth );
- }
- /**
- * Updates the scissor with the values from the given render context.
- *
- * @param {RenderContext} renderContext - The render context.
- */
- updateScissor( renderContext ) {
- const { currentPass } = this.get( renderContext );
- const { x, y, width, height } = renderContext.scissorValue;
- currentPass.setScissorRect( x, y, width, height );
- }
- /**
- * Returns the clear color and alpha into a single
- * color object.
- *
- * @return {Color4} The clear color.
- */
- getClearColor() {
- const clearColor = super.getClearColor();
- // only premultiply alpha when alphaMode is "premultiplied"
- if ( this.renderer.alpha === true ) {
- clearColor.r *= clearColor.a;
- clearColor.g *= clearColor.a;
- clearColor.b *= clearColor.a;
- }
- return clearColor;
- }
- /**
- * Performs a clear operation.
- *
- * @param {boolean} color - Whether the color buffer should be cleared or not.
- * @param {boolean} depth - Whether the depth buffer should be cleared or not.
- * @param {boolean} stencil - Whether the stencil buffer should be cleared or not.
- * @param {?RenderContext} [renderTargetContext=null] - The render context of the current set render target.
- */
- clear( color, depth, stencil, renderTargetContext = null ) {
- const device = this.device;
- const renderer = this.renderer;
- let colorAttachments = [];
- let depthStencilAttachment;
- let supportsDepth;
- let supportsStencil;
- if ( color ) {
- const clearColor = this.getClearColor();
- _clearValue.r = clearColor.r;
- _clearValue.g = clearColor.g;
- _clearValue.b = clearColor.b;
- _clearValue.a = clearColor.a;
- }
- if ( renderTargetContext === null ) {
- supportsDepth = renderer.depth;
- supportsStencil = renderer.stencil;
- const descriptor = this._getDefaultRenderPassDescriptor();
- if ( color ) {
- colorAttachments = descriptor.colorAttachments;
- const colorAttachment = colorAttachments[ 0 ];
- colorAttachment.clearValue = _clearValue;
- colorAttachment.loadOp = GPULoadOp.Clear;
- colorAttachment.storeOp = GPUStoreOp.Store;
- }
- if ( supportsDepth || supportsStencil ) {
- depthStencilAttachment = descriptor.depthStencilAttachment;
- }
- } else {
- supportsDepth = renderTargetContext.depth;
- supportsStencil = renderTargetContext.stencil;
- const clearConfig = {
- loadOp: color ? GPULoadOp.Clear : GPULoadOp.Load,
- clearValue: color ? _clearValue : undefined
- };
- if ( supportsDepth ) {
- clearConfig.depthLoadOp = depth ? GPULoadOp.Clear : GPULoadOp.Load;
- clearConfig.depthClearValue = depth ? renderer.getClearDepth() : undefined;
- clearConfig.depthStoreOp = GPUStoreOp.Store;
- }
- if ( supportsStencil ) {
- clearConfig.stencilLoadOp = stencil ? GPULoadOp.Clear : GPULoadOp.Load;
- clearConfig.stencilClearValue = stencil ? renderer.getClearStencil() : undefined;
- clearConfig.stencilStoreOp = GPUStoreOp.Store;
- }
- const descriptor = this._getRenderPassDescriptor( renderTargetContext, clearConfig );
- colorAttachments = descriptor.colorAttachments;
- depthStencilAttachment = descriptor.depthStencilAttachment;
- }
- if ( supportsDepth && depthStencilAttachment ) {
- if ( depth ) {
- depthStencilAttachment.depthLoadOp = GPULoadOp.Clear;
- depthStencilAttachment.depthClearValue = renderer.getClearDepth();
- depthStencilAttachment.depthStoreOp = GPUStoreOp.Store;
- } else {
- depthStencilAttachment.depthLoadOp = GPULoadOp.Load;
- depthStencilAttachment.depthStoreOp = GPUStoreOp.Store;
- }
- }
- //
- if ( supportsStencil && depthStencilAttachment ) {
- if ( stencil ) {
- depthStencilAttachment.stencilLoadOp = GPULoadOp.Clear;
- depthStencilAttachment.stencilClearValue = renderer.getClearStencil();
- depthStencilAttachment.stencilStoreOp = GPUStoreOp.Store;
- } else {
- depthStencilAttachment.stencilLoadOp = GPULoadOp.Load;
- depthStencilAttachment.stencilStoreOp = GPUStoreOp.Store;
- }
- }
- //
- _commandEncoderDescriptor.label = 'clear';
- const encoder = device.createCommandEncoder( _commandEncoderDescriptor );
- _commandEncoderDescriptor.reset();
- const currentPass = encoder.beginRenderPass( {
- colorAttachments,
- depthStencilAttachment
- } );
- currentPass.end();
- submit( device, encoder.finish() );
- }
- // compute
- /**
- * This method is executed at the beginning of a compute call and
- * prepares the state for upcoming compute tasks.
- *
- * @param {Node|Array<Node>} computeGroup - The compute node(s).
- */
- beginCompute( computeGroup ) {
- const groupGPU = this.get( computeGroup );
- //
- const label = 'computeGroup_' + computeGroup.id;
- _computePassDescriptor.label = label;
- _commandEncoderDescriptor.label = label;
- this.initTimestampQuery( TimestampQuery.COMPUTE, this.getTimestampUID( computeGroup ), _computePassDescriptor );
- groupGPU.cmdEncoderGPU = this.device.createCommandEncoder( _commandEncoderDescriptor );
- groupGPU.passEncoderGPU = groupGPU.cmdEncoderGPU.beginComputePass( _computePassDescriptor );
- groupGPU.currentPipeline = null;
- _commandEncoderDescriptor.reset();
- _computePassDescriptor.reset();
- }
- /**
- * Executes a compute command for the given compute node.
- *
- * @param {Node|Array<Node>} computeGroup - The group of compute nodes of a compute call. Can be a single compute node.
- * @param {Node} computeNode - The compute node.
- * @param {Array<BindGroup>} bindings - The bindings.
- * @param {ComputePipeline} pipeline - The compute pipeline.
- * @param {number|Array<number>|IndirectStorageBufferAttribute} [dispatchSize=null]
- * - A single number representing count, or
- * - An array [x, y, z] representing dispatch size, or
- * - A IndirectStorageBufferAttribute for indirect dispatch size.
- */
- compute( computeGroup, computeNode, bindings, pipeline, dispatchSize = null ) {
- const computeNodeData = this.get( computeNode );
- const groupGPU = this.get( computeGroup );
- const { passEncoderGPU } = groupGPU;
- // pipeline
- const pipelineGPU = this.get( pipeline ).pipeline;
- if ( groupGPU.currentPipeline !== pipelineGPU ) {
- passEncoderGPU.setPipeline( pipelineGPU );
- groupGPU.currentPipeline = pipelineGPU;
- }
- // bind groups
- for ( let i = 0, l = bindings.length; i < l; i ++ ) {
- const bindGroup = bindings[ i ];
- const bindingsData = this.get( bindGroup );
- passEncoderGPU.setBindGroup( i, bindingsData.group );
- }
- if ( dispatchSize === null ) {
- dispatchSize = computeNode.dispatchSize || computeNode.count;
- }
- // When the dispatchSize is set with a StorageBuffer from the GPU.
- if ( dispatchSize && dispatchSize.isIndirectStorageBufferAttribute ) {
- const dispatchBuffer = this.get( dispatchSize ).buffer;
- passEncoderGPU.dispatchWorkgroupsIndirect( dispatchBuffer, 0 );
- return;
- }
- if ( typeof dispatchSize === 'number' ) {
- // If a single number is given, we calculate the dispatch size based on the workgroup size
- const count = dispatchSize;
- if ( computeNodeData.dispatchSize === undefined || computeNodeData.count !== count ) {
- // cache dispatch size to avoid recalculating it every time
- computeNodeData.dispatchSize = [ 0, 1, 1 ];
- computeNodeData.count = count;
- const workgroupSize = computeNode.workgroupSize;
- let size = workgroupSize[ 0 ];
- for ( let i = 1; i < workgroupSize.length; i ++ )
- size *= workgroupSize[ i ];
- const dispatchCount = Math.ceil( count / size );
- //
- const maxComputeWorkgroupsPerDimension = this.device.limits.maxComputeWorkgroupsPerDimension;
- dispatchSize = [ dispatchCount, 1, 1 ];
- if ( dispatchCount > maxComputeWorkgroupsPerDimension ) {
- dispatchSize[ 0 ] = Math.min( dispatchCount, maxComputeWorkgroupsPerDimension );
- dispatchSize[ 1 ] = Math.ceil( dispatchCount / maxComputeWorkgroupsPerDimension );
- }
- computeNodeData.dispatchSize = dispatchSize;
- }
- dispatchSize = computeNodeData.dispatchSize;
- }
- //
- passEncoderGPU.dispatchWorkgroups(
- dispatchSize[ 0 ],
- dispatchSize[ 1 ] || 1,
- dispatchSize[ 2 ] || 1
- );
- }
- /**
- * This method is executed at the end of a compute call and
- * finalizes work after compute tasks.
- *
- * @param {Node|Array<Node>} computeGroup - The compute node(s).
- */
- finishCompute( computeGroup ) {
- const groupData = this.get( computeGroup );
- groupData.passEncoderGPU.end();
- submit( this.device, groupData.cmdEncoderGPU.finish() );
- }
- /**
- * Internal draw function that performs the draw with the given pass encoder.
- *
- * @private
- * @param {RenderObject} renderObject - The render object.
- * @param {Info} info - Holds a series of statistical information about the GPU memory and the rendering process.
- * @param {Object} renderContextData - The render context data object, holding current pass state and occlusion query tracking.
- * @param {GPURenderPipeline} pipelineGPU - The GPU render pipeline.
- * @param {Array<BindGroup>} bindings - The bind groups.
- * @param {Array<BufferAttribute>} vertexBuffers - The vertex buffers.
- * @param {{vertexCount: number, firstVertex: number, instanceCount: number, firstInstance: number}} drawParams - The draw parameters.
- * @param {GPURenderPassEncoder|GPURenderBundleEncoder} passEncoderGPU - The GPU pass encoder used for recording draw commands.
- * @param {Object} currentSets - Tracking object for currently set pipeline, attributes, bind groups, and index state.
- */
- _draw( renderObject, info, renderContextData, pipelineGPU, bindings, vertexBuffers, drawParams, passEncoderGPU, currentSets ) {
- const { object, material, context } = renderObject;
- const index = renderObject.getIndex();
- const hasIndex = ( index !== null );
- // pipeline
- if ( currentSets.pipeline !== pipelineGPU ) {
- passEncoderGPU.setPipeline( pipelineGPU );
- currentSets.pipeline = pipelineGPU;
- }
- // bind groups
- const currentBindingGroups = currentSets.bindingGroups;
- for ( let i = 0, l = bindings.length; i < l; i ++ ) {
- const bindGroup = bindings[ i ];
- if ( currentBindingGroups[ i ] !== bindGroup.id ) {
- const bindingsData = this.get( bindGroup );
- passEncoderGPU.setBindGroup( i, bindingsData.group );
- currentBindingGroups[ i ] = bindGroup.id;
- }
- }
- // attributes
- // index
- if ( hasIndex === true ) {
- if ( currentSets.index !== index ) {
- const buffer = this.get( index ).buffer;
- const indexFormat = ( index.array instanceof Uint16Array ) ? GPUIndexFormat.Uint16 : GPUIndexFormat.Uint32;
- passEncoderGPU.setIndexBuffer( buffer, indexFormat );
- currentSets.index = index;
- }
- }
- for ( let i = 0, l = vertexBuffers.length; i < l; i ++ ) {
- const vertexBuffer = vertexBuffers[ i ];
- if ( currentSets.attributes[ i ] !== vertexBuffer ) {
- const buffer = this.get( vertexBuffer ).buffer;
- passEncoderGPU.setVertexBuffer( i, buffer );
- currentSets.attributes[ i ] = vertexBuffer;
- }
- }
- // stencil
- if ( context.stencil === true && material.stencilWrite === true && renderContextData.currentStencilRef !== material.stencilRef ) {
- passEncoderGPU.setStencilReference( material.stencilRef );
- renderContextData.currentStencilRef = material.stencilRef;
- }
- if ( object.isBatchedMesh === true ) {
- const starts = object._multiDrawStarts;
- const counts = object._multiDrawCounts;
- const drawCount = object._multiDrawCount;
- let bytesPerElement = ( hasIndex === true ) ? index.array.BYTES_PER_ELEMENT : 1;
- if ( material.wireframe ) {
- bytesPerElement = object.geometry.attributes.position.count > 65535 ? 4 : 2;
- }
- for ( let i = 0; i < drawCount; i ++ ) {
- if ( hasIndex === true ) {
- passEncoderGPU.drawIndexed( counts[ i ], 1, starts[ i ] / bytesPerElement, 0, i );
- } else {
- passEncoderGPU.draw( counts[ i ], 1, starts[ i ], i );
- }
- info.update( object, counts[ i ], 1 );
- }
- } else if ( hasIndex === true ) {
- const { vertexCount: indexCount, instanceCount, firstVertex: firstIndex } = drawParams;
- const indirect = renderObject.getIndirect();
- if ( indirect !== null ) {
- const buffer = this.get( indirect ).buffer;
- const indirectOffset = renderObject.getIndirectOffset();
- const indirectOffsets = Array.isArray( indirectOffset ) ? indirectOffset : [ indirectOffset ];
- for ( let i = 0; i < indirectOffsets.length; i ++ ) {
- passEncoderGPU.drawIndexedIndirect( buffer, indirectOffsets[ i ] );
- }
- } else {
- passEncoderGPU.drawIndexed( indexCount, instanceCount, firstIndex, 0, 0 );
- }
- info.update( object, indexCount, instanceCount );
- } else {
- const { vertexCount, instanceCount, firstVertex } = drawParams;
- const indirect = renderObject.getIndirect();
- if ( indirect !== null ) {
- const buffer = this.get( indirect ).buffer;
- const indirectOffset = renderObject.getIndirectOffset();
- const indirectOffsets = Array.isArray( indirectOffset ) ? indirectOffset : [ indirectOffset ];
- for ( let i = 0; i < indirectOffsets.length; i ++ ) {
- passEncoderGPU.drawIndirect( buffer, indirectOffsets[ i ] );
- }
- } else {
- passEncoderGPU.draw( vertexCount, instanceCount, firstVertex, 0 );
- }
- info.update( object, vertexCount, instanceCount );
- }
- }
- // render object
- /**
- * Executes a draw command for the given render object.
- *
- * @param {RenderObject} renderObject - The render object to draw.
- * @param {Info} info - Holds a series of statistical information about the GPU memory and the rendering process.
- */
- draw( renderObject, info ) {
- const { object, context, pipeline } = renderObject;
- const renderContextData = this.get( context );
- const pipelineData = this.get( pipeline );
- const pipelineGPU = pipelineData.pipeline;
- // Skip if pipeline has error
- if ( pipelineData.error === true ) return;
- const drawParams = renderObject.getDrawParameters();
- if ( drawParams === null ) return;
- const bindings = renderObject.getBindings();
- // vertex buffers
- const vertexBuffers = renderObject.getVertexBuffers();
- if ( renderObject.camera.isArrayCamera && renderObject.camera.cameras.length > 0 ) {
- const cameraData = this.get( renderObject.camera );
- const cameras = renderObject.camera.cameras;
- const cameraIndex = renderObject.getBindingGroup( 'cameraIndex' );
- if ( cameraData.indexesGPU === undefined || cameraData.indexesGPU.length !== cameras.length ) {
- const bindingsData = this.get( cameraIndex );
- const indexesGPU = [];
- const data = new Uint32Array( [ 0, 0, 0, 0 ] );
- for ( let i = 0, len = cameras.length; i < len; i ++ ) {
- data[ 0 ] = i;
- const { layoutGPU } = bindingsData.layout;
- const bindGroupIndex = this.bindingUtils.createBindGroupIndex( data, layoutGPU );
- indexesGPU.push( bindGroupIndex );
- }
- cameraData.indexesGPU = indexesGPU; // TODO: Create a global library for this
- }
- const pixelRatio = this.renderer.getPixelRatio();
- for ( let i = 0, len = cameras.length; i < len; i ++ ) {
- const subCamera = cameras[ i ];
- if ( object.layers.test( subCamera.layers ) ) {
- const vp = subCamera.viewport;
- let pass = renderContextData.currentPass;
- let sets = renderContextData.currentSets;
- const isBundleEncoder = renderContextData.bundleEncoders !== undefined;
- if ( isBundleEncoder ) {
- const bundleEncoder = renderContextData.bundleEncoders[ i ];
- const bundleSets = renderContextData.bundleSets[ i ];
- pass = bundleEncoder;
- sets = bundleSets;
- }
- // GPURenderBundleEncoder does not support setViewport, only GPURenderPassEncoder does
- if ( vp && ! isBundleEncoder ) {
- pass.setViewport(
- Math.floor( vp.x * pixelRatio ),
- Math.floor( vp.y * pixelRatio ),
- Math.floor( vp.width * pixelRatio ),
- Math.floor( vp.height * pixelRatio ),
- context.viewportValue.minDepth,
- context.viewportValue.maxDepth
- );
- }
- // Set camera index binding for this layer
- if ( cameraIndex && cameraData.indexesGPU ) {
- const indexPos = bindings.indexOf( cameraIndex );
- pass.setBindGroup( indexPos, cameraData.indexesGPU[ i ] );
- sets.bindingGroups[ indexPos ] = cameraIndex.id;
- }
- this._draw( renderObject, info, renderContextData, pipelineGPU, bindings, vertexBuffers, drawParams, pass, sets );
- }
- }
- } else {
- // Regular single camera rendering
- if ( renderContextData.currentPass ) {
- // Handle occlusion queries
- if ( renderContextData.occlusionQuerySet !== undefined ) {
- const lastObject = renderContextData.lastOcclusionObject;
- if ( lastObject !== object ) {
- if ( lastObject !== null && lastObject.occlusionTest === true ) {
- renderContextData.currentPass.endOcclusionQuery();
- renderContextData.occlusionQueryIndex ++;
- }
- if ( object.occlusionTest === true ) {
- renderContextData.currentPass.beginOcclusionQuery( renderContextData.occlusionQueryIndex );
- renderContextData.occlusionQueryObjects[ renderContextData.occlusionQueryIndex ] = object;
- }
- renderContextData.lastOcclusionObject = object;
- }
- }
- this._draw( renderObject, info, renderContextData, pipelineGPU, bindings, vertexBuffers, drawParams, renderContextData.currentPass, renderContextData.currentSets );
- }
- }
- }
- // cache key
- /**
- * Returns `true` if the render pipeline requires an update.
- *
- * @param {RenderObject} renderObject - The render object.
- * @return {boolean} Whether the render pipeline requires an update or not.
- */
- needsRenderUpdate( renderObject ) {
- const data = this.get( renderObject );
- const { object, material } = renderObject;
- const utils = this.utils;
- const sampleCount = utils.getSampleCountRenderContext( renderObject.context );
- const colorSpace = utils.getCurrentColorSpace( renderObject.context );
- const colorFormat = utils.getCurrentColorFormat( renderObject.context );
- const depthStencilFormat = utils.getCurrentDepthStencilFormat( renderObject.context );
- const primitiveTopology = utils.getPrimitiveTopology( object, material );
- const frontFaceCW = ( object.isMesh && object.matrixWorld.determinantAffine() < 0 );
- let needsUpdate = false;
- if ( data.material !== material || data.materialVersion !== material.version ||
- data.transparent !== material.transparent || data.blending !== material.blending || data.premultipliedAlpha !== material.premultipliedAlpha ||
- data.blendSrc !== material.blendSrc || data.blendDst !== material.blendDst || data.blendEquation !== material.blendEquation ||
- data.blendSrcAlpha !== material.blendSrcAlpha || data.blendDstAlpha !== material.blendDstAlpha || data.blendEquationAlpha !== material.blendEquationAlpha ||
- data.colorWrite !== material.colorWrite || data.depthWrite !== material.depthWrite || data.depthTest !== material.depthTest || data.depthFunc !== material.depthFunc ||
- data.stencilWrite !== material.stencilWrite || data.stencilFunc !== material.stencilFunc ||
- data.stencilFail !== material.stencilFail || data.stencilZFail !== material.stencilZFail || data.stencilZPass !== material.stencilZPass ||
- data.stencilFuncMask !== material.stencilFuncMask || data.stencilWriteMask !== material.stencilWriteMask ||
- data.side !== material.side || data.alphaToCoverage !== material.alphaToCoverage ||
- data.sampleCount !== sampleCount || data.colorSpace !== colorSpace ||
- data.colorFormat !== colorFormat || data.depthStencilFormat !== depthStencilFormat ||
- data.primitiveTopology !== primitiveTopology ||
- data.frontFaceCW !== frontFaceCW ||
- data.clippingContextCacheKey !== renderObject.clippingContextCacheKey
- ) {
- data.material = material; data.materialVersion = material.version;
- data.transparent = material.transparent; data.blending = material.blending; data.premultipliedAlpha = material.premultipliedAlpha;
- data.blendSrc = material.blendSrc; data.blendDst = material.blendDst; data.blendEquation = material.blendEquation;
- data.blendSrcAlpha = material.blendSrcAlpha; data.blendDstAlpha = material.blendDstAlpha; data.blendEquationAlpha = material.blendEquationAlpha;
- data.colorWrite = material.colorWrite;
- data.depthWrite = material.depthWrite; data.depthTest = material.depthTest; data.depthFunc = material.depthFunc;
- data.stencilWrite = material.stencilWrite; data.stencilFunc = material.stencilFunc;
- data.stencilFail = material.stencilFail; data.stencilZFail = material.stencilZFail; data.stencilZPass = material.stencilZPass;
- data.stencilFuncMask = material.stencilFuncMask; data.stencilWriteMask = material.stencilWriteMask;
- data.side = material.side; data.alphaToCoverage = material.alphaToCoverage;
- data.sampleCount = sampleCount;
- data.colorSpace = colorSpace;
- data.colorFormat = colorFormat;
- data.depthStencilFormat = depthStencilFormat;
- data.primitiveTopology = primitiveTopology;
- data.frontFaceCW = frontFaceCW;
- data.clippingContextCacheKey = renderObject.clippingContextCacheKey;
- needsUpdate = true;
- }
- return needsUpdate;
- }
- /**
- * Returns a cache key that is used to identify render pipelines.
- *
- * @param {RenderObject} renderObject - The render object.
- * @return {string} The cache key.
- */
- getRenderCacheKey( renderObject ) {
- const { object, material } = renderObject;
- const utils = this.utils;
- const renderContext = renderObject.context;
- // meshes with negative scale have a different frontFace render pipeline
- // descriptor value so the following must be honored in the cache key
- const frontFaceCW = ( object.isMesh && object.matrixWorld.determinantAffine() < 0 );
- return [
- material.transparent, material.blending, material.premultipliedAlpha,
- material.blendSrc, material.blendDst, material.blendEquation,
- material.blendSrcAlpha, material.blendDstAlpha, material.blendEquationAlpha,
- material.colorWrite,
- material.depthWrite, material.depthTest, material.depthFunc,
- material.stencilWrite, material.stencilFunc,
- material.stencilFail, material.stencilZFail, material.stencilZPass,
- material.stencilFuncMask, material.stencilWriteMask,
- material.side,
- frontFaceCW,
- utils.getSampleCountRenderContext( renderContext ),
- utils.getCurrentColorSpace( renderContext ), utils.getCurrentColorFormat( renderContext ), utils.getCurrentDepthStencilFormat( renderContext ),
- utils.getPrimitiveTopology( object, material ),
- renderObject.getGeometryCacheKey(),
- renderObject.clippingContextCacheKey
- ].join();
- }
- // textures
- /**
- * Updates a GPU sampler for the given texture.
- *
- * @param {Texture} texture - The texture to update the sampler for.
- * @param {TextureNode} textureNode - The texture node to update the sampler with.
- * @return {string} The current sampler key.
- */
- updateSampler( texture, textureNode ) {
- return this.textureUtils.updateSampler( texture, textureNode );
- }
- /**
- * Creates a default texture for the given texture that can be used
- * as a placeholder until the actual texture is ready for usage.
- *
- * @param {Texture} texture - The texture to create a default texture for.
- * @return {boolean} Whether the sampler has been updated or not.
- */
- createDefaultTexture( texture ) {
- return this.textureUtils.createDefaultTexture( texture );
- }
- /**
- * Defines a texture on the GPU for the given texture object.
- *
- * @param {Texture} texture - The texture.
- * @param {Object} [options={}] - Optional configuration parameter.
- */
- createTexture( texture, options ) {
- this.textureUtils.createTexture( texture, options );
- }
- /**
- * Uploads the updated texture data to the GPU.
- *
- * @param {Texture} texture - The texture.
- * @param {Object} [options={}] - Optional configuration parameter.
- */
- updateTexture( texture, options ) {
- this.textureUtils.updateTexture( texture, options );
- }
- /**
- * Generates mipmaps for the given texture.
- *
- * @param {Texture} texture - The texture.
- */
- generateMipmaps( texture ) {
- this.textureUtils.generateMipmaps( texture );
- }
- /**
- * Destroys the GPU data for the given texture object.
- *
- * @param {Texture} texture - The texture.
- * @param {boolean} [isDefaultTexture=false] - Whether the texture uses a default GPU texture or not.
- */
- destroyTexture( texture, isDefaultTexture = false ) {
- this.textureUtils.destroyTexture( texture, isDefaultTexture );
- }
- /**
- * Returns texture data as a typed array.
- *
- * @async
- * @param {Texture} texture - The texture to copy.
- * @param {number} x - The x coordinate of the copy origin.
- * @param {number} y - The y coordinate of the copy origin.
- * @param {number} width - The width of the copy.
- * @param {number} height - The height of the copy.
- * @param {number} faceIndex - The face index.
- * @return {Promise<TypedArray>} A Promise that resolves with a typed array when the copy operation has finished.
- */
- async copyTextureToBuffer( texture, x, y, width, height, faceIndex ) {
- return this.textureUtils.copyTextureToBuffer( texture, x, y, width, height, faceIndex );
- }
- /**
- * Inits a time stamp query for the given render context.
- *
- * @param {string} type - The type of the timestamp query (e.g. 'render', 'compute').
- * @param {number} uid - Unique id for the context (e.g. render context id).
- * @param {Object} descriptor - The query descriptor.
- */
- initTimestampQuery( type, uid, descriptor ) {
- if ( ! this.trackTimestamp ) return;
- if ( ! this.timestampQueryPool[ type ] ) {
- // TODO: Variable maxQueries?
- this.timestampQueryPool[ type ] = new WebGPUTimestampQueryPool( this.device, type, 2048 );
- }
- const timestampQueryPool = this.timestampQueryPool[ type ];
- const baseOffset = timestampQueryPool.allocateQueriesForContext( uid );
- _renderPassTimestampWrites.querySet = timestampQueryPool.querySet;
- _renderPassTimestampWrites.beginningOfPassWriteIndex = baseOffset;
- _renderPassTimestampWrites.endOfPassWriteIndex = baseOffset + 1;
- descriptor.timestampWrites = _renderPassTimestampWrites;
- }
- // node builder
- /**
- * Returns a node builder for the given render object.
- *
- * @param {RenderObject} object - The render object.
- * @param {Renderer} renderer - The renderer.
- * @return {WGSLNodeBuilder} The node builder.
- */
- createNodeBuilder( object, renderer ) {
- return new WGSLNodeBuilder( object, renderer );
- }
- // program
- /**
- * Creates a shader program from the given programmable stage.
- *
- * @param {ProgrammableStage} program - The programmable stage.
- */
- createProgram( program ) {
- const programGPU = this.get( program );
- _shaderModuleDescriptor.label = program.stage + ( program.name !== '' ? `_${ program.name }` : '' );
- _shaderModuleDescriptor.code = program.code;
- programGPU.module = {
- module: this.device.createShaderModule( _shaderModuleDescriptor ),
- entryPoint: 'main'
- };
- _shaderModuleDescriptor.reset();
- }
- /**
- * Destroys the shader program of the given programmable stage.
- *
- * @param {ProgrammableStage} program - The programmable stage.
- */
- destroyProgram( program ) {
- this.delete( program );
- }
- // pipelines
- /**
- * Creates a render pipeline for the given render object.
- *
- * @param {RenderObject} renderObject - The render object.
- * @param {Array<Promise>} promises - An array of compilation promises which are used in `compileAsync()`.
- */
- createRenderPipeline( renderObject, promises ) {
- this.pipelineUtils.createRenderPipeline( renderObject, promises );
- }
- /**
- * Creates a compute pipeline for the given compute node.
- *
- * @param {ComputePipeline} computePipeline - The compute pipeline.
- * @param {Array<BindGroup>} bindings - The bindings.
- */
- createComputePipeline( computePipeline, bindings ) {
- this.pipelineUtils.createComputePipeline( computePipeline, bindings );
- }
- /**
- * Prepares the state for encoding render bundles.
- *
- * @param {RenderContext} renderContext - The render context.
- */
- beginBundle( renderContext ) {
- const renderContextData = this.get( renderContext );
- renderContextData._currentPass = renderContextData.currentPass;
- renderContextData._currentSets = renderContextData.currentSets;
- renderContextData.currentSets = { attributes: {}, bindingGroups: [], pipeline: null, index: null };
- renderContextData.currentPass = this.pipelineUtils.createBundleEncoder( renderContext );
- }
- /**
- * After processing render bundles this method finalizes related work.
- *
- * @param {RenderContext} renderContext - The render context.
- * @param {RenderBundle} bundle - The render bundle.
- */
- finishBundle( renderContext, bundle ) {
- const renderContextData = this.get( renderContext );
- const bundleEncoder = renderContextData.currentPass;
- const bundleGPU = bundleEncoder.finish();
- this.get( bundle ).bundleGPU = bundleGPU;
- // restore render pass state
- renderContextData.currentSets = renderContextData._currentSets;
- renderContextData.currentPass = renderContextData._currentPass;
- }
- /**
- * Adds a render bundle to the render context data.
- *
- * @param {RenderContext} renderContext - The render context.
- * @param {RenderBundle} bundle - The render bundle to add.
- */
- addBundle( renderContext, bundle ) {
- const renderContextData = this.get( renderContext );
- renderContextData.renderBundles.push( this.get( bundle ).bundleGPU );
- }
- // bindings
- /**
- * Creates a uniform buffer.
- *
- * @param {Buffer} uniformBuffer - The uniform buffer.
- */
- createUniformBuffer( uniformBuffer ) {
- const uniformBufferData = this.get( uniformBuffer );
- if ( uniformBufferData.buffer === undefined ) {
- const byteLength = uniformBuffer.byteLength;
- const usage = GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST;
- const visibilities = [];
- if ( uniformBuffer.visibility & GPUShaderStage.VERTEX ) {
- visibilities.push( 'vertex' );
- }
- if ( uniformBuffer.visibility & GPUShaderStage.FRAGMENT ) {
- visibilities.push( 'fragment' );
- }
- if ( uniformBuffer.visibility & GPUShaderStage.COMPUTE ) {
- visibilities.push( 'compute' );
- }
- const bufferVisibility = `(${visibilities.join( ',' )})`;
- _bufferDescriptor.label = `bindingBuffer${uniformBuffer.id}_${uniformBuffer.name}_${bufferVisibility}`;
- _bufferDescriptor.size = byteLength;
- _bufferDescriptor.usage = usage;
- const bufferGPU = this.device.createBuffer( _bufferDescriptor );
- _bufferDescriptor.reset();
- uniformBufferData.buffer = bufferGPU;
- }
- }
- /**
- * Destroys the GPU data for the given uniform buffer.
- *
- * @param {Buffer} uniformBuffer - The uniform buffer.
- */
- destroyUniformBuffer( uniformBuffer ) {
- const uniformBufferData = this.get( uniformBuffer );
- uniformBufferData.buffer.destroy();
- this.delete( uniformBuffer );
- }
- /**
- * Creates bindings from the given bind group definition.
- *
- * @param {BindGroup} bindGroup - The bind group.
- * @param {Array<BindGroup>} bindings - Array of bind groups.
- * @param {number} cacheIndex - The cache index.
- * @param {number} version - The version.
- */
- createBindings( bindGroup, bindings, cacheIndex, version ) {
- this.bindingUtils.createBindings( bindGroup, bindings, cacheIndex, version );
- }
- /**
- * Updates the given bind group definition.
- *
- * @param {BindGroup} bindGroup - The bind group.
- * @param {Array<BindGroup>} bindings - Array of bind groups.
- * @param {number} cacheIndex - The cache index.
- * @param {number} version - The version.
- */
- updateBindings( bindGroup, bindings, cacheIndex, version ) {
- this.bindingUtils.createBindings( bindGroup, bindings, cacheIndex, version );
- }
- /**
- * Updates a buffer binding.
- *
- * @param {Buffer} binding - The buffer binding to update.
- */
- updateBinding( binding ) {
- this.bindingUtils.updateBinding( binding );
- }
- /**
- * Delete data associated with the current bind group.
- *
- * @param {BindGroup} bindGroup - The bind group.
- */
- deleteBindGroupData( bindGroup ) {
- this.bindingUtils.deleteBindGroupData( bindGroup );
- }
- // attributes
- /**
- * Creates the buffer of an indexed shader attribute.
- *
- * @param {BufferAttribute} attribute - The indexed buffer attribute.
- */
- createIndexAttribute( attribute ) {
- let usage = GPUBufferUsage.INDEX | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST;
- if ( attribute.isStorageBufferAttribute || attribute.isStorageInstancedBufferAttribute ) {
- usage |= GPUBufferUsage.STORAGE;
- }
- this.attributeUtils.createAttribute( attribute, usage );
- }
- /**
- * Creates the GPU buffer of a shader attribute.
- *
- * @param {BufferAttribute} attribute - The buffer attribute.
- */
- createAttribute( attribute ) {
- this.attributeUtils.createAttribute( attribute, GPUBufferUsage.VERTEX | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST );
- }
- /**
- * Creates the GPU buffer of a storage attribute.
- *
- * @param {BufferAttribute} attribute - The buffer attribute.
- */
- createStorageAttribute( attribute ) {
- this.attributeUtils.createAttribute( attribute, GPUBufferUsage.STORAGE | GPUBufferUsage.VERTEX | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST );
- }
- /**
- * Creates the GPU buffer of an indirect storage attribute.
- *
- * @param {BufferAttribute} attribute - The buffer attribute.
- */
- createIndirectStorageAttribute( attribute ) {
- this.attributeUtils.createAttribute( attribute, GPUBufferUsage.STORAGE | GPUBufferUsage.INDIRECT | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST );
- }
- /**
- * Updates the GPU buffer of a shader attribute.
- *
- * @param {BufferAttribute} attribute - The buffer attribute to update.
- */
- updateAttribute( attribute ) {
- this.attributeUtils.updateAttribute( attribute );
- }
- /**
- * Destroys the GPU buffer of a shader attribute.
- *
- * @param {BufferAttribute} attribute - The buffer attribute to destroy.
- */
- destroyAttribute( attribute ) {
- this.attributeUtils.destroyAttribute( attribute );
- }
- // canvas
- /**
- * Triggers an update of the default render pass descriptor.
- */
- updateSize() {
- this.delete( this.renderer.getCanvasTarget() );
- }
- // utils public
- /**
- * Checks if the given feature is supported by the backend.
- *
- * @param {string} name - The feature's name.
- * @return {boolean} Whether the feature is supported or not.
- */
- hasFeature( name ) {
- if ( GPUFeatureMap[ name ] !== undefined ) name = GPUFeatureMap[ name ];
- return this.device.features.has( name );
- }
- /**
- * Copies data of the given source texture to the given destination texture.
- *
- * @param {Texture} srcTexture - The source texture.
- * @param {Texture} dstTexture - The destination texture.
- * @param {?(Box3|Box2)} [srcRegion=null] - The region of the source texture to copy.
- * @param {?(Vector2|Vector3)} [dstPosition=null] - The destination position of the copy.
- * @param {number} [srcLevel=0] - The mipmap level to copy.
- * @param {number} [dstLevel=0] - The destination mip level to copy to.
- */
- copyTextureToTexture( srcTexture, dstTexture, srcRegion = null, dstPosition = null, srcLevel = 0, dstLevel = 0 ) {
- let dstX = 0;
- let dstY = 0;
- let dstZ = 0;
- let srcX = 0;
- let srcY = 0;
- let srcZ = 0;
- let srcWidth = srcTexture.image.width;
- let srcHeight = srcTexture.image.height;
- let srcDepth = 1;
- if ( srcRegion !== null ) {
- if ( srcRegion.isBox3 === true ) {
- srcX = srcRegion.min.x;
- srcY = srcRegion.min.y;
- srcZ = srcRegion.min.z;
- srcWidth = srcRegion.max.x - srcRegion.min.x;
- srcHeight = srcRegion.max.y - srcRegion.min.y;
- srcDepth = srcRegion.max.z - srcRegion.min.z;
- } else {
- // Assume it's a Box2
- srcX = srcRegion.min.x;
- srcY = srcRegion.min.y;
- srcWidth = srcRegion.max.x - srcRegion.min.x;
- srcHeight = srcRegion.max.y - srcRegion.min.y;
- srcDepth = 1;
- }
- }
- if ( dstPosition !== null ) {
- dstX = dstPosition.x;
- dstY = dstPosition.y;
- dstZ = dstPosition.z || 0;
- }
- _commandEncoderDescriptor.label = 'copyTextureToTexture_' + srcTexture.id + '_' + dstTexture.id;
- const encoder = this.device.createCommandEncoder( _commandEncoderDescriptor );
- _commandEncoderDescriptor.reset();
- const sourceGPU = this.get( srcTexture ).texture;
- const destinationGPU = this.get( dstTexture ).texture;
- _texelCopyTextureInfoSrc.texture = sourceGPU;
- _texelCopyTextureInfoSrc.mipLevel = srcLevel;
- _texelCopyTextureInfoSrc.origin.x = srcX;
- _texelCopyTextureInfoSrc.origin.y = srcY;
- _texelCopyTextureInfoSrc.origin.z = srcZ;
- _texelCopyTextureInfoDst.texture = destinationGPU;
- _texelCopyTextureInfoDst.mipLevel = dstLevel;
- _texelCopyTextureInfoDst.origin.x = dstX;
- _texelCopyTextureInfoDst.origin.y = dstY;
- _texelCopyTextureInfoDst.origin.z = dstZ;
- _extent3D.width = srcWidth;
- _extent3D.height = srcHeight;
- _extent3D.depthOrArrayLayers = srcDepth;
- encoder.copyTextureToTexture(
- _texelCopyTextureInfoSrc,
- _texelCopyTextureInfoDst,
- _extent3D
- );
- _texelCopyTextureInfoSrc.reset();
- _texelCopyTextureInfoDst.reset();
- _extent3D.reset();
- submit( this.device, encoder.finish() );
- if ( dstLevel === 0 && dstTexture.generateMipmaps ) {
- this.textureUtils.generateMipmaps( dstTexture );
- }
- }
- /**
- * Copies the current bound framebuffer to the given texture.
- *
- * @param {Texture} texture - The destination texture.
- * @param {RenderContext} renderContext - The render context.
- * @param {Vector4} rectangle - A four dimensional vector defining the origin and dimension of the copy.
- */
- copyFramebufferToTexture( texture, renderContext, rectangle ) {
- const renderContextData = this.get( renderContext );
- let sourceGPU = null;
- if ( renderContext.renderTarget ) {
- if ( texture.isDepthTexture ) {
- sourceGPU = this.get( renderContext.depthTexture ).texture;
- } else {
- sourceGPU = this.get( renderContext.textures[ 0 ] ).texture;
- }
- } else {
- if ( texture.isDepthTexture ) {
- sourceGPU = this.textureUtils.getDepthBuffer( renderContext.depth, renderContext.stencil );
- } else {
- sourceGPU = this.context.getCurrentTexture();
- }
- }
- const destinationGPU = this.get( texture ).texture;
- if ( sourceGPU.format !== destinationGPU.format ) {
- error( 'WebGPUBackend: copyFramebufferToTexture: Source and destination formats do not match.', sourceGPU.format, destinationGPU.format );
- return;
- }
- let encoder;
- if ( renderContextData.currentPass ) {
- renderContextData.currentPass.end();
- encoder = renderContextData.encoder;
- } else {
- _commandEncoderDescriptor.label = 'copyFramebufferToTexture_' + texture.id;
- encoder = this.device.createCommandEncoder( _commandEncoderDescriptor );
- _commandEncoderDescriptor.reset();
- }
- _texelCopyTextureInfoSrc.texture = sourceGPU;
- _texelCopyTextureInfoSrc.origin.x = rectangle.x;
- _texelCopyTextureInfoSrc.origin.y = rectangle.y;
- _texelCopyTextureInfoDst.texture = destinationGPU;
- _extent3D.width = rectangle.z;
- _extent3D.height = rectangle.w;
- encoder.copyTextureToTexture(
- _texelCopyTextureInfoSrc,
- _texelCopyTextureInfoDst,
- _extent3D
- );
- _texelCopyTextureInfoSrc.reset();
- _texelCopyTextureInfoDst.reset();
- _extent3D.reset();
- // mipmaps must be genereated with the same encoder otherwise the copied texture data
- // might be out-of-sync, see #31768
- if ( texture.generateMipmaps ) {
- this.textureUtils.generateMipmaps( texture, encoder );
- }
- if ( renderContextData.currentPass ) {
- const { descriptor } = renderContextData;
- for ( let i = 0; i < descriptor.colorAttachments.length; i ++ ) {
- descriptor.colorAttachments[ i ].loadOp = GPULoadOp.Load;
- }
- if ( renderContext.depth ) descriptor.depthStencilAttachment.depthLoadOp = GPULoadOp.Load;
- if ( renderContext.stencil ) descriptor.depthStencilAttachment.stencilLoadOp = GPULoadOp.Load;
- renderContextData.currentPass = encoder.beginRenderPass( descriptor );
- renderContextData.currentSets = { attributes: {}, bindingGroups: [], pipeline: null, index: null };
- if ( renderContext.viewport ) {
- this.updateViewport( renderContext );
- }
- if ( renderContext.scissor ) {
- this.updateScissor( renderContext );
- }
- } else {
- submit( this.device, encoder.finish() );
- }
- }
- /**
- * Checks if the given compatibility is supported by the backend.
- *
- * @param {string} name - The compatibility name.
- * @return {boolean} Whether the compatibility is supported or not.
- */
- hasCompatibility( name ) {
- if ( this._compatibility[ name ] !== undefined ) {
- return this._compatibility[ name ];
- }
- return super.hasCompatibility( name );
- }
- dispose() {
- this.bindingUtils.dispose();
- this.textureUtils.dispose();
- if ( this.occludedResolveCache ) {
- for ( const buffer of this.occludedResolveCache.values() ) {
- buffer.destroy();
- }
- this.occludedResolveCache.clear();
- }
- if ( this.timestampQueryPool ) {
- for ( const queryPool of Object.values( this.timestampQueryPool ) ) {
- if ( queryPool !== null ) queryPool.dispose();
- }
- }
- if ( this.parameters.device === undefined && this.device !== null ) {
- this.device.destroy();
- }
- }
- }
- export default WebGPUBackend;
|