# https://github.com/jheinen/GR.jl # significant contributions by @jheinen import GR export GR # -------------------------------------------------------------------------------------- gr_linetype(k) = (auto = 1, solid = 1, dash = 2, dot = 3, dashdot = 4, dashdotdot = -1)[k] gr_markertype(k) = ( auto = 1, none = -1, circle = -1, rect = -7, diamond = -13, utriangle = -3, dtriangle = -5, ltriangle = -18, rtriangle = -17, pentagon = -21, hexagon = -22, heptagon = -23, octagon = -24, cross = 2, xcross = 5, (+) = 2, x = 5, star4 = -25, star5 = -26, star6 = -27, star7 = -28, star8 = -29, vline = -30, hline = -31, )[k] gr_halign(k) = (left = 1, hcenter = 2, right = 3)[k] gr_valign(k) = (top = 1, vcenter = 3, bottom = 5)[k] const gr_font_family = Dict( # compat: "times" => 101, "helvetica" => 105, "courier" => 109, "bookman" => 114, "newcenturyschlbk" => 118, "avantgarde" => 122, "palatino" => 126, "serif-roman" => 232, "sans-serif" => 233, # https://gr-framework.org/fonts.html: "times roman" => 101, "times italic" => 102, "times bold" => 103, "times bold italic" => 104, "helvetica" => 105, "helvetica oblique" => 106, "helvetica bold" => 107, "helvetica bold oblique" => 108, "courier" => 109, "courier oblique" => 110, "courier bold" => 111, "courier bold oblique" => 112, "symbol" => 113, "bookman light" => 114, "bookman light italic" => 115, "bookman demi" => 116, "bookman demi italic" => 117, "new century schoolbook roman" => 118, "new century schoolbook italic" => 119, "new century schoolbook bold" => 120, "new century schoolbook bold italic" => 121, "avantgarde book" => 122, "avantgarde book oblique" => 123, "avantgarde demi" => 124, "avantgarde demi oblique" => 125, "palatino roman" => 126, "palatino italic" => 127, "palatino bold" => 128, "palatino bold italic" => 129, "zapf chancery medium italic" => 130, "zapf dingbats" => 131, "computer modern" => 232, "dejavu sans" => 233, ) # -------------------------------------------------------------------------------------- gr_color(c) = gr_color(c, color_type(c)) gr_color(c, ::Type{<:AbstractRGB}) = UInt32( round(UInt, clamp(alpha(c) * 255, 0, 255)) << 24 + round(UInt, clamp(blue(c) * 255, 0, 255)) << 16 + round(UInt, clamp(green(c) * 255, 0, 255)) << 8 + round(UInt, clamp(red(c) * 255, 0, 255)), ) function gr_color(c, ::Type{<:AbstractGray}) g = round(UInt, clamp(gray(c) * 255, 0, 255)) α = round(UInt, clamp(alpha(c) * 255, 0, 255)) rgba = UInt32(α << 24 + g << 16 + g << 8 + g) end gr_color(c, ::Type) = gr_color(RGBA(c), RGB) set_RGBA_alpha(alpha, c::RGBA) = RGBA(red(c), green(c), blue(c), alpha) set_RGBA_alpha(alpha::Nothing, c::RGBA) = c function gr_getcolorind(c) gr_set_transparency(float(alpha(c))) convert(Int, GR.inqcolorfromrgb(red(c), green(c), blue(c))) end gr_set_linecolor(c) = GR.setlinecolorind(gr_getcolorind(_cycle(c, 1))) gr_set_fillcolor(c) = GR.setfillcolorind(gr_getcolorind(_cycle(c, 1))) gr_set_markercolor(c) = GR.setmarkercolorind(gr_getcolorind(_cycle(c, 1))) gr_set_bordercolor(c) = GR.setbordercolorind(gr_getcolorind(_cycle(c, 1))) gr_set_textcolor(c) = GR.settextcolorind(gr_getcolorind(_cycle(c, 1))) gr_set_transparency(α::Real) = GR.settransparency(clamp(α, 0, 1)) gr_set_transparency(::Nothing) = GR.settransparency(1) gr_set_transparency(c, α) = gr_set_transparency(α) gr_set_transparency(c::Colorant, ::Nothing) = gr_set_transparency(c) gr_set_transparency(c::Colorant) = GR.settransparency(alpha(c)) gr_set_arrowstyle(s::Symbol) = GR.setarrowstyle( get( ( simple = 1, hollow = 3, filled = 4, triangle = 5, filledtriangle = 6, closed = 6, open = 5, ), s, 1, ), ) gr_set_fillstyle(::Nothing) = GR.setfillintstyle(GR.INTSTYLE_SOLID) function gr_set_fillstyle(s::Symbol) GR.setfillintstyle(GR.INTSTYLE_HATCH) GR.setfillstyle(get(((/) = 9, (\) = 10, (|) = 7, (-) = 8, (+) = 11, (x) = 6), s, 9)) end # -------------------------------------------------------------------------------------- # draw line segments, splitting x/y into contiguous/finite segments # note: this can be used for shapes by passing func `GR.fillarea` function gr_polyline(x, y, func = GR.polyline; arrowside = :none, arrowstyle = :simple) iend = 0 n = length(x) while iend < n - 1 # set istart to the first index that is finite istart = -1 for j in (iend + 1):n if isfinite(x[j]) && isfinite(y[j]) istart = j break end end if istart > 0 # iend is the last finite index iend = -1 for j in (istart + 1):n if isfinite(x[j]) && isfinite(y[j]) iend = j else break end end end # if we found a start and end, draw the line segment, otherwise we're done if istart > 0 && iend > 0 func(x[istart:iend], y[istart:iend]) if arrowside in (:head, :both) gr_set_arrowstyle(arrowstyle) GR.drawarrow(x[iend - 1], y[iend - 1], x[iend], y[iend]) end if arrowside in (:tail, :both) gr_set_arrowstyle(arrowstyle) GR.drawarrow(x[istart + 1], y[istart + 1], x[istart], y[istart]) end else break end end end function gr_polyline3d(x, y, z, func = GR.polyline3d) iend = 0 n = length(x) while iend < n - 1 # set istart to the first index that is finite istart = -1 for j in (iend + 1):n if isfinite(x[j]) && isfinite(y[j]) && isfinite(z[j]) istart = j break end end if istart > 0 # iend is the last finite index iend = -1 for j in (istart + 1):n if isfinite(x[j]) && isfinite(y[j]) && isfinite(z[j]) iend = j else break end end end # if we found a start and end, draw the line segment, otherwise we're done if istart > 0 && iend > 0 func(x[istart:iend], y[istart:iend], z[istart:iend]) else break end end end gr_inqtext(x, y, s) = gr_inqtext(x, y, string(s)) function gr_inqtext(x, y, s::AbstractString) if (occursin('\\', s) || occursin("10^{", s)) && match(r".*\$[^\$]+?\$.*", String(s)) == nothing GR.inqtextext(x, y, s) else GR.inqtext(x, y, s) end end gr_text(x, y, s) = gr_text(x, y, string(s)) function gr_text(x, y, s::AbstractString) if (occursin('\\', s) || occursin("10^{", s)) && match(r".*\$[^\$]+?\$.*", String(s)) == nothing GR.textext(x, y, s) else GR.text(x, y, s) end end function gr_polaraxes(rmin::Real, rmax::Real, sp::Subplot) GR.savestate() xaxis = sp[:xaxis] yaxis = sp[:yaxis] α = 0:45:315 a = α .+ 90 sinf = sind.(a) cosf = cosd.(a) rtick_values, rtick_labels = get_ticks(sp, yaxis, update = false) #draw angular grid if xaxis[:grid] gr_set_line( xaxis[:gridlinewidth], xaxis[:gridstyle], xaxis[:foreground_color_grid], sp, ) gr_set_transparency(xaxis[:foreground_color_grid], xaxis[:gridalpha]) for i in eachindex(α) GR.polyline([sinf[i], 0], [cosf[i], 0]) end end #draw radial grid if yaxis[:grid] gr_set_line( yaxis[:gridlinewidth], yaxis[:gridstyle], yaxis[:foreground_color_grid], sp, ) gr_set_transparency(yaxis[:foreground_color_grid], yaxis[:gridalpha]) for i in eachindex(rtick_values) r = (rtick_values[i] - rmin) / (rmax - rmin) if r <= 1.0 && r >= 0.0 GR.drawarc(-r, r, -r, r, 0, 359) end end GR.drawarc(-1, 1, -1, 1, 0, 359) end #prepare to draw ticks gr_set_transparency(1) GR.setlinecolorind(90) GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_HALF) #draw angular ticks if xaxis[:showaxis] GR.drawarc(-1, 1, -1, 1, 0, 359) for i in eachindex(α) x, y = GR.wctondc(1.1 * sinf[i], 1.1 * cosf[i]) GR.textext(x, y, string((360 - α[i]) % 360, "^o")) end end #draw radial ticks if yaxis[:showaxis] for i in eachindex(rtick_values) r = (rtick_values[i] - rmin) / (rmax - rmin) if r <= 1.0 && r >= 0.0 x, y = GR.wctondc(0.05, r) gr_text(x, y, _cycle(rtick_labels, i)) end end end GR.restorestate() end # using the axis extrema and limit overrides, return the min/max value for this axis gr_x_axislims(sp::Subplot) = axis_limits(sp, :x) gr_y_axislims(sp::Subplot) = axis_limits(sp, :y) gr_z_axislims(sp::Subplot) = axis_limits(sp, :z) gr_xy_axislims(sp::Subplot) = gr_x_axislims(sp)..., gr_y_axislims(sp)... function gr_fill_viewport(vp::AVec{Float64}, c) GR.savestate() GR.selntran(0) GR.setscale(0) GR.setfillintstyle(GR.INTSTYLE_SOLID) gr_set_fillcolor(c) GR.fillrect(vp...) GR.selntran(1) GR.restorestate() end # --------------------------------------------------------- # draw ONE Shape function gr_draw_marker(series, xi, yi, clims, i, msize, strokewidth, shape::Shape) sx, sy = coords(shape) # convert to ndc coords (percentages of window) ... w, h = get_size(series) f = msize / (w + h) xi, yi = GR.wctondc(xi, yi) # ... convert back to world coordinates xs_ys = GR.ndctowc.(xi .+ sx .* f, yi .+ sy .* f) xs, ys = getindex.(xs_ys, 1), getindex.(xs_ys, 2) # draw the interior mc = get_markercolor(series, clims, i) gr_set_fill(mc) gr_set_transparency(mc, get_markeralpha(series, i)) GR.fillarea(xs, ys) # draw the shapes msc = get_markerstrokecolor(series, i) gr_set_line(strokewidth, :solid, msc, series) gr_set_transparency(msc, get_markerstrokealpha(series, i)) GR.polyline(xs, ys) end function gr_nominal_size(s) w, h = get_size(s) min(w, h) / 500 end # draw ONE symbol marker function gr_draw_marker(series, xi, yi, clims, i, msize, strokewidth, shape::Symbol) GR.setborderwidth(strokewidth) gr_set_bordercolor(get_markerstrokecolor(series, i)) gr_set_markercolor(get_markercolor(series, clims, i)) gr_set_transparency(get_markeralpha(series, i)) GR.setmarkertype(gr_markertype(shape)) GR.setmarkersize(0.3msize / gr_nominal_size(series)) GR.polymarker([xi], [yi]) end # --------------------------------------------------------- function gr_set_line(lw, style, c, s) # s can be Subplot or Series GR.setlinetype(gr_linetype(style)) GR.setlinewidth(get_thickness_scaling(s) * max(0, lw / gr_nominal_size(s))) gr_set_linecolor(c) end function gr_set_fill(c) #, a) gr_set_fillcolor(c) #, a) GR.setfillintstyle(GR.INTSTYLE_SOLID) end # this stores the conversion from a font pointsize to "percentage of window height" # (which is what GR uses). `s` can be a Series, Subplot or Plot gr_point_mult(s) = 1.5 * get_thickness_scaling(s) * px / pt / maximum(get_size(s)) # set the font attributes. function gr_set_font( f::Font, s; halign = f.halign, valign = f.valign, color = f.color, rotation = f.rotation, ) family = lowercase(f.family) GR.setcharheight(gr_point_mult(s) * f.pointsize) GR.setcharup(sind(-rotation), cosd(-rotation)) if haskey(gr_font_family, family) GR.settextfontprec( gr_font_family[family], gr_font_family[family] >= 200 ? 3 : GR.TEXT_PRECISION_STRING, ) end gr_set_textcolor(plot_color(color)) GR.settextalign(gr_halign(halign), gr_valign(valign)) end function gr_w3tondc(x, y, z) xw, yw, zw = GR.wc3towc(x, y, z) x, y = GR.wctondc(xw, yw) return x, y end # -------------------------------------------------------------------------------------- # viewport plot area function gr_viewport_from_bbox( sp::Subplot{GRBackend}, bb::BoundingBox, w, h, viewport_canvas, ) viewport = zeros(4) viewport[1] = viewport_canvas[2] * (left(bb) / w) viewport[2] = viewport_canvas[2] * (right(bb) / w) viewport[3] = viewport_canvas[4] * (1.0 - bottom(bb) / h) viewport[4] = viewport_canvas[4] * (1.0 - top(bb) / h) if hascolorbar(sp) viewport[2] -= 0.1 * (1 + RecipesPipeline.is3d(sp) / 2) end viewport end # change so we're focused on the viewport area function gr_set_viewport_cmap(sp::Subplot, viewport_plotarea) GR.setviewport( viewport_plotarea[2] + (RecipesPipeline.is3d(sp) ? 0.07 : 0.02), viewport_plotarea[2] + (RecipesPipeline.is3d(sp) ? 0.10 : 0.05), viewport_plotarea[3], viewport_plotarea[4], ) end function gr_set_viewport_polar(viewport_plotarea) xmin, xmax, ymin, ymax = viewport_plotarea ymax -= 0.05 * (xmax - xmin) xcenter = 0.5 * (xmin + xmax) ycenter = 0.5 * (ymin + ymax) r = 0.5 * NaNMath.min(xmax - xmin, ymax - ymin) GR.setviewport(xcenter - r, xcenter + r, ycenter - r, ycenter + r) GR.setwindow(-1, 1, -1, 1) r end struct GRColorbar gradients fills lines GRColorbar() = new([], [], []) end function gr_update_colorbar!(cbar::GRColorbar, series::Series) style = colorbar_style(series) style === nothing && return list = style == cbar_gradient ? cbar.gradients : style == cbar_fill ? cbar.fills : style == cbar_lines ? cbar.lines : error("Unknown colorbar style: $style.") push!(list, series) end function gr_contour_levels(series::Series, clims) levels = contour_levels(series, clims) if isfilledcontour(series) # GR implicitly uses the maximal z value as the highest level levels = levels[1:(end - 1)] end levels end function gr_colorbar_colors(series::Series, clims) if iscontour(series) levels = gr_contour_levels(series, clims) if isfilledcontour(series) # GR.contourf uses a color range according to supplied levels zrange = ignorenan_extrema(levels) else # GR.contour uses a color range according to data range zrange = clims end colors = 1000 .+ 255 .* (levels .- zrange[1]) ./ (zrange[2] - zrange[1]) else colors = 1000:1255 end round.(Int, colors) end function _cbar_unique(values, propname) out = last(values) if any(x != out for x in values) @warn "Multiple series with different $propname share a colorbar. " * "Colorbar may not reflect all series correctly." end out end # add the colorbar function gr_draw_colorbar(cbar::GRColorbar, sp::Subplot, clims, viewport_plotarea) GR.savestate() xmin, xmax = gr_xy_axislims(sp)[1:2] zmin, zmax = clims[1:2] gr_set_viewport_cmap(sp, viewport_plotarea) GR.setscale(0) GR.setwindow(xmin, xmax, zmin, zmax) if !isempty(cbar.gradients) series = cbar.gradients gr_set_gradient(_cbar_unique(get_colorgradient.(series), "color")) gr_set_transparency(_cbar_unique(get_fillalpha.(series), "fill alpha")) GR.cellarray(xmin, xmax, zmax, zmin, 1, 256, 1000:1255) end if !isempty(cbar.fills) series = cbar.fills GR.setfillintstyle(GR.INTSTYLE_SOLID) gr_set_gradient(_cbar_unique(get_colorgradient.(series), "color")) gr_set_transparency(_cbar_unique(get_fillalpha.(series), "fill alpha")) levels = _cbar_unique(contour_levels.(series, Ref(clims)), "levels") # GR implicitly uses the maximal z value as the highest level if levels[end] < clims[2] @warn "GR: highest contour level less than maximal z value is not supported." # replace levels, rather than assign to levels[end], to ensure type # promotion in case levels is an integer array levels = [levels[1:(end - 1)]; clims[2]] end colors = gr_colorbar_colors(last(series), clims) for (from, to, color) in zip(levels[1:(end - 1)], levels[2:end], colors) GR.setfillcolorind(color) GR.fillrect(xmin, xmax, from, to) end end if !isempty(cbar.lines) series = cbar.lines gr_set_gradient(_cbar_unique(get_colorgradient.(series), "color")) gr_set_line( _cbar_unique(get_linewidth.(series), "line width"), _cbar_unique(get_linestyle.(series), "line style"), _cbar_unique(get_linecolor.(series, Ref(clims)), "line color"), sp, ) gr_set_transparency(_cbar_unique(get_linealpha.(series), "line alpha")) levels = _cbar_unique(contour_levels.(series, Ref(clims)), "levels") colors = gr_colorbar_colors(last(series), clims) for (line, color) in zip(levels, colors) GR.setlinecolorind(color) GR.polyline([xmin, xmax], [line, line]) end end 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 text(sp[:colorbar_title], colorbartitlefont(sp)) end gr_set_font(title.font, sp) GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_TOP) GR.setcharup(-1, 0) gr_text(viewport_plotarea[2] + 0.1, gr_view_ycenter(viewport_plotarea), title.str) GR.restorestate() end gr_view_xcenter(viewport_plotarea) = 0.5 * (viewport_plotarea[1] + viewport_plotarea[2]) gr_view_ycenter(viewport_plotarea) = 0.5 * (viewport_plotarea[3] + viewport_plotarea[4]) gr_view_xposition(viewport_plotarea, position) = viewport_plotarea[1] + position * (viewport_plotarea[2] - viewport_plotarea[1]) gr_view_yposition(viewport_plotarea, position) = viewport_plotarea[3] + position * (viewport_plotarea[4] - viewport_plotarea[3]) function position(symb) if symb == :top || symb == :right return 0.95 elseif symb == :left || symb == :bottom return 0.05 end return 0.5 end function alignment(symb) if symb == :top || symb == :right return GR.TEXT_HALIGN_RIGHT elseif symb == :left || symb == :bottom return GR.TEXT_HALIGN_LEFT end return GR.TEXT_HALIGN_CENTER end # -------------------------------------------------------------------------------------- function gr_set_gradient(c) grad = _as_gradient(c) for (i, z) in enumerate(range(0, stop = 1, length = 256)) c = grad[z] GR.setcolorrep(999 + i, red(c), green(c), blue(c)) end grad end function gr_set_gradient(series::Series) color = get_colorgradient(series) color !== nothing && gr_set_gradient(color) end # this is our new display func... set up the viewport_canvas, compute bounding boxes, and display each subplot function gr_display(plt::Plot, fmt = "") GR.clearws() dpi_factor = fmt == "png" ? plt[:dpi] / Plots.DPI : 1 # collect some monitor/display sizes in meters and pixels display_width_meters, display_height_meters, display_width_px, display_height_px = GR.inqdspsize() display_width_ratio = display_width_meters / display_width_px display_height_ratio = display_height_meters / display_height_px # compute the viewport_canvas, normalized to the larger dimension viewport_canvas = Float64[0, 1, 0, 1] w, h = get_size(plt) if w > h ratio = float(h) / w msize = display_width_ratio * w * dpi_factor GR.setwsviewport(0, msize, 0, msize * ratio) GR.setwswindow(0, 1, 0, ratio) viewport_canvas[3] *= ratio viewport_canvas[4] *= ratio else ratio = float(w) / h msize = display_height_ratio * h * dpi_factor GR.setwsviewport(0, msize * ratio, 0, msize) GR.setwswindow(0, ratio, 0, 1) viewport_canvas[1] *= ratio viewport_canvas[2] *= ratio end # fill in the viewport_canvas background gr_fill_viewport(viewport_canvas, plt[:background_color_outside]) # subplots: for sp in plt.subplots gr_display(sp, w * px, h * px, viewport_canvas) end GR.updatews() end function gr_set_tickfont(sp, letter) axis = sp[get_attr_symbol(letter, :axis)] gr_set_font( tickfont(axis), sp, rotation = axis[:rotation], color = axis[:tickfontcolor], ) end # size of the text with no rotation function gr_text_size(str) GR.savestate() GR.selntran(0) GR.setcharup(0, 1) xs, ys = gr_inqtext(0, 0, string(str)) l, r = extrema(xs) b, t = extrema(ys) w = r - l h = t - b GR.restorestate() return w, h end # size of the text with rotation applied function gr_text_size(str, rot) GR.savestate() GR.selntran(0) GR.setcharup(0, 1) xs, ys = gr_inqtext(0, 0, string(str)) l, r = extrema(xs) b, t = extrema(ys) w = text_box_width(r - l, t - b, rot) h = text_box_height(r - l, t - b, rot) GR.restorestate() return w, h end text_box_width(w, h, rot) = abs(cosd(rot)) * w + abs(cosd(rot + 90)) * h text_box_height(w, h, rot) = abs(sind(rot)) * w + abs(sind(rot + 90)) * h function gr_get_3d_axis_angle(cvs, nt, ft, letter) length(cvs) < 2 && return 0 tickpoints = [gr_w3tondc(sort_3d_axes(cv, nt, ft, letter)...) for cv in cvs] dx = tickpoints[2][1] - tickpoints[1][1] dy = tickpoints[2][2] - tickpoints[1][2] return atand(dy, dx) end function gr_get_ticks_size(ticks, rot) w, h = 0.0, 0.0 for (cv, dv) in zip(ticks...) wi, hi = gr_text_size(dv, rot) w = NaNMath.max(w, wi) h = NaNMath.max(h, hi) end return w, h end function labelfunc(scale::Symbol, backend::GRBackend) texfunc = labelfunc_tex(scale) # replace dash with \minus (U+2212) label -> replace(texfunc(label), "-" => "−") end function gr_axis_height(sp, axis) GR.savestate() ticks = get_ticks(sp, axis, update = false) gr_set_font(tickfont(axis), sp) h = ( ticks in (nothing, false, :none) ? 0 : last(gr_get_ticks_size(ticks, axis[:rotation])) ) if axis[:guide] != "" gr_set_font(guidefont(axis), sp) h += last(gr_text_size(axis[:guide])) end GR.restorestate() return h end function gr_axis_width(sp, axis) GR.savestate() ticks = get_ticks(sp, axis, update = false) gr_set_font(tickfont(axis), sp) w = ( ticks in (nothing, false, :none) ? 0 : first(gr_get_ticks_size(ticks, axis[:rotation])) ) if axis[:guide] != "" gr_set_font(guidefont(axis), sp) w += last(gr_text_size(axis[:guide])) end GR.restorestate() return w end function _update_min_padding!(sp::Subplot{GRBackend}) dpi = sp.plt[:thickness_scaling] ENV["GKS_ENCODING"] = "utf8" if !haskey(ENV, "GKSwstype") if isijulia() ENV["GKSwstype"] = "svg" end end # Add margin given by the user leftpad = 2mm + sp[:left_margin] toppad = 2mm + sp[:top_margin] rightpad = 2mm + sp[:right_margin] bottompad = 2mm + sp[:bottom_margin] # Add margin for title if sp[:title] != "" gr_set_font(titlefont(sp), sp) l = last(last(gr_text_size(sp[:title]))) h = 1mm + get_size(sp)[2] * l * px toppad += h end if RecipesPipeline.is3d(sp) xaxis, yaxis, zaxis = sp[:xaxis], sp[:yaxis], sp[:zaxis] xticks, yticks, zticks = get_ticks(sp, xaxis), get_ticks(sp, yaxis), get_ticks(sp, zaxis) # Add margin for x and y ticks h = 0mm if !isempty(first(xticks)) gr_set_font( tickfont(xaxis), sp, halign = (:left, :hcenter, :right)[sign(xaxis[:rotation]) + 2], valign = (xaxis[:mirror] ? :bottom : :top), rotation = xaxis[:rotation], ) l = 0.01 + last(gr_get_ticks_size(xticks, xaxis[:rotation])) h = max(h, 1mm + get_size(sp)[2] * l * px) end if !isempty(first(yticks)) gr_set_font( tickfont(yaxis), sp, halign = (:left, :hcenter, :right)[sign(yaxis[:rotation]) + 2], valign = (yaxis[:mirror] ? :bottom : :top), rotation = yaxis[:rotation], ) l = 0.01 + last(gr_get_ticks_size(yticks, yaxis[:rotation])) h = max(h, 1mm + get_size(sp)[2] * l * px) end if h > 0mm if xaxis[:mirror] || yaxis[:mirror] toppad += h end if !xaxis[:mirror] || !yaxis[:mirror] bottompad += h end end if !isempty(first(zticks)) gr_set_font( tickfont(zaxis), sp, halign = (zaxis[:mirror] ? :left : :right), valign = (:top, :vcenter, :bottom)[sign(zaxis[:rotation]) + 2], rotation = zaxis[:rotation], color = zaxis[:tickfontcolor], ) l = 0.01 + first(gr_get_ticks_size(zticks, zaxis[:rotation])) w = 1mm + get_size(sp)[1] * l * px if zaxis[:mirror] rightpad += w else leftpad += w end end # Add margin for x or y label h = 0mm if xaxis[:guide] != "" gr_set_font(guidefont(sp[:xaxis]), sp) l = last(gr_text_size(sp[:xaxis][:guide])) h = max(h, 1mm + get_size(sp)[2] * l * px) end if yaxis[:guide] != "" gr_set_font(guidefont(sp[:yaxis]), sp) l = last(gr_text_size(sp[:yaxis][:guide])) h = max(h, 1mm + get_size(sp)[2] * l * px) end if h > 0mm if ( xaxis[:guide_position] == :top || (xaxis[:guide_position] == :auto && xaxis[:mirror] == true) ) toppad += h else bottompad += h end end # Add margin for z label if zaxis[:guide] != "" gr_set_font(guidefont(sp[:zaxis]), sp) l = last(gr_text_size(sp[:zaxis][:guide])) w = 1mm + get_size(sp)[2] * l * px if ( zaxis[:guide_position] == :right || (zaxis[:guide_position] == :auto && zaxis[:mirror] == true) ) rightpad += w else leftpad += w end end else # Add margin for x and y ticks xticks, yticks = get_ticks(sp, sp[:xaxis]), get_ticks(sp, sp[:yaxis]) if !isempty(first(xticks)) gr_set_tickfont(sp, :x) l = 0.01 + last(gr_get_ticks_size(xticks, sp[:xaxis][:rotation])) h = 1mm + get_size(sp)[2] * l * px if sp[:xaxis][:mirror] toppad += h else bottompad += h end end if !isempty(first(yticks)) gr_set_tickfont(sp, :y) l = 0.01 + first(gr_get_ticks_size(yticks, sp[:yaxis][:rotation])) w = 1mm + get_size(sp)[1] * l * px if sp[:yaxis][:mirror] rightpad += w else leftpad += w end end # Add margin for x label if sp[:xaxis][:guide] != "" gr_set_font(guidefont(sp[:xaxis]), sp) l = last(gr_text_size(sp[:xaxis][:guide])) h = 1mm + get_size(sp)[2] * l * px if ( sp[:xaxis][:guide_position] == :top || (sp[:xaxis][:guide_position] == :auto && sp[:xaxis][:mirror] == true) ) toppad += h else bottompad += h end end # Add margin for y label if sp[:yaxis][:guide] != "" gr_set_font(guidefont(sp[:yaxis]), sp) l = last(gr_text_size(sp[:yaxis][:guide])) w = 1mm + get_size(sp)[2] * l * px if ( sp[:yaxis][:guide_position] == :right || (sp[:yaxis][:guide_position] == :auto && sp[:yaxis][:mirror] == true) ) rightpad += w else leftpad += w end end end if sp[:colorbar_title] != "" rightpad += 4mm end sp.minpad = Tuple(dpi * [leftpad, toppad, rightpad, bottompad]) end function is_equally_spaced(v) d = collect(v[2:end] .- v[1:(end - 1)]) all(d .≈ d[1]) end remap(x, lo, hi) = (x - lo) / (hi - lo) function get_z_normalized(z, clims...) isnan(z) && return 256 / 255 return remap(clamp(z, clims...), clims...) end function gr_clims(args...) if args[1][:clims] != :auto return get_clims(args[1]) end lo, hi = get_clims(args...) if lo == hi if lo == 0 hi = one(hi) elseif lo < 0 hi = zero(hi) else lo = zero(lo) end end return lo, hi end function gr_display(sp::Subplot{GRBackend}, w, h, viewport_canvas) _update_min_padding!(sp) # the viewports for this subplot viewport_subplot = gr_viewport_from_bbox(sp, bbox(sp), w, h, viewport_canvas) viewport_plotarea = gr_viewport_from_bbox(sp, plotarea(sp), w, h, viewport_canvas) # update viewport_plotarea leg = gr_get_legend_geometry(viewport_plotarea, sp) gr_update_viewport_legend!(viewport_plotarea, sp, leg) gr_update_viewport_ratio!(viewport_plotarea, sp) # fill in the plot area background gr_fill_plotarea(sp, viewport_plotarea) # set our plot area view GR.setviewport(viewport_plotarea...) # set the scale flags and window gr_set_window(sp, viewport_plotarea) # draw the axes gr_draw_axes(sp, viewport_plotarea) gr_add_title(sp, viewport_plotarea, viewport_subplot) # this needs to be here to point the colormap to the right indices GR.setcolormap(1000 + GR.COLORMAP_COOLWARM) # init the colorbar cbar = GRColorbar() for series in series_list(sp) gr_add_series(sp, series) gr_update_colorbar!(cbar, series) end # draw the colorbar hascolorbar(sp) && gr_draw_colorbar(cbar, sp, gr_clims(sp), viewport_plotarea) # add the legend gr_add_legend(sp, leg, viewport_plotarea) # add annotations for ann in sp[:annotations] x, y, val = locate_annotation(sp, ann...) x, y = if RecipesPipeline.is3d(sp) gr_w3tondc(x, y, z) else GR.wctondc(x, y) end gr_set_font(val.font, sp) gr_text(x, y, val.str) end end ## Legend function gr_add_legend(sp, leg, viewport_plotarea) if !(sp[:legend_position] in (:none, :inline)) GR.savestate() GR.selntran(0) GR.setscale(0) gr_set_font(legendfont(sp), sp) if leg.w > 0 xpos, ypos = gr_legend_pos(sp, leg, viewport_plotarea) GR.setfillintstyle(GR.INTSTYLE_SOLID) gr_set_fillcolor(sp[:legend_background_color]) GR.fillrect( xpos - leg.leftw, xpos + leg.textw + leg.rightw, ypos + leg.dy, ypos - leg.h, ) # Allocating white space for actual legend width here gr_set_line(1, :solid, sp[:legend_foreground_color], sp) GR.drawrect( xpos - leg.leftw, xpos + leg.textw + leg.rightw, ypos + leg.dy, ypos - leg.h, ) # Drawing actual legend width here i = 0 if sp[:legend_title] !== nothing GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_HALF) gr_set_font(legendtitlefont(sp), sp) gr_text(xpos - 0.03 + 0.5 * leg.w, ypos, string(sp[:legend_title])) ypos -= leg.dy gr_set_font(legendfont(sp), sp) end for series in series_list(sp) clims = gr_clims(sp, series) should_add_to_legend(series) || continue st = series[:seriestype] lc = get_linecolor(series, clims) gr_set_line(sp[:legend_font_pointsize] / 8, get_linestyle(series), lc, sp) if ( (st == :shape || series[:fillrange] !== nothing) && series[:ribbon] === nothing ) fc = get_fillcolor(series, clims) gr_set_fill(fc) fs = get_fillstyle(series, i) gr_set_fillstyle(fs) l, r = xpos - leg.width_factor * 3.5, xpos - leg.width_factor / 2 b, t = ypos - 0.4 * leg.dy, ypos + 0.4 * leg.dy x = [l, r, r, l, l] y = [b, b, t, t, b] gr_set_transparency(fc, get_fillalpha(series)) gr_polyline(x, y, GR.fillarea) lc = get_linecolor(series, clims) gr_set_transparency(lc, get_linealpha(series)) gr_set_line(get_linewidth(series), get_linestyle(series), lc, sp) st == :shape && gr_polyline(x, y) end if st in (:path, :straightline, :path3d) gr_set_transparency(lc, get_linealpha(series)) if series[:fillrange] === nothing || series[:ribbon] !== nothing GR.polyline( [xpos - leg.width_factor * 3.5, xpos - leg.width_factor / 2], [ypos, ypos], ) else GR.polyline( [xpos - leg.width_factor * 3.5, xpos - leg.width_factor / 2], [ypos + 0.4 * leg.dy, ypos + 0.4 * leg.dy], ) end end if series[:markershape] != :none ms = first(series[:markersize]) msw = first(series[:markerstrokewidth]) s, sw = if ms > 0 0.8 * sp[:legend_font_pointsize], 0.8 * sp[:legend_font_pointsize] * msw / ms else 0, 0.8 * sp[:legend_font_pointsize] * msw / 8 end gr_draw_markers(series, xpos - leg.width_factor * 2, ypos, clims, s, sw) end lab = series[:label] GR.settextalign(GR.TEXT_HALIGN_LEFT, GR.TEXT_VALIGN_HALF) gr_set_textcolor(plot_color(sp[:legend_font_color])) gr_text(xpos, ypos, string(lab)) ypos -= leg.dy end end GR.selntran(1) GR.restorestate() end end function gr_legend_pos(sp::Subplot, leg, viewport_plotarea) s = sp[:legend_position] s isa Real && return gr_legend_pos(s, leg, viewport_plotarea) if s isa Tuple{<:Real,Symbol} if s[2] !== :outer return gr_legend_pos(s[1], leg, viewport_plotarea) end xaxis, yaxis = sp[:xaxis], sp[:yaxis] xmirror = xaxis[:guide_position] == :top || (xaxis[:guide_position] == :auto && xaxis[:mirror] == true) ymirror = yaxis[:guide_position] == :right || (yaxis[:guide_position] == :auto && yaxis[:mirror] == true) axisclearance = [ !ymirror * gr_axis_width(sp, sp[:yaxis]), ymirror * gr_axis_width(sp, sp[:yaxis]), !xmirror * gr_axis_height(sp, sp[:xaxis]), xmirror * gr_axis_height(sp, sp[:xaxis]), ] return gr_legend_pos(s[1], leg, viewport_plotarea; axisclearance) end s isa Symbol || return gr_legend_pos(s, viewport_plotarea) str = string(s) if str == "best" str = "topright" end if occursin("outer", str) xaxis, yaxis = sp[:xaxis], sp[:yaxis] xmirror = xaxis[:guide_position] == :top || (xaxis[:guide_position] == :auto && xaxis[:mirror] == true) ymirror = yaxis[:guide_position] == :right || (yaxis[:guide_position] == :auto && yaxis[:mirror] == true) end if occursin("right", str) if occursin("outer", str) # As per https://github.com/jheinen/GR.jl/blob/master/src/jlgr.jl#L525 xpos = viewport_plotarea[2] + leg.xoffset + leg.leftw + ymirror * gr_axis_width(sp, sp[:yaxis]) else xpos = viewport_plotarea[2] - leg.rightw - leg.textw - leg.xoffset end elseif occursin("left", str) if occursin("outer", str) xpos = viewport_plotarea[1] - !ymirror * gr_axis_width(sp, sp[:yaxis]) - leg.xoffset * 2 - leg.rightw - leg.textw else xpos = viewport_plotarea[1] + leg.leftw + leg.xoffset end else xpos = (viewport_plotarea[2] - viewport_plotarea[1]) / 2 + viewport_plotarea[1] + leg.leftw - leg.rightw - leg.textw - leg.xoffset * 2 end if occursin("top", str) if s == :outertop ypos = viewport_plotarea[4] + leg.yoffset + leg.h + xmirror * gr_axis_height(sp, sp[:xaxis]) else ypos = viewport_plotarea[4] - leg.yoffset - leg.dy end elseif occursin("bottom", str) if s == :outerbottom ypos = viewport_plotarea[3] - leg.yoffset - leg.dy - !xmirror * gr_axis_height(sp, sp[:xaxis]) else ypos = viewport_plotarea[3] + leg.yoffset + leg.h end else # Adding min y to shift legend pos to correct graph (#2377) ypos = (viewport_plotarea[4] - viewport_plotarea[3] + leg.h) / 2 + viewport_plotarea[3] end return xpos, ypos end function gr_legend_pos(v::Tuple{S,T}, viewport_plotarea) where {S<:Real,T<:Real} xpos = v[1] * (viewport_plotarea[2] - viewport_plotarea[1]) + viewport_plotarea[1] ypos = v[2] * (viewport_plotarea[4] - viewport_plotarea[3]) + viewport_plotarea[3] (xpos, ypos) end function gr_legend_pos(theta::Real, leg, viewport_plotarea; axisclearance = nothing) xcenter = +(viewport_plotarea[1:2]...) / 2 ycenter = +(viewport_plotarea[3:4]...) / 2 if isnothing(axisclearance) # Inner # rectangle where the anchor can legally be xmin = viewport_plotarea[1] + leg.xoffset + leg.leftw xmax = viewport_plotarea[2] - leg.xoffset - leg.rightw - leg.textw ymin = viewport_plotarea[3] + leg.yoffset + leg.h ymax = viewport_plotarea[4] - leg.yoffset - leg.dy else # Outer xmin = viewport_plotarea[1] - leg.xoffset - leg.rightw - leg.textw - axisclearance[1] xmax = viewport_plotarea[2] + leg.xoffset + leg.leftw + axisclearance[2] ymin = viewport_plotarea[3] - leg.yoffset - leg.dy - axisclearance[3] ymax = viewport_plotarea[4] + leg.yoffset + leg.h + axisclearance[4] end return legend_pos_from_angle(theta, xmin, xcenter, xmax, ymin, ycenter, ymax) end function gr_get_legend_geometry(viewport_plotarea, sp) legendn = legendw = dy = 0 if sp[:legend_position] != :none GR.savestate() GR.selntran(0) GR.setcharup(0, 1) GR.setscale(0) if sp[:legend_title] !== nothing gr_set_font(legendtitlefont(sp), sp) legendn += 1 tbx, tby = gr_inqtext(0, 0, string(sp[:legend_title])) l, r = extrema(tbx) b, t = extrema(tby) legendw = r - l dy = t - b end gr_set_font(legendfont(sp), sp) for series in series_list(sp) should_add_to_legend(series) || continue legendn += 1 tbx, tby = gr_inqtext(0, 0, string(series[:label])) l, r = extrema(tbx) b, t = extrema(tby) legendw = max(legendw, r - l) # Holds text width right now dy = max(dy, t - b) end GR.setscale(1) GR.selntran(1) GR.restorestate() end legend_width_factor = 0.02 legend_textw = legendw legend_rightw = legend_width_factor legend_leftw = legend_width_factor * 4 total_legendw = legend_textw + legend_leftw + legend_rightw x_legend_offset = (viewport_plotarea[2] - viewport_plotarea[1]) / 30 y_legend_offset = (viewport_plotarea[4] - viewport_plotarea[3]) / 30 dy *= get(sp[:extra_kwargs], :legend_hfactor, 1) legendh = dy * legendn return ( w = legendw, h = legendh, dy = dy, leftw = legend_leftw, textw = legend_textw, rightw = legend_rightw, xoffset = x_legend_offset, yoffset = y_legend_offset, width_factor = legend_width_factor, ) end ## Viewport, window and scale function gr_update_viewport_legend!(viewport_plotarea, sp, leg) s = sp[:legend_position] xaxis, yaxis = sp[:xaxis], sp[:yaxis] xmirror = xaxis[:guide_position] == :top || (xaxis[:guide_position] == :auto && xaxis[:mirror] == true) ymirror = yaxis[:guide_position] == :right || (yaxis[:guide_position] == :auto && yaxis[:mirror] == true) if s isa Tuple{<:Real,Symbol} if s[2] === :outer (x, y) = gr_legend_pos(sp, leg, viewport_plotarea) # Dry run, to figure out if x < viewport_plotarea[1] viewport_plotarea[1] += leg.leftw + leg.textw + leg.rightw + leg.xoffset + !ymirror * gr_axis_width(sp, sp[:yaxis]) elseif x > viewport_plotarea[2] viewport_plotarea[2] -= leg.leftw + leg.textw + leg.rightw + leg.xoffset end if y < viewport_plotarea[3] viewport_plotarea[3] += leg.h + leg.dy + leg.yoffset + !xmirror * gr_axis_height(sp, sp[:xaxis]) elseif y > viewport_plotarea[4] viewport_plotarea[4] -= leg.h + leg.dy + leg.yoffset end end end leg_str = string(s) if occursin("outer", leg_str) if occursin("right", leg_str) viewport_plotarea[2] -= leg.leftw + leg.textw + leg.rightw + leg.xoffset elseif occursin("left", leg_str) viewport_plotarea[1] += leg.leftw + leg.textw + leg.rightw + leg.xoffset + !ymirror * gr_axis_width(sp, sp[:yaxis]) elseif occursin("top", leg_str) viewport_plotarea[4] -= leg.h + leg.dy + leg.yoffset elseif occursin("bottom", leg_str) viewport_plotarea[3] += leg.h + leg.dy + leg.yoffset + !xmirror * gr_axis_height(sp, sp[:xaxis]) end end if s == :inline if sp[:yaxis][:mirror] viewport_plotarea[1] += leg.w else viewport_plotarea[2] -= leg.w end end end function gr_update_viewport_ratio!(viewport_plotarea, sp) ratio = get_aspect_ratio(sp) if ratio != :none xmin, xmax, ymin, ymax = gr_xy_axislims(sp) if ratio == :equal ratio = 1 end viewport_ratio = (viewport_plotarea[2] - viewport_plotarea[1]) / (viewport_plotarea[4] - viewport_plotarea[3]) window_ratio = (xmax - xmin) / (ymax - ymin) / ratio if window_ratio < viewport_ratio viewport_center = 0.5 * (viewport_plotarea[1] + viewport_plotarea[2]) viewport_size = (viewport_plotarea[2] - viewport_plotarea[1]) * window_ratio / viewport_ratio viewport_plotarea[1] = viewport_center - 0.5 * viewport_size viewport_plotarea[2] = viewport_center + 0.5 * viewport_size elseif window_ratio > viewport_ratio viewport_center = 0.5 * (viewport_plotarea[3] + viewport_plotarea[4]) viewport_size = (viewport_plotarea[4] - viewport_plotarea[3]) * viewport_ratio / window_ratio viewport_plotarea[3] = viewport_center - 0.5 * viewport_size viewport_plotarea[4] = viewport_center + 0.5 * viewport_size end end end function gr_set_window(sp, viewport_plotarea) if ispolar(sp) gr_set_viewport_polar(viewport_plotarea) else xmin, xmax, ymin, ymax = gr_xy_axislims(sp) needs_3d = needs_any_3d_axes(sp) if needs_3d zmin, zmax = gr_z_axislims(sp) zok = zmax > zmin else zok = true end scaleop = 0 if xmax > xmin && ymax > ymin && zok sp[:xaxis][:scale] == :log10 && (scaleop |= GR.OPTION_X_LOG) sp[:yaxis][:scale] == :log10 && (scaleop |= GR.OPTION_Y_LOG) needs_3d && sp[:zaxis][:scale] == :log10 && (scaleop |= GR.OPTION_Z_LOG) sp[:xaxis][:flip] && (scaleop |= GR.OPTION_FLIP_X) sp[:yaxis][:flip] && (scaleop |= GR.OPTION_FLIP_Y) needs_3d && sp[:zaxis][:flip] && (scaleop |= GR.OPTION_FLIP_Z) # NOTE: setwindow sets the "data coordinate" limits of the current "viewport" GR.setwindow(xmin, xmax, ymin, ymax) GR.setscale(scaleop) end end end function gr_fill_plotarea(sp, viewport_plotarea) if !RecipesPipeline.is3d(sp) gr_fill_viewport(viewport_plotarea, plot_color(sp[:background_color_inside])) end end ## Axes function gr_draw_axes(sp, viewport_plotarea) GR.setlinewidth(sp.plt[:thickness_scaling]) if RecipesPipeline.is3d(sp) # set space xmin, xmax, ymin, ymax = gr_xy_axislims(sp) zmin, zmax = gr_z_axislims(sp) camera = round.(Int, sp[:camera]) warn_invalid(val) = if val < 0 || val > 90 @warn "camera: $(val)° ∉ [0°, 90°]" end warn_invalid.(camera) GR.setspace(zmin, zmax, camera...) # fill the plot area gr_set_fill(plot_color(sp[:background_color_inside])) plot_area_x = [xmin, xmin, xmin, xmax, xmax, xmax, xmin] plot_area_y = [ymin, ymin, ymax, ymax, ymax, ymin, ymin] plot_area_z = [zmin, zmax, zmax, zmax, zmin, zmin, zmin] x_bg, y_bg = RecipesPipeline.unzip(GR.wc3towc.(plot_area_x, plot_area_y, plot_area_z)) GR.fillarea(x_bg, y_bg) for letter in (:x, :y, :z) gr_draw_axis_3d(sp, letter, viewport_plotarea) end elseif ispolar(sp) r = gr_set_viewport_polar(viewport_plotarea) #rmin, rmax = GR.adjustrange(ignorenan_minimum(r), ignorenan_maximum(r)) rmin, rmax = axis_limits(sp, :y) gr_polaraxes(rmin, rmax, sp) elseif sp[:framestyle] != :none for letter in (:x, :y) gr_draw_axis(sp, letter, viewport_plotarea) end end end function gr_draw_axis(sp, letter, viewport_plotarea) ax = axis_drawing_info(sp, letter) axis = sp[get_attr_symbol(letter, :axis)] # draw segments gr_draw_grid(sp, axis, ax.grid_segments) gr_draw_minorgrid(sp, axis, ax.minorgrid_segments) gr_draw_spine(sp, axis, ax.segments) gr_draw_border(sp, axis, ax.border_segments) gr_draw_ticks(sp, axis, ax.tick_segments) # labels gr_label_ticks(sp, letter, ax.ticks) gr_label_axis(sp, letter, viewport_plotarea) end function gr_draw_axis_3d(sp, letter, viewport_plotarea) ax = axis_drawing_info_3d(sp, letter) axis = sp[get_attr_symbol(letter, :axis)] # draw segments gr_draw_grid(sp, axis, ax.grid_segments, gr_polyline3d) gr_draw_minorgrid(sp, axis, ax.minorgrid_segments, gr_polyline3d) gr_draw_spine(sp, axis, ax.segments, gr_polyline3d) gr_draw_border(sp, axis, ax.border_segments, gr_polyline3d) gr_draw_ticks(sp, axis, ax.tick_segments, gr_polyline3d) # labels GR.setscale(0) gr_label_ticks_3d(sp, letter, ax.ticks) gr_label_axis_3d(sp, letter) gr_set_window(sp, viewport_plotarea) end function gr_draw_grid(sp, axis, segments, func = gr_polyline) if axis[:grid] gr_set_line( axis[:gridlinewidth], axis[:gridstyle], axis[:foreground_color_grid], sp, ) gr_set_transparency(axis[:foreground_color_grid], axis[:gridalpha]) func(coords(segments)...) end end function gr_draw_minorgrid(sp, axis, segments, func = gr_polyline) if axis[:minorgrid] gr_set_line( axis[:minorgridlinewidth], axis[:minorgridstyle], axis[:foreground_color_minor_grid], sp, ) gr_set_transparency(axis[:foreground_color_minor_grid], axis[:minorgridalpha]) func(coords(segments)...) end end function gr_draw_spine(sp, axis, segments, func = gr_polyline) if axis[:showaxis] gr_set_line(1, :solid, axis[:foreground_color_border], sp) gr_set_transparency(1.0) GR.setclip(0) func(coords(segments)...) GR.setclip(1) end end function gr_draw_border(sp, axis, segments, func = gr_polyline) intensity = sp[:framestyle] == :semi ? 0.5 : 1 if sp[:framestyle] in (:box, :semi) GR.setclip(0) gr_set_line(intensity, :solid, axis[:foreground_color_border], sp) gr_set_transparency(axis[:foreground_color_border], intensity) func(coords(segments)...) GR.setclip(1) end end function gr_draw_ticks(sp, axis, segments, func = gr_polyline) if axis[:showaxis] if sp[:framestyle] in (:zerolines, :grid) gr_set_line(1, :solid, axis[:foreground_color_grid], sp) gr_set_transparency( axis[:foreground_color_grid], axis[:tick_direction] == :out ? axis[:gridalpha] : 0, ) else gr_set_line(1, :solid, axis[:foreground_color_axis], sp) end GR.setclip(0) func(coords(segments)...) GR.setclip(1) end end function gr_label_ticks(sp, letter, ticks) axis = sp[get_attr_symbol(letter, :axis)] isy = letter === :y oletter = isy ? :x : :y oaxis = sp[get_attr_symbol(oletter, :axis)] oamin, oamax = axis_limits(sp, oletter) gr_set_tickfont(sp, letter) out_factor = ifelse(axis[:tick_direction] === :out, 1.5, 1) x_base_offset = isy ? -1.5e-2 * out_factor : 0 y_base_offset = isy ? 0 : -8e-3 * out_factor rot = axis[:rotation] % 360 ov = sp[:framestyle] == :origin ? 0 : xor(oaxis[:flip], axis[:mirror]) ? oamax : oamin sgn = axis[:mirror] ? -1 : 1 sgn2 = iseven(Int(floor(rot / 90))) ? -1 : 1 sgn3 = if isy -360 < rot < -180 || 0 < rot < 180 ? 1 : -1 else rot < -270 || -90 < rot < 90 || rot > 270 ? 1 : -1 end for (cv, dv) in zip(ticks...) x, y = GR.wctondc(reverse_if((cv, ov), isy)...) sz_rot = gr_text_size(dv, rot) sz = gr_text_size(dv) x_offset = x_base_offset y_offset = y_base_offset if isy x_offset += -first(sz_rot) / 2 if rot % 90 != 0 y_offset += sgn2 * last(sz_rot) / 2 + sgn3 * last(sz) * cosd(rot) / 2 end else if rot % 90 != 0 x_offset += sgn2 * first(sz_rot) / 2 + sgn3 * last(sz) * sind(rot) / 2 end y_offset += -last(sz_rot) / 2 end gr_text(x + sgn * x_offset, y + sgn * y_offset, dv) end end function gr_label_ticks(sp, letter, ticks::Nothing) end function gr_label_ticks_3d(sp, letter, ticks) near_letter = letter in (:x, :z) ? :y : :x far_letter = letter in (:x, :y) ? :z : :x isy = letter === :y isz = letter === :z ax = sp[get_attr_symbol(letter, :axis)] nax = sp[get_attr_symbol(near_letter, :axis)] fax = sp[get_attr_symbol(far_letter, :axis)] amin, amax = axis_limits(sp, letter) namin, namax = axis_limits(sp, near_letter) famin, famax = axis_limits(sp, far_letter) n0, n1 = isy ? (namax, namin) : (namin, namax) # find out which axes we are dealing with i = findfirst(==(letter), (:x, :y, :z)) letters = axes_shift((:x, :y, :z), 1 - i) asyms = get_attr_symbol.(letters, :axis) # get axis objects, ticks and minor ticks # regardless of the `letter` we now use the convention that `x` in variable names refer to # the first axesm `y` to the second, etc ... ylims, zlims = axis_limits.(Ref(sp), letters[2:3]) xax, yax, zax = getindex.(Ref(sp), asyms) gr_set_tickfont(sp, letter) nt = sp[:framestyle] == :origin ? 0 : ax[:mirror] ? n1 : n0 ft = sp[:framestyle] == :origin ? 0 : ax[:mirror] ? famax : famin rot = mod(ax[:rotation], 360) sgn = ax[:mirror] ? -1 : 1 cvs, dvs = ticks ax[:flip] && reverse!(cvs) axisθ = isz ? 270 : mod(gr_get_3d_axis_angle(cvs, nt, ft, letter), 360) # issue: doesn't work with 1 tick axisϕ = mod(axisθ - 90, 360) out_factor = ifelse(ax[:tick_direction] === :out, 1.5, 1) axisoffset = out_factor * 1.2e-2 x_base_offset = axisoffset * cosd(axisϕ) y_base_offset = axisoffset * sind(axisϕ) sgn2a = sgn2b = sgn3 = 0 if axisθ != 0 || rot % 90 != 0 sgn2a = (axisθ != 90) && (axisθ == 0 && (rot < 90 || 180 ≤ rot < 270)) || (axisθ == 270) || (axisθ < 90 && (axisθ < rot < 90 || axisθ + 180 < rot < 270)) || (axisθ > 270 && (rot < 90 || axisθ - 180 < rot < 270 || rot > axisθ)) ? -1 : 1 end if (axisθ - 90) % 180 != 0 || (rot - 90) % 180 != 0 sgn2b = axisθ == 0 || (axisθ == 90 && (90 ≤ rot < 180 || 270 ≤ rot < 360)) || (axisθ == 270 && (rot < 90 || 180 ≤ rot < 270)) || (axisθ < 90 && (axisθ < rot < 180 || axisθ + 180 < rot)) || (axisθ > 270 && (rot < axisθ - 180 || 180 ≤ rot < axisθ)) ? -1 : 1 end if !(axisθ == 0 && rot % 180 == 0) && ((rot - 90) % 180 != 0) sgn3 = (axisθ == 0 && 90 < rot < 270) || (axisθ == 90 && rot < 180) || (axisθ == 270 && rot > 180) || (axisθ < 90 && (rot < axisθ || 90 ≤ rot < 180 || axisθ + 180 < rot < 270)) || (axisθ > 270 && (90 ≤ rot < axisθ - 180 || 180 ≤ rot < 270 || rot > axisθ)) ? -1 : 1 end for (cv, dv) in zip((cvs, dvs)...) xi, yi = gr_w3tondc(sort_3d_axes(cv, nt, ft, letter)...) sz_rot = gr_text_size(dv, rot) sz = gr_text_size(dv) x_offset = x_base_offset + sgn2a * first(sz_rot) / 2 + sgn3 * last(sz) * sind(rot) / 2 y_offset = y_base_offset + sgn2b * last(sz_rot) / 2 + sgn3 * last(sz) * cosd(rot) / 2 gr_text(xi + sgn * x_offset, yi + sgn * y_offset, dv) end end function gr_label_axis(sp, letter, viewport_plotarea) axis = sp[get_attr_symbol(letter, :axis)] mirror = axis[:mirror] # guide if axis[:guide] != "" GR.savestate() gr_set_font(guidefont(axis), sp) guide_position = axis[:guide_position] angle = float(axis[:guidefontrotation]) # github.com/JuliaPlots/Plots.jl/issues/3089 if letter === :y angle += 180.0 # default angle = 0. should yield GR.setcharup(-1, 0) i.e. 180° GR.setcharup(cosd(angle), sind(angle)) ypos = gr_view_yposition(viewport_plotarea, position(axis[:guidefontvalign])) yalign = alignment(axis[:guidefontvalign]) if guide_position === :right || (guide_position == :auto && mirror) GR.settextalign(yalign, GR.TEXT_VALIGN_BOTTOM) xpos = viewport_plotarea[2] + 0.03 + mirror * gr_axis_width(sp, axis) else GR.settextalign(yalign, GR.TEXT_VALIGN_TOP) xpos = viewport_plotarea[1] - 0.03 - !mirror * gr_axis_width(sp, axis) end else angle += 90.0 # default angle = 0. should yield GR.setcharup(0, 1) i.e. 90° GR.setcharup(cosd(angle), sind(angle)) xpos = gr_view_xposition(viewport_plotarea, position(axis[:guidefonthalign])) xalign = alignment(axis[:guidefonthalign]) if guide_position === :top || (guide_position == :auto && mirror) GR.settextalign(xalign, GR.TEXT_VALIGN_TOP) ypos = viewport_plotarea[4] + 0.015 + (mirror ? gr_axis_height(sp, axis) : 0.015) else GR.settextalign(xalign, GR.TEXT_VALIGN_BOTTOM) ypos = viewport_plotarea[3] - 0.015 - (mirror ? 0.015 : gr_axis_height(sp, axis)) end end gr_text(xpos, ypos, axis[:guide]) GR.restorestate() end end function gr_label_axis_3d(sp, letter) ax = sp[get_attr_symbol(letter, :axis)] if ax[:guide] != "" near_letter = letter in (:x, :z) ? :y : :x far_letter = letter in (:x, :y) ? :z : :x nax = sp[get_attr_symbol(near_letter, :axis)] fax = sp[get_attr_symbol(far_letter, :axis)] amin, amax = axis_limits(sp, letter) namin, namax = axis_limits(sp, near_letter) famin, famax = axis_limits(sp, far_letter) n0, n1 = letter === :y ? (namax, namin) : (namin, namax) GR.savestate() gr_set_font( guidefont(ax), sp, halign = (:left, :hcenter, :right)[sign(ax[:rotation]) + 2], valign = ax[:mirror] ? :bottom : :top, rotation = ax[:rotation], # color = ax[:guidefontcolor], ) ag = (amin + amax) / 2 ng = ax[:mirror] ? n1 : n0 fg = ax[:mirror] ? famax : famin x, y = gr_w3tondc(sort_3d_axes(ag, ng, fg, letter)...) if letter in (:x, :y) h = gr_axis_height(sp, ax) x_offset = letter === :x ? -h : h y_offset = -h else x_offset = -0.03 - gr_axis_width(sp, ax) y_offset = 0 end letter === :z && GR.setcharup(-1, 0) sgn = ax[:mirror] ? -1 : 1 gr_text(x + sgn * x_offset, y + sgn * y_offset, ax[:guide]) GR.restorestate() end end function gr_add_title(sp, viewport_plotarea, viewport_subplot) if sp[:title] != "" GR.savestate() gr_set_font(titlefont(sp), sp) loc = sp[:titlelocation] if loc == :left xpos = viewport_plotarea[1] halign = GR.TEXT_HALIGN_LEFT elseif loc == :right xpos = viewport_plotarea[2] halign = GR.TEXT_HALIGN_RIGHT else xpos = gr_view_xcenter(viewport_plotarea) halign = GR.TEXT_HALIGN_CENTER end GR.settextalign(halign, GR.TEXT_VALIGN_TOP) gr_text(xpos, viewport_subplot[4], sp[:title]) GR.restorestate() end end ## Series function gr_add_series(sp, series) st = series[:seriestype] # update the current stored gradient gr_set_gradient(series) GR.savestate() x, y, z = (handle_surface(series[letter]) for letter in (:x, :y, :z)) xscale, yscale = sp[:xaxis][:scale], sp[:yaxis][:scale] frng = series[:fillrange] # recompute data if ispolar(sp) && z === nothing rmin, rmax = axis_limits(sp, :y) if frng !== nothing _, frng = convert_to_polar(x, frng, (rmin, rmax)) end x, y = convert_to_polar(x, y, (rmin, rmax)) end clims = gr_clims(sp, series) # add custom frame shapes to markershape? series_annotations_shapes!(series) # ------------------------------------------------------- # draw the series if st in (:path, :scatter, :straightline) if st === :straightline x, y = straightline_data(series) end gr_draw_segments(series, x, y, frng, clims) if series[:markershape] !== :none gr_draw_markers(series, x, y, clims) end elseif st === :shape gr_draw_shapes(series, clims) elseif st in (:path3d, :scatter3d) gr_draw_segments_3d(series, x, y, z, clims) if st === :scatter3d || series[:markershape] !== :none # TODO: Do we need to transform to 2d coordinates here? x2, y2 = RecipesPipeline.unzip(map(GR.wc3towc, x, y, z)) gr_draw_markers(series, x2, y2, clims) end elseif st === :contour gr_draw_contour(series, x, y, z, clims) elseif st in (:surface, :wireframe, :mesh3d) gr_draw_surface(series, x, y, z, clims) elseif st === :volume sp[:legend_position] = :none GR.gr3.clear() dmin, dmax = GR.gr3.volume(y.v, 0) elseif st === :heatmap # `z` is already transposed, so we need to reverse before passing its size. x, y = heatmap_edges(x, xscale, y, yscale, reverse(size(z)), ispolar(series)) gr_draw_heatmap(series, x, y, z, clims) elseif st === :image gr_draw_image(series, x, y, z, clims) end # this is all we need to add the series_annotations text anns = series[:series_annotations] for (xi, yi, str, fnt) in EachAnn(anns, x, y) gr_set_font(fnt, sp) gr_text(GR.wctondc(xi, yi)..., str) end if sp[:legend_position] == :inline && should_add_to_legend(series) gr_set_font(legendfont(sp), sp) gr_set_textcolor(plot_color(sp[:legend_font_color])) if sp[:yaxis][:mirror] (_, i) = sp[:xaxis][:flip] ? findmax(x) : findmin(x) GR.settextalign(GR.TEXT_HALIGN_RIGHT, GR.TEXT_VALIGN_HALF) offset = -0.01 else (_, i) = sp[:xaxis][:flip] ? findmin(x) : findmax(x) GR.settextalign(GR.TEXT_HALIGN_LEFT, GR.TEXT_VALIGN_HALF) offset = 0.01 end (x_l, y_l) = GR.wctondc(x[i], y[i]) gr_text(x_l + offset, y_l, series[:label]) end GR.restorestate() end function gr_draw_segments(series, x, y, fillrange, clims) st = series[:seriestype] if x !== nothing && length(x) > 1 segments = series_segments(series, st; check = true) # do area fill if fillrange !== nothing GR.setfillintstyle(GR.INTSTYLE_SOLID) fr_from, fr_to = (is_2tuple(fillrange) ? fillrange : (y, fillrange)) for segment in segments i, rng = segment.attr_index, segment.range fc = get_fillcolor(series, clims, i) gr_set_fillcolor(fc) fs = get_fillstyle(series, i) gr_set_fillstyle(fs) fx = _cycle(x, vcat(rng, reverse(rng))) fy = vcat(_cycle(fr_from, rng), _cycle(fr_to, reverse(rng))) gr_set_transparency(fc, get_fillalpha(series, i)) GR.fillarea(fx, fy) end end # draw the line(s) if st in (:path, :straightline) for segment in segments i, rng = segment.attr_index, segment.range lc = get_linecolor(series, clims, i) gr_set_line(get_linewidth(series, i), get_linestyle(series, i), lc, series) arrowside = isa(series[:arrow], Arrow) ? series[:arrow].side : :none arrowstyle = isa(series[:arrow], Arrow) ? series[:arrow].style : :simple gr_set_fillcolor(lc) gr_set_transparency(lc, get_linealpha(series, i)) gr_polyline(x[rng], y[rng]; arrowside = arrowside, arrowstyle = arrowstyle) end end end end function gr_draw_segments_3d(series, x, y, z, clims) if series[:seriestype] === :path3d && length(x) > 1 lz = series[:line_z] segments = series_segments(series, :path3d; check = true) for segment in segments i, rng = segment.attr_index, segment.range lc = get_linecolor(series, clims, i) gr_set_line(get_linewidth(series, i), get_linestyle(series, i), lc, series) gr_set_transparency(lc, get_linealpha(series, i)) GR.polyline3d(x[rng], y[rng], z[rng]) end end end function gr_draw_markers( series::Series, x, y, clims, msize = series[:markersize], strokewidth = series[:markerstrokewidth], ) isempty(x) && return GR.setfillintstyle(GR.INTSTYLE_SOLID) shapes = series[:markershape] if shapes != :none for segment in series_segments(series, :scatter) i = segment.attr_index rng = intersect(eachindex(x), segment.range) if !isempty(rng) ms = get_thickness_scaling(series) * _cycle(msize, i) msw = get_thickness_scaling(series) * _cycle(strokewidth, i) shape = _cycle(shapes, i) for j in rng gr_draw_marker( series, _cycle(x, j), _cycle(y, j), clims, i, ms, msw, shape, ) end end end end end function gr_draw_shapes(series, clims) x, y = shape_data(series) for segment in series_segments(series, :shape) i, rng = segment.attr_index, segment.range if length(rng) > 1 # connect to the beginning rng = vcat(rng, rng[1]) # get the segments xseg, yseg = x[rng], y[rng] # draw the interior fc = get_fillcolor(series, clims, i) gr_set_fill(fc) fs = get_fillstyle(series, i) gr_set_fillstyle(fs) gr_set_transparency(fc, get_fillalpha(series, i)) GR.fillarea(xseg, yseg) # draw the shapes lc = get_linecolor(series, clims, i) gr_set_line(get_linewidth(series, i), get_linestyle(series, i), lc, series) gr_set_transparency(lc, get_linealpha(series, i)) GR.polyline(xseg, yseg) end end end function gr_draw_contour(series, x, y, z, clims) GR.setspace(clims[1], clims[2], 0, 90) gr_set_line(get_linewidth(series), get_linestyle(series), get_linecolor(series), series) gr_set_transparency(get_fillalpha(series)) is_lc_black = let black = plot_color(:black) plot_color(series[:linecolor]) in (black, [black]) end h = gr_contour_levels(series, clims) if series[:fillrange] !== nothing if series[:fillcolor] != series[:linecolor] && !is_lc_black @warn "GR: filled contour only supported with black contour lines" end GR.contourf(x, y, h, z, series[:contour_labels] == true ? 1 : 0) else coff = is_lc_black ? 0 : 1000 GR.contour(x, y, h, z, coff + (series[:contour_labels] == true ? 1 : 0)) end end function gr_draw_surface(series, x, y, z, clims) e_kwargs = series[:extra_kwargs] st = series[:seriestype] if st === :surface if ndims(x) == ndims(y) == ndims(z) == 2 GR.gr3.surface(x', y', z, GR.OPTION_3D_MESH) else fillalpha = get_fillalpha(series) fillcolor = get_fillcolor(series) # NOTE: setting nx = 0 or ny = 0 disables GR.gridit interpolation nx, ny = get(e_kwargs, :nx, 200), get(e_kwargs, :ny, 200) if length(x) == length(y) == length(z) && nx > 0 && ny > 0 x, y, z = GR.gridit(x, y, z, nx, ny) end d_opt = get(e_kwargs, :display_option, GR.OPTION_COLORED_MESH) if (!isnothing(fillalpha) && fillalpha < 1) || alpha(first(fillcolor)) < 1 gr_set_transparency(fillcolor, fillalpha) GR.surface(x, y, z, d_opt) else GR.gr3.surface(x, y, z, d_opt) end end elseif st === :wireframe GR.setfillcolorind(0) GR.surface(x, y, z, get(e_kwargs, :display_option, GR.OPTION_FILLED_MESH)) elseif st === :mesh3d if series[:connections] isa AbstractVector{<:AbstractVector{Int}} # Combination of any polygon types cns = [[length(polyinds), polyinds...] for polyinds in series[:connections]] elseif series[:connections] isa AbstractVector{NTuple{N,Int}} where {N} # Only N-gons - connections have to be 1-based (indexing) N = length(series[:connections][1]) cns = [[N, polyinds...] for polyinds in series[:connections]] elseif series[:connections] isa NTuple{3,<:AbstractVector{Int}} # Only triangles - connections have to be 0-based (indexing) ci, cj, ck = series[:connections] if !(length(ci) == length(cj) == length(ck)) throw( ArgumentError( "Argument connections must consist of equally sized arrays.", ), ) end cns = [([3, ci[i] + 1, cj[i] + 1, ck[i] + 1]) for i in eachindex(ci)] else throw( ArgumentError( "Unsupported `:connections` type $(typeof(series[:connections])) for seriestype=$st", ), ) end fillalpha = get_fillalpha(series) n_polygons = length(cns) facecolor = if series[:fillcolor] isa AbstractArray series[:fillcolor] else fill(series[:fillcolor], n_polygons) end facecolor = map(fc -> set_RGBA_alpha(fillalpha, fc), facecolor) GR.setborderwidth(get_linewidth(series)) GR.setbordercolorind(gr_getcolorind(get_linecolor(series))) GR.polygonmesh3d(x, y, z, vcat(cns...), signed.(gr_color.(facecolor))) else throw(ArgumentError("Not handled !")) end end function gr_draw_heatmap(series, x, y, z, clims) fillgrad = _as_gradient(series[:fillcolor]) GR.setspace(clims..., 0, 90) w, h = length(x) - 1, length(y) - 1 if !ispolar(series) && is_uniformly_spaced(x) && is_uniformly_spaced(y) # For uniformly spaced data use GR.drawimage, which can be # much faster than GR.nonuniformcellarray, especially for # 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, _z = if series[:subplot][:colorbar_scale] == :identity plot_color.(get(fillgrad, z, clims), series[:fillalpha]), z elseif series[:subplot][:colorbar_scale] == :log10 z_log = replace(x -> isinf(x) ? NaN : x, log10.(z)) z_normalized = get_z_normalized.(z_log, log10.(clims)...) plot_color.(map(z -> get(fillgrad, z), z_normalized), series[:fillalpha]), z_log end for i in eachindex(colors) if isnan(_z[i]) 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, _z = if series[:subplot][:colorbar_scale] == :identity get_z_normalized.(z, clims...), z elseif series[:subplot][:colorbar_scale] == :log10 z_log = replace(x -> isinf(x) ? NaN : x, log10.(z)) get_z_normalized.(z_log, log10.(clims)...), z_log end rgba = Int32[round(Int32, 1000 + _i * 255) for _i in z_normalized] background_color_ind = gr_getcolorind(plot_color(series[:subplot][:background_color_inside])) for i in eachindex(rgba) if isnan(_z[i]) rgba[i] = background_color_ind end end if !ispolar(series) GR.nonuniformcellarray(x, y, w, h, rgba) else if y[1] < 0 @warn "'y[1] < 0' (rmin) is not yet supported." end xmin, xmax, ymin, ymax = gr_xy_axislims(series[:subplot]) GR.setwindow(-ymax, ymax, -ymax, ymax) GR.nonuniformpolarcellarray(rad2deg.(x), y, w, h, rgba) end end end function gr_draw_image(series, x, y, z, clims) w, h = size(z) xmin, xmax = ignorenan_extrema(x) ymin, ymax = ignorenan_extrema(y) rgba = gr_color.(z) GR.drawimage(xmin, xmax, ymax, ymin, w, h, rgba) end # ---------------------------------------------------------------- for (mime, fmt) in ( "application/pdf" => "pdf", "image/png" => "png", "application/postscript" => "ps", "image/svg+xml" => "svg", ) @eval function _show(io::IO, ::MIME{Symbol($mime)}, plt::Plot{GRBackend}) ENV["GKS_ENCODING"] = "utf8" GR.emergencyclosegks() filepath = tempname() * "." * $fmt env = get(ENV, "GKSwstype", "0") ENV["GKSwstype"] = $fmt ENV["GKS_FILEPATH"] = filepath gr_display(plt, $fmt) GR.emergencyclosegks() write(io, read(filepath, String)) rm(filepath) if env != "0" ENV["GKSwstype"] = env else pop!(ENV, "GKSwstype") end end end function _display(plt::Plot{GRBackend}) ENV["GKS_ENCODING"] = "utf8" if plt[:display_type] == :inline GR.emergencyclosegks() filepath = tempname() * ".pdf" ENV["GKSwstype"] = "pdf" ENV["GKS_FILEPATH"] = filepath gr_display(plt) GR.emergencyclosegks() content = string( "\033]1337;File=inline=1;preserveAspectRatio=0:", base64encode(open(read, filepath)), "\a", ) println(content) rm(filepath) else ENV["GKS_DOUBLE_BUF"] = true gr_display(plt) end end closeall(::GRBackend) = GR.emergencyclosegks()