@kitware/vtk.js 28.10.2 → 28.11.0

package/Rendering/WebGPU/StorageBuffer.js

@@ -1,14 +1,21 @@
-import macro from '../../macros.js';
+import { m as macro } from '../../macros2.js';
 import vtkWebGPUBufferManager from './BufferManager.js';
 import vtkWebGPUTypes from './Types.js';
 
-var BufferUsage = vtkWebGPUBufferManager.BufferUsage;
-var vtkErrorMacro = macro.vtkErrorMacro; // ----------------------------------------------------------------------------
+const {
+  BufferUsage
+} = vtkWebGPUBufferManager;
+const {
+  vtkErrorMacro
+} = macro;
+
+// ----------------------------------------------------------------------------
 // vtkWebGPUStorageBuffer - similar to the UniformBuffer class
 // but YOU are responsible for layout issues and alignment.
 // The order you add entries is the order they will be layed out
 // in memory. But you must follow layout rules.
 // ----------------------------------------------------------------------------
+
 // ----------------------------------------------------------------------------
 // vtkWebGPUStorageBuffer methods
 // ----------------------------------------------------------------------------
@@ -16,182 +23,149 @@ var vtkErrorMacro = macro.vtkErrorMacro; // ------------------------------------
 function vtkWebGPUStorageBuffer(publicAPI, model) {
   // Set our className
   model.classHierarchy.push('vtkWebGPUStorageBuffer');
-
-  publicAPI.addEntry = function (name, type) {
+  publicAPI.addEntry = (name, type) => {
     if (model._bufferEntryNames.has(name)) {
-      vtkErrorMacro("entry named ".concat(name, " already exists"));
+      vtkErrorMacro(`entry named ${name} already exists`);
       return;
     }
-
     model._bufferEntryNames.set(name, model.bufferEntries.length);
-
-    var sizeInBytes = vtkWebGPUTypes.getByteStrideFromShaderFormat(type);
+    const sizeInBytes = vtkWebGPUTypes.getByteStrideFromShaderFormat(type);
     model.bufferEntries.push({
-      name: name,
-      type: type,
-      sizeInBytes: sizeInBytes,
+      name,
+      type,
+      sizeInBytes,
       offset: model.sizeInBytes,
       nativeType: vtkWebGPUTypes.getNativeTypeFromShaderFormat(type)
     });
     model.sizeInBytes += sizeInBytes;
   };
-
-  publicAPI.send = function (device) {
+  publicAPI.send = device => {
     if (!model._buffer) {
-      var req = {
+      const req = {
         nativeArray: model.Float32Array,
         usage: BufferUsage.Storage,
         label: model.label
       };
       model._buffer = device.getBufferManager().getBuffer(req);
       model.bindGroupTime.modified();
-
       model._sendTime.modified();
-
       return;
     }
-
     device.getHandle().queue.writeBuffer(model._buffer.getHandle(), 0, model.arrayBuffer, 0, model.sizeInBytes * model.numberOfInstances);
-
     model._sendTime.modified();
   };
-
-  publicAPI.createView = function (type) {
+  publicAPI.createView = type => {
     if (type in model === false) {
       if (!model.arrayBuffer) {
         model.arrayBuffer = new ArrayBuffer(model.sizeInBytes * model.numberOfInstances);
       }
-
       model[type] = macro.newTypedArray(type, model.arrayBuffer);
     }
   };
-
-  publicAPI.setValue = function (name, instance, val) {
-    var idx = model._bufferEntryNames.get(name);
-
+  publicAPI.setValue = (name, instance, val) => {
+    const idx = model._bufferEntryNames.get(name);
     if (idx === undefined) {
-      vtkErrorMacro("entry named ".concat(name, " not found in UBO"));
+      vtkErrorMacro(`entry named ${name} not found in UBO`);
       return;
     }
-
-    var entry = model.bufferEntries[idx];
+    const entry = model.bufferEntries[idx];
     publicAPI.createView(entry.nativeType);
-    var view = model[entry.nativeType];
+    const view = model[entry.nativeType];
     view[(entry.offset + instance * model.sizeInBytes) / view.BYTES_PER_ELEMENT] = val;
   };
-
-  publicAPI.setArray = function (name, instance, arr) {
-    var idx = model._bufferEntryNames.get(name);
-
+  publicAPI.setArray = (name, instance, arr) => {
+    const idx = model._bufferEntryNames.get(name);
     if (idx === undefined) {
-      vtkErrorMacro("entry named ".concat(name, " not found in UBO"));
+      vtkErrorMacro(`entry named ${name} not found in UBO`);
       return;
     }
-
-    var entry = model.bufferEntries[idx];
+    const entry = model.bufferEntries[idx];
     publicAPI.createView(entry.nativeType);
-    var view = model[entry.nativeType];
-    var ioffset = (entry.offset + instance * model.sizeInBytes) / view.BYTES_PER_ELEMENT;
-
-    for (var i = 0; i < arr.length; i++) {
+    const view = model[entry.nativeType];
+    const ioffset = (entry.offset + instance * model.sizeInBytes) / view.BYTES_PER_ELEMENT;
+    for (let i = 0; i < arr.length; i++) {
       view[ioffset + i] = arr[i];
     }
   };
-
-  publicAPI.setAllInstancesFromArray = function (name, arr) {
-    var idx = model._bufferEntryNames.get(name);
-
+  publicAPI.setAllInstancesFromArray = (name, arr) => {
+    const idx = model._bufferEntryNames.get(name);
     if (idx === undefined) {
-      vtkErrorMacro("entry named ".concat(name, " not found in UBO"));
+      vtkErrorMacro(`entry named ${name} not found in UBO`);
       return;
     }
-
-    var entry = model.bufferEntries[idx];
+    const entry = model.bufferEntries[idx];
     publicAPI.createView(entry.nativeType);
-    var view = model[entry.nativeType];
-    var numComponents = arr.length / model.numberOfInstances;
-
-    for (var inst = 0; inst < model.numberOfInstances; inst++) {
-      var ioffset = (entry.offset + inst * model.sizeInBytes) / view.BYTES_PER_ELEMENT;
-
-      for (var i = 0; i < numComponents; i++) {
+    const view = model[entry.nativeType];
+    const numComponents = arr.length / model.numberOfInstances;
+    for (let inst = 0; inst < model.numberOfInstances; inst++) {
+      const ioffset = (entry.offset + inst * model.sizeInBytes) / view.BYTES_PER_ELEMENT;
+      for (let i = 0; i < numComponents; i++) {
         view[ioffset + i] = arr[inst * numComponents + i];
       }
     }
   };
-
-  publicAPI.setAllInstancesFromArrayColorToFloat = function (name, arr) {
-    var idx = model._bufferEntryNames.get(name);
-
+  publicAPI.setAllInstancesFromArrayColorToFloat = (name, arr) => {
+    const idx = model._bufferEntryNames.get(name);
     if (idx === undefined) {
-      vtkErrorMacro("entry named ".concat(name, " not found in UBO"));
+      vtkErrorMacro(`entry named ${name} not found in UBO`);
       return;
     }
-
-    var entry = model.bufferEntries[idx];
+    const entry = model.bufferEntries[idx];
     publicAPI.createView(entry.nativeType);
-    var view = model[entry.nativeType];
-    var numComponents = arr.length / model.numberOfInstances;
-
-    for (var inst = 0; inst < model.numberOfInstances; inst++) {
-      var ioffset = (entry.offset + inst * model.sizeInBytes) / view.BYTES_PER_ELEMENT;
-
-      for (var i = 0; i < numComponents; i++) {
+    const view = model[entry.nativeType];
+    const numComponents = arr.length / model.numberOfInstances;
+    for (let inst = 0; inst < model.numberOfInstances; inst++) {
+      const ioffset = (entry.offset + inst * model.sizeInBytes) / view.BYTES_PER_ELEMENT;
+      for (let i = 0; i < numComponents; i++) {
         view[ioffset + i] = arr[inst * numComponents + i] / 255.0;
       }
     }
   };
-
-  publicAPI.setAllInstancesFromArray3x3To4x4 = function (name, arr) {
-    var idx = model._bufferEntryNames.get(name);
-
+  publicAPI.setAllInstancesFromArray3x3To4x4 = (name, arr) => {
+    const idx = model._bufferEntryNames.get(name);
     if (idx === undefined) {
-      vtkErrorMacro("entry named ".concat(name, " not found in UBO"));
+      vtkErrorMacro(`entry named ${name} not found in UBO`);
       return;
     }
-
-    var entry = model.bufferEntries[idx];
+    const entry = model.bufferEntries[idx];
     publicAPI.createView(entry.nativeType);
-    var view = model[entry.nativeType];
-    var numComponents = 9;
-
-    for (var inst = 0; inst < model.numberOfInstances; inst++) {
-      var ioffset = (entry.offset + inst * model.sizeInBytes) / view.BYTES_PER_ELEMENT;
-
-      for (var j = 0; j < 3; j++) {
-        for (var i = 0; i < 3; i++) {
+    const view = model[entry.nativeType];
+    const numComponents = 9;
+    for (let inst = 0; inst < model.numberOfInstances; inst++) {
+      const ioffset = (entry.offset + inst * model.sizeInBytes) / view.BYTES_PER_ELEMENT;
+      for (let j = 0; j < 3; j++) {
+        for (let i = 0; i < 3; i++) {
           view[ioffset + j * 4 + i] = arr[inst * numComponents + j * 3 + i];
         }
       }
     }
   };
-
-  publicAPI.getSendTime = function () {
-    return model._sendTime.getMTime();
-  };
-
-  publicAPI.getShaderCode = function (binding, group) {
-    var lines = ["struct ".concat(model.label, "StructEntry\n{")];
-
-    for (var i = 0; i < model.bufferEntries.length; i++) {
-      var entry = model.bufferEntries[i];
-      lines.push("  ".concat(entry.name, ": ").concat(entry.type, ","));
+  publicAPI.getSendTime = () => model._sendTime.getMTime();
+  publicAPI.getShaderCode = (binding, group) => {
+    const lines = [`struct ${model.label}StructEntry\n{`];
+    for (let i = 0; i < model.bufferEntries.length; i++) {
+      const entry = model.bufferEntries[i];
+      lines.push(`  ${entry.name}: ${entry.type},`);
     }
-
-    lines.push("\n};\nstruct ".concat(model.label, "Struct\n{\n  values: array<").concat(model.label, "StructEntry>,\n};\n@binding(").concat(binding, ") @group(").concat(group, ") var<storage, read> ").concat(model.label, ": ").concat(model.label, "Struct;\n"));
+    lines.push(`
+};
+struct ${model.label}Struct
+{
+  values: array<${model.label}StructEntry>,
+};
+@binding(${binding}) @group(${group}) var<storage, read> ${model.label}: ${model.label}Struct;
+`);
     return lines.join('\n');
   };
-
-  publicAPI.getBindGroupEntry = function () {
-    var foo = {
+  publicAPI.getBindGroupEntry = () => {
+    const foo = {
       resource: {
         buffer: model._buffer.getHandle()
       }
     };
     return foo;
   };
-
-  publicAPI.clearData = function () {
+  publicAPI.clearData = () => {
     model.numberOfInstances = 0;
     model.sizeInBytes = 0;
     model.bufferEntries = [];
@@ -200,25 +174,30 @@ function vtkWebGPUStorageBuffer(publicAPI, model) {
     delete model.arrayBuffer;
     delete model.Float32Array;
   };
-} // ----------------------------------------------------------------------------
+}
+
+// ----------------------------------------------------------------------------
 // Object factory
 // ----------------------------------------------------------------------------
 
-
-var DEFAULT_VALUES = {
+const DEFAULT_VALUES = {
   bufferEntries: null,
   bufferEntryNames: null,
   sizeInBytes: 0,
   label: null,
   numberOfInstances: 1
-}; // ----------------------------------------------------------------------------
+};
+
+// ----------------------------------------------------------------------------
 
 function extend(publicAPI, model) {
-  var initialValues = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : {};
-  Object.assign(model, DEFAULT_VALUES, initialValues); // Build VTK API
+  let initialValues = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : {};
+  Object.assign(model, DEFAULT_VALUES, initialValues);
 
-  macro.obj(publicAPI, model); // Internal objects
+  // Build VTK API
+  macro.obj(publicAPI, model);
 
+  // Internal objects
   model._bufferEntryNames = new Map();
   model.bufferEntries = [];
   model._sendTime = {};
@@ -228,24 +207,30 @@ function extend(publicAPI, model) {
   model.bindGroupTime = {};
   macro.obj(model.bindGroupTime, {
     mtime: 0
-  }); // default SSBO desc
+  });
 
+  // default SSBO desc
   model.bindGroupLayoutEntry = model.bindGroupLayoutEntry || {
     buffer: {
       type: 'read-only-storage'
     }
   };
   macro.get(publicAPI, model, ['bindGroupTime']);
-  macro.setGet(publicAPI, model, ['device', 'bindGroupLayoutEntry', 'label', 'numberOfInstances', 'sizeInBytes']); // Object methods
+  macro.setGet(publicAPI, model, ['device', 'bindGroupLayoutEntry', 'label', 'numberOfInstances', 'sizeInBytes']);
 
+  // Object methods
   vtkWebGPUStorageBuffer(publicAPI, model);
-} // ----------------------------------------------------------------------------
+}
+
+// ----------------------------------------------------------------------------
+
+const newInstance = macro.newInstance(extend, 'vtkWebGPUStorageBuffer');
 
-var newInstance = macro.newInstance(extend, 'vtkWebGPUStorageBuffer'); // ----------------------------------------------------------------------------
+// ----------------------------------------------------------------------------
 
 var vtkWebGPUStorageBuffer$1 = {
-  newInstance: newInstance,
-  extend: extend
+  newInstance,
+  extend
 };
 
 export { vtkWebGPUStorageBuffer$1 as default, extend, newInstance };