t0 and reinit! for NoiseFunction

SciML · Aug 17, 2022 · 510eeb3 · 510eeb3
1 parent 382d81c
commit 510eeb3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/noise_interfaces/common.jl b/src/noise_interfaces/common.jl
@@ -76,7 +76,7 @@ function DiffEqBase.reinit!(W::AbstractNoiseProcess, dt;
 end
 
 function DiffEqBase.reinit!(W::NoiseFunction, dt;
-                            t0 = zero(W.curt),
+                            t0 = W.t0,
                             erase_sol = true,
                             setup_next = false)
     W.curt = t0

diff --git a/src/types.jl b/src/types.jl
@@ -542,6 +542,7 @@ mutable struct NoiseFunction{T, N, wType, zType, Tt, T2, T3, inplace} <:
     dt::Tt
     dW::T2
     dZ::T3
+    t0::Tt
     reset::Bool
 
     function NoiseFunction{iip}(t0, W, Z = nothing;
@@ -560,7 +561,7 @@ mutable struct NoiseFunction{T, N, wType, zType, Tt, T2, T3, inplace} <:
         end
         new{typeof(noise_prototype), ndims(noise_prototype), typeof(W), typeof(Z),
             typeof(curt), typeof(curW), typeof(curZ), iip}(W, Z, curt, curW, curZ,
-                                                           dt, dW, dZ, reset)
+                                                           dt, dW, dZ, t0, reset)
     end
 end