@hraness/dawg 0.6.0 → 0.7.0

package/src/audio/strings/body.ts

@@ -120,26 +120,39 @@ export function bodyFor(name: string, size: number): Body {
 /** Body over `x` in place: direct + modes, then a 10 Hz DC blocker. */
 export function applyBody(x: Float64Array, body: Body, sr: number): void {
   if (body.modes.length === 0 && body.direct === 1) return;
-  const input = Float64Array.from(x);
-  for (let i = 0; i < x.length; i += 1) x[i] = body.direct * input[i]!;
-  for (const [f, q, g] of body.modes) {
-    if (f >= 0.45 * sr) continue;
+  // Every mode per sample in one pass, summed in mode order (the same
+  // arithmetic as a pass per mode, so the output is bit-identical), with
+  // the coefficients in flat arrays: about 7x faster under JSC.
+  const live = body.modes.filter(([f]) => f < 0.45 * sr);
+  const m = live.length;
+  const B0 = new Float64Array(m);
+  const A1 = new Float64Array(m);
+  const A2 = new Float64Array(m);
+  const G = new Float64Array(m);
+  const Z1 = new Float64Array(m);
+  const Z2 = new Float64Array(m);
+  for (let k = 0; k < m; k += 1) {
+    const [f, q, g] = live[k]!;
     const w = (2 * Math.PI * f) / sr;
     const alpha = Math.sin(w) / (2 * q);
     const a0 = 1 + alpha;
-    const b0 = alpha / a0;
-    const b2 = -alpha / a0;
-    const a1 = (-2 * Math.cos(w)) / a0;
-    const a2 = (1 - alpha) / a0;
-    let z1 = 0;
-    let z2 = 0;
-    for (let i = 0; i < x.length; i += 1) {
-      const xi = input[i]!;
-      const y = b0 * xi + z1;
-      z1 = -a1 * y + z2;
-      z2 = b2 * xi - a2 * y;
-      x[i] = x[i]! + g * y;
+    B0[k] = alpha / a0;
+    A1[k] = (-2 * Math.cos(w)) / a0;
+    A2[k] = (1 - alpha) / a0;
+    G[k] = g;
+  }
+  const direct = body.direct;
+  for (let i = 0; i < x.length; i += 1) {
+    const xi = x[i]!;
+    let acc = direct * xi;
+    for (let k = 0; k < m; k += 1) {
+      const b0 = B0[k]!;
+      const y = b0 * xi + Z1[k]!;
+      Z1[k] = -A1[k]! * y + Z2[k]!;
+      Z2[k] = -b0 * xi - A2[k]! * y;
+      acc = acc + G[k]! * y;
     }
+    x[i] = acc;
   }
   const R = Math.exp((-2 * Math.PI * 10) / sr);
   let x1 = 0;