Merge pull request #63 from kanav99/resize_jac

Adds `resize!` for JacobianCache
JuliaDiff · Jul 2, 2019 · 48897f0 · 48897f0
2 parents c4be5ff + aecb30e
commit 48897f0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/DiffEqDiffTools.jl b/src/DiffEqDiffTools.jl
@@ -4,6 +4,8 @@ module DiffEqDiffTools
 
 using LinearAlgebra, SparseArrays, StaticArrays
 
+import Base: resize!
+
 include("function_wrappers.jl")
 include("finitediff.jl")
 include("derivatives.jl")

diff --git a/src/jacobians.jl b/src/jacobians.jl
@@ -1,4 +1,4 @@
-struct JacobianCache{CacheType1,CacheType2,CacheType3,ColorType,fdtype,returntype,inplace}
+mutable struct JacobianCache{CacheType1,CacheType2,CacheType3,ColorType,fdtype,returntype,inplace}
     x1  :: CacheType1
     fx  :: CacheType2
     fx1 :: CacheType3
@@ -450,3 +450,11 @@ function finite_difference_jacobian!(
     end
     J
 end
+
+function resize!(cache::JacobianCache, i::Int)
+    resize!(cache.x1,  i)
+    resize!(cache.fx,  i)
+    cache.fx1 != nothing && resize!(cache.fx1, i)
+    cache.color = 1:i
+    nothing
+end