195 lines
5 KiB
Python
195 lines
5 KiB
Python
import PySimpleGUIWeb as sg
|
|
import numpy as np
|
|
from matplotlib.backends.backend_tkagg import FigureCanvasAgg
|
|
import matplotlib.figure
|
|
import matplotlib.pyplot as plt
|
|
import io
|
|
|
|
|
|
from matplotlib import cm
|
|
from mpl_toolkits.mplot3d.axes3d import get_test_data
|
|
from matplotlib.ticker import NullFormatter # useful for `logit` scale
|
|
|
|
|
|
def create_axis_grid():
|
|
from mpl_toolkits.axes_grid1.axes_rgb import RGBAxes
|
|
|
|
plt.close('all')
|
|
|
|
def get_demo_image():
|
|
# prepare image
|
|
delta = 0.5
|
|
|
|
extent = (-3, 4, -4, 3)
|
|
x = np.arange(-3.0, 4.001, delta)
|
|
y = np.arange(-4.0, 3.001, delta)
|
|
X, Y = np.meshgrid(x, y)
|
|
Z1 = np.exp(-X ** 2 - Y ** 2)
|
|
Z2 = np.exp(-(X - 1) ** 2 - (Y - 1) ** 2)
|
|
Z = (Z1 - Z2) * 2
|
|
|
|
return Z, extent
|
|
|
|
def get_rgb():
|
|
Z, extent = get_demo_image()
|
|
|
|
Z[Z < 0] = 0.
|
|
Z = Z / Z.max()
|
|
|
|
R = Z[:13, :13]
|
|
G = Z[2:, 2:]
|
|
B = Z[:13, 2:]
|
|
|
|
return R, G, B
|
|
|
|
fig = plt.figure(1)
|
|
ax = RGBAxes(fig, [0.1, 0.1, 0.8, 0.8])
|
|
|
|
r, g, b = get_rgb()
|
|
kwargs = dict(origin="lower", interpolation="nearest")
|
|
ax.imshow_rgb(r, g, b, **kwargs)
|
|
|
|
ax.RGB.set_xlim(0., 9.5)
|
|
ax.RGB.set_ylim(0.9, 10.6)
|
|
|
|
plt.draw()
|
|
return plt.gcf()
|
|
|
|
|
|
|
|
def create_figure():
|
|
# ------------------------------- START OF YOUR MATPLOTLIB CODE -------------------------------
|
|
fig = matplotlib.figure.Figure(figsize=(5, 4), dpi=100)
|
|
t = np.arange(0, 3, .01)
|
|
fig.add_subplot(111).plot(t, 2 * np.sin(2 * np.pi * t))
|
|
|
|
return fig
|
|
|
|
|
|
def create_subplot_3d():
|
|
|
|
|
|
fig = plt.figure()
|
|
|
|
ax = fig.add_subplot(1, 2, 1, projection='3d')
|
|
X = np.arange(-5, 5, 0.25)
|
|
Y = np.arange(-5, 5, 0.25)
|
|
X, Y = np.meshgrid(X, Y)
|
|
R = np.sqrt(X ** 2 + Y ** 2)
|
|
Z = np.sin(R)
|
|
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.jet,
|
|
linewidth=0, antialiased=False)
|
|
ax.set_zlim3d(-1.01, 1.01)
|
|
|
|
fig.colorbar(surf, shrink=0.5, aspect=5)
|
|
|
|
ax = fig.add_subplot(1, 2, 2, projection='3d')
|
|
X, Y, Z = get_test_data(0.05)
|
|
ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)
|
|
return fig
|
|
|
|
|
|
|
|
|
|
def create_pyplot_scales():
|
|
|
|
plt.close('all')
|
|
# Fixing random state for reproducibility
|
|
np.random.seed(19680801)
|
|
|
|
# make up some data in the interval ]0, 1[
|
|
y = np.random.normal(loc=0.5, scale=0.4, size=1000)
|
|
y = y[(y > 0) & (y < 1)]
|
|
y.sort()
|
|
x = np.arange(len(y))
|
|
|
|
# plot with various axes scales
|
|
plt.figure(1)
|
|
|
|
# linear
|
|
plt.subplot(221)
|
|
plt.plot(x, y)
|
|
plt.yscale('linear')
|
|
plt.title('linear')
|
|
plt.grid(True)
|
|
|
|
# log
|
|
plt.subplot(222)
|
|
plt.plot(x, y)
|
|
plt.yscale('log')
|
|
plt.title('log')
|
|
plt.grid(True)
|
|
|
|
# symmetric log
|
|
plt.subplot(223)
|
|
plt.plot(x, y - y.mean())
|
|
plt.yscale('symlog', linthreshy=0.01)
|
|
plt.title('symlog')
|
|
plt.grid(True)
|
|
|
|
# logit
|
|
plt.subplot(224)
|
|
plt.plot(x, y)
|
|
plt.yscale('logit')
|
|
plt.title('logit')
|
|
plt.grid(True)
|
|
# Format the minor tick labels of the y-axis into empty strings with
|
|
# `NullFormatter`, to avoid cumbering the axis with too many labels.
|
|
plt.gca().yaxis.set_minor_formatter(NullFormatter())
|
|
# Adjust the subplot layout, because the logit one may take more space
|
|
# than usual, due to y-tick labels like "1 - 10^{-3}"
|
|
plt.subplots_adjust(top=0.92, bottom=0.08, left=0.10, right=0.95, hspace=0.25,
|
|
wspace=0.35)
|
|
return plt.gcf()
|
|
|
|
# ----------------------------- The draw figure helpful function -----------------------------
|
|
|
|
def draw_figure(fig, element):
|
|
"""
|
|
Draws the previously created "figure" in the supplied Image Element
|
|
|
|
:param fig: a Matplotlib figure
|
|
:param element: an Image Element
|
|
:return: The figure canvas
|
|
"""
|
|
|
|
plt.close('all') # erases previously drawn plots
|
|
canv = FigureCanvasAgg(fig)
|
|
buf = io.BytesIO()
|
|
canv.print_figure(buf, format='png')
|
|
if buf is None:
|
|
return None
|
|
buf.seek(0)
|
|
element.update(data=buf.read())
|
|
return canv
|
|
|
|
# ----------------------------- The GUI Section -----------------------------
|
|
|
|
def main():
|
|
dictionary_of_figures = {'Axis Grid': create_axis_grid,
|
|
'Subplot 3D': create_subplot_3d,
|
|
'Scales': create_pyplot_scales,
|
|
'Basic Figure': create_figure}
|
|
|
|
layout = [
|
|
[sg.T('Matplotlib Example', font='Any 20')],
|
|
[sg.Listbox(dictionary_of_figures.keys(), size=(15, 5), key='-LB-'), sg.Image(key='-IMAGE-')],
|
|
[sg.B('Draw'), sg.B('Exit')],
|
|
]
|
|
|
|
window = sg.Window('Title', layout, finalize=True)
|
|
|
|
image_element = window['-IMAGE-'] # type: sg.Image
|
|
|
|
while True:
|
|
event, values = window.read()
|
|
if event == 'Exit' or event == sg.WIN_CLOSED:
|
|
break
|
|
if event == 'Draw':
|
|
func = dictionary_of_figures[values['-LB-'][0]]
|
|
draw_figure(func(), image_element)
|
|
window.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|