mantao/Plots.jl
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add colorbar

Browse Source
This commit is contained in:
t-bltg 2022-01-12 19:43:49 +01:00
parent 1effd45e8b
commit 143c77334b
1 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/backends/unicodeplots.jl
View File

@ -104,39 +104,41 @@ function addUnicodeSeries!(
st = series[:seriestype] st = series[:seriestype]
# get the series data and label # get the series data and label
x, y = if st == :straightline x, y = if st === :straightline
straightline_data(series) straightline_data(series)
elseif st == :shape elseif st === :shape
shape_data(series) shape_data(series)
else else
float(series[:x]), float(series[:y]) float(series[:x]), float(series[:y])
end end
# special handling (src/interface) # special handling (src/interface)
if st == :histogram2d if st === :histogram2d
kw[:xlim][:] .= kw[:ylim][:] .= 0 kw[:xlim][:] .= kw[:ylim][:] .= 0
return UnicodePlots.densityplot(x, y; kw...) return UnicodePlots.densityplot(x, y; kw...)
elseif st == :heatmap elseif st === :spy
return UnicodePlots.heatmap( return UnicodePlots.spy(series[:z].surf; kw...)
series[:z].surf; elseif st in (:contour, :heatmap)
kw_hm_ct = (;
zlabel = sp[:colorbar_title], zlabel = sp[:colorbar_title],
colormap = up_cmap(series), colormap = up_cmap(series),
kw..., colorbar = hascolorbar(sp),
) )
elseif st == :spy if st === :contour
return UnicodePlots.spy(series[:z].surf; kw...)
elseif st == :contour
return UnicodePlots.contourplot( return UnicodePlots.contourplot(
x, y, series[:z].surf; x, y, series[:z].surf;
levels=series[:levels], colormap = up_cmap(series), kw... levels = series[:levels], kw_hm_ct..., kw...
) )
else
return UnicodePlots.heatmap(series[:z].surf; kw_hm_ct..., kw...)
end
end end
# now use the ! functions to add to the plot # now use the ! functions to add to the plot
if st in (:path, :straightline, :shape) if st in (:path, :straightline, :shape)
func = UnicodePlots.lineplot! func = UnicodePlots.lineplot!
series_kw = (; head_tail = series[:arrow] isa Arrow ? series[:arrow].side : nothing) series_kw = (; head_tail = series[:arrow] isa Arrow ? series[:arrow].side : nothing)
elseif st == :scatter || series[:markershape] != :none elseif st === :scatter || series[:markershape] != :none
func = UnicodePlots.scatterplot! func = UnicodePlots.scatterplot!
series_kw = (; marker = series[:markershape]) series_kw = (; marker = series[:markershape])
else else