@audio/wma-decode 1.0.0

package/LICENSE

@@ -0,0 +1,12 @@
+wma-decode - WMA audio decoder
+Copyright (c) 2025 audiojs
+
+This library is free software; you can redistribute it and/or modify it
+under the terms of the GNU Lesser General Public License as published by
+the Free Software Foundation; either version 2.1 of the License, or (at
+your option) any later version.
+
+This library is distributed in the hope that it will be useful, but
+WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
+License for more details.

package/README.md

@@ -0,0 +1,76 @@
+# wma-decode
+
+Decode WMA audio to PCM float samples. ASF demuxer in pure JS, WMA decoding via [RockBox](https://www.rockbox.org/) fixed-point decoder compiled to WASM (70 KB).
+
+Part of [audio-decode](https://github.com/audiojs/audio-decode).
+
+## Install
+
+```
+npm i @audio/wma-decode
+```
+
+## Usage
+
+```js
+import decode from '@audio/wma-decode'
+
+let { channelData, sampleRate } = await decode(wmaBuffer)
+```
+
+### Streaming
+
+```js
+import { decoder } from '@audio/wma-decode'
+
+let dec = await decoder()
+let result = dec.decode(chunk)
+dec.free()
+```
+
+### ASF demuxer only
+
+```js
+import { demuxASF } from '@audio/wma-decode'
+
+let { channels, sampleRate, bitRate, packets } = demuxASF(buffer)
+```
+
+## API
+
+### `decode(src): Promise<AudioData>`
+
+Whole-file decode. Accepts `Uint8Array` or `ArrayBuffer`.
+
+### `decoder(): Promise<WMADecoder>`
+
+Creates a decoder instance.
+
+- **`dec.decode(data)`** — decode chunk, returns `{ channelData, sampleRate }`
+- **`dec.flush()`** — flush (returns empty)
+- **`dec.free()`** — release WASM memory
+
+### `demuxASF(buf): ASFInfo`
+
+Parse ASF container without decoding. Returns stream properties and raw packets.
+
+## Formats
+
+- WMA v1 (0x0160)
+- WMA v2 (0x0161)
+
+WMA Pro and Lossless are not supported. An FFmpeg-based build is available via `build-ffmpeg.sh` for those formats.
+
+## Building WASM
+
+```
+npm run build
+```
+
+RockBox source is included in `lib/rockbox-wma/` (3 files, 152 KB).
+
+## License
+
+GPL-2.0+ (RockBox)
+
+<a href="https://github.com/krishnized/license/">ॐ</a>

package/package.json

@@ -0,0 +1,45 @@
+{
+  "name": "@audio/wma-decode",
+  "version": "1.0.0",
+  "description": "Decode WMA audio via RockBox WASM",
+  "type": "module",
+  "main": "wma-decode.js",
+  "types": "wma-decode.d.ts",
+  "exports": {
+    ".": "./wma-decode.js",
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "wma-decode.js",
+    "wma-decode.d.ts",
+    "src/wma.wasm.cjs",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "bash build.sh",
+    "prepack": "npm run build",
+    "test": "node test.js"
+  },
+  "keywords": [
+    "wma",
+    "windows-media-audio",
+    "asf",
+    "audio",
+    "decode",
+    "decoder",
+    "wasm",
+    "rockbox"
+  ],
+  "publishConfig": { "access": "public" },
+  "license": "GPL-2.0-or-later",
+  "author": "audiojs",
+  "homepage": "https://github.com/audiojs/wma-decode#readme",
+  "bugs": "https://github.com/audiojs/wma-decode/issues",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/audiojs/wma-decode.git"
+  },
+  "engines": {
+    "node": ">=16"
+  }
+}

package/wma-decode.d.ts

@@ -0,0 +1,33 @@
+interface AudioData {
+  channelData: Float32Array[];
+  sampleRate: number;
+}
+
+interface WmaDecoder {
+  decode(data: Uint8Array): AudioData;
+  flush(): AudioData;
+  free(): void;
+}
+
+/** Decode WMA audio buffer to PCM samples */
+export default function decode(src: ArrayBuffer | Uint8Array): Promise<AudioData>;
+
+/** Create streaming decoder instance */
+export function decoder(): Promise<WmaDecoder>;
+
+/** Parse ASF data packet, extract compressed audio payloads */
+export function parsePacket(pkt: Uint8Array, packetSize: number): Uint8Array[];
+
+/** Parse ASF container, extract audio properties and raw packets */
+export function demuxASF(buf: Uint8Array): {
+  channels: number;
+  sampleRate: number;
+  bitRate: number;
+  blockAlign: number;
+  bitsPerSample: number;
+  formatTag: number;
+  codecData: Uint8Array | null;
+  duration: number;
+  packetSize: number;
+  packets: Uint8Array[];
+};

package/wma-decode.js

@@ -0,0 +1,488 @@
+/**
+ * WMA decoder — ASF demuxer (pure JS) + RockBox wmadec (WASM)
+ * Decodes WMA v1/v2 in ASF containers
+ *
+ * let { channelData, sampleRate } = await decode(wmabuf)
+ */
+
+const EMPTY = Object.freeze({ channelData: [], sampleRate: 0 })
+
+// ASF GUIDs (16 bytes each, little-endian)
+const GUID = {
+	header:      [0x30,0x26,0xb2,0x75,0x8e,0x66,0xcf,0x11,0xa6,0xd9,0x00,0xaa,0x00,0x62,0xce,0x6c],
+	fileProps:   [0xa1,0xdc,0xab,0x8c,0x47,0xa9,0xcf,0x11,0x8e,0xe4,0x00,0xc0,0x0c,0x20,0x53,0x65],
+	streamProps: [0x91,0x07,0xdc,0xb7,0xb7,0xa9,0xcf,0x11,0x8e,0xe6,0x00,0xc0,0x0c,0x20,0x53,0x65],
+	audioMedia:  [0x40,0x9e,0x69,0xf8,0x4d,0x5b,0xcf,0x11,0xa8,0xfd,0x00,0x80,0x5f,0x5c,0x44,0x2b],
+	data:        [0x36,0x26,0xb2,0x75,0x8e,0x66,0xcf,0x11,0xa6,0xd9,0x00,0xaa,0x00,0x62,0xce,0x6c],
+	headerExt:   [0xb5,0x03,0xbf,0x5f,0x2e,0xa9,0xcf,0x11,0x8e,0xe3,0x00,0xc0,0x0c,0x20,0x53,0x65],
+}
+
+function guidEq(buf, off, guid) {
+	for (let i = 0; i < 16; i++) if (buf[off + i] !== guid[i]) return false
+	return true
+}
+
+// Little-endian readers
+function u16(b, o) { return b[o] | (b[o + 1] << 8) }
+function u32(b, o) { return (b[o] | (b[o + 1] << 8) | (b[o + 2] << 16) | (b[o + 3] << 24)) >>> 0 }
+function u64(b, o) {
+	// JS safe up to 2^53; read low 32 + high 32
+	let lo = u32(b, o), hi = u32(b, o + 4)
+	return hi * 0x100000000 + lo
+}
+
+/**
+ * Parse ASF container — extract audio stream properties + data packets
+ * @param {Uint8Array} buf
+ * @returns {{ channels, sampleRate, bitRate, blockAlign, bitsPerSample, formatTag, codecData, duration, packetSize, packets }}
+ */
+export function demuxASF(buf) {
+	if (!(buf instanceof Uint8Array)) buf = new Uint8Array(buf)
+	if (buf.length < 30) throw Error('Not an ASF/WMA file')
+
+	// Verify ASF Header Object GUID
+	if (!guidEq(buf, 0, GUID.header)) throw Error('Not an ASF/WMA file')
+
+	let headerSize = u64(buf, 16)
+	let numObjects = u32(buf, 24)
+	// reserved1(1) + reserved2(1) at offset 28-29
+
+	let audio = null    // WAVEFORMATEX fields
+	let packetSize = 0
+	let duration = 0    // in seconds
+	let datOff = 0      // data object offset
+	let datSize = 0     // data object total size
+	let totalPackets = 0
+
+	// Parse header sub-objects
+	let pos = 30
+	let headerEnd = Math.min(Number(headerSize), buf.length)
+	for (let i = 0; i < numObjects && pos < headerEnd - 24; i++) {
+		let objSize = u64(buf, pos + 16)
+		if (objSize < 24) break
+		let objEnd = pos + Math.min(Number(objSize), headerEnd - pos)
+
+		if (guidEq(buf, pos, GUID.streamProps)) {
+			audio = parseStreamProps(buf, pos + 24, objEnd) || audio
+		} else if (guidEq(buf, pos, GUID.fileProps)) {
+			let fp = parseFileProps(buf, pos + 24, objEnd)
+			if (fp) { packetSize = fp.packetSize; duration = fp.duration }
+		} else if (guidEq(buf, pos, GUID.headerExt)) {
+			// Header Extension Object — nested sub-objects skipped (no metadata needed for decode)
+		}
+
+		pos = objEnd
+	}
+
+	if (!audio) throw Error('No audio stream in ASF')
+
+	// Find Data Object after header
+	pos = Number(headerSize)
+	if (pos + 50 <= buf.length && guidEq(buf, pos, GUID.data)) {
+		datSize = u64(buf, pos + 16)
+		// fileId(16) + totalPackets(8) + reserved(2) = 26 bytes after object header
+		totalPackets = u64(buf, pos + 24 + 16)
+		datOff = pos + 24 + 26 // start of packet data
+	}
+
+	// Extract packets
+	let packets = []
+	if (datOff && packetSize > 0) {
+		let datEnd = Math.min(pos + Number(datSize), buf.length)
+		let ppos = datOff
+		while (ppos + packetSize <= datEnd) {
+			packets.push(buf.subarray(ppos, ppos + packetSize))
+			ppos += packetSize
+		}
+	} else if (datOff && totalPackets > 0) {
+		// Variable-size packets — fallback: treat remaining data as one blob
+		let datEnd = Math.min(pos + Number(datSize), buf.length)
+		if (datOff < datEnd) packets.push(buf.subarray(datOff, datEnd))
+	}
+
+	return {
+		channels: audio.channels,
+		sampleRate: audio.sampleRate,
+		bitRate: audio.avgBytesPerSec * 8,
+		blockAlign: audio.blockAlign,
+		bitsPerSample: audio.bitsPerSample,
+		formatTag: audio.formatTag,
+		codecData: audio.codecData,
+		duration,
+		packetSize,
+		packets
+	}
+}
+
+/**
+ * Parse Stream Properties Object body
+ * Returns audio WAVEFORMATEX fields if this is an audio stream, null otherwise
+ */
+function parseStreamProps(buf, start, end) {
+	// Stream Type GUID(16) + Error Correction Type GUID(16) + Time Offset(8)
+	// + Type-Specific Data Length(4) + Error Correction Data Length(4)
+	// + Flags(2) + Reserved(4)
+	if (start + 54 > end) return null
+
+	// Check that stream type is Audio Media
+	if (!guidEq(buf, start, GUID.audioMedia)) return null
+
+	// flags(2) + reserved(4) = 6 bytes after type/ec data lengths
+	let waveOff = start + 54
+
+	if (waveOff + 18 > end) return null
+
+	// WAVEFORMATEX
+	let formatTag = u16(buf, waveOff)
+	let channels = u16(buf, waveOff + 2)
+	let sampleRate = u32(buf, waveOff + 4)
+	let avgBytesPerSec = u32(buf, waveOff + 8)
+	let blockAlign = u16(buf, waveOff + 12)
+	let bitsPerSample = u16(buf, waveOff + 14)
+	let cbSize = u16(buf, waveOff + 16)
+
+	// Codec-specific data follows WAVEFORMATEX
+	let codecData = null
+	if (cbSize > 0 && waveOff + 18 + cbSize <= end) {
+		codecData = buf.subarray(waveOff + 18, waveOff + 18 + cbSize)
+	}
+
+	return { formatTag, channels, sampleRate, avgBytesPerSec, blockAlign, bitsPerSample, codecData }
+}
+
+/**
+ * Parse File Properties Object body
+ * Returns { packetSize, duration } or null
+ */
+function parseFileProps(buf, start, end) {
+	// File ID(16) + File Size(8) + Creation Date(8) + Data Packets Count(8)
+	// + Play Duration(8) + Send Duration(8) + Preroll(8)
+	// + Flags(4) + Min Data Packet Size(4) + Max Data Packet Size(4)
+	// + Max Bitrate(4)
+	if (start + 80 > end) return null
+
+	// Play Duration is 100-ns units (offset 40)
+	let playDur = u64(buf, start + 40)
+	// Preroll in ms (offset 56)
+	let preroll = u64(buf, start + 56)
+	// Duration in seconds
+	let duration = Math.max(0, playDur / 10000000 - preroll / 1000)
+
+	// Max packet size at offset 72 (min at 68, always equal for WMA)
+	let packetSize = u32(buf, start + 72)
+
+	return { packetSize, duration }
+}
+
+/**
+ * Parse ASF data packet — extract compressed audio payload(s)
+ * Returns array of payload buffers
+ */
+export function parsePacket(pkt, packetSize) {
+	if (!pkt || pkt.length < 3) return []
+
+	let pos = 0
+
+	// Error Correction — first byte flags
+	let ecFlags = pkt[pos++]
+	if (ecFlags & 0x80) {
+		// Error correction present
+		let ecLen = ecFlags & 0x0F
+		// Opaque data flag is bit 4
+		pos += ecLen
+	}
+
+	if (pos >= pkt.length) return []
+
+	// Payload Parsing Information
+	let ppFlags = pkt[pos++]
+	let lenFlags = pkt[pos++]
+
+	let multiplePayloads = !!(ppFlags & 0x01)
+	let seqType = (ppFlags >> 1) & 0x03
+	let padType = (ppFlags >> 3) & 0x03
+	let pktLenType = (ppFlags >> 5) & 0x03
+
+	let repType = lenFlags & 0x03
+	let offType = (lenFlags >> 2) & 0x03
+	let medNumType = (lenFlags >> 4) & 0x03
+
+	// Skip optional packet length field
+	if (pktLenType === 1) pos += 1
+	else if (pktLenType === 2) pos += 2
+	else if (pktLenType === 3) pos += 4
+
+	// Sequence (skip)
+	if (seqType === 1) pos += 1
+	else if (seqType === 2) pos += 2
+	else if (seqType === 3) pos += 4
+
+	// Padding length
+	let padLen = 0
+	if (padType === 1) { padLen = pkt[pos++] }
+	else if (padType === 2) { padLen = u16(pkt, pos); pos += 2 }
+	else if (padType === 3) { padLen = u32(pkt, pos); pos += 4 }
+
+	// Send time (4 bytes) + duration (2 bytes)
+	pos += 6
+
+	if (pos >= pkt.length) return []
+
+	let payloads = []
+
+	if (!multiplePayloads) {
+		// Single payload — rest of packet minus padding
+		pos += 1 // stream number (always 1 byte)
+
+		// Media object number
+		pos += fieldSize(medNumType)
+
+		// Offset into media object
+		pos += fieldSize(offType)
+
+		// Replicated data length
+		let repLen = readVarField(pkt, pos, repType)
+		pos += fieldSize(repType)
+
+		// Skip replicated data
+		if (repLen === 1) {
+			// Compressed payload: repLen==1 means compressed
+			pos += 1 // presentation time delta
+		} else {
+			pos += repLen
+		}
+
+		let payloadLen = pkt.length - pos - padLen
+		if (payloadLen > 0 && pos + payloadLen <= pkt.length) {
+			payloads.push(pkt.subarray(pos, pos + payloadLen))
+		}
+	} else {
+		// Multiple payloads
+		let payloadFlags = pkt[pos++]
+		let numPayloads = payloadFlags & 0x3F
+		let payLenType = (payloadFlags >> 6) & 0x03
+
+		for (let i = 0; i < numPayloads && pos < pkt.length; i++) {
+			// Stream number (1 byte, lower 7 = number, bit 7 = key frame)
+			pos += 1
+
+			// Media object number
+			pos += fieldSize(medNumType)
+
+			// Offset into media object
+			pos += fieldSize(offType)
+
+			// Replicated data
+			let repLen = readVarField(pkt, pos, repType)
+			pos += fieldSize(repType)
+
+			if (repLen === 1) {
+				pos += 1 // compressed: presentation time delta
+			} else {
+				pos += repLen
+			}
+
+			// Payload length
+			let payLen = 0
+			if (payLenType === 1) { payLen = pkt[pos++] }
+			else if (payLenType === 2) { payLen = u16(pkt, pos); pos += 2 }
+			else if (payLenType === 3) { payLen = u32(pkt, pos); pos += 4 }
+			else { payLen = pkt.length - pos }
+
+			if (payLen > 0 && pos + payLen <= pkt.length) {
+				payloads.push(pkt.subarray(pos, pos + payLen))
+			}
+			pos += payLen
+		}
+	}
+
+	return payloads
+}
+
+function fieldSize(type) {
+	return type === 0 ? 0 : type === 1 ? 1 : type === 2 ? 2 : 4
+}
+
+function readVarField(buf, off, type) {
+	if (type === 0) return 0
+	if (type === 1) return buf[off] || 0
+	if (type === 2) return (off + 2 <= buf.length) ? u16(buf, off) : 0
+	return (off + 4 <= buf.length) ? u32(buf, off) : 0
+}
+
+// ===== WMA format tag names =====
+
+const WMA_FORMATS = {
+	0x0160: 'WMAv1',
+	0x0161: 'WMAv2',
+	0x0162: 'WMAPro',
+	0x0163: 'WMALossless',
+}
+
+// ===== WASM module loader =====
+
+let _modP
+
+async function getMod() {
+	if (_modP) return _modP
+	let p = (async () => {
+		let createWMA
+		if (typeof process !== 'undefined' && process.versions?.node) {
+			let m = 'module'
+			let { createRequire } = await import(m)
+			createWMA = createRequire(import.meta.url)('./src/wma.wasm.cjs')
+		} else {
+			let mod = await import('./src/wma.wasm.cjs')
+			createWMA = mod.default || mod
+		}
+		return createWMA()
+	})()
+	_modP = p
+	try { return await p }
+	catch (e) { _modP = null; throw e }
+}
+
+/**
+ * Whole-file decode
+ * @param {Uint8Array|ArrayBuffer} src
+ * @returns {Promise<{channelData: Float32Array[], sampleRate: number}>}
+ */
+export default async function decode(src) {
+	let buf = src instanceof Uint8Array ? src : new Uint8Array(src)
+	let dec = await decoder()
+	try {
+		return dec.decode(buf)
+	} finally {
+		dec.free()
+	}
+}
+
+/**
+ * Create decoder instance
+ * @returns {Promise<{decode(chunk: Uint8Array): {channelData, sampleRate}, flush(), free()}>}
+ */
+export async function decoder() {
+	return new WMADecoder(await getMod())
+}
+
+class WMADecoder {
+	constructor(mod) {
+		this.m = mod
+		this.h = null
+		this.sr = 0
+		this.ch = 0
+		this.done = false
+		this._ptr = 0
+		this._cap = 0
+	}
+
+	decode(data) {
+		if (this.done) throw Error('Decoder already freed')
+		if (!data?.length) return EMPTY
+
+		let buf = data instanceof Uint8Array ? data : new Uint8Array(data)
+		let asf = demuxASF(buf)
+
+		if (!asf.packets.length) return EMPTY
+
+		let fmt = WMA_FORMATS[asf.formatTag]
+		if (!fmt) throw Error('Unsupported WMA format tag: 0x' + asf.formatTag.toString(16))
+
+		this.sr = asf.sampleRate
+		this.ch = asf.channels
+
+		let m = this.m
+
+		// Init WASM decoder with audio properties
+		let extraPtr = 0, extraLen = 0
+		if (asf.codecData?.length) {
+			extraPtr = this._alloc(asf.codecData.length)
+			m.HEAPU8.set(asf.codecData, extraPtr)
+			extraLen = asf.codecData.length
+		}
+
+		this.h = m._wma_create(
+			asf.channels, asf.sampleRate, asf.bitRate,
+			asf.blockAlign, asf.formatTag, asf.bitsPerSample,
+			extraPtr, extraLen
+		)
+		if (!this.h) throw Error('WMA decoder init failed')
+
+		// Extract payloads from packets and decode
+		// Each payload is one blockAlign-sized WMA superframe
+		let chunks = []
+		let totalPerCh = 0
+		let channels = asf.channels
+		let errors = 0
+		let ba = asf.blockAlign
+
+		for (let pkt of asf.packets) {
+			let payloads = parsePacket(pkt, asf.packetSize)
+			for (let payload of payloads) {
+				// Split payload into blockAlign-sized frames
+				for (let off = 0; off + ba <= payload.length; off += ba) {
+					let frame = payload.subarray(off, off + ba)
+					let ptr = this._alloc(ba)
+					m.HEAPU8.set(frame, ptr)
+					let out = m._wma_decode(this.h, ptr, ba)
+					if (!out) { errors++; continue }
+
+					let n = m._wma_samples()
+					let sr = m._wma_samplerate()
+					if (sr) this.sr = sr
+					let ch = m._wma_channels()
+					if (ch) channels = ch
+
+					let spc = n / channels
+					let samples = new Float32Array(m.HEAPF32.buffer, out, n).slice()
+					chunks.push({ data: samples, ch: channels, spc })
+					totalPerCh += spc
+				}
+			}
+		}
+
+		if (!totalPerCh) {
+			if (errors) throw Error(errors + ' frame(s) failed to decode')
+			return EMPTY
+		}
+
+		// De-interleave
+		let channelData = Array.from({ length: channels }, () => new Float32Array(totalPerCh))
+		let pos = 0
+		for (let { data, ch, spc } of chunks) {
+			for (let c = 0; c < ch; c++) {
+				let out = channelData[c]
+				for (let s = 0; s < spc; s++) out[pos + s] = data[s * ch + c]
+			}
+			pos += spc
+		}
+
+		return { channelData, sampleRate: this.sr }
+	}
+
+	flush() { return EMPTY }
+
+	free() {
+		if (this.done) return
+		this.done = true
+		if (this.h) {
+			this.m._wma_close(this.h)
+			this.m._wma_free_buf()
+			this.h = null
+		}
+		if (this._ptr) {
+			this.m._free(this._ptr)
+			this._ptr = 0
+			this._cap = 0
+		}
+	}
+
+	_alloc(len) {
+		if (len > this._cap) {
+			if (this._ptr) this.m._free(this._ptr)
+			this._cap = len
+			this._ptr = this.m._malloc(len)
+		}
+		return this._ptr
+	}
+}