forked from M-Labs/thermostat
128 lines
3.2 KiB
Python
128 lines
3.2 KiB
Python
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
import matplotlib.animation as animation
|
|
from threading import Thread, Lock
|
|
from pytec.client import Client
|
|
|
|
TIME_WINDOW = 300.0
|
|
|
|
class Series:
|
|
def __init__(self, scale=1.0):
|
|
self.scale = scale
|
|
self.x_data = []
|
|
self.y_data = []
|
|
|
|
def append(self, x, y):
|
|
self.x_data.append(x)
|
|
self.y_data.append(self.scale * y)
|
|
|
|
def clip(self, min_x):
|
|
drop = 0
|
|
while drop < len(self.x_data) and self.x_data[drop] < min_x:
|
|
drop += 1
|
|
self.x_data = self.x_data[drop:]
|
|
self.y_data = self.y_data[drop:]
|
|
|
|
series = {
|
|
'adc': Series(),
|
|
'sens': Series(0.0001),
|
|
'temperature': Series(),
|
|
'i_set': Series(),
|
|
'vref': Series(),
|
|
'dac_feedback': Series(),
|
|
'i_tec': Series(),
|
|
'tec_i': Series(),
|
|
'tec_u_meas': Series(),
|
|
}
|
|
series_lock = Lock()
|
|
|
|
quit = False
|
|
|
|
def recv_data(tec):
|
|
print("reporting")
|
|
for data in tec.report_mode():
|
|
if data['channel'] == 0:
|
|
series_lock.acquire()
|
|
try:
|
|
time = data['time'] / 1000.0
|
|
for k, s in series.iteritems():
|
|
v = data[k]
|
|
if data.has_key(k) and type(v) is float:
|
|
s.append(time, v)
|
|
finally:
|
|
series_lock.release()
|
|
|
|
if quit:
|
|
break
|
|
|
|
tec = Client()
|
|
print("connected")
|
|
thread = Thread(target=recv_data, args=(tec,))
|
|
thread.start()
|
|
|
|
fig, ax = plt.subplots()
|
|
|
|
for k, s in series.iteritems():
|
|
s.plot, = ax.plot([], [], label=k)
|
|
ax.legend()
|
|
|
|
def animate(i):
|
|
min_x, max_x, min_y, max_y = None, None, None, None
|
|
|
|
series_lock.acquire()
|
|
try:
|
|
for s in series.itervalues():
|
|
s.plot.set_data(s.x_data, s.y_data)
|
|
|
|
if len(s.x_data) > 0:
|
|
min_x_ = min(s.x_data)
|
|
if min_x is None:
|
|
min_x = min_x_
|
|
else:
|
|
min_x = min(min_x, min_x_)
|
|
max_x_ = max(s.x_data)
|
|
if max_x is None:
|
|
max_x = max_x_
|
|
else:
|
|
max_x = max(max_x, max_x_)
|
|
if len(s.y_data) > 0:
|
|
min_y_ = min(s.y_data)
|
|
if min_y is None:
|
|
min_y = min_y_
|
|
else:
|
|
min_y = min(min_y, min_y_)
|
|
max_y_ = max(s.y_data)
|
|
if max_y is None:
|
|
max_y = max_y_
|
|
else:
|
|
max_y = max(max_y, max_y_)
|
|
|
|
if min_x is not None and max_x - TIME_WINDOW > min_x:
|
|
for s in series.itervalues():
|
|
s.clip(max_x - TIME_WINDOW)
|
|
finally:
|
|
series_lock.release()
|
|
|
|
margin_y = 0.01 * (max_y - min_y)
|
|
ax.set_xlim(min_x, max_x)
|
|
ax.set_ylim(min_y - margin_y, max_y + margin_y)
|
|
|
|
|
|
ani = animation.FuncAnimation(
|
|
fig, animate, interval=1, blit=False, save_count=50)
|
|
|
|
# To save the animation, use e.g.
|
|
#
|
|
# ani.save("movie.mp4")
|
|
#
|
|
# or
|
|
#
|
|
# writer = animation.FFMpegWriter(
|
|
# fps=15, metadata=dict(artist='Me'), bitrate=1800)
|
|
# ani.save("movie.mp4", writer=writer)
|
|
|
|
print("show")
|
|
plt.show()
|
|
quit = True
|
|
thread.join()
|