Merge ceaf8f0 into d59f692

SciML · Jul 12, 2019 · 1467d29 · 1467d29
2 parents d59f692 + ceaf8f0
commit 1467d29
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/integrators/integrator_utils.jl b/src/integrators/integrator_utils.jl
@@ -48,7 +48,11 @@ function modify_dt_for_tstops!(integrator)
   end
 end
 
-function savevalues!(integrator::ODEIntegrator,force_save=false,reduce_size=true)::Tuple{Bool,Bool}
+# Want to extend savevalues! for DDEIntegrator
+savevalues!(integrator::ODEIntegrator, force_save = false, reduce_size = true) =
+  _savevalues!(integrator, force_save, reduce_size)
+
+function _savevalues!(integrator, force_save, reduce_size)::Tuple{Bool,Bool}
   saved, savedexactly = false, false
   !integrator.opts.save_on && return saved, savedexactly
   while !isempty(integrator.opts.saveat) && integrator.tdir*top(integrator.opts.saveat) <= integrator.tdir*integrator.t # Perform saveat
@@ -113,7 +117,10 @@ function savevalues!(integrator::ODEIntegrator,force_save=false,reduce_size=true
   return saved, savedexactly
 end
 
-function postamble!(integrator::ODEIntegrator)
+# Want to extend postamble! for DDEIntegrator
+postamble!(integrator::ODEIntegrator) = _postamble!(integrator)
+
+function _postamble!(integrator)
   solution_endpoint_match_cur_integrator!(integrator)
   resize!(integrator.sol.t,integrator.saveiter)
   resize!(integrator.sol.u,integrator.saveiter)