deepfilternet3-assets 1.0.0 → 1.0.2

package/README.md

@@ -1,13 +1,13 @@
-# @mezon/deepfilternet3-assets
+# deepfilternet3-assets
 
 DeepFilterNet3 WASM and model assets for noise suppression.
 
 ## Installation
 
 ```bash
-npm install @mezon/deepfilternet3-assets
+npm install deepfilternet3-assets
 # or
-yarn add @mezon/deepfilternet3-assets
+yarn add deepfilternet3-assets
 ```
 
 ## Contents
@@ -21,7 +21,7 @@ This package includes:
 ### Getting Asset Paths (Node.js)
 
 ```javascript
-const { getWasmPath, getModelPath, getAssetPaths } = require('@mezon/deepfilternet3-assets');
+const { getWasmPath, getModelPath, getAssetPaths } = require('deepfilternet3-assets');
 
 // Get individual paths
 const wasmPath = getWasmPath();
@@ -36,7 +36,7 @@ const { wasm, model } = getAssetPaths();
 For web applications, you can use the relative paths:
 
 ```javascript
-const { WASM_RELATIVE_PATH, MODEL_RELATIVE_PATH } = require('@mezon/deepfilternet3-assets');
+const { WASM_RELATIVE_PATH, MODEL_RELATIVE_PATH } = require('deepfilternet3-assets');
 
 // Then use with your bundler's asset handling
 // For example, with Webpack or Vite
@@ -48,7 +48,7 @@ You can copy these files to your static assets directory or CDN:
 
 ```javascript
 const fs = require('fs');
-const { getAssetPaths } = require('@mezon/deepfilternet3-assets');
+const { getAssetPaths } = require('deepfilternet3-assets');
 
 const assets = getAssetPaths();
 
@@ -62,7 +62,7 @@ fs.copyFileSync(assets.model, './public/models/DeepFilterNet3_onnx.tar.gz');
 This package includes TypeScript type definitions:
 
 ```typescript
-import { getAssetPaths, WASM_RELATIVE_PATH, MODEL_RELATIVE_PATH } from '@mezon/deepfilternet3-assets';
+import { getAssetPaths, WASM_RELATIVE_PATH, MODEL_RELATIVE_PATH } from 'deepfilternet3-assets';
 
 const { wasm, model } = getAssetPaths();
 

package/package.json

