|
35 | 35 | # Check if the event occured
|
36 | 36 | if typeof(callback.idxs) <: Void
|
37 | 37 | previous_condition = callback.condition(integrator.tprev,integrator.uprev,integrator)
|
38 |
| - else |
| 38 | + elseif typeof(callback.idxs) <: Number |
39 | 39 | previous_condition = callback.condition(integrator.tprev,integrator.uprev[callback.idxs],integrator)
|
| 40 | + else |
| 41 | + previous_condition = callback.condition(integrator.tprev,@view(integrator.uprev[callback.idxs]),integrator) |
40 | 42 | end
|
41 | 43 | if isapprox(previous_condition,0,rtol=callback.reltol,atol=callback.abstol)
|
42 | 44 | prev_sign = 0.0
|
|
45 | 47 | end
|
46 | 48 | prev_sign_index = 1
|
47 | 49 | if typeof(callback.idxs) <: Void
|
48 |
| - next_sign = sign(callback.condition(integrator.tprev+integrator.dt,integrator.u,integrator)) |
| 50 | + next_sign = sign(callback.condition(integrator.t,integrator.u,integrator)) |
| 51 | + elseif typeof(callback.idxs) <: Number |
| 52 | + next_sign = sign(callback.condition(integrator.t,integrator.u[callback.idxs],integrator)) |
49 | 53 | else
|
50 |
| - next_sign = sign(callback.condition(integrator.tprev+integrator.dt,integrator.u[callback.idxs],integrator)) |
| 54 | + next_sign = sign(callback.condition(integrator.t,@view(integrator.u[callback.idxs]),integrator)) |
51 | 55 | end
|
52 | 56 | if ((prev_sign<0 && !(typeof(callback.affect!)<:Void)) || (prev_sign>0 && !(typeof(callback.affect_neg!)<:Void))) && prev_sign*next_sign<0
|
53 | 57 | event_occurred = true
|
|
183 | 187 | end
|
184 | 188 |
|
185 | 189 | integrator.u_modified = true
|
186 |
| - if callback.condition(integrator.tprev+integrator.dt,integrator.u,integrator) |
| 190 | + if callback.condition(integrator.t,integrator.u,integrator) |
187 | 191 | callback.affect!(integrator)
|
188 | 192 | @inbounds if callback.save_positions[2]
|
189 | 193 | savevalues!(integrator)
|
|
0 commit comments