def __init__(self,
g_id=0,
samples=500,
rate=0.02,
port=None,
plot_widget=None):
super(PlotManager, self).__init__()
# mp.Process.__init__(self)
self.worker = Worker()
self.graph_id = g_id
self.samples = int(samples)
self.rate = float(rate)
self.port = port
self.plot = []
self.plot.append(plot_widget)
self.configure_timers()
self.color_dict = ({
0: '#6c6d70',
1: '#EB340D',
2: '#0D46EB',
3: '#d01bd3',
4: '#ed9615',
5: '#298909'
})
def start(self):
self.worker = Worker(graph_id=self.graph_id,
samples=self.samples,
rate=self.rate,
port=self.port)
if self.worker.start():
self.timer_plot.start(20)
def __init__(self, samples=500, rate=0.02, port=None):
super(PlotManager, self).__init__()
# mp.Process.__init__(self)
self.samples = int(samples)
self.rate = int(rate)
self.port = port
self.plotfunc = dict()
self.clist = []
self.worker = Worker()
self.computation = Function()
self.configure_timers()
self.color_dict = ({
0: '#6c6d70',
1: '#EB340D',
2: '#0D46EB',
3: '#d01bd3',
4: '#ed9615',
5: '#298909'
})
self.line = 0
# Log
self.directory = FileManager()
self.csvfile = None
self.mywriter = None
self.fieldnames = ['Channel']
self.filename = ''
self.data = dict()
# self.logenable= False
self.writeheader = False
def _init_worker(self):
for worker_index in range(MAX_WORKER_COUNT):
worker = Worker(worker_index, self.__master)
self.__workers.append(worker)
self.__worker_index_to_fd.append(-1)
LOG.info('init {} workers successful'.format(MAX_WORKER_COUNT))
return
def test_one(self):
time = 2
speed = 0.02
error = 0.95
zeros = 0
passed = False
samples = int(time * (1/speed))
samples_plus_error = int(samples / error)
worker = Worker(graph_id=0,
samples=500,
rate=0.2,
port=None)
worker.run()
sleep(time)
worker.stop()
worker.get_plot_value()
for v in worker.getybuffer(0):
print(v)
class PlotManager(QObject):
def __init__(self,
g_id=0,
samples=500,
rate=0.02,
port=None,
plot_widget=None):
super(PlotManager, self).__init__()
# mp.Process.__init__(self)
self.worker = Worker()
self.graph_id = g_id
self.samples = int(samples)
self.rate = float(rate)
self.port = port
self.plot = []
self.plot.append(plot_widget)
self.configure_timers()
self.color_dict = ({
0: '#6c6d70',
1: '#EB340D',
2: '#0D46EB',
3: '#d01bd3',
4: '#ed9615',
5: '#298909'
})
def start(self):
self.worker = Worker(graph_id=self.graph_id,
samples=self.samples,
rate=self.rate,
port=self.port)
if self.worker.start():
self.timer_plot.start(20)
def stop(self):
self.timer_plot.stop()
self.worker.stop()
def configure_timers(self):
"""
Configures specific elements of the QTimers.
:return:
"""
self.timer_plot = QTimer()
self.timer_plot.timeout.connect(self.update_plot)
def update_plot(self, read_line=-1):
if self.worker.is_running():
self.worker.get_plot_value()
widget_num = len(self.plot)
count = self.worker.get_channel_num()
c = 1
# Clear all data before plot
for p in self.plot:
p.plotItem.clear()
# Plot data
# Plot all channel in 1 graph
for i in range(count):
pen = pg.mkPen(self.color_dict[i], width=1, style=None)
self.plot[0].plotItem.plot(self.worker.getxbuffer(),
self.worker.getybuffer(i),
pen=pen)
# Individual plot
while c <= widget_num - 1:
pen = pg.mkPen(self.color_dict[c - 1], width=1, style=None)
self.plot[c].plotItem.plot(self.worker.getxbuffer(),
self.worker.getybuffer(c - 1),
pen=pen)
c += 1
else:
self.stop()
print('Manager fail to open port')
def is_running(self):
return self.worker.is_running()
def update_parameter(self, s, r):
self.samples = int(s)
self.rate = float(r)
print('Manager update')
class PlotManager(QObject):
path = '..\\SerialPlot\\data'
def __init__(self, samples=500, rate=0.02, port=None):
super(PlotManager, self).__init__()
# mp.Process.__init__(self)
self.samples = int(samples)
self.rate = int(rate)
self.port = port
self.plotfunc = dict()
self.clist = []
self.worker = Worker()
self.computation = Function()
self.configure_timers()
self.color_dict = ({
0: '#6c6d70',
1: '#EB340D',
2: '#0D46EB',
3: '#d01bd3',
4: '#ed9615',
5: '#298909'
})
self.line = 0
# Log
self.directory = FileManager()
self.csvfile = None
self.mywriter = None
self.fieldnames = ['Channel']
self.filename = ''
self.data = dict()
# self.logenable= False
self.writeheader = False
def start(self):
self.worker = Worker(samples=self.samples,
rate=self.rate,
port=self.port)
if self.worker.start():
self.timer_plot.start(20)
def stop(self):
self.timer_plot.stop()
self.worker.stop()
def configure_timers(self):
"""
Configures specific elements of the QTimers.
:return:
"""
self.timer_plot = QTimer()
self.timer_plot.timeout.connect(self.update_plot)
def update_plot(self):
if self.worker.is_running():
self.worker.get_plot_value()
self.data.clear()
size = self.samples / 2
for f in self.plotfunc.keys():
self.plotfunc[f].plotItem.clear()
if f == 'Raw Data':
for c in self.clist:
self.graph(self.plotfunc[f], self.worker.getxbuffer(),
self.worker.getybuffer(c), c)
self.data[c, f] = self.worker.getybuffer(c)
elif f == 'Autocorrelation':
for c in self.clist:
x, y = self.computation.autocorrelation_plot(
self.worker.getybuffer(c))
self.graph(self.plotfunc[f], x, y, c)
self.data[c, f] = y
# if self.logenable:
# # Header for log file
# if self.writeheader:
# self.mywriter.writerow(self.fieldnames)
# self.writeheader = False
# # Data
# processed_data = []
# for c in self.clist:
# d = 0
# while d < size:
# field = 1
# processed_data.clear()
# processed_data.append(c)
# while field < len(self.fieldnames):
# processed_data.append(self.data[(c,self.fieldnames[field])][d])
# field += 1
# if len(processed_data) > 1:
# self.mywriter.writerow(processed_data)
# d += 1
else:
self.stop()
print('Manager fail to open port')
def is_running(self):
return self.worker.is_running()
def update_plotDict(self, pf):
self.plotfunc = pf
self.fieldnames.clear()
self.fieldnames.append('Channel')
for header in pf.keys():
self.fieldnames.append(header)
self.writeheader = True
def update_samples(self, s):
# if self.is_running():
# self.stop()
self.samples = int(s)
print(self.samples)
# self.start()
def add_channel(self, c):
if c not in self.clist:
self.clist.append(c)
def remove_channel(self, c):
for ch in self.clist:
if ch == c:
self.clist.remove(c)
def setup_serial(self, sample, brate, port):
self.samples = int(sample)
self.rate = int(brate)
self.port = port
print('Setting up serial port!')
def get_port(self):
return self.worker.get_port()
def csv_checked(self, fileready, fname):
self.logenable = fileready
self.filename = fname
if fileready:
self.csvfile = self.directory.create_file(self.path, fname)
self.mywriter = csv.writer(self.csvfile, delimiter=',')
else:
if self.directory.check_path(
os.path.join(self.path, fname + '.csv')):
self.csvfile.close()
def count_channel(self):
return self.worker.get_channel_num()
def graph(self, plot_widget, x, y, i):
pen = pg.mkPen(self.color_dict[i], width=1, style=None)
plot_widget.plotItem.plot(x, y, pen=pen)