improve dense plotting

SciML · Feb 5, 2017 · 8b22d5f · 8b22d5f
1 parent 9e763ee
commit 8b22d5f
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/solutions/solution_interface.jl b/src/solutions/solution_interface.jl
@@ -53,15 +53,13 @@ end
 =#
 
 @recipe function f(sol::DESolution;
-                   plot_analytic=false,denseplot=true,plotdensity=10*length(sol),vars=nothing)
-
-  if typeof(sol) <: AbstractSDESolution
-    denseplot = false
-  end
+                   plot_analytic=false,
+                   denseplot = (sol.dense || typeof(sol.prob) <: AbstractDiscreteProblem) && !(typeof(sol) <: AbstractSDESolution),
+                   plotdensity=typeof(sol.prob) <: AbstractDiscreteProblem ? 100*length(sol) : 10*length(sol),vars=nothing)
 
   int_vars = interpret_vars(vars,sol)
 
-  if denseplot && sol.dense
+  if denseplot
     # Generate the points from the plot from dense function
     plott = collect(Ranges.linspace(sol.t[1],sol.t[end],plotdensity))
     plot_timeseries = sol(plott)