support MVector

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "RecursiveArrayTools"
 uuid = "731186ca-8d62-57ce-b412-fbd966d074cd"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "2.26.1"
+version = "2.26.2"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/utils.jl

@@ -55,6 +55,17 @@ function recursivefill!(b::AbstractArray{T,N},a::T2) where {T<:Union{Number,Bool
   fill!(b, a)
 end
 
+function recursivefill!(b::AbstractArray{T,N},a) where {T<:MArray,N}
+  @inbounds for i in eachindex(b)
+    if isassigned(b,i)
+      recursivefill!(b[i],a)
+    else
+      b[i] = zero(eltype(b))
+      recursivefill!(b[i], a)
+    end
+  end
+end
+
 function recursivefill!(b::AbstractArray{T,N},a) where {T<:AbstractArray,N}
   @inbounds for i in eachindex(b)
     recursivefill!(b[i], a)

test/utils_test.jl

@@ -93,4 +93,9 @@ recursivefill!(x, 1.0)
 x = [similar(x[1]), similar(x[1])]
 recursivefill!(x, true)
 @test x[1] == MVector{10}(ones(10))
+@test x[2] == MVector{10}(ones(10))
+
+x = similar(x)
+recursivefill!(x, true)
+@test x[1] == MVector{10}(ones(10))
 @test x[2] == MVector{10}(ones(10))