@parastud/react-native-vban 0.1.0

package/src/vban/pcm.ts

@@ -0,0 +1,99 @@
+/**
+ * Convert between VBAN wire PCM formats and normalized Float32 samples in the
+ * range [-1, 1). Samples are interleaved by channel, matching the VBAN payload
+ * layout (frame0_ch0, frame0_ch1, frame1_ch0, ...).
+ *
+ * Int16 is the common case (Voicemeeter's default), but Byte8/Int24/Int32 and
+ * native Float32 are handled too. Float64 and the packed 10/12-bit formats are
+ * not implemented yet and throw.
+ */
+
+import { DATA_TYPE_SIZE, VbanDataType } from './constants';
+
+/** Decode a raw VBAN payload into interleaved Float32 samples in [-1, 1). */
+export function pcmToFloat32(payload: Uint8Array, dataType: VbanDataType): Float32Array {
+  const sampleSize = DATA_TYPE_SIZE[dataType];
+  const count = Math.floor(payload.length / sampleSize);
+  const out = new Float32Array(count);
+  const view = new DataView(payload.buffer, payload.byteOffset, payload.byteLength);
+
+  switch (dataType) {
+    case VbanDataType.Byte8:
+      for (let i = 0; i < count; i++) out[i] = view.getInt8(i) / 128;
+      break;
+    case VbanDataType.Int16:
+      for (let i = 0; i < count; i++) out[i] = view.getInt16(i * 2, true) / 32768;
+      break;
+    case VbanDataType.Int24:
+      for (let i = 0; i < count; i++) {
+        const b0 = payload[i * 3];
+        const b1 = payload[i * 3 + 1];
+        const b2 = payload[i * 3 + 2];
+        let v = b0 | (b1 << 8) | (b2 << 16);
+        if (v & 0x800000) v |= ~0xffffff; // sign-extend 24 -> 32 bit
+        out[i] = v / 8388608;
+      }
+      break;
+    case VbanDataType.Int32:
+      for (let i = 0; i < count; i++) out[i] = view.getInt32(i * 4, true) / 2147483648;
+      break;
+    case VbanDataType.Float32:
+      for (let i = 0; i < count; i++) out[i] = view.getFloat32(i * 4, true);
+      break;
+    default:
+      throw new Error(`pcmToFloat32: unsupported data type ${VbanDataType[dataType]}`);
+  }
+  return out;
+}
+
+/** Clamp to [-1, 1) so encoding never wraps around. */
+function clamp(x: number): number {
+  return x < -1 ? -1 : x > 0.999969482421875 ? 0.999969482421875 : x;
+}
+
+/** Encode interleaved Float32 samples into a raw VBAN payload. */
+export function float32ToPcm(samples: Float32Array, dataType: VbanDataType): Uint8Array {
+  const sampleSize = DATA_TYPE_SIZE[dataType];
+  const out = new Uint8Array(samples.length * sampleSize);
+  const view = new DataView(out.buffer);
+
+  switch (dataType) {
+    case VbanDataType.Byte8:
+      for (let i = 0; i < samples.length; i++) view.setInt8(i, Math.round(clamp(samples[i]) * 128));
+      break;
+    case VbanDataType.Int16:
+      for (let i = 0; i < samples.length; i++) {
+        view.setInt16(i * 2, Math.round(clamp(samples[i]) * 32768), true);
+      }
+      break;
+    case VbanDataType.Int24:
+      for (let i = 0; i < samples.length; i++) {
+        const v = Math.round(clamp(samples[i]) * 8388608) & 0xffffff;
+        out[i * 3] = v & 0xff;
+        out[i * 3 + 1] = (v >> 8) & 0xff;
+        out[i * 3 + 2] = (v >> 16) & 0xff;
+      }
+      break;
+    case VbanDataType.Int32:
+      for (let i = 0; i < samples.length; i++) {
+        view.setInt32(i * 4, Math.round(clamp(samples[i]) * 2147483648), true);
+      }
+      break;
+    case VbanDataType.Float32:
+      for (let i = 0; i < samples.length; i++) view.setFloat32(i * 4, samples[i], true);
+      break;
+    default:
+      throw new Error(`float32ToPcm: unsupported data type ${VbanDataType[dataType]}`);
+  }
+  return out;
+}
+
+/** Peak absolute amplitude of interleaved samples, useful for a level meter. */
+export function peakLevel(samples: Float32Array): number {
+  let peak = 0;
+  for (let i = 0; i < samples.length; i++) {
+    const a = samples[i] < 0 ? -samples[i] : samples[i];
+    if (a > peak) peak = a;
+  }
+  return peak;
+}