Small fixes to callbacks

src/callbacks/density_reinit.jl

@@ -6,7 +6,9 @@ Callback to reinitialize the density field when using [`ContinuityDensity`](@ref
 # Keywords
 - `interval=0`:              Reinitialize the density every `interval` time steps.
 - `dt`:                      Reinitialize the density in regular intervals of `dt` in terms
-                             of integration time.
+                             of integration time. This callback does not add extra time
+                             steps / `tstops`; instead, reinitialization is triggered at
+                             the first solver step after each `dt` interval has elapsed.
 - `reinit_initial_solution`: Reinitialize the initial solution (default=false)
 """
 mutable struct DensityReinitializationCallback{I}
@@ -102,4 +104,6 @@ function (reinit_callback::DensityReinitializationCallback)(integrator)
     @trixi_timeit timer() "reinit density" reinit_density!(vu_ode, semi)
 
     reinit_callback.last_t = integrator.t
+
+    u_modified!(integrator, true)
 end

src/callbacks/solution_saving.jl

@@ -104,7 +104,9 @@ function SolutionSavingCallback(; interval::Integer=0, dt=0.0,
                                                -1, Ref("UnknownVersion"))
 
     if length(save_times) > 0
-        return PresetTimeCallback(save_times, solution_callback)
+        return PresetTimeCallback(save_times, solution_callback,
+                                  save_positions=(false, false),
+                                  initialize=(initialize_save_cb!))
     elseif dt > 0
         # Add a `tstop` every `dt`, and save the final solution
         return PeriodicCallback(solution_callback, dt,

src/callbacks/steady_state_reached.jl

@@ -50,25 +50,20 @@ function SteadyStateReachedCallback(; interval::Integer=0, dt=0.0,
     end
 end
 
-# `affect!` (`PeriodicCallback`)
+# `affect!` (`PeriodicCallback` and `DiscreteCallback`)
 function (cb::SteadyStateReachedCallback)(integrator)
-    steady_state_condition!(cb, integrator) || return nothing
+    if steady_state_condition!(cb, integrator)
+        print_summary(integrator)
 
-    print_summary(integrator)
-
-    terminate!(integrator)
-end
-
-# `affect!` (`DiscreteCallback`)
-function (cb::SteadyStateReachedCallback{Int})(integrator)
-    print_summary(integrator)
-
-    terminate!(integrator)
+        terminate!(integrator)
+    end
 end
 
 # `condition` (`DiscreteCallback`)
 function (steady_state_callback::SteadyStateReachedCallback)(vu_ode, t, integrator)
-    return steady_state_condition!(steady_state_callback, integrator)
+    (; interval) = steady_state_callback
+
+    return condition_integrator_interval(integrator, interval)
 end
 
 @inline function steady_state_condition!(cb, integrator)

src/general/initial_condition.jl

@@ -368,8 +368,8 @@ function InitialCondition(sol::ODESolution, system, semi; use_final_velocity=fal
 
     v_ode, u_ode = sol.u[end].x
 
+    v = wrap_v(v_ode, system, semi)
     u = wrap_u(u_ode, system, semi)
-    v = wrap_u(v_ode, system, semi)
 
     # Check if particles come too close especially when the surface exhibits large curvature
     too_close = find_too_close_particles(u, min_particle_distance)

src/schemes/fluid/weakly_compressible_sph/system.jl

@@ -392,6 +392,13 @@ end
 
 function reinit_density!(system::WeaklyCompressibleSPHSystem, v, u,
                          v_ode, u_ode, semi)
+    (; density_calculator) = system
+
+    reinit_density!(system, density_calculator, v, u, v_ode, u_ode, semi)
+end
+
+function reinit_density!(system::WeaklyCompressibleSPHSystem, ::ContinuityDensity, v, u,
+                         v_ode, u_ode, semi)
     # Compute density with `SummationDensity` and store the result in `v`,
     # overwriting the previous integrated density.
     summation_density!(system, semi, u, u_ode, v[end, :])
@@ -407,6 +414,11 @@ function reinit_density!(system::WeaklyCompressibleSPHSystem, v, u,
     return system
 end
 
+function reinit_density!(system::WeaklyCompressibleSPHSystem, ::SummationDensity, v, u,
+                         v_ode, u_ode, semi)
+    return system
+end
+
 function reinit_density!(system, v, u, v_ode, u_ode, semi)
     return system
 end