@mediabunny/ac3 0.1.0

package/src/encoder.ts

@@ -0,0 +1,191 @@
+/*!
+ * Copyright (c) 2026-present, Vanilagy and contributors
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/.
+ */
+
+import {
+	CustomAudioEncoder,
+	AudioCodec,
+	AudioSample,
+	EncodedPacket,
+	registerEncoder,
+} from 'mediabunny';
+import { sendCommand } from './worker-client';
+import { assert } from './shared';
+import { AC3_SAMPLE_RATES, EAC3_REDUCED_SAMPLE_RATES } from '../../../shared/ac3-misc';
+
+class Ac3Encoder extends CustomAudioEncoder {
+	private ctx = 0;
+	private encoderFrameSize = 0;
+	private sampleRate = 0;
+	private numberOfChannels = 0;
+	private chunkMetadata: EncodedAudioChunkMetadata = {};
+
+	// Accumulate interleaved f32 samples until we have a full frame
+	private pendingBuffer = new Float32Array(2 ** 16);
+	private pendingFrames = 0;
+	private nextSampleTimestampInSamples: number | null = null;
+	private nextPacketTimestampInSamples: number | null = null;
+
+	static override supports(codec: AudioCodec, config: AudioEncoderConfig): boolean {
+		const sampleRates = codec === 'eac3'
+			? [...AC3_SAMPLE_RATES, ...EAC3_REDUCED_SAMPLE_RATES]
+			: AC3_SAMPLE_RATES;
+
+		return (codec === 'ac3' || codec === 'eac3')
+			&& config.numberOfChannels >= 1
+			&& config.numberOfChannels <= 8
+			&& sampleRates.includes(config.sampleRate);
+	}
+
+	async init() {
+		assert(this.config.bitrate);
+		this.sampleRate = this.config.sampleRate;
+		this.numberOfChannels = this.config.numberOfChannels;
+
+		const result = await sendCommand({
+			type: 'init-encoder',
+			data: {
+				codec: this.codec,
+				numberOfChannels: this.config.numberOfChannels,
+				sampleRate: this.config.sampleRate,
+				bitrate: this.config.bitrate,
+			},
+		});
+
+		this.ctx = result.ctx;
+		this.encoderFrameSize = result.frameSize;
+
+		this.resetInternalState();
+	}
+
+	private resetInternalState() {
+		this.pendingFrames = 0;
+		this.nextSampleTimestampInSamples = null;
+		this.nextPacketTimestampInSamples = null;
+
+		this.chunkMetadata = {
+			decoderConfig: {
+				codec: this.codec === 'ac3' ? 'ac-3' : 'ec-3',
+				numberOfChannels: this.config.numberOfChannels,
+				sampleRate: this.config.sampleRate,
+			},
+		};
+	}
+
+	async encode(audioSample: AudioSample) {
+		if (this.nextSampleTimestampInSamples === null) {
+			this.nextSampleTimestampInSamples = Math.round(audioSample.timestamp * this.sampleRate);
+			this.nextPacketTimestampInSamples = this.nextSampleTimestampInSamples;
+		}
+
+		const channels = this.numberOfChannels;
+		const incomingFrames = audioSample.numberOfFrames;
+
+		// Extract interleaved f32 data
+		const totalBytes = audioSample.allocationSize({ format: 'f32', planeIndex: 0 });
+		const audioBytes = new Uint8Array(totalBytes);
+		audioSample.copyTo(audioBytes, { format: 'f32', planeIndex: 0 });
+		const incomingData = new Float32Array(audioBytes.buffer);
+
+		const requiredSamples = (this.pendingFrames + incomingFrames) * channels;
+		if (requiredSamples > this.pendingBuffer.length) {
+			let newSize = this.pendingBuffer.length;
+			while (newSize < requiredSamples) {
+				newSize *= 2;
+			}
+			const newBuffer = new Float32Array(newSize);
+			newBuffer.set(this.pendingBuffer.subarray(0, this.pendingFrames * channels));
+			this.pendingBuffer = newBuffer;
+		}
+		this.pendingBuffer.set(incomingData, this.pendingFrames * channels);
+		this.pendingFrames += incomingFrames;
+
+		while (this.pendingFrames >= this.encoderFrameSize) {
+			await this.encodeOneFrame();
+		}
+	}
+
+	async flush() {
+		// Pad remaining samples with silence to fill a full frame
+		if (this.pendingFrames > 0) {
+			const channels = this.numberOfChannels;
+			const frameSize = this.encoderFrameSize;
+			const usedSamples = this.pendingFrames * channels;
+			const frameSamples = frameSize * channels;
+
+			this.pendingBuffer.fill(0, usedSamples, frameSamples);
+			this.pendingFrames = frameSize;
+
+			await this.encodeOneFrame();
+		}
+
+		await sendCommand({ type: 'flush-encoder', data: { ctx: this.ctx } });
+
+		this.resetInternalState();
+	}
+
+	close() {
+		void sendCommand({ type: 'close-encoder', data: { ctx: this.ctx } });
+	}
+
+	private async encodeOneFrame() {
+		assert(this.nextSampleTimestampInSamples !== null);
+		assert(this.nextPacketTimestampInSamples !== null);
+
+		const channels = this.numberOfChannels;
+		const frameSize = this.encoderFrameSize;
+		const frameSamples = frameSize * channels;
+
+		const frameData = this.pendingBuffer.slice(0, frameSamples);
+
+		// Shift remaining using copyWithin
+		this.pendingFrames -= frameSize;
+		if (this.pendingFrames > 0) {
+			this.pendingBuffer.copyWithin(0, frameSamples, frameSamples + this.pendingFrames * channels);
+		}
+
+		const audioData = frameData.buffer;
+		const result = await sendCommand({
+			type: 'encode',
+			data: {
+				ctx: this.ctx,
+				audioData,
+				timestamp: this.nextSampleTimestampInSamples,
+			},
+		}, [audioData]);
+
+		this.nextSampleTimestampInSamples += frameSize;
+
+		// We always get exactly one packet because we encode the correct frame size
+		const packet = new EncodedPacket(
+			new Uint8Array(result.encodedData),
+			'key',
+			this.nextPacketTimestampInSamples / this.sampleRate,
+			result.duration / this.sampleRate,
+		);
+
+		this.nextPacketTimestampInSamples += result.duration;
+
+		this.onPacket(
+			packet,
+			this.chunkMetadata,
+		);
+
+		this.chunkMetadata = {};
+	}
+}
+
+/**
+ * Registers AC-3 and E-AC-3 encoders, which Mediabunny will then use automatically when applicable. Make sure to call
+ * this function before starting any encoding task.
+ *
+ * @group \@mediabunny/ac3
+ * @public
+ */
+export const registerAc3Encoder = () => {
+	registerEncoder(Ac3Encoder);
+};

