working on pgfplots
This commit is contained in:
Thomas Breloff
parent
d1d8d196ba
commit
bde3c2e19d
@ -427,7 +427,7 @@ function gr_display(sp::Subplot{GRBackend}, w, h, viewport_canvas)
|
|||||||
# @show "SUBPLOT",sp.attr[:subplot_index] bbox(sp) plotarea(sp) viewport_subplot viewport_plotarea
|
# @show "SUBPLOT",sp.attr[:subplot_index] bbox(sp) plotarea(sp) viewport_subplot viewport_plotarea
|
||||||
|
|
||||||
# fill in the plot area background
|
# fill in the plot area background
|
||||||
bg = sp.attr[:background_color_inside]
|
bg = getColor(sp.attr[:background_color_inside])
|
||||||
gr_fill_viewport(viewport_plotarea, bg)
|
gr_fill_viewport(viewport_plotarea, bg)
|
||||||
# # # # c = getColor(d[:background_color_inside])
|
# # # # c = getColor(d[:background_color_inside])
|
||||||
dark_bg = 0.21 * bg.r + 0.72 * bg.g + 0.07 * bg.b < 0.9
|
dark_bg = 0.21 * bg.r + 0.72 * bg.g + 0.07 * bg.b < 0.9
|
||||||
|
|||||||
@ -74,16 +74,18 @@ function _initialize_backend(::PGFPlotsBackend; kw...)
|
|||||||
@eval begin
|
@eval begin
|
||||||
import PGFPlots
|
import PGFPlots
|
||||||
export PGFPlots
|
export PGFPlots
|
||||||
# TODO: other initialization that needs to be eval-ed
|
|
||||||
end
|
end
|
||||||
# TODO: other initialization
|
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
|
# --------------------------------------------------------------------------------------
|
||||||
|
|
||||||
const _pgfplots_linestyles = KW(
|
const _pgfplots_linestyles = KW(
|
||||||
:solid => "solid",
|
:solid => "solid",
|
||||||
:dash => "dashed",
|
:dash => "dashed",
|
||||||
:dot => "dotted",
|
:dot => "dotted",
|
||||||
:dashdot => "dashdotted",
|
:dashdot => "dashdotted",
|
||||||
:dashdotdot => "dashdotdotted"
|
:dashdotdot => "dashdotdotted",
|
||||||
)
|
)
|
||||||
|
|
||||||
const _pgfplots_markers = KW(
|
const _pgfplots_markers = KW(
|
||||||
@ -97,292 +99,443 @@ const _pgfplots_markers = KW(
|
|||||||
:star5 => "star",
|
:star5 => "star",
|
||||||
:star6 => "asterisk",
|
:star6 => "asterisk",
|
||||||
:diamond => "diamond*",
|
:diamond => "diamond*",
|
||||||
:pentagon => "pentagon*"
|
:pentagon => "pentagon*",
|
||||||
)
|
)
|
||||||
|
|
||||||
const _pgfplots_legend_pos = KW(
|
const _pgfplots_legend_pos = KW(
|
||||||
:bottomleft => "south west",
|
:bottomleft => "south west",
|
||||||
:bottomright => "south east",
|
:bottomright => "south east",
|
||||||
:topright => "north east",
|
:topright => "north east",
|
||||||
:topleft => "north west"
|
:topleft => "north west",
|
||||||
)
|
)
|
||||||
|
|
||||||
function _pgfplots_get_color(kwargs, symb)
|
|
||||||
c = typeof(kwargs[symb]) == Symbol ? convertColor(kwargs[symb]) : kwargs[symb].c
|
const _pgf_series_extrastyle = KW(
|
||||||
# We need to convert to decimals here because pgfplot will error
|
:steppre => "const plot mark right",
|
||||||
# for colors in engineering notation
|
:stepmid => "const plot mark mid",
|
||||||
r_str = @sprintf("%.8f", float(c.r))
|
:steppost => "const plot",
|
||||||
g_str = @sprintf("%.8f", float(c.g))
|
:sticks => "ycomb",
|
||||||
b_str = @sprintf("%.8f", float(c.b))
|
:ysticks => "ycomb",
|
||||||
"{rgb,1:red,$(r_str);green,$(g_str);blue,$(b_str)}"
|
:xsticks => "xcomb",
|
||||||
|
)
|
||||||
|
|
||||||
|
# --------------------------------------------------------------------------------------
|
||||||
|
|
||||||
|
|
||||||
|
# # function _pgfplots_get_color(kwargs, symb)
|
||||||
|
# # c = typeof(kwargs[symb]) == Symbol ? convertColor(kwargs[symb]) : kwargs[symb].c
|
||||||
|
# function _pgfplots_get_color(c)
|
||||||
|
# getColor(c)
|
||||||
|
# # We need to convert to decimals here because pgfplot will error
|
||||||
|
# # for colors in engineering notation
|
||||||
|
# r_str = @sprintf("%.8f", float(c.r))
|
||||||
|
# g_str = @sprintf("%.8f", float(c.g))
|
||||||
|
# b_str = @sprintf("%.8f", float(c.b))
|
||||||
|
# "{rgb,1:red,$(r_str);green,$(g_str);blue,$(b_str)}"
|
||||||
|
# end
|
||||||
|
|
||||||
|
# # function _pgfplots_get_linestyle!(kwargs, plt)
|
||||||
|
# function _pgfplots_get_linestyle!(style, kw, d)
|
||||||
|
# ls = d[:linestyle]
|
||||||
|
# if haskey(_pgfplots_linestyles, ls)
|
||||||
|
# push!(style, _pgfplots_linestyles[ls])
|
||||||
|
# end
|
||||||
|
#
|
||||||
|
# push!(style, "line width = $(d[:linewidth]) pt")
|
||||||
|
# end
|
||||||
|
|
||||||
|
# takes in color,alpha, and returns color and alpha appropriate for pgf style
|
||||||
|
function pgf_color(c, a = nothing)
|
||||||
|
c = getColor(c)
|
||||||
|
cstr = @sprintf("{rgb,1:red,%.8f;green,%.8f;blue,%.8f}", red(c), green(c), blue(c))
|
||||||
|
a = (a == nothing ? alpha(c) : a)
|
||||||
|
cstr, a
|
||||||
end
|
end
|
||||||
|
|
||||||
function _pgfplots_get_linestyle!(kwargs, plt)
|
function pgf_fillstyle(d::KW)
|
||||||
ls = plt[:linestyle]
|
cstr,a = pgf_color(d[:fillcolor], d[:fillalpha])
|
||||||
if haskey(_pgfplots_linestyles, ls)
|
"fill = $cstr, fill opacity=$a"
|
||||||
push!(kwargs[:style], _pgfplots_linestyles[ls])
|
end
|
||||||
end
|
|
||||||
|
|
||||||
push!(kwargs[:style], "line width = $(plt[:linewidth]) pt")
|
function pgf_linestyle(d::KW)
|
||||||
|
cstr,a = pgf_color(d[:linecolor], d[:linealpha])
|
||||||
|
"""
|
||||||
|
color = $cstr,
|
||||||
|
draw opacity=$a,
|
||||||
|
line width=$(d[:linewidth]),
|
||||||
|
$(get(_pgfplots_linestyles, d[:linestyle], "solid"))
|
||||||
|
"""
|
||||||
|
end
|
||||||
|
|
||||||
|
function pgf_marker(d::KW)
|
||||||
|
shape = d[:markershape]
|
||||||
|
cstr, a = pgf_color(d[:markercolor], d[:markeralpha])
|
||||||
|
cstr_stroke, a_stroke = pgf_color(d[:markerstrokecolor], d[:markerstrokealpha])
|
||||||
|
"""
|
||||||
|
mark = $(get(_pgfplots_markers, shape, "o*")),
|
||||||
|
mark size = $(0.5 * d[:markersize]),
|
||||||
|
mark options = {
|
||||||
|
color = $cstr_stroke, draw opacity = $a_stroke,
|
||||||
|
fill = $cstr, fill opacity = $a,
|
||||||
|
line width = $(d[:markerstrokewidth]),
|
||||||
|
rotate = $(shape == :dtriangle ? 180 : 0),
|
||||||
|
$(get(_pgfplots_linestyles, d[:markerstrokestyle], "solid"))
|
||||||
|
}
|
||||||
|
"""
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
function _pgfplots_get_marker!(kwargs, plt)
|
# # function _pgfplots_get_marker!(kwargs, plt)
|
||||||
# Control marker shape, size, colors, alphas, and stroke width
|
# function _pgfplots_get_marker!(style, kw, d)
|
||||||
mark = plt[:markershape]
|
# # Control marker shape, size, colors, alphas, and stroke width
|
||||||
α = plt[:markeralpha] == nothing ? 1.0 : plt[:markeralpha]
|
# mark = d[:markershape]
|
||||||
push!(kwargs[:style], "mark = " * _pgfplots_markers[mark],
|
# α = d[:markeralpha] == nothing ? 1.0 : d[:markeralpha]
|
||||||
"mark size = $(plt[:markersize]/2)",
|
# push!(style, "mark = " * _pgfplots_markers[mark],
|
||||||
"mark options = {color=$(_pgfplots_get_color(plt, :markerstrokecolor))",
|
# "mark size = $(d[:markersize]/2)",
|
||||||
"fill=$(_pgfplots_get_color(plt, :markercolor))",
|
# "mark options = {color=$(_pgfplots_get_color(d[:markerstrokecolor]))",
|
||||||
"fill opacity = $α",
|
# "fill=$(_pgfplots_get_color(d[:markercolor]))",
|
||||||
"line width=$(plt[:markerstrokewidth])")
|
# "fill opacity = $α",
|
||||||
|
# "line width=$(d[:markerstrokewidth])")
|
||||||
# Rotate the marker if :dtriangle was chosen
|
#
|
||||||
mark == :dtriangle && push!(kwargs[:style], "rotate=180")
|
# # Rotate the marker if :dtriangle was chosen
|
||||||
|
# mark == :dtriangle && push!(style, "rotate=180")
|
||||||
# Apply marker stroke style if it is a valid PGFPlots stroke style
|
#
|
||||||
if haskey(_pgfplots_linestyles, plt[:markerstrokestyle])
|
# # Apply marker stroke style if it is a valid PGFPlots stroke style
|
||||||
push!(kwargs[:style], _pgfplots_linestyles[plt[:markerstrokestyle]])
|
# if haskey(_pgfplots_linestyles, d[:markerstrokestyle])
|
||||||
end
|
# push!(style, _pgfplots_linestyles[d[:markerstrokestyle]])
|
||||||
|
# end
|
||||||
# End the open mark options bracker
|
#
|
||||||
push!(kwargs[:style], "}")
|
# # End the open mark options bracker
|
||||||
end
|
# push!(style, "}")
|
||||||
|
|
||||||
function _pgfplots_get_series_color!(kwargs, plt)
|
|
||||||
α = plt[:seriesalpha] == nothing ? 1.0 : plt[:seriesalpha]
|
|
||||||
push!(kwargs[:style], "color=$(_pgfplots_get_color(plt, :seriescolor))",
|
|
||||||
"draw opacity = $α")
|
|
||||||
end
|
|
||||||
|
|
||||||
function _pgfplots_get_line_color!(kwargs, plt)
|
|
||||||
α = plt[:linealpha] == nothing ? 1.0 : plt[:linealpha]
|
|
||||||
kwargs[:style] *= ", color=$(_pgfplots_get_color(plt, :linecolor))" *
|
|
||||||
", draw opacity = $α"
|
|
||||||
end
|
|
||||||
|
|
||||||
function _pgfplots_get_fill_color!(kwargs, plt)
|
|
||||||
α = plt[:fillalpha] == nothing ? 1.0 : plt[:fillalpha]
|
|
||||||
kwargs[:style] *= ", fill=$(_pgfplots_get_color(plt, :fillcolor))" *
|
|
||||||
", fill opacity = $α"
|
|
||||||
end
|
|
||||||
|
|
||||||
function _pgfplots_get_label!(kwargs, plt)
|
|
||||||
if plt[:label] != nothing && plt[:legend] != :none
|
|
||||||
kwargs[:legendentry] = plt[:label]
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
function _pgfplots_get_plot_kwargs(plt)
|
|
||||||
kwargs = KW()
|
|
||||||
kwargs[:style] = []
|
|
||||||
_pgfplots_get_linestyle!(kwargs, plt)
|
|
||||||
_pgfplots_get_marker!(kwargs, plt)
|
|
||||||
_pgfplots_get_series_color!(kwargs, plt)
|
|
||||||
_pgfplots_get_label!(kwargs, plt)
|
|
||||||
kwargs[:style] = join(kwargs[:style], ',')
|
|
||||||
kwargs
|
|
||||||
end
|
|
||||||
|
|
||||||
function _pgfplots_axis(plt_series)
|
|
||||||
line_type = plt_series[:seriestype]
|
|
||||||
plt_kwargs = _pgfplots_get_plot_kwargs(plt_series)
|
|
||||||
if line_type == :path
|
|
||||||
PGFPlots.Linear(plt_series[:x], plt_series[:y]; plt_kwargs...)
|
|
||||||
elseif line_type == :path3d
|
|
||||||
PGFPlots.Linear3(plt_series[:x], plt_series[:y], plt_series[:z]; plt_kwargs...)
|
|
||||||
elseif line_type == :scatter
|
|
||||||
PGFPlots.Scatter(plt_series[:x], plt_series[:y]; plt_kwargs...)
|
|
||||||
elseif line_type == :steppre
|
|
||||||
plt_kwargs[:style] *= ", const plot mark right"
|
|
||||||
PGFPlots.Linear(plt_series[:x], plt_series[:y]; plt_kwargs...)
|
|
||||||
elseif line_type == :stepmid
|
|
||||||
plt_kwargs[:style] *= ", const plot mark mid"
|
|
||||||
PGFPlots.Linear(plt_series[:x], plt_series[:y]; plt_kwargs...)
|
|
||||||
elseif line_type == :steppost
|
|
||||||
plt_kwargs[:style] *= ", const plot"
|
|
||||||
PGFPlots.Linear(plt_series[:x], plt_series[:y]; plt_kwargs...)
|
|
||||||
elseif line_type == :hist
|
|
||||||
#TODO patch this in PGFPlots.jl instead; the problem is that PGFPlots will
|
|
||||||
# save _all_ data points in the figure which can be quite heavy
|
|
||||||
plt_hist = hist(plt_series[:y])
|
|
||||||
plt_kwargs[:style] *= ", ybar interval"
|
|
||||||
_pgfplots_get_line_color!(plt_kwargs, plt_series)
|
|
||||||
_pgfplots_get_fill_color!(plt_kwargs, plt_series)
|
|
||||||
PGFPlots.Linear(plt_hist[1][1:end-1]+plt_hist[1].step/2, plt_hist[2]; plt_kwargs...)
|
|
||||||
elseif line_type == :hist2d
|
|
||||||
PGFPlots.Histogram2(plt_series[:x], plt_series[:y])
|
|
||||||
elseif line_type == :bar
|
|
||||||
plt_kwargs[:style] *= ", ybar"
|
|
||||||
_pgfplots_get_line_color!(plt_kwargs, plt_series)
|
|
||||||
_pgfplots_get_fill_color!(plt_kwargs, plt_series)
|
|
||||||
PGFPlots.Linear(plt_series[:x], plt_series[:y]; plt_kwargs...)
|
|
||||||
elseif line_type == :sticks || line_type == :ysticks
|
|
||||||
plt_kwargs[:style] *= ", ycomb"
|
|
||||||
PGFPlots.Linear(plt_series[:x], plt_series[:y]; plt_kwargs...)
|
|
||||||
elseif line_type == :xsticks
|
|
||||||
plt_kwargs[:style] *= ", xcomb"
|
|
||||||
PGFPlots.Linear(plt_series[:x], plt_series[:y]; plt_kwargs...)
|
|
||||||
elseif line_type == :contour
|
|
||||||
PGFPlots.Contour(plt_series[:z].surf, plt_series[:x], plt_series[:y])
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
# function _create_plot(pkg::PGFPlotsBackend, d::KW)
|
|
||||||
# # TODO: create the window/canvas/context that is the plot within the backend (call it `o`)
|
|
||||||
# # TODO: initialize the plot... title, xlabel, bgcolor, etc
|
|
||||||
# Plot(nothing, pkg, 0, d, KW[])
|
|
||||||
# end
|
# end
|
||||||
|
|
||||||
|
|
||||||
# function _series_added(::PGFPlotsBackend, plt::Plot, d::KW)
|
|
||||||
# # TODO: add one series to the underlying package
|
|
||||||
# push!(plt.seriesargs, d)
|
|
||||||
# plt
|
|
||||||
# end
|
|
||||||
|
|
||||||
function _add_annotations{X,Y,V}(plt::Plot{PGFPlotsBackend}, anns::AVec{@compat(Tuple{X,Y,V})})
|
# # function _pgfplots_get_series_color!(kwargs, plt)
|
||||||
# set or add to the annotation_list
|
# function _pgfplots_get_series_color!(style, kw, d)
|
||||||
if haskey(plt.attr, :annotation_list)
|
# c = getColor(d[:seriescolor])
|
||||||
append!(plt.attr[:annotation_list], anns)
|
# α = d[:seriesalpha] == nothing ? 1.0 : d[:seriesalpha]
|
||||||
else
|
# push!(style, "color=$(_pgfplots_get_color(d[:seriescolor]))",
|
||||||
plt.attr[:annotation_list] = anns
|
# "draw opacity = $α")
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
# ----------------------------------------------------------------
|
|
||||||
|
|
||||||
# function _before_update_plot(plt::Plot{PGFPlotsBackend})
|
|
||||||
# end
|
|
||||||
|
|
||||||
# TODO: override this to update plot items (title, xlabel, etc) after creation
|
|
||||||
function _update_plot_object(plt::Plot{PGFPlotsBackend}, d::KW)
|
|
||||||
end
|
|
||||||
|
|
||||||
# function _update_plot_pos_size(plt::AbstractPlot{PGFPlotsBackend}, d::KW)
|
|
||||||
# end
|
|
||||||
|
|
||||||
# ----------------------------------------------------------------
|
|
||||||
|
|
||||||
# accessors for x/y data
|
|
||||||
|
|
||||||
# function getxy(plt::Plot{PGFPlotsBackend}, i::Int)
|
|
||||||
# d = plt.seriesargs[i]
|
|
||||||
# d[:x], d[:y]
|
|
||||||
# end
|
# end
|
||||||
#
|
#
|
||||||
# function setxy!{X,Y}(plt::Plot{PGFPlotsBackend}, xy::Tuple{X,Y}, i::Integer)
|
# # function _pgfplots_get_line_color!(kwargs, plt)
|
||||||
# d = plt.seriesargs[i]
|
# function _pgfplots_get_line_color!(style, kw, d)
|
||||||
# d[:x], d[:y] = xy
|
# α = d[:linealpha] == nothing ? 1.0 : d[:linealpha]
|
||||||
# plt
|
# style *= ", color=$(_pgfplots_get_color(d[:linecolor]))" *
|
||||||
# end
|
# ", draw opacity = $α"
|
||||||
|
|
||||||
# ----------------------------------------------------------------
|
|
||||||
|
|
||||||
# function _create_subplot(subplt::Subplot{PGFPlotsBackend}, isbefore::Bool)
|
|
||||||
# # TODO: build the underlying Subplot object. this is where you might layout the panes within a GUI window, for example
|
|
||||||
# true
|
|
||||||
# end
|
|
||||||
|
|
||||||
# function _expand_limits(lims, plt::Plot{PGFPlotsBackend}, isx::Bool)
|
|
||||||
# # TODO: call expand limits for each plot data
|
|
||||||
# end
|
# end
|
||||||
#
|
#
|
||||||
# function _remove_axis(plt::Plot{PGFPlotsBackend}, isx::Bool)
|
# # function _pgfplots_get_fill_color!(kwargs, plt)
|
||||||
# # TODO: if plot is inner subplot, might need to remove ticks or axis labels
|
# function _pgfplots_get_fill_color!(style, kw, d)
|
||||||
|
# α = d[:fillalpha] == nothing ? 1.0 : d[:fillalpha]
|
||||||
|
# style *= ", fill=$(_pgfplots_get_color(d[:fillcolor]))" *
|
||||||
|
# ", fill opacity = $α"
|
||||||
|
# end
|
||||||
|
|
||||||
|
# # function _pgfplots_get_label!(kwargs, plt)
|
||||||
|
# function _pgfplots_get_label!(kw::KW, series::Series)
|
||||||
|
# if d[:label] != nothing && d[:legend] != :none
|
||||||
|
# kwargs[:legendentry] = d[:label]
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
|
||||||
|
# --------------------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# function _pgfplots_get_plot_kwargs(plt)
|
||||||
|
# style = []
|
||||||
|
# kw = KW()
|
||||||
|
# # kw[:style] = []
|
||||||
|
# _pgfplots_get_linestyle!(style, kw, plt)
|
||||||
|
# _pgfplots_get_marker!(style, kw, plt)
|
||||||
|
# _pgfplots_get_series_color!(style, kw, plt)
|
||||||
|
# _pgfplots_get_label!(style, kw, plt)
|
||||||
|
# kw[:style] = join(style, ',')
|
||||||
|
# kw
|
||||||
|
# end
|
||||||
|
|
||||||
|
|
||||||
|
function pgf_series(sp::Subplot, series::Series)
|
||||||
|
d = series.d
|
||||||
|
st = d[:seriestype]
|
||||||
|
style = []
|
||||||
|
kw = KW()
|
||||||
|
|
||||||
|
push!(style, pgf_linestyle(d))
|
||||||
|
push!(style, pgf_marker(d))
|
||||||
|
push!(style, pgf_fillstyle(d))
|
||||||
|
|
||||||
|
# add to legend?
|
||||||
|
if should_add_to_legend(series)
|
||||||
|
kw[:legendentry] = d[:label]
|
||||||
|
end
|
||||||
|
|
||||||
|
# function args
|
||||||
|
args = if st == :contour
|
||||||
|
d[:z].surf, d[:x], d[:y]
|
||||||
|
elseif is3d(st)
|
||||||
|
d[:x], d[:y], d[:z]
|
||||||
|
else
|
||||||
|
d[:x], d[:y]
|
||||||
|
end
|
||||||
|
|
||||||
|
# include additional style, then add to the kw
|
||||||
|
if haskey(_pgf_series_extrastyle, st)
|
||||||
|
push!(style, _pgf_series_extrastyle[st])
|
||||||
|
end
|
||||||
|
kw[:style] = join(style, ',')
|
||||||
|
|
||||||
|
# build/return the series object
|
||||||
|
func = if st == :path3d
|
||||||
|
PGFPlots.Linear3
|
||||||
|
elseif st == :scatter
|
||||||
|
PGFPlots.Scatter
|
||||||
|
elseif st == :hist2d
|
||||||
|
PGFPlots.Histogram2
|
||||||
|
elseif st == :contour
|
||||||
|
PGFPlots.Contour
|
||||||
|
else
|
||||||
|
PGFPlots.Linear
|
||||||
|
end
|
||||||
|
func(args...; kw...)
|
||||||
|
|
||||||
|
# # now return the series object
|
||||||
|
# func, args = if st == :path
|
||||||
|
# PGFPlots.Linear, (d[:x], d[:y])
|
||||||
|
# elseif st == :path3d
|
||||||
|
# PGFPlots.Linear3(d[:x], d[:y], d[:z]; kw...)
|
||||||
|
# elseif st == :scatter
|
||||||
|
# PGFPlots.Scatter(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :steppre
|
||||||
|
# kw[:style] *= ", const plot mark right"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :stepmid
|
||||||
|
# kw[:style] *= ", const plot mark mid"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :steppost
|
||||||
|
# kw[:style] *= ", const plot"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# # elseif st == :hist
|
||||||
|
# # #TODO patch this in PGFPlots.jl instead; the problem is that PGFPlots will
|
||||||
|
# # # save _all_ data points in the figure which can be quite heavy
|
||||||
|
# # plt_hist = hist(d[:y])
|
||||||
|
# # kw[:style] *= ", ybar interval"
|
||||||
|
# # _pgfplots_get_line_color!(kw, d)
|
||||||
|
# # _pgfplots_get_fill_color!(kw, d)
|
||||||
|
# # PGFPlots.Linear(plt_hist[1][1:end-1]+plt_hist[1].step/2, plt_hist[2]; kw...)
|
||||||
|
# elseif st == :hist2d
|
||||||
|
# PGFPlots.Histogram2(d[:x], d[:y])
|
||||||
|
# # elseif st == :bar
|
||||||
|
# # kw[:style] *= ", ybar"
|
||||||
|
# # _pgfplots_get_line_color!(kw, d)
|
||||||
|
# # _pgfplots_get_fill_color!(kw, d)
|
||||||
|
# # PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :sticks || st == :ysticks
|
||||||
|
# kw[:style] *= ", ycomb"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :xsticks
|
||||||
|
# kw[:style] *= ", xcomb"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :contour
|
||||||
|
# PGFPlots.Contour(d[:z].surf, d[:x], d[:y])
|
||||||
|
# end
|
||||||
|
|
||||||
|
|
||||||
|
# if st == :path
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :path3d
|
||||||
|
# PGFPlots.Linear3(d[:x], d[:y], d[:z]; kw...)
|
||||||
|
# elseif st == :scatter
|
||||||
|
# PGFPlots.Scatter(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :steppre
|
||||||
|
# kw[:style] *= ", const plot mark right"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :stepmid
|
||||||
|
# kw[:style] *= ", const plot mark mid"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :steppost
|
||||||
|
# kw[:style] *= ", const plot"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :hist
|
||||||
|
# #TODO patch this in PGFPlots.jl instead; the problem is that PGFPlots will
|
||||||
|
# # save _all_ data points in the figure which can be quite heavy
|
||||||
|
# plt_hist = hist(d[:y])
|
||||||
|
# kw[:style] *= ", ybar interval"
|
||||||
|
# _pgfplots_get_line_color!(kw, d)
|
||||||
|
# _pgfplots_get_fill_color!(kw, d)
|
||||||
|
# PGFPlots.Linear(plt_hist[1][1:end-1]+plt_hist[1].step/2, plt_hist[2]; kw...)
|
||||||
|
# elseif st == :hist2d
|
||||||
|
# PGFPlots.Histogram2(d[:x], d[:y])
|
||||||
|
# elseif st == :bar
|
||||||
|
# kw[:style] *= ", ybar"
|
||||||
|
# _pgfplots_get_line_color!(kw, d)
|
||||||
|
# _pgfplots_get_fill_color!(kw, d)
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :sticks || st == :ysticks
|
||||||
|
# kw[:style] *= ", ycomb"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :xsticks
|
||||||
|
# kw[:style] *= ", xcomb"
|
||||||
|
# PGFPlots.Linear(d[:x], d[:y]; kw...)
|
||||||
|
# elseif st == :contour
|
||||||
|
# PGFPlots.Contour(d[:z].surf, d[:x], d[:y])
|
||||||
|
# end
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
# ----------------------------------------------------------------
|
||||||
|
|
||||||
|
function pgf_axis(sp::Subplot, letter)
|
||||||
|
axis = sp.attr[symbol(letter,:axis)]
|
||||||
|
style = []
|
||||||
|
kw = KW()
|
||||||
|
|
||||||
|
# axis guide
|
||||||
|
kw[symbol(letter,:label)] = axis[:guide]
|
||||||
|
|
||||||
|
# flip/reverse?
|
||||||
|
axis[:flip] && push!(style, "$letter dir=reverse")
|
||||||
|
|
||||||
|
# scale
|
||||||
|
scale = axis[:scale]
|
||||||
|
if scale in (:log2, :ln, :log10)
|
||||||
|
kw[symbol(letter,:mode)] = "log"
|
||||||
|
scale == :ln || push!(style, "log basis $letter=$(scale == :log2 ? 2 : 10)")
|
||||||
|
end
|
||||||
|
|
||||||
|
# limits
|
||||||
|
lims = axis_limits(axis)
|
||||||
|
kw[symbol(letter,:min)] = lims[1]
|
||||||
|
kw[symbol(letter,:max)] = lims[2]
|
||||||
|
|
||||||
|
# return the style list and KW args
|
||||||
|
style, kw
|
||||||
|
end
|
||||||
|
|
||||||
|
# function _pgfplots_get_axis_kwargs(d)
|
||||||
|
# axisargs = KW()
|
||||||
|
# for arg in (:xguide, :yguide, :zguide, :title)
|
||||||
|
# axisargs[arg] = d[arg]
|
||||||
|
# end
|
||||||
|
# axisargs[:style] = []
|
||||||
|
# d[:xflip] == true && push!(axisargs[:style], "x dir=reverse")
|
||||||
|
# d[:yflip] == true && push!(axisargs[:style], "y dir=reverse")
|
||||||
|
# if d[:xscale] in (:log, :log2, :ln, :log10)
|
||||||
|
# axisargs[:xmode] = "log"
|
||||||
|
# if d[:xscale] == :log2
|
||||||
|
# push!(axisargs[:style], "log basis x=2")
|
||||||
|
# elseif d[:xscale] in (:log, :log10)
|
||||||
|
# push!(axisargs[:style], "log basis x=10")
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
# if d[:yscale] in (:log, :log2, :ln, :log10)
|
||||||
|
# axisargs[:ymode] = "log"
|
||||||
|
# if d[:yscale] == :log2
|
||||||
|
# push!(axisargs[:style], "log basis y=2")
|
||||||
|
# elseif d[:yscale] in (:log, :log10)
|
||||||
|
# push!(axisargs[:style], "log basis x=10")
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
# if d[:zscale] in (:log, :log2, :ln, :log10)
|
||||||
|
# axisargs[:zmode] = "log"
|
||||||
|
# if d[:zscale] == :log2
|
||||||
|
# push!(axisargs[:style], "log basis z=2")
|
||||||
|
# elseif d[:zscale] in (:log, :log10)
|
||||||
|
# push!(axisargs[:style], "log basis x=10")
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
#
|
||||||
|
# # Control background color
|
||||||
|
# push!(axisargs[:style], "axis background/.style={fill=$(_pgfplots_get_color(d, :background_color))}")
|
||||||
|
# # Control x/y-limits
|
||||||
|
# if d[:xlims] !== :auto
|
||||||
|
# axisargs[:xmin] = d[:xlims][1]
|
||||||
|
# axisargs[:xmax] = d[:xlims][2]
|
||||||
|
# end
|
||||||
|
# if d[:ylims] !== :auto
|
||||||
|
# axisargs[:ymin] = d[:ylims][1]
|
||||||
|
# axisargs[:ymax] = d[:ylims][2]
|
||||||
|
# end
|
||||||
|
#
|
||||||
|
# d[:grid] == true && push!(axisargs[:style], "grid = major")
|
||||||
|
#
|
||||||
|
# if d[:aspect_ratio] == :equal || d[:aspect_ratio] == 1
|
||||||
|
# axisargs[:axisEqual] = "true"
|
||||||
|
# end
|
||||||
|
#
|
||||||
|
# if ((d[:legend] != :none) || (d[:legend] != :best)) && (d[:legend] in keys(_pgfplots_legend_pos))
|
||||||
|
# axisargs[:legendPos] = _pgfplots_legend_pos[d[:legend]]
|
||||||
|
# end
|
||||||
|
# axisargs[:style] = join(axisargs[:style], ',')
|
||||||
|
# axisargs
|
||||||
# end
|
# end
|
||||||
|
|
||||||
# ----------------------------------------------------------------
|
# ----------------------------------------------------------------
|
||||||
|
|
||||||
function _pgfplots_get_axis_kwargs(d)
|
# ################# This is the important method to implement!!! #################
|
||||||
axisargs = KW()
|
# function _make_pgf_plot(plt::Plot{PGFPlotsBackend})
|
||||||
for arg in (:xguide, :yguide, :zguide, :title)
|
# os = Any[]
|
||||||
axisargs[arg] = d[arg]
|
# # We need to send the :legend KW to the axis
|
||||||
end
|
# for plt_series in plt.seriesargs
|
||||||
axisargs[:style] = []
|
# plt_series[:legend] = plt.attr[:legend]
|
||||||
d[:xflip] == true && push!(axisargs[:style], "x dir=reverse")
|
# push!(os, _pgfplots_axis(plt_series))
|
||||||
d[:yflip] == true && push!(axisargs[:style], "y dir=reverse")
|
# end
|
||||||
if d[:xscale] in (:log, :log2, :ln, :log10)
|
# axisargs =_pgfplots_get_axis_kwargs(plt.attr)
|
||||||
axisargs[:xmode] = "log"
|
# w, h = map(px2mm, plt.attr[:size])
|
||||||
if d[:xscale] == :log2
|
# plt.o = PGFPlots.Axis([os...]; width = "$w mm", height = "$h mm", axisargs...)
|
||||||
push!(axisargs[:style], "log basis x=2")
|
# end
|
||||||
elseif d[:xscale] in (:log, :log10)
|
|
||||||
push!(axisargs[:style], "log basis x=10")
|
function _make_pgf_plot!(plt::Plot)
|
||||||
|
plt.o = []
|
||||||
|
for sp in plt.subplots
|
||||||
|
# first build the PGFPlots.Axis object
|
||||||
|
style = []
|
||||||
|
kw = KW()
|
||||||
|
|
||||||
|
# add to style/kw for each axis
|
||||||
|
for letter in (:x, :y, :z)
|
||||||
|
if letter != :z || is3d(sp)
|
||||||
|
axisstyle, axiskw = pgf_axis(sp, letter)
|
||||||
|
merge!(kw, axiskw)
|
||||||
|
end
|
||||||
end
|
end
|
||||||
end
|
|
||||||
if d[:yscale] in (:log, :log2, :ln, :log10)
|
# bounding box values are in mm
|
||||||
axisargs[:ymode] = "log"
|
bb = bbox(sp)
|
||||||
if d[:yscale] == :log2
|
push!(style, "xshift = $(left(bb).value)mm")
|
||||||
push!(axisargs[:style], "log basis y=2")
|
push!(style, "yshift = $((height(bb) - (bottom(bb))).value)mm") # bb origin is top-left, pgf is bottom-left
|
||||||
elseif d[:yscale] in (:log, :log10)
|
push!(style, "width = $(width(bb).value)mm")
|
||||||
push!(axisargs[:style], "log basis x=10")
|
push!(style, "height = $(height(bb).value)mm")
|
||||||
|
push!(style, "title = \"$(sp.attr[:title])\"")
|
||||||
|
push!(style, "axis background/.style={fill=$(pgf_color(sp.attr[:background_color_inside])[1])}")
|
||||||
|
|
||||||
|
sp.attr[:grid] && push!(style, "grid = major")
|
||||||
|
sp.attr[:aspect_ratio] in (1, :equal) && (kw[:axisEqual] = "true")
|
||||||
|
|
||||||
|
legpos = sp.attr[:legend]
|
||||||
|
if haskey(_pgfplots_legend_pos, legpos)
|
||||||
|
kw[:legendPos] = _pgfplots_legend_pos[legpos]
|
||||||
end
|
end
|
||||||
end
|
|
||||||
if d[:zscale] in (:log, :log2, :ln, :log10)
|
o = PGFPlots.Axis(; style = style, kw...)
|
||||||
axisargs[:zmode] = "log"
|
|
||||||
if d[:zscale] == :log2
|
# add the series object to the PGFPlots.Axis
|
||||||
push!(axisargs[:style], "log basis z=2")
|
for series in series_list(sp)
|
||||||
elseif d[:zscale] in (:log, :log10)
|
push!(o, pgf_series(sp, series))
|
||||||
push!(axisargs[:style], "log basis x=10")
|
|
||||||
end
|
end
|
||||||
end
|
|
||||||
|
|
||||||
# Control background color
|
# add the PGFPlots.Axis to the list
|
||||||
push!(axisargs[:style], "axis background/.style={fill=$(_pgfplots_get_color(d, :background_color))}")
|
push!(plt.o, o)
|
||||||
# Control x/y-limits
|
|
||||||
if d[:xlims] !== :auto
|
|
||||||
axisargs[:xmin] = d[:xlims][1]
|
|
||||||
axisargs[:xmax] = d[:xlims][2]
|
|
||||||
end
|
end
|
||||||
if d[:ylims] !== :auto
|
|
||||||
axisargs[:ymin] = d[:ylims][1]
|
|
||||||
axisargs[:ymax] = d[:ylims][2]
|
|
||||||
end
|
|
||||||
|
|
||||||
d[:grid] == true && push!(axisargs[:style], "grid = major")
|
|
||||||
|
|
||||||
if d[:aspect_ratio] == :equal || d[:aspect_ratio] == 1
|
|
||||||
axisargs[:axisEqual] = "true"
|
|
||||||
end
|
|
||||||
|
|
||||||
if ((d[:legend] != :none) || (d[:legend] != :best)) && (d[:legend] in keys(_pgfplots_legend_pos))
|
|
||||||
axisargs[:legendPos] = _pgfplots_legend_pos[d[:legend]]
|
|
||||||
end
|
|
||||||
axisargs[:style] = join(axisargs[:style], ',')
|
|
||||||
axisargs
|
|
||||||
end
|
end
|
||||||
|
|
||||||
# ----------------------------------------------------------------
|
|
||||||
|
|
||||||
################# This is the important method to implement!!! #################
|
function _writemime(io::IO, mime::MIME"image/svg+xml", plt::Plot{PGFPlotsBackend})
|
||||||
function _make_pgf_plot(plt::Plot{PGFPlotsBackend})
|
_make_pgf_plot!(plt)
|
||||||
os = Any[]
|
|
||||||
# We need to send the :legend KW to the axis
|
|
||||||
for plt_series in plt.seriesargs
|
|
||||||
plt_series[:legend] = plt.attr[:legend]
|
|
||||||
push!(os, _pgfplots_axis(plt_series))
|
|
||||||
end
|
|
||||||
axisargs =_pgfplots_get_axis_kwargs(plt.attr)
|
|
||||||
w, h = map(px2mm, plt.attr[:size])
|
|
||||||
plt.o = PGFPlots.Axis([os...]; width = "$w mm", height = "$h mm", axisargs...)
|
|
||||||
end
|
|
||||||
|
|
||||||
function Base.writemime(io::IO, mime::MIME"image/svg+xml", plt::AbstractPlot{PGFPlotsBackend})
|
|
||||||
plt.o = _make_pgf_plot(plt)
|
|
||||||
writemime(io, mime, plt.o)
|
writemime(io, mime, plt.o)
|
||||||
end
|
end
|
||||||
|
|
||||||
# function Base.writemime(io::IO, ::MIME"text/html", plt::AbstractPlot{PGFPlotsBackend})
|
function _display(plt::Plot{PGFPlotsBackend})
|
||||||
# end
|
_make_pgf_plot!(plt)
|
||||||
|
|
||||||
function Base.display(::PlotsDisplay, plt::AbstractPlot{PGFPlotsBackend})
|
|
||||||
plt.o = _make_pgf_plot(plt)
|
|
||||||
display(plt.o)
|
display(plt.o)
|
||||||
end
|
end
|
||||||
|
|
||||||
# function Base.display(::PlotsDisplay, plt::Subplot{PGFPlotsBackend})
|
|
||||||
# # TODO: display/show the subplot
|
|
||||||
# end
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user