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

use a.''foo'' instead of a.foo

Browse Source
This commit is contained in:
Daniel Schwabeneder 2019-03-15 15:15:59 +01:00
parent b21d98affd
commit b79ea119ea
1 changed files with 102 additions and 102 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

204
src/backends/pyplot.jl
View File

@ -64,8 +64,8 @@ py_color(c::Colorant, α) = py_color(plot_color(c, α))
function py_colormap(grad::ColorGradient) function py_colormap(grad::ColorGradient)
pyvals = [(z, py_color(grad[z])) for z in grad.values] pyvals = [(z, py_color(grad[z])) for z in grad.values]
cm = pycolors."LinearSegmentedColormap".from_list("tmp", pyvals) cm = pycolors."LinearSegmentedColormap"."from_list"("tmp", pyvals)
cm.set_bad(color=(0,0,0,0.0), alpha=0.0) cm."set_bad"(color=(0,0,0,0.0), alpha=0.0)
cm cm
end end
py_colormap(c) = py_colormap(cgrad()) py_colormap(c) = py_colormap(cgrad())
@ -74,7 +74,7 @@ py_colormap(c) = py_colormap(cgrad())
function py_shading(c, z) function py_shading(c, z)
cmap = py_colormap(c) cmap = py_colormap(c)
ls = pycolors."LightSource"(270,45) ls = pycolors."LightSource"(270,45)
ls.shade(z, cmap, vert_exag=0.1, blend_mode="soft") ls."shade"(z, cmap, vert_exag=0.1, blend_mode="soft")
end end
# get the style (solid, dashed, etc) # get the style (solid, dashed, etc)
@ -204,7 +204,7 @@ end
function py_mask_nans(z) function py_mask_nans(z)
# pynp["ma"][:masked_invalid](z))) # pynp["ma"][:masked_invalid](z)))
PyCall.pycall(pynp."ma".masked_invalid, Any, z) PyCall.pycall(pynp."ma"."masked_invalid", Any, z)
# pynp["ma"][:masked_where](pynp["isnan"](z),z) # pynp["ma"][:masked_where](pynp["isnan"](z),z)
end end
@ -248,22 +248,22 @@ py_fillcolormap(series::Series) = py_colormap(series[:fillcolor])
# Figure utils -- F*** matplotlib for making me work so hard to figure this crap out # Figure utils -- F*** matplotlib for making me work so hard to figure this crap out
# the drawing surface # the drawing surface
py_canvas(fig) = fig.canvas py_canvas(fig) = fig."canvas"
# the object controlling draw commands # the object controlling draw commands
py_renderer(fig) = py_canvas(fig).get_renderer() py_renderer(fig) = py_canvas(fig)."get_renderer"()
# draw commands... paint the screen (probably updating internals too) # draw commands... paint the screen (probably updating internals too)
py_drawfig(fig) = fig.draw(py_renderer(fig)) py_drawfig(fig) = fig."draw"(py_renderer(fig))
# py_drawax(ax) = ax[:draw](py_renderer(ax[:get_figure]())) # py_drawax(ax) = ax[:draw](py_renderer(ax[:get_figure]()))
# get a vector [left, right, bottom, top] in PyPlot coords (origin is bottom-left!) # get a vector [left, right, bottom, top] in PyPlot coords (origin is bottom-left!)
py_extents(obj) = obj.get_window_extent().get_points() py_extents(obj) = obj."get_window_extent"()."get_points"()
# compute a bounding box (with origin top-left), however pyplot gives coords with origin bottom-left # compute a bounding box (with origin top-left), however pyplot gives coords with origin bottom-left
function py_bbox(obj) function py_bbox(obj)
fl, fr, fb, ft = py_extents(obj.get_figure()) fl, fr, fb, ft = py_extents(obj."get_figure"())
l, r, b, t = py_extents(obj) l, r, b, t = py_extents(obj)
BoundingBox(l*px, (ft-t)*px, (r-l)*px, (t-b)*px) BoundingBox(l*px, (ft-t)*px, (r-l)*px, (t-b)*px)
end end
@ -339,7 +339,7 @@ function py_init_subplot(plt::Plot{PyPlotBackend}, sp::Subplot{PyPlotBackend})
proj = (proj in (nothing,:none) ? nothing : string(proj)) proj = (proj in (nothing,:none) ? nothing : string(proj))
# add a new axis, and force it to create a new one by setting a distinct label # add a new axis, and force it to create a new one by setting a distinct label
ax = fig.add_axes( ax = fig."add_axes"(
[0,0,1,1], [0,0,1,1],
label = string(gensym()), label = string(gensym()),
projection = proj projection = proj
@ -440,7 +440,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
# push!(handles, handle) # push!(handles, handle)
# else # else
for (i, rng) in enumerate(iter_segments(series)) for (i, rng) in enumerate(iter_segments(series))
handle = ax.plot((arg[rng] for arg in xyargs)...; handle = ax."plot"((arg[rng] for arg in xyargs)...;
label = i == 1 ? series[:label] : "", label = i == 1 ? series[:label] : "",
zorder = series[:series_plotindex], zorder = series[:series_plotindex],
color = py_color(get_linecolor(series, clims, i), get_linealpha(series, i)), color = py_color(get_linecolor(series, clims, i), get_linealpha(series, i)),
@ -468,7 +468,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
:linestyle => py_linestyle(st, get_linestyle(series)), :linestyle => py_linestyle(st, get_linestyle(series)),
) )
add_arrows(x, y) do xyprev, xy add_arrows(x, y) do xyprev, xy
ax.annotate("", ax."annotate"("",
xytext = (0.001xyprev[1] + 0.999xy[1], 0.001xyprev[2] + 0.999xy[2]), xytext = (0.001xyprev[1] + 0.999xy[1], 0.001xyprev[2] + 0.999xy[2]),
xy = xy, xy = xy,
arrowprops = arrowprops, arrowprops = arrowprops,
@ -514,7 +514,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
for i=1:length(y) for i=1:length(y)
extrakw[:c] = _cycle(markercolor, i) extrakw[:c] = _cycle(markercolor, i)
push!(handle, ax.scatter(_cycle(x,i), _cycle(y,i); push!(handle, ax."scatter"(_cycle(x,i), _cycle(y,i);
label = series[:label], label = series[:label],
zorder = series[:series_plotindex] + 0.5, zorder = series[:series_plotindex] + 0.5,
marker = py_marker(_cycle(shapes,i)), marker = py_marker(_cycle(shapes,i)),
@ -553,7 +553,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
continue continue
end end
push!(handle, ax.scatter(cur_x_list, cur_y_list; push!(handle, ax."scatter"(cur_x_list, cur_y_list;
label = series[:label], label = series[:label],
zorder = series[:series_plotindex] + 0.5, zorder = series[:series_plotindex] + 0.5,
marker = prev_marker, marker = prev_marker,
@ -574,7 +574,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
end end
if !isempty(cur_color_list) if !isempty(cur_color_list)
push!(handle, ax.scatter(cur_x_list, cur_y_list; push!(handle, ax."scatter"(cur_x_list, cur_y_list;
label = series[:label], label = series[:label],
zorder = series[:series_plotindex] + 0.5, zorder = series[:series_plotindex] + 0.5,
marker = prev_marker, marker = prev_marker,
@ -589,7 +589,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
push!(handles, handle) push!(handles, handle)
else else
# do a normal scatter plot # do a normal scatter plot
handle = ax.scatter(xyargs...; handle = ax."scatter"(xyargs...;
label = series[:label], label = series[:label],
zorder = series[:series_plotindex] + 0.5, zorder = series[:series_plotindex] + 0.5,
marker = py_marker(series[:markershape]), marker = py_marker(series[:markershape]),
@ -603,7 +603,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
end end
if st == :hexbin if st == :hexbin
handle = ax.hexbin(x, y; handle = ax."hexbin"(x, y;
label = series[:label], label = series[:label],
zorder = series[:series_plotindex], zorder = series[:series_plotindex],
gridsize = series[:bins], gridsize = series[:bins],
@ -635,7 +635,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
end end
# contour lines # contour lines
handle = ax.contour(x, y, z, levelargs...; handle = ax."contour"(x, y, z, levelargs...;
label = series[:label], label = series[:label],
zorder = series[:series_plotindex], zorder = series[:series_plotindex],
linewidths = py_thickness_scale(plt, series[:linewidth]), linewidths = py_thickness_scale(plt, series[:linewidth]),
@ -643,13 +643,13 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
extrakw... extrakw...
) )
if series[:contour_labels] == true if series[:contour_labels] == true
PyPlot.clabel(handle, handle.levels) PyPlot."clabel"(handle, handle.levels)
end end
push!(handles, handle) push!(handles, handle)
# contour fills # contour fills
if series[:fillrange] != nothing if series[:fillrange] != nothing
handle = ax.contourf(x, y, z, levelargs...; handle = ax."contourf"(x, y, z, levelargs...;
label = series[:label], label = series[:label],
zorder = series[:series_plotindex] + 0.5, zorder = series[:series_plotindex] + 0.5,
extrakw... extrakw...
@ -690,7 +690,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
if series[:contours] if series[:contours]
for (zdir,mat) in (("x",x), ("y",y), ("z",z)) for (zdir,mat) in (("x",x), ("y",y), ("z",z))
offset = (zdir == "y" ? ignorenan_maximum : ignorenan_minimum)(mat) offset = (zdir == "y" ? ignorenan_maximum : ignorenan_minimum)(mat)
handle = ax.contourf(x, y, z, levelargs...; handle = ax."contourf"(x, y, z, levelargs...;
zdir = zdir, zdir = zdir,
cmap = py_fillcolormap(series), cmap = py_fillcolormap(series),
offset = (zdir == "y" ? ignorenan_maximum : ignorenan_minimum)(mat) # where to draw the contour plane offset = (zdir == "y" ? ignorenan_maximum : ignorenan_minimum)(mat) # where to draw the contour plane
@ -702,7 +702,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
elseif typeof(z) <: AbstractVector elseif typeof(z) <: AbstractVector
# tri-surface plot (http://matplotlib.org/mpl_toolkits/mplot3d/tutorial.html#tri-surface-plots) # tri-surface plot (http://matplotlib.org/mpl_toolkits/mplot3d/tutorial.html#tri-surface-plots)
handle = ax.plot_trisurf(x, y, z; handle = ax."plot_trisurf"(x, y, z;
label = series[:label], label = series[:label],
zorder = series[:series_plotindex], zorder = series[:series_plotindex],
cmap = py_fillcolormap(series), cmap = py_fillcolormap(series),
@ -727,7 +727,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
else else
z # hopefully it's in a data format that will "just work" with imshow z # hopefully it's in a data format that will "just work" with imshow
end end
handle = ax.imshow(z; handle = ax."imshow"(z;
zorder = series[:series_plotindex], zorder = series[:series_plotindex],
cmap = py_colormap(cgrad([:black, :white])), cmap = py_colormap(cgrad([:black, :white])),
vmin = 0.0, vmin = 0.0,
@ -737,7 +737,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
push!(handles, handle) push!(handles, handle)
# expand extrema... handle is AxesImage object # expand extrema... handle is AxesImage object
xmin, xmax, ymax, ymin = handle.get_extent() xmin, xmax, ymax, ymin = handle."get_extent"()
expand_extrema!(sp, xmin, xmax, ymin, ymax) expand_extrema!(sp, xmin, xmax, ymin, ymax)
# sp[:yaxis].series[:flip] = true # sp[:yaxis].series[:flip] = true
end end
@ -752,7 +752,7 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
discrete_colorbar_values = dvals discrete_colorbar_values = dvals
end end
handle = ax.pcolormesh(x, y, py_mask_nans(z); handle = ax."pcolormesh"(x, y, py_mask_nans(z);
label = series[:label], label = series[:label],
zorder = series[:series_plotindex], zorder = series[:series_plotindex],
cmap = py_fillcolormap(series), cmap = py_fillcolormap(series),
@ -778,14 +778,14 @@ function py_add_series(plt::Plot{PyPlotBackend}, series::Series)
linestyle = py_linestyle(st, get_linestyle(series, i)), linestyle = py_linestyle(st, get_linestyle(series, i)),
fill = true fill = true
) )
push!(handle, ax.add_patch(patches)) push!(handle, ax."add_patch"(patches))
end end
end end
push!(handles, handle) push!(handles, handle)
end end
if st == :pie if st == :pie
handle = ax.pie(y; handle = ax."pie"(y;
# colors = # a vector of colors? # colors = # a vector of colors?
labels = pie_labels(sp, series) labels = pie_labels(sp, series)
)[1] )[1]
@ -852,16 +852,16 @@ function py_set_ticks(ax, ticks, letter)
for dir in (:top,:bottom,:left,:right) for dir in (:top,:bottom,:left,:right)
kw[dir] = kw[Symbol(:label,dir)] = false kw[dir] = kw[Symbol(:label,dir)] = false
end end
axis.set_tick_params(;which="both", kw...) axis."set_tick_params"(;which="both", kw...)
return return
end end
ttype = ticksType(ticks) ttype = ticksType(ticks)
if ttype == :ticks if ttype == :ticks
axis.set_ticks(ticks) axis."set_ticks"(ticks)
elseif ttype == :ticks_and_labels elseif ttype == :ticks_and_labels
axis.set_ticks(ticks[1]) axis."set_ticks"(ticks[1])
axis.set_ticklabels(ticks[2]) axis."set_ticklabels"(ticks[2])
else else
error("Invalid input for $(letter)ticks: $ticks") error("Invalid input for $(letter)ticks: $ticks")
end end
@ -912,12 +912,12 @@ end
function py_set_axis_colors(sp, ax, a::Axis) function py_set_axis_colors(sp, ax, a::Axis)
for (loc, spine) in ax.spines for (loc, spine) in ax.spines
spine.set_color(py_color(a[:foreground_color_border])) spine."set_color"(py_color(a[:foreground_color_border]))
end end
axissym = Symbol(a[:letter], :axis) axissym = Symbol(a[:letter], :axis)
if PyCall.hasproperty(ax, axissym) if PyCall.hasproperty(ax, axissym)
tickcolor = sp[:framestyle] in (:zerolines, :grid) ? py_color(plot_color(a[:foreground_color_grid], a[:gridalpha])) : py_color(a[:foreground_color_axis]) tickcolor = sp[:framestyle] in (:zerolines, :grid) ? py_color(plot_color(a[:foreground_color_grid], a[:gridalpha])) : py_color(a[:foreground_color_axis])
ax.tick_params(axis=string(a[:letter]), which="both", ax."tick_params"(axis=string(a[:letter]), which="both",
colors=tickcolor, colors=tickcolor,
labelcolor=py_color(a[:tickfontcolor])) labelcolor=py_color(a[:tickfontcolor]))
getproperty(ax, axissym).label.set_color(py_color(a[:guidefontcolor])) getproperty(ax, axissym).label.set_color(py_color(a[:guidefontcolor]))
@ -932,14 +932,14 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
# update the fig # update the fig
w, h = plt[:size] w, h = plt[:size]
fig = plt.o fig = plt.o
fig.clear() fig."clear"()
dpi = plt[:dpi] dpi = plt[:dpi]
fig.set_size_inches(w/DPI, h/DPI, forward = true) fig."set_size_inches"(w/DPI, h/DPI, forward = true)
getproperty(fig, set_facecolor_sym)(py_color(plt[:background_color_outside])) getproperty(fig, set_facecolor_sym)(py_color(plt[:background_color_outside]))
fig.set_dpi(plt[:dpi]) fig."set_dpi"(plt[:dpi])
# resize the window # resize the window
PyPlot.plt.get_current_fig_manager().resize(w, h) PyPlot.plt."get_current_fig_manager"().resize(w, h)
# initialize subplots # initialize subplots
for sp in plt.subplots for sp in plt.subplots
@ -973,10 +973,10 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
else else
:title :title
end end
getproperty(ax, func).set_text(sp[:title]) getproperty(ax, func)."set_text"(sp[:title])
getproperty(ax, func).set_fontsize(py_thickness_scale(plt, sp[:titlefontsize])) getproperty(ax, func)."set_fontsize"(py_thickness_scale(plt, sp[:titlefontsize]))
getproperty(ax, func).set_family(sp[:titlefontfamily]) getproperty(ax, func)."set_family"(sp[:titlefontfamily])
getproperty(ax, func).set_color(py_color(sp[:titlefontcolor])) getproperty(ax, func)."set_color"(py_color(sp[:titlefontcolor]))
# ax[:set_title](sp[:title], loc = loc) # ax[:set_title](sp[:title], loc = loc)
end end
@ -996,7 +996,7 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
kw[:boundaries] = vcat(0, kw[:values] + 0.5) kw[:boundaries] = vcat(0, kw[:values] + 0.5)
elseif any(colorbar_series[attr] != nothing for attr in (:line_z, :fill_z, :marker_z)) elseif any(colorbar_series[attr] != nothing for attr in (:line_z, :fill_z, :marker_z))
cmin, cmax = get_clims(sp) cmin, cmax = get_clims(sp)
norm = pycolors.Normalize(vmin = cmin, vmax = cmax) norm = pycolors."Normalize"(vmin = cmin, vmax = cmax)
f = if colorbar_series[:line_z] != nothing f = if colorbar_series[:line_z] != nothing
py_linecolormap py_linecolormap
elseif colorbar_series[:fill_z] != nothing elseif colorbar_series[:fill_z] != nothing
@ -1004,21 +1004,21 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
else else
py_markercolormap py_markercolormap
end end
cmap = pycmap.ScalarMappable(norm = norm, cmap = f(colorbar_series)) cmap = pycmap."ScalarMappable"(norm = norm, cmap = f(colorbar_series))
cmap.set_array([]) cmap."set_array"([])
handle = cmap handle = cmap
end end
# create and store the colorbar object (handle) and the axis that it is drawn on. # create and store the colorbar object (handle) and the axis that it is drawn on.
# note: the colorbar axis is positioned independently from the subplot axis # note: the colorbar axis is positioned independently from the subplot axis
fig = plt.o fig = plt.o
cbax = fig.add_axes([0.8,0.1,0.03,0.8], label = string(gensym())) cbax = fig."add_axes"([0.8,0.1,0.03,0.8], label = string(gensym()))
cb = fig.colorbar(handle; cax = cbax, kw...) cb = fig."colorbar"(handle; cax = cbax, kw...)
cb.set_label(sp[:colorbar_title],size=py_thickness_scale(plt, sp[:yaxis][:guidefontsize]),family=sp[:yaxis][:guidefontfamily], color = py_color(sp[:yaxis][:guidefontcolor])) cb."set_label"(sp[:colorbar_title],size=py_thickness_scale(plt, sp[:yaxis][:guidefontsize]),family=sp[:yaxis][:guidefontfamily], color = py_color(sp[:yaxis][:guidefontcolor]))
for lab in cb.ax.yaxis.get_ticklabels() for lab in cb."ax"."yaxis"."get_ticklabels"()
lab.set_fontsize(py_thickness_scale(plt, sp[:yaxis][:tickfontsize])) lab."set_fontsize"(py_thickness_scale(plt, sp[:yaxis][:tickfontsize]))
lab.set_family(sp[:yaxis][:tickfontfamily]) lab."set_family"(sp[:yaxis][:tickfontfamily])
lab.set_color(py_color(sp[:yaxis][:tickfontcolor])) lab."set_color"(py_color(sp[:yaxis][:tickfontcolor]))
end end
sp.attr[:cbar_handle] = cb sp.attr[:cbar_handle] = cb
sp.attr[:cbar_ax] = cbax sp.attr[:cbar_ax] = cbax
@ -1026,28 +1026,28 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
# framestyle # framestyle
if !ispolar(sp) && !is3d(sp) if !ispolar(sp) && !is3d(sp)
ax.spines["left"].set_linewidth(py_thickness_scale(plt, 1)) ax.spines["left"]."set_linewidth"(py_thickness_scale(plt, 1))
ax.spines["bottom"].set_linewidth(py_thickness_scale(plt, 1)) ax.spines["bottom"]."set_linewidth"(py_thickness_scale(plt, 1))
if sp[:framestyle] == :semi if sp[:framestyle] == :semi
intensity = 0.5 intensity = 0.5
ax.spines["right"].set_alpha(intensity) ax.spines["right"]."set_alpha"(intensity)
ax.spines["top"].set_alpha(intensity) ax.spines["top"]."set_alpha"(intensity)
ax.spines["right"].set_linewidth(py_thickness_scale(plt, intensity)) ax.spines["right"]."set_linewidth"(py_thickness_scale(plt, intensity))
ax.spines["top"].set_linewidth(py_thickness_scale(plt, intensity)) ax.spines["top"]."set_linewidth"(py_thickness_scale(plt, intensity))
elseif sp[:framestyle] in (:axes, :origin) elseif sp[:framestyle] in (:axes, :origin)
ax.spines["right"].set_visible(false) ax.spines["right"]."set_visible"(false)
ax.spines["top"].set_visible(false) ax.spines["top"]."set_visible"(false)
if sp[:framestyle] == :origin if sp[:framestyle] == :origin
ax.spines["bottom"].set_position("zero") ax.spines["bottom"]."set_position"("zero")
ax.spines["left"].set_position("zero") ax.spines["left"]."set_position"("zero")
end end
elseif sp[:framestyle] in (:grid, :none, :zerolines) elseif sp[:framestyle] in (:grid, :none, :zerolines)
for (loc, spine) in ax.spines for (loc, spine) in ax.spines
spine.set_visible(false) spine."set_visible"(false)
end end
if sp[:framestyle] == :zerolines if sp[:framestyle] == :zerolines
ax.axhline(y = 0, color = py_color(sp[:xaxis][:foreground_color_axis]), lw = py_thickness_scale(plt, 0.75)) ax."axhline"(y = 0, color = py_color(sp[:xaxis][:foreground_color_axis]), lw = py_thickness_scale(plt, 0.75))
ax.axvline(x = 0, color = py_color(sp[:yaxis][:foreground_color_axis]), lw = py_thickness_scale(plt, 0.75)) ax."axvline"(x = 0, color = py_color(sp[:yaxis][:foreground_color_axis]), lw = py_thickness_scale(plt, 0.75))
end end
end end
end end
@ -1060,17 +1060,17 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
pyaxis = getproperty(ax, axissym) pyaxis = getproperty(ax, axissym)
if axis[:mirror] && letter != :z if axis[:mirror] && letter != :z
pos = letter == :x ? "top" : "right" pos = letter == :x ? "top" : "right"
pyaxis.set_label_position(pos) # the guides pyaxis."set_label_position"(pos) # the guides
pyaxis.set_ticks_position("both") # the hash marks pyaxis."set_ticks_position"("both") # the hash marks
getproperty(pyaxis, Symbol(:tick_, pos))() # the tick labels getproperty(pyaxis, Symbol(:tick_, pos))() # the tick labels
end end
if axis[:guide_position] != :auto && letter != :z if axis[:guide_position] != :auto && letter != :z
pyaxis.set_label_position(axis[:guide_position]) pyaxis."set_label_position"(axis[:guide_position])
end end
py_set_scale(ax, axis) py_set_scale(ax, axis)
axis[:ticks] != :native ? py_set_lims(ax, axis) : nothing axis[:ticks] != :native ? py_set_lims(ax, axis) : nothing
if ispolar(sp) && letter == :y if ispolar(sp) && letter == :y
ax.set_rlabel_position(90) ax."set_rlabel_position"(90)
end end
ticks = sp[:framestyle] == :none ? nothing : get_ticks(axis) ticks = sp[:framestyle] == :none ? nothing : get_ticks(axis)
# don't show the 0 tick label for the origin framestyle # don't show the 0 tick label for the origin framestyle
@ -1078,28 +1078,28 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
ticks[2][ticks[1] .== 0] .= "" ticks[2][ticks[1] .== 0] .= ""
end end
axis[:ticks] != :native ? py_set_ticks(ax, ticks, letter) : nothing axis[:ticks] != :native ? py_set_ticks(ax, ticks, letter) : nothing
pyaxis.set_tick_params(direction = axis[:tick_direction] == :out ? "out" : "in") pyaxis."set_tick_params"(direction = axis[:tick_direction] == :out ? "out" : "in")
getproperty(ax, Symbol("set_", letter, "label"))(axis[:guide]) getproperty(ax, Symbol("set_", letter, "label"))(axis[:guide])
if get(axis.plotattributes, :flip, false) if get(axis.plotattributes, :flip, false)
getproperty(ax, Symbol("invert_", letter, "axis"))() getproperty(ax, Symbol("invert_", letter, "axis"))()
end end
pyaxis.label.set_fontsize(py_thickness_scale(plt, axis[:guidefontsize])) pyaxis."label"."set_fontsize"(py_thickness_scale(plt, axis[:guidefontsize]))
pyaxis.label.set_family(axis[:guidefontfamily]) pyaxis."label"."set_family"(axis[:guidefontfamily])
for lab in getproperty(ax, Symbol("get_", letter, "ticklabels"))() for lab in getproperty(ax, Symbol("get_", letter, "ticklabels"))()
lab.set_fontsize(py_thickness_scale(plt, axis[:tickfontsize])) lab."set_fontsize"(py_thickness_scale(plt, axis[:tickfontsize]))
lab.set_family(axis[:tickfontfamily]) lab."set_family"(axis[:tickfontfamily])
lab.set_rotation(axis[:rotation]) lab."set_rotation"(axis[:rotation])
end end
if axis[:grid] && !(ticks in (:none, nothing, false)) if axis[:grid] && !(ticks in (:none, nothing, false))
fgcolor = py_color(axis[:foreground_color_grid]) fgcolor = py_color(axis[:foreground_color_grid])
pyaxis.grid(true, pyaxis."grid"(true,
color = fgcolor, color = fgcolor,
linestyle = py_linestyle(:line, axis[:gridstyle]), linestyle = py_linestyle(:line, axis[:gridstyle]),
linewidth = py_thickness_scale(plt, axis[:gridlinewidth]), linewidth = py_thickness_scale(plt, axis[:gridlinewidth]),
alpha = axis[:gridalpha]) alpha = axis[:gridalpha])
ax.set_axisbelow(true) ax."set_axisbelow"(true)
else else
pyaxis.grid(false) pyaxis."grid"(false)
end end
py_set_axis_colors(sp, ax, axis) py_set_axis_colors(sp, ax, axis)
end end
@ -1109,13 +1109,13 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
kw = KW() kw = KW()
for dir in (:top, :bottom) for dir in (:top, :bottom)
if ispolar(sp) if ispolar(sp)
ax.spines."polar".set_visible(false) ax."spines"."polar"."set_visible"(false)
else else
getproperty(ax.spines, dir).set_visible(false) getproperty(ax.spines, dir).set_visible(false)
end end
kw[dir] = kw[Symbol(:label,dir)] = false kw[dir] = kw[Symbol(:label,dir)] = false
end end
ax.xaxis.set_tick_params(; which="both", kw...) ax."xaxis"."set_tick_params"(; which="both", kw...)
end end
if !sp[:yaxis][:showaxis] if !sp[:yaxis][:showaxis]
kw = KW() kw = KW()
@ -1125,20 +1125,20 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
end end
kw[dir] = kw[Symbol(:label,dir)] = false kw[dir] = kw[Symbol(:label,dir)] = false
end end
ax.yaxis.set_tick_params(; which="both", kw...) ax."yaxis"."set_tick_params"(; which="both", kw...)
end end
# aspect ratio # aspect ratio
aratio = sp[:aspect_ratio] aratio = sp[:aspect_ratio]
if aratio != :none if aratio != :none
ax.set_aspect(isa(aratio, Symbol) ? string(aratio) : aratio, anchor = "C") ax."set_aspect"(isa(aratio, Symbol) ? string(aratio) : aratio, anchor = "C")
end end
#camera/view angle #camera/view angle
if is3d(sp) if is3d(sp)
#convert azimuthal to match GR behaviour #convert azimuthal to match GR behaviour
#view_init(elevation, azimuthal) so reverse :camera args #view_init(elevation, azimuthal) so reverse :camera args
ax.view_init((sp[:camera].-(90,0))[end:-1:1]...) ax."view_init"((sp[:camera].-(90,0))[end:-1:1]...)
end end
# legend # legend
@ -1149,8 +1149,8 @@ function _before_layout_calcs(plt::Plot{PyPlotBackend})
# link axes # link axes
x_ax_link, y_ax_link = sp[:xaxis].sps[1].o, sp[:yaxis].sps[1].o x_ax_link, y_ax_link = sp[:xaxis].sps[1].o, sp[:yaxis].sps[1].o
ax != x_ax_link && ax.get_shared_x_axes().join(ax, sp[:xaxis].sps[1].o) ax != x_ax_link && ax."get_shared_x_axes"()."join"(ax, sp[:xaxis].sps[1].o)
ax != y_ax_link && ax.get_shared_y_axes().join(ax, sp[:yaxis].sps[1].o) ax != y_ax_link && ax."get_shared_y_axes"()."join"(ax, sp[:yaxis].sps[1].o)
end end
py_drawfig(fig) py_drawfig(fig)
end end
@ -1181,7 +1181,7 @@ function _update_min_padding!(sp::Subplot{PyPlotBackend})
# optionally add the width of colorbar labels and colorbar to rightpad # optionally add the width of colorbar labels and colorbar to rightpad
if haskey(sp.attr, :cbar_ax) if haskey(sp.attr, :cbar_ax)
bb = py_bbox(sp.attr[:cbar_handle].ax.get_yticklabels()) bb = py_bbox(sp.attr[:cbar_handle]."ax"."get_yticklabels"())
sp.attr[:cbar_width] = _cbar_width + width(bb) + 2.3mm + (sp[:colorbar_title] == "" ? 0px : 30px) sp.attr[:cbar_width] = _cbar_width + width(bb) + 2.3mm + (sp[:colorbar_title] == "" ? 0px : 30px)
rightpad = rightpad + sp.attr[:cbar_width] rightpad = rightpad + sp.attr[:cbar_width]
end end
@ -1202,13 +1202,13 @@ end
function py_add_annotations(sp::Subplot{PyPlotBackend}, x, y, val) function py_add_annotations(sp::Subplot{PyPlotBackend}, x, y, val)
ax = sp.o ax = sp.o
ax.annotate(val, xy = (x,y), zorder = 999) ax."annotate"(val, xy = (x,y), zorder = 999)
end end
function py_add_annotations(sp::Subplot{PyPlotBackend}, x, y, val::PlotText) function py_add_annotations(sp::Subplot{PyPlotBackend}, x, y, val::PlotText)
ax = sp.o ax = sp.o
ax.annotate(val.str, ax."annotate"(val.str,
xy = (x,y), xy = (x,y),
family = val.font.family, family = val.font.family,
color = py_color(val.font.color), color = py_color(val.font.color),
@ -1244,14 +1244,14 @@ function py_add_legend(plt::Plot, sp::Subplot, ax)
if should_add_to_legend(series) if should_add_to_legend(series)
# add a line/marker and a label # add a line/marker and a label
push!(handles, if series[:seriestype] == :shape || series[:fillrange] != nothing push!(handles, if series[:seriestype] == :shape || series[:fillrange] != nothing
pypatches.Patch( pypatches."Patch"(
edgecolor = py_color(get_linecolor(series, clims), get_linealpha(series)), edgecolor = py_color(get_linecolor(series, clims), get_linealpha(series)),
facecolor = py_color(get_fillcolor(series, clims), get_fillalpha(series)), facecolor = py_color(get_fillcolor(series, clims), get_fillalpha(series)),
linewidth = py_thickness_scale(plt, clamp(get_linewidth(series), 0, 5)), linewidth = py_thickness_scale(plt, clamp(get_linewidth(series), 0, 5)),
linestyle = py_linestyle(series[:seriestype], get_linestyle(series)) linestyle = py_linestyle(series[:seriestype], get_linestyle(series))
) )
elseif series[:seriestype] in (:path, :straightline, :scatter) elseif series[:seriestype] in (:path, :straightline, :scatter)
PyPlot.plt.Line2D((0,1),(0,0), PyPlot.plt."Line2D"((0,1),(0,0),
color = py_color(get_linecolor(series, clims), get_linealpha(series)), color = py_color(get_linecolor(series, clims), get_linealpha(series)),
linewidth = py_thickness_scale(plt, clamp(get_linewidth(series), 0, 5)), linewidth = py_thickness_scale(plt, clamp(get_linewidth(series), 0, 5)),
linestyle = py_linestyle(:path, get_linestyle(series)), linestyle = py_linestyle(:path, get_linestyle(series)),
@ -1268,7 +1268,7 @@ function py_add_legend(plt::Plot, sp::Subplot, ax)
# if anything was added, call ax.legend and set the colors # if anything was added, call ax.legend and set the colors
if !isempty(handles) if !isempty(handles)
leg = ax.legend(handles, leg = ax."legend"(handles,
labels, labels,
loc = get(_pyplot_legend_pos, leg, "best"), loc = get(_pyplot_legend_pos, leg, "best"),
scatterpoints = 1, scatterpoints = 1,
@ -1277,13 +1277,13 @@ function py_add_legend(plt::Plot, sp::Subplot, ax)
edgecolor = py_color(sp[:foreground_color_legend]), edgecolor = py_color(sp[:foreground_color_legend]),
framealpha = alpha(plot_color(sp[:background_color_legend])), framealpha = alpha(plot_color(sp[:background_color_legend])),
) )
frame = leg.get_frame() frame = leg."get_frame"()
frame.set_linewidth(py_thickness_scale(plt, 1)) frame."set_linewidth"(py_thickness_scale(plt, 1))
leg.set_zorder(1000) leg."set_zorder"(1000)
sp[:legendtitle] != nothing && leg.set_title(sp[:legendtitle]) sp[:legendtitle] != nothing && leg."set_title"(sp[:legendtitle])
for txt in leg.get_texts() for txt in leg."get_texts"()
PyPlot.plt.setp(txt, color = py_color(sp[:legendfontcolor]), family = sp[:legendfontfamily]) PyPlot.plt."setp"(txt, color = py_color(sp[:legendfontcolor]), family = sp[:legendfontfamily])
end end
end end
end end
@ -1301,7 +1301,7 @@ function _update_plot_object(plt::Plot{PyPlotBackend})
figw, figh = sp.plt[:size] figw, figh = sp.plt[:size]
figw, figh = figw*px, figh*px figw, figh = figw*px, figh*px
pcts = bbox_to_pcts(sp.plotarea, figw, figh) pcts = bbox_to_pcts(sp.plotarea, figw, figh)
ax.set_position(pcts) ax."set_position"(pcts)
# set the cbar position if there is one # set the cbar position if there is one
if haskey(sp.attr, :cbar_ax) if haskey(sp.attr, :cbar_ax)
@ -1311,7 +1311,7 @@ function _update_plot_object(plt::Plot{PyPlotBackend})
has_toplabel = !(1e-7 < max(abs(ex.emax), abs(ex.emin)) < 1e7) has_toplabel = !(1e-7 < max(abs(ex.emax), abs(ex.emin)) < 1e7)
cb_bbox = BoundingBox(right(sp.bbox)-cbw+1mm, top(sp.bbox) + (has_toplabel ? 4mm : 2mm), _cbar_width-1mm, height(sp.bbox) - (has_toplabel ? 6mm : 4mm)) cb_bbox = BoundingBox(right(sp.bbox)-cbw+1mm, top(sp.bbox) + (has_toplabel ? 4mm : 2mm), _cbar_width-1mm, height(sp.bbox) - (has_toplabel ? 6mm : 4mm))
pcts = bbox_to_pcts(cb_bbox, figw, figh) pcts = bbox_to_pcts(cb_bbox, figw, figh)
sp.attr[:cbar_ax].set_position(pcts) sp.attr[:cbar_ax]."set_position"(pcts)
end end
end end
PyPlot.draw() PyPlot.draw()
@ -1320,7 +1320,7 @@ end
# ----------------------------------------------------------------- # -----------------------------------------------------------------
# display/output # display/output
_display(plt::Plot{PyPlotBackend}) = plt.o.show() _display(plt::Plot{PyPlotBackend}) = plt.o."show"()
@ -1338,16 +1338,16 @@ const _pyplot_mimeformats = Dict(
for (mime, fmt) in _pyplot_mimeformats for (mime, fmt) in _pyplot_mimeformats
@eval function _show(io::IO, ::MIME{Symbol($mime)}, plt::Plot{PyPlotBackend}) @eval function _show(io::IO, ::MIME{Symbol($mime)}, plt::Plot{PyPlotBackend})
fig = plt.o fig = plt.o
fig.canvas.print_figure( fig."canvas"."print_figure"(
io, io,
format=$fmt, format=$fmt,
# bbox_inches = "tight", # bbox_inches = "tight",
# figsize = map(px2inch, plt[:size]), # figsize = map(px2inch, plt[:size]),
facecolor = fig.get_facecolor(), facecolor = fig."get_facecolor"(),
edgecolor = "none", edgecolor = "none",
dpi = plt[:dpi] dpi = plt[:dpi]
) )
end end
end end
closeall(::PyPlotBackend) = PyPlot.plt.close("all") closeall(::PyPlotBackend) = PyPlot.plt."close"("all")