Implement TimeKind for BifurcationSolution/Solver

tkf · Jun 23, 2018 · 4f3bb15 · 4f3bb15
1 parent 3962b37
commit 4f3bb15
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/base/display.jl b/src/base/display.jl
@@ -4,10 +4,6 @@ tkindstr(x) = tkindstr(timekind(x))
 tkindstr(::Discrete) = "Discrete"
 tkindstr(::Continuous) = "Continuous"
 
-tkindstr(sol::BifurcationSolution) =
-    isempty(sol.sweeps) ? "?" : tkindstr(sol.sweeps[1])  # FIXME
-tkindstr(solver::BifurcationSolver) = tkindstr(solver.sol)
-
 function print_header(io::IO, point::Union{SpecialPoint,
                                            SpecialPointInterval})
     print(io, aliasof(point), " <",

diff --git a/src/base/solution.jl b/src/base/solution.jl
@@ -94,3 +94,5 @@ end
 BifurcationSolution(super::ContinuationSolution,
                     SweepType::Type{<: BifurcationSweep}) =
     BifurcationSolution(super, SweepType[])
+
+TimeKind(::Type{<: BifurcationSolution{_, W}}) where {_, W} = TimeKind(W)
diff --git a/src/base/solver.jl b/src/base/solver.jl
@@ -25,6 +25,8 @@ function BifurcationSolver(prob::BifurcationProblem,
     return BifurcationSolver(super, prob, opts, cache, sol)
 end
 
+TimeKind(::Type{<: BifurcationSolver{_, P}}) where {_, P} = TimeKind(P)
+
 
 function step!(solver::BifurcationSolver)
     step!(solver.super)