Skip to content

Commit 15065bb

Browse files
committed
format
1 parent c872ca9 commit 15065bb

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

src/systems/callbacks.jl

+10-5
Original file line numberDiff line numberDiff line change
@@ -887,7 +887,8 @@ Compile an affect defined by a set of equations. Systems with algebraic equation
887887
function compile_equational_affect(
888888
aff::Union{AffectSystem, Vector{Equation}}, sys; reset_jumps = false, kwargs...)
889889
if aff isa AbstractVector
890-
aff = make_affect(aff; iv = get_iv(sys), alg_eqs = alg_equations(sys), warn_no_algebraic = false)
890+
aff = make_affect(
891+
aff; iv = get_iv(sys), alg_eqs = alg_equations(sys), warn_no_algebraic = false)
891892
end
892893
affsys = system(aff)
893894
ps_to_update = discretes(aff)
@@ -925,7 +926,9 @@ function compile_equational_affect(
925926
wrap_code = add_integrator_header(sys, integ, :p),
926927
expression = Val{false}, outputidxs = p_idxs, wrap_mtkparameters, cse = false)
927928

928-
return let dvs_to_update = dvs_to_update, ps_to_update = ps_to_update, reset_jumps = reset_jumps, u_up! = u_up!, p_up! = p_up!
929+
return let dvs_to_update = dvs_to_update, ps_to_update = ps_to_update,
930+
reset_jumps = reset_jumps, u_up! = u_up!, p_up! = p_up!
931+
929932
function explicit_affect!(integ)
930933
isempty(dvs_to_update) || u_up!(integ)
931934
isempty(ps_to_update) || p_up!(integ)
@@ -935,7 +938,7 @@ function compile_equational_affect(
935938
else
936939
return let dvs_to_update = dvs_to_update, aff_map = aff_map, sys_map = sys_map,
937940
affsys = affsys, ps_to_update = ps_to_update, aff = aff, sys = sys
938-
941+
939942
dvs_to_access = unknowns(affsys)
940943
ps_to_access = parameters(affsys)
941944

@@ -946,12 +949,14 @@ function compile_equational_affect(
946949
affu_getters = [getsym(affsys, sys_map[u]) for u in dvs_to_update]
947950
affp_getters = [getsym(affsys, sys_map[p]) for p in ps_to_update]
948951

949-
affprob = ImplicitDiscreteProblem(affsys, [dv => 0 for dv in dvs_to_access], (0, 0), [p => 0 for p in ps_to_access];
952+
affprob = ImplicitDiscreteProblem(affsys, [dv => 0 for dv in dvs_to_access],
953+
(0, 0), [p => 0 for p in ps_to_access];
950954
build_initializeprob = false, check_length = false)
951955

952956
function implicit_affect!(integ)
953957
pmap = Pair[p => getp(integ) for (p, getp) in zip(ps_to_access, p_getters)]
954-
u0map = Pair[u => getu(integ) for (u, getu) in zip(dvs_to_access, u_getters)]
958+
u0map = Pair[u => getu(integ)
959+
for (u, getu) in zip(dvs_to_access, u_getters)]
955960
affprob = remake(affprob, u0 = u0map, p = pmap, tspan = (integ.t, integ.t))
956961
affsol = init(affprob, IDSolve())
957962
(check_error(affsol) === ReturnCode.InitialFailure) &&

0 commit comments

Comments
 (0)