more integrator interface on stochastic

SciML · Nov 15, 2017 · 743813d · 743813d
1 parent 80afc6b
commit 743813d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/src/StochasticDiffEq.jl b/src/StochasticDiffEq.jl
@@ -26,9 +26,10 @@ module StochasticDiffEq
 
   import DiffEqBase: solve, solve!, init, step!, build_solution, initialize!
 
-  import DiffEqBase: resize!,deleteat!,addat!,full_cache,user_cache, u_cache,du_cache,
+  # Integrator Interface
+  import DiffEqBase: resize!,deleteat!,addat!,full_cache,user_cache,u_cache,du_cache,
                      resize_non_user_cache!,deleteat_non_user_cache!,addat_non_user_cache!,
-                     terminate!,get_du, get_dt,get_proposed_dt,
+                     terminate!,get_du, get_dt,get_proposed_dt,set_proposed_dt!,
                      u_modified!,savevalues!,add_tstop!,add_saveat!,set_reltol!,
                      set_abstol!
 

diff --git a/src/integrators/integrator_interface.jl b/src/integrators/integrator_interface.jl
@@ -28,6 +28,19 @@ end
 
 (integrator::SDEIntegrator)(val::AbstractArray,t::Union{Number,AbstractArray},deriv::Type=Val{0};idxs=nothing) = current_interpolant!(val,t,integrator,idxs,deriv)
 
+function u_modified!(integrator::SDEIntegrator,bool::Bool)
+  integrator.u_modified = bool
+end
+
+get_proposed_dt(integrator::SDEIntegrator) = integrator.dtpropose
+set_proposed_dt!(integrator::SDEIntegrator,dt::Number) = (integrator.dtpropose = dt)
+
+function set_proposed_dt!(integrator::SDEIntegrator,integrator2::SDEIntegrator)
+  integrator.dtpropose = integrator2.dtpropose
+  integrator.qold = integrator2.qold
+  integrator.erracc = integrator2.erracc
+  integrator.dtacc = integrator2.dtacc
+end
 
 user_cache(integrator::SDEIntegrator) = user_cache(integrator.cache)
 u_cache(integrator::SDEIntegrator) = u_cache(integrator.cache)