@danielsimonjr/mathts-functions 0.25.1 → 0.26.0

package/dist/index.js

@@ -36301,6 +36301,11 @@ var createSolveODE = /* @__PURE__ */ factory(
       }
       return true;
     }
+    function _rmsNorm(v) {
+      let s = 0;
+      for (let i = 0; i < v.length; i++) s += v[i] * v[i];
+      return Math.sqrt(s / v.length);
+    }
     function _rkWasmLoop(f, tspan, y0, options, butcherTableau) {
       const wasm = wasmLoader.getModule();
       if (!wasm) return null;
@@ -36313,12 +36318,21 @@ var createSolveODE = /* @__PURE__ */ factory(
       const t0 = tspan[0];
       const tf = tspan[1];
       const isForwards = tf > t0;
-      const steps = 1;
       const tol = options.tol ? options.tol : 1e-4;
       const minDelta = options.minDelta ? options.minDelta : 0.2;
       const maxDelta = options.maxDelta ? options.maxDelta : 5;
       const maxIter = options.maxIter ? options.maxIter : 1e4;
-      let h = options.firstStep ? isForwards ? options.firstStep : -options.firstStep : (tf - t0) / steps;
+      let h;
+      if (options.firstStep) {
+        h = isForwards ? options.firstStep : -options.firstStep;
+      } else {
+        const f0 = f(t0, y0);
+        const d0 = _rmsNorm(y0);
+        const d1 = _rmsNorm(Array.isArray(f0) ? f0 : [f0]);
+        const h0 = d0 < 1e-5 || d1 < 1e-5 ? 1e-6 : 0.01 * (d0 / d1);
+        const hMag = Math.min(h0, Math.abs(tf - t0));
+        h = isForwards ? hMag : -hMag;
+      }
       const maxStepVal = options.maxStep;
       const minStepVal = options.minStep;
       const deltaB = new Float64Array(numStages);
@@ -36468,10 +36482,31 @@ var createSolveODE = /* @__PURE__ */ factory(
           bignumber2(butcherTableau.b),
           bignumber2(butcherTableau.bp)
         ] : [butcherTableau.a, butcherTableau.c, butcherTableau.b, butcherTableau.bp];
-        let h = firstStep ? isForwards ? firstStep : unaryMinus2(firstStep) : divide2(subtract2(tf, t0), steps);
+        let h;
+        if (firstStep) {
+          h = isForwards ? firstStep : unaryMinus2(firstStep);
+        } else if (!hasBigNumbers && !tspan.some(isUnit) && y0.every((v) => typeof v === "number")) {
+          const f0 = f(t0, y0);
+          const f0arr = Array.isArray(f0) ? f0 : [f0];
+          const d0 = _rmsNorm(y0);
+          const d1 = _rmsNorm(f0arr);
+          const h0 = d0 < 1e-5 || d1 < 1e-5 ? 1e-6 : 0.01 * (d0 / d1);
+          const hMag = Math.min(h0, Math.abs(tf - t0));
+          h = isForwards ? hMag : -hMag;
+        } else {
+          h = divide2(subtract2(tf, t0), steps);
+        }
         const t = [t0];
         const y = [y0];
         const deltaB = subtract2(b, bp);
+        function stageCombo(coeffs, stages) {
+          const m = Math.min(coeffs.length, stages.length);
+          let acc = multiply2(coeffs[0], stages[0]);
+          for (let j = 1; j < m; j++) {
+            acc = add2(acc, multiply2(coeffs[j], stages[j]));
+          }
+          return acc;
+        }
         let n = 0;
         let iter = 0;
         const ongoing = _createOngoing(isForwards);
@@ -36481,14 +36516,14 @@ var createSolveODE = /* @__PURE__ */ factory(
           h = trimStep(t[n], tf, h);
           k.push(f(t[n], y[n]));
           for (let i = 1; i < c.length; ++i) {
-            k.push(f(add2(t[n], multiply2(c[i], h)), add2(y[n], multiply2(h, a[i], k))));
+            k.push(f(add2(t[n], multiply2(c[i], h)), add2(y[n], multiply2(h, stageCombo(a[i], k)))));
           }
-          const TE = max2(
-            abs2(map2(multiply2(deltaB, k), (X) => isUnit(X) ? X.value : X))
-          );
+          const errComb = stageCombo(deltaB, k);
+          const errArr = Array.isArray(errComb) ? errComb : [errComb];
+          const TE = max2(abs2(map2(errArr, (X) => isUnit(X) ? X.value : X)));
           if (TE < tol && tol / TE > 1 / 4) {
             t.push(add2(t[n], h));
-            y.push(add2(y[n], multiply2(h, b, k)));
+            y.push(add2(y[n], multiply2(h, stageCombo(b, k))));
             n++;
           }
           let delta = 0.84 * (tol / TE) ** (1 / 5);

package/dist/numeric/solveODE.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"solveODE.d.ts","sourceRoot":"","sources":["../../src/numeric/solveODE.ts"],"names":[],"mappings":"AAmEA,eAAO,MAAM,cAAc,yDA8kB1B,CAAC"}
+{"version":3,"file":"solveODE.d.ts","sourceRoot":"","sources":["../../src/numeric/solveODE.ts"],"names":[],"mappings":"AAmEA,eAAO,MAAM,cAAc,yDA8nB1B,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@danielsimonjr/mathts-functions",
-  "version": "0.25.1",
+  "version": "0.26.0",
   "description": "Mathematical functions for MathTS - arithmetic, algebra, trigonometry, statistics, and more",
   "author": "Daniel Simon Jr.",
   "license": "MIT",