rusty-chromaprint-wasm 0.1.0

package/README.md

@@ -0,0 +1,30 @@
+# chromaprint-wasm
+
+WASM bindings for [rusty-chromaprint](https://github.com/darksv/rusty-chromaprint). Uses preset test2 (AcoustID/fpcalc compatible).
+
+## Build
+
+```bash
+# Install wasm-pack if needed: cargo install wasm-pack
+wasm-pack build --target web
+# Or for Node: wasm-pack build --target nodejs
+```
+
+Output in `pkg/`: `.wasm`, `.js` glue, and (with wasm-pack) TypeScript definitions.
+
+## API (from JS)
+
+- **`new Fingerprinter()`** – create a fingerprinter
+- **`.start(sampleRate, channels)`** – e.g. `44100`, `1` or `2`
+- **`.consume(samples)`** – `samples` is an `Int16Array` (interleaved if stereo)
+- **`.finish()`** – call when all samples have been fed
+- **`.getFingerprint()`** – returns `Uint32Array` (raw)
+- **`.getCompressedFingerprint()`** – returns base64 string (AcoustID format)
+
+- **`fingerprintFromSamples(sampleRate, channels, samples)`** – one-shot; returns `{ raw: Uint32Array, compressed: string }`
+
+- **`matchFingerprints(raw1, raw2)`** – both `Uint32Array`; returns array of `{ start1, end1, start2, end2, duration, score }` (times in seconds)
+
+## Audio input
+
+Decode in the browser (e.g. `decodeAudioData`), get channel data as Float32, convert to i16 (e.g. `samples[i] = Math.max(-32768, Math.min(32767, channelData[i] * 32768))`), then pass as `Int16Array` to `consume()` or `fingerprintFromSamples`.

package/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "rusty-chromaprint-wasm",
+  "type": "module",
+  "description": "WASM bindings for rusty-chromaprint audio fingerprinting",
+  "version": "0.1.0",
+  "files": [
+    "rusty_chromaprint_wasm_bg.wasm",
+    "rusty_chromaprint_wasm.js",
+    "rusty_chromaprint_wasm.d.ts"
+  ],
+  "main": "rusty_chromaprint_wasm.js",
+  "types": "rusty_chromaprint_wasm.d.ts",
+  "sideEffects": [
+    "./snippets/*"
+  ]
+}

package/rusty_chromaprint_wasm.d.ts

