added in a check to ensure that nonexistent data is not accessed for …

…initial state prediction
JuliaMPC · Jul 7, 2020 · 981c303 · 981c303
1 parent 20a5dbb
commit 981c303
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/PrettyPlots/NLOptControl_plots.jl b/src/PrettyPlots/NLOptControl_plots.jl
@@ -132,7 +132,7 @@ function statePlot(n,idx::Int64,st::Int64,args...;kwargs...)
 
     plot!(time,vals,line=_pretty_defaults[:plant_lines][1],label=string(legend_string,"plant"));
   end
-  if _pretty_defaults[:X0p]
+  if _pretty_defaults[:X0p] && length(n.r.ip.X0p) > idx
     X0 = n.r.ip.X0p[idx][1]
     t = n.r.ip.X0p[idx][2]
     scatter!((t,X0[st]),marker=_pretty_defaults[:X0p_marker],label="X0p")