# https://github.com/Evizero/UnicodePlots.jl supportedArgs(::UnicodePlotsBackend) = [ # :annotations, # :args, # :axis, # :background_color, # :linecolor, # :fill, # :foreground_color, :group, # :heatmap_c, # :kwargs, :label, # :layout, :legend, :seriescolor, :seriesalpha, :linestyle, :seriestype, # :linewidth, :markershape, # :markercolor, # :markersize, # :markerstrokewidth, # :markerstrokecolor, # :markerstrokestyle, # :n, :bins, # :nc, # :nr, # :pos, # :reg, # :ribbon, :show, :size, :title, :window_title, :x, :xguide, :xlims, # :xticks, :y, :yguide, :ylims, # :yrightlabel, # :yticks, # :xscale, # :yscale, # :xflip, # :yflip, # :z, ] supportedAxes(::UnicodePlotsBackend) = [:auto, :left] supportedTypes(::UnicodePlotsBackend) = [:none, :line, :path, :steppre, :steppost, :sticks, :scatter, :hist2d, :hexbin, :hist, :bar, :hline, :vline] supportedStyles(::UnicodePlotsBackend) = [:auto, :solid] supportedMarkers(::UnicodePlotsBackend) = [:none, :auto, :ellipse] supportedScales(::UnicodePlotsBackend) = [:identity] subplotSupported(::UnicodePlotsBackend) = true # -------------------------------------------------------------------------------------- function _initialize_backend(::UnicodePlotsBackend; kw...) @eval begin import UnicodePlots export UnicodePlots end end # ------------------------------- # do all the magic here... build it all at once, since we need to know about all the series at the very beginning function rebuildUnicodePlot!(plt::Plot) # figure out the plotting area xlim = [xmin, xmax] and ylim = [ymin, ymax] sargs = plt.seriesargs iargs = plt.attr # get the x/y limits if get(iargs, :xlims, :auto) == :auto xlim = [Inf, -Inf] for d in sargs _expand_limits(xlim, d[:x]) end else xmin, xmax = iargs[:xlims] xlim = [xmin, xmax] end if get(iargs, :ylims, :auto) == :auto ylim = [Inf, -Inf] for d in sargs _expand_limits(ylim, d[:y]) end else ymin, ymax = iargs[:ylims] ylim = [ymin, ymax] end # we set x/y to have a single point, since we need to create the plot with some data. # since this point is at the bottom left corner of the plot, it shouldn't actually be shown x = Float64[xlim[1]] y = Float64[ylim[1]] # create a plot window with xlim/ylim set, but the X/Y vectors are outside the bounds width, height = iargs[:size] o = UnicodePlots.Plot(x, y; width = width, height = height, title = iargs[:title], # labels = iargs[:legend], xlim = xlim, ylim = ylim) # set the axis labels UnicodePlots.xlabel!(o, iargs[:xguide]) UnicodePlots.ylabel!(o, iargs[:yguide]) # now use the ! functions to add to the plot for d in sargs addUnicodeSeries!(o, d, iargs[:legend] != :none, xlim, ylim) end # save the object plt.o = o end # add a single series function addUnicodeSeries!(o, d::KW, addlegend::Bool, xlim, ylim) # get the function, or special handling for step/bar/hist st = d[:seriestype] # handle hline/vline separately if st in (:hline,:vline) for yi in d[:y] if st == :hline UnicodePlots.lineplot!(o, xlim, [yi,yi]) else UnicodePlots.lineplot!(o, [yi,yi], ylim) end end return end stepstyle = :post if st == :path func = UnicodePlots.lineplot! elseif st == :scatter || d[:markershape] != :none func = UnicodePlots.scatterplot! elseif st == :steppost func = UnicodePlots.stairs! elseif st == :steppre func = UnicodePlots.stairs! stepstyle = :pre else error("Linestyle $st not supported by UnicodePlots") end # get the series data and label x, y = [collect(float(d[s])) for s in (:x, :y)] label = addlegend ? d[:label] : "" # if we happen to pass in allowed color symbols, great... otherwise let UnicodePlots decide color = d[:linecolor] in UnicodePlots.color_cycle ? d[:linecolor] : :auto # add the series func(o, x, y; color = color, name = label, style = stepstyle) end # function handlePlotColors(::UnicodePlotsBackend, d::KW) # # TODO: something special for unicodeplots, since it doesn't take kindly to people messing with its color palette # d[:color_palette] = [RGB(0,0,0)] # end # ------------------------------- # function _create_plot(pkg::UnicodePlotsBackend, d::KW) # plt = Plot(nothing, pkg, 0, d, KW[]) function _create_backend_figure(plt::Plot{UnicodePlotsBackend}) # do we want to give a new default size? if !haskey(plt.attr, :size) || plt.attr[:size] == default(:size) plt.attr[:size] = (60,20) end plt.attr[:color_palette] = [RGB(0,0,0)] nothing # plt end function _series_added(plt::Plot{UnicodePlotsBackend}, series::Series) d = series.d # TODO don't need these once the "bar" series recipe is done if d[:seriestype] in (:sticks, :bar) d = barHack(; d...) elseif d[:seriestype] == :hist d = barHack(; histogramHack(; d...)...) end # push!(plt.seriesargs, d) # plt end function _update_plot_object(plt::Plot{UnicodePlotsBackend}, d::KW) for k in (:title, :xguide, :yguide, :xlims, :ylims) if haskey(d, k) plt.attr[k] = d[k] end end end # ------------------------------- # since this is such a hack, it's only callable using `png`... should error during normal `writemime` function png(plt::AbstractPlot{UnicodePlotsBackend}, fn::@compat(AbstractString)) fn = addExtension(fn, "png") # make some whitespace and show the plot println("\n\n\n\n\n\n") gui(plt) @osx_only begin # BEGIN HACK # wait while the plot gets drawn sleep(0.5) # use osx screen capture when my terminal is maximized and cursor starts at the bottom (I know, right?) # TODO: compute size of plot to adjust these numbers (or maybe implement something good??) run(`screencapture -R50,600,700,420 $fn`) # END HACK (phew) return end error("Can only savepng on osx with UnicodePlots (though even then I wouldn't do it)") end # ------------------------------- # we don't do very much for subplots... just stack them vertically # function _create_subplot(subplt::Subplot{UnicodePlotsBackend}, isbefore::Bool) # isbefore && return false # true # end function Base.display(::PlotsDisplay, plt::Plot{UnicodePlotsBackend}) rebuildUnicodePlot!(plt) show(plt.o) end # function Base.display(::PlotsDisplay, subplt::Subplot{UnicodePlotsBackend}) # for plt in subplt.plts # gui(plt) # end # end