def create_waterfall_plots(figures: dict, opt: DotDict) -> None:
""" Main loop for waterfall plot creation. """
LOG.debug(f" ...creating Waterfall Plot")
for fig_id, fig_cont in figures.items():
LOG.debug(f' Plotting Figure: {fig_id}.')
fig_cont.fig.canvas.set_window_title(fig_id)
_plot_waterfall(fig_cont, opt.line_width, opt.cmap,
opt.common_plane_colors)
plot_lines(fig_cont, opt.lines)
_format_axes(fig_cont, opt.limits, opt.ncol_legend)
output_plot(fig_cont)
if opt.show:
plt.show()
def create_stem_plots(figures: dict, opt: DotDict) -> None:
""" Main loop for stem-plot creation. """
LOG.debug(f" ...creating Stem Plots")
for fig_id, fig_cont in figures.items():
LOG.debug(f' Plotting Figure: {fig_id}.')
fig_cont.fig.canvas.set_window_title(fig_id)
_plot_stems(fig_cont)
plot_lines(fig_cont, opt.lines)
_format_axes(fig_cont, opt.limits)
if opt.ncol_legend > 0:
_create_legend(fig_cont.axes[0], fig_cont.data.keys(), opt.lines, opt.ncol_legend)
output_plot(fig_cont)
if opt.show:
plt.show()
def _create_plots(fig_collection, opt):
"""Main plotting routine."""
for fig_container in fig_collection.figs.values():
for idx_ax, ax_id in enumerate(fig_container.axes_ids):
ax, data, xlabel, ylabel = fig_container[ax_id]
_plot_vlines(ax, opt.vertical_lines)
_plot_data(ax, data, opt.change_marker, opt.errorbar_alpha)
_set_axes_layout(ax, opt.x_lim, opt.y_lim, ylabel, xlabel)
# plt.show(block=False) # for debugging
if idx_ax == 0 or not opt.single_legend:
pannot.make_top_legend(ax, opt.ncol_legend)
output_plot(fig_container)
if opt.show:
plt.show()
