feat: support for model-requested horizons
This commit is contained in:
parent
685de96eec
commit
ac4e066bf8
24 changed files with 170 additions and 93 deletions
|
|
@ -18,8 +18,8 @@ let fder y yd =
|
|||
yd.{0} <- y.{1};
|
||||
yd.{1} <- (mu *. (1.0 -. (y.{0} *. y.{0})) *. y.{1}) -. y.{0};
|
||||
yd
|
||||
let fout _ _ y = of_array [| y.{0}; y.{1} |]
|
||||
let step { i; lx } _ =
|
||||
let fout _ _ _ y = of_array [| y.{0}; y.{1} |]
|
||||
let step { i; lx } _ _ =
|
||||
let lx = if i then of_array [| x0; y0 |] else lx in
|
||||
of_array [| lx.{0}; lx.{1} |], { lx; i=false }
|
||||
let cget s = s.lx
|
||||
|
|
@ -31,8 +31,8 @@ let horizon _ = max_float
|
|||
let van_der_pol () : (_, _, carray, carray, carray, zarray, carray) hnode =
|
||||
let yd = cmake csize in
|
||||
let zout = cmake zsize in
|
||||
let fder _ _ y = fder y yd in
|
||||
let fzer _ _ _ = zout in
|
||||
let fder _ _ _ y = fder y yd in
|
||||
let fzer _ _ _ _ = zout in
|
||||
let state = { lx=of_array [| x0; y0 |]; i=true } in
|
||||
let reset _ _ = state in
|
||||
HNode { state; fder; fzer; fout; step; reset; horizon;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue