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.
- package/.claude/settings.local.json +11 -0
- package/CLAUDE.md +98 -0
- package/README.MD +100 -24
- package/images/E5.png +0 -0
- package/images/HIRES.png +0 -0
- package/images/OREGO.png +0 -0
- package/images/POLL-full.png +0 -0
- package/images/ROBER.png +0 -0
- package/images/VDP0L.png +0 -0
- package/index.js +1 -1
- package/index.js.map +1 -1
- package/index.ts +2 -1
- package/jest.config.js +1 -0
- package/package.json +1 -1
- package/src/examples/bioreactor.js +125 -0
- package/src/examples/bioreactor.js.map +1 -0
- package/src/examples/bioreactor.ts +132 -0
- package/src/examples/cyclic-model.js +3 -1
- package/src/examples/cyclic-model.js.map +1 -1
- package/src/examples/cyclic-model.ts +3 -1
- package/src/solver-tools/ab4-method.js +332 -0
- package/src/solver-tools/ab4-method.js.map +1 -0
- package/src/solver-tools/ab4-method.ts +395 -0
- package/src/solver-tools/ab5-method.js +334 -0
- package/src/solver-tools/ab5-method.js.map +1 -0
- package/src/solver-tools/ab5-method.ts +397 -0
- package/src/solver-tools/index.js +6 -0
- package/src/solver-tools/index.js.map +1 -1
- package/src/solver-tools/index.ts +6 -0
- package/src/solver-tools/lsoda/blas.js +246 -0
- package/src/solver-tools/lsoda/blas.js.map +1 -0
- package/src/solver-tools/lsoda/blas.ts +267 -0
- package/src/solver-tools/lsoda/cfode.js +73 -0
- package/src/solver-tools/lsoda/cfode.js.map +1 -0
- package/src/solver-tools/lsoda/cfode.ts +83 -0
- package/src/solver-tools/lsoda/common.js +94 -0
- package/src/solver-tools/lsoda/common.js.map +1 -0
- package/src/solver-tools/lsoda/common.ts +135 -0
- package/src/solver-tools/lsoda/corfailure.js +25 -0
- package/src/solver-tools/lsoda/corfailure.js.map +1 -0
- package/src/solver-tools/lsoda/corfailure.ts +30 -0
- package/src/solver-tools/lsoda/correction.js +101 -0
- package/src/solver-tools/lsoda/correction.js.map +1 -0
- package/src/solver-tools/lsoda/correction.ts +119 -0
- package/src/solver-tools/lsoda/dense.js +110 -0
- package/src/solver-tools/lsoda/dense.js.map +1 -0
- package/src/solver-tools/lsoda/dense.ts +129 -0
- package/src/solver-tools/lsoda/index.js +4 -0
- package/src/solver-tools/lsoda/index.js.map +1 -0
- package/src/solver-tools/lsoda/index.ts +5 -0
- package/src/solver-tools/lsoda/intdy.js +43 -0
- package/src/solver-tools/lsoda/intdy.js.map +1 -0
- package/src/solver-tools/lsoda/intdy.ts +50 -0
- package/src/solver-tools/lsoda/lsoda.js +567 -0
- package/src/solver-tools/lsoda/lsoda.js.map +1 -0
- package/src/solver-tools/lsoda/lsoda.ts +632 -0
- package/src/solver-tools/lsoda/methodswitch.js +97 -0
- package/src/solver-tools/lsoda/methodswitch.js.map +1 -0
- package/src/solver-tools/lsoda/methodswitch.ts +107 -0
- package/src/solver-tools/lsoda/orderswitch.js +86 -0
- package/src/solver-tools/lsoda/orderswitch.js.map +1 -0
- package/src/solver-tools/lsoda/orderswitch.ts +95 -0
- package/src/solver-tools/lsoda/prja.js +43 -0
- package/src/solver-tools/lsoda/prja.js.map +1 -0
- package/src/solver-tools/lsoda/prja.ts +49 -0
- package/src/solver-tools/lsoda/scaleh.js +30 -0
- package/src/solver-tools/lsoda/scaleh.js.map +1 -0
- package/src/solver-tools/lsoda/scaleh.ts +33 -0
- package/src/solver-tools/lsoda/solsy.js +14 -0
- package/src/solver-tools/lsoda/solsy.js.map +1 -0
- package/src/solver-tools/lsoda/solsy.ts +18 -0
- package/src/solver-tools/lsoda/stoda.js +259 -0
- package/src/solver-tools/lsoda/stoda.js.map +1 -0
- package/src/solver-tools/lsoda/stoda.ts +286 -0
- package/src/solver-tools/lsoda-method.js +120 -0
- package/src/solver-tools/lsoda-method.js.map +1 -0
- package/src/solver-tools/lsoda-method.ts +136 -0
- package/src/solver-tools/mrt-method.js +81 -18
- package/src/solver-tools/mrt-method.js.map +1 -1
- package/src/solver-tools/mrt-method.ts +111 -21
- package/src/solver-tools/rk3-method.js +210 -0
- package/src/solver-tools/rk3-method.js.map +1 -0
- package/src/solver-tools/rk3-method.ts +257 -0
- package/src/solver-tools/rk4-method.js +236 -0
- package/src/solver-tools/rk4-method.js.map +1 -0
- package/src/solver-tools/rk4-method.ts +286 -0
- package/src/solver-tools/rkdp-method.js +247 -0
- package/src/solver-tools/rkdp-method.js.map +1 -0
- package/src/solver-tools/rkdp-method.ts +300 -0
- package/src/solver-tools/ros34prw-method.js +1 -1
- package/src/solver-tools/ros34prw-method.js.map +1 -1
- package/src/solver-tools/ros34prw-method.ts +1 -1
- package/src/solver-tools/ros3prw-method.js +1 -1
- package/src/solver-tools/ros3prw-method.js.map +1 -1
- package/src/solver-tools/ros3prw-method.ts +1 -1
- package/src/solver-tools/solver-defs.js +6 -0
- package/src/solver-tools/solver-defs.js.map +1 -1
- package/src/solver-tools/solver-defs.ts +6 -0
- package/src/tests/correctness.test.js +6 -4
- package/src/tests/correctness.test.js.map +1 -1
- package/src/tests/correctness.test.ts +6 -4
- package/src/tests/lsoda-lib-correctness.test.js +223 -0
- package/src/tests/lsoda-lib-correctness.test.js.map +1 -0
- package/src/tests/lsoda-lib-correctness.test.ts +283 -0
- package/src/tests/lsoda-lib-performance.test.js +95 -0
- package/src/tests/lsoda-lib-performance.test.js.map +1 -0
- package/src/tests/lsoda-lib-performance.test.ts +126 -0
- package/src/tests/performance.test.js +10 -8
- package/src/tests/performance.test.js.map +1 -1
- package/src/tests/performance.test.ts +10 -8
- package/src/tests/test-defs.js +14 -1
- package/src/tests/test-defs.js.map +1 -1
- package/src/tests/test-defs.ts +15 -1
- package/src/worker-tools/solving.js +13 -1
- package/src/worker-tools/solving.js.map +1 -1
- package/src/worker-tools/solving.ts +20 -1
- package/docs/.nojekyll +0 -1
- package/docs/assets/hierarchy.js +0 -1
- package/docs/assets/highlight.css +0 -99
- package/docs/assets/icons.js +0 -18
- package/docs/assets/icons.svg +0 -1
- package/docs/assets/main.js +0 -60
- package/docs/assets/navigation.js +0 -1
- package/docs/assets/search.js +0 -1
- package/docs/assets/style.css +0 -1633
- package/docs/classes/BasicModelPipelineCreator.html +0 -10
- package/docs/classes/Callback.html +0 -7
- package/docs/classes/CyclicModelPipelineCreator.html +0 -10
- package/docs/classes/IterCheckerCallback.html +0 -8
- package/docs/classes/PipelineCreator.html +0 -6
- package/docs/classes/TimeCheckerCallback.html +0 -8
- package/docs/classes/UpdatesModelPipelineCreator.html +0 -10
- package/docs/functions/applyPipeline.html +0 -9
- package/docs/functions/getCallback.html +0 -3
- package/docs/functions/getIVP.html +0 -6
- package/docs/functions/getInputVector.html +0 -2
- package/docs/functions/getIvp2WebWorker.html +0 -6
- package/docs/functions/getJScode.html +0 -6
- package/docs/functions/getOutputNames.html +0 -4
- package/docs/functions/getPipelineCreator.html +0 -7
- package/docs/functions/mrt.html +0 -11
- package/docs/functions/printE5.html +0 -76
- package/docs/functions/printHires.html +0 -76
- package/docs/functions/printOrego.html +0 -66
- package/docs/functions/printPollution.html +0 -163
- package/docs/functions/printRobertson.html +0 -60
- package/docs/functions/printVdpol.html +0 -66
- package/docs/functions/ros34prw.html +0 -8
- package/docs/functions/ros3prw.html +0 -8
- package/docs/functions/solveIvp.html +0 -5
- package/docs/hierarchy.html +0 -1
- package/docs/index.html +0 -683
- package/docs/media/CONTRIBUTING.md +0 -74
- package/docs/media/POLL.png +0 -0
- package/docs/media/VDPOL.png +0 -0
- package/docs/media/acid-prod.gif +0 -0
- package/docs/media/basic-use.js +0 -40
- package/docs/media/basic-use.js.map +0 -1
- package/docs/media/basic-use.ts +0 -44
- package/docs/media/check-methods.js +0 -59
- package/docs/media/check-methods.js.map +0 -1
- package/docs/media/check-methods.ts +0 -81
- package/docs/media/corr-probs.js +0 -132
- package/docs/media/corr-probs.js.map +0 -1
- package/docs/media/corr-probs.ts +0 -155
- package/docs/media/cyclic-model.js +0 -100
- package/docs/media/cyclic-model.js.map +0 -1
- package/docs/media/cyclic-model.ts +0 -110
- package/docs/media/e5.js +0 -36
- package/docs/media/e5.js.map +0 -1
- package/docs/media/e5.ts +0 -36
- package/docs/media/hires.js +0 -39
- package/docs/media/hires.js.map +0 -1
- package/docs/media/hires.ts +0 -40
- package/docs/media/model-updates.js +0 -112
- package/docs/media/model-updates.js.map +0 -1
- package/docs/media/model-updates.ts +0 -122
- package/docs/media/orego.js +0 -24
- package/docs/media/orego.js.map +0 -1
- package/docs/media/orego.ts +0 -25
- package/docs/media/pipeline-use.js +0 -87
- package/docs/media/pipeline-use.js.map +0 -1
- package/docs/media/pipeline-use.ts +0 -98
- package/docs/media/pollution.js +0 -132
- package/docs/media/pollution.js.map +0 -1
- package/docs/media/pollution.ts +0 -133
- package/docs/media/print-benchmark.js +0 -581
- package/docs/media/print-benchmark.js.map +0 -1
- package/docs/media/print-benchmark.ts +0 -594
- package/docs/media/robertson.js +0 -19
- package/docs/media/robertson.js.map +0 -1
- package/docs/media/robertson.ts +0 -19
- package/docs/media/scripting.js +0 -30
- package/docs/media/scripting.js.map +0 -1
- package/docs/media/scripting.ts +0 -35
- package/docs/media/vdpol.js +0 -17
- package/docs/media/vdpol.js.map +0 -1
- package/docs/media/vdpol.ts +0 -17
- package/docs/modules.html +0 -1
- package/docs/types/Arg.html +0 -12
- package/docs/types/DifEqs.html +0 -8
- package/docs/types/Func.html +0 -7
- package/docs/types/IVP.html +0 -40
- package/docs/types/IVP2WebWorker.html +0 -12
- package/docs/types/Input.html +0 -8
- package/docs/types/Loop.html +0 -7
- package/docs/types/ODEs.html +0 -31
- package/docs/types/Output.html +0 -6
- package/docs/types/Pipeline.html +0 -6
- package/docs/types/SolverOptions.html +0 -8
- package/docs/types/Update.html +0 -9
- 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 @@
|
|
|
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
|
+
}
|