@cognipilot/rumoca 0.9.4

package/rumoca_bind_wasm_diffsol.d.ts

@@ -0,0 +1,49 @@
+/* tslint:disable */
+/* eslint-disable */
+
+/**
+ * Simulate a pre-lowered model with the stiff/implicit diffsol backend.
+ * `input_json` is the `{ solve_model, t_end, dt }` payload from the main
+ * module's `lower_model_to_solve_json`. Returns the same `{ "payload": ... }`
+ * JSON shape as the main module's `simulate_model`, so callers treat both
+ * paths uniformly.
+ */
+export function simulate_solve_model_diffsol(input_json: string): string;
+
+export function start(): void;
+
+export type InitInput = RequestInfo | URL | Response | BufferSource | WebAssembly.Module;
+
+export interface InitOutput {
+    readonly memory: WebAssembly.Memory;
+    readonly simulate_solve_model_diffsol: (a: number, b: number) => [number, number, number, number];
+    readonly start: () => void;
+    readonly __wbindgen_free: (a: number, b: number, c: number) => void;
+    readonly __wbindgen_malloc: (a: number, b: number) => number;
+    readonly __wbindgen_realloc: (a: number, b: number, c: number, d: number) => number;
+    readonly __wbindgen_externrefs: WebAssembly.Table;
+    readonly __externref_table_dealloc: (a: 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/rumoca_bind_wasm_diffsol.js

@@ -0,0 +1,270 @@
+/* @ts-self-types="./rumoca_bind_wasm_diffsol.d.ts" */
+
+/**
+ * Simulate a pre-lowered model with the stiff/implicit diffsol backend.
+ * `input_json` is the `{ solve_model, t_end, dt }` payload from the main
+ * module's `lower_model_to_solve_json`. Returns the same `{ "payload": ... }`
+ * JSON shape as the main module's `simulate_model`, so callers treat both
+ * paths uniformly.
+ * @param {string} input_json
+ * @returns {string}
+ */
+export function simulate_solve_model_diffsol(input_json) {
+    let deferred3_0;
+    let deferred3_1;
+    try {
+        const ptr0 = passStringToWasm0(input_json, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ret = wasm.simulate_solve_model_diffsol(ptr0, len0);
+        var ptr2 = ret[0];
+        var len2 = ret[1];
+        if (ret[3]) {
+            ptr2 = 0; len2 = 0;
+            throw takeFromExternrefTable0(ret[2]);
+        }
+        deferred3_0 = ptr2;
+        deferred3_1 = len2;
+        return getStringFromWasm0(ptr2, len2);
+    } finally {
+        wasm.__wbindgen_free(deferred3_0, deferred3_1, 1);
+    }
+}
+
+export function start() {
+    wasm.start();
+}
+
+function __wbg_get_imports() {
+    const import0 = {
+        __proto__: null,
+        __wbg___wbindgen_throw_be289d5034ed271b: function(arg0, arg1) {
+            throw new Error(getStringFromWasm0(arg0, arg1));
+        },
+        __wbg_error_7534b8e9a36f1ab4: function(arg0, arg1) {
+            let deferred0_0;
+            let deferred0_1;
+            try {
+                deferred0_0 = arg0;
+                deferred0_1 = arg1;
+                console.error(getStringFromWasm0(arg0, arg1));
+            } finally {
+                wasm.__wbindgen_free(deferred0_0, deferred0_1, 1);
+            }
+        },
+        __wbg_new_8a6f238a6ece86ea: function() {
+            const ret = new Error();
+            return ret;
+        },
+        __wbg_stack_0ed75d68575b0f3c: function(arg0, arg1) {
+            const ret = arg1.stack;
+            const ptr1 = passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+            const len1 = WASM_VECTOR_LEN;
+            getDataViewMemory0().setInt32(arg0 + 4 * 1, len1, true);
+            getDataViewMemory0().setInt32(arg0 + 4 * 0, ptr1, true);
+        },
+        __wbindgen_cast_0000000000000001: 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,
+        "./rumoca_bind_wasm_diffsol_bg.js": import0,
+    };
+}
+
+let cachedDataViewMemory0 = null;
+function getDataViewMemory0() {
+    if (cachedDataViewMemory0 === null || cachedDataViewMemory0.buffer.detached === true || (cachedDataViewMemory0.buffer.detached === undefined && cachedDataViewMemory0.buffer !== wasm.memory.buffer)) {
+        cachedDataViewMemory0 = new DataView(wasm.memory.buffer);
+    }
+    return cachedDataViewMemory0;
+}
+
+function getStringFromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return decodeText(ptr, len);
+}
+
+let cachedUint8ArrayMemory0 = null;
+function getUint8ArrayMemory0() {
+    if (cachedUint8ArrayMemory0 === null || cachedUint8ArrayMemory0.byteLength === 0) {
+        cachedUint8ArrayMemory0 = new Uint8Array(wasm.memory.buffer);
+    }
+    return cachedUint8ArrayMemory0;
+}
+
+function passStringToWasm0(arg, malloc, realloc) {
+    if (realloc === undefined) {
+        const buf = cachedTextEncoder.encode(arg);
+        const ptr = malloc(buf.length, 1) >>> 0;
+        getUint8ArrayMemory0().subarray(ptr, ptr + buf.length).set(buf);
+        WASM_VECTOR_LEN = buf.length;
+        return ptr;
+    }
+
+    let len = arg.length;
+    let ptr = malloc(len, 1) >>> 0;
+
+    const mem = getUint8ArrayMemory0();
+
+    let offset = 0;
+
+    for (; offset < len; offset++) {
+        const code = arg.charCodeAt(offset);
+        if (code > 0x7F) break;
+        mem[ptr + offset] = code;
+    }
+    if (offset !== len) {
+        if (offset !== 0) {
+            arg = arg.slice(offset);
+        }
+        ptr = realloc(ptr, len, len = offset + arg.length * 3, 1) >>> 0;
+        const view = getUint8ArrayMemory0().subarray(ptr + offset, ptr + len);
+        const ret = cachedTextEncoder.encodeInto(arg, view);
+
+        offset += ret.written;
+        ptr = realloc(ptr, len, offset, 1) >>> 0;
+    }
+
+    WASM_VECTOR_LEN = offset;
+    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));
+}
+
+const cachedTextEncoder = new TextEncoder();
+
+if (!('encodeInto' in cachedTextEncoder)) {
+    cachedTextEncoder.encodeInto = function (arg, view) {
+        const buf = cachedTextEncoder.encode(arg);
+        view.set(buf);
+        return {
+            read: arg.length,
+            written: buf.length
+        };
+    };
+}
+
+let WASM_VECTOR_LEN = 0;
+
+let wasmModule, wasm;
+function __wbg_finalize_init(instance, module) {
+    wasm = instance.exports;
+    wasmModule = module;
+    cachedDataViewMemory0 = 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('rumoca_bind_wasm_diffsol_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 };

package/rumoca_diffsol.js

@@ -0,0 +1,109 @@
+// Lazy diffsol (stiff/implicit) simulation driver for the @cognipilot/rumoca
+// package.
+//
+// Why this exists: the main rumoca WASM module is SIMD-free and loads on every
+// browser. The diffsol solver pulls relaxed-SIMD (via faer/pulp), and WASM
+// validates a whole module at instantiation — so a single combined module
+// would hard-fail the entire package on browsers without relaxed-SIMD. Instead
+// the diffsol code ships as a *separate* tiny module (~0.4 MB gz,
+// `rumoca_bind_wasm_diffsol.js`) that is loaded only when (a) the browser
+// supports relaxed-SIMD and (b) the user actually runs a stiff solve.
+//
+// Usage (the main module is already initialized by the caller):
+//   import { diffsolAvailable, simulateWithDiffsol } from "@cognipilot/rumoca/diffsol";
+//   const ok = await diffsolAvailable(pkgBase);          // grey out the UI if false
+//   const result = await simulateWithDiffsol(mainModule, pkgBase, src, model, tEnd, dt);
+//
+// `pkgBase` is the same base URL the main `rumoca_bind_wasm.js` was imported
+// from (the addon wasm sits next to it).
+
+// Minimal module that uses a 1-operand relaxed-SIMD op
+// (`i32x4.relaxed_trunc_f32x4_s`, 0xFD 0x101). `WebAssembly.validate` returns
+// false on engines without relaxed-SIMD. This is a cheap *hint* for greying out
+// the UI; the authoritative check is whether the addon actually instantiates
+// (see `diffsolAvailable`). NOTE: verify in a real browser as part of the
+// editor smoke test.
+const RELAXED_SIMD_PROBE = new Uint8Array([
+  0x00, 0x61, 0x73, 0x6d, 0x01, 0x00, 0x00, 0x00, // \0asm v1
+  0x01, 0x05, 0x01, 0x60, 0x00, 0x01, 0x7b,       // type: () -> v128
+  0x03, 0x02, 0x01, 0x00,                         // func 0 : type 0
+  0x0a, 0x19, 0x01, 0x17, 0x00,                   // code: 1 body, size 23, 0 locals
+  0xfd, 0x0c, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // v128.const 0
+  0xfd, 0x81, 0x02,                               // i32x4.relaxed_trunc_f32x4_s
+  0x0b,                                           // end
+]);
+
+/** Quick, non-authoritative hint: does this engine validate relaxed-SIMD? */
+export function relaxedSimdSupported() {
+  try {
+    return WebAssembly.validate(RELAXED_SIMD_PROBE);
+  } catch {
+    return false;
+  }
+}
+
+let addonPromise = null;
+
+/**
+ * Lazily import + initialize the diffsol addon module from `pkgBase`. Resolves
+ * to the module's exports, or rejects (e.g. `CompileError` on a browser without
+ * relaxed-SIMD). Cached after the first call.
+ */
+export function loadDiffsolAddon(pkgBase) {
+  if (!addonPromise) {
+    addonPromise = (async () => {
+      const mod = await import(pkgBase + "rumoca_bind_wasm_diffsol.js");
+      await mod.default(); // run wasm-bindgen init (instantiates the wasm)
+      return mod;
+    })();
+    // Allow a retry if loading failed (e.g. transient network).
+    addonPromise.catch(() => {
+      addonPromise = null;
+    });
+  }
+  return addonPromise;
+}
+
+/**
+ * Authoritative check: true iff the diffsol addon actually instantiates in this
+ * browser. Downloads/compiles the addon once (cached). Use this to enable or
+ * grey out the stiff-solver options.
+ */
+export async function diffsolAvailable(pkgBase) {
+  try {
+    await loadDiffsolAddon(pkgBase);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * Run a stiff (diffsol) simulation: the main module lowers the model to a
+ * SolveModel JSON, the addon simulates it. Returns the parsed
+ * `{ payload, ... }` object (same shape as the main module's `simulate_model`).
+ * Throws a clear error if the addon can't load (old browser) — callers should
+ * fall back to rk45 / the GPU path or surface the message.
+ */
+export async function simulateWithDiffsol(mainModule, pkgBase, source, modelName, tEnd, dt) {
+  if (typeof mainModule.lower_model_to_solve_json !== "function") {
+    throw new Error(
+      "this rumoca build predates the diffsol addon; rebuild the package",
+    );
+  }
+  let addon;
+  try {
+    addon = await loadDiffsolAddon(pkgBase);
+  } catch (err) {
+    throw new Error(
+      "the stiff (diffsol) solver needs a browser with WebAssembly relaxed-SIMD " +
+        "(Chrome 114+, Firefox 120+, Safari 16.4+). Use the rk45 solver or the GPU " +
+        `path instead. (${err && err.message ? err.message : err})`,
+    );
+  }
+  // The payload carries the lowered model plus the resolved t_end/dt (so an
+  // annotation-driven duration is honored even when the caller passes 0).
+  const prepJson = mainModule.lower_model_to_solve_json(source, modelName, tEnd, dt);
+  const resultJson = addon.simulate_solve_model_diffsol(prepJson);
+  return JSON.parse(resultJson);
+}