diff options
| -rwxr-xr-x | src/sketch.py | 171 |
1 files changed, 51 insertions, 120 deletions
diff --git a/src/sketch.py b/src/sketch.py index b7500c8..5f170b2 100755 --- a/src/sketch.py +++ b/src/sketch.py @@ -19,14 +19,24 @@ import numpy as np # Our interface for simulation / hardware import qpsk +# For debugging +import logging + +#================================================ +# Debugging tools + +logging.basicConfig(format="[%(levelname)s] %(asctime)s %(message)s", level=logging.DEBUG) +logger = logging.getLogger(__name__) + +#================================================ +# Initialize DearPyGUI -# Create GL context and initialize DearPyGUI create_context() create_viewport() setup_dearpygui() # Show demo for dev -show_demo() +# show_demo() #================================================ # Custom GNURadio blocks @@ -43,15 +53,16 @@ class pygui_plot_block(gr.sync_block): # TODO: parametrize with nsamples self.buffer = np.linspace(0, 1, 1000) - def construct_gui(self, width, height): + def construct_gui(self, width, height, **kwargs): # Create plot - with plot(label=self.title, height=height, width=width): + with plot(label=self.title, height=height, width=width, **kwargs): add_plot_legend() add_plot_axis(mvXAxis, label=self.xlabel) add_plot_axis(mvYAxis, label=self.ylabel) def work(self, input_items, output_items): # This is a sink (no output) + # TODO: add samples to display buffer return 0 @@ -64,26 +75,34 @@ class pygui_constellation_block(gr.sync_block): self.xlabel = xlabel selt.ylabel = ylabel + def construct_gui(self, width, height, **kwargs): + # TODO: Create a widget based ons scatter plot + pass + + def work(self, input_items, output_items): + # This is a sink (no output) + # TODO: add samples to display buffer + return 0 + #================================================ # GUI Callback functions -def _on_flowgraph_close(): - pass - -def _on_rx_node_link(sender, app_data): +# Flow graph window +def on_rx_node_link(sender, app_data): link_id_1, link_id_2 = app_data - print(app_data) add_node_link(link_id_1, link_id_2, parent=sender) -def _on_rx_node_delink(sender, app_data): +def on_rx_node_delink(sender, app_data): link_id = app_data delete_item(link_id) #================================================ # Set up GNURadio simulation +# Global variables, yes I know they are ugly sim = qpsk.qpsk_nogui() +sim_running = False # Catch signals def sim_sig_handler(sig=None, frame=None): @@ -96,37 +115,38 @@ signal.signal(signal.SIGINT, sim_sig_handler) signal.signal(signal.SIGTERM, sim_sig_handler); # Add GUI blocks -locked_time_plot = pygui_plot_block(title="Locked signal", xlabel="t") +locked_time_plot = pygui_plot_block(title="Locked signal", ylabel="Amplitude", xlabel="Time") sim.connect((sim.digital_costas_loop_cc_0, 0), (locked_time_plot, 0)) #================================================ -# Simulation Control Window +# Settings Window + +with window(label="Settings", width=200, height=400, pos=(25, 450), tag="sim_win"): + add_button(label="Toggle Fullscreen", callback= toggle_viewport_fullscreen) -with window(label="Simulation", width=200, height=400, pos=(25, 450), tag="sim_win"): - add_text("Simulation state:") + add_text("Simulation running:") + add_text("false", tag="sim_running_lbl") with group(horizontal=True): def on_sim_start_btn_clicked(): sim.start() - - configure_item("sim_start_btn", enabled=False) - configure_item("sim_stop_btn", enabled=True) + sim_runnig = True + logger.debug("Started simulation") def on_sim_stop_btn_clicked(): sim.stop() sim.wait() - - configure_item("sim_start_btn", enabled=True) - configure_item("sim_stop_btn", enabled=False) + sim_running = False + logger.debug("Stopped simulation") add_button(label="Start", tag="sim_start_btn", callback=on_sim_start_btn_clicked) - add_button(label="Stop", enabled=False, tag="sim_stop_btn", callback=on_sim_stop_btn_clicked) + add_button(label="Stop", tag="sim_stop_btn", callback=on_sim_stop_btn_clicked) #================================================ # Flow Graph Window -with window(label="RX DSP Flow Graph", width=800, height=400, on_close=_on_flowgraph_close, pos=(25,25), tag="rx_win"): - with node_editor(callback=_on_rx_node_link, delink_callback=_on_rx_node_delink): +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") @@ -142,12 +162,12 @@ with window(label="RX DSP Flow Graph", width=800, height=400, on_close=_on_flowg with node_attribute(tag="eq_in", attribute_type=mvNode_Attr_Input): add_text("Input") - with node_attribute(tag="eq_out", attribute_type=mvNode_Attr_Output): - add_text("Equalized") - 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") @@ -167,98 +187,8 @@ with window(label="RX DSP Flow Graph", width=800, height=400, on_close=_on_flowg with window(label="Time domain plots", width=800, height=400, pos=(850,25), tag="time_plots_win"): locked_time_plot.construct_gui(width=780, height=300) - #================================================ -# QPSK Window - -# q = qp.qpsk_nogui() - -# with window(label="QPSK", width=1500, height=800, on_close=_on_flowgraph_close, pos=(100,100), 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 Source Random") -# print(q.analog_random_source_x_0) -# # TO DO: Signal Plot -# -# with node(label="Constellation Modulator", pos=(200,200)): -# with node_attribute(tag="modul_in", attribute_type=mvNode_Attr_Input): -# add_text("Input") -# add_text(f'Sample {q.sps}') -# -# with node_attribute(tag="modul_out", attribute_type=mvNode_Attr_Output): -# add_text("Output") -# -# with node(label="Channel Model", pos=(420,100)): -# with node_attribute(tag="channel_in", attribute_type=mvNode_Attr_Input): -# add_text("Input") -# -# with node_attribute(tag="channel_out", attribute_type=mvNode_Attr_Output): -# add_text("Output") -# -# with node(label="Clock Sync", pos=(550,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=(700,100)): -# with node_attribute(attribute_type=mvNode_Attr_Static): -# add_input_float(label="Gain", width=150) -# -# with node_attribute(tag="eq_in", attribute_type=mvNode_Attr_Input): -# add_text("Input") -# -# with node_attribute(tag="eq_out", attribute_type=mvNode_Attr_Output): -# add_text("Equalized") -# -# -# with node(label="Phase Locked Loop", pos=(950, 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_node_link(get_alias_id("src_out"), get_alias_id("modul_in")) -# add_node_link(get_alias_id("modul_out"), get_alias_id("channel_in")) -# add_node_link(get_alias_id("channel_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")) -# - - -#================================================ -# Plot Time Window Test - -# creating data -# sindatax = [] -# sindatay = [] -# for i in range(0, 500): -# sindatax.append(i / 1000) -# sindatay.append(0.5 + 0.5 * sin(50 * i / 1000)) -# -# with window(label="Plots"): -# add_text("Hello world") -# # create plot -# with plot(label="Line Series", height=400, width=400): -# # optionally create legend -# add_plot_legend() -# -# # REQUIRED: create x and y axes -# add_plot_axis(mvXAxis, label="x") -# add_plot_axis(mvYAxis, label="y", tag="y_axis") -# -# # series belong to a y axis -# add_line_series(sindatax, sindatay, label="0.5 + 0.5 * sin(x)", parent="y_axis") - -#================================================ - - +# Start GUI # Start window and main loop show_viewport() @@ -267,6 +197,7 @@ start_dearpygui() # clean up gui destroy_context() -# Stop GNURadio -sim.stop() -sim.wait() +# Stop GNURadio +if sim_running: + sim.stop() + sim.wait() |