Finish bifurcation function

SciML · Nov 1, 2023 · 2065fb4 · 2065fb4
1 parent 06aea47
commit 2065fb4
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/ext/CatalystBifurcationKitExtension/bifurcation_kit_extension.jl b/ext/CatalystBifurcationKitExtension/bifurcation_kit_extension.jl
@@ -3,19 +3,16 @@
 # Creates a BifurcationProblem, using a ReactionSystem as an input.
 function BK.BifurcationProblem(rs::ReactionSystem, u0_bif, ps, bif_par, args...; plot_var=nothing, record_from_solution=BK.record_sol_default, jac=true, u0=[], kwargs...)
 
-    println("HERE")
-
     # Converts symbols to symbolics.
     (bif_par isa Symbol) && (bif_par = rs.var_to_name[bif_par])
     (plot_var isa Symbol) && (plot_var = rs.var_to_name[plot_var])
-    (u0_bif isa Vector{Symbol}) && (u0_bif = symmap_to_varmap(rs, u0_bif))
-    (ps isa Vector{Symbol}) && (ps = symmap_to_varmap(rs, ps))
-
-    println("HERE")
+    (u0_bif isa Vector{Pair{Symbol, Float64}}) && (u0_bif = symmap_to_varmap(rs, u0_bif))
+    (ps isa Vector{Pair{Symbol, Float64}}) && (ps = symmap_to_varmap(rs, ps))
+    (u0 isa Vector{Pair{Symbol, Float64}}) && (u0 = symmap_to_varmap(rs, u0))
 
     # Creates NonlinearSystem.
-    conservationlaw_errorcheck(rs, vcat(ps, u0))
-    nsys = covnert(NonlinearSystem, rs; defaults=u0)
+    #conservationlaw_errorcheck(rs, vcat(ps, u0))
+    nsys = convert(NonlinearSystem, rs; remove_conserved=true, defaults=Dict(u0))
 
     # Makes BifurcationProblem.
     return BK.BifurcationProblem(nsys, u0_bif, ps, bif_par, args...; plot_var=plot_var, record_from_solution=record_from_solution, jac=jac, kwargs...)