Move StateKind trait to Continuations

tkf · Jun 20, 2018 · f2c6baf · f2c6baf
1 parent ab8e81b
commit f2c6baf
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 9 deletions.
diff --git a/src/Bifurcations.jl b/src/Bifurcations.jl
@@ -11,6 +11,10 @@ import .Continuations: get_prob_cache, get_u0, residual!, residual_jacobian!,
     residual, isindomain
 const _C = AbstractProblemCache
 
+# StateKind trait:
+using .Continuations: statekind, MutableState, ImmutableState
+import .Continuations: StateKind
+
 include("codim1/codim1.jl")
 using .Codim1: timekind, Continuous, Discrete
 import .Codim1: TimeKind

diff --git a/src/continuations/base.jl b/src/continuations/base.jl
@@ -1,3 +1,11 @@
+abstract type StateKind end
+struct MutableState <: StateKind end
+struct ImmutableState <: StateKind end
+
+statekind(::T) where T = StateKind(T)
+# TODO: Use StateKind everywhere instead of boolean iip type parameter.
+
+
 """
 Definition of continuation problem.
 

diff --git a/src/diffeq.jl b/src/diffeq.jl
@@ -4,15 +4,6 @@ using Setfield: Lens, set, get
 const DEP{iip} = AbstractODEProblem{uType, tType, iip} where {uType, tType}
 
 
-abstract type StateKind end
-struct MutableState <: StateKind end
-struct ImmutableState <: StateKind end
-
-statekind(::T) where T = StateKind(T)
-# TODO: Move StateKind to continuations/base.jl and use it everywhere.
-# It's only used in here at the moment.
-
-
 TimeKind(::Type{<: DiscreteProblem}) = Discrete()
 TimeKind(::Type{<: AbstractODEProblem}) = Continuous()
 StateKind(::Type{<: DEP{true}}) = MutableState()