@@ -0,0 +1,88 @@
+/* tslint:disable */
+/* eslint-disable */
+
+/**
+ * Create a new fingerprinter. Uses the default preset (compatible with AcoustID).
+ */
+export class Fingerprinter {
+    free(): void;
+    [Symbol.dispose](): void;
+    /**
+     * Feed interleaved i16 samples (e.g. from decodeAudioData → getChannelData → convert to i16).
+     * Pass an Int16Array from JS.
+     */
+    consume(samples: Int16Array): void;
+    /**
+     * Call when all samples have been fed.
+     */
+    finish(): void;
+    /**
+     * Compressed fingerprint as base64 (AcoustID/fpcalc format). Only valid after finish().
+     */
+    getCompressedFingerprint(): string;
+    /**
+     * Raw fingerprint (u32 array). Only valid after finish().
+     */
+    getFingerprint(): Uint32Array;
+    constructor();
+    /**
+     * Start fingerprinting. Call with your audio's sample rate (e.g. 44100) and channel count (1 or 2).
+     */
+    start(sample_rate: number, channels: number): void;
+}
+
+/**
+ * One-shot: compute fingerprint from interleaved i16 samples.
+ * Returns `{ raw: Uint32Array, compressed: string }`.
+ */
+export function fingerprintFromSamples(sample_rate: number, channels: number, samples: Int16Array): any;
+
+/**
+ * Match two raw fingerprints (u32 arrays). Returns segments of similar audio.
+ * Each segment has start1, end1, start2, end2, duration (seconds) and score.
+ */
+export function matchFingerprints(raw1: Uint32Array, raw2: Uint32Array): any;
+
+export type InitInput = RequestInfo | URL | Response | BufferSource | WebAssembly.Module;
+
+export interface InitOutput {
+    readonly memory: WebAssembly.Memory;
+    readonly __wbg_fingerprinter_free: (a: number, b: number) => void;
+    readonly fingerprintFromSamples: (a: number, b: number, c: number, d: number) => [number, number, number];
+    readonly fingerprinter_consume: (a: number, b: number, c: number) => void;
+    readonly fingerprinter_finish: (a: number) => void;
+    readonly fingerprinter_getCompressedFingerprint: (a: number) => [number, number];
+    readonly fingerprinter_getFingerprint: (a: number) => [number, number];
+    readonly fingerprinter_new: () => [number, number, number];
+    readonly fingerprinter_start: (a: number, b: number, c: number) => [number, number];
+    readonly matchFingerprints: (a: number, b: number, c: number, d: number) => [number, number, number];
+    readonly __wbindgen_exn_store: (a: number) => void;
+    readonly __externref_table_alloc: () => number;
+    readonly __wbindgen_externrefs: WebAssembly.Table;
+    readonly __wbindgen_malloc: (a: number, b: number) => number;
+    readonly __externref_table_dealloc: (a: number) => void;
+    readonly __wbindgen_free: (a: number, b: number, c: number) => void;
+    readonly __wbindgen_start: () => void;
+}
+
+export type SyncInitInput = BufferSource | WebAssembly.Module;
+
+/**
+ * Instantiates the given `module`, which can either be bytes or
+ * a precompiled `WebAssembly.Module`.
+ *
+ * @param {{ module: SyncInitInput }} module - Passing `SyncInitInput` directly is deprecated.
+ *
+ * @returns {InitOutput}
+ */
+export function initSync(module: { module: SyncInitInput } | SyncInitInput): InitOutput;
+
+/**
+ * If `module_or_path` is {RequestInfo} or {URL}, makes a request and
+ * for everything else, calls `WebAssembly.instantiate` directly.
+ *
+ * @param {{ module_or_path: InitInput | Promise<InitInput> }} module_or_path - Passing `InitInput` directly is deprecated.
+ *
+ * @returns {Promise<InitOutput>}
+ */
+export default function __wbg_init (module_or_path?: { module_or_path: InitInput | Promise<InitInput> } | InitInput | Promise<InitInput>): Promise<InitOutput>;

package/rusty_chromaprint_wasm.js

