full static array support by isinplace instead of array checks

Author: ChrisRackauckas

src/integrators/integrator_utils.jl

@@ -233,13 +233,13 @@ end
 
   #Update uprev
   if alg_extrapolates(integrator.alg)
-    if typeof(integrator.u) <: AbstractArray
+    if isinplace(integrator.sol.prob)
       recursivecopy!(integrator.uprev2,integrator.uprev)
     else
       integrator.uprev2 = integrator.uprev
     end
   end
-  if typeof(integrator.u) <: AbstractArray
+  if isinplace(integrator.sol.prob)
     recursivecopy!(integrator.uprev,integrator.u)
   else
     integrator.uprev = integrator.u
@@ -260,7 +260,7 @@ end
     elseif integrator.reeval_fsal || (typeof(integrator.alg)<:DP8 && !integrator.opts.calck) || (typeof(integrator.alg)<:Union{Rosenbrock23,Rosenbrock32} && !integrator.opts.adaptive)
       reset_fsal!(integrator)
     else # Do not reeval_fsal, instead copy! over
-      if typeof(integrator.fsalfirst) <: Union{AbstractArray,ArrayPartition}
+      if isinplace(integrator.sol.prob)
         recursivecopy!(integrator.fsalfirst,integrator.fsallast)
       else
         integrator.fsalfirst = integrator.fsallast

test/REQUIRE

@@ -6,3 +6,4 @@ ParameterizedFunctions 0.5.0
 ODEInterfaceDiffEq
 DiffEqBase 0.10.0
 SpecialMatrices
+StaticArrays

test/runtests.jl

@@ -34,6 +34,7 @@ tic()
 @time @testset "saveat Tests" begin include("ode/ode_saveat_tests.jl") end
 (LONGER_TESTS) && @time @testset "Feagin Tests" begin include("ode/ode_feagin_tests.jl") end
 @time @testset "Number Type Tests" begin include("ode/ode_numbertype_tests.jl") end
+@time @testset "Static Array Tests" begin include("static_array_tests.jl") end
 @time @testset "Data Array Tests" begin include("data_array_test.jl") end
 @time @testset "Ndim Complex Tests" begin include("ode/ode_ndim_complex_tests.jl") end
 @time @testset "Iterator Tests" begin include("iterator_tests.jl") end

test/static_array_tests.jl

@@ -0,0 +1,25 @@
+using StaticArrays
+using DiffEqBase, OrdinaryDiffEq
+
+u0 = zeros(MVector{2,Float64}, 2) + 1
+u0[1] = ones(MVector{2,Float64}) + 1
+f = (t,u,du) -> du .= u
+ode = ODEProblem(f, u0, (0.,1.))
+sol = solve(ode, Euler(), dt=1.e-2)
+
+u0 = zeros(SVector{2,Float64}, 2) + 1
+u0[1] = ones(SVector{2,Float64}) + 1
+ode = ODEProblem(f, u0, (0.,1.))
+sol = solve(ode, Euler(), dt=1.e-2)
+
+sol = solve(ode, SSPRK22(), dt=1.e-2)
+
+
+u0 = zero(MVector{2,Float64}) + 1
+ode = ODEProblem(f, u0, (0.,1.))
+sol = solve(ode, Euler(), dt=1.e-2)
+
+u0 = zero(SVector{2,Float64}) + 1
+f = (t,u) -> u
+ode = ODEProblem(f, u0, (0.,1.))
+sol = solve(ode, Euler(), dt=1.e-2)