@@ -14,16 +14,17 @@ function generate_ODENLStepData(sys::System, u0, p, mm = calculate_massmatrix(sy
14
14
end
15
15
nlprob = NonlinearProblem (nlsys, op; build_initializeprob = false )
16
16
17
- subsetidxs = [findfirst (isequal (y),unknowns (sys)) for y in unknowns (nlsys)]
17
+ subsetidxs = [findfirst (isequal (y), unknowns (sys)) for y in unknowns (nlsys)]
18
18
set_gamma_c = setsym (nlsys, (ODE_GAMMA... , ODE_C))
19
19
set_outer_tmp = setsym (nlsys, outer_tmp)
20
20
set_inner_tmp = setsym (nlsys, inner_tmp)
21
21
nlprobmap = generate_nlprobmap (sys, nlsys)
22
22
23
- return SciMLBase. ODENLStepData (nlprob, subsetidxs, set_gamma_c, set_outer_tmp, set_inner_tmp, nlprobmap)
23
+ return SciMLBase. ODENLStepData (
24
+ nlprob, subsetidxs, set_gamma_c, set_outer_tmp, set_inner_tmp, nlprobmap)
24
25
end
25
26
26
- const ODE_GAMMA = @parameters γ₁ₘₜₖ, γ₂ₘₜₖ, γ₃ₘₜₖ
27
+ const ODE_GAMMA = @parameters γ₁ₘₜₖ, γ₂ₘₜₖ, γ₃ₘₜₖ
27
28
const ODE_C = only (@parameters cₘₜₖ)
28
29
29
30
function get_outer_tmp (n:: Int )
@@ -55,7 +56,8 @@ function inner_nlsystem(sys::System, mm)
55
56
56
57
new_dvs = unknowns (sys)
57
58
new_ps = [parameters (sys); [gamma1, gamma2, gamma3, c, inner_tmp, outer_tmp]]
58
- nlsys = mtkcompile (System (new_eqs, new_dvs, new_ps; name = :nlsys ); split = is_split (sys))
59
+ nlsys = mtkcompile (
60
+ System (new_eqs, new_dvs, new_ps; name = :nlsys ); split = is_split (sys))
59
61
return nlsys, outer_tmp, inner_tmp
60
62
end
61
63
0 commit comments