@@ -0,0 +1,356 @@
+/* @ts-self-types="./rusty_chromaprint_wasm.d.ts" */
+
+/**
+ * Create a new fingerprinter. Uses the default preset (compatible with AcoustID).
+ */
+export class Fingerprinter {
+    __destroy_into_raw() {
+        const ptr = this.__wbg_ptr;
+        this.__wbg_ptr = 0;
+        FingerprinterFinalization.unregister(this);
+        return ptr;
+    }
+    free() {
+        const ptr = this.__destroy_into_raw();
+        wasm.__wbg_fingerprinter_free(ptr, 0);
+    }
+    /**
+     * Feed interleaved i16 samples (e.g. from decodeAudioData → getChannelData → convert to i16).
+     * Pass an Int16Array from JS.
+     * @param {Int16Array} samples
+     */
+    consume(samples) {
+        const ptr0 = passArray16ToWasm0(samples, wasm.__wbindgen_malloc);
+        const len0 = WASM_VECTOR_LEN;
+        wasm.fingerprinter_consume(this.__wbg_ptr, ptr0, len0);
+    }
+    /**
+     * Call when all samples have been fed.
+     */
+    finish() {
+        wasm.fingerprinter_finish(this.__wbg_ptr);
+    }
+    /**
+     * Compressed fingerprint as base64 (AcoustID/fpcalc format). Only valid after finish().
+     * @returns {string}
+     */
+    getCompressedFingerprint() {
+        let deferred1_0;
+        let deferred1_1;
+        try {
+            const ret = wasm.fingerprinter_getCompressedFingerprint(this.__wbg_ptr);
+            deferred1_0 = ret[0];
+            deferred1_1 = ret[1];
+            return getStringFromWasm0(ret[0], ret[1]);
+        } finally {
+            wasm.__wbindgen_free(deferred1_0, deferred1_1, 1);
+        }
+    }
+    /**
+     * Raw fingerprint (u32 array). Only valid after finish().
+     * @returns {Uint32Array}
+     */
+    getFingerprint() {
+        const ret = wasm.fingerprinter_getFingerprint(this.__wbg_ptr);
+        var v1 = getArrayU32FromWasm0(ret[0], ret[1]).slice();
+        wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
+        return v1;
+    }
+    constructor() {
+        const ret = wasm.fingerprinter_new();
+        if (ret[2]) {
+            throw takeFromExternrefTable0(ret[1]);
+        }
+        this.__wbg_ptr = ret[0] >>> 0;
+        FingerprinterFinalization.register(this, this.__wbg_ptr, this);
+        return this;
+    }
+    /**
+     * Start fingerprinting. Call with your audio's sample rate (e.g. 44100) and channel count (1 or 2).
+     * @param {number} sample_rate
+     * @param {number} channels
+     */
+    start(sample_rate, channels) {
+        const ret = wasm.fingerprinter_start(this.__wbg_ptr, sample_rate, channels);
+        if (ret[1]) {
+            throw takeFromExternrefTable0(ret[0]);
+        }
+    }
+}
+if (Symbol.dispose) Fingerprinter.prototype[Symbol.dispose] = Fingerprinter.prototype.free;
+
+/**
+ * One-shot: compute fingerprint from interleaved i16 samples.
+ * Returns `{ raw: Uint32Array, compressed: string }`.
+ * @param {number} sample_rate
+ * @param {number} channels
+ * @param {Int16Array} samples
+ * @returns {any}
+ */
+export function fingerprintFromSamples(sample_rate, channels, samples) {
+    const ptr0 = passArray16ToWasm0(samples, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.fingerprintFromSamples(sample_rate, channels, ptr0, len0);
+    if (ret[2]) {
+        throw takeFromExternrefTable0(ret[1]);
+    }
+    return takeFromExternrefTable0(ret[0]);
+}
+
+/**
+ * Match two raw fingerprints (u32 arrays). Returns segments of similar audio.
+ * Each segment has start1, end1, start2, end2, duration (seconds) and score.
+ * @param {Uint32Array} raw1
+ * @param {Uint32Array} raw2
+ * @returns {any}
+ */
+export function matchFingerprints(raw1, raw2) {
+    const ptr0 = passArray32ToWasm0(raw1, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ptr1 = passArray32ToWasm0(raw2, wasm.__wbindgen_malloc);
+    const len1 = WASM_VECTOR_LEN;
+    const ret = wasm.matchFingerprints(ptr0, len0, ptr1, len1);
+    if (ret[2]) {
+        throw takeFromExternrefTable0(ret[1]);
+    }
+    return takeFromExternrefTable0(ret[0]);
+}
+
+function __wbg_get_imports() {
+    const import0 = {
+        __proto__: null,
+        __wbg_Error_83742b46f01ce22d: function(arg0, arg1) {
+            const ret = Error(getStringFromWasm0(arg0, arg1));
+            return ret;
+        },
+        __wbg___wbindgen_throw_6ddd609b62940d55: function(arg0, arg1) {
+            throw new Error(getStringFromWasm0(arg0, arg1));
+        },
+        __wbg_new_a70fbab9066b301f: function() {
+            const ret = new Array();
+            return ret;
+        },
+        __wbg_new_ab79df5bd7c26067: function() {
+            const ret = new Object();
+            return ret;
+        },
+        __wbg_new_from_slice_898ac63cbd46f332: function(arg0, arg1) {
+            const ret = new Uint32Array(getArrayU32FromWasm0(arg0, arg1));
+            return ret;
+        },
+        __wbg_push_e87b0e732085a946: function(arg0, arg1) {
+            const ret = arg0.push(arg1);
+            return ret;
+        },
+        __wbg_set_7eaa4f96924fd6b3: function() { return handleError(function (arg0, arg1, arg2) {
+            const ret = Reflect.set(arg0, arg1, arg2);
+            return ret;
+        }, arguments); },
+        __wbindgen_cast_0000000000000001: function(arg0) {
+            // Cast intrinsic for `F64 -> Externref`.
+            const ret = arg0;
+            return ret;
+        },
+        __wbindgen_cast_0000000000000002: function(arg0, arg1) {
+            // Cast intrinsic for `Ref(String) -> Externref`.
+            const ret = getStringFromWasm0(arg0, arg1);
+            return ret;
+        },
+        __wbindgen_init_externref_table: function() {
+            const table = wasm.__wbindgen_externrefs;
+            const offset = table.grow(4);
+            table.set(0, undefined);
+            table.set(offset + 0, undefined);
+            table.set(offset + 1, null);
+            table.set(offset + 2, true);
+            table.set(offset + 3, false);
+        },
+    };
+    return {
+        __proto__: null,
+        "./rusty_chromaprint_wasm_bg.js": import0,
+    };
+}
+
+const FingerprinterFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_fingerprinter_free(ptr >>> 0, 1));
+
+function addToExternrefTable0(obj) {
+    const idx = wasm.__externref_table_alloc();
+    wasm.__wbindgen_externrefs.set(idx, obj);
+    return idx;
+}
+
+function getArrayU32FromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return getUint32ArrayMemory0().subarray(ptr / 4, ptr / 4 + len);
+}
+
+function getStringFromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return decodeText(ptr, len);
+}
+
+let cachedUint16ArrayMemory0 = null;
+function getUint16ArrayMemory0() {
+    if (cachedUint16ArrayMemory0 === null || cachedUint16ArrayMemory0.byteLength === 0) {
+        cachedUint16ArrayMemory0 = new Uint16Array(wasm.memory.buffer);
+    }
+    return cachedUint16ArrayMemory0;
+}
+
+let cachedUint32ArrayMemory0 = null;
+function getUint32ArrayMemory0() {
+    if (cachedUint32ArrayMemory0 === null || cachedUint32ArrayMemory0.byteLength === 0) {
+        cachedUint32ArrayMemory0 = new Uint32Array(wasm.memory.buffer);
+    }
+    return cachedUint32ArrayMemory0;
+}
+
+let cachedUint8ArrayMemory0 = null;
+function getUint8ArrayMemory0() {
+    if (cachedUint8ArrayMemory0 === null || cachedUint8ArrayMemory0.byteLength === 0) {
+        cachedUint8ArrayMemory0 = new Uint8Array(wasm.memory.buffer);
+    }
+    return cachedUint8ArrayMemory0;
+}
+
+function handleError(f, args) {
+    try {
+        return f.apply(this, args);
+    } catch (e) {
+        const idx = addToExternrefTable0(e);
+        wasm.__wbindgen_exn_store(idx);
+    }
+}
+
+function passArray16ToWasm0(arg, malloc) {
+    const ptr = malloc(arg.length * 2, 2) >>> 0;
+    getUint16ArrayMemory0().set(arg, ptr / 2);
+    WASM_VECTOR_LEN = arg.length;
+    return ptr;
+}
+
+function passArray32ToWasm0(arg, malloc) {
+    const ptr = malloc(arg.length * 4, 4) >>> 0;
+    getUint32ArrayMemory0().set(arg, ptr / 4);
+    WASM_VECTOR_LEN = arg.length;
+    return ptr;
+}
+
+function takeFromExternrefTable0(idx) {
+    const value = wasm.__wbindgen_externrefs.get(idx);
+    wasm.__externref_table_dealloc(idx);
+    return value;
+}
+
+let cachedTextDecoder = new TextDecoder('utf-8', { ignoreBOM: true, fatal: true });
+cachedTextDecoder.decode();
+const MAX_SAFARI_DECODE_BYTES = 2146435072;
+let numBytesDecoded = 0;
+function decodeText(ptr, len) {
+    numBytesDecoded += len;
+    if (numBytesDecoded >= MAX_SAFARI_DECODE_BYTES) {
+        cachedTextDecoder = new TextDecoder('utf-8', { ignoreBOM: true, fatal: true });
+        cachedTextDecoder.decode();
+        numBytesDecoded = len;
+    }
+    return cachedTextDecoder.decode(getUint8ArrayMemory0().subarray(ptr, ptr + len));
+}
+
+let WASM_VECTOR_LEN = 0;
+
+let wasmModule, wasm;
+function __wbg_finalize_init(instance, module) {
+    wasm = instance.exports;
+    wasmModule = module;
+    cachedUint16ArrayMemory0 = null;
+    cachedUint32ArrayMemory0 = null;
+    cachedUint8ArrayMemory0 = null;
+    wasm.__wbindgen_start();
+    return wasm;
+}
+
+async function __wbg_load(module, imports) {
+    if (typeof Response === 'function' && module instanceof Response) {
+        if (typeof WebAssembly.instantiateStreaming === 'function') {
+            try {
+                return await WebAssembly.instantiateStreaming(module, imports);
+            } catch (e) {
+                const validResponse = module.ok && expectedResponseType(module.type);
+
+                if (validResponse && module.headers.get('Content-Type') !== 'application/wasm') {
+                    console.warn("`WebAssembly.instantiateStreaming` failed because your server does not serve Wasm with `application/wasm` MIME type. Falling back to `WebAssembly.instantiate` which is slower. Original error:\n", e);
+
+                } else { throw e; }
+            }
+        }
+
+        const bytes = await module.arrayBuffer();
+        return await WebAssembly.instantiate(bytes, imports);
+    } else {
+        const instance = await WebAssembly.instantiate(module, imports);
+
+        if (instance instanceof WebAssembly.Instance) {
+            return { instance, module };
+        } else {
+            return instance;
+        }
+    }
+
+    function expectedResponseType(type) {
+        switch (type) {
+            case 'basic': case 'cors': case 'default': return true;
+        }
+        return false;
+    }
+}
+
+function initSync(module) {
+    if (wasm !== undefined) return wasm;
+
+
+    if (module !== undefined) {
+        if (Object.getPrototypeOf(module) === Object.prototype) {
+            ({module} = module)
+        } else {
+            console.warn('using deprecated parameters for `initSync()`; pass a single object instead')
+        }
+    }
+
+    const imports = __wbg_get_imports();
+    if (!(module instanceof WebAssembly.Module)) {
+        module = new WebAssembly.Module(module);
+    }
+    const instance = new WebAssembly.Instance(module, imports);
+    return __wbg_finalize_init(instance, module);
+}
+
+async function __wbg_init(module_or_path) {
+    if (wasm !== undefined) return wasm;
+
+
+    if (module_or_path !== undefined) {
+        if (Object.getPrototypeOf(module_or_path) === Object.prototype) {
+            ({module_or_path} = module_or_path)
+        } else {
+            console.warn('using deprecated parameters for the initialization function; pass a single object instead')
+        }
+    }
+
+    if (module_or_path === undefined) {
+        module_or_path = new URL('rusty_chromaprint_wasm_bg.wasm', import.meta.url);
+    }
+    const imports = __wbg_get_imports();
+
+    if (typeof module_or_path === 'string' || (typeof Request === 'function' && module_or_path instanceof Request) || (typeof URL === 'function' && module_or_path instanceof URL)) {
+        module_or_path = fetch(module_or_path);
+    }
+
+    const { instance, module } = await __wbg_load(await module_or_path, imports);
+
+    return __wbg_finalize_init(instance, module);
+}
+
+export { initSync, __wbg_init as default };