@kitware/vtk.js 24.5.2 → 24.5.5

package/Rendering/WebGPU/BufferManager.js

@@ -1,6 +1,7 @@
 import _defineProperty from '@babel/runtime/helpers/defineProperty';
-import { newInstance as newInstance$1, obj, setGet, newTypedArray, vtkDebugMacro as vtkDebugMacro$1 } from '../../macros.js';
+import { newInstance as newInstance$1, obj, setGet, vtkErrorMacro as vtkErrorMacro$1, newTypedArray, vtkDebugMacro as vtkDebugMacro$1 } from '../../macros.js';
 import { j as cross, l as normalize } from '../../Common/Core/Math/index.js';
+import vtkDataArray from '../../Common/Core/DataArray.js';
 import vtkWebGPUBuffer from './Buffer.js';
 import vtkWebGPUTypes from './Types.js';
 import vtkProperty from '../Core/Property.js';
@@ -12,7 +13,9 @@ function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { va
 var BufferUsage = Constants.BufferUsage,
     PrimitiveTypes = Constants.PrimitiveTypes;
 var Representation = vtkProperty.Representation;
-var vtkDebugMacro = vtkDebugMacro$1; // the webgpu constants all show up as undefined
+var vtkDebugMacro = vtkDebugMacro$1,
+    vtkErrorMacro = vtkErrorMacro$1;
+var VtkDataTypes = vtkDataArray.VtkDataTypes; // the webgpu constants all show up as undefined
 
 /* eslint-disable no-undef */
 // ----------------------------------------------------------------------------
@@ -62,6 +65,65 @@ var cellCounters = {
   }
 };
 
+function _getFormatForDataArray(dataArray) {
+  var format;
+
+  switch (dataArray.getDataType()) {
+    case VtkDataTypes.UNSIGNED_CHAR:
+      format = 'uint8';
+      break;
+
+    case VtkDataTypes.FLOAT:
+      format = 'float32';
+      break;
+
+    case VtkDataTypes.UNSIGNED_INT:
+      format = 'uint32';
+      break;
+
+    case VtkDataTypes.INT:
+      format = 'sint32';
+      break;
+
+    case VtkDataTypes.DOUBLE:
+      format = 'float32';
+      break;
+
+    case VtkDataTypes.UNSIGNED_SHORT:
+      format = 'uint16';
+      break;
+
+    case VtkDataTypes.SHORT:
+      format = 'sin16';
+      break;
+
+    default:
+      format = 'float32';
+      break;
+  }
+
+  switch (dataArray.getNumberOfComponents()) {
+    case 2:
+      format += 'x2';
+      break;
+
+    case 3:
+      // only 32bit types support x3
+      if (!format.contains('32')) {
+        vtkErrorMacro("unsupported x3 type for ".concat(format));
+      }
+
+      format += 'x3';
+      break;
+
+    case 4:
+      format += 'x4';
+      break;
+  }
+
+  return format;
+}
+
 function getPrimitiveName(primType) {
   switch (primType) {
     case PrimitiveTypes.Points:
@@ -443,27 +505,34 @@ function vtkWebGPUBufferManager(publicAPI, model) {
   } // is the buffer already present?
 
 
-  publicAPI.hasBuffer = function (req) {
-    if (req.owner) {
-      // if a matching buffer already exists then return true
-      var hash = req.time + req.format + req.usage + req.hash;
-      return model.device.hasCachedObject(req.owner, hash);
-    }
-
-    return false;
+  publicAPI.hasBuffer = function (hash) {
+    return model.device.hasCachedObject(hash);
   };
 
   publicAPI.getBuffer = function (req) {
     // if we have a source the get/create/cache the buffer
-    if (req.owner) {
-      // if a matching buffer already exists then return it
-      var hash = req.time + req.format + req.usage + req.hash;
-      return model.device.getCachedObject(req.owner, hash, _createBuffer, req);
+    if (req.hash) {
+      return model.device.getCachedObject(req.hash, _createBuffer, req);
     }
 
     return _createBuffer(req);
   };
 
+  publicAPI.getBufferForPointArray = function (dataArray, cells, primitiveType, representation) {
+    var format = _getFormatForDataArray(dataArray);
+
+    var buffRequest = {
+      hash: "PA".concat(representation, "P").concat(primitiveType, "D").concat(dataArray.getMTime(), "C").concat(cells.getMTime()).concat(format),
+      usage: BufferUsage.PointArray,
+      format: format,
+      dataArray: dataArray,
+      cells: cells,
+      primitiveType: primitiveType,
+      representation: representation
+    };
+    return publicAPI.getBuffer(buffRequest);
+  };
+
   publicAPI.getFullScreenQuadBuffer = function () {
     if (model.fullScreenQuadBuffer) {
       return model.fullScreenQuadBuffer;