Also support array indexing in bc

SciML · Dec 19, 2024 · eb13500 · eb13500
1 parent 665592e
commit eb13500
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/BoundaryValueDiffEqCore/src/misc_utils.jl b/lib/BoundaryValueDiffEqCore/src/misc_utils.jl
@@ -4,3 +4,11 @@
     t
     cache::C
 end
+
+Base.size(e::EvalSol) = (size(e.u[1])..., length(e.u))
+Base.size(e::EvalSol, i) = size(e)[i]
+
+Base.axes(e::EvalSol) = Base.OneTo.(size(e))
+Base.axes(e::EvalSol, i) = Base.OneTo.(size(e)[d])
+
+Base.getindex(e::EvalSol, args...) = Base.getindex(VectorOfArray(e.u), args...)