UNPKG

@evan/wasm

Version:
96 lines (95 loc) 3.17 kB
var __defProp = Object.defineProperty; var __markAsModule = (target) => __defProp(target, "__esModule", { value: true }); var __export = (target, all) => { __markAsModule(target); for (var name in all) __defProp(target, name, { get: all[name], enumerable: true }); }; __export(exports, { compress: () => compress, compress_raw: () => compress_raw, decompress: () => decompress, decompress_raw: () => decompress_raw, decompress_raw_with: () => decompress_raw_with, decompress_with: () => decompress_with }); let wasm; { const module2 = new WebAssembly.Module(require('fs').readFileSync(require('path').join(__dirname, '../../src/zlib/simd.wasm'))); const instance = new WebAssembly.Instance(module2, { wasi_snapshot_preview1: { fd_write() { }, proc_exit() { } }, env: { __sys_getcwd() { }, emscripten_notify_memory_growth() { } } }); wasm = instance.exports; } class mem { static length() { return wasm.wlen(); } static alloc(size) { return wasm.walloc(size); } static free(ptr, size) { return wasm.wfree(ptr, size); } static u8(ptr, size) { return new Uint8Array(wasm.memory.buffer, ptr, size); } static u32(ptr, size) { return new Uint32Array(wasm.memory.buffer, ptr, size); } static copy_and_free(ptr, size) { let slice = mem.u8(ptr, size).slice(); return wasm.wfree(ptr, size), slice; } } function compress(buffer, level = 3) { const ptr = mem.alloc(buffer.length); mem.u8(ptr, buffer.length).set(buffer); return mem.copy_and_free(wasm.compress(ptr, buffer.length, level), mem.length()); } function compress_raw(buffer, level = 3) { const ptr = mem.alloc(buffer.length); mem.u8(ptr, buffer.length).set(buffer); return mem.copy_and_free(wasm.compress_raw(ptr, buffer.length, level), mem.length()); } function decompress(buffer, limit = 0) { const ptr = mem.alloc(buffer.length); mem.u8(ptr, buffer.length).set(buffer); const x = wasm.decompress(ptr, buffer.length, limit); if (x === 0) throw new Error("zlib: failed to decompress"); return mem.copy_and_free(x, mem.length()); } function decompress_raw(buffer, limit = 0) { const ptr = mem.alloc(buffer.length); mem.u8(ptr, buffer.length).set(buffer); const x = wasm.decompress_raw(ptr, buffer.length, limit); if (x === 0) throw new Error("zlib: failed to decompress (raw)"); return mem.copy_and_free(x, mem.length()); } function decompress_with(buffer, limit = 0, transform) { const ptr = mem.alloc(buffer.length); mem.u8(ptr, buffer.length).set(buffer); const x = wasm.decompress(ptr, buffer.length, limit); if (x === 0) throw new Error("zlib: failed to decompress"); const u8 = mem.u8(x, mem.length()); const value = transform(u8); return mem.free(x, u8.length), value; } function decompress_raw_with(buffer, limit = 0, transform) { const ptr = mem.alloc(buffer.length); mem.u8(ptr, buffer.length).set(buffer); const x = wasm.decompress_raw(ptr, buffer.length, limit); if (x === 0) throw new Error("zlib: failed to decompress (raw)"); const u8 = mem.u8(x, mem.length()); const value = transform(u8); return mem.free(x, u8.length), value; }