package/src/index.ts

@@ -0,0 +1,21 @@
+/*!
+ * Copyright (c) 2026-present, Vanilagy and contributors
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/.
+ */
+
+const AC3_LOADED_SYMBOL = Symbol.for('@mediabunny/ac3 loaded');
+if ((globalThis as Record<symbol, unknown>)[AC3_LOADED_SYMBOL]) {
+	console.error(
+		'[WARNING]\n@mediabunny/ac3 was loaded twice.'
+		+ ' This will likely cause the encoder/decoder not to work correctly.'
+		+ ' Check if multiple dependencies are importing different versions of @mediabunny/ac3,'
+		+ ' or if something is being bundled incorrectly.',
+	);
+}
+(globalThis as Record<symbol, unknown>)[AC3_LOADED_SYMBOL] = true;
+
+export { registerAc3Decoder } from './decoder';
+export { registerAc3Encoder } from './encoder';

package/src/shared.ts

@@ -0,0 +1,103 @@
+/*!
+ * Copyright (c) 2026-present, Vanilagy and contributors
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/.
+ */
+
+export type WorkerCommand = {
+	type: 'init-decoder';
+	data: {
+		codec: string;
+	};
+} | {
+	type: 'decode';
+	data: {
+		ctx: number;
+		encodedData: ArrayBuffer;
+		timestamp: number;
+	};
+} | {
+	type: 'flush-decoder';
+	data: {
+		ctx: number;
+	};
+} | {
+	type: 'close-decoder';
+	data: {
+		ctx: number;
+	};
+} | {
+	type: 'init-encoder';
+	data: {
+		codec: string;
+		numberOfChannels: number;
+		sampleRate: number;
+		bitrate: number;
+	};
+} | {
+	type: 'encode';
+	data: {
+		ctx: number;
+		audioData: ArrayBuffer;
+		timestamp: number;
+	};
+} | {
+	type: 'flush-encoder';
+	data: {
+		ctx: number;
+	};
+} | {
+	type: 'close-encoder';
+	data: {
+		ctx: number;
+	};
+};
+
+export type WorkerResponseData = {
+	type: 'init-decoder';
+	ctx: number;
+	frameSize: number;
+} | {
+	type: 'decode';
+	pcmData: ArrayBuffer;
+	format: AudioSampleFormat;
+	channels: number;
+	sampleRate: number;
+	sampleCount: number;
+	pts: number;
+} | {
+	type: 'flush-decoder';
+} | {
+	type: 'close-decoder';
+} | {
+	type: 'init-encoder';
+	ctx: number;
+	frameSize: number;
+} | {
+	type: 'encode';
+	encodedData: ArrayBuffer;
+	pts: number;
+	duration: number;
+} | {
+	type: 'flush-encoder';
+} | {
+	type: 'close-encoder';
+};
+
+export type WorkerResponse = {
+	id: number;
+} & ({
+	success: true;
+	data: WorkerResponseData;
+} | {
+	success: false;
+	error: unknown;
+});
+
+export function assert(x: unknown): asserts x {
+	if (!x) {
+		throw new Error('Assertion failed.');
+	}
+}

