u0 substitution in ODAEProblem

Author: YingboMa

src/structural_transformation/codegen.jl

@@ -121,7 +121,17 @@ function gen_nlsolve(eqs, vars, u0map::AbstractDict; checkbounds=true)
     params = setdiff(allvars, vars) # these are not the subject of the root finding
 
     # splatting to tighten the type
-    u0 = [map(var->get(u0map, var, 1e-3), vars)...]
+    u0 = []
+    for v in vars
+        v in keys(u0map) || (push!(u0, 1e-3); continue)
+        u = substitute(v, u0map)
+        for i in 1:1000
+            u = substitute(u, u0map)
+            u isa Number && (push!(u0, u); continue)
+        end
+        u isa Number || error("$v doesn't have a default.")
+    end
+    u0 = [u0...]
     # specialize on the scalar case
     isscalar = length(u0) == 1
     u0 = isscalar ? u0[1] : SVector(u0...)