UNPKG

@kitware/vtk.js

Version:

Visualization Toolkit for the Web

58 lines (44 loc) 2.27 kB
import { m as macro } from '../../macros2.js'; import vtkWebGPUShaderCache from './ShaderCache.js'; import vtkWebGPUSimpleMapper from './SimpleMapper.js'; // ---------------------------------------------------------------------------- // vtkWebGPUFullScreenQuad methods // ---------------------------------------------------------------------------- function vtkWebGPUFullScreenQuad(publicAPI, model) { // Set our className model.classHierarchy.push('vtkWebGPUFullScreenQuad'); publicAPI.replaceShaderPosition = (hash, pipeline, vertexInput) => { const vDesc = pipeline.getShaderDescription('vertex'); vDesc.addBuiltinOutput('vec4<f32>', '@builtin(position) Position'); vDesc.addOutput('vec4<f32>', 'vertexVC'); let code = vDesc.getCode(); code = vtkWebGPUShaderCache.substitute(code, '//VTK::Position::Impl', ['output.tcoordVS = vec2<f32>(vertexBC.x * 0.5 + 0.5, 1.0 - vertexBC.y * 0.5 - 0.5);', 'output.Position = vec4<f32>(vertexBC, 1.0);', 'output.vertexVC = vec4<f32>(vertexBC, 1);']).result; vDesc.setCode(code); }; model.shaderReplacements.set('replaceShaderPosition', publicAPI.replaceShaderPosition); publicAPI.updateBuffers = () => { const buff = model.device.getBufferManager().getFullScreenQuadBuffer(); model.vertexInput.addBuffer(buff, ['vertexBC']); model.numberOfVertices = 6; }; } // ---------------------------------------------------------------------------- // Object factory // ---------------------------------------------------------------------------- const DEFAULT_VALUES = {}; // ---------------------------------------------------------------------------- function extend(publicAPI, model, initialValues = {}) { Object.assign(model, DEFAULT_VALUES, initialValues); // Inheritance vtkWebGPUSimpleMapper.extend(publicAPI, model, initialValues); // Object methods vtkWebGPUFullScreenQuad(publicAPI, model); } // ---------------------------------------------------------------------------- const newInstance = macro.newInstance(extend, 'vtkWebGPUFullScreenQuad'); // ---------------------------------------------------------------------------- var vtkWebGPUFullScreenQuad$1 = { newInstance, extend }; export { vtkWebGPUFullScreenQuad$1 as default, extend, newInstance };