@nayan-ui/engine 0.0.0-stage → 1.0.0-beta.4

package/lib/module/world/TransformBatch.js

@@ -0,0 +1,128 @@
+"use strict";
+
+import { resolveColor } from "./color.js";
+export const TRANSFORM_STRIDE = 16;
+
+/**
+ * Explicit draw-only updates, applied in one native call by `flush()`. `world.update` flushes
+ * `world.transforms` before stepping; flush yourself before reading positions or animating. Records and scratch storage are reused; values are
+ * copied when queued, never retained by reference. The scalar methods (`position`, `rotation`,
+ * `scale`, `color`, `transform`) allocate nothing; consecutive calls for the same entity share one
+ * record.
+ *
+ * ```ts
+ * const batch = world.transforms;
+ * for (const p of pieces) batch.transform(p.entity, p.x, 0, p.z, 0, p.qy, 0, p.qw);
+ * batch.flush();
+ * ```
+ */
+export class TransformBatch {
+  data = new Float64Array(TRANSFORM_STRIDE * 32);
+  count = 0;
+  constructor(commit) {
+    this.commit = commit;
+  }
+  record(entity, flags) {
+    // Merge into the previous record when it updates the same entity (the layout is fixed per field).
+    if (this.count > 0) {
+      const last = (this.count - 1) * TRANSFORM_STRIDE;
+      if (this.data[last] === entity) {
+        this.data[last + 1] = this.data[last + 1] | flags;
+        return last;
+      }
+    }
+    const offset = this.count * TRANSFORM_STRIDE;
+    if (offset + TRANSFORM_STRIDE > this.data.length) {
+      const grown = new Float64Array(this.data.length * 2);
+      grown.set(this.data);
+      this.data = grown;
+    }
+    this.count++;
+    this.data[offset] = entity;
+    this.data[offset + 1] = flags;
+    return offset;
+  }
+  position(entity, x, y, z) {
+    const offset = this.record(entity, 2);
+    this.data[offset + 2] = x;
+    this.data[offset + 3] = y;
+    this.data[offset + 4] = z;
+  }
+
+  /** Rotation quaternion [x, y, z, w]. */
+  rotation(entity, x, y, z, w) {
+    const offset = this.record(entity, 4);
+    this.data[offset + 5] = x;
+    this.data[offset + 6] = y;
+    this.data[offset + 7] = z;
+    this.data[offset + 8] = w;
+  }
+
+  /** Uniform scale, or per axis. */
+  scale(entity, x, y = x, z = x) {
+    const offset = this.record(entity, 8);
+    this.data[offset + 9] = x;
+    this.data[offset + 10] = y;
+    this.data[offset + 11] = z;
+  }
+
+  /** RGBA color; alpha defaults to 1. */
+  color(entity, r, g, b, a = 1) {
+    const offset = this.record(entity, 16);
+    this.data[offset + 12] = r;
+    this.data[offset + 13] = g;
+    this.data[offset + 14] = b;
+    this.data[offset + 15] = a;
+  }
+
+  /** Position and rotation together (the common per-frame update for a moving piece). */
+  transform(entity, x, y, z, qx, qy, qz, qw) {
+    const offset = this.record(entity, 6);
+    this.data[offset + 2] = x;
+    this.data[offset + 3] = y;
+    this.data[offset + 4] = z;
+    this.data[offset + 5] = qx;
+    this.data[offset + 6] = qy;
+    this.data[offset + 7] = qz;
+    this.data[offset + 8] = qw;
+  }
+
+  /** Queued records (merged per entity). */
+  get size() {
+    return this.count;
+  }
+  set(entity, options) {
+    const {
+      position,
+      rotation,
+      scale,
+      color
+    } = options;
+    const flags = (position ? 2 : 0) | (rotation ? 4 : 0) | (scale !== undefined ? 8 : 0) | (color ? 16 : 0);
+    if (!flags) return;
+    const offset = this.record(entity, flags);
+    if (position) this.data.set(position, offset + 2);
+    if (rotation) this.data.set(rotation, offset + 5);
+    if (typeof scale === 'number') this.data.fill(scale, offset + 9, offset + 12);else if (scale) this.data.set(scale, offset + 9);
+    if (color) this.writeColor(offset, color);
+  }
+  writeColor(offset, input) {
+    const color = resolveColor(input);
+    this.data[offset + 12] = color[0];
+    this.data[offset + 13] = color[1];
+    this.data[offset + 14] = color[2];
+    this.data[offset + 15] = color[3] ?? 1;
+  }
+  flush() {
+    if (!this.count) return 0;
+    try {
+      return this.commit(this.data, this.count);
+    } finally {
+      this.count = 0;
+    }
+  }
+  clear() {
+    this.count = 0;
+  }
+}
+//# sourceMappingURL=TransformBatch.js.map