package/src/worker-client.ts

@@ -0,0 +1,69 @@
+/*!
+ * Copyright (c) 2026-present, Vanilagy and contributors
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/.
+ */
+
+import { assert, type WorkerCommand, type WorkerResponse, type WorkerResponseData } from './shared';
+// @ts-expect-error An esbuild plugin handles this, TypeScript doesn't need to understand
+import createWorker from './codec.worker';
+
+let workerPromise: Promise<Worker> | null;
+let nextMessageId = 0;
+const pendingMessages = new Map<number, {
+	resolve: (value: WorkerResponseData) => void;
+	reject: (reason?: unknown) => void;
+}>();
+
+export const sendCommand = async <T extends string>(
+	command: WorkerCommand & { type: T },
+	transferables?: Transferable[],
+) => {
+	const worker = await ensureWorker();
+
+	return new Promise<WorkerResponseData & { type: T }>((resolve, reject) => {
+		const id = nextMessageId++;
+		pendingMessages.set(id, {
+			resolve: resolve as (value: WorkerResponseData) => void,
+			reject,
+		});
+
+		if (transferables) {
+			worker.postMessage({ id, command }, transferables);
+		} else {
+			worker.postMessage({ id, command });
+		}
+	});
+};
+
+const ensureWorker = () => {
+	return workerPromise ??= (async () => {
+		// eslint-disable-next-line @typescript-eslint/no-unsafe-call
+		const worker = (await createWorker()) as Worker;
+
+		const onMessage = (data: WorkerResponse) => {
+			const pending = pendingMessages.get(data.id);
+			assert(pending !== undefined);
+
+			pendingMessages.delete(data.id);
+			if (data.success) {
+				pending.resolve(data.data);
+			} else {
+				pending.reject(data.error);
+			}
+		};
+
+		if (worker.addEventListener) {
+			worker.addEventListener('message', event => onMessage(event.data as WorkerResponse));
+		} else {
+			const nodeWorker = worker as unknown as {
+				on: (event: string, listener: (data: never) => void) => void;
+			};
+			nodeWorker.on('message', onMessage);
+		}
+
+		return worker;
+	})();
+};