diff-grok 1.0.10 → 1.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (212) hide show
  1. package/.claude/settings.local.json +11 -0
  2. package/CLAUDE.md +98 -0
  3. package/README.MD +100 -24
  4. package/images/E5.png +0 -0
  5. package/images/HIRES.png +0 -0
  6. package/images/OREGO.png +0 -0
  7. package/images/POLL-full.png +0 -0
  8. package/images/ROBER.png +0 -0
  9. package/images/VDP0L.png +0 -0
  10. package/index.js +1 -1
  11. package/index.js.map +1 -1
  12. package/index.ts +2 -1
  13. package/jest.config.js +1 -0
  14. package/package.json +1 -1
  15. package/src/examples/bioreactor.js +125 -0
  16. package/src/examples/bioreactor.js.map +1 -0
  17. package/src/examples/bioreactor.ts +132 -0
  18. package/src/examples/cyclic-model.js +3 -1
  19. package/src/examples/cyclic-model.js.map +1 -1
  20. package/src/examples/cyclic-model.ts +3 -1
  21. package/src/solver-tools/ab4-method.js +332 -0
  22. package/src/solver-tools/ab4-method.js.map +1 -0
  23. package/src/solver-tools/ab4-method.ts +395 -0
  24. package/src/solver-tools/ab5-method.js +334 -0
  25. package/src/solver-tools/ab5-method.js.map +1 -0
  26. package/src/solver-tools/ab5-method.ts +397 -0
  27. package/src/solver-tools/index.js +6 -0
  28. package/src/solver-tools/index.js.map +1 -1
  29. package/src/solver-tools/index.ts +6 -0
  30. package/src/solver-tools/lsoda/blas.js +246 -0
  31. package/src/solver-tools/lsoda/blas.js.map +1 -0
  32. package/src/solver-tools/lsoda/blas.ts +267 -0
  33. package/src/solver-tools/lsoda/cfode.js +73 -0
  34. package/src/solver-tools/lsoda/cfode.js.map +1 -0
  35. package/src/solver-tools/lsoda/cfode.ts +83 -0
  36. package/src/solver-tools/lsoda/common.js +94 -0
  37. package/src/solver-tools/lsoda/common.js.map +1 -0
  38. package/src/solver-tools/lsoda/common.ts +135 -0
  39. package/src/solver-tools/lsoda/corfailure.js +25 -0
  40. package/src/solver-tools/lsoda/corfailure.js.map +1 -0
  41. package/src/solver-tools/lsoda/corfailure.ts +30 -0
  42. package/src/solver-tools/lsoda/correction.js +101 -0
  43. package/src/solver-tools/lsoda/correction.js.map +1 -0
  44. package/src/solver-tools/lsoda/correction.ts +119 -0
  45. package/src/solver-tools/lsoda/dense.js +110 -0
  46. package/src/solver-tools/lsoda/dense.js.map +1 -0
  47. package/src/solver-tools/lsoda/dense.ts +129 -0
  48. package/src/solver-tools/lsoda/index.js +4 -0
  49. package/src/solver-tools/lsoda/index.js.map +1 -0
  50. package/src/solver-tools/lsoda/index.ts +5 -0
  51. package/src/solver-tools/lsoda/intdy.js +43 -0
  52. package/src/solver-tools/lsoda/intdy.js.map +1 -0
  53. package/src/solver-tools/lsoda/intdy.ts +50 -0
  54. package/src/solver-tools/lsoda/lsoda.js +567 -0
  55. package/src/solver-tools/lsoda/lsoda.js.map +1 -0
  56. package/src/solver-tools/lsoda/lsoda.ts +632 -0
  57. package/src/solver-tools/lsoda/methodswitch.js +97 -0
  58. package/src/solver-tools/lsoda/methodswitch.js.map +1 -0
  59. package/src/solver-tools/lsoda/methodswitch.ts +107 -0
  60. package/src/solver-tools/lsoda/orderswitch.js +86 -0
  61. package/src/solver-tools/lsoda/orderswitch.js.map +1 -0
  62. package/src/solver-tools/lsoda/orderswitch.ts +95 -0
  63. package/src/solver-tools/lsoda/prja.js +43 -0
  64. package/src/solver-tools/lsoda/prja.js.map +1 -0
  65. package/src/solver-tools/lsoda/prja.ts +49 -0
  66. package/src/solver-tools/lsoda/scaleh.js +30 -0
  67. package/src/solver-tools/lsoda/scaleh.js.map +1 -0
  68. package/src/solver-tools/lsoda/scaleh.ts +33 -0
  69. package/src/solver-tools/lsoda/solsy.js +14 -0
  70. package/src/solver-tools/lsoda/solsy.js.map +1 -0
  71. package/src/solver-tools/lsoda/solsy.ts +18 -0
  72. package/src/solver-tools/lsoda/stoda.js +259 -0
  73. package/src/solver-tools/lsoda/stoda.js.map +1 -0
  74. package/src/solver-tools/lsoda/stoda.ts +286 -0
  75. package/src/solver-tools/lsoda-method.js +120 -0
  76. package/src/solver-tools/lsoda-method.js.map +1 -0
  77. package/src/solver-tools/lsoda-method.ts +136 -0
  78. package/src/solver-tools/mrt-method.js +81 -18
  79. package/src/solver-tools/mrt-method.js.map +1 -1
  80. package/src/solver-tools/mrt-method.ts +111 -21
  81. package/src/solver-tools/rk3-method.js +210 -0
  82. package/src/solver-tools/rk3-method.js.map +1 -0
  83. package/src/solver-tools/rk3-method.ts +257 -0
  84. package/src/solver-tools/rk4-method.js +236 -0
  85. package/src/solver-tools/rk4-method.js.map +1 -0
  86. package/src/solver-tools/rk4-method.ts +286 -0
  87. package/src/solver-tools/rkdp-method.js +247 -0
  88. package/src/solver-tools/rkdp-method.js.map +1 -0
  89. package/src/solver-tools/rkdp-method.ts +300 -0
  90. package/src/solver-tools/ros34prw-method.js +1 -1
  91. package/src/solver-tools/ros34prw-method.js.map +1 -1
  92. package/src/solver-tools/ros34prw-method.ts +1 -1
  93. package/src/solver-tools/ros3prw-method.js +1 -1
  94. package/src/solver-tools/ros3prw-method.js.map +1 -1
  95. package/src/solver-tools/ros3prw-method.ts +1 -1
  96. package/src/solver-tools/solver-defs.js +6 -0
  97. package/src/solver-tools/solver-defs.js.map +1 -1
  98. package/src/solver-tools/solver-defs.ts +6 -0
  99. package/src/tests/correctness.test.js +6 -4
  100. package/src/tests/correctness.test.js.map +1 -1
  101. package/src/tests/correctness.test.ts +6 -4
  102. package/src/tests/lsoda-lib-correctness.test.js +223 -0
  103. package/src/tests/lsoda-lib-correctness.test.js.map +1 -0
  104. package/src/tests/lsoda-lib-correctness.test.ts +283 -0
  105. package/src/tests/lsoda-lib-performance.test.js +95 -0
  106. package/src/tests/lsoda-lib-performance.test.js.map +1 -0
  107. package/src/tests/lsoda-lib-performance.test.ts +126 -0
  108. package/src/tests/performance.test.js +10 -8
  109. package/src/tests/performance.test.js.map +1 -1
  110. package/src/tests/performance.test.ts +10 -8
  111. package/src/tests/test-defs.js +14 -1
  112. package/src/tests/test-defs.js.map +1 -1
  113. package/src/tests/test-defs.ts +15 -1
  114. package/src/worker-tools/solving.js +13 -1
  115. package/src/worker-tools/solving.js.map +1 -1
  116. package/src/worker-tools/solving.ts +20 -1
  117. package/docs/.nojekyll +0 -1
  118. package/docs/assets/hierarchy.js +0 -1
  119. package/docs/assets/highlight.css +0 -99
  120. package/docs/assets/icons.js +0 -18
  121. package/docs/assets/icons.svg +0 -1
  122. package/docs/assets/main.js +0 -60
  123. package/docs/assets/navigation.js +0 -1
  124. package/docs/assets/search.js +0 -1
  125. package/docs/assets/style.css +0 -1633
  126. package/docs/classes/BasicModelPipelineCreator.html +0 -10
  127. package/docs/classes/Callback.html +0 -7
  128. package/docs/classes/CyclicModelPipelineCreator.html +0 -10
  129. package/docs/classes/IterCheckerCallback.html +0 -8
  130. package/docs/classes/PipelineCreator.html +0 -6
  131. package/docs/classes/TimeCheckerCallback.html +0 -8
  132. package/docs/classes/UpdatesModelPipelineCreator.html +0 -10
  133. package/docs/functions/applyPipeline.html +0 -9
  134. package/docs/functions/getCallback.html +0 -3
  135. package/docs/functions/getIVP.html +0 -6
  136. package/docs/functions/getInputVector.html +0 -2
  137. package/docs/functions/getIvp2WebWorker.html +0 -6
  138. package/docs/functions/getJScode.html +0 -6
  139. package/docs/functions/getOutputNames.html +0 -4
  140. package/docs/functions/getPipelineCreator.html +0 -7
  141. package/docs/functions/mrt.html +0 -11
  142. package/docs/functions/printE5.html +0 -76
  143. package/docs/functions/printHires.html +0 -76
  144. package/docs/functions/printOrego.html +0 -66
  145. package/docs/functions/printPollution.html +0 -163
  146. package/docs/functions/printRobertson.html +0 -60
  147. package/docs/functions/printVdpol.html +0 -66
  148. package/docs/functions/ros34prw.html +0 -8
  149. package/docs/functions/ros3prw.html +0 -8
  150. package/docs/functions/solveIvp.html +0 -5
  151. package/docs/hierarchy.html +0 -1
  152. package/docs/index.html +0 -683
  153. package/docs/media/CONTRIBUTING.md +0 -74
  154. package/docs/media/POLL.png +0 -0
  155. package/docs/media/VDPOL.png +0 -0
  156. package/docs/media/acid-prod.gif +0 -0
  157. package/docs/media/basic-use.js +0 -40
  158. package/docs/media/basic-use.js.map +0 -1
  159. package/docs/media/basic-use.ts +0 -44
  160. package/docs/media/check-methods.js +0 -59
  161. package/docs/media/check-methods.js.map +0 -1
  162. package/docs/media/check-methods.ts +0 -81
  163. package/docs/media/corr-probs.js +0 -132
  164. package/docs/media/corr-probs.js.map +0 -1
  165. package/docs/media/corr-probs.ts +0 -155
  166. package/docs/media/cyclic-model.js +0 -100
  167. package/docs/media/cyclic-model.js.map +0 -1
  168. package/docs/media/cyclic-model.ts +0 -110
  169. package/docs/media/e5.js +0 -36
  170. package/docs/media/e5.js.map +0 -1
  171. package/docs/media/e5.ts +0 -36
  172. package/docs/media/hires.js +0 -39
  173. package/docs/media/hires.js.map +0 -1
  174. package/docs/media/hires.ts +0 -40
  175. package/docs/media/model-updates.js +0 -112
  176. package/docs/media/model-updates.js.map +0 -1
  177. package/docs/media/model-updates.ts +0 -122
  178. package/docs/media/orego.js +0 -24
  179. package/docs/media/orego.js.map +0 -1
  180. package/docs/media/orego.ts +0 -25
  181. package/docs/media/pipeline-use.js +0 -87
  182. package/docs/media/pipeline-use.js.map +0 -1
  183. package/docs/media/pipeline-use.ts +0 -98
  184. package/docs/media/pollution.js +0 -132
  185. package/docs/media/pollution.js.map +0 -1
  186. package/docs/media/pollution.ts +0 -133
  187. package/docs/media/print-benchmark.js +0 -581
  188. package/docs/media/print-benchmark.js.map +0 -1
  189. package/docs/media/print-benchmark.ts +0 -594
  190. package/docs/media/robertson.js +0 -19
  191. package/docs/media/robertson.js.map +0 -1
  192. package/docs/media/robertson.ts +0 -19
  193. package/docs/media/scripting.js +0 -30
  194. package/docs/media/scripting.js.map +0 -1
  195. package/docs/media/scripting.ts +0 -35
  196. package/docs/media/vdpol.js +0 -17
  197. package/docs/media/vdpol.js.map +0 -1
  198. package/docs/media/vdpol.ts +0 -17
  199. package/docs/modules.html +0 -1
  200. package/docs/types/Arg.html +0 -12
  201. package/docs/types/DifEqs.html +0 -8
  202. package/docs/types/Func.html +0 -7
  203. package/docs/types/IVP.html +0 -40
  204. package/docs/types/IVP2WebWorker.html +0 -12
  205. package/docs/types/Input.html +0 -8
  206. package/docs/types/Loop.html +0 -7
  207. package/docs/types/ODEs.html +0 -31
  208. package/docs/types/Output.html +0 -6
  209. package/docs/types/Pipeline.html +0 -6
  210. package/docs/types/SolverOptions.html +0 -8
  211. package/docs/types/Update.html +0 -9
  212. package/docs/types/Wrapper.html +0 -8
