Implement contkind(::Codim2LCProblem)

tkf · Aug 16, 2018 · 77439e3 · 77439e3
1 parent ae1b1cd
commit 77439e3
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/src/base/contkind.jl b/src/base/contkind.jl
@@ -12,6 +12,9 @@ struct HopfCont <: TwoParamCont end
 # Codim1LimitCycle
 struct LimitCycleCont <: OneParamCont end
 
+# Codim2LimitCycle
+struct FoldLimitCycleCont <: TwoParamCont end
+
 
 #=
 contkind(::T) where T = ContinuationKind(T)

diff --git a/src/codim2lc/codim2lc.jl b/src/codim2lc/codim2lc.jl
@@ -16,6 +16,9 @@ import ..BifurcationsBase: TimeKind
 using ..BifurcationsBase: statekind, MutableState, ImmutableState
 import ..BifurcationsBase: StateKind
 
+# ContinuationKind trait:
+using ..BifurcationsBase: contkind, FoldLimitCycleCont
+
 import ..BifurcationsBase: BifurcationProblem
 
 using Setfield: Lens

diff --git a/src/codim2lc/solver.jl b/src/codim2lc/solver.jl
@@ -84,6 +84,9 @@ const Codim2LCSolver{
         } =
     BifurcationSolver{R, P, C, S}
 
+# TODO: trait
+BifurcationsBase.contkind(::Codim2LCProblem) = FoldLimitCycleCont()
+
 function re_analyze!(solver::Codim2LCSolver, u::AbstractVector)
     residual_jacobian!(as(solver.cache, ContinuationCache), u)
     analyze!(solver.cache, solver.opts)