@forgeax/engine-rhi 0.1.2

package/src/queue-writes.ts

@@ -0,0 +1,62 @@
+/**
+ * Resolve the source range of a queue.writeBuffer call.
+ *
+ * WebGPU defines dataOffset and size in elements for typed-array sources, but
+ * in bytes for ArrayBuffer and DataView sources. Keeping that conversion in
+ * the shared RHI package makes the browser, native, and debug paths agree.
+ */
+export interface BufferWriteRange {
+  readonly sourceOffset: number;
+  readonly sourceSize: number;
+  readonly byteOffset: number;
+  readonly byteSize: number;
+  readonly bytes: Uint8Array;
+}
+
+function elementSize(data: ArrayBufferView | ArrayBuffer): number {
+  if (data instanceof ArrayBuffer) return 1;
+  const bytesPerElement = (data as ArrayBufferView & { BYTES_PER_ELEMENT?: number })
+    .BYTES_PER_ELEMENT;
+  return typeof bytesPerElement === 'number' && bytesPerElement > 0 ? bytesPerElement : 1;
+}
+
+function sourceByteView(data: ArrayBufferView | ArrayBuffer): Uint8Array {
+  return data instanceof ArrayBuffer
+    ? new Uint8Array(data)
+    : new Uint8Array(data.buffer, data.byteOffset, data.byteLength);
+}
+
+/**
+ * Return the exact source bytes selected by a queue.writeBuffer range.
+ * Undefined means the supplied element range is invalid.
+ */
+export function resolveBufferWriteRange(
+  data: ArrayBufferView | ArrayBuffer,
+  dataOffset?: number,
+  size?: number,
+): BufferWriteRange | undefined {
+  const bytes = sourceByteView(data);
+  const bytesPerElement = elementSize(data);
+  const elementCount = bytes.byteLength / bytesPerElement;
+  const sourceOffset = dataOffset ?? 0;
+  const sourceSize = size ?? elementCount - sourceOffset;
+  if (
+    !Number.isInteger(sourceOffset) ||
+    !Number.isInteger(sourceSize) ||
+    sourceOffset < 0 ||
+    sourceSize < 0 ||
+    sourceOffset > elementCount ||
+    sourceSize > elementCount - sourceOffset
+  ) {
+    return undefined;
+  }
+  const byteOffset = sourceOffset * bytesPerElement;
+  const byteSize = sourceSize * bytesPerElement;
+  return {
+    sourceOffset,
+    sourceSize,
+    byteOffset,
+    byteSize,
+    bytes: bytes.subarray(byteOffset, byteOffset + byteSize),
+  };
+}