def __init__(self, parent):
"""
Create a new control panel.
@param parent the wx parent window
"""
WIDTH = 90
self.parent = parent
wx.Panel.__init__(self, parent, style=wx.SUNKEN_BORDER)
parent[SHOW_CONTROL_PANEL_KEY] = True
parent.subscribe(SHOW_CONTROL_PANEL_KEY, self.Show)
control_box = wx.BoxSizer(wx.VERTICAL)
##################################################
# Persistence
##################################################
forms.check_box(
sizer=control_box,
parent=self,
label='Persistence',
ps=parent,
key=USE_PERSISTENCE_KEY,
)
#static text and slider for analog alpha
persist_alpha_text = forms.static_text(
sizer=control_box,
parent=self,
label='Analog Alpha',
converter=forms.float_converter(lambda x: '%.4f' % x),
ps=parent,
key=PERSIST_ALPHA_KEY,
width=50,
)
persist_alpha_slider = forms.log_slider(
sizer=control_box,
parent=self,
min_exp=PERSIST_ALPHA_MIN_EXP,
max_exp=PERSIST_ALPHA_MAX_EXP,
num_steps=SLIDER_STEPS,
ps=parent,
key=PERSIST_ALPHA_KEY,
)
for widget in (persist_alpha_text, persist_alpha_slider):
parent.subscribe(USE_PERSISTENCE_KEY, widget.Enable)
widget.Enable(parent[USE_PERSISTENCE_KEY])
parent.subscribe(USE_PERSISTENCE_KEY, widget.ShowItems)
#allways show initially, so room is reserved for them
widget.ShowItems(True) # (parent[USE_PERSISTENCE_KEY])
parent.subscribe(USE_PERSISTENCE_KEY, self._update_layout)
##################################################
# Axes Options
##################################################
control_box.AddStretchSpacer()
axes_options_box = forms.static_box_sizer(
parent=self,
sizer=control_box,
label='Axes Options',
bold=True,
orient=wx.VERTICAL,
)
##################################################
# Scope Mode Box
##################################################
scope_mode_box = wx.BoxSizer(wx.VERTICAL)
axes_options_box.Add(scope_mode_box, 0, wx.EXPAND)
#x axis divs
forms.incr_decr_buttons(
parent=self,
sizer=scope_mode_box,
label='Secs/Div',
on_incr=self._on_incr_t_divs,
on_decr=self._on_decr_t_divs,
)
#y axis divs
y_buttons_scope = forms.incr_decr_buttons(
parent=self,
sizer=scope_mode_box,
label='Counts/Div',
on_incr=self._on_incr_y_divs,
on_decr=self._on_decr_y_divs,
)
#y axis ref lvl
y_off_buttons_scope = forms.incr_decr_buttons(
parent=self,
sizer=scope_mode_box,
label='Y Offset',
on_incr=self._on_incr_y_off,
on_decr=self._on_decr_y_off,
)
#t axis ref lvl
scope_mode_box.AddSpacer(5)
forms.slider(
parent=self,
sizer=scope_mode_box,
ps=parent,
key=T_FRAC_OFF_KEY,
label='T Offset',
minimum=0,
maximum=1,
num_steps=1000,
)
scope_mode_box.AddSpacer(5)
##################################################
# XY Mode Box
##################################################
xy_mode_box = wx.BoxSizer(wx.VERTICAL)
axes_options_box.Add(xy_mode_box, 0, wx.EXPAND)
#x div controls
x_buttons = forms.incr_decr_buttons(
parent=self,
sizer=xy_mode_box,
label='X/Div',
on_incr=self._on_incr_x_divs,
on_decr=self._on_decr_x_divs,
)
#y div controls
y_buttons = forms.incr_decr_buttons(
parent=self,
sizer=xy_mode_box,
label='Y/Div',
on_incr=self._on_incr_y_divs,
on_decr=self._on_decr_y_divs,
)
#x offset controls
x_off_buttons = forms.incr_decr_buttons(
parent=self,
sizer=xy_mode_box,
label='X Off',
on_incr=self._on_incr_x_off,
on_decr=self._on_decr_x_off,
)
#y offset controls
y_off_buttons = forms.incr_decr_buttons(
parent=self,
sizer=xy_mode_box,
label='Y Off',
on_incr=self._on_incr_y_off,
on_decr=self._on_decr_y_off,
)
for widget in (y_buttons_scope, y_off_buttons_scope, x_buttons,
y_buttons, x_off_buttons, y_off_buttons):
parent.subscribe(AUTORANGE_KEY, widget.Disable)
widget.Disable(parent[AUTORANGE_KEY])
xy_mode_box.ShowItems(False)
#autorange check box
forms.check_box(
parent=self,
sizer=axes_options_box,
label='Autorange',
ps=parent,
key=AUTORANGE_KEY,
)
##################################################
# Channel Options
##################################################
TRIGGER_PAGE_INDEX = parent.num_inputs
XY_PAGE_INDEX = parent.num_inputs + 1
control_box.AddStretchSpacer()
chan_options_box = forms.static_box_sizer(
parent=self,
sizer=control_box,
label='Channel Options',
bold=True,
orient=wx.VERTICAL,
)
options_notebook = wx.Notebook(self)
options_notebook_args = list()
CHANNELS = [('Ch %d' % (i + 1), i) for i in range(parent.num_inputs)]
##################################################
# Channel Menu Boxes
##################################################
for i in range(parent.num_inputs):
channel_menu_panel = wx.Panel(options_notebook)
options_notebook_args.append(
(channel_menu_panel, i, 'Ch%d' % (i + 1)))
channel_menu_box = wx.BoxSizer(wx.VERTICAL)
channel_menu_panel.SetSizer(channel_menu_box)
#ac couple check box
channel_menu_box.AddStretchSpacer()
forms.drop_down(
parent=channel_menu_panel,
sizer=channel_menu_box,
ps=parent,
key=common.index_key(AC_COUPLE_KEY, i),
choices=map(lambda x: x[1], COUPLING_MODES),
labels=map(lambda x: x[0], COUPLING_MODES),
label='Coupling',
width=WIDTH,
)
#marker
channel_menu_box.AddStretchSpacer()
forms.drop_down(
parent=channel_menu_panel,
sizer=channel_menu_box,
ps=parent,
key=common.index_key(MARKER_KEY, i),
choices=map(lambda x: x[1], MARKER_TYPES),
labels=map(lambda x: x[0], MARKER_TYPES),
label='Marker',
width=WIDTH,
)
channel_menu_box.AddStretchSpacer()
##################################################
# Trigger Menu Box
##################################################
trigger_menu_panel = wx.Panel(options_notebook)
options_notebook_args.append(
(trigger_menu_panel, TRIGGER_PAGE_INDEX, 'Trig'))
trigger_menu_box = wx.BoxSizer(wx.VERTICAL)
trigger_menu_panel.SetSizer(trigger_menu_box)
#trigger mode
forms.drop_down(
parent=trigger_menu_panel,
sizer=trigger_menu_box,
ps=parent,
key=TRIGGER_MODE_KEY,
choices=map(lambda x: x[1], TRIGGER_MODES),
labels=map(lambda x: x[0], TRIGGER_MODES),
label='Mode',
width=WIDTH,
)
#trigger slope
trigger_slope_chooser = forms.drop_down(
parent=trigger_menu_panel,
sizer=trigger_menu_box,
ps=parent,
key=TRIGGER_SLOPE_KEY,
choices=map(lambda x: x[1], TRIGGER_SLOPES),
labels=map(lambda x: x[0], TRIGGER_SLOPES),
label='Slope',
width=WIDTH,
)
#trigger channel
trigger_channel_chooser = forms.drop_down(
parent=trigger_menu_panel,
sizer=trigger_menu_box,
ps=parent,
key=TRIGGER_CHANNEL_KEY,
choices=map(lambda x: x[1], CHANNELS),
labels=map(lambda x: x[0], CHANNELS),
label='Channel',
width=WIDTH,
)
#trigger level
hbox = wx.BoxSizer(wx.HORIZONTAL)
trigger_menu_box.Add(hbox, 0, wx.EXPAND)
hbox.Add(wx.StaticText(trigger_menu_panel, label='Level:'), 1,
wx.ALIGN_CENTER_VERTICAL)
trigger_level_button = forms.single_button(
parent=trigger_menu_panel,
sizer=hbox,
label='50%',
callback=parent.set_auto_trigger_level,
style=wx.BU_EXACTFIT,
)
hbox.AddSpacer(WIDTH - 60)
trigger_level_buttons = forms.incr_decr_buttons(
parent=trigger_menu_panel,
sizer=hbox,
on_incr=self._on_incr_trigger_level,
on_decr=self._on_decr_trigger_level,
)
def disable_all(trigger_mode):
for widget in (trigger_slope_chooser, trigger_channel_chooser,
trigger_level_buttons, trigger_level_button):
widget.Disable(trigger_mode == gr.gr_TRIG_MODE_FREE)
parent.subscribe(TRIGGER_MODE_KEY, disable_all)
disable_all(parent[TRIGGER_MODE_KEY])
##################################################
# XY Menu Box
##################################################
if parent.num_inputs > 1:
xy_menu_panel = wx.Panel(options_notebook)
options_notebook_args.append((xy_menu_panel, XY_PAGE_INDEX, 'XY'))
xy_menu_box = wx.BoxSizer(wx.VERTICAL)
xy_menu_panel.SetSizer(xy_menu_box)
#x and y channel choosers
xy_menu_box.AddStretchSpacer()
forms.drop_down(
parent=xy_menu_panel,
sizer=xy_menu_box,
ps=parent,
key=X_CHANNEL_KEY,
choices=map(lambda x: x[1], CHANNELS),
labels=map(lambda x: x[0], CHANNELS),
label='Channel X',
width=WIDTH,
)
xy_menu_box.AddStretchSpacer()
forms.drop_down(
parent=xy_menu_panel,
sizer=xy_menu_box,
ps=parent,
key=Y_CHANNEL_KEY,
choices=map(lambda x: x[1], CHANNELS),
labels=map(lambda x: x[0], CHANNELS),
label='Channel Y',
width=WIDTH,
)
#marker
xy_menu_box.AddStretchSpacer()
forms.drop_down(
parent=xy_menu_panel,
sizer=xy_menu_box,
ps=parent,
key=XY_MARKER_KEY,
choices=map(lambda x: x[1], MARKER_TYPES),
labels=map(lambda x: x[0], MARKER_TYPES),
label='Marker',
width=WIDTH,
)
xy_menu_box.AddStretchSpacer()
##################################################
# Setup Options Notebook
##################################################
forms.notebook(
parent=self,
sizer=chan_options_box,
notebook=options_notebook,
ps=parent,
key=CHANNEL_OPTIONS_KEY,
pages=map(lambda x: x[0], options_notebook_args),
choices=map(lambda x: x[1], options_notebook_args),
labels=map(lambda x: x[2], options_notebook_args),
)
#gui handling for channel options changing
def options_notebook_changed(chan_opt):
try:
parent[TRIGGER_SHOW_KEY] = chan_opt == TRIGGER_PAGE_INDEX
parent[XY_MODE_KEY] = chan_opt == XY_PAGE_INDEX
except wx.PyDeadObjectError:
pass
parent.subscribe(CHANNEL_OPTIONS_KEY, options_notebook_changed)
#gui handling for xy mode changing
def xy_mode_changed(mode):
#ensure xy tab is selected
if mode and parent[CHANNEL_OPTIONS_KEY] != XY_PAGE_INDEX:
parent[CHANNEL_OPTIONS_KEY] = XY_PAGE_INDEX
#ensure xy tab is not selected
elif not mode and parent[CHANNEL_OPTIONS_KEY] == XY_PAGE_INDEX:
parent[CHANNEL_OPTIONS_KEY] = 0
#show/hide control buttons
scope_mode_box.ShowItems(not mode)
xy_mode_box.ShowItems(mode)
control_box.Layout()
parent.subscribe(XY_MODE_KEY, xy_mode_changed)
xy_mode_changed(parent[XY_MODE_KEY])
##################################################
# Run/Stop Button
##################################################
#run/stop
control_box.AddStretchSpacer()
forms.toggle_button(
sizer=control_box,
parent=self,
true_label='Stop',
false_label='Run',
ps=parent,
key=RUNNING_KEY,
)
#set sizer
self.SetSizerAndFit(control_box)
#mouse wheel event
def on_mouse_wheel(event):
if not parent[XY_MODE_KEY]:
if event.GetWheelRotation() < 0: self._on_incr_t_divs(event)
else: self._on_decr_t_divs(event)
parent.plotter.Bind(wx.EVT_MOUSEWHEEL, on_mouse_wheel)
def __init__(
self,
parent,
title='',
sample_rate=1,
size=scope_window.DEFAULT_WIN_SIZE,
v_scale=0,
t_scale=0,
v_offset=0,
xy_mode=False,
ac_couple=False,
num_inputs=1,
trig_mode=scope_window.DEFAULT_TRIG_MODE,
y_axis_label='Counts',
frame_rate=scope_window.DEFAULT_FRAME_RATE,
use_persistence=False,
persist_alpha=None,
**kwargs #do not end with a comma
):
#ensure analog alpha
if persist_alpha is None:
actual_frame_rate = float(frame_rate)
analog_cutoff_freq = 0.5 # Hertz
#calculate alpha from wanted cutoff freq
persist_alpha = 1.0 - math.exp(
-2.0 * math.pi * analog_cutoff_freq / actual_frame_rate)
if not t_scale: t_scale = 10.0 / sample_rate
#init
gr.hier_block2.__init__(
self,
"scope_sink",
gr.io_signature(num_inputs, num_inputs, self._item_size),
gr.io_signature(0, 0, 0),
)
#scope
msgq = gr.msg_queue(2)
scope = gr.oscope_sink_f(sample_rate, msgq)
#controller
self.controller = pubsub()
self.controller.subscribe(SAMPLE_RATE_KEY, scope.set_sample_rate)
self.controller.publish(SAMPLE_RATE_KEY, scope.sample_rate)
self.controller.subscribe(DECIMATION_KEY, scope.set_decimation_count)
self.controller.publish(DECIMATION_KEY, scope.get_decimation_count)
self.controller.subscribe(TRIGGER_LEVEL_KEY, scope.set_trigger_level)
self.controller.publish(TRIGGER_LEVEL_KEY, scope.get_trigger_level)
self.controller.subscribe(TRIGGER_MODE_KEY, scope.set_trigger_mode)
self.controller.publish(TRIGGER_MODE_KEY, scope.get_trigger_mode)
self.controller.subscribe(TRIGGER_SLOPE_KEY, scope.set_trigger_slope)
self.controller.publish(TRIGGER_SLOPE_KEY, scope.get_trigger_slope)
self.controller.subscribe(TRIGGER_CHANNEL_KEY,
scope.set_trigger_channel)
self.controller.publish(TRIGGER_CHANNEL_KEY, scope.get_trigger_channel)
actual_num_inputs = self._real and num_inputs or num_inputs * 2
#init ac couple
for i in range(actual_num_inputs):
self.controller[common.index_key(AC_COUPLE_KEY, i)] = ac_couple
#start input watcher
common.input_watcher(msgq, self.controller, MSG_KEY)
#create window
self.win = scope_window.scope_window(
parent=parent,
controller=self.controller,
size=size,
title=title,
frame_rate=frame_rate,
num_inputs=actual_num_inputs,
sample_rate_key=SAMPLE_RATE_KEY,
t_scale=t_scale,
v_scale=v_scale,
v_offset=v_offset,
xy_mode=xy_mode,
trig_mode=trig_mode,
y_axis_label=y_axis_label,
ac_couple_key=AC_COUPLE_KEY,
trigger_level_key=TRIGGER_LEVEL_KEY,
trigger_mode_key=TRIGGER_MODE_KEY,
trigger_slope_key=TRIGGER_SLOPE_KEY,
trigger_channel_key=TRIGGER_CHANNEL_KEY,
decimation_key=DECIMATION_KEY,
msg_key=MSG_KEY,
use_persistence=use_persistence,
persist_alpha=persist_alpha,
)
common.register_access_methods(self, self.win)
#connect
if self._real:
for i in range(num_inputs):
self.wxgui_connect(
(self, i),
ac_couple_block(self.controller,
common.index_key(AC_COUPLE_KEY, i),
SAMPLE_RATE_KEY),
(scope, i),
)
else:
for i in range(num_inputs):
c2f = gr.complex_to_float()
self.wxgui_connect((self, i), c2f)
for j in range(2):
self.connect(
(c2f, j),
ac_couple_block(
self.controller,
common.index_key(AC_COUPLE_KEY, 2 * i + j),
SAMPLE_RATE_KEY),
(scope, 2 * i + j),
)
def __init__(
self,
parent,
controller,
size,
title,
frame_rate,
num_inputs,
sample_rate_key,
t_scale,
v_scale,
v_offset,
xy_mode,
ac_couple_key,
trigger_level_key,
trigger_mode_key,
trigger_slope_key,
trigger_channel_key,
decimation_key,
msg_key,
use_persistence,
persist_alpha,
trig_mode,
y_axis_label,
):
pubsub.pubsub.__init__(self)
#check num inputs
assert num_inputs <= len(CHANNEL_COLOR_SPECS)
#setup
self.sampleses = None
self.num_inputs = num_inputs
autorange = not v_scale
self.autorange_ts = 0
v_scale = v_scale or 1
self.frame_rate_ts = 0
#proxy the keys
self.proxy(MSG_KEY, controller, msg_key)
self.proxy(SAMPLE_RATE_KEY, controller, sample_rate_key)
self.proxy(TRIGGER_LEVEL_KEY, controller, trigger_level_key)
self.proxy(TRIGGER_MODE_KEY, controller, trigger_mode_key)
self.proxy(TRIGGER_SLOPE_KEY, controller, trigger_slope_key)
self.proxy(TRIGGER_CHANNEL_KEY, controller, trigger_channel_key)
self.proxy(DECIMATION_KEY, controller, decimation_key)
#initialize values
self[RUNNING_KEY] = True
self[XY_MARKER_KEY] = 2.0
self[CHANNEL_OPTIONS_KEY] = 0
self[XY_MODE_KEY] = xy_mode
self[X_CHANNEL_KEY] = 0
self[Y_CHANNEL_KEY] = self.num_inputs - 1
self[AUTORANGE_KEY] = autorange
self[T_PER_DIV_KEY] = t_scale
self[X_PER_DIV_KEY] = v_scale
self[Y_PER_DIV_KEY] = v_scale
self[T_OFF_KEY] = 0
self[X_OFF_KEY] = v_offset
self[Y_OFF_KEY] = v_offset
self[T_DIVS_KEY] = 8
self[X_DIVS_KEY] = 8
self[Y_DIVS_KEY] = 8
self[Y_AXIS_LABEL] = y_axis_label
self[FRAME_RATE_KEY] = frame_rate
self[TRIGGER_LEVEL_KEY] = 0
self[TRIGGER_CHANNEL_KEY] = 0
self[TRIGGER_MODE_KEY] = trig_mode
self[TRIGGER_SLOPE_KEY] = gr.gr_TRIG_SLOPE_POS
self[T_FRAC_OFF_KEY] = 0.5
self[USE_PERSISTENCE_KEY] = use_persistence
self[PERSIST_ALPHA_KEY] = persist_alpha
if self[TRIGGER_MODE_KEY] == gr.gr_TRIG_MODE_STRIPCHART:
self[T_FRAC_OFF_KEY] = 0.0
for i in range(num_inputs):
self.proxy(common.index_key(AC_COUPLE_KEY, i), controller,
common.index_key(ac_couple_key, i))
#init panel and plot
wx.Panel.__init__(self, parent, style=wx.SIMPLE_BORDER)
self.plotter = plotter.channel_plotter(self)
self.plotter.SetSize(wx.Size(*size))
self.plotter.SetSizeHints(*size)
self.plotter.set_title(title)
self.plotter.enable_legend(True)
self.plotter.enable_point_label(True)
self.plotter.enable_grid_lines(True)
self.plotter.set_use_persistence(use_persistence)
self.plotter.set_persist_alpha(persist_alpha)
#setup the box with plot and controls
self.control_panel = control_panel(self)
main_box = wx.BoxSizer(wx.HORIZONTAL)
main_box.Add(self.plotter, 1, wx.EXPAND)
main_box.Add(self.control_panel, 0, wx.EXPAND)
self.SetSizerAndFit(main_box)
#register events for message
self.subscribe(MSG_KEY, self.handle_msg)
#register events for grid
for key in [
common.index_key(MARKER_KEY, i) for i in range(self.num_inputs)
] + [
TRIGGER_LEVEL_KEY,
TRIGGER_MODE_KEY,
T_PER_DIV_KEY,
X_PER_DIV_KEY,
Y_PER_DIV_KEY,
T_OFF_KEY,
X_OFF_KEY,
Y_OFF_KEY,
T_DIVS_KEY,
X_DIVS_KEY,
Y_DIVS_KEY,
XY_MODE_KEY,
AUTORANGE_KEY,
T_FRAC_OFF_KEY,
TRIGGER_SHOW_KEY,
XY_MARKER_KEY,
X_CHANNEL_KEY,
Y_CHANNEL_KEY,
]:
self.subscribe(key, self.update_grid)
#register events for plotter settings
self.subscribe(USE_PERSISTENCE_KEY, self.plotter.set_use_persistence)
self.subscribe(PERSIST_ALPHA_KEY, self.plotter.set_persist_alpha)
#initial update
self.update_grid()
def handle_samples(self):
"""
Handle the cached samples from the scope input.
Perform ac coupling, triggering, and auto ranging.
"""
if not self.sampleses: return
sampleses = self.sampleses
if self[XY_MODE_KEY]:
self[DECIMATION_KEY] = 1
x_samples = sampleses[self[X_CHANNEL_KEY]]
y_samples = sampleses[self[Y_CHANNEL_KEY]]
#autorange
if self[AUTORANGE_KEY] and time.time(
) - self.autorange_ts > AUTORANGE_UPDATE_RATE:
x_min, x_max = common.get_min_max(x_samples)
y_min, y_max = common.get_min_max(y_samples)
#adjust the x per div
x_per_div = common.get_clean_num(
(x_max - x_min) / self[X_DIVS_KEY])
if x_per_div != self[X_PER_DIV_KEY]:
self[X_PER_DIV_KEY] = x_per_div
return
#adjust the x offset
x_off = x_per_div * round((x_max + x_min) / 2 / x_per_div)
if x_off != self[X_OFF_KEY]:
self[X_OFF_KEY] = x_off
return
#adjust the y per div
y_per_div = common.get_clean_num(
(y_max - y_min) / self[Y_DIVS_KEY])
if y_per_div != self[Y_PER_DIV_KEY]:
self[Y_PER_DIV_KEY] = y_per_div
return
#adjust the y offset
y_off = y_per_div * round((y_max + y_min) / 2 / y_per_div)
if y_off != self[Y_OFF_KEY]:
self[Y_OFF_KEY] = y_off
return
self.autorange_ts = time.time()
#plot xy channel
self.plotter.set_waveform(
channel='XY',
samples=(x_samples, y_samples),
color_spec=CHANNEL_COLOR_SPECS[0],
marker=self[XY_MARKER_KEY],
)
#turn off each waveform
for i, samples in enumerate(sampleses):
self.plotter.clear_waveform(channel='Ch%d' % (i + 1))
else:
#autorange
if self[AUTORANGE_KEY] and time.time(
) - self.autorange_ts > AUTORANGE_UPDATE_RATE:
bounds = [common.get_min_max(samples) for samples in sampleses]
y_min = numpy.min([bound[0] for bound in bounds])
y_max = numpy.max([bound[1] for bound in bounds])
#adjust the y per div
y_per_div = common.get_clean_num(
(y_max - y_min) / self[Y_DIVS_KEY])
if y_per_div != self[Y_PER_DIV_KEY]:
self[Y_PER_DIV_KEY] = y_per_div
return
#adjust the y offset
y_off = y_per_div * round((y_max + y_min) / 2 / y_per_div)
if y_off != self[Y_OFF_KEY]:
self[Y_OFF_KEY] = y_off
return
self.autorange_ts = time.time()
#number of samples to scale to the screen
actual_rate = self.get_actual_rate()
time_span = self[T_PER_DIV_KEY] * self[T_DIVS_KEY]
num_samps = int(round(time_span * actual_rate))
#handle the time offset
t_off = self[T_FRAC_OFF_KEY] * (len(sampleses[0]) / actual_rate -
time_span)
if t_off != self[T_OFF_KEY]:
self[T_OFF_KEY] = t_off
return
samps_off = int(round(actual_rate * self[T_OFF_KEY]))
#adjust the decim so that we use about half the samps
self[DECIMATION_KEY] = int(
round(time_span * self[SAMPLE_RATE_KEY] /
(0.5 * len(sampleses[0]))))
#num samps too small, auto increment the time
if num_samps < 2:
self[T_PER_DIV_KEY] = common.get_clean_incr(
self[T_PER_DIV_KEY])
#num samps in bounds, plot each waveform
elif num_samps <= len(sampleses[0]):
for i, samples in enumerate(sampleses):
#plot samples
self.plotter.set_waveform(
channel='Ch%d' % (i + 1),
samples=samples[samps_off:num_samps + samps_off],
color_spec=CHANNEL_COLOR_SPECS[i],
marker=self[common.index_key(MARKER_KEY, i)],
trig_off=self.trigger_offset,
)
#turn XY channel off
self.plotter.clear_waveform(channel='XY')
#keep trigger level within range
if self[TRIGGER_LEVEL_KEY] > self.get_y_max():
self[TRIGGER_LEVEL_KEY] = self.get_y_max()
return
if self[TRIGGER_LEVEL_KEY] < self.get_y_min():
self[TRIGGER_LEVEL_KEY] = self.get_y_min()
return
#disable the trigger channel
if not self[TRIGGER_SHOW_KEY] or self[XY_MODE_KEY] or self[
TRIGGER_MODE_KEY] == gr.gr_TRIG_MODE_FREE:
self.plotter.clear_waveform(channel='Trig')
else: #show trigger channel
trigger_level = self[TRIGGER_LEVEL_KEY]
trigger_point = (len(self.sampleses[0]) -
1) / self.get_actual_rate() / 2.0
self.plotter.set_waveform(
channel='Trig',
samples=([
self.get_t_min(), trigger_point, trigger_point,
trigger_point, trigger_point,
self.get_t_max()
], [
trigger_level, trigger_level,
self.get_y_max(),
self.get_y_min(), trigger_level, trigger_level
]),
color_spec=TRIGGER_COLOR_SPEC,
)
#update the plotter
self.plotter.update()
def __init__(self, parent):
"""
Create a new control panel.
Args:
parent: the wx parent window
"""
WIDTH = 90
self.parent = parent
wx.Panel.__init__(self, parent, style=wx.SUNKEN_BORDER)
parent[SHOW_CONTROL_PANEL_KEY] = True
parent.subscribe(SHOW_CONTROL_PANEL_KEY, self.Show)
control_box = wx.BoxSizer(wx.VERTICAL)
##################################################
# Persistence
##################################################
forms.check_box(
sizer=control_box, parent=self, label='Persistence',
ps=parent, key=USE_PERSISTENCE_KEY,
)
#static text and slider for analog alpha
persist_alpha_text = forms.static_text(
sizer=control_box, parent=self, label='Analog Alpha',
converter=forms.float_converter(lambda x: '%.4f'%x),
ps=parent, key=PERSIST_ALPHA_KEY, width=50,
)
persist_alpha_slider = forms.log_slider(
sizer=control_box, parent=self,
min_exp=PERSIST_ALPHA_MIN_EXP,
max_exp=PERSIST_ALPHA_MAX_EXP,
num_steps=SLIDER_STEPS,
ps=parent, key=PERSIST_ALPHA_KEY,
)
for widget in (persist_alpha_text, persist_alpha_slider):
parent.subscribe(USE_PERSISTENCE_KEY, widget.Enable)
widget.Enable(parent[USE_PERSISTENCE_KEY])
parent.subscribe(USE_PERSISTENCE_KEY, widget.ShowItems)
#always show initially, so room is reserved for them
widget.ShowItems(True) # (parent[USE_PERSISTENCE_KEY])
parent.subscribe(USE_PERSISTENCE_KEY, self._update_layout)
##################################################
# Axes Options
##################################################
control_box.AddStretchSpacer()
axes_options_box = forms.static_box_sizer(
parent=self, sizer=control_box, label='Axes Options',
bold=True, orient=wx.VERTICAL,
)
##################################################
# Scope Mode Box
##################################################
scope_mode_box = wx.BoxSizer(wx.VERTICAL)
axes_options_box.Add(scope_mode_box, 0, wx.EXPAND)
#x axis divs
forms.incr_decr_buttons(
parent=self, sizer=scope_mode_box, label='Secs/Div',
on_incr=self._on_incr_t_divs, on_decr=self._on_decr_t_divs,
)
#y axis divs
y_buttons_scope = forms.incr_decr_buttons(
parent=self, sizer=scope_mode_box, label='Counts/Div',
on_incr=self._on_incr_y_divs, on_decr=self._on_decr_y_divs,
)
#y axis ref lvl
y_off_buttons_scope = forms.incr_decr_buttons(
parent=self, sizer=scope_mode_box, label='Y Offset',
on_incr=self._on_incr_y_off, on_decr=self._on_decr_y_off,
)
#t axis ref lvl
scope_mode_box.AddSpacer(5)
forms.slider(
parent=self, sizer=scope_mode_box,
ps=parent, key=T_FRAC_OFF_KEY, label='T Offset',
minimum=0, maximum=1, num_steps=1000,
)
scope_mode_box.AddSpacer(5)
##################################################
# XY Mode Box
##################################################
xy_mode_box = wx.BoxSizer(wx.VERTICAL)
axes_options_box.Add(xy_mode_box, 0, wx.EXPAND)
#x div controls
x_buttons = forms.incr_decr_buttons(
parent=self, sizer=xy_mode_box, label='X/Div',
on_incr=self._on_incr_x_divs, on_decr=self._on_decr_x_divs,
)
#y div controls
y_buttons = forms.incr_decr_buttons(
parent=self, sizer=xy_mode_box, label='Y/Div',
on_incr=self._on_incr_y_divs, on_decr=self._on_decr_y_divs,
)
#x offset controls
x_off_buttons = forms.incr_decr_buttons(
parent=self, sizer=xy_mode_box, label='X Off',
on_incr=self._on_incr_x_off, on_decr=self._on_decr_x_off,
)
#y offset controls
y_off_buttons = forms.incr_decr_buttons(
parent=self, sizer=xy_mode_box, label='Y Off',
on_incr=self._on_incr_y_off, on_decr=self._on_decr_y_off,
)
for widget in (y_buttons_scope, y_off_buttons_scope, x_buttons, y_buttons, x_off_buttons, y_off_buttons):
parent.subscribe(AUTORANGE_KEY, widget.Disable)
widget.Disable(parent[AUTORANGE_KEY])
xy_mode_box.ShowItems(False)
#autorange check box
forms.check_box(
parent=self, sizer=axes_options_box, label='Autorange',
ps=parent, key=AUTORANGE_KEY,
)
##################################################
# Channel Options
##################################################
TRIGGER_PAGE_INDEX = parent.num_inputs
XY_PAGE_INDEX = parent.num_inputs+1
control_box.AddStretchSpacer()
chan_options_box = forms.static_box_sizer(
parent=self, sizer=control_box, label='Channel Options',
bold=True, orient=wx.VERTICAL,
)
options_notebook = wx.Notebook(self)
options_notebook_args = list()
CHANNELS = [('Ch %d'%(i+1), i) for i in range(parent.num_inputs)]
##################################################
# Channel Menu Boxes
##################################################
for i in range(parent.num_inputs):
channel_menu_panel = wx.Panel(options_notebook)
options_notebook_args.append((channel_menu_panel, i, 'Ch%d'%(i+1)))
channel_menu_box = wx.BoxSizer(wx.VERTICAL)
channel_menu_panel.SetSizer(channel_menu_box)
#ac couple check box
channel_menu_box.AddStretchSpacer()
forms.drop_down(
parent=channel_menu_panel, sizer=channel_menu_box,
ps=parent, key=common.index_key(AC_COUPLE_KEY, i),
choices=map(lambda x: x[1], COUPLING_MODES),
labels=map(lambda x: x[0], COUPLING_MODES),
label='Coupling', width=WIDTH,
)
#marker
channel_menu_box.AddStretchSpacer()
forms.drop_down(
parent=channel_menu_panel, sizer=channel_menu_box,
ps=parent, key=common.index_key(MARKER_KEY, i),
choices=map(lambda x: x[1], MARKER_TYPES),
labels=map(lambda x: x[0], MARKER_TYPES),
label='Marker', width=WIDTH,
)
channel_menu_box.AddStretchSpacer()
##################################################
# Trigger Menu Box
##################################################
trigger_menu_panel = wx.Panel(options_notebook)
options_notebook_args.append((trigger_menu_panel, TRIGGER_PAGE_INDEX, 'Trig'))
trigger_menu_box = wx.BoxSizer(wx.VERTICAL)
trigger_menu_panel.SetSizer(trigger_menu_box)
#trigger mode
forms.drop_down(
parent=trigger_menu_panel, sizer=trigger_menu_box,
ps=parent, key=TRIGGER_MODE_KEY,
choices=map(lambda x: x[1], TRIGGER_MODES),
labels=map(lambda x: x[0], TRIGGER_MODES),
label='Mode', width=WIDTH,
)
#trigger slope
trigger_slope_chooser = forms.drop_down(
parent=trigger_menu_panel, sizer=trigger_menu_box,
ps=parent, key=TRIGGER_SLOPE_KEY,
choices=map(lambda x: x[1], TRIGGER_SLOPES),
labels=map(lambda x: x[0], TRIGGER_SLOPES),
label='Slope', width=WIDTH,
)
#trigger channel
trigger_channel_chooser = forms.drop_down(
parent=trigger_menu_panel, sizer=trigger_menu_box,
ps=parent, key=TRIGGER_CHANNEL_KEY,
choices=map(lambda x: x[1], CHANNELS),
labels=map(lambda x: x[0], CHANNELS),
label='Channel', width=WIDTH,
)
#trigger level
hbox = wx.BoxSizer(wx.HORIZONTAL)
trigger_menu_box.Add(hbox, 0, wx.EXPAND)
hbox.Add(wx.StaticText(trigger_menu_panel, label='Level:'), 1, wx.ALIGN_CENTER_VERTICAL)
trigger_level_button = forms.single_button(
parent=trigger_menu_panel, sizer=hbox, label='50%',
callback=parent.set_auto_trigger_level, style=wx.BU_EXACTFIT,
)
hbox.AddSpacer(WIDTH-60)
trigger_level_buttons = forms.incr_decr_buttons(
parent=trigger_menu_panel, sizer=hbox,
on_incr=self._on_incr_trigger_level, on_decr=self._on_decr_trigger_level,
)
def disable_all(trigger_mode):
for widget in (trigger_slope_chooser, trigger_channel_chooser, trigger_level_buttons, trigger_level_button):
widget.Disable(trigger_mode == wxgui.TRIG_MODE_FREE)
parent.subscribe(TRIGGER_MODE_KEY, disable_all)
disable_all(parent[TRIGGER_MODE_KEY])
##################################################
# XY Menu Box
##################################################
if parent.num_inputs > 1:
xy_menu_panel = wx.Panel(options_notebook)
options_notebook_args.append((xy_menu_panel, XY_PAGE_INDEX, 'XY'))
xy_menu_box = wx.BoxSizer(wx.VERTICAL)
xy_menu_panel.SetSizer(xy_menu_box)
#x and y channel choosers
xy_menu_box.AddStretchSpacer()
forms.drop_down(
parent=xy_menu_panel, sizer=xy_menu_box,
ps=parent, key=X_CHANNEL_KEY,
choices=map(lambda x: x[1], CHANNELS),
labels=map(lambda x: x[0], CHANNELS),
label='Channel X', width=WIDTH,
)
xy_menu_box.AddStretchSpacer()
forms.drop_down(
parent=xy_menu_panel, sizer=xy_menu_box,
ps=parent, key=Y_CHANNEL_KEY,
choices=map(lambda x: x[1], CHANNELS),
labels=map(lambda x: x[0], CHANNELS),
label='Channel Y', width=WIDTH,
)
#marker
xy_menu_box.AddStretchSpacer()
forms.drop_down(
parent=xy_menu_panel, sizer=xy_menu_box,
ps=parent, key=XY_MARKER_KEY,
choices=map(lambda x: x[1], MARKER_TYPES),
labels=map(lambda x: x[0], MARKER_TYPES),
label='Marker', width=WIDTH,
)
xy_menu_box.AddStretchSpacer()
##################################################
# Setup Options Notebook
##################################################
forms.notebook(
parent=self, sizer=chan_options_box,
notebook=options_notebook,
ps=parent, key=CHANNEL_OPTIONS_KEY,
pages=map(lambda x: x[0], options_notebook_args),
choices=map(lambda x: x[1], options_notebook_args),
labels=map(lambda x: x[2], options_notebook_args),
)
#gui handling for channel options changing
def options_notebook_changed(chan_opt):
try:
parent[TRIGGER_SHOW_KEY] = chan_opt == TRIGGER_PAGE_INDEX
parent[XY_MODE_KEY] = chan_opt == XY_PAGE_INDEX
except wx.PyDeadObjectError: pass
parent.subscribe(CHANNEL_OPTIONS_KEY, options_notebook_changed)
#gui handling for xy mode changing
def xy_mode_changed(mode):
#ensure xy tab is selected
if mode and parent[CHANNEL_OPTIONS_KEY] != XY_PAGE_INDEX:
parent[CHANNEL_OPTIONS_KEY] = XY_PAGE_INDEX
#ensure xy tab is not selected
elif not mode and parent[CHANNEL_OPTIONS_KEY] == XY_PAGE_INDEX:
parent[CHANNEL_OPTIONS_KEY] = 0
#show/hide control buttons
scope_mode_box.ShowItems(not mode)
xy_mode_box.ShowItems(mode)
control_box.Layout()
parent.subscribe(XY_MODE_KEY, xy_mode_changed)
xy_mode_changed(parent[XY_MODE_KEY])
##################################################
# Run/Stop Button
##################################################
#run/stop
control_box.AddStretchSpacer()
forms.toggle_button(
sizer=control_box, parent=self,
true_label='Stop', false_label='Run',
ps=parent, key=RUNNING_KEY,
)
#set sizer
self.SetSizerAndFit(control_box)
#mouse wheel event
def on_mouse_wheel(event):
if not parent[XY_MODE_KEY]:
if event.GetWheelRotation() < 0: self._on_incr_t_divs(event)
else: self._on_decr_t_divs(event)
parent.plotter.Bind(wx.EVT_MOUSEWHEEL, on_mouse_wheel)
def handle_samples(self): """ Handle the cached samples from the scope input. Perform ac coupling, triggering, and auto ranging. """ if not self.sampleses: return sampleses = self.sampleses if self[XY_MODE_KEY]: self[DECIMATION_KEY] = 1 x_samples = sampleses[self[X_CHANNEL_KEY]] y_samples = sampleses[self[Y_CHANNEL_KEY]] #autorange if self[AUTORANGE_KEY] and time.time() - self.autorange_ts > AUTORANGE_UPDATE_RATE: x_min, x_max = common.get_min_max(x_samples) y_min, y_max = common.get_min_max(y_samples) #adjust the x per div x_per_div = common.get_clean_num((x_max-x_min)/self[X_DIVS_KEY]) if x_per_div != self[X_PER_DIV_KEY]: self[X_PER_DIV_KEY] = x_per_div; return #adjust the x offset x_off = x_per_div*round((x_max+x_min)/2/x_per_div) if x_off != self[X_OFF_KEY]: self[X_OFF_KEY] = x_off; return #adjust the y per div y_per_div = common.get_clean_num((y_max-y_min)/self[Y_DIVS_KEY]) if y_per_div != self[Y_PER_DIV_KEY]: self[Y_PER_DIV_KEY] = y_per_div; return #adjust the y offset y_off = y_per_div*round((y_max+y_min)/2/y_per_div) if y_off != self[Y_OFF_KEY]: self[Y_OFF_KEY] = y_off; return self.autorange_ts = time.time() #plot xy channel self.plotter.set_waveform( channel='XY', samples=(x_samples, y_samples), color_spec=CHANNEL_COLOR_SPECS[0], marker=self[XY_MARKER_KEY], ) #turn off each waveform for i, samples in enumerate(sampleses): self.plotter.clear_waveform(channel='Ch%d'%(i+1)) else: #autorange if self[AUTORANGE_KEY] and time.time() - self.autorange_ts > AUTORANGE_UPDATE_RATE: bounds = [common.get_min_max(samples) for samples in sampleses] y_min = numpy.min([bound[0] for bound in bounds]) y_max = numpy.max([bound[1] for bound in bounds]) #adjust the y per div y_per_div = common.get_clean_num((y_max-y_min)/self[Y_DIVS_KEY]) if y_per_div != self[Y_PER_DIV_KEY]: self[Y_PER_DIV_KEY] = y_per_div; return #adjust the y offset y_off = y_per_div*round((y_max+y_min)/2/y_per_div) if y_off != self[Y_OFF_KEY]: self[Y_OFF_KEY] = y_off; return self.autorange_ts = time.time() #number of samples to scale to the screen actual_rate = self.get_actual_rate() time_span = self[T_PER_DIV_KEY]*self[T_DIVS_KEY] num_samps = int(round(time_span*actual_rate)) #handle the time offset t_off = self[T_FRAC_OFF_KEY]*(len(sampleses[0])/actual_rate - time_span) if t_off != self[T_OFF_KEY]: self[T_OFF_KEY] = t_off; return samps_off = int(round(actual_rate*self[T_OFF_KEY])) #adjust the decim so that we use about half the samps self[DECIMATION_KEY] = int(round( time_span*self[SAMPLE_RATE_KEY]/(0.5*len(sampleses[0])) ) ) #num samps too small, auto increment the time if num_samps < 2: self[T_PER_DIV_KEY] = common.get_clean_incr(self[T_PER_DIV_KEY]) #num samps in bounds, plot each waveform elif num_samps <= len(sampleses[0]): for i, samples in enumerate(sampleses): #plot samples self.plotter.set_waveform( channel='Ch%d'%(i+1), samples=samples[samps_off:num_samps+samps_off], color_spec=CHANNEL_COLOR_SPECS[i], marker=self[common.index_key(MARKER_KEY, i)], trig_off=self.trigger_offset, ) #turn XY channel off self.plotter.clear_waveform(channel='XY') #keep trigger level within range if self[TRIGGER_LEVEL_KEY] > self.get_y_max(): self[TRIGGER_LEVEL_KEY] = self.get_y_max(); return if self[TRIGGER_LEVEL_KEY] < self.get_y_min(): self[TRIGGER_LEVEL_KEY] = self.get_y_min(); return #disable the trigger channel if not self[TRIGGER_SHOW_KEY] or self[XY_MODE_KEY] or self[TRIGGER_MODE_KEY] == wxgui.TRIG_MODE_FREE: self.plotter.clear_waveform(channel='Trig') else: #show trigger channel trigger_level = self[TRIGGER_LEVEL_KEY] trigger_point = (len(self.sampleses[0])-1)/self.get_actual_rate()/2.0 self.plotter.set_waveform( channel='Trig', samples=( [self.get_t_min(), trigger_point, trigger_point, trigger_point, trigger_point, self.get_t_max()], [trigger_level, trigger_level, self.get_y_max(), self.get_y_min(), trigger_level, trigger_level] ), color_spec=TRIGGER_COLOR_SPEC, ) #update the plotter self.plotter.update()
def __init__(
self,
parent,
controller,
size,
title,
frame_rate,
num_inputs,
sample_rate_key,
t_scale,
v_scale,
v_offset,
xy_mode,
ac_couple_key,
trigger_level_key,
trigger_mode_key,
trigger_slope_key,
trigger_channel_key,
decimation_key,
msg_key,
use_persistence,
persist_alpha,
trig_mode,
y_axis_label,
):
pubsub.pubsub.__init__(self)
#check num inputs
assert num_inputs <= len(CHANNEL_COLOR_SPECS)
#setup
self.sampleses = None
self.num_inputs = num_inputs
autorange = not v_scale
self.autorange_ts = 0
v_scale = v_scale or 1
self.frame_rate_ts = 0
#proxy the keys
self.proxy(MSG_KEY, controller, msg_key)
self.proxy(SAMPLE_RATE_KEY, controller, sample_rate_key)
self.proxy(TRIGGER_LEVEL_KEY, controller, trigger_level_key)
self.proxy(TRIGGER_MODE_KEY, controller, trigger_mode_key)
self.proxy(TRIGGER_SLOPE_KEY, controller, trigger_slope_key)
self.proxy(TRIGGER_CHANNEL_KEY, controller, trigger_channel_key)
self.proxy(DECIMATION_KEY, controller, decimation_key)
#initialize values
self[RUNNING_KEY] = True
self[XY_MARKER_KEY] = 2.0
self[CHANNEL_OPTIONS_KEY] = 0
self[XY_MODE_KEY] = xy_mode
self[X_CHANNEL_KEY] = 0
self[Y_CHANNEL_KEY] = self.num_inputs-1
self[AUTORANGE_KEY] = autorange
self[T_PER_DIV_KEY] = t_scale
self[X_PER_DIV_KEY] = v_scale
self[Y_PER_DIV_KEY] = v_scale
self[T_OFF_KEY] = 0
self[X_OFF_KEY] = v_offset
self[Y_OFF_KEY] = v_offset
self[T_DIVS_KEY] = 8
self[X_DIVS_KEY] = 8
self[Y_DIVS_KEY] = 8
self[Y_AXIS_LABEL] = y_axis_label
self[FRAME_RATE_KEY] = frame_rate
self[TRIGGER_LEVEL_KEY] = 0
self[TRIGGER_CHANNEL_KEY] = 0
self[TRIGGER_MODE_KEY] = trig_mode
self[TRIGGER_SLOPE_KEY] = wxgui.TRIG_SLOPE_POS
self[T_FRAC_OFF_KEY] = 0.5
self[USE_PERSISTENCE_KEY] = use_persistence
self[PERSIST_ALPHA_KEY] = persist_alpha
if self[TRIGGER_MODE_KEY] == wxgui.TRIG_MODE_STRIPCHART:
self[T_FRAC_OFF_KEY] = 0.0
for i in range(num_inputs):
self.proxy(common.index_key(AC_COUPLE_KEY, i), controller, common.index_key(ac_couple_key, i))
#init panel and plot
wx.Panel.__init__(self, parent, style=wx.SIMPLE_BORDER)
self.plotter = plotter.channel_plotter(self)
self.plotter.SetSize(wx.Size(*size))
self.plotter.SetSizeHints(*size)
self.plotter.set_title(title)
self.plotter.enable_legend(True)
self.plotter.enable_point_label(True)
self.plotter.enable_grid_lines(True)
self.plotter.set_use_persistence(use_persistence)
self.plotter.set_persist_alpha(persist_alpha)
#setup the box with plot and controls
self.control_panel = control_panel(self)
main_box = wx.BoxSizer(wx.HORIZONTAL)
main_box.Add(self.plotter, 1, wx.EXPAND)
main_box.Add(self.control_panel, 0, wx.EXPAND)
self.SetSizerAndFit(main_box)
#register events for message
self.subscribe(MSG_KEY, self.handle_msg)
#register events for grid
for key in [common.index_key(MARKER_KEY, i) for i in range(self.num_inputs)] + [
TRIGGER_LEVEL_KEY, TRIGGER_MODE_KEY,
T_PER_DIV_KEY, X_PER_DIV_KEY, Y_PER_DIV_KEY,
T_OFF_KEY, X_OFF_KEY, Y_OFF_KEY,
T_DIVS_KEY, X_DIVS_KEY, Y_DIVS_KEY,
XY_MODE_KEY, AUTORANGE_KEY, T_FRAC_OFF_KEY,
TRIGGER_SHOW_KEY, XY_MARKER_KEY, X_CHANNEL_KEY, Y_CHANNEL_KEY,
]: self.subscribe(key, self.update_grid)
#register events for plotter settings
self.subscribe(USE_PERSISTENCE_KEY, self.plotter.set_use_persistence)
self.subscribe(PERSIST_ALPHA_KEY, self.plotter.set_persist_alpha)
#initial update
self.update_grid()
def __init__( self, parent, title='', sample_rate=1, size=scope_window.DEFAULT_WIN_SIZE, v_scale=0, t_scale=0, xy_mode=False, ac_couple=False, num_inputs=1, frame_rate=scope_window.DEFAULT_FRAME_RATE, **kwargs #do not end with a comma ): if not t_scale: t_scale = 10.0/sample_rate #init gr.hier_block2.__init__( self, "scope_sink", gr.io_signature(num_inputs, num_inputs, self._item_size), gr.io_signature(0, 0, 0), ) #scope msgq = gr.msg_queue(2) scope = gr.oscope_sink_f(sample_rate, msgq) #controller self.controller = pubsub() self.controller.subscribe(SAMPLE_RATE_KEY, scope.set_sample_rate) self.controller.publish(SAMPLE_RATE_KEY, scope.sample_rate) self.controller.subscribe(DECIMATION_KEY, scope.set_decimation_count) self.controller.publish(DECIMATION_KEY, scope.get_decimation_count) self.controller.subscribe(TRIGGER_LEVEL_KEY, scope.set_trigger_level) self.controller.publish(TRIGGER_LEVEL_KEY, scope.get_trigger_level) self.controller.subscribe(TRIGGER_MODE_KEY, scope.set_trigger_mode) self.controller.publish(TRIGGER_MODE_KEY, scope.get_trigger_mode) self.controller.subscribe(TRIGGER_SLOPE_KEY, scope.set_trigger_slope) self.controller.publish(TRIGGER_SLOPE_KEY, scope.get_trigger_slope) self.controller.subscribe(TRIGGER_CHANNEL_KEY, scope.set_trigger_channel) self.controller.publish(TRIGGER_CHANNEL_KEY, scope.get_trigger_channel) #connect if self._real: for i in range(num_inputs): self.connect( (self, i), ac_couple_block(self.controller, common.index_key(AC_COUPLE_KEY, i), ac_couple, SAMPLE_RATE_KEY), (scope, i), ) else: for i in range(num_inputs): c2f = gr.complex_to_float() self.connect((self, i), c2f) for j in range(2): self.connect( (c2f, j), ac_couple_block(self.controller, common.index_key(AC_COUPLE_KEY, 2*i+j), ac_couple, SAMPLE_RATE_KEY), (scope, 2*i+j), ) num_inputs *= 2 #start input watcher common.input_watcher(msgq, self.controller, MSG_KEY) #create window self.win = scope_window.scope_window( parent=parent, controller=self.controller, size=size, title=title, frame_rate=frame_rate, num_inputs=num_inputs, sample_rate_key=SAMPLE_RATE_KEY, t_scale=t_scale, v_scale=v_scale, xy_mode=xy_mode, ac_couple_key=AC_COUPLE_KEY, trigger_level_key=TRIGGER_LEVEL_KEY, trigger_mode_key=TRIGGER_MODE_KEY, trigger_slope_key=TRIGGER_SLOPE_KEY, trigger_channel_key=TRIGGER_CHANNEL_KEY, decimation_key=DECIMATION_KEY, msg_key=MSG_KEY, ) common.register_access_methods(self, self.win)
def __init__(
self,
parent,
title='',
sample_rate=1,
size=scope_window.DEFAULT_WIN_SIZE,
v_scale=0,
t_scale=0,
v_offset=0,
xy_mode=False,
ac_couple=False,
num_inputs=1,
trig_mode=scope_window.DEFAULT_TRIG_MODE,
y_axis_label='Counts',
frame_rate=scope_window.DEFAULT_FRAME_RATE,
use_persistence=False,
persist_alpha=None,
**kwargs #do not end with a comma
):
#ensure analog alpha
if persist_alpha is None:
actual_frame_rate=float(frame_rate)
analog_cutoff_freq=0.5 # Hertz
#calculate alpha from wanted cutoff freq
persist_alpha = 1.0 - math.exp(-2.0*math.pi*analog_cutoff_freq/actual_frame_rate)
if not t_scale: t_scale = 10.0/sample_rate
#init
gr.hier_block2.__init__(
self,
"scope_sink",
gr.io_signature(num_inputs, num_inputs, self._item_size),
gr.io_signature(0, 0, 0),
)
#scope
msgq = gr.msg_queue(2)
scope = gr.oscope_sink_f(sample_rate, msgq)
#controller
self.controller = pubsub()
self.controller.subscribe(SAMPLE_RATE_KEY, scope.set_sample_rate)
self.controller.publish(SAMPLE_RATE_KEY, scope.sample_rate)
self.controller.subscribe(DECIMATION_KEY, scope.set_decimation_count)
self.controller.publish(DECIMATION_KEY, scope.get_decimation_count)
self.controller.subscribe(TRIGGER_LEVEL_KEY, scope.set_trigger_level)
self.controller.publish(TRIGGER_LEVEL_KEY, scope.get_trigger_level)
self.controller.subscribe(TRIGGER_MODE_KEY, scope.set_trigger_mode)
self.controller.publish(TRIGGER_MODE_KEY, scope.get_trigger_mode)
self.controller.subscribe(TRIGGER_SLOPE_KEY, scope.set_trigger_slope)
self.controller.publish(TRIGGER_SLOPE_KEY, scope.get_trigger_slope)
self.controller.subscribe(TRIGGER_CHANNEL_KEY, scope.set_trigger_channel)
self.controller.publish(TRIGGER_CHANNEL_KEY, scope.get_trigger_channel)
actual_num_inputs = self._real and num_inputs or num_inputs*2
#init ac couple
for i in range(actual_num_inputs):
self.controller[common.index_key(AC_COUPLE_KEY, i)] = ac_couple
#start input watcher
common.input_watcher(msgq, self.controller, MSG_KEY)
#create window
self.win = scope_window.scope_window(
parent=parent,
controller=self.controller,
size=size,
title=title,
frame_rate=frame_rate,
num_inputs=actual_num_inputs,
sample_rate_key=SAMPLE_RATE_KEY,
t_scale=t_scale,
v_scale=v_scale,
v_offset=v_offset,
xy_mode=xy_mode,
trig_mode=trig_mode,
y_axis_label=y_axis_label,
ac_couple_key=AC_COUPLE_KEY,
trigger_level_key=TRIGGER_LEVEL_KEY,
trigger_mode_key=TRIGGER_MODE_KEY,
trigger_slope_key=TRIGGER_SLOPE_KEY,
trigger_channel_key=TRIGGER_CHANNEL_KEY,
decimation_key=DECIMATION_KEY,
msg_key=MSG_KEY,
use_persistence=use_persistence,
persist_alpha=persist_alpha,
)
common.register_access_methods(self, self.win)
#connect
if self._real:
for i in range(num_inputs):
self.wxgui_connect(
(self, i),
ac_couple_block(self.controller, common.index_key(AC_COUPLE_KEY, i), SAMPLE_RATE_KEY),
(scope, i),
)
else:
for i in range(num_inputs):
c2f = gr.complex_to_float()
self.wxgui_connect((self, i), c2f)
for j in range(2):
self.connect(
(c2f, j),
ac_couple_block(self.controller, common.index_key(AC_COUPLE_KEY, 2*i+j), SAMPLE_RATE_KEY),
(scope, 2*i+j),
)
