Compare commits

...

2 Commits

Author SHA1 Message Date
ee094241bb PyThermostat: Add entry points for runnables
Forms a more convienient interface.
2024-11-18 17:51:18 +08:00
0ff0dbc3ec PyThermostat: Add main function to plot.py 2024-11-18 17:51:18 +08:00
2 changed files with 124 additions and 108 deletions

View File

@ -5,13 +5,15 @@ import matplotlib.animation as animation
from threading import Thread, Lock
from pythermostat.client import Client
TIME_WINDOW = 300.0
tec = Client()
target_temperature = tec.get_pid()[0]['target']
print("Channel 0 target temperature: {:.3f}".format(target_temperature))
def main():
TIME_WINDOW = 300.0
class Series:
tec = Client()
target_temperature = tec.get_pid()[0]['target']
print("Channel 0 target temperature: {:.3f}".format(target_temperature))
class Series:
def __init__(self, conv=lambda x: x):
self.conv = conv
self.x_data = []
@ -28,7 +30,7 @@ class Series:
self.x_data = self.x_data[drop:]
self.y_data = self.y_data[drop:]
series = {
series = {
# 'adc': Series(),
# 'sens': Series(lambda x: x * 0.0001),
'temperature': Series(),
@ -41,12 +43,12 @@ series = {
'tec_i': Series(),
'tec_u_meas': Series(),
# 'interval': Series(),
}
series_lock = Lock()
}
series_lock = Lock()
quit = False
quit = False
def recv_data(tec):
def recv_data(tec):
global last_packet_time
while True:
data = tec.get_report()
@ -65,16 +67,16 @@ def recv_data(tec):
break
time.sleep(0.05)
thread = Thread(target=recv_data, args=(tec,))
thread.start()
thread = Thread(target=recv_data, args=(tec,))
thread.start()
fig, ax = plt.subplots()
fig, ax = plt.subplots()
for k, s in series.items():
for k, s in series.items():
s.plot, = ax.plot([], [], label=k)
legend = ax.legend()
legend = ax.legend()
def animate(i):
def animate(i):
min_x, max_x, min_y, max_y = None, None, None, None
series_lock.acquire()
@ -119,13 +121,17 @@ def animate(i):
margin_y = 0.01 * (max_y - min_y)
ax.set_ylim(min_y - margin_y, max_y + margin_y)
global legend
nonlocal legend
legend.remove()
legend = ax.legend()
ani = animation.FuncAnimation(
ani = animation.FuncAnimation(
fig, animate, interval=1, blit=False, save_count=50)
plt.show()
quit = True
thread.join()
plt.show()
quit = True
thread.join()
if __name__ == "__main__":
main()

View File

@ -9,4 +9,14 @@ setup(
license="GPLv3",
install_requires=["setuptools"],
packages=find_packages(),
entry_points={
"gui_scripts": [
"thermostat_plot = plot:main",
],
"console_scripts": [
"thermostat_autotune = autotune:main",
"thermostat_test = test:main",
]
},
py_modules=["autotune", "plot", "test"],
)