RPMPythonInterface/main_matplotlib.py at main · EdwardL33/RPMPythonInterface · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import matplotlib.pyplot as plt
import numpy as np

from matplotlib.widgets import TextBox, Button

fig, FULL_ax = plt.subplots(layout="constrained")
# fig.subplots_adjust(bottom=0.2)
FULL_ax.axis('off')

import matplotlib.gridspec as gridspec
gs0 = gridspec.GridSpec(1, 2, figure=fig, width_ratios=[1, 4])
gs1 = gridspec.GridSpecFromSubplotSpec(5, 1, subplot_spec=gs0[0], height_ratios=[4, 2, 2, 1, 1])

plot_ax = fig.add_subplot(gs0[1])
text_ax = fig.add_subplot(gs1[0])
start_ax = fig.add_subplot(gs1[1])
stop_ax = fig.add_subplot(gs1[2])
man_ax = fig.add_subplot(gs1[3])
upload_ax = fig.add_subplot(gs1[4])

t = np.arange(-2.0, 2.0, 0.001)
l, = plot_ax.plot(t, np.zeros_like(t), lw=2)


def submit(expression):
    """
    Update the plotted function to the new math *expression*.

    *expression* is a string using "t" as its independent variable, e.g.
    "t ** 3".
    """
    ydata = eval(expression, {'np': np}, {'t': t})
    l.set_ydata(ydata)
    plot_ax.relim()
    plot_ax.autoscale_view()
    plt.draw()

def start(event):
    print("Start")

def stop(event):
    print("Sto9p")

def manual(event):
    print("Man")

def upload(event):
    print("Upload")

# axbox = fig.add_axes([0.1, 0.05, 0.6, 0.075])
# axprev = fig.add_axes([0.7, 0.05, 0.1, 0.075])
# axnext = fig.add_axes([0.81, 0.05, 0.1, 0.075])
text_box = TextBox(text_ax, "", textalignment="left")
#text_box.on_submit(submit)
text_box.set_val("t ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\nt ** 2\nasdjaisjda\naduhsudas\n")  # Trigger `submit` with the initial string.
bnext = Button(start_ax, 'Start')
bnext.on_clicked(start)
bprev = Button(stop_ax, 'Stop')
bprev.on_clicked(stop)
bprev = Button(man_ax, 'Manual')
bprev.on_clicked(manual)
bprev = Button(upload_ax, 'Upload')
bprev.on_clicked(upload)

plt.show()