@@ -1,11 +1,13 @@
 {
     "name": "deepfilternet3-assets",
-    "version": "1.0.0",
+    "version": "1.0.2",
     "description": "DeepFilterNet3 WASM and model assets for noise suppression",
     "main": "index.js",
     "types": "index.d.ts",
     "files": [
         "pkg/df_bg.wasm",
+        "pkg/df.js",
+        "pkg/df.d.ts",
         "models/DeepFilterNet3_onnx.tar.gz",
         "index.js",
         "index.d.ts",

package/pkg/df.d.ts

@@ -0,0 +1,104 @@
+/* tslint:disable */
+/* eslint-disable */
+/**
+* Create a DeepFilterNet Model
+*
+* Args:
+*     - path: File path to a DeepFilterNet tar.gz onnx model
+*     - atten_lim: Attenuation limit in dB.
+*
+* Returns:
+*     - DF state doing the full processing: stft, DNN noise reduction, istft.
+* @param {Uint8Array} model_bytes
+* @param {number} atten_lim
+* @returns {number}
+*/
+export function df_create(model_bytes: Uint8Array, atten_lim: number): number;
+/**
+* Get DeepFilterNet frame size in samples.
+* @param {number} st
+* @returns {number}
+*/
+export function df_get_frame_length(st: number): number;
+/**
+* @param {number} st
+* @param {boolean} enabled
+* @param {number} desired_output_rms
+* @param {number} distortion_factor
+* @param {number} snr_thresh
+*/
+export function df_set_agc_params(st: number, enabled: boolean, desired_output_rms: number, distortion_factor: number, snr_thresh: number): void;
+/**
+* Set DeepFilterNet attenuation limit.
+*
+* Args:
+*     - lim_db: New attenuation limit in dB.
+* @param {number} st
+* @param {number} lim_db
+*/
+export function df_set_atten_lim(st: number, lim_db: number): void;
+/**
+* Set DeepFilterNet post filter beta. A beta of 0 disables the post filter.
+*
+* Args:
+*     - beta: Post filter attenuation. Suitable range between 0.05 and 0;
+* @param {number} st
+* @param {number} beta
+*/
+export function df_set_post_filter_beta(st: number, beta: number): void;
+/**
+* Processes a chunk of samples.
+*
+* Args:
+*     - df_state: Created via df_create()
+*     - input: Input buffer of length df_get_frame_length()
+*     - output: Output buffer of length df_get_frame_length()
+*
+* Returns:
+*     - Local SNR of the current frame.
+* @param {number} st
+* @param {Float32Array} input
+* @returns {Float32Array}
+*/
+export function df_process_frame(st: number, input: Float32Array): Float32Array;
+/**
+*/
+export class DFState {
+  free(): void;
+}
+
+export type InitInput = RequestInfo | URL | Response | BufferSource | WebAssembly.Module;
+
+export interface InitOutput {
+  readonly memory: WebAssembly.Memory;
+  readonly __wbg_dfstate_free: (a: number) => void;
+  readonly df_create: (a: number, b: number, c: number) => number;
+  readonly df_get_frame_length: (a: number) => number;
+  readonly df_process_frame: (a: number, b: number, c: number) => number;
+  readonly df_set_agc_params: (a: number, b: number, c: number, d: number, e: number) => void;
+  readonly df_set_atten_lim: (a: number, b: number) => void;
+  readonly df_set_post_filter_beta: (a: number, b: number) => void;
+  readonly __wbindgen_malloc: (a: number, b: number) => number;
+  readonly __wbindgen_exn_store: (a: number) => void;
+}
+
+export type SyncInitInput = BufferSource | WebAssembly.Module;
+/**
+* Instantiates the given `module`, which can either be bytes or
+* a precompiled `WebAssembly.Module`.
+*
+* @param {SyncInitInput} module
+*
+* @returns {InitOutput}
+*/
+export function initSync(module: SyncInitInput): InitOutput;
+
+/**
+* If `module_or_path` is {RequestInfo} or {URL}, makes a request and
+* for everything else, calls `WebAssembly.instantiate` directly.
+*
+* @param {InitInput | Promise<InitInput>} module_or_path
+*
+* @returns {Promise<InitOutput>}
+*/
+export default function __wbg_init (module_or_path?: InitInput | Promise<InitInput>): Promise<InitOutput>;

package/pkg/df.js

@@ -0,0 +1,399 @@
+let wasm;
+
+const heap = new Array(128).fill(undefined);
+
+heap.push(undefined, null, true, false);
+
+function getObject(idx) { return heap[idx]; }
+
+let heap_next = heap.length;
+
+function dropObject(idx) {
+    if (idx < 132) return;
+    heap[idx] = heap_next;
+    heap_next = idx;
+}
+
+function takeObject(idx) {
+    const ret = getObject(idx);
+    dropObject(idx);
+    return ret;
+}
+
+const cachedTextDecoder = (typeof TextDecoder !== 'undefined' ? new TextDecoder('utf-8', { ignoreBOM: true, fatal: true }) : { decode: () => { throw Error('TextDecoder not available') } } );
+
+if (typeof TextDecoder !== 'undefined') { cachedTextDecoder.decode(); };
+
+let cachedUint8Memory0 = null;
+
+function getUint8Memory0() {
+    if (cachedUint8Memory0 === null || cachedUint8Memory0.byteLength === 0) {
+        cachedUint8Memory0 = new Uint8Array(wasm.memory.buffer);
+    }
+    return cachedUint8Memory0;
+}
+
+function getStringFromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return cachedTextDecoder.decode(getUint8Memory0().subarray(ptr, ptr + len));
+}
+
+function addHeapObject(obj) {
+    if (heap_next === heap.length) heap.push(heap.length + 1);
+    const idx = heap_next;
+    heap_next = heap[idx];
+
+    heap[idx] = obj;
+    return idx;
+}
+
+let WASM_VECTOR_LEN = 0;
+
+function passArray8ToWasm0(arg, malloc) {
+    const ptr = malloc(arg.length * 1, 1) >>> 0;
+    getUint8Memory0().set(arg, ptr / 1);
+    WASM_VECTOR_LEN = arg.length;
+    return ptr;
+}
+/**
+* Create a DeepFilterNet Model
+*
+* Args:
+*     - path: File path to a DeepFilterNet tar.gz onnx model
+*     - atten_lim: Attenuation limit in dB.
+*
+* Returns:
+*     - DF state doing the full processing: stft, DNN noise reduction, istft.
+* @param {Uint8Array} model_bytes
+* @param {number} atten_lim
+* @returns {number}
+*/
+export function df_create(model_bytes, atten_lim) {
+    const ptr0 = passArray8ToWasm0(model_bytes, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.df_create(ptr0, len0, atten_lim);
+    return ret >>> 0;
+}
+
+/**
+* Get DeepFilterNet frame size in samples.
+* @param {number} st
+* @returns {number}
+*/
+export function df_get_frame_length(st) {
+    const ret = wasm.df_get_frame_length(st);
+    return ret >>> 0;
+}
+
+/**
+* @param {number} st
+* @param {boolean} enabled
+* @param {number} desired_output_rms
+* @param {number} distortion_factor
+* @param {number} snr_thresh
+*/
+export function df_set_agc_params(st, enabled, desired_output_rms, distortion_factor, snr_thresh) {
+    wasm.df_set_agc_params(st, enabled, desired_output_rms, distortion_factor, snr_thresh);
+}
+
+/**
+* Set DeepFilterNet attenuation limit.
+*
+* Args:
+*     - lim_db: New attenuation limit in dB.
+* @param {number} st
+* @param {number} lim_db
+*/
+export function df_set_atten_lim(st, lim_db) {
+    wasm.df_set_atten_lim(st, lim_db);
+}
+
+/**
+* Set DeepFilterNet post filter beta. A beta of 0 disables the post filter.
+*
+* Args:
+*     - beta: Post filter attenuation. Suitable range between 0.05 and 0;
+* @param {number} st
+* @param {number} beta
+*/
+export function df_set_post_filter_beta(st, beta) {
+    wasm.df_set_post_filter_beta(st, beta);
+}
+
+let cachedFloat32Memory0 = null;
+
+function getFloat32Memory0() {
+    if (cachedFloat32Memory0 === null || cachedFloat32Memory0.byteLength === 0) {
+        cachedFloat32Memory0 = new Float32Array(wasm.memory.buffer);
+    }
+    return cachedFloat32Memory0;
+}
+
+function passArrayF32ToWasm0(arg, malloc) {
+    const ptr = malloc(arg.length * 4, 4) >>> 0;
+    getFloat32Memory0().set(arg, ptr / 4);
+    WASM_VECTOR_LEN = arg.length;
+    return ptr;
+}
+/**
+* Processes a chunk of samples.
+*
+* Args:
+*     - df_state: Created via df_create()
+*     - input: Input buffer of length df_get_frame_length()
+*     - output: Output buffer of length df_get_frame_length()
+*
+* Returns:
+*     - Local SNR of the current frame.
+* @param {number} st
+* @param {Float32Array} input
+* @returns {Float32Array}
+*/
+export function df_process_frame(st, input) {
+    const ptr0 = passArrayF32ToWasm0(input, wasm.__wbindgen_malloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.df_process_frame(st, ptr0, len0);
+    return takeObject(ret);
+}
+
+function handleError(f, args) {
+    try {
+        return f.apply(this, args);
+    } catch (e) {
+        wasm.__wbindgen_exn_store(addHeapObject(e));
+    }
+}
+
+const DFStateFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_dfstate_free(ptr >>> 0));
+/**
+*/
+export class DFState {
+
+    __destroy_into_raw() {
+        const ptr = this.__wbg_ptr;
+        this.__wbg_ptr = 0;
+        DFStateFinalization.unregister(this);
+        return ptr;
+    }
+
+    free() {
+        const ptr = this.__destroy_into_raw();
+        wasm.__wbg_dfstate_free(ptr);
+    }
+}
+
+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) {
+                if (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 __wbg_get_imports() {
+    const imports = {};
+    imports.wbg = {};
+    imports.wbg.__wbindgen_object_drop_ref = function(arg0) {
+        takeObject(arg0);
+    };
+    imports.wbg.__wbg_crypto_566d7465cdbb6b7a = function(arg0) {
+        const ret = getObject(arg0).crypto;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbindgen_is_object = function(arg0) {
+        const val = getObject(arg0);
+        const ret = typeof(val) === 'object' && val !== null;
+        return ret;
+    };
+    imports.wbg.__wbg_process_dc09a8c7d59982f6 = function(arg0) {
+        const ret = getObject(arg0).process;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_versions_d98c6400c6ca2bd8 = function(arg0) {
+        const ret = getObject(arg0).versions;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_node_caaf83d002149bd5 = function(arg0) {
+        const ret = getObject(arg0).node;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbindgen_is_string = function(arg0) {
+        const ret = typeof(getObject(arg0)) === 'string';
+        return ret;
+    };
+    imports.wbg.__wbg_require_94a9da52636aacbf = function() { return handleError(function () {
+        const ret = module.require;
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbindgen_is_function = function(arg0) {
+        const ret = typeof(getObject(arg0)) === 'function';
+        return ret;
+    };
+    imports.wbg.__wbindgen_string_new = function(arg0, arg1) {
+        const ret = getStringFromWasm0(arg0, arg1);
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_msCrypto_0b84745e9245cdf6 = function(arg0) {
+        const ret = getObject(arg0).msCrypto;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_randomFillSync_290977693942bf03 = function() { return handleError(function (arg0, arg1) {
+        getObject(arg0).randomFillSync(takeObject(arg1));
+    }, arguments) };
+    imports.wbg.__wbg_getRandomValues_260cc23a41afad9a = function() { return handleError(function (arg0, arg1) {
+        getObject(arg0).getRandomValues(getObject(arg1));
+    }, arguments) };
+    imports.wbg.__wbg_newnoargs_e258087cd0daa0ea = function(arg0, arg1) {
+        const ret = new Function(getStringFromWasm0(arg0, arg1));
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_new_63b92bc8671ed464 = function(arg0) {
+        const ret = new Uint8Array(getObject(arg0));
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_new_9efabd6b6d2ce46d = function(arg0) {
+        const ret = new Float32Array(getObject(arg0));
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_buffer_12d079cc21e14bdb = function(arg0) {
+        const ret = getObject(arg0).buffer;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_newwithbyteoffsetandlength_aa4a17c33a06e5cb = function(arg0, arg1, arg2) {
+        const ret = new Uint8Array(getObject(arg0), arg1 >>> 0, arg2 >>> 0);
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_newwithlength_e9b4878cebadb3d3 = function(arg0) {
+        const ret = new Uint8Array(arg0 >>> 0);
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_set_a47bac70306a19a7 = function(arg0, arg1, arg2) {
+        getObject(arg0).set(getObject(arg1), arg2 >>> 0);
+    };
+    imports.wbg.__wbg_subarray_a1f73cd4b5b42fe1 = function(arg0, arg1, arg2) {
+        const ret = getObject(arg0).subarray(arg1 >>> 0, arg2 >>> 0);
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_newwithbyteoffsetandlength_4a659d079a1650e0 = function(arg0, arg1, arg2) {
+        const ret = new Float32Array(getObject(arg0), arg1 >>> 0, arg2 >>> 0);
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_self_ce0dbfc45cf2f5be = function() { return handleError(function () {
+        const ret = self.self;
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbg_window_c6fb939a7f436783 = function() { return handleError(function () {
+        const ret = window.window;
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbg_globalThis_d1e6af4856ba331b = function() { return handleError(function () {
+        const ret = globalThis.globalThis;
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbg_global_207b558942527489 = function() { return handleError(function () {
+        const ret = global.global;
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbindgen_is_undefined = function(arg0) {
+        const ret = getObject(arg0) === undefined;
+        return ret;
+    };
+    imports.wbg.__wbg_call_27c0f87801dedf93 = function() { return handleError(function (arg0, arg1) {
+        const ret = getObject(arg0).call(getObject(arg1));
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbindgen_object_clone_ref = function(arg0) {
+        const ret = getObject(arg0);
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbg_call_b3ca7c6051f9bec1 = function() { return handleError(function (arg0, arg1, arg2) {
+        const ret = getObject(arg0).call(getObject(arg1), getObject(arg2));
+        return addHeapObject(ret);
+    }, arguments) };
+    imports.wbg.__wbindgen_memory = function() {
+        const ret = wasm.memory;
+        return addHeapObject(ret);
+    };
+    imports.wbg.__wbindgen_throw = function(arg0, arg1) {
+        throw new Error(getStringFromWasm0(arg0, arg1));
+    };
+
+    return imports;
+}
+
+function __wbg_init_memory(imports, maybe_memory) {
+
+}
+
+function __wbg_finalize_init(instance, module) {
+    wasm = instance.exports;
+    __wbg_init.__wbindgen_wasm_module = module;
+    cachedFloat32Memory0 = null;
+    cachedUint8Memory0 = null;
+
+
+    return wasm;
+}
+
+function initSync(module) {
+    if (wasm !== undefined) return wasm;
+
+    const imports = __wbg_get_imports();
+
+    __wbg_init_memory(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(input) {
+    if (wasm !== undefined) return wasm;
+
+    if (typeof input === 'undefined') {
+        input = new URL('df_bg.wasm', import.meta.url);
+    }
+    const imports = __wbg_get_imports();
+
+    if (typeof input === 'string' || (typeof Request === 'function' && input instanceof Request) || (typeof URL === 'function' && input instanceof URL)) {
+        input = fetch(input);
+    }
+
+    __wbg_init_memory(imports);
+
+    const { instance, module } = await __wbg_load(await input, imports);
+
+    return __wbg_finalize_init(instance, module);
+}
+
+export { initSync }
+export default __wbg_init;