Move GUI files into their dir

2 files changed, 199 insertions, 0 deletions

diff --git a/src/gui/gui.py b/src/gui/gui.py
new file mode 100755
index 0000000..b2cbebb
--- /dev/null
+++ b/src/gui/gui.py
@@ -0,0 +1,131 @@
+#!/usr/bin/env python3
+
+# Python stdlib
+import sys
+
+# Grahical libraries
+from dearpygui.dearpygui import *
+import dearpygui._dearpygui as internal_dpg
+from dearpygui.demo import show_demo
+
+# Detect (unix) signals
+import signal
+
+# Mathematics
+import numpy as np
+
+# For debugging
+import logging
+
+# Remote resources
+import net
+
+#================================================
+# Debugging tools
+
+logging.basicConfig(format="[%(levelname)s] %(asctime)s %(message)s", level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+#================================================
+# Initialize DearPyGUI
+
+create_context()
+create_viewport(title="Fading Demonstrator")
+setup_dearpygui()
+
+# Show demo for dev
+show_demo()
+
+
+#================================================
+# GUI Callback functions
+
+# Flow graph window
+def on_rx_node_link(sender, app_data):
+    link_id_1, link_id_2 = app_data
+    add_node_link(link_id_1, link_id_2, parent=sender)
+
+def on_rx_node_delink(sender, app_data):
+    link_id = app_data
+    delete_item(link_id)
+
+#================================================
+# Settings Window
+
+with window(label="Settings", width=200, height=400, pos=(25, 450), tag="sim_win"):
+    with child_window(autosize_x=True, height=100):
+        add_button(label="Toggle Fullscreen", callback= toggle_viewport_fullscreen)
+
+#================================================
+# Flow Graph Window
+
+with window(label="RX DSP Flow Graph", width=800, height=400, pos=(25,25), tag="rx_win"):
+    with node_editor(callback=on_rx_node_link, delink_callback=on_rx_node_delink):
+        with node(label="USRP Source", pos=(20,100)):
+            with node_attribute(tag="src_out", attribute_type=mvNode_Attr_Output):
+                add_text("Signal from antenna")
+
+        with node(label="Clock Sync", pos=(200,200)):
+            with node_attribute(tag="clksync_in", attribute_type=mvNode_Attr_Input):
+                add_text("Input")
+
+            with node_attribute(tag="clksync_out", attribute_type=mvNode_Attr_Output):
+                add_text("Synchronized")
+
+        with node(label="Equalizer", pos=(350,100)):
+            with node_attribute(tag="eq_in", attribute_type=mvNode_Attr_Input):
+                add_text("Input")
+
+            with node_attribute(attribute_type=mvNode_Attr_Static):
+                add_knob_float(label="Gain")
+
+            with node_attribute(tag="eq_out", attribute_type=mvNode_Attr_Output):
+                add_text("Equalized")
+
+        with node(label="Phase Locked Loop", pos=(600, 200)):
+            with node_attribute(tag="pll_in", attribute_type=mvNode_Attr_Input):
+                add_text("Input")
+
+            with node_attribute(tag="pll_out", attribute_type=mvNode_Attr_Output):
+                add_text("Locked")
+                add_knob_float(label="Loop BW")
+
+
+        add_node_link(get_alias_id("src_out"), get_alias_id("clksync_in"))
+        add_node_link(get_alias_id("clksync_out"), get_alias_id("eq_in"))
+        add_node_link(get_alias_id("eq_out"), get_alias_id("pll_in"))
+
+#================================================
+# Network plots Window
+
+recv_plot = net.network_plot(url="udp://localhost:31415", nsamples=100, label="Test", height=300, width=800)
+
+plots = {
+    recv_plot: "plt_ampl"
+}
+
+with window(label="Time domain plots", width=800, height=400, pos=(850,25)):
+    with recv_plot:
+        add_plot_axis(mvXAxis, label="Time")
+        add_plot_axis(mvYAxis, label="Amplitude", tag="plt_ampl")
+
+        add_line_series(recv_plot.x_data, recv_plot.y_data, parent="plt_ampl")
+
+#================================================
+# Start GUI and main loop
+
+# Start window and main loop
+show_viewport()
+
+# Main loop
+while is_dearpygui_running():
+    for plt, tag in plots.items():
+        plt.refresh_series(tag)
+
+    render_dearpygui_frame()
+
+#================================================
+# Close everything
+
+# clean up gui
+destroy_context()
diff --git a/src/gui/net.py b/src/gui/net.py
new file mode 100644
index 0000000..2c91bb8
--- /dev/null
+++ b/src/gui/net.py
@@ -0,0 +1,68 @@
+import select
+import socket
+from urllib.parse import urlparse
+
+import numpy as np
+from numpy_ringbuffer import RingBuffer
+import dearpygui.dearpygui as dpg
+
+
+class udpsource:
+    """
+    Creates an UDP listening socket
+    """
+    def __init__(self, url):
+        self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        self.url = urlparse(url)
+
+    def __del__(self):
+        self.sock.close()
+
+    def bind(self):
+        self.sock.setblocking(False)
+        self.sock.bind((self.url.hostname, self.url.port))
+        # self.sock.listen()
+
+    def read(self, nbytes):
+        ready_to_read, ready_to_write, in_err = \
+                select.select([self.sock], [], [], 1)
+
+        if ready_to_read:
+            data = sock.recv(nbytes)
+            print(data)
+        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