feat: support setindex! for nonlinear integrators

Author: AayushSabharwal

src/NonlinearSolve.jl

@@ -53,7 +53,7 @@ using PrecompileTools: @recompile_invalidations, @compile_workload, @setup_workl
     using StaticArraysCore: StaticArray, SVector, SArray, MArray, Size, SMatrix
     using SymbolicIndexingInterface: SymbolicIndexingInterface, ParameterIndexingProxy,
                                      symbolic_container, parameter_values, state_values,
-                                     getu
+                                     getu, setu
 end
 
 @reexport using SciMLBase, SimpleNonlinearSolve

src/abstract_types.jl

@@ -220,6 +220,10 @@ function Base.getindex(cache::AbstractNonlinearSolveCache, sym)
     return getu(cache, sym)(cache)
 end
 
+function Base.setindex!(cache::AbstractNonlinearSolveCache, val, sym)
+    return setu(cache, sym)(cache, val)
+end
+
 function Base.show(io::IO, cache::AbstractNonlinearSolveCache)
     __show_cache(io, cache, 0)
 end