@@ -0,0 +1,30 @@
1
+ import {LsodaContext, MXNCF} from './common';
2
+
3
+ /**
4
+ * Handle corrector convergence failure.
5
+ * Retracts yh array and increments ncf.
6
+ * Returns 2 if unrecoverable, 1 if step can be retried.
7
+ */
8
+ export function corfailure(ctx: LsodaContext, told: number): number {
9
+ const c = ctx.common!;
10
+ const neq = ctx.neq;
11
+ const hmin = ctx.opt!.hmin;
12
+
13
+ c.ncf++;
14
+ c.rmax = 2.;
15
+ c.tn = told;
16
+
17
+ for (let j = c.nq; j >= 1; j--) {
18
+ for (let i1 = j; i1 <= c.nq; i1++) {
19
+ for (let i = 1; i <= neq; i++)
20
+ c.yh[i1][i] -= c.yh[i1 + 1][i];
21
+ }
22
+ }
23
+
24
+ if (Math.abs(c.h) <= hmin * 1.00001 || c.ncf === MXNCF)
25
+ return 2;
26
+
27
+
28
+ c.ipup = c.miter;
29
+ return 1;
30
+ }
@@ -0,0 +1,101 @@
1
+ import { ETA, MAXCOR } from './common';
2
+ import { vmnorm } from './blas';
3
+ import { prja } from './prja';
4
+ import { solsy } from './solsy';
5
+ import { corfailure } from './corfailure';
6
+ /**
7
+ * Corrector iteration.
8
+ * Returns corflag: 0 = converged, 1 = reduce step & redo prediction, 2 = failure.
9
+ */
10
+ export function correction(ctx, y, pnorm, cs, told) {
11
+ const c = ctx.common;
12
+ const neq = ctx.neq;
13
+ cs.m = 0;
14
+ let rate = 0.;
15
+ cs.del = 0.;
16
+ for (let i = 1; i <= neq; i++)
17
+ y[i] = c.yh[1][i];
18
+ ctx.func(c.tn, y.subarray(1), c.savf.subarray(1), ctx.data);
19
+ c.nfe++;
20
+ while (true) {
21
+ if (cs.m === 0) {
22
+ if (c.ipup > 0) {
23
+ const ierpj = prja(ctx, y);
24
+ c.jcur = 1;
25
+ c.ipup = 0;
26
+ c.rc = 1.;
27
+ c.nslp = c.nst;
28
+ c.crate = 0.7;
29
+ if (!ierpj)
30
+ return corfailure(ctx, told);
31
+ }
32
+ for (let i = 1; i <= neq; i++)
33
+ c.acor[i] = 0.;
34
+ }
35
+ if (c.miter === 0) {
36
+ // Functional iteration
37
+ for (let i = 1; i <= neq; i++) {
38
+ c.savf[i] = c.h * c.savf[i] - c.yh[2][i];
39
+ y[i] = c.savf[i] - c.acor[i];
40
+ }
41
+ cs.del = vmnorm(neq, y, c.ewt);
42
+ for (let i = 1; i <= neq; i++) {
43
+ y[i] = c.yh[1][i] + c.el[1] * c.savf[i];
44
+ c.acor[i] = c.savf[i];
45
+ }
46
+ }
47
+ else {
48
+ // Chord method
49
+ for (let i = 1; i <= neq; i++)
50
+ y[i] = c.h * c.savf[i] - (c.yh[2][i] + c.acor[i]);
51
+ solsy(ctx, y);
52
+ cs.del = vmnorm(neq, y, c.ewt);
53
+ for (let i = 1; i <= neq; i++) {
54
+ c.acor[i] += y[i];
55
+ y[i] = c.yh[1][i] + c.el[1] * c.acor[i];
56
+ }
57
+ }
58
+ // Test for convergence
59
+ if (cs.del <= 100. * pnorm * ETA)
60
+ break;
61
+ if (cs.m !== 0 || c.meth !== 1) {
62
+ if (cs.m !== 0) {
63
+ let rm = 1024.0;
64
+ if (cs.del <= 1024. * cs.delp)
65
+ rm = cs.del / cs.delp;
66
+ rate = Math.max(rate, rm);
67
+ c.crate = Math.max(0.2 * c.crate, rm);
68
+ }
69
+ const conit = 0.5 / (c.nq + 2);
70
+ const dcon = cs.del * Math.min(1., 1.5 * c.crate) / (c.tesco[c.nq][2] * conit);
71
+ if (dcon <= 1.) {
72
+ c.pdest = Math.max(c.pdest, rate / Math.abs(c.h * c.el[1]));
73
+ if (c.pdest !== 0.)
74
+ c.pdlast = c.pdest;
75
+ break;
76
+ }
77
+ }
78
+ // Corrector iteration failed to converge
79
+ cs.m++;
80
+ if (cs.m === MAXCOR || (cs.m >= 2 && cs.del > 2. * cs.delp)) {
81
+ if (c.miter === 0 || c.jcur === 1)
82
+ return corfailure(ctx, told);
83
+ c.ipup = c.miter;
84
+ // Restart corrector if Jacobian is recomputed
85
+ cs.m = 0;
86
+ rate = 0.;
87
+ cs.del = 0.;
88
+ for (let i = 1; i <= neq; i++)
89
+ y[i] = c.yh[1][i];
90
+ ctx.func(c.tn, y.subarray(1), c.savf.subarray(1), ctx.data);
91
+ c.nfe++;
92
+ }
93
+ else {
94
+ cs.delp = cs.del;
95
+ ctx.func(c.tn, y.subarray(1), c.savf.subarray(1), ctx.data);
96
+ c.nfe++;
97
+ }
98
+ }
99
+ return 0;
100
+ }
101
+ //# sourceMappingURL=correction.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"correction.js","sourceRoot":"","sources":["correction.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,GAAG,EAAE,MAAM,EAAC,MAAM,UAAU,CAAC;AACnD,OAAO,EAAC,MAAM,EAAC,MAAM,QAAQ,CAAC;AAC9B,OAAO,EAAC,IAAI,EAAC,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAC,KAAK,EAAC,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AASxC;;;GAGG;AACH,MAAM,UAAU,UAAU,CACxB,GAAiB,EAAE,CAAe,EAAE,KAAa,EACjD,EAAmB,EAAE,IAAY;IAEjC,MAAM,CAAC,GAAG,GAAG,CAAC,MAAO,CAAC;IACtB,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC;IAEpB,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;IACT,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC;IAEZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE;QAC3B,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAEpB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;IAC5D,CAAC,CAAC,GAAG,EAAE,CAAC;IAER,OAAO,IAAI,EAAE,CAAC;QACZ,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YACf,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBACf,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBAC3B,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;gBACX,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC;gBACX,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC;gBACV,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC;gBACf,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC;gBACd,IAAI,CAAC,KAAK;oBACR,OAAO,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YACjC,CAAC;YACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE;gBAC3B,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YAClB,uBAAuB;YACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACzC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC/B,CAAC;YACD,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACxC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,eAAe;YACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE;gBAC3B,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YACpD,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YACd,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,IAAI,EAAE,CAAC,GAAG,IAAI,IAAI,GAAG,KAAK,GAAG,GAAG;YAC9B,MAAM;QAER,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC/B,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;gBACf,IAAI,EAAE,GAAG,MAAM,CAAC;gBAChB,IAAI,EAAE,CAAC,GAAG,IAAI,KAAK,GAAG,EAAE,CAAC,IAAI;oBAC3B,EAAE,GAAG,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC;gBACxB,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC1B,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACxC,CAAC;YACD,MAAM,KAAK,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;YAC/B,MAAM,IAAI,GAAG,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;YAC/E,IAAI,IAAI,IAAI,EAAE,EAAE,CAAC;gBACf,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5D,IAAI,CAAC,CAAC,KAAK,KAAK,EAAE;oBAChB,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC;gBACrB,MAAM;YACR,CAAC;QACH,CAAC;QAED,yCAAyC;QACzC,EAAE,CAAC,CAAC,EAAE,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,KAAK,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,GAAG,GAAG,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5D,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC;gBAC/B,OAAO,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YAE/B,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,KAAK,CAAC;YACjB,8CAA8C;YAC9C,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;YACT,IAAI,GAAG,EAAE,CAAC;YACV,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC;YACZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE;gBAC3B,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACpB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;YAC5D,CAAC,CAAC,GAAG,EAAE,CAAC;QACV,CAAC;aAAM,CAAC;YACN,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,GAAG,CAAC;YACjB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;YAC5D,CAAC,CAAC,GAAG,EAAE,CAAC;QACV,CAAC;IACH,CAAC;IAED,OAAO,CAAC,CAAC;AACX,CAAC"}
@@ -0,0 +1,119 @@
1
+ import {LsodaContext, ETA, MAXCOR} from './common';
2
+ import {vmnorm} from './blas';
3
+ import {prja} from './prja';
4
+ import {solsy} from './solsy';
5
+ import {corfailure} from './corfailure';
6
+
7
+ /** Mutable state passed between correction and stoda */
8
+ export interface CorrectionState {
9
+ del: number;
10
+ delp: number;
11
+ m: number;
12
+ }
13
+
14
+ /**
15
+ * Corrector iteration.
16
+ * Returns corflag: 0 = converged, 1 = reduce step & redo prediction, 2 = failure.
17
+ */
18
+ export function correction(
19
+ ctx: LsodaContext, y: Float64Array, pnorm: number,
20
+ cs: CorrectionState, told: number,
21
+ ): number {
22
+ const c = ctx.common!;
23
+ const neq = ctx.neq;
24
+
25
+ cs.m = 0;
26
+ let rate = 0.;
27
+ cs.del = 0.;
28
+
29
+ for (let i = 1; i <= neq; i++)
30
+ y[i] = c.yh[1][i];
31
+
32
+ ctx.func(c.tn, y.subarray(1), c.savf.subarray(1), ctx.data);
33
+ c.nfe++;
34
+
35
+ while (true) {
36
+ if (cs.m === 0) {
37
+ if (c.ipup > 0) {
38
+ const ierpj = prja(ctx, y);
39
+ c.jcur = 1;
40
+ c.ipup = 0;
41
+ c.rc = 1.;
42
+ c.nslp = c.nst;
43
+ c.crate = 0.7;
44
+ if (!ierpj)
45
+ return corfailure(ctx, told);
46
+ }
47
+ for (let i = 1; i <= neq; i++)
48
+ c.acor[i] = 0.;
49
+ }
50
+
51
+ if (c.miter === 0) {
52
+ // Functional iteration
53
+ for (let i = 1; i <= neq; i++) {
54
+ c.savf[i] = c.h * c.savf[i] - c.yh[2][i];
55
+ y[i] = c.savf[i] - c.acor[i];
56
+ }
57
+ cs.del = vmnorm(neq, y, c.ewt);
58
+ for (let i = 1; i <= neq; i++) {
59
+ y[i] = c.yh[1][i] + c.el[1] * c.savf[i];
60
+ c.acor[i] = c.savf[i];
61
+ }
62
+ } else {
63
+ // Chord method
64
+ for (let i = 1; i <= neq; i++)
65
+ y[i] = c.h * c.savf[i] - (c.yh[2][i] + c.acor[i]);
66
+ solsy(ctx, y);
67
+ cs.del = vmnorm(neq, y, c.ewt);
68
+ for (let i = 1; i <= neq; i++) {
69
+ c.acor[i] += y[i];
70
+ y[i] = c.yh[1][i] + c.el[1] * c.acor[i];
71
+ }
72
+ }
73
+
74
+ // Test for convergence
75
+ if (cs.del <= 100. * pnorm * ETA)
76
+ break;
77
+
78
+ if (cs.m !== 0 || c.meth !== 1) {
79
+ if (cs.m !== 0) {
80
+ let rm = 1024.0;
81
+ if (cs.del <= 1024. * cs.delp)
82
+ rm = cs.del / cs.delp;
83
+ rate = Math.max(rate, rm);
84
+ c.crate = Math.max(0.2 * c.crate, rm);
85
+ }
86
+ const conit = 0.5 / (c.nq + 2);
87
+ const dcon = cs.del * Math.min(1., 1.5 * c.crate) / (c.tesco[c.nq][2] * conit);
88
+ if (dcon <= 1.) {
89
+ c.pdest = Math.max(c.pdest, rate / Math.abs(c.h * c.el[1]));
90
+ if (c.pdest !== 0.)
91
+ c.pdlast = c.pdest;
92
+ break;
93
+ }
94
+ }
95
+
96
+ // Corrector iteration failed to converge
97
+ cs.m++;
98
+ if (cs.m === MAXCOR || (cs.m >= 2 && cs.del > 2. * cs.delp)) {
99
+ if (c.miter === 0 || c.jcur === 1)
100
+ return corfailure(ctx, told);
101
+
102
+ c.ipup = c.miter;
103
+ // Restart corrector if Jacobian is recomputed
104
+ cs.m = 0;
105
+ rate = 0.;
106
+ cs.del = 0.;
107
+ for (let i = 1; i <= neq; i++)
108
+ y[i] = c.yh[1][i];
109
+ ctx.func(c.tn, y.subarray(1), c.savf.subarray(1), ctx.data);
110
+ c.nfe++;
111
+ } else {
112
+ cs.delp = cs.del;
113
+ ctx.func(c.tn, y.subarray(1), c.savf.subarray(1), ctx.data);
114
+ c.nfe++;
115
+ }
116
+ }
117
+
118
+ return 0;
119
+ }
@@ -0,0 +1,110 @@
1
+ /**
2
+ * Dense output interpolator. Given a sequence of Nordsieck snapshots
3
+ * captured during integration, evaluates the solution at arbitrary points
4
+ * via Taylor/Horner interpolation.
5
+ */
6
+ export class DenseOutput {
7
+ constructor(snapshots, neq) {
8
+ if (snapshots.length === 0)
9
+ throw new Error('[DenseOutput] no snapshots collected');
10
+ this.snaps = snapshots;
11
+ this.neq = neq;
12
+ }
13
+ /** Leftmost reachable time. */
14
+ get tMin() {
15
+ const s = this.snaps[0];
16
+ return s.tn - s.hu;
17
+ }
18
+ /** Rightmost reachable time. */
19
+ get tMax() {
20
+ return this.snaps[this.snaps.length - 1].tn;
21
+ }
22
+ /**
23
+ * Evaluate the solution at a single time point.
24
+ * Uses Horner's method on the Nordsieck array of the enclosing step.
25
+ */
26
+ evaluateAt(t) {
27
+ const snap = this.findSnap(t);
28
+ return this.horner(snap, t, new Float64Array(this.neq));
29
+ }
30
+ /**
31
+ * Evaluate the solution at each point in a sorted ascending array.
32
+ * Uses a linear scan over snapshots for efficiency.
33
+ */
34
+ solveAtTimes(tArray) {
35
+ if (tArray.length === 0)
36
+ return [];
37
+ const snaps = this.snaps;
38
+ // Validate range
39
+ const tFirst = tArray[0];
40
+ const tLast = tArray[tArray.length - 1];
41
+ if (tFirst < this.tMin - 1e-14 * Math.abs(this.tMin))
42
+ throw new Error(`[DenseOutput] t=${tFirst} is before tMin=${this.tMin}`);
43
+ if (tLast > this.tMax + 1e-14 * Math.abs(this.tMax))
44
+ throw new Error(`[DenseOutput] t=${tLast} is after tMax=${this.tMax}`);
45
+ const dky = new Float64Array(this.neq);
46
+ const result = [];
47
+ for (let i = 0; i < this.neq; ++i)
48
+ result.push(new Float64Array(tArray.length));
49
+ let si = 0; // snapshot pointer
50
+ for (let qi = 0; qi < tArray.length; qi++) {
51
+ const t = tArray[qi];
52
+ // Advance snapshot pointer until we find an interval containing t
53
+ while (si < snaps.length - 1 && t > snaps[si].tn + 1e-14 * Math.abs(snaps[si].tn))
54
+ si++;
55
+ this.horner(snaps[si], t, dky);
56
+ for (let i = 0; i < this.neq; ++i)
57
+ result[i][qi] = dky[i];
58
+ }
59
+ return result;
60
+ }
61
+ /**
62
+ * Evaluate the solution on a uniform grid from tStart to tEnd with given step.
63
+ */
64
+ solveOnGrid(tStart, tEnd, step) {
65
+ if (step <= 0)
66
+ throw new Error('[DenseOutput] step must be positive');
67
+ const n = Math.floor((tEnd - tStart) / step) + 1;
68
+ const tArr = new Float64Array(n);
69
+ for (let i = 0; i < n - 1; i++)
70
+ tArr[i] = tStart + i * step;
71
+ tArr[n - 1] = tEnd;
72
+ const t = tArr;
73
+ const y = this.solveAtTimes(t);
74
+ return { t, y };
75
+ }
76
+ /** Find the snapshot whose interval contains t via binary search. */
77
+ findSnap(t) {
78
+ const snaps = this.snaps;
79
+ // Quick boundary checks
80
+ if (t <= snaps[0].tn)
81
+ return snaps[0];
82
+ if (t >= snaps[snaps.length - 1].tn)
83
+ return snaps[snaps.length - 1];
84
+ // Binary search: find first snap with tn >= t
85
+ let lo = 0;
86
+ let hi = snaps.length - 1;
87
+ while (lo < hi) {
88
+ const mid = (lo + hi) >>> 1;
89
+ if (snaps[mid].tn < t)
90
+ lo = mid + 1;
91
+ else
92
+ hi = mid;
93
+ }
94
+ return snaps[lo];
95
+ }
96
+ /** Horner evaluation of the Nordsieck polynomial at time t. */
97
+ horner(snap, t, dky) {
98
+ const { nq, h, yh } = snap;
99
+ const neq = this.neq;
100
+ const s = (t - snap.tn) / h;
101
+ for (let i = 0; i < neq; i++) {
102
+ let val = yh[nq][i];
103
+ for (let j = nq - 1; j >= 0; j--)
104
+ val = yh[j][i] + s * val;
105
+ dky[i] = val;
106
+ }
107
+ return dky;
108
+ }
109
+ }
110
+ //# sourceMappingURL=dense.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"dense.js","sourceRoot":"","sources":["dense.ts"],"names":[],"mappings":"AAEA;;;;GAIG;AACH,MAAM,OAAO,WAAW;IAItB,YAAY,SAA8B,EAAE,GAAW;QACrD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;QACvB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAED,+BAA+B;IAC/B,IAAI,IAAI;QACN,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxB,OAAO,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;IACrB,CAAC;IAED,gCAAgC;IAChC,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;IAC9C,CAAC;IAED;;;OAGG;IACH,UAAU,CAAC,CAAS;QAClB,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC9B,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACH,YAAY,CAAC,MAAoB;QAC/B,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QAEzB,iBAAiB;QACjB,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACxC,IAAI,MAAM,GAAG,IAAI,CAAC,IAAI,GAAG,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;YAClD,MAAM,IAAI,KAAK,CAAC,mBAAmB,MAAM,mBAAmB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3E,IAAI,KAAK,GAAG,IAAI,CAAC,IAAI,GAAG,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,mBAAmB,KAAK,kBAAkB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAEzE,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,MAAM,GAAmB,EAAE,CAAC;QAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAE/C,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,mBAAmB;QAE/B,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC;YAC1C,MAAM,CAAC,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC;YACrB,kEAAkE;YAClE,OAAO,EAAE,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;gBAC/E,EAAE,EAAE,CAAC;YAEP,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;YAE/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;gBAC/B,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,MAAc,EAAE,IAAY,EAAE,IAAY;QACpD,IAAI,IAAI,IAAI,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QAEzD,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;QACjD,MAAM,IAAI,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;YAC5B,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC;QAC9B,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC;QAEnB,MAAM,CAAC,GAAG,IAAI,CAAC;QACf,MAAM,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,OAAO,EAAC,CAAC,EAAE,CAAC,EAAC,CAAC;IAChB,CAAC;IAED,qEAAqE;IAC7D,QAAQ,CAAC,CAAS;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QAEzB,wBAAwB;QACxB,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE;YAAE,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE;YAAE,OAAO,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAEpE,8CAA8C;QAC9C,IAAI,EAAE,GAAG,CAAC,CAAC;QAAC,IAAI,EAAE,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;QACtC,OAAO,EAAE,GAAG,EAAE,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;YAC5B,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC;gBACnB,EAAE,GAAG,GAAG,GAAG,CAAC,CAAC;;gBAEb,EAAE,GAAG,GAAG,CAAC;QACb,CAAC;QACD,OAAO,KAAK,CAAC,EAAE,CAAC,CAAC;IACnB,CAAC;IAED,+DAA+D;IACvD,MAAM,CAAC,IAAuB,EAAE,CAAS,EAAE,GAAiB;QAClE,MAAM,EAAC,EAAE,EAAE,CAAC,EAAE,EAAE,EAAC,GAAG,IAAI,CAAC;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;QACrB,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;QAE5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7B,IAAI,GAAG,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACpB,KAAK,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE;gBAC9B,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;YAC3B,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;QACf,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;CACF"}
@@ -0,0 +1,129 @@
1
+ import {NordsieckSnapshot} from './common';
2
+
3
+ /**
4
+ * Dense output interpolator. Given a sequence of Nordsieck snapshots
5
+ * captured during integration, evaluates the solution at arbitrary points
6
+ * via Taylor/Horner interpolation.
7
+ */
8
+ export class DenseOutput {
9
+ private snaps: NordsieckSnapshot[];
10
+ private neq: number;
11
+
12
+ constructor(snapshots: NordsieckSnapshot[], neq: number) {
13
+ if (snapshots.length === 0)
14
+ throw new Error('[DenseOutput] no snapshots collected');
15
+ this.snaps = snapshots;
16
+ this.neq = neq;
17
+ }
18
+
19
+ /** Leftmost reachable time. */
20
+ get tMin(): number {
21
+ const s = this.snaps[0];
22
+ return s.tn - s.hu;
23
+ }
24
+
25
+ /** Rightmost reachable time. */
26
+ get tMax(): number {
27
+ return this.snaps[this.snaps.length - 1].tn;
28
+ }
29
+
30
+ /**
31
+ * Evaluate the solution at a single time point.
32
+ * Uses Horner's method on the Nordsieck array of the enclosing step.
33
+ */
34
+ evaluateAt(t: number): Float64Array {
35
+ const snap = this.findSnap(t);
36
+ return this.horner(snap, t, new Float64Array(this.neq));
37
+ }
38
+
39
+ /**
40
+ * Evaluate the solution at each point in a sorted ascending array.
41
+ * Uses a linear scan over snapshots for efficiency.
42
+ */
43
+ solveAtTimes(tArray: Float64Array): Float64Array[] {
44
+ if (tArray.length === 0) return [];
45
+ const snaps = this.snaps;
46
+
47
+ // Validate range
48
+ const tFirst = tArray[0];
49
+ const tLast = tArray[tArray.length - 1];
50
+ if (tFirst < this.tMin - 1e-14 * Math.abs(this.tMin))
51
+ throw new Error(`[DenseOutput] t=${tFirst} is before tMin=${this.tMin}`);
52
+ if (tLast > this.tMax + 1e-14 * Math.abs(this.tMax))
53
+ throw new Error(`[DenseOutput] t=${tLast} is after tMax=${this.tMax}`);
54
+
55
+ const dky = new Float64Array(this.neq);
56
+ const result: Float64Array[] = [];
57
+
58
+ for (let i = 0; i < this.neq; ++i)
59
+ result.push(new Float64Array(tArray.length));
60
+
61
+ let si = 0; // snapshot pointer
62
+
63
+ for (let qi = 0; qi < tArray.length; qi++) {
64
+ const t = tArray[qi];
65
+ // Advance snapshot pointer until we find an interval containing t
66
+ while (si < snaps.length - 1 && t > snaps[si].tn + 1e-14 * Math.abs(snaps[si].tn))
67
+ si++;
68
+
69
+ this.horner(snaps[si], t, dky);
70
+
71
+ for (let i = 0; i < this.neq; ++i)
72
+ result[i][qi] = dky[i];
73
+ }
74
+ return result;
75
+ }
76
+
77
+ /**
78
+ * Evaluate the solution on a uniform grid from tStart to tEnd with given step.
79
+ */
80
+ solveOnGrid(tStart: number, tEnd: number, step: number): { t: Float64Array, y: Float64Array[] } {
81
+ if (step <= 0)
82
+ throw new Error('[DenseOutput] step must be positive');
83
+
84
+ const n = Math.floor((tEnd - tStart) / step) + 1;
85
+ const tArr = new Float64Array(n);
86
+ for (let i = 0; i < n - 1; i++)
87
+ tArr[i] = tStart + i * step;
88
+ tArr[n - 1] = tEnd;
89
+
90
+ const t = tArr;
91
+ const y = this.solveAtTimes(t);
92
+ return {t, y};
93
+ }
94
+
95
+ /** Find the snapshot whose interval contains t via binary search. */
96
+ private findSnap(t: number): NordsieckSnapshot {
97
+ const snaps = this.snaps;
98
+
99
+ // Quick boundary checks
100
+ if (t <= snaps[0].tn) return snaps[0];
101
+ if (t >= snaps[snaps.length - 1].tn) return snaps[snaps.length - 1];
102
+
103
+ // Binary search: find first snap with tn >= t
104
+ let lo = 0; let hi = snaps.length - 1;
105
+ while (lo < hi) {
106
+ const mid = (lo + hi) >>> 1;
107
+ if (snaps[mid].tn < t)
108
+ lo = mid + 1;
109
+ else
110
+ hi = mid;
111
+ }
112
+ return snaps[lo];
113
+ }
114
+
115
+ /** Horner evaluation of the Nordsieck polynomial at time t. */
116
+ private horner(snap: NordsieckSnapshot, t: number, dky: Float64Array): Float64Array {
117
+ const {nq, h, yh} = snap;
118
+ const neq = this.neq;
119
+ const s = (t - snap.tn) / h;
120
+
121
+ for (let i = 0; i < neq; i++) {
122
+ let val = yh[nq][i];
123
+ for (let j = nq - 1; j >= 0; j--)
124
+ val = yh[j][i] + s * val;
125
+ dky[i] = val;
126
+ }
127
+ return dky;
128
+ }
129
+ }
@@ -0,0 +1,4 @@
1
+ export { Lsoda, lsoda, lsodaPrepare, lsodaReset, lsodaFree } from './lsoda';
2
+ export { LsodaContext } from './common';
3
+ export { DenseOutput } from './dense';
4
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,EAAC,MAAM,SAAS,CAAC;AAG1E,OAAO,EAAC,YAAY,EAAC,MAAM,UAAU,CAAC;AACtC,OAAO,EAAC,WAAW,EAAC,MAAM,SAAS,CAAC"}
@@ -0,0 +1,5 @@
1
+ export {Lsoda, lsoda, lsodaPrepare, lsodaReset, lsodaFree} from './lsoda';
2
+ export type {OdeFunction, LsodaSolveResult} from './lsoda';
3
+ export type {LsodaOpt, LsodaFunc, NordsieckSnapshot} from './common';
4
+ export {LsodaContext} from './common';
5
+ export {DenseOutput} from './dense';
@@ -0,0 +1,43 @@
1
+ import { ETA } from './common';
2
+ /**
3
+ * Interpolate solution at time t using Nordsieck history array.
4
+ * k: derivative order (0 for solution)
5
+ * dky: output array (1-indexed)
6
+ * Returns 0 on success, negative on error.
7
+ */
8
+ export function intdy(ctx, t, k, dky) {
9
+ const c = ctx.common;
10
+ const neq = ctx.neq;
11
+ if (k < 0 || k > c.nq) {
12
+ console.error(`[intdy] k = ${k} illegal`);
13
+ return -1;
14
+ }
15
+ const tp = c.tn - c.hu - 100. * ETA * (c.tn + c.hu);
16
+ if ((t - tp) * (t - c.tn) > 0.) {
17
+ console.error(`intdy -- t = ${t} illegal. t not in interval tcur - hu to tcur`);
18
+ return -2;
19
+ }
20
+ const s = (t - c.tn) / c.h;
21
+ let ic = 1;
22
+ for (let jj = (c.nq + 1) - k; jj <= c.nq; jj++)
23
+ ic *= jj;
24
+ let cc = ic;
25
+ for (let i = 1; i <= neq; i++)
26
+ dky[i] = cc * c.yh[c.nq + 1][i];
27
+ for (let j = c.nq - 1; j >= k; j--) {
28
+ const jp1 = j + 1;
29
+ ic = 1;
30
+ for (let jj = jp1 - k; jj <= j; jj++)
31
+ ic *= jj;
32
+ cc = ic;
33
+ for (let i = 1; i <= neq; i++)
34
+ dky[i] = cc * c.yh[jp1][i] + s * dky[i];
35
+ }
36
+ if (k === 0)
37
+ return 0;
38
+ const r = Math.pow(c.h, -k);
39
+ for (let i = 1; i <= neq; i++)
40
+ dky[i] *= r;
41
+ return 0;
42
+ }
43
+ //# sourceMappingURL=intdy.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"intdy.js","sourceRoot":"","sources":["intdy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,GAAG,EAAC,MAAM,UAAU,CAAC;AAE3C;;;;;GAKG;AACH,MAAM,UAAU,KAAK,CAAC,GAAiB,EAAE,CAAS,EAAE,CAAS,EAAE,GAAiB;IAC9E,MAAM,CAAC,GAAG,GAAG,CAAC,MAAO,CAAC;IACtB,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC;IAEpB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC;QACtB,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QAC1C,OAAO,CAAC,CAAC,CAAC;IACZ,CAAC;IAED,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,GAAG,IAAI,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC;IACpD,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC;QAC/B,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC,+CAA+C,CAAC,CAAC;QAChF,OAAO,CAAC,CAAC,CAAC;IACZ,CAAC;IAED,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IAC3B,IAAI,EAAE,GAAG,CAAC,CAAC;IACX,KAAK,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE;QAC5C,EAAE,IAAI,EAAE,CAAC;IACX,IAAI,EAAE,GAAG,EAAE,CAAC;IAEZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE;QAC3B,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAElC,KAAK,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;QAClB,EAAE,GAAG,CAAC,CAAC;QACP,KAAK,IAAI,EAAE,GAAG,GAAG,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE;YAClC,EAAE,IAAI,EAAE,CAAC;QACX,EAAE,GAAG,EAAE,CAAC;QACR,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE;YAC3B,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAEtB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE;QAC3B,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO,CAAC,CAAC;AACX,CAAC"}
@@ -0,0 +1,50 @@
1
+ import {LsodaContext, ETA} from './common';
2
+
3
+ /**
4
+ * Interpolate solution at time t using Nordsieck history array.
5
+ * k: derivative order (0 for solution)
6
+ * dky: output array (1-indexed)
7
+ * Returns 0 on success, negative on error.
8
+ */
9
+ export function intdy(ctx: LsodaContext, t: number, k: number, dky: Float64Array): number {
10
+ const c = ctx.common!;
11
+ const neq = ctx.neq;
12
+
13
+ if (k < 0 || k > c.nq) {
14
+ console.error(`[intdy] k = ${k} illegal`);
15
+ return -1;
16
+ }
17
+
18
+ const tp = c.tn - c.hu - 100. * ETA * (c.tn + c.hu);
19
+ if ((t - tp) * (t - c.tn) > 0.) {
20
+ console.error(`intdy -- t = ${t} illegal. t not in interval tcur - hu to tcur`);
21
+ return -2;
22
+ }
23
+
24
+ const s = (t - c.tn) / c.h;
25
+ let ic = 1;
26
+ for (let jj = (c.nq + 1) - k; jj <= c.nq; jj++)
27
+ ic *= jj;
28
+ let cc = ic;
29
+
30
+ for (let i = 1; i <= neq; i++)
31
+ dky[i] = cc * c.yh[c.nq + 1][i];
32
+
33
+ for (let j = c.nq - 1; j >= k; j--) {
34
+ const jp1 = j + 1;
35
+ ic = 1;
36
+ for (let jj = jp1 - k; jj <= j; jj++)
37
+ ic *= jj;
38
+ cc = ic;
39
+ for (let i = 1; i <= neq; i++)
40
+ dky[i] = cc * c.yh[jp1][i] + s * dky[i];
41
+ }
42
+
43
+ if (k === 0) return 0;
44
+
45
+ const r = Math.pow(c.h, -k);
46
+ for (let i = 1; i <= neq; i++)
47
+ dky[i] *= r;
48
+
49
+ return 0;
50
+ }