mirror of
https://github.com/wxWidgets/Phoenix.git
synced 2025-12-16 09:40:07 +01:00
1011 lines
38 KiB
Python
1011 lines
38 KiB
Python
# -*- coding: utf-8 -*-
|
|
# pylint: disable=E1101, C0330, C0103
|
|
# E1101: Module X has no Y member
|
|
# C0330: Wrong continued indentation
|
|
# C0103: Invalid attribute/variable/method name
|
|
"""
|
|
demo.py
|
|
=======
|
|
|
|
.. highlight:: shell
|
|
|
|
This is a demo showing some of the capabilities of the :mod:`wx.lib.plot`
|
|
package.
|
|
|
|
It is intended to be run as a standalone script via::
|
|
|
|
user@host:.../site-packages/wx/lib/plot$ python examples/demo.py
|
|
|
|
"""
|
|
__docformat__ = "restructuredtext en"
|
|
|
|
# Third Party
|
|
import wx
|
|
from wx.lib import plot as wxplot
|
|
|
|
# Needs NumPy
|
|
try:
|
|
import numpy as np
|
|
except ImportError:
|
|
msg = """
|
|
This module requires the NumPy module, which could not be
|
|
imported. It probably is not installed (it's not part of the
|
|
standard Python distribution). See the Numeric Python site
|
|
(http://numpy.scipy.org) for information on downloading source or
|
|
binaries, or just try `pip install numpy` and it will probably
|
|
work."""
|
|
raise ImportError("NumPy not found.\n" + msg)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
### Drawing Functions
|
|
# ---------------------------------------------------------------------------
|
|
def _draw1Objects():
|
|
"""Sin, Cos, and Points"""
|
|
# 100 points sin function, plotted as green circles
|
|
data1 = 2. * np.pi * np.arange(-200, 200) / 200.
|
|
data1.shape = (200, 2)
|
|
data1[:, 1] = np.sin(data1[:, 0])
|
|
markers1 = wxplot.PolyMarker(data1,
|
|
legend='Green Markers',
|
|
colour='green',
|
|
marker='circle',
|
|
size=1,
|
|
)
|
|
|
|
# 50 points cos function, plotted as red line and markers
|
|
data1 = 2. * np.pi * np.arange(-100, 100) / 100.
|
|
data1.shape = (100, 2)
|
|
data1[:, 1] = np.cos(data1[:, 0])
|
|
lines = wxplot.PolySpline(data1, legend='Red Line', colour='red')
|
|
markers3 = wxplot.PolyMarker(data1,
|
|
legend='Red Dot',
|
|
colour='red',
|
|
marker='circle',
|
|
size=1,
|
|
)
|
|
|
|
# A few more points...
|
|
pi = np.pi
|
|
pts = [(0., 0.), (pi / 4., 1.), (pi / 2, 0.), (3. * pi / 4., -1)]
|
|
markers2 = wxplot.PolyMarker(pts,
|
|
legend='Cross Legend',
|
|
colour='blue',
|
|
marker='cross',
|
|
)
|
|
line2 = wxplot.PolyLine(pts, drawstyle='steps-post')
|
|
|
|
return wxplot.PlotGraphics([markers1, lines, markers3, markers2, line2],
|
|
"Graph Title",
|
|
"X Axis",
|
|
"Y Axis",
|
|
)
|
|
|
|
|
|
def _draw2Objects():
|
|
"""Sin, Cos, Points, and lines between points"""
|
|
# 100 points sin function, plotted as green dots
|
|
data1 = 2. * np.pi * np.arange(200) / 200.
|
|
data1.shape = (100, 2)
|
|
data1[:, 1] = np.sin(data1[:, 0])
|
|
line1 = wxplot.PolySpline(data1,
|
|
legend='Green Line',
|
|
colour='green',
|
|
width=6,
|
|
style=wx.PENSTYLE_DOT)
|
|
|
|
# 25 points cos function, plotted as red dot-dash with steps.
|
|
data1 = 2. * np.pi * np.arange(50) / 50.
|
|
data1.shape = (25, 2)
|
|
data1[:, 1] = np.cos(data1[:, 0])
|
|
line2 = wxplot.PolyLine(data1,
|
|
legend='Red Line',
|
|
colour='red',
|
|
width=2,
|
|
style=wx.PENSTYLE_DOT_DASH,
|
|
drawstyle='steps-post',
|
|
)
|
|
|
|
# data points for the 25pt cos function.
|
|
pts2 = wxplot.PolyMarker(data1,
|
|
legend='Red Points',
|
|
colour='red',
|
|
size=1.5,
|
|
)
|
|
|
|
# A few more points...
|
|
pi = np.pi
|
|
pts = [(0., 0.), (pi / 4., 1.), (pi / 2, 0.), (3. * pi / 4., -1)]
|
|
markers1 = wxplot.PolyMarker(pts,
|
|
legend='Cross Hatch Square',
|
|
colour='blue',
|
|
width=3,
|
|
size=6,
|
|
fillcolour='red',
|
|
fillstyle=wx.CROSSDIAG_HATCH,
|
|
marker='square',
|
|
)
|
|
marker_line = wxplot.PolyLine(pts,
|
|
legend='Cross Hatch Square',
|
|
colour='blue',
|
|
width=3,
|
|
)
|
|
|
|
return wxplot.PlotGraphics([markers1, line1, line2, pts2, marker_line],
|
|
"Big Markers with Different Line Styles")
|
|
|
|
|
|
def _draw3Objects():
|
|
"""Various Marker Types"""
|
|
markerList = ['circle', 'dot', 'square', 'triangle', 'triangle_down',
|
|
'cross', 'plus', 'circle']
|
|
m = []
|
|
for i in range(len(markerList)):
|
|
m.append(wxplot.PolyMarker([(2 * i + .5, i + .5)],
|
|
legend=markerList[i],
|
|
colour='blue',
|
|
marker=markerList[i])
|
|
)
|
|
return wxplot.PlotGraphics(m,
|
|
title="Selection of Markers",
|
|
xLabel="Minimal Axis",
|
|
yLabel="No Axis")
|
|
|
|
|
|
def _draw4Objects():
|
|
"""25,000 point line and markers"""
|
|
# Points
|
|
data1 = np.random.normal(loc=7.5e5, scale=70000, size=50000)
|
|
data1.shape = (25000, 2)
|
|
markers2 = wxplot.PolyMarker(data1,
|
|
legend='Dots',
|
|
colour='blue',
|
|
marker='square',
|
|
size=1,
|
|
)
|
|
|
|
# Line
|
|
data1 = np.arange(5e5, 1e6, 10)
|
|
data1.shape = (25000, 2)
|
|
line1 = wxplot.PolyLine(data1, legend='Wide Line', colour='green', width=4)
|
|
|
|
return wxplot.PlotGraphics([markers2, line1],
|
|
"25,000 Points",
|
|
"Value X",
|
|
"")
|
|
|
|
|
|
def _draw5Objects():
|
|
"""Empty graph with axes but no points"""
|
|
points = []
|
|
line1 = wxplot.PolyLine(points,
|
|
legend='Wide Line',
|
|
colour='green',
|
|
width=5)
|
|
return wxplot.PlotGraphics([line1],
|
|
"Empty Plot With Just Axes",
|
|
"Value X",
|
|
"Value Y")
|
|
|
|
|
|
def _draw6Objects():
|
|
"""Faking a Bar graph"""
|
|
points1 = [(1, 0), (1, 10)]
|
|
line1 = wxplot.PolyLine(points1, colour='green', legend='Feb.', width=10)
|
|
points1g = [(2, 0), (2, 4)]
|
|
line1g = wxplot.PolyLine(points1g, colour='red', legend='Mar.', width=10)
|
|
points1b = [(3, 0), (3, 6)]
|
|
line1b = wxplot.PolyLine(points1b, colour='blue', legend='Apr.', width=10)
|
|
|
|
points2 = [(4, 0), (4, 12)]
|
|
line2 = wxplot.PolyLine(points2, colour='Yellow', legend='May', width=10)
|
|
points2g = [(5, 0), (5, 8)]
|
|
line2g = wxplot.PolyLine(points2g,
|
|
colour='orange',
|
|
legend='June',
|
|
width=10)
|
|
points2b = [(6, 0), (6, 4)]
|
|
line2b = wxplot.PolyLine(points2b, colour='brown', legend='July', width=10)
|
|
|
|
return wxplot.PlotGraphics([line1, line1g, line1b, line2, line2g, line2b],
|
|
"Bar Graph - (Turn on Grid, Legend)",
|
|
"Months",
|
|
"Number of Students")
|
|
|
|
|
|
def _draw7Objects():
|
|
"""Log10 on both axes"""
|
|
x = np.arange(-1000, 1000, 1)
|
|
y1 = 4.5 * x ** 2
|
|
y2 = 2.2 * x ** 3
|
|
points1 = np.transpose([x, y1])
|
|
points2 = np.transpose([x, y2])
|
|
line1 = wxplot.PolyLine(points1,
|
|
legend='quadratic',
|
|
colour='blue',
|
|
width=1)
|
|
line2 = wxplot.PolyLine(points2, legend='cubic', colour='red', width=1)
|
|
return wxplot.PlotGraphics([line1, line2],
|
|
"double log plot",
|
|
"Value X",
|
|
"Value Y")
|
|
|
|
|
|
def _draw8Objects():
|
|
"""
|
|
Box plot
|
|
"""
|
|
data1 = np.array([np.nan, 337, 607, 583, 512, 531, 558, 381, 621, 574,
|
|
538, 577, 679, 415, 454, 417, 635, 319, 350, 183,
|
|
863, 337, 607, 583, 512, 531, 558, 381, 621, 574,
|
|
538, 577, 679, 415, 454, 417, 635, 319, 350, 97])
|
|
data2 = np.array([912, 337, 607, 583, 512, 531, 558, 381, 621, 574,
|
|
538, 532, 829, 82, 454, 417, 635, 319, 350, 183,
|
|
863, 337, 607, 583, 512, 531, 558, 866, 621, 574,
|
|
538, 577, 679, 415, 326, 417, 635, 319, 350, 97])
|
|
|
|
data2 = data2 * 0.9
|
|
|
|
data1 = np.array([(0, x) for x in data1])
|
|
data2 = np.array([(1, x) for x in data2])
|
|
|
|
data3 = np.random.gamma(2, 2, 500) * 30 + 100
|
|
data3 = np.array([(2, x) for x in data3])
|
|
|
|
boxplot = wxplot.BoxPlot(data1, legend="0.0: Weights")
|
|
boxplot2 = wxplot.BoxPlot(data2, legend="1.0: Heights")
|
|
boxplot3 = wxplot.BoxPlot(data3, legend="2.0: GammaDistribution")
|
|
return wxplot.PlotGraphics([boxplot, boxplot2, boxplot3],
|
|
"Box Plot",
|
|
"",
|
|
"Value")
|
|
|
|
def _draw9Objects():
|
|
"""
|
|
Histogram
|
|
|
|
The following must be true:
|
|
len(bspec) = len(hist_data) + 1
|
|
"""
|
|
|
|
data = 2.25 * np.random.randn(50) + 3
|
|
heights, bins = np.histogram(data, bins=10)
|
|
|
|
hist_fixed_binsize = wxplot.PolyHistogram(heights, bins)
|
|
|
|
# Bins can also be arbitrary widths:
|
|
hist_variable_binsize = wxplot.PolyHistogram(
|
|
[2, 3, 4, 6, 3],
|
|
[18, 19, 22, 25, 26, 27],
|
|
fillcolour=wx.BLUE,
|
|
edgewidth=2,
|
|
edgecolour=wx.RED,
|
|
)
|
|
|
|
return wxplot.PlotGraphics(
|
|
[hist_fixed_binsize, hist_variable_binsize],
|
|
"Histogram with fixed binsize (left) and variable binsize (right)",
|
|
"Value",
|
|
"Count",
|
|
)
|
|
|
|
|
|
def _draw10Objects():
|
|
"""
|
|
A real bar graph
|
|
"""
|
|
bar_height = np.array([1, 5, 8, 16, 12, 15, 18, 23, 4, 7, 9, 6])
|
|
bar_location = np.array([0, 2, 4, 6, 8, 10, 11, 12, 16, 20, 30])
|
|
data = list(zip(bar_location, bar_height))
|
|
bars = [wxplot.PolyBars(data)]
|
|
|
|
return wxplot.PlotGraphics(bars, "Histogram", "XValue", "Count")
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
### Demo Application
|
|
# ---------------------------------------------------------------------------
|
|
class PlotDemoApp(object):
|
|
def __init__(self):
|
|
self.app = wx.App()
|
|
self.frame = PlotDemoMainFrame(None, -1, "PlotCanvas")
|
|
self.frame.Show(True)
|
|
self.app.MainLoop()
|
|
|
|
|
|
class PlotDemoMainFrame(wx.Frame):
|
|
# -----------------------------------------------------------------------
|
|
### UI Initialization
|
|
# -----------------------------------------------------------------------
|
|
def __init__(self, parent, wxid, title):
|
|
wx.Frame.__init__(self, parent, wxid, title,
|
|
wx.DefaultPosition, (800, 600))
|
|
|
|
# Now Create the menu bar and items
|
|
self.mainmenu = wx.MenuBar()
|
|
|
|
self._init_file_menu()
|
|
self._init_plot_menu()
|
|
self._init_options_menu()
|
|
self._init_help_menu()
|
|
|
|
self.SetMenuBar(self.mainmenu)
|
|
|
|
# A status bar to tell people what's happening
|
|
self.CreateStatusBar(1)
|
|
|
|
self.client = wxplot.PlotCanvas(self)
|
|
# define the function for drawing pointLabels
|
|
# self.client.SetPointLabelFunc(self.DrawPointLabel)
|
|
self.client.pointLabelFunc = self.DrawPointLabel
|
|
# Create mouse event for showing cursor coords in status bar
|
|
self.client.canvas.Bind(wx.EVT_LEFT_DOWN, self.OnMouseLeftDown)
|
|
# Show closest point when enabled
|
|
self.client.canvas.Bind(wx.EVT_MOTION, self.OnMotion)
|
|
|
|
self.OnPlotDraw1(None)
|
|
|
|
wx.CallAfter(wx.MessageBox,
|
|
"Various plot types can be shown using the Plot menu. " +
|
|
"Check out the Options menu too.",
|
|
"wx.lib.plot Demo")
|
|
|
|
|
|
def _init_file_menu(self):
|
|
""" Create the "File" menu items. """
|
|
menu = wx.Menu()
|
|
menu.Append(200, 'Page Setup...', 'Setup the printer page')
|
|
self.Bind(wx.EVT_MENU, self.OnFilePageSetup, id=200)
|
|
menu.Append(201, 'Print Preview...', 'Show the current plot on page')
|
|
self.Bind(wx.EVT_MENU, self.OnFilePrintPreview, id=201)
|
|
menu.Append(202, 'Print...', 'Print the current plot')
|
|
self.Bind(wx.EVT_MENU, self.OnFilePrint, id=202)
|
|
menu.Append(203, 'Save Plot...', 'Save current plot')
|
|
self.Bind(wx.EVT_MENU, self.OnSaveFile, id=203)
|
|
menu.Append(wx.ID_EXIT, 'E&xit', 'Enough of this already!')
|
|
self.Bind(wx.EVT_MENU, self.OnFileExit, id=wx.ID_EXIT)
|
|
self.mainmenu.Append(menu, '&File')
|
|
|
|
def _init_plot_menu(self):
|
|
""" Create the "Plot" menu items. """
|
|
menu = wx.Menu()
|
|
menu.Append(206, 'Draw1 - sin, cos',
|
|
'Draw Sin and Cos curves')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw1, id=206)
|
|
menu.Append(207, 'Draw2 - sin, cos, large joined markers',
|
|
'Draw Sin and Cos curves with some large joined makers')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw2, id=207)
|
|
menu.Append(208, 'Draw3 - various markers',
|
|
'Demo various markers')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw3, id=208)
|
|
menu.Append(209, 'Draw4 - 25k pts',
|
|
'Example of drawing many points quickly')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw4, id=209)
|
|
menu.Append(210, 'Draw5 - empty plot',
|
|
'An empty plot')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw5, id=210)
|
|
menu.Append(260, 'Draw6 - fake bar graph via lines',
|
|
'Fake a bar graph by using lines.')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw6, id=260)
|
|
menu.Append(261, 'Draw7 - log-log',
|
|
'Plot a Log-Log graph')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw7, id=261)
|
|
menu.Append(262, 'Draw8 - Box Plots',
|
|
'Show off some box plots')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw8, id=262)
|
|
menu.Append(263, 'Draw9 - Histogram',
|
|
'Plot a histogram')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw9, id=263)
|
|
menu.Append(264, 'Draw10 - real bar graph',
|
|
'Plot a real bar chart (not faked with lines)')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotDraw10, id=264)
|
|
|
|
menu.AppendSeparator()
|
|
|
|
menu.Append(211, '&Redraw', 'Redraw plots')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotRedraw, id=211)
|
|
menu.Append(212, '&Clear', 'Clear canvas')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotClear, id=212)
|
|
menu.Append(213, '&Scale', 'Scale canvas')
|
|
self.Bind(wx.EVT_MENU, self.OnPlotScale, id=213)
|
|
|
|
menu.AppendSeparator()
|
|
|
|
menu.Append(225, 'Scroll Up 1', 'Move View Up 1 Unit')
|
|
self.Bind(wx.EVT_MENU, self.OnScrUp, id=225)
|
|
menu.Append(230, 'Scroll Rt 2', 'Move View Right 2 Units')
|
|
self.Bind(wx.EVT_MENU, self.OnScrRt, id=230)
|
|
menu.Append(235, '&Plot Reset', 'Reset to original plot')
|
|
self.Bind(wx.EVT_MENU, self.OnReset, id=235)
|
|
|
|
self.mainmenu.Append(menu, '&Plot')
|
|
|
|
|
|
def _init_options_menu(self):
|
|
""" Create the "Options" menu items. """
|
|
menu = wx.Menu()
|
|
|
|
menu.Append(214, 'Enable &Zoom',
|
|
'Enable Mouse Zoom', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableZoom, id=214)
|
|
|
|
menu.Append(217, 'Enable &Drag',
|
|
'Activates dragging mode', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableDrag, id=217)
|
|
|
|
item = menu.Append(-1, 'Enable &Scrollbars (if needed)',
|
|
'Enable Scrollbars (if needed)', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableScrollbars, item)
|
|
|
|
menu.Append(222, 'Enable &Point Label',
|
|
'Show Closest Point', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnablePointLabel, id=222)
|
|
|
|
menu.Append(223, 'Enable &Anti-Aliasing',
|
|
'Smooth output', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAntiAliasing, id=223)
|
|
|
|
menu.Append(224, 'Enable &High-Resolution AA',
|
|
'Draw in higher resolution', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableHiRes, id=224)
|
|
|
|
menu.AppendSeparator()
|
|
|
|
menu.Append(226, 'Enable Center Lines',
|
|
'Draw center lines', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableCenterLines, id=226)
|
|
|
|
menu.Append(227, 'Enable Diagonal Lines',
|
|
'Draw diagonal lines', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableDiagonals, id=227)
|
|
|
|
menu.Append(220, 'Enable &Legend',
|
|
'Turn on Legend', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableLegend, id=220)
|
|
|
|
### SubMenu for Grid
|
|
submenu = wx.Menu()
|
|
self.gridSubMenu = submenu
|
|
|
|
submenu.AppendCheckItem(2151, "X Gridlines", "Enable X gridlines")
|
|
submenu.AppendCheckItem(2152, "Y Gridlines", "Enable Y gridlines")
|
|
submenu.AppendCheckItem(2153, "All Gridlines", "Enable All gridlines")
|
|
submenu.Check(2151, True)
|
|
submenu.Check(2152, True)
|
|
submenu.Check(2153, True)
|
|
|
|
self.Bind(wx.EVT_MENU, self.OnEnableGridX, id=2151)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableGridY, id=2152)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableGridAll, id=2153)
|
|
|
|
menu.AppendSubMenu(submenu, 'Enable Grid', 'Turn on Grid')
|
|
|
|
### SubMenu for Axes
|
|
submenu = wx.Menu()
|
|
submenu_items = ("Bottom", "Left", "Top", "Right",
|
|
"Bottom+Left", "All")
|
|
self.axesSubMenu = submenu
|
|
for _i, item in enumerate(submenu_items, 2401):
|
|
submenu.AppendCheckItem(_i, item, "Enables {} axis".format(item))
|
|
submenu.Check(_i, True)
|
|
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesBottom, id=2401)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesLeft, id=2402)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesTop, id=2403)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesRight, id=2404)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesBottomLeft, id=2405)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesAll, id=2406)
|
|
|
|
menu.AppendSubMenu(submenu, 'Enable Axes', 'Enables the display of the Axes')
|
|
|
|
submenu = wx.Menu()
|
|
submenu_items = ("Bottom", "Left", "Top", "Right")
|
|
help_txt = "Enables {} axis values"
|
|
self.axesValuesSubMenu = submenu
|
|
for _i, item in enumerate(submenu_items, 2451):
|
|
submenu.AppendCheckItem(_i, item, help_txt.format(item))
|
|
|
|
submenu.Check(2451, True)
|
|
submenu.Check(2452, True)
|
|
submenu.Check(2453, False)
|
|
submenu.Check(2454, False)
|
|
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesValuesBottom, id=2451)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesValuesLeft, id=2452)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesValuesTop, id=2453)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesValuesRight, id=2454)
|
|
|
|
menu.AppendSubMenu(submenu, 'Enable Axes Values', 'Enables the display of the axes values')
|
|
|
|
submenu = wx.Menu()
|
|
submenu_items = ("Bottom", "Left", "Top", "Right",
|
|
"Bottom+Left", "All")
|
|
help_txt = "Enables {} ticks"
|
|
self.ticksSubMenu = submenu
|
|
for _i, item in enumerate(submenu_items, 2501):
|
|
submenu.AppendCheckItem(_i, item, help_txt.format(item))
|
|
|
|
submenu.Check(2501, False)
|
|
submenu.Check(2502, False)
|
|
submenu.Check(2503, False)
|
|
submenu.Check(2504, False)
|
|
|
|
self.Bind(wx.EVT_MENU, self.OnEnableTicksBottom, id=2501)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableTicksLeft, id=2502)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableTicksTop, id=2503)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableTicksRight, id=2504)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableTicksBottomLeft, id=2505)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableTicksAll, id=2506)
|
|
|
|
menu.AppendSubMenu(submenu, 'Enable Ticks','Enables the display of the ticks')
|
|
|
|
menu.Append(255, 'Enable Plot Title',
|
|
'Enables the plot title', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnablePlotTitle, id=255)
|
|
menu.Check(255, True)
|
|
|
|
menu.Append(270, 'Enable Axes Labels',
|
|
'Enables the X and Y axes labels', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnEnableAxesLabels, id=270)
|
|
menu.Check(270, True)
|
|
|
|
menu.Append(271, 'Enable Log-Y',
|
|
'Changes the Y axis to log10 scale', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnLogY, id=271)
|
|
menu.Append(272, 'Enable Log-X',
|
|
'Changes the X axis to log10 scale', kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnLogX, id=272)
|
|
|
|
menu.Append(273, 'Enable Abs(X)',
|
|
'Applies absolute value transform to X axis',
|
|
kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnAbsX, id=273)
|
|
menu.Append(274, 'Enable Abs(Y)',
|
|
'Applies absolute value transform to Y axis',
|
|
kind=wx.ITEM_CHECK)
|
|
self.Bind(wx.EVT_MENU, self.OnAbsY, id=274)
|
|
|
|
menu.AppendSeparator()
|
|
|
|
menu.Append(231, 'Set Gray Background',
|
|
'Change background colour to gray')
|
|
self.Bind(wx.EVT_MENU, self.OnBackgroundGray, id=231)
|
|
menu.Append(232, 'Set &White Background',
|
|
'Change background colour to white')
|
|
self.Bind(wx.EVT_MENU, self.OnBackgroundWhite, id=232)
|
|
menu.Append(233, 'Set Red Label Text',
|
|
'Change label text colour to red')
|
|
self.Bind(wx.EVT_MENU, self.OnForegroundRed, id=233)
|
|
menu.Append(234, 'Set &Black Label Text',
|
|
'Change label text colour to black')
|
|
self.Bind(wx.EVT_MENU, self.OnForegroundBlack, id=234)
|
|
|
|
self.mainmenu.Append(menu, '&Options')
|
|
|
|
self.plot_options_menu = menu
|
|
|
|
def _init_help_menu(self):
|
|
""" Create the "Help" menu items. """
|
|
menu = wx.Menu()
|
|
menu.Append(wx.ID_ABOUT, '&About', 'About this thing...')
|
|
self.Bind(wx.EVT_MENU, self.OnHelpAbout, id=wx.ID_ABOUT)
|
|
self.mainmenu.Append(menu, '&Help')
|
|
|
|
# -----------------------------------------------------------------------
|
|
### Event Handling
|
|
# -----------------------------------------------------------------------
|
|
|
|
|
|
def OnMouseLeftDown(self, event):
|
|
s = "Left Mouse Down at Point: (%.4f, %.4f)" % self.client.GetXY(
|
|
event)
|
|
self.SetStatusText(s)
|
|
event.Skip() # allows plotCanvas OnMouseLeftDown to be called
|
|
|
|
def OnMotion(self, event):
|
|
# show closest point (when enbled)
|
|
if self.client.enablePointLabel:
|
|
# make up dict with info for the pointLabel
|
|
# I've decided to mark the closest point on the closest curve
|
|
dlst = self.client.GetClosestPoint(
|
|
self.client.GetXY(event),
|
|
pointScaled=True,
|
|
)
|
|
if dlst != []: # returns [] if none
|
|
curveNum, legend, pIndex, pointXY, scaledXY, distance = dlst
|
|
# make up dictionary to pass to my user function (see
|
|
# DrawPointLabel)
|
|
mDataDict = {"curveNum": curveNum,
|
|
"legend": legend,
|
|
"pIndex": pIndex,
|
|
"pointXY": pointXY,
|
|
"scaledXY": scaledXY}
|
|
# pass dict to update the pointLabel
|
|
self.client.UpdatePointLabel(mDataDict)
|
|
event.Skip() # go to next handler
|
|
|
|
def OnFilePageSetup(self, event):
|
|
self.client.PageSetup()
|
|
|
|
def OnFilePrintPreview(self, event):
|
|
self.client.PrintPreview()
|
|
|
|
def OnFilePrint(self, event):
|
|
self.client.Printout()
|
|
|
|
def OnSaveFile(self, event):
|
|
self.client.SaveFile()
|
|
|
|
def OnFileExit(self, event):
|
|
self.Close()
|
|
|
|
|
|
# -----------------------------------------------------------------------
|
|
### PlotDraw Events
|
|
# -----------------------------------------------------------------------
|
|
def OnPlotDraw1(self, event):
|
|
""" Sin, Cos, and Points """
|
|
self.resetDefaults()
|
|
self.client.Draw(_draw1Objects())
|
|
|
|
def OnPlotDraw2(self, event):
|
|
""" Sin, Cos, Points, and lines between points """
|
|
self.resetDefaults()
|
|
self.client.Draw(_draw2Objects())
|
|
|
|
def OnPlotDraw3(self, event):
|
|
""" Various Marker Types """
|
|
self.resetDefaults()
|
|
self.client.SetFont(wx.Font(10,
|
|
wx.FONTFAMILY_SCRIPT,
|
|
wx.FONTSTYLE_NORMAL,
|
|
wx.FONTWEIGHT_NORMAL)
|
|
)
|
|
self.client.fontSizeAxis = 20
|
|
self.client.fontSizeLegend = 12
|
|
self.client.xSpec = 'min'
|
|
self.client.ySpec = 'none'
|
|
self.client.Draw(_draw3Objects())
|
|
|
|
def OnPlotDraw4(self, event):
|
|
""" 25,000 point line and markers """
|
|
self.resetDefaults()
|
|
drawObj = _draw4Objects()
|
|
self.client.Draw(drawObj)
|
|
# profile
|
|
# start = _time.perf_counter()
|
|
# for x in range(10):
|
|
# self.client.Draw(drawObj)
|
|
## print("10x of Draw4 took: %f sec."%(_time.perf_counter() - start))
|
|
# profile end
|
|
|
|
def OnPlotDraw5(self, event):
|
|
""" Empty plot with just axes """
|
|
self.resetDefaults()
|
|
drawObj = _draw5Objects()
|
|
# make the axis X= (0,5), Y=(0,10)
|
|
# (default with None is X= (-1,1), Y= (-1,1))
|
|
self.client.Draw(drawObj, xAxis=(0, 5), yAxis=(0, 10))
|
|
|
|
def OnPlotDraw6(self, event):
|
|
""" Bar Graph Example """
|
|
self.resetDefaults()
|
|
# self.client.SetEnableLegend(True) #turn on Legend
|
|
# self.client.SetEnableGrid(True) #turn on Grid
|
|
self.client.xSpec = 'none'
|
|
self.client.ySpec = 'auto'
|
|
self.client.Draw(_draw6Objects(), xAxis=(0, 7))
|
|
|
|
def OnPlotDraw7(self, event):
|
|
""" log scale example """
|
|
self.resetDefaults()
|
|
self.plot_options_menu.Check(271, True)
|
|
self.plot_options_menu.Check(272, True)
|
|
self.client.logScale = (True, True)
|
|
self.client.Draw(_draw7Objects())
|
|
|
|
def OnPlotDraw8(self, event):
|
|
""" Box Plot example """
|
|
self.resetDefaults()
|
|
self.client.Draw(_draw8Objects())
|
|
|
|
def OnPlotDraw9(self, event):
|
|
""" Histogram example """
|
|
self.resetDefaults()
|
|
self.client.Draw(_draw9Objects())
|
|
|
|
def OnPlotDraw10(self, event):
|
|
""" Bar Chart example """
|
|
self.resetDefaults()
|
|
self.client.Draw(_draw10Objects())
|
|
|
|
def OnPlotRedraw(self, event):
|
|
self.client.Redraw()
|
|
|
|
def OnPlotClear(self, event):
|
|
self.client.Clear()
|
|
|
|
def OnPlotScale(self, event):
|
|
if self.client.last_draw is not None:
|
|
graphics, xAxis, yAxis = self.client.last_draw
|
|
self.client.Draw(graphics, (1, 3.05), (0, 1))
|
|
|
|
# -----------------------------------------------------------------------
|
|
### Other Events
|
|
# -----------------------------------------------------------------------
|
|
def OnScrUp(self, event):
|
|
self.client.ScrollUp(1)
|
|
|
|
def OnScrRt(self, event):
|
|
self.client.ScrollRight(2)
|
|
|
|
def OnReset(self, event):
|
|
self.client.Reset()
|
|
|
|
def OnHelpAbout(self, event):
|
|
from wx.lib.dialogs import ScrolledMessageDialog
|
|
about = ScrolledMessageDialog(self, __doc__, "About...")
|
|
about.ShowModal()
|
|
|
|
def OnBackgroundGray(self, event):
|
|
self.client.SetBackgroundColour("#CCCCCC")
|
|
self.client.Redraw()
|
|
|
|
def OnBackgroundWhite(self, event):
|
|
self.client.SetBackgroundColour("white")
|
|
self.client.Redraw()
|
|
|
|
def OnForegroundRed(self, event):
|
|
self.client.SetForegroundColour("red")
|
|
self.client.Redraw()
|
|
|
|
def OnForegroundBlack(self, event):
|
|
self.client.SetForegroundColour("black")
|
|
self.client.Redraw()
|
|
|
|
# -----------------------------------------------------------------------
|
|
### Enable Grid Events
|
|
# -----------------------------------------------------------------------
|
|
def _checkOtherGridMenuItems(self):
|
|
""" check or uncheck the submenu items """
|
|
self.gridSubMenu.Check(2151, self.client.enableGrid[0])
|
|
self.gridSubMenu.Check(2152, self.client.enableGrid[1])
|
|
self.gridSubMenu.Check(2153, all(self.client.enableGrid))
|
|
|
|
def OnEnableGridX(self, event):
|
|
old = self.client.enableGrid
|
|
self.client.enableGrid = (event.IsChecked(), old[1])
|
|
self._checkOtherGridMenuItems()
|
|
|
|
def OnEnableGridY(self, event):
|
|
old = self.client.enableGrid
|
|
self.client.enableGrid = (old[0], event.IsChecked())
|
|
self._checkOtherGridMenuItems()
|
|
|
|
def OnEnableGridAll(self, event):
|
|
self.client.enableGrid = event.IsChecked()
|
|
self._checkOtherGridMenuItems()
|
|
self.gridSubMenu.Check(2151, event.IsChecked())
|
|
self.gridSubMenu.Check(2152, event.IsChecked())
|
|
|
|
# -----------------------------------------------------------------------
|
|
### Enable Axes Events
|
|
# -----------------------------------------------------------------------
|
|
def _checkOtherAxesMenuItems(self):
|
|
""" check or uncheck the submenu items """
|
|
self.axesSubMenu.Check(2401, self.client.enableAxes[0])
|
|
self.axesSubMenu.Check(2402, self.client.enableAxes[1])
|
|
self.axesSubMenu.Check(2403, self.client.enableAxes[2])
|
|
self.axesSubMenu.Check(2404, self.client.enableAxes[3])
|
|
self.axesSubMenu.Check(2405, all(self.client.enableAxes[:2]))
|
|
self.axesSubMenu.Check(2406, all(self.client.enableAxes))
|
|
|
|
def OnEnableAxesBottom(self, event):
|
|
old = self.client.enableAxes
|
|
self.client.enableAxes = (event.IsChecked(), old[1], old[2], old[3])
|
|
self._checkOtherAxesMenuItems()
|
|
|
|
def OnEnableAxesLeft(self, event):
|
|
old = self.client.enableAxes
|
|
self.client.enableAxes = (old[0], event.IsChecked(), old[2], old[3])
|
|
self._checkOtherAxesMenuItems()
|
|
|
|
def OnEnableAxesTop(self, event):
|
|
old = self.client.enableAxes
|
|
self.client.enableAxes = (old[0], old[1], event.IsChecked(), old[3])
|
|
self._checkOtherAxesMenuItems()
|
|
|
|
def OnEnableAxesRight(self, event):
|
|
old = self.client.enableAxes
|
|
self.client.enableAxes = (old[0], old[1], old[2], event.IsChecked())
|
|
self._checkOtherAxesMenuItems()
|
|
|
|
def OnEnableAxesBottomLeft(self, event):
|
|
checked = event.IsChecked()
|
|
old = self.client.enableAxes
|
|
self.client.enableAxes = (checked, checked, old[2], old[3])
|
|
self._checkOtherAxesMenuItems()
|
|
|
|
def OnEnableAxesAll(self, event):
|
|
checked = event.IsChecked()
|
|
self.client.enableAxes = (checked, checked, checked, checked)
|
|
self._checkOtherAxesMenuItems()
|
|
|
|
# -----------------------------------------------------------------------
|
|
### Enable Ticks Events
|
|
# -----------------------------------------------------------------------
|
|
def _checkOtherTicksMenuItems(self):
|
|
""" check or uncheck the submenu items """
|
|
self.ticksSubMenu.Check(2501, self.client.enableTicks[0])
|
|
self.ticksSubMenu.Check(2502, self.client.enableTicks[1])
|
|
self.ticksSubMenu.Check(2503, self.client.enableTicks[2])
|
|
self.ticksSubMenu.Check(2504, self.client.enableTicks[3])
|
|
self.ticksSubMenu.Check(2505, all(self.client.enableTicks[:2]))
|
|
self.ticksSubMenu.Check(2506, all(self.client.enableTicks))
|
|
|
|
def OnEnableTicksBottom(self, event):
|
|
old = self.client.enableTicks
|
|
self.client.enableTicks = (event.IsChecked(), old[1],
|
|
old[2], old[3])
|
|
self._checkOtherTicksMenuItems()
|
|
|
|
def OnEnableTicksLeft(self, event):
|
|
old = self.client.enableTicks
|
|
self.client.enableTicks = (old[0], event.IsChecked(),
|
|
old[2], old[3])
|
|
self._checkOtherTicksMenuItems()
|
|
|
|
def OnEnableTicksTop(self, event):
|
|
old = self.client.enableTicks
|
|
self.client.enableTicks = (old[0], old[1],
|
|
event.IsChecked(), old[3])
|
|
self._checkOtherTicksMenuItems()
|
|
|
|
def OnEnableTicksRight(self, event):
|
|
old = self.client.enableTicks
|
|
self.client.enableTicks = (old[0], old[1],
|
|
old[2], event.IsChecked())
|
|
self._checkOtherTicksMenuItems()
|
|
|
|
def OnEnableTicksBottomLeft(self, event):
|
|
checked = event.IsChecked()
|
|
old = self.client.enableTicks
|
|
self.client.enableTicks = (checked, checked, old[2], old[3])
|
|
self._checkOtherTicksMenuItems()
|
|
|
|
def OnEnableTicksAll(self, event):
|
|
checked = event.IsChecked()
|
|
self.client.enableTicks = tuple([checked] * 4)
|
|
self._checkOtherTicksMenuItems()
|
|
|
|
# -----------------------------------------------------------------------
|
|
### Enable AxesValues Events
|
|
# -----------------------------------------------------------------------
|
|
def OnEnableAxesValuesBottom(self, event):
|
|
old = self.client.enableAxesValues
|
|
self.client.enableAxesValues = (event.IsChecked(), old[1],
|
|
old[2], old[3])
|
|
|
|
def OnEnableAxesValuesLeft(self, event):
|
|
old = self.client.enableAxesValues
|
|
self.client.enableAxesValues = (old[0], event.IsChecked(),
|
|
old[2], old[3])
|
|
|
|
def OnEnableAxesValuesTop(self, event):
|
|
old = self.client.enableAxesValues
|
|
self.client.enableAxesValues = (old[0], old[1],
|
|
event.IsChecked(), old[3])
|
|
|
|
def OnEnableAxesValuesRight(self, event):
|
|
old = self.client.enableAxesValues
|
|
self.client.enableAxesValues = (old[0], old[1],
|
|
old[2], event.IsChecked())
|
|
|
|
# -----------------------------------------------------------------------
|
|
### Other Enable Events
|
|
# -----------------------------------------------------------------------
|
|
def OnEnableZoom(self, event):
|
|
self.client.enableZoom = event.IsChecked()
|
|
if self.mainmenu.IsChecked(217):
|
|
self.mainmenu.Check(217, False)
|
|
|
|
def OnEnableDrag(self, event):
|
|
self.client.enableDrag = event.IsChecked()
|
|
if self.mainmenu.IsChecked(214):
|
|
self.mainmenu.Check(214, False)
|
|
|
|
def OnEnableScrollbars(self, event):
|
|
self.client.showScrollbars = event.IsChecked()
|
|
|
|
def OnEnableLegend(self, event):
|
|
self.client.enableLegend = event.IsChecked()
|
|
|
|
def OnEnablePointLabel(self, event):
|
|
self.client.enablePointLabel = event.IsChecked()
|
|
|
|
def OnEnableAntiAliasing(self, event):
|
|
self.client.enableAntiAliasing = event.IsChecked()
|
|
|
|
def OnEnableHiRes(self, event):
|
|
self.client.enableHiRes = event.IsChecked()
|
|
|
|
def OnEnablePlotTitle(self, event):
|
|
self.client.enablePlotTitle = event.IsChecked()
|
|
|
|
def OnEnableAxesLabels(self, event):
|
|
self.client.enableAxesLabels = event.IsChecked()
|
|
|
|
def OnEnableCenterLines(self, event):
|
|
self.client.enableCenterLines = event.IsChecked()
|
|
|
|
def OnEnableDiagonals(self, event):
|
|
self.client.enableDiagonals = event.IsChecked()
|
|
|
|
def OnLogX(self, event):
|
|
self.client.logScale = (event.IsChecked(), self.client.logScale[1])
|
|
self.client.Redraw()
|
|
|
|
def OnLogY(self, event):
|
|
self.client.logScale = (self.client.logScale[0], event.IsChecked())
|
|
self.client.Redraw()
|
|
|
|
def OnAbsX(self, event):
|
|
self.client.absScale = (event.IsChecked(), self.client.absScale[1])
|
|
self.client.Redraw()
|
|
|
|
def OnAbsY(self, event):
|
|
self.client.absScale = (self.client.absScale[0], event.IsChecked())
|
|
self.client.Redraw()
|
|
|
|
def resetDefaults(self):
|
|
"""Just to reset the fonts back to the PlotCanvas defaults"""
|
|
self.client.SetFont(wx.Font(10,
|
|
wx.FONTFAMILY_SWISS,
|
|
wx.FONTSTYLE_NORMAL,
|
|
wx.FONTWEIGHT_NORMAL)
|
|
)
|
|
self.client.fontSizeAxis = 10
|
|
self.client.fontSizeLegend = 7
|
|
self.client.logScale = (False, False)
|
|
self.client.xSpec = 'auto'
|
|
self.client.ySpec = 'auto'
|
|
|
|
def DrawPointLabel(self, dc, mDataDict):
|
|
"""
|
|
This is the function that defines how the pointLabels are plotted
|
|
|
|
:param dc: DC that will be passed
|
|
:param mDataDict: Dictionary of data that you want to use
|
|
for the pointLabel
|
|
|
|
As an example I have decided I want a box at the curve point
|
|
with some text information about the curve plotted below.
|
|
Any wxDC method can be used.
|
|
|
|
"""
|
|
dc.SetPen(wx.Pen(wx.BLACK))
|
|
dc.SetBrush(wx.Brush(wx.BLACK, wx.BRUSHSTYLE_SOLID))
|
|
|
|
sx, sy = mDataDict["scaledXY"] # scaled x,y of closest point
|
|
# 10by10 square centered on point
|
|
dc.DrawRectangle(int(sx - 5), int(sy - 5), 10, 10)
|
|
px, py = mDataDict["pointXY"]
|
|
cNum = mDataDict["curveNum"]
|
|
pntIn = mDataDict["pIndex"]
|
|
legend = mDataDict["legend"]
|
|
# make a string to display
|
|
s = "Crv# %i, '%s', Pt. (%.2f,%.2f), PtInd %i" % (
|
|
cNum, legend, px, py, pntIn)
|
|
dc.DrawText(s, int(sx), int(sy + 1))
|
|
|
|
|
|
def run_demo():
|
|
"""
|
|
Run the :mod:`wx.lib.plot` demo application.
|
|
"""
|
|
PlotDemoApp()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run_demo()
|