Merge pull request #16 from utkarsh530/u/fix

Add left & right params for Bracketing methods
SciML · Feb 1, 2021 · c4f918c · c4f918c
2 parents 87d6e58 + 7c3a4f6
commit c4f918c
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/solutions/nonlinear_solutions.jl b/src/solutions/nonlinear_solutions.jl
@@ -1,13 +1,15 @@
 """
 $(TYPEDEF)
 """
-struct NonlinearSolution{T,N,uType,R,P,A,O} <: AbstractNonlinearSolution{T,N}
+struct NonlinearSolution{T,N,uType,R,P,A,O,uType2} <: AbstractNonlinearSolution{T,N}
   u::uType
   resid::R
   prob::P
   alg::A
   retcode::Symbol
   original::O
+  left::uType2
+  right::uType2
 end
 
 const SteadyStateSolution = NonlinearSolution
@@ -16,22 +18,24 @@ function build_solution(prob::AbstractNonlinearProblem,
                         alg,u,resid;calculate_error = true,
                         retcode = :Default,
                         original = nothing,
+                        left = nothing, 
+                        right = nothing,
                         kwargs...)
 
   T = eltype(eltype(u))
   N = length((size(prob.u0)...,))
 
   NonlinearSolution{T,N,typeof(u),typeof(resid),
-                    typeof(prob),typeof(alg),typeof(original)}(
-                    u,resid,prob,alg,retcode,original)
+                    typeof(prob),typeof(alg),typeof(original),typeof(left)}(
+                    u,resid,prob,alg,retcode,original,left,right)
 end
 
-function sensitivity_solution(sol::AbstractNonlinearProblem,u)
+function sensitivity_solution(sol::AbstractNonlinearSolution,u)
   T = eltype(eltype(u))
   N = length((size(sol.prob.u0)...,))
 
   NonlinearSolution{T,N,typeof(u),typeof(sol.resid),
                     typeof(sol.prob),typeof(sol.alg),
-                    typeof(sol.original)}(
-                    u,sol.resid,sol.prob,sol.alg,sol.retcode,sol.original)
+                    typeof(sol.original),typeof(sol.left)}(
+                    u,sol.resid,sol.prob,sol.alg,sol.retcode,sol.original,sol.left,sol.right)
 end