# https://github.com/jheinen/GR.jl # significant contributions by @jheinen import GR export GR # -------------------------------------------------------------------------------------- const gr_linetype = KW( :auto => 1, :solid => 1, :dash => 2, :dot => 3, :dashdot => 4, :dashdotdot => -1 ) const gr_markertype = KW( :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 ) const gr_halign = KW( :left => 1, :hcenter => 2, :right => 3 ) const gr_valign = KW( :top => 1, :vcenter => 3, :bottom => 5 ) const gr_font_family = Dict( "times" => 1, "helvetica" => 5, "courier" => 9, "bookman" => 14, "newcenturyschlbk" => 18, "avantgarde" => 22, "palatino" => 26 ) # -------------------------------------------------------------------------------------- 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_textcolor(c) = GR.settextcolorind(gr_getcolorind(_cycle(c,1))) gr_set_transparency(α::Real) = GR.settransparency(clamp(α, 0, 1)) function gr_set_transparency(::Nothing) 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) iend = 0 n = length(x) while iend < n-1 # set istart to the first index that is finite istart = -1 for j = 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 = 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.drawarrow(x[iend-1], y[iend-1], x[iend], y[iend]) end if arrowside in (:tail,:both) GR.drawarrow(x[istart+1], y[istart+1], x[istart], y[istart]) end else break end end end gr_inqtext(x, y, s::Symbol) = gr_inqtext(x, y, string(s)) function gr_inqtext(x, y, s) if length(s) >= 2 && s[1] == '$' && s[end] == '$' GR.inqmathtex(x, y, s[2:end-1]) elseif findfirst(isequal('\\'), s) != nothing || occursin("10^{", s) GR.inqtextext(x, y, s) else GR.inqtext(x, y, s) end end gr_text(x, y, s::Symbol) = gr_text(x, y, string(s)) function gr_text(x, y, s) if length(s) >= 2 && s[1] == '$' && s[end] == '$' GR.mathtex(x, y, s[2:end-1]) elseif findfirst(isequal('\\'), s) != nothing || occursin("10^{", s) 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(yaxis) if yaxis[:formatter] in (:scientific, :auto) && yaxis[:ticks] in (:auto, :native) rtick_labels = convert_sci_unicode.(rtick_labels) end #draw angular grid if xaxis[:grid] gr_set_line(xaxis[:gridlinewidth], xaxis[:gridstyle], xaxis[:foreground_color_grid]) gr_set_transparency(xaxis[:gridalpha]) for i in 1:length(α) 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]) gr_set_transparency(yaxis[:gridalpha]) for i in 1:length(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 1:length(α) 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 1:length(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[:xaxis]) gr_y_axislims(sp::Subplot) = axis_limits(sp[:yaxis]) gr_z_axislims(sp::Subplot) = axis_limits(sp[:zaxis]) gr_xy_axislims(sp::Subplot) = gr_x_axislims(sp)..., gr_y_axislims(sp)... function gr_lims(axis::Axis, adjust::Bool, expand = nothing) if expand != nothing expand_extrema!(axis, expand) end lims = axis_limits(axis) if adjust GR.adjustrange(lims...) else lims end end function gr_fill_viewport(vp::AVec{Float64}, c) GR.savestate() GR.selntran(0) GR.setfillintstyle(GR.INTSTYLE_SOLID) gr_set_fillcolor(c) GR.fillrect(vp...) GR.selntran(1) GR.restorestate() end normalize_zvals(args...) = nothing function normalize_zvals(zv::AVec, clims::NTuple{2, <:Real}) vmin, vmax = ignorenan_extrema(zv) isfinite(clims[1]) && (vmin = clims[1]) isfinite(clims[2]) && (vmax = clims[2]) if vmin == vmax zeros(length(zv)) else clamp.((zv .- vmin) ./ (vmax .- vmin), 0, 1) end end # --------------------------------------------------------- # draw ONE Shape function gr_draw_marker(xi, yi, msize, shape::Shape) sx, sy = coords(shape) # convert to ndc coords (percentages of window) GR.selntran(0) w, h = gr_plot_size f = msize / (w + h) xi, yi = GR.wctondc(xi, yi) GR.fillarea(xi .+ sx .* f, yi .+ sy .* f) GR.selntran(1) end # draw ONE symbol marker function gr_draw_marker(xi, yi, msize::Number, shape::Symbol) GR.setmarkertype(gr_markertype[shape]) w, h = gr_plot_size GR.setmarkersize(0.3msize / ((w + h) * 0.001)) GR.polymarker([xi], [yi]) end # draw the markers, one at a time function gr_draw_markers(series::Series, x, y, clims, msize = series[:markersize]) isempty(x) && return GR.setfillintstyle(GR.INTSTYLE_SOLID) shapes = series[:markershape] if shapes != :none for i=1:length(x) msi = _cycle(msize, i) shape = _cycle(shapes, i) cfunc = isa(shape, Shape) ? gr_set_fillcolor : gr_set_markercolor # draw a filled in shape, slightly bigger, to estimate a stroke if series[:markerstrokewidth] > 0 cfunc(get_markerstrokecolor(series, i)) gr_set_transparency(get_markerstrokealpha(series, i)) gr_draw_marker(x[i], y[i], msi + series[:markerstrokewidth], shape) end # draw the shape - don't draw filled area if marker shape is 1D if !(shape in (:hline, :vline, :+, :x)) cfunc(get_markercolor(series, clims, i)) gr_set_transparency(get_markeralpha(series, i)) gr_draw_marker(x[i], y[i], msi, shape) end end end end # --------------------------------------------------------- function gr_set_line(lw, style, c) #, a) GR.setlinetype(gr_linetype[style]) w, h = gr_plot_size GR.setlinewidth(_gr_thickness_scaling[1] * max(0, lw / ((w + h) * 0.001))) gr_set_linecolor(c) #, a) 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) const _gr_point_mult = 0.0018 * ones(1) const _gr_thickness_scaling = ones(1) # set the font attributes... assumes _gr_point_mult has been populated already function gr_set_font(f::Font; halign = f.halign, valign = f.valign, color = f.color, rotation = f.rotation) family = lowercase(f.family) GR.setcharheight(_gr_point_mult[1] * f.pointsize) GR.setcharup(sind(-rotation), cosd(-rotation)) if haskey(gr_font_family, family) GR.settextfontprec(100 + gr_font_family[family], GR.TEXT_PRECISION_STRING) end gr_set_textcolor(color) GR.settextalign(gr_halign[halign], gr_valign[valign]) end function gr_nans_to_infs!(z) for (i,zi) in enumerate(z) if zi == NaN z[i] = Inf end end end # -------------------------------------------------------------------------------------- # viewport plot area # this stays constant for a given subplot while displaying that subplot. # values are [xmin, xmax, ymin, ymax]. they range [0,1]. const viewport_plotarea = zeros(4) # the size of the current plot in pixels const gr_plot_size = [600.0, 400.0] const gr_colorbar_ratio = 0.1 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 is3d(sp) vp = viewport[:] extent = min(vp[2] - vp[1], vp[4] - vp[3]) viewport[1] = 0.5 * (vp[1] + vp[2] - extent) viewport[2] = 0.5 * (vp[1] + vp[2] + extent) viewport[3] = 0.5 * (vp[3] + vp[4] - extent) viewport[4] = 0.5 * (vp[3] + vp[4] + extent) end if hascolorbar(sp) viewport[2] -= gr_colorbar_ratio end viewport end # change so we're focused on the viewport area function gr_set_viewport_cmap(sp::Subplot) GR.setviewport( viewport_plotarea[2] + (is3d(sp) ? 0.07 : 0.02), viewport_plotarea[2] + (is3d(sp) ? 0.10 : 0.05), viewport_plotarea[3], viewport_plotarea[4] ) end # reset the viewport to the plot area function gr_set_viewport_plotarea() GR.setviewport( viewport_plotarea[1], viewport_plotarea[2], viewport_plotarea[3], viewport_plotarea[4] ) end function gr_set_viewport_polar() 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 # add the colorbar function gr_colorbar(sp::Subplot, clims) xmin, xmax = gr_xy_axislims(sp)[1:2] gr_set_viewport_cmap(sp) l = zeros(Int32, 1, 256) l[1,:] = Int[round(Int, _i) for _i in range(1000, stop=1255, length=256)] GR.setscale(0) GR.setwindow(xmin, xmax, clims[1], clims[2]) GR.cellarray(xmin, xmax, clims[2], clims[1], 1, length(l), l) ztick = 0.5 * GR.tick(clims[1], clims[2]) GR.axes(0, ztick, xmax, clims[1], 0, 1, 0.005) gr_set_font(guidefont(sp[:yaxis])) GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_TOP) GR.setcharup(-1, 0) gr_text(viewport_plotarea[2] + gr_colorbar_ratio, gr_view_ycenter(), sp[:colorbar_title]) gr_set_viewport_plotarea() end gr_view_xcenter() = 0.5 * (viewport_plotarea[1] + viewport_plotarea[2]) gr_view_ycenter() = 0.5 * (viewport_plotarea[3] + viewport_plotarea[4]) function gr_legend_pos(s::Symbol,w,h) str = string(s) if str == "best" str = "topright" end if occursin("right", str) xpos = viewport_plotarea[2] - 0.05 - w elseif occursin("left", str) xpos = viewport_plotarea[1] + 0.11 else xpos = (viewport_plotarea[2]-viewport_plotarea[1])/2 - w/2 +.04 end if occursin("top", str) ypos = viewport_plotarea[4] - 0.06 elseif occursin("bottom", str) ypos = viewport_plotarea[3] + h + 0.06 else ypos = (viewport_plotarea[4]-viewport_plotarea[3])/2 + h/2 end (xpos,ypos) end function gr_legend_pos(v::Tuple{S,T},w,h) 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 # -------------------------------------------------------------------------------------- const _gr_gradient_alpha = ones(256) function gr_set_gradient(c) grad = isa(c, ColorGradient) ? c : cgrad() for (i,z) in enumerate(range(0, stop=1, length=256)) c = grad[z] GR.setcolorrep(999+i, red(c), green(c), blue(c)) _gr_gradient_alpha[i] = alpha(c) end grad 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() _gr_thickness_scaling[1] = plt[:thickness_scaling] dpi_factor = plt[:dpi] / Plots.DPI if fmt == "svg" dpi_factor *= 4 end # 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 = plt[:size] gr_plot_size[:] = [w, h] 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]) # update point mult px_per_pt = px / pt _gr_point_mult[1] = 1.5 * _gr_thickness_scaling[1] * px_per_pt / max(h,w) # subplots: for sp in plt.subplots gr_display(sp, w*px, h*px, viewport_canvas) end GR.updatews() end function gr_set_xticks_font(sp) flip = sp[:yaxis][:flip] mirror = sp[:xaxis][:mirror] gr_set_font(tickfont(sp[:xaxis]), halign = (:left, :hcenter, :right)[sign(sp[:xaxis][:rotation]) + 2], valign = (mirror ? :bottom : :top), rotation = sp[:xaxis][:rotation]) return flip, mirror end function gr_set_yticks_font(sp) flip = sp[:xaxis][:flip] mirror = sp[:yaxis][:mirror] gr_set_font(tickfont(sp[:yaxis]), halign = (mirror ? :left : :right), valign = (:top, :vcenter, :bottom)[sign(sp[:yaxis][:rotation]) + 2], rotation = sp[:yaxis][:rotation]) return flip, mirror end function gr_get_ticks_size(ticks, i) GR.savestate() GR.selntran(0) l = 0.0 for (cv, dv) in zip(ticks...) tb = gr_inqtext(0, 0, string(dv))[i] tb_min, tb_max = extrema(tb) l = max(l, tb_max - tb_min) end GR.restorestate() return l end function _update_min_padding!(sp::Subplot{GRBackend}) dpi = sp.plt[:thickness_scaling] if !haskey(ENV, "GKSwstype") if isijulia() ENV["GKSwstype"] = "svg" end end # Add margin given by the user leftpad = 4mm + sp[:left_margin] toppad = 2mm + sp[:top_margin] rightpad = 4mm + sp[:right_margin] bottompad = 2mm + sp[:bottom_margin] # Add margin for title if sp[:title] != "" toppad += 5mm end # Add margin for x and y ticks xticks, yticks = axis_drawing_info(sp)[1:2] if !(xticks in (nothing, false, :none)) flip, mirror = gr_set_xticks_font(sp) l = gr_get_ticks_size(xticks, 2) if mirror toppad += 1mm + gr_plot_size[2] * l * px else bottompad += 1mm + gr_plot_size[2] * l * px end end if !(yticks in (nothing, false, :none)) flip, mirror = gr_set_yticks_font(sp) l = gr_get_ticks_size(yticks, 1) if mirror rightpad += 1mm + gr_plot_size[1] * l * px else leftpad += 1mm + gr_plot_size[1] * l * px end end # Add margin for x label if sp[:xaxis][:guide] != "" bottompad += 4mm end # Add margin for y label if sp[:yaxis][:guide] != "" leftpad += 4mm end if sp[:colorbar_title] != "" rightpad += 4mm end sp.minpad = Tuple(dpi * [leftpad, toppad, rightpad, bottompad]) 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) # get data limits data_lims = gr_xy_axislims(sp) xy_lims = data_lims ratio = sp[:aspect_ratio] if ratio != :none if ratio == :equal ratio = 1 end viewport_ratio = (viewport_plotarea[2] - viewport_plotarea[1]) / (viewport_plotarea[4] - viewport_plotarea[3]) window_ratio = (data_lims[2] - data_lims[1]) / (data_lims[4] - data_lims[3]) / 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 # fill in the plot area background bg = plot_color(sp[:background_color_inside]) gr_fill_viewport(viewport_plotarea, bg) # reduced from before... set some flags based on the series in this subplot # TODO: can these be generic flags? outside_ticks = false cmap = hascolorbar(sp) draw_axes = sp[:framestyle] != :none # axes_2d = true for series in series_list(sp) st = series[:seriestype] if st == :pie draw_axes = false end if st == :heatmap outside_ticks = true for ax in (sp[:xaxis], sp[:yaxis]) v = series[ax[:letter]] if diff(collect(extrema(diff(v))))[1] > 1e-6*std(v) @warn("GR: heatmap only supported with equally spaced data.") end end x, y = heatmap_edges(series[:x], sp[:xaxis][:scale]), heatmap_edges(series[:y], sp[:yaxis][:scale]) xy_lims = x[1], x[end], y[1], y[end] expand_extrema!(sp[:xaxis], x) expand_extrema!(sp[:yaxis], y) data_lims = gr_xy_axislims(sp) end end # set our plot area view gr_set_viewport_plotarea() # these are the Axis objects, which hold scale, lims, etc xaxis = sp[:xaxis] yaxis = sp[:yaxis] zaxis = sp[:zaxis] # set the scale flags and window xmin, xmax, ymin, ymax = data_lims scaleop = 0 xtick, ytick = 1, 1 if xmax > xmin && ymax > ymin # NOTE: for log axes, the major_x and major_y - if non-zero (omit labels) - control the minor grid lines (1 = draw 9 minor grid lines, 2 = no minor grid lines) # NOTE: for log axes, the x_tick and y_tick - if non-zero (omit axes) - only affect the output appearance (1 = nomal, 2 = scientiic notation) xaxis[:scale] == :log10 && (scaleop |= GR.OPTION_X_LOG) yaxis[:scale] == :log10 && (scaleop |= GR.OPTION_Y_LOG) xaxis[:flip] && (scaleop |= GR.OPTION_FLIP_X) yaxis[:flip] && (scaleop |= GR.OPTION_FLIP_Y) if scaleop & GR.OPTION_X_LOG == 0 majorx = 1 #5 xtick = GR.tick(xmin, xmax) / majorx else # log axis xtick = 2 # scientific notation majorx = 2 # no minor grid lines end if scaleop & GR.OPTION_Y_LOG == 0 majory = 1 #5 ytick = GR.tick(ymin, ymax) / majory else # log axis ytick = 2 # scientific notation majory = 2 # no minor grid lines end # NOTE: setwindow sets the "data coordinate" limits of the current "viewport" GR.setwindow(xmin, xmax, ymin, ymax) GR.setscale(scaleop) end # draw the axes gr_set_font(tickfont(xaxis)) GR.setlinewidth(sp.plt[:thickness_scaling]) if is3d(sp) zmin, zmax = gr_lims(zaxis, true) clims = sp[:clims] if is_2tuple(clims) isfinite(clims[1]) && (zmin = clims[1]) isfinite(clims[2]) && (zmax = clims[2]) end GR.setspace(zmin, zmax, round.(Int, sp[:camera])...) xtick = GR.tick(xmin, xmax) / 2 ytick = GR.tick(ymin, ymax) / 2 ztick = GR.tick(zmin, zmax) / 2 ticksize = 0.01 * (viewport_plotarea[2] - viewport_plotarea[1]) if xaxis[:grid] gr_set_line(xaxis[:gridlinewidth], xaxis[:gridstyle], xaxis[:foreground_color_grid]) gr_set_transparency(xaxis[:gridalpha]) GR.grid3d(xtick, 0, 0, xmin, ymax, zmin, 2, 0, 0) end if yaxis[:grid] gr_set_line(yaxis[:gridlinewidth], yaxis[:gridstyle], yaxis[:foreground_color_grid]) gr_set_transparency(yaxis[:gridalpha]) GR.grid3d(0, ytick, 0, xmin, ymax, zmin, 0, 2, 0) end if zaxis[:grid] gr_set_line(zaxis[:gridlinewidth], zaxis[:gridstyle], zaxis[:foreground_color_grid]) gr_set_transparency(zaxis[:gridalpha]) GR.grid3d(0, 0, ztick, xmin, ymax, zmin, 0, 0, 2) end gr_set_line(1, :solid, xaxis[:foreground_color_axis]) gr_set_transparency(1) GR.axes3d(xtick, 0, ztick, xmin, ymin, zmin, 2, 0, 2, -ticksize) GR.axes3d(0, ytick, 0, xmax, ymin, zmin, 0, 2, 0, ticksize) elseif ispolar(sp) r = gr_set_viewport_polar() #rmin, rmax = GR.adjustrange(ignorenan_minimum(r), ignorenan_maximum(r)) rmin, rmax = axis_limits(sp[:yaxis]) gr_polaraxes(rmin, rmax, sp) elseif draw_axes if xmax > xmin && ymax > ymin GR.setwindow(xmin, xmax, ymin, ymax) end xticks, yticks, xspine_segs, yspine_segs, xtick_segs, ytick_segs, xgrid_segs, ygrid_segs, xminorgrid_segs, yminorgrid_segs, xborder_segs, yborder_segs = axis_drawing_info(sp) # @show xticks yticks #spine_segs grid_segs # draw the grid lines if xaxis[:grid] # gr_set_linecolor(sp[:foreground_color_grid]) # GR.grid(xtick, ytick, 0, 0, majorx, majory) gr_set_line(xaxis[:gridlinewidth], xaxis[:gridstyle], xaxis[:foreground_color_grid]) gr_set_transparency(xaxis[:gridalpha]) gr_polyline(coords(xgrid_segs)...) end if yaxis[:grid] gr_set_line(yaxis[:gridlinewidth], yaxis[:gridstyle], yaxis[:foreground_color_grid]) gr_set_transparency(yaxis[:gridalpha]) gr_polyline(coords(ygrid_segs)...) end if xaxis[:minorgrid] # gr_set_linecolor(sp[:foreground_color_grid]) # GR.grid(xtick, ytick, 0, 0, majorx, majory) gr_set_line(xaxis[:minorgridlinewidth], xaxis[:minorgridstyle], xaxis[:foreground_color_minor_grid]) gr_set_transparency(xaxis[:minorgridalpha]) gr_polyline(coords(xminorgrid_segs)...) end if yaxis[:minorgrid] gr_set_line(yaxis[:minorgridlinewidth], yaxis[:minorgridstyle], yaxis[:foreground_color_minor_grid]) gr_set_transparency(yaxis[:minorgridalpha]) gr_polyline(coords(yminorgrid_segs)...) end gr_set_transparency(1.0) # axis lines if xaxis[:showaxis] gr_set_line(1, :solid, xaxis[:foreground_color_axis]) GR.setclip(0) gr_polyline(coords(xspine_segs)...) end if yaxis[:showaxis] gr_set_line(1, :solid, yaxis[:foreground_color_axis]) GR.setclip(0) gr_polyline(coords(yspine_segs)...) end GR.setclip(1) # axis ticks if xaxis[:showaxis] if sp[:framestyle] in (:zerolines, :grid) gr_set_line(1, :solid, xaxis[:foreground_color_grid]) gr_set_transparency(xaxis[:gridalpha]) else gr_set_line(1, :solid, xaxis[:foreground_color_axis]) end GR.setclip(0) gr_polyline(coords(xtick_segs)...) end if yaxis[:showaxis] if sp[:framestyle] in (:zerolines, :grid) gr_set_line(1, :solid, yaxis[:foreground_color_grid]) gr_set_transparency(yaxis[:gridalpha]) else gr_set_line(1, :solid, yaxis[:foreground_color_axis]) end GR.setclip(0) gr_polyline(coords(ytick_segs)...) end GR.setclip(1) # tick marks if !(xticks in (:none, nothing, false)) && xaxis[:showaxis] # x labels flip, mirror = gr_set_xticks_font(sp) for (cv, dv) in zip(xticks...) # use xor ($) to get the right y coords xi, yi = GR.wctondc(cv, sp[:framestyle] == :origin ? 0 : xor(flip, mirror) ? ymax : ymin) # @show cv dv ymin xi yi flip mirror (flip $ mirror) if xaxis[:ticks] in (:auto, :native) # ensure correct dispatch in gr_text for automatic log ticks if xaxis[:scale] in _logScales dv = string(dv, "\\ ") elseif xaxis[:formatter] in (:scientific, :auto) dv = convert_sci_unicode(dv) end end gr_text(xi, yi + (mirror ? 1 : -1) * 5e-3 * (xaxis[:tick_direction] == :out ? 1.5 : 1.0), string(dv)) end end if !(yticks in (:none, nothing, false)) && yaxis[:showaxis] # y labels flip, mirror = gr_set_yticks_font(sp) for (cv, dv) in zip(yticks...) # use xor ($) to get the right y coords xi, yi = GR.wctondc(sp[:framestyle] == :origin ? 0 : xor(flip, mirror) ? xmax : xmin, cv) # @show cv dv xmin xi yi if yaxis[:ticks] in (:auto, :native) # ensure correct dispatch in gr_text for automatic log ticks if yaxis[:scale] in _logScales dv = string(dv, "\\ ") elseif yaxis[:formatter] in (:scientific, :auto) dv = convert_sci_unicode(dv) end end gr_text(xi + (mirror ? 1 : -1) * 1e-2 * (yaxis[:tick_direction] == :out ? 1.5 : 1.0), yi, string(dv)) end end # border intensity = sp[:framestyle] == :semi ? 0.5 : 1.0 if sp[:framestyle] in (:box, :semi) gr_set_line(intensity, :solid, xaxis[:foreground_color_border]) gr_set_transparency(intensity) gr_polyline(coords(xborder_segs)...) gr_set_line(intensity, :solid, yaxis[:foreground_color_border]) gr_set_transparency(intensity) gr_polyline(coords(yborder_segs)...) end end # end # add the guides GR.savestate() if sp[:title] != "" gr_set_font(titlefont(sp)) loc = sp[:title_location] 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() halign = GR.TEXT_HALIGN_CENTER end GR.settextalign(halign, GR.TEXT_VALIGN_TOP) gr_text(xpos, viewport_subplot[4], sp[:title]) end if xaxis[:guide] != "" gr_set_font(guidefont(xaxis)) if xaxis[:guide_position] == :top GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_TOP) gr_text(gr_view_xcenter(), viewport_subplot[4], xaxis[:guide]) else GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_BOTTOM) gr_text(gr_view_xcenter(), viewport_subplot[3], xaxis[:guide]) end end if yaxis[:guide] != "" gr_set_font(guidefont(yaxis)) GR.setcharup(-1, 0) if yaxis[:guide_position] == :left GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_BOTTOM) gr_text(viewport_subplot[2], gr_view_ycenter(), yaxis[:guide]) else GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_TOP) gr_text(viewport_subplot[1], gr_view_ycenter(), yaxis[:guide]) end end GR.restorestate() gr_set_font(tickfont(xaxis)) # this needs to be here to point the colormap to the right indices GR.setcolormap(1000 + GR.COLORMAP_COOLWARM) # calculate the colorbar limits once for a subplot clims = get_clims(sp) for (idx, series) in enumerate(series_list(sp)) st = series[:seriestype] # update the current stored gradient if st in (:contour, :surface, :wireframe, :heatmap) gr_set_gradient(series[:fillcolor]) #, series[:fillalpha]) elseif series[:marker_z] != nothing series[:markercolor] = gr_set_gradient(series[:markercolor]) elseif series[:line_z] != nothing series[:linecolor] = gr_set_gradient(series[:linecolor]) elseif series[:fill_z] != nothing series[:fillcolor] = gr_set_gradient(series[:fillcolor]) end GR.savestate() # update the bounding window if ispolar(sp) gr_set_viewport_polar() else xmin, xmax, ymin, ymax = data_lims if xmax > xmin && ymax > ymin GR.setwindow(xmin, xmax, ymin, ymax) end end x, y, z = series[:x], series[:y], series[:z] frng = series[:fillrange] # add custom frame shapes to markershape? series_annotations_shapes!(series) # ------------------------------------------------------- # recompute data if typeof(z) <: Surface z = vec(transpose_z(series, z.surf, false)) elseif ispolar(sp) if frng != nothing _, frng = convert_to_polar(x, frng, (rmin, rmax)) end x, y = convert_to_polar(x, y, (rmin, rmax)) end if st == :straightline x, y = straightline_data(series) end if st in (:path, :scatter, :straightline) if length(x) > 1 lz = series[:line_z] segments = iter_segments(series) # do area fill if frng != nothing GR.setfillintstyle(GR.INTSTYLE_SOLID) fr_from, fr_to = (is_2tuple(frng) ? frng : (y, frng)) for (i, rng) in enumerate(segments) gr_set_fillcolor(get_fillcolor(series, clims, i)) fx = _cycle(x, vcat(rng, reverse(rng))) fy = vcat(_cycle(fr_from,rng), _cycle(fr_to,reverse(rng))) gr_set_transparency(get_fillalpha(series, i)) GR.fillarea(fx, fy) end end # draw the line(s) if st in (:path, :straightline) for (i, rng) in enumerate(segments) gr_set_line(get_linewidth(series, i), get_linestyle(series, i), get_linecolor(series, clims, i)) #, series[:linealpha]) gr_set_transparency(get_linealpha(series, i)) arrowside = isa(series[:arrow], Arrow) ? series[:arrow].side : :none gr_polyline(x[rng], y[rng]; arrowside = arrowside) end end end if series[:markershape] != :none gr_draw_markers(series, x, y, clims) end elseif st == :contour zmin, zmax = clims GR.setspace(zmin, zmax, 0, 90) if typeof(series[:levels]) <: AbstractArray h = series[:levels] else h = series[:levels] > 1 ? range(zmin, stop=zmax, length=series[:levels]) : [(zmin + zmax) / 2] end if series[:fillrange] != nothing GR.surface(x, y, z, GR.OPTION_CELL_ARRAY) else GR.setlinetype(gr_linetype[get_linestyle(series)]) GR.setlinewidth(max(0, get_linewidth(series) / (sum(gr_plot_size) * 0.001))) if plot_color(series[:linecolor]) == [plot_color(:black)] GR.contour(x, y, h, z, 0 + (series[:contour_labels] == true ? 1 : 0)) else GR.contour(x, y, h, z, 1000 + (series[:contour_labels] == true ? 1 : 0)) end end # create the colorbar of contour levels if cmap gr_set_line(1, :solid, yaxis[:foreground_color_axis]) gr_set_viewport_cmap(sp) l = (length(h) > 1) ? round.(Int32, 1000 .+ (h .- ignorenan_minimum(h)) ./ (ignorenan_maximum(h) - ignorenan_minimum(h)) .* 255) : Int32[1000, 1255] GR.setwindow(xmin, xmax, zmin, zmax) GR.cellarray(xmin, xmax, zmax, zmin, 1, length(l), l) ztick = 0.5 * GR.tick(zmin, zmax) GR.axes(0, ztick, xmax, zmin, 0, 1, 0.005) gr_set_viewport_plotarea() end elseif st in [:surface, :wireframe] if st == :surface if length(x) == length(y) == length(z) GR.trisurface(x, y, z) else GR.gr3.surface(x, y, z, GR.OPTION_COLORED_MESH) end else GR.setfillcolorind(0) GR.surface(x, y, z, GR.OPTION_FILLED_MESH) end elseif st == :heatmap xmin, xmax, ymin, ymax = xy_lims zmin, zmax = clims m, n = length(x), length(y) xinds = sort(1:m, rev = xaxis[:flip]) yinds = sort(1:n, rev = yaxis[:flip]) z = reshape(reshape(z, m, n)[xinds, yinds], m*n) GR.setspace(zmin, zmax, 0, 90) grad = isa(series[:fillcolor], ColorGradient) ? series[:fillcolor] : cgrad() colors = [plot_color(grad[clamp((zi-zmin) / (zmax-zmin), 0, 1)], series[:fillalpha]) for zi=z] rgba = map(c -> UInt32( round(UInt, alpha(c) * 255) << 24 + round(UInt, blue(c) * 255) << 16 + round(UInt, green(c) * 255) << 8 + round(UInt, red(c) * 255) ), colors) w, h = length(x), length(y) GR.drawimage(xmin, xmax, ymax, ymin, w, h, rgba) elseif st in (:path3d, :scatter3d) # draw path if st == :path3d if length(x) > 1 lz = series[:line_z] segments = iter_segments(series) for (i, rng) in enumerate(segments) gr_set_line(get_linewidth(series, i), get_linestyle(series, i), get_linecolor(series, clims, i)) #, series[:linealpha]) gr_set_transparency(get_linealpha(series, i)) GR.polyline3d(x[rng], y[rng], z[rng]) end end end # draw markers if st == :scatter3d || series[:markershape] != :none x2, y2 = unzip(map(GR.wc3towc, x, y, z)) gr_draw_markers(series, x2, y2, clims) end # TODO: replace with pie recipe elseif st == :pie GR.selntran(0) GR.setfillintstyle(GR.INTSTYLE_SOLID) xmin, xmax, ymin, ymax = viewport_plotarea ymax -= 0.1 * (xmax - xmin) xcenter = 0.5 * (xmin + xmax) ycenter = 0.5 * (ymin + ymax) if xmax - xmin > ymax - ymin r = 0.5 * (ymax - ymin) xmin, xmax = xcenter - r, xcenter + r else r = 0.5 * (xmax - xmin) ymin, ymax = ycenter - r, ycenter + r end labels = pie_labels(sp, series) slices = series[:y] numslices = length(slices) total = sum(slices) a1 = 0 x = zeros(3) y = zeros(3) for i in 1:numslices a2 = round(Int, a1 + (slices[i] / total) * 360.0) GR.setfillcolorind(980 + (i-1) % 20) GR.fillarc(xmin, xmax, ymin, ymax, a1, a2) α = 0.5 * (a1 + a2) cosf = r * cos(α * pi / 180) sinf = r * sin(α * pi / 180) x[1] = xcenter + cosf y[1] = ycenter + sinf x[2] = x[1] + 0.1 * cosf y[2] = y[1] + 0.1 * sinf y[3] = y[2] if 90 <= α < 270 x[3] = x[2] - 0.05 GR.settextalign(GR.TEXT_HALIGN_RIGHT, GR.TEXT_VALIGN_HALF) gr_text(x[3] - 0.01, y[3], string(labels[i])) else x[3] = x[2] + 0.05 GR.settextalign(GR.TEXT_HALIGN_LEFT, GR.TEXT_VALIGN_HALF) gr_text(x[3] + 0.01, y[3], string(labels[i])) end gr_polyline(x, y) a1 = a2 end GR.selntran(1) elseif st == :shape x, y = shape_data(series) for (i,rng) in enumerate(iter_segments(x, y)) 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 gr_set_fill(get_fillcolor(series, clims, i)) gr_set_transparency(get_fillalpha(series, i)) GR.fillarea(xseg, yseg) # draw the shapes gr_set_line(get_linewidth(series, i), get_linestyle(series, i), get_linecolor(series, clims, i)) gr_set_transparency(get_linealpha(series, i)) GR.polyline(xseg, yseg) end end elseif st == :image z = transpose_z(series, series[:z].surf, true)' w, h = length(x), length(y) xinds = sort(1:w, rev = xaxis[:flip]) yinds = sort(1:h, rev = yaxis[:flip]) z = z[xinds, yinds] xmin, xmax = ignorenan_extrema(series[:x]); ymin, ymax = ignorenan_extrema(series[:y]) if eltype(z) <: Colors.AbstractGray grey = round.(UInt8, float(z) * 255) rgba = map(c -> UInt32( 0xff000000 + UInt(c)<<16 + UInt(c)<<8 + UInt(c) ), grey) else rgba = map(c -> UInt32( round(UInt, alpha(c) * 255) << 24 + round(UInt, blue(c) * 255) << 16 + round(UInt, green(c) * 255) << 8 + round(UInt, red(c) * 255) ), z) end GR.drawimage(xmin, xmax, ymax, ymin, w, h, rgba) 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) gr_text(GR.wctondc(xi, yi)..., str) end GR.restorestate() end # draw the colorbar GR.savestate() # special colorbar with steps is drawn for contours if cmap && any(series[:seriestype] != :contour for series in series_list(sp)) gr_set_line(1, :solid, yaxis[:foreground_color_axis]) gr_set_transparency(1) gr_colorbar(sp, clims) end GR.restorestate() # add the legend if sp[:legend] != :none GR.savestate() GR.selntran(0) GR.setscale(0) gr_set_font(legendfont(sp)) w = 0 i = 0 n = 0 if sp[:legendtitle] != nothing tbx, tby = gr_inqtext(0, 0, string(sp[:legendtitle])) w = tbx[3] - tbx[1] n += 1 end for series in series_list(sp) should_add_to_legend(series) || continue n += 1 if typeof(series[:label]) <: Array i += 1 lab = series[:label][i] else lab = series[:label] end tbx, tby = gr_inqtext(0, 0, string(lab)) w = max(w, tbx[3] - tbx[1]) end if w > 0 dy = _gr_point_mult[1] * sp[:legendfontsize] * 1.75 h = dy*n (xpos,ypos) = gr_legend_pos(sp[:legend],w,h) GR.setfillintstyle(GR.INTSTYLE_SOLID) gr_set_fillcolor(sp[:background_color_legend]) GR.fillrect(xpos - 0.08, xpos + w + 0.02, ypos + dy, ypos - dy * n) gr_set_line(1, :solid, sp[:foreground_color_legend]) GR.drawrect(xpos - 0.08, xpos + w + 0.02, ypos + dy, ypos - dy * n) i = 0 if sp[:legendtitle] != nothing GR.settextalign(GR.TEXT_HALIGN_CENTER, GR.TEXT_VALIGN_HALF) gr_set_textcolor(sp[:legendfontcolor]) gr_set_transparency(1) gr_text(xpos - 0.03 + 0.5*w, ypos, string(sp[:legendtitle])) ypos -= dy end for series in series_list(sp) should_add_to_legend(series) || continue st = series[:seriestype] gr_set_line(get_linewidth(series), get_linestyle(series), get_linecolor(series, clims)) #, series[:linealpha]) if (st == :shape || series[:fillrange] != nothing) && series[:ribbon] == nothing gr_set_fill(get_fillcolor(series, clims)) #, series[:fillalpha]) l, r = xpos-0.07, xpos-0.01 b, t = ypos-0.4dy, ypos+0.4dy x = [l, r, r, l, l] y = [b, b, t, t, b] gr_set_transparency(get_fillalpha(series)) gr_polyline(x, y, GR.fillarea) gr_set_transparency(get_linealpha(series)) gr_set_line(get_linewidth(series), get_linestyle(series), get_linecolor(series, clims)) st == :shape && gr_polyline(x, y) end if st in (:path, :straightline) gr_set_transparency(get_linealpha(series)) if series[:fillrange] == nothing || series[:ribbon] != nothing GR.polyline([xpos - 0.07, xpos - 0.01], [ypos, ypos]) else GR.polyline([xpos - 0.07, xpos - 0.01], [ypos+0.4dy, ypos+0.4dy]) end end if series[:markershape] != :none gr_draw_markers(series, xpos - .035, ypos, clims, 6) end if typeof(series[:label]) <: Array i += 1 lab = series[:label][i] else lab = series[:label] end GR.settextalign(GR.TEXT_HALIGN_LEFT, GR.TEXT_VALIGN_HALF) gr_set_textcolor(sp[:legendfontcolor]) gr_text(xpos, ypos, string(lab)) ypos -= dy end end GR.selntran(1) GR.restorestate() end # add annotations GR.savestate() # update the bounding window if ispolar(sp) gr_set_viewport_polar() else xmin, xmax, ymin, ymax = data_lims if xmax > xmin && ymax > ymin GR.setwindow(xmin, xmax, ymin, ymax) end end for ann in sp[:annotations] x, y, val = locate_annotation(sp, ann...) x, y = if is3d(sp) # GR.wc3towc(x, y, z) else GR.wctondc(x, y) end gr_set_font(val.font) gr_text(x, y, val.str) end GR.restorestate() end # ---------------------------------------------------------------- const _gr_mimeformats = Dict( "application/pdf" => "pdf", "image/png" => "png", "application/postscript" => "ps", "image/svg+xml" => "svg", ) const _gr_wstype_default = @static if Sys.islinux() "x11" # "cairox11" elseif Sys.isapple() "quartz" else "use_default" end const _gr_wstype = Ref(get(ENV, "GKSwstype", _gr_wstype_default)) gr_set_output(wstype::String) = (_gr_wstype[] = wstype) for (mime, fmt) in _gr_mimeformats @eval function _show(io::IO, ::MIME{Symbol($mime)}, plt::Plot{GRBackend}) 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}) 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 if _gr_wstype[] != "use_default" ENV["GKSwstype"] = _gr_wstype[] end gr_display(plt) end end closeall(::GRBackend) = GR.emergencyclosegks()