diff options
author | Nao Pross <np@0hm.ch> | 2021-11-27 14:46:47 +0100 |
---|---|---|
committer | Nao Pross <np@0hm.ch> | 2021-11-27 14:46:47 +0100 |
commit | f263caa2db965ded17cbeb02280dc0174bd73587 (patch) | |
tree | 3479b25649ca15132e371a90bc116eb9fc6f2e84 /src/net.py | |
parent | Test interpolation angefangen (diff) | |
download | Fading-f263caa2db965ded17cbeb02280dc0174bd73587.tar.gz Fading-f263caa2db965ded17cbeb02280dc0174bd73587.zip |
Move network code and update nix files
Diffstat (limited to '')
-rw-r--r-- | src/net.py | 35 |
1 files changed, 35 insertions, 0 deletions
@@ -3,6 +3,8 @@ import socket from urllib.parse import urlparse import numpy as np +from numpy_ringbuffer import RingBuffer +import dearpygui.dearpygui as dpg class udpsource: @@ -31,3 +33,36 @@ class udpsource: else: return None + +class network_plot(udpsource): + def __init__(self, url, nsamples, **kwargs): + udpsource.__init__(self, url) + + self.nsamples = nsamples + self.plot = dpg.plot(**kwargs) + + # create buffer and fill with zeroes + self.buffer = RingBuffer(capacity=nsamples, dtype=(float, 2)) + for i in range(nsamples): + # TODO: remove random data used for testing + self.buffer.append(np.array([i, 1 + np.random.rand() / 5])) + + self.bind() + + def __enter__(self): + return self.plot.__enter__() + + def __exit__(self, t, val, tb): + self.plot.__exit__(t, val, tb) + + @property + def x_data(self): + return np.array(self.buffer[:,0]) + + @property + def y_data(self): + return np.array(self.buffer[:,1]) + + def refresh_series(self, tag): + dpg.set_value(tag, [self.x_data, self.y_data]) + pass |