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

Add :log10-colorbar_scale support in GR

Browse Source
This commit is contained in:
Lukas Hauertmann 2021-10-31 11:16:59 +01:00
parent 19b70c06ff
commit d1d4b4fe33
2 changed files with 38 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/backends.jl
View File

@ -377,6 +377,8 @@ const _gr_attr = merge_with_base_supported([
:colorbar,
:colorbar_title,
:colorbar_entry,
:colorbar_scale,
:clims,
:fill_z,
:line_z,
:marker_z,

40
src/backends/gr.jl
View File

@ -569,8 +569,11 @@ function gr_draw_colorbar(cbar::GRColorbar, sp::Subplot, clims, viewport_plotare
ztick = 0.5 * GR.tick(zmin, zmax)
gr_set_line(1, :solid, plot_color(:black), sp)
if sp[:colorbar_scale] == :log10
GR.setscale(2)
end
GR.axes(0, ztick, xmax, zmin, 0, 1, 0.005)
title = if isa(sp[:colorbar_title], PlotText)
sp[:colorbar_title]
else
@ -949,6 +952,9 @@ function get_z_normalized(z, clims...)
end
function gr_clims(args...)
if args[1][:clims] != :auto
return args[1][:clims]
end
lo, hi = get_clims(args...)
if lo == hi
if lo == 0
@ -992,7 +998,7 @@ function gr_display(sp::Subplot{GRBackend}, w, h, viewport_canvas)
# init the colorbar
cbar = GRColorbar()
for series in series_list(sp)
gr_add_series(sp, series)
gr_update_colorbar!(cbar, series)
@ -2085,15 +2091,41 @@ function gr_draw_heatmap(series, x, y, z, clims)
# pdf output, and also supports alpha values.
# Note that drawimage draws uniformly spaced data correctly
# even on log scales, where it is visually non-uniform.
colors = plot_color.(get(fillgrad, z, clims), series[:fillalpha])
colors = if series[:subplot][:colorbar_scale] == :identity
colors = plot_color.(get(fillgrad, z, clims), series[:fillalpha])
elseif series[:subplot][:colorbar_scale] == :log10
z_log = log10.(z)
z_normalized = get_z_normalized.(z_log, log10.(clims)...)
colors = plot_color.(map(z -> get(fillgrad, z), z_normalized), series[:fillalpha])
end
for i in eachindex(colors)
if z[i] < (clims[1])
colors[i] = set_RGBA_alpha(0, colors[i])
end
end
rgba = gr_color.(colors)
GR.drawimage(first(x), last(x), last(y), first(y), w, h, rgba)
else
if something(series[:fillalpha], 1) < 1
@warn "GR: transparency not supported in non-uniform heatmaps. Alpha values ignored."
end
z_normalized = get_z_normalized.(z, clims...)
z_normalized = if series[:subplot][:colorbar_scale] == :identity
get_z_normalized.(z, clims...)
elseif series[:subplot][:colorbar_scale] == :log10
z_log = log10.(z)
z_log = map(z -> isinf(z) ? Inf : z, z_log)
z_min = minimum(z_log)
z_log = map(z -> isinf(z) ? z_min : z, z_log)
get_z_normalized.(z_log, minimum(z_log), maximum(z_log))
end
rgba = Int32[round(Int32, 1000 + _i * 255) for _i in z_normalized]
for i in eachindex(rgba)
if z[i] < (clims[1])
rgba[i] = 0 # White, not transparent
elseif z[i] > (clims[end])
rgba[i] = 1255
end
end
if !ispolar(series)
GR.nonuniformcellarray(x, y, w, h, rgba)
else