UNPKG

@thewtex/vtk.js-esm

Version:

Visualization Toolkit for the Web

154 lines (118 loc) 5.5 kB
import macro from '../../macro.js'; import vtkViewNode from '../SceneGraph/ViewNode.js'; import { registerOverride } from './ViewNodeFactory.js'; import { j as transpose, i as identity, d as copy, g as invert, t as translate } from '../../vendor/gl-matrix/esm/mat4.js'; // vtkWebGPUActor methods // ---------------------------------------------------------------------------- function vtkWebGPUActor(publicAPI, model) { // Set our className model.classHierarchy.push('vtkWebGPUActor'); // Builds myself. publicAPI.buildPass = function (prepass) { if (prepass) { model.WebGPURenderer = publicAPI.getFirstAncestorOfType('vtkWebGPURenderer'); model.WebGPURenderWindow = model.WebGPURenderer.getFirstAncestorOfType('vtkWebGPURenderWindow'); if (model.propID === undefined) { model.propID = model.WebGPURenderWindow.getUniquePropID(); } publicAPI.prepareNodes(); publicAPI.addMissingNode(model.renderable.getMapper()); publicAPI.removeUnusedNodes(); } }; // we draw textures, then mapper, then post pass textures publicAPI.traverseOpaquePass = function (renderPass) { if (!model.renderable || !model.renderable.getVisibility() || !model.renderable.getIsOpaque() || model.WebGPURenderer.getSelector() && !model.renderable.getPickable()) { return; } publicAPI.apply(renderPass, true); if (model.children[0]) { model.children[0].traverse(renderPass); } publicAPI.apply(renderPass, false); }; // we draw textures, then mapper, then post pass textures publicAPI.traverseTranslucentPass = function (renderPass) { if (!model.renderable || !model.renderable.getVisibility() || model.renderable.getIsOpaque() || model.WebGPURenderer.getSelector() && !model.renderable.getPickable()) { return; } publicAPI.apply(renderPass, true); if (model.children[0]) { model.children[0].traverse(renderPass); } publicAPI.apply(renderPass, false); }; publicAPI.queryPass = function (prepass, renderPass) { if (prepass) { if (!model.renderable || !model.renderable.getVisibility()) { return; } if (model.renderable.getIsOpaque()) { renderPass.incrementOpaqueActorCount(); } else { renderPass.incrementTranslucentActorCount(); } } }; publicAPI.getBufferShift = function (wgpuRen) { publicAPI.getKeyMatrices(wgpuRen); return model.bufferShift; }; publicAPI.getKeyMatrices = function (wgpuRen) { // has the actor or stabilization center changed? if (Math.max(model.renderable.getMTime(), wgpuRen.getStabilizedTime()) > model.keyMatricesTime.getMTime()) { model.renderable.computeMatrix(); var mcwc = model.renderable.getMatrix(); // compute the net shift var center = wgpuRen.getStabilizedCenterByReference(); model.bufferShift[0] = mcwc[3] - center[0]; model.bufferShift[1] = mcwc[7] - center[1]; model.bufferShift[2] = mcwc[11] - center[2]; transpose(model.keyMatrices.bcwc, mcwc); if (model.renderable.getIsIdentity()) { identity(model.keyMatrices.normalMatrix); } else { // we use bcwc BEFORE the translate below (just to get transposed mcvc) copy(model.keyMatrices.normalMatrix, model.keyMatrices.bcwc); // zero out translation model.keyMatrices.normalMatrix[3] = 0.0; model.keyMatrices.normalMatrix[7] = 0.0; model.keyMatrices.normalMatrix[11] = 0.0; invert(model.keyMatrices.normalMatrix, model.keyMatrices.normalMatrix); transpose(model.keyMatrices.normalMatrix, model.keyMatrices.normalMatrix); } // only meed the buffer shift to get to world translate(model.keyMatrices.bcwc, model.keyMatrices.bcwc, [-model.bufferShift[0], -model.bufferShift[1], -model.bufferShift[2]]); // to get to stabilized we also need the center translate(model.keyMatrices.bcsc, model.keyMatrices.bcwc, [-center[0], -center[1], -center[2]]); model.keyMatricesTime.modified(); } return model.keyMatrices; }; } // ---------------------------------------------------------------------------- // Object factory // ---------------------------------------------------------------------------- var DEFAULT_VALUES = { keyMatricesTime: null, keyMatrices: null, propID: undefined, bufferShift: undefined }; // ---------------------------------------------------------------------------- function extend(publicAPI, model) { var initialValues = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : {}; Object.assign(model, DEFAULT_VALUES, initialValues); // Inheritance vtkViewNode.extend(publicAPI, model, initialValues); model.keyMatricesTime = {}; macro.obj(model.keyMatricesTime, { mtime: 0 }); model.keyMatrices = { normalMatrix: new Float64Array(16), bcwc: new Float64Array(16), bcsc: new Float64Array(16) }; macro.get(publicAPI, model, ['propID', 'keyMatricesTime']); model.bufferShift = [0, 0, 0, 0]; // Object methods vtkWebGPUActor(publicAPI, model); } // ---------------------------------------------------------------------------- var newInstance = macro.newInstance(extend); // ---------------------------------------------------------------------------- var index = { newInstance: newInstance, extend: extend }; // Register ourself to WebGPU backend if imported registerOverride('vtkActor', newInstance); export default index; export { extend, newInstance };