def set_bits_per_symbol(self, bits_per_symbol):
self.bits_per_symbol = bits_per_symbol
self.set_payload_mod(
digital.qam_constellation(
constellation_points=2**self.bits_per_symbol))
self.analog_noise_source_x_0.set_amplitude(
np.sqrt(10**(np.log10(1 / self.bits_per_symbol) -
(self.mag / 10.0))) * 1)
self.blocks_repack_bits_bb_0_0_1.set_k_and_l(self.bits_per_symbol, 1)
self.blocks_repack_bits_bb_0_0_1_1.set_k_and_l(self.bits_per_symbol, 1)
def make_constellation_object(params):
gr_params = {}
constellation_str = params['constellation']
gr_params['m'] = params['order']
gr_params['differential'] = params.get('differential',
False) # give it a default
mod_code_str = params.get('mod_code', 'GRAY')
if mod_code_str == 'GRAY' or mod_code_str == 'gray':
gr_params['mod_code'] = digital.mod_codes.GRAY_CODE
elif mod_code_str == 'no_code' or mod_code_str == 'NO_CODE':
gr_params['mod_code'] = digital.mod_codes.NO_CODE
else:
logger.error(
'The provided mod code for the generic modulator is not recognized'
)
raise AttributeError('I do not recognize this mod_code')
logger.debug('these are the params %s', gr_params)
if constellation_str == 'psk':
return digital.psk_constellation(**gr_params)
elif constellation_str == 'qam':
return digital.qam_constellation(**gr_params)
else:
logger.error('The provided constellation is not supported')
raise AttributeError('I do not recognize this constellation')
def __init__(self):
gr.top_block.__init__(self, "BER Simulation")
Qt.QWidget.__init__(self)
self.setWindowTitle("BER Simulation")
qtgui.util.check_set_qss()
try:
self.setWindowIcon(Qt.QIcon.fromTheme('gnuradio-grc'))
except:
pass
self.top_scroll_layout = Qt.QVBoxLayout()
self.setLayout(self.top_scroll_layout)
self.top_scroll = Qt.QScrollArea()
self.top_scroll.setFrameStyle(Qt.QFrame.NoFrame)
self.top_scroll_layout.addWidget(self.top_scroll)
self.top_scroll.setWidgetResizable(True)
self.top_widget = Qt.QWidget()
self.top_scroll.setWidget(self.top_widget)
self.top_layout = Qt.QVBoxLayout(self.top_widget)
self.top_grid_layout = Qt.QGridLayout()
self.top_layout.addLayout(self.top_grid_layout)
self.settings = Qt.QSettings("GNU Radio", "ber_sim")
try:
if StrictVersion(Qt.qVersion()) < StrictVersion("5.0.0"):
self.restoreGeometry(
self.settings.value("geometry").toByteArray())
else:
self.restoreGeometry(self.settings.value("geometry"))
except:
pass
##################################################
# Variables
##################################################
self.constellation = constellation = digital.qam_constellation(
64, True, digital.mod_codes.GRAY_CODE, True)
self.bits_per_symbol = bits_per_symbol = constellation.bits_per_symbol(
)
self.EbN0 = EbN0 = 15
self.sym_map = sym_map = [
0, 1, 2, 3, 6, 7, 4, 5, 25, 24, 27, 26, 31, 30, 29, 28, 18, 19, 16,
17, 20, 21, 22, 23, 11, 10, 9, 8, 13, 12, 15, 14
]
self.samp_rate = samp_rate = 100000
self.const_points = const_points = [(-7 + 7j), (-3 + 7j), (1 + 7j),
(5 + 7j), (-5 + 5j), (-1 + 5j),
(3 + 5j), (7 + 5j), (-7 + 3j),
(-3 + 3j), (1 + 3j), (5 + 3j),
(-5 + 1j), (-1 + 1j), (3 + 1j),
(7 + 1j), (-7 - 1j), (-3 - 1j),
(1 - 1j), (5 - 1j), (-5 - 3j),
(-1 - 3j), (3 - 3j), (7 - 3j),
(-7 - 5j), (-3 - 5j), (1 - 5j),
(5 - 5j), (-5 - 7j), (-1 - 7j),
(3 - 7j), (7 - 7j)]
self.EsN0 = EsN0 = EbN0 + 10 * math.log10(bits_per_symbol)
##################################################
# Blocks
##################################################
self.qtgui_number_sink_0 = qtgui.number_sink(gr.sizeof_float, 0,
qtgui.NUM_GRAPH_HORIZ, 1)
self.qtgui_number_sink_0.set_update_time(0.10)
self.qtgui_number_sink_0.set_title("")
labels = ['', '', '', '', '', '', '', '', '', '']
units = ['', '', '', '', '', '', '', '', '', '']
colors = [("black", "black"), ("black", "black"), ("black", "black"),
("black", "black"), ("black", "black"), ("black", "black"),
("black", "black"), ("black", "black"), ("black", "black"),
("black", "black")]
factor = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
for i in range(1):
self.qtgui_number_sink_0.set_min(i, 0)
self.qtgui_number_sink_0.set_max(i, 1)
self.qtgui_number_sink_0.set_color(i, colors[i][0], colors[i][1])
if len(labels[i]) == 0:
self.qtgui_number_sink_0.set_label(i, "Data {0}".format(i))
else:
self.qtgui_number_sink_0.set_label(i, labels[i])
self.qtgui_number_sink_0.set_unit(i, units[i])
self.qtgui_number_sink_0.set_factor(i, factor[i])
self.qtgui_number_sink_0.enable_autoscale(False)
self._qtgui_number_sink_0_win = sip.wrapinstance(
self.qtgui_number_sink_0.pyqwidget(), Qt.QWidget)
self.top_grid_layout.addWidget(self._qtgui_number_sink_0_win)
self.qtgui_const_sink_x_0 = qtgui.const_sink_c(
1024, #size
"", #name
1 #number of inputs
)
self.qtgui_const_sink_x_0.set_update_time(0.10)
self.qtgui_const_sink_x_0.set_y_axis(-2, 2)
self.qtgui_const_sink_x_0.set_x_axis(-2, 2)
self.qtgui_const_sink_x_0.set_trigger_mode(qtgui.TRIG_MODE_FREE,
qtgui.TRIG_SLOPE_POS, 0.0,
0, "")
self.qtgui_const_sink_x_0.enable_autoscale(False)
self.qtgui_const_sink_x_0.enable_grid(False)
self.qtgui_const_sink_x_0.enable_axis_labels(True)
labels = ['', '', '', '', '', '', '', '', '', '']
widths = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
colors = [
"blue", "red", "red", "red", "red", "red", "red", "red", "red",
"red"
]
styles = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
markers = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
alphas = [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
for i in range(1):
if len(labels[i]) == 0:
self.qtgui_const_sink_x_0.set_line_label(
i, "Data {0}".format(i))
else:
self.qtgui_const_sink_x_0.set_line_label(i, labels[i])
self.qtgui_const_sink_x_0.set_line_width(i, widths[i])
self.qtgui_const_sink_x_0.set_line_color(i, colors[i])
self.qtgui_const_sink_x_0.set_line_style(i, styles[i])
self.qtgui_const_sink_x_0.set_line_marker(i, markers[i])
self.qtgui_const_sink_x_0.set_line_alpha(i, alphas[i])
self._qtgui_const_sink_x_0_win = sip.wrapinstance(
self.qtgui_const_sink_x_0.pyqwidget(), Qt.QWidget)
self.top_grid_layout.addWidget(self._qtgui_const_sink_x_0_win)
self.fec_ber_bf_0 = fec.ber_bf(False, 100, -7.0)
self.digital_constellation_decoder_cb_0 = digital.constellation_decoder_cb(
constellation.base())
self.digital_chunks_to_symbols_xx_0 = digital.chunks_to_symbols_bc(
constellation.points(), 1)
self.blocks_throttle_0 = blocks.throttle(gr.sizeof_char * 1, samp_rate,
True)
self.blocks_add_xx_0 = blocks.add_vcc(1)
self.analog_random_source_x_0 = blocks.vector_source_b(
list(
map(int,
numpy.random.randint(0, constellation.arity(), 10000000))),
True)
self.analog_noise_source_x_0 = analog.noise_source_c(
analog.GR_GAUSSIAN, 1.0 / math.sqrt(10**(EsN0 / 10)), 42)
self._EbN0_range = Range(-2, 30, 1, 15, 200)
self._EbN0_win = RangeWidget(self._EbN0_range, self.set_EbN0,
'Eb / N0 (dB)', "counter_slider", float)
self.top_grid_layout.addWidget(self._EbN0_win)
##################################################
# Connections
##################################################
self.connect((self.analog_noise_source_x_0, 0),
(self.blocks_add_xx_0, 1))
self.connect((self.analog_random_source_x_0, 0),
(self.blocks_throttle_0, 0))
self.connect((self.blocks_add_xx_0, 0),
(self.digital_constellation_decoder_cb_0, 0))
self.connect((self.blocks_add_xx_0, 0), (self.qtgui_const_sink_x_0, 0))
self.connect((self.blocks_throttle_0, 0),
(self.digital_chunks_to_symbols_xx_0, 0))
self.connect((self.blocks_throttle_0, 0), (self.fec_ber_bf_0, 0))
self.connect((self.digital_chunks_to_symbols_xx_0, 0),
(self.blocks_add_xx_0, 0))
self.connect((self.digital_constellation_decoder_cb_0, 0),
(self.fec_ber_bf_0, 1))
self.connect((self.fec_ber_bf_0, 0), (self.qtgui_number_sink_0, 0))
def __init__(self, doc, ip="127.0.0.1"):
gr.top_block.__init__(self, "RL Demod training")
self.doc = doc
self.plot_lst = []
self.widget_lst = []
##################################################
# Parameters
##################################################
self.ip = ip
##################################################
# Variables
##################################################
self.pilot_symbols = pilot_symbols = ((
1,
1,
1,
-1,
), )
self.pilot_carriers = pilot_carriers = ((
-21,
-7,
7,
21,
), )
self.packet_length_tag_key = packet_length_tag_key = "packet_length"
self.occupied_carriers = occupied_carriers = (
list(range(-22, -21)) + list(range(-20, -7)) + list(range(-6, 0)) +
list(range(1, 7)) + list(range(8, 21)) + list(range(22, 23)), )
self.length_tag_key_0 = length_tag_key_0 = "frame_len"
self.length_tag_key = length_tag_key = "packet_len"
self.header_mod = header_mod = digital.constellation_bpsk()
self.fft_len = fft_len = 64
self.errors = errors = 0
self.bits_per_symbol = bits_per_symbol = 4
self.ber = ber = 0
self.variable_qtgui_label_0_0_val = variable_qtgui_label_0_0 = ber
self.variable_qtgui_label_0_val = variable_qtgui_label_0 = errors
self.variable_0 = variable_0 = len(occupied_carriers[0])
self.training_mod = training_mod = 1
self.timestamp = timestamp = '1'
self.t_state = t_state = 1
self.sync_word2 = sync_word2 = [
0, 0, 0, 0, 0, 0, -1, -1, -1, -1, 1, 1, -1, -1, -1, 1, -1, 1, 1, 1,
1, 1, -1, -1, -1, -1, -1, 1, -1, -1, 1, -1, 0, 1, -1, 1, 1, 1, -1,
1, 1, 1, -1, 1, 1, 1, 1, -1, 1, -1, -1, -1, 1, -1, 1, -1, -1, -1,
-1, 0, 0, 0, 0, 0
]
self.sync_word1 = sync_word1 = [
0., 0., 0., 0., 0., 0., 0., 1.41421356, 0., -1.41421356, 0.,
1.41421356, 0., -1.41421356, 0., -1.41421356, 0., -1.41421356, 0.,
1.41421356, 0., -1.41421356, 0., 1.41421356, 0., -1.41421356, 0.,
-1.41421356, 0., -1.41421356, 0., -1.41421356, 0., 1.41421356, 0.,
-1.41421356, 0., 1.41421356, 0., 1.41421356, 0., 1.41421356, 0.,
-1.41421356, 0., 1.41421356, 0., 1.41421356, 0., 1.41421356, 0.,
-1.41421356, 0., 1.41421356, 0., 1.41421356, 0., 1.41421356, 0.,
0., 0., 0., 0., 0.
]
self.snr_stop = snr_stop = 29
self.snr_step = snr_step = 0.5
self.snr_start = snr_start = 0
self.samp_rate = samp_rate = 0.3e6
self.rx_lr = rx_lr = -2
self.rolloff = rolloff = 0
self.puncpat = puncpat = '11'
self.payload_mod = payload_mod = digital.qam_constellation(
constellation_points=2**bits_per_symbol)
self.payload_equalizer = payload_equalizer = digital.ofdm_equalizer_static(
fft_len, occupied_carriers, pilot_carriers, pilot_symbols, 1)
self.packet_len = packet_len = 400
self.mag = mag = 10
self.ldpc_enc = ldpc_enc = fec.ldpc_encoder_make(
gr.prefix() + "/share/gnuradio/fec/ldpc/" +
"n_0100_k_0042_gap_02.alist")
self.ldpc_dec = ldpc_dec = fec.ldpc_decoder.make(
gr.prefix() + "/share/gnuradio/fec/ldpc/" +
"n_0100_k_0042_gap_02.alist", 0.5, 50)
self.header_formatter = header_formatter = digital.packet_header_ofdm(
occupied_carriers,
n_syms=1,
len_tag_key=packet_length_tag_key,
frame_len_tag_key=length_tag_key_0,
bits_per_header_sym=header_mod.bits_per_symbol(),
bits_per_payload_sym=8,
scramble_header=False)
self.header_equalizer = header_equalizer = digital.ofdm_equalizer_simpledfe(
fft_len, header_mod.base(), occupied_carriers, pilot_carriers,
pilot_symbols)
self.hdr_format = hdr_format = digital.header_format_ofdm(
occupied_carriers,
1,
length_tag_key,
)
self.gain_rx = gain_rx = 5
self.freq = freq = 900e6
self.bits_per_symbol_0 = bits_per_symbol_0 = len(occupied_carriers[0])
##################################################
# Blocks
##################################################
self.training_mod_textbox = bokehgui.textbox(
self.widget_lst, str(1), 'Alternate training every' + ": ")
self.training_mod_textbox.add_callback(
lambda attr, old, new: self.set_training_mod(int(new)))
self._t_state_options = [
1,
0,
]
self._t_state_labels = [
'On',
'Off',
]
self.t_state_radiobutton = bokehgui.radiobutton(self.widget_lst,
None,
self._t_state_labels,
inline=True)
self.t_state_radiobutton.add_callback(
lambda new: self.set_t_state(int(self._t_state_options[new])))
self.rx_lr_slider = bokehgui.slider(self.widget_lst,
'RX Learning rate (log)' + ":", -5,
0, 0.1, 1, -2)
self.rx_lr_slider.add_callback(
lambda attr, old, new: self.set_rx_lr(new))
self.mag_slider = bokehgui.slider(self.widget_lst, 'Eb/N0' + ":", 0,
40, 0.1, 1, 10)
self.mag_slider.add_callback(lambda attr, old, new: self.set_mag(new))
self.learning_ber_bf_0 = learning.ber_bf(False, 100, -7.0, 1)
self.gain_rx_slider = bokehgui.slider(self.widget_lst,
'Amplitude Rx' + ":", 0, 90, 0.5,
1, 5)
self.gain_rx_slider.add_callback(
lambda attr, old, new: self.set_gain_rx(new))
self.blocks_probe_signal_x_0 = blocks.probe_signal_f()
self.zeromq_sub_msg_source_0_0 = zeromq.sub_msg_source(
"tcp://" + ip + ":50001", 1000)
self.zeromq_pub_msg_sink_0_0_0 = zeromq.pub_msg_sink(
'tcp://*:50002', 1000)
self.variable_qtgui_label_0_0 = bokehgui.label(
self.widget_lst, str(variable_qtgui_label_0_0), 'BER' + ": ")
self.variable_qtgui_label_0 = bokehgui.label(
self.widget_lst, str(variable_qtgui_label_0), 'Error count' + ": ")
self.uhd_usrp_source_0 = uhd.usrp_source(
",".join(("", "")),
uhd.stream_args(
cpu_format="fc32",
args='',
channels=list(range(0, 1)),
),
)
self.uhd_usrp_source_0.set_samp_rate(samp_rate)
self.uhd_usrp_source_0.set_time_now(uhd.time_spec(time.time()),
uhd.ALL_MBOARDS)
self.uhd_usrp_source_0.set_center_freq(freq, 0)
self.uhd_usrp_source_0.set_antenna("TX/RX", 0)
self.uhd_usrp_source_0.set_rx_agc(False, 0)
self.uhd_usrp_source_0.set_gain(gain_rx, 0)
self.timestamp_textbox = bokehgui.textbox(self.widget_lst, str('1'),
'Load timestamp' + ": ")
self.timestamp_textbox.add_callback(
lambda attr, old, new: self.set_timestamp(str(new)))
self.learning_dl_demod_0 = learning.dl_demod(
'packet_num', False, bits_per_symbol, packet_len, 20, training_mod,
10**rx_lr, t_state,
'/home/cyrille/Gnu-Radio/modules/gr-learning/examples/saved_models'
)
self.learning_align_0 = learning.align('packet_num', packet_len, 1, 48)
self.fft_vxx_1 = fft.fft_vcc(fft_len, True, (), True, 1)
self.fft_vxx_0 = fft.fft_vcc(fft_len, True, (), True, 1)
def _errors_probe():
while True:
val = self.learning_ber_bf_0.total_errors()
try:
self.set_errors(val)
except AttributeError:
pass
time.sleep(1.0 / (0.5))
_errors_thread = threading.Thread(target=_errors_probe)
_errors_thread.daemon = True
_errors_thread.start()
self.digital_packet_headerparser_b_0 = digital.packet_headerparser_b(
header_formatter.base())
self.digital_ofdm_sync_sc_cfb_0 = digital.ofdm_sync_sc_cfb(
fft_len, fft_len // 4, False, 0.9)
self.digital_ofdm_serializer_vcc_payload = digital.ofdm_serializer_vcc(
fft_len, occupied_carriers, length_tag_key_0,
packet_length_tag_key, 1, '', True)
self.digital_ofdm_serializer_vcc_header = digital.ofdm_serializer_vcc(
fft_len, occupied_carriers, length_tag_key_0, '', 0, '', True)
self.digital_ofdm_frame_equalizer_vcvc_1 = digital.ofdm_frame_equalizer_vcvc(
payload_equalizer.base(), fft_len // 4, length_tag_key_0, True, 0)
self.digital_ofdm_frame_equalizer_vcvc_0 = digital.ofdm_frame_equalizer_vcvc(
header_equalizer.base(), fft_len // 4, length_tag_key_0, True, 1)
self.digital_ofdm_chanest_vcvc_0 = digital.ofdm_chanest_vcvc(
sync_word1, sync_word2, 1, 0, 3, False)
self.digital_header_payload_demux_0 = digital.header_payload_demux(
3, fft_len, fft_len // 4, length_tag_key_0, "", True,
gr.sizeof_gr_complex, "rx_time", int(samp_rate), (), 0)
self.digital_constellation_decoder_cb_0 = digital.constellation_decoder_cb(
header_mod.base())
self.bokehgui_time_sink_x_0 = bokehgui.time_sink_f_proc(
1024, samp_rate, "Decoded minus label", 1)
self.bokehgui_time_sink_x_0_plot = bokehgui.time_sink_f(
self.doc,
self.plot_lst,
self.bokehgui_time_sink_x_0,
is_message=False)
labels = ['', '', '', '', '', '', '', '', '', '']
legend_list = []
for i in range(1):
if len(labels[i]) == 0:
legend_list.append("Data {0}".format(i))
else:
legend_list.append(labels[i])
self.bokehgui_time_sink_x_0_plot.initialize(log_x=False,
log_y=False,
update_time=500,
legend_list=legend_list)
self.bokehgui_time_sink_x_0_plot.set_y_axis([-10, 10])
self.bokehgui_time_sink_x_0_plot.set_y_label('Difference' + '(' + "" +
')')
self.bokehgui_time_sink_x_0_plot.set_x_label('Symbol' + '(' + "" + ')')
self.bokehgui_time_sink_x_0_plot.enable_tags(-1, True)
self.bokehgui_time_sink_x_0_plot.set_trigger_mode(
bokehgui.TRIG_MODE_FREE, bokehgui.TRIG_SLOPE_POS, 0.0, 0, 0, "")
self.bokehgui_time_sink_x_0_plot.enable_grid(False)
self.bokehgui_time_sink_x_0_plot.enable_axis_labels(True)
self.bokehgui_time_sink_x_0_plot.disable_legend(not True)
self.bokehgui_time_sink_x_0_plot.set_layout(*((2, 0, 1, 3)))
colors = [
"blue", "red", "green", "black", "cyan", "magenta", "yellow",
"blue", "blue", "blue"
]
widths = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
styles = [
"solid", "solid", "solid", "solid", "solid", "solid", "solid",
"solid", "solid", "solid"
]
markers = [None, None, None, None, None, None, None, None, None, None]
alphas = [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
for i in range(1):
self.bokehgui_time_sink_x_0_plot.format_line(
i, colors[i], widths[i], styles[i], markers[i], alphas[i])
self.bokehgui_time_const_x_0 = bokehgui.time_sink_c_proc(
1024, samp_rate, "Received noisy constellation", 1)
self.bokehgui_time_const_x_0_plot = bokehgui.const_sink_c(
self.doc,
self.plot_lst,
self.bokehgui_time_const_x_0,
is_message=False)
labels = ['', '', '', '', '', '', '', '', '', '']
legend_list = []
for i in range(1):
if len(labels[i]) == 0:
if (i % 2 == 0):
legend_list.append("Re{{Data {0}}}".format(i / 2))
else:
legend_list.append("Im{{Data {0}}}".format(i / 2))
else:
legend_list.append(labels[i])
self.bokehgui_time_const_x_0_plot.initialize(update_time=500,
legend_list=legend_list)
self.bokehgui_time_const_x_0_plot.set_y_axis([-2, 2])
self.bokehgui_time_const_x_0_plot.set_y_label('Q Channel' + '(' + "" +
')')
self.bokehgui_time_const_x_0_plot.set_x_label('I Channel' + '(' + "" +
')')
self.bokehgui_time_const_x_0_plot.enable_tags(-1, False)
self.bokehgui_time_const_x_0_plot.set_trigger_mode(
bokehgui.TRIG_MODE_FREE, bokehgui.TRIG_SLOPE_POS, 0.0, 0, 0, "")
self.bokehgui_time_const_x_0_plot.enable_grid(True)
self.bokehgui_time_const_x_0_plot.enable_axis_labels(True)
self.bokehgui_time_const_x_0_plot.disable_legend(not True)
self.bokehgui_time_const_x_0_plot.set_layout(*((0, 1, 2, 2)))
colors = [
"blue", "red", "green", "black", "cyan", "magenta", "yellow",
"blue", "blue", "blue"
]
widths = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
markers = ['o', 'o', 'o', 'o', 'o', 'o', 'o', 'o', 'o', 'o']
alphas = [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
for i in range(1):
self.bokehgui_time_const_x_0_plot.format_line(
i, colors[i], widths[i], 'None', markers[i], alphas[i])
self.bokehgui_frequency_sink_x_0 = bokehgui.freq_sink_c_proc(
1024, firdes.WIN_BLACKMAN_hARRIS, freq, samp_rate, "Rx signal", 1)
self.bokehgui_frequency_sink_x_0_plot = bokehgui.freq_sink_c(
self.doc,
self.plot_lst,
self.bokehgui_frequency_sink_x_0,
is_message=False)
labels = ['', '', '', '', '', '', '', '', '', '']
legend_list = []
for i in range(1):
if len(labels[i]) == 0:
legend_list.append("Data {0}".format(i))
else:
legend_list.append(labels[i])
self.bokehgui_frequency_sink_x_0_plot.initialize(
update_time=100, legend_list=legend_list)
self.bokehgui_frequency_sink_x_0_plot.set_y_axis([-140, 10])
self.bokehgui_frequency_sink_x_0_plot.set_y_label('Relative Gain' +
'(' + 'dB' + ')')
self.bokehgui_frequency_sink_x_0_plot.set_x_label('Frequency' + '(' +
"Hz" + ')')
self.bokehgui_frequency_sink_x_0_plot.set_trigger_mode(
bokehgui.TRIG_MODE_FREE, 0.0, 0, "")
self.bokehgui_frequency_sink_x_0_plot.enable_grid(False)
self.bokehgui_frequency_sink_x_0_plot.enable_axis_labels(True)
self.bokehgui_frequency_sink_x_0_plot.disable_legend(not True)
self.bokehgui_frequency_sink_x_0_plot.set_layout(*((3, 0, 1, 3)))
self.bokehgui_frequency_sink_x_0_plot.enable_max_hold()
colors = [
"blue", "red", "green", "black", "cyan", "magenta", "yellow",
"dark red", "dark green", "dark blue"
]
widths = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
styles = [
"solid", "solid", "solid", "solid", "solid", "solid", "solid",
"solid", "solid", "solid"
]
markers = [None, None, None, None, None, None, None, None, None, None]
alphas = [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
for i in range(1):
self.bokehgui_frequency_sink_x_0_plot.format_line(
i, colors[i], widths[i], styles[i], markers[i], alphas[i])
self.blocks_sub_xx_0 = blocks.sub_ff(1)
self.blocks_repack_bits_bb_0_0_1_1 = blocks.repack_bits_bb(
bits_per_symbol, 1, '', False, gr.GR_LSB_FIRST)
self.blocks_repack_bits_bb_0_0_1 = blocks.repack_bits_bb(
bits_per_symbol, 1, '', False, gr.GR_LSB_FIRST)
self.blocks_pdu_to_tagged_stream_0_0 = blocks.pdu_to_tagged_stream(
blocks.byte_t, 'packet_len')
self.blocks_null_sink_0 = blocks.null_sink(gr.sizeof_float *
bits_per_symbol)
self.blocks_multiply_xx_0 = blocks.multiply_vcc(1)
self.blocks_delay_0 = blocks.delay(gr.sizeof_gr_complex * 1,
fft_len + 0 * (fft_len // 4) + 10)
self.blocks_char_to_float_0_0_0 = blocks.char_to_float(1, 1)
self.blocks_char_to_float_0_0 = blocks.char_to_float(1, 1)
self.blocks_add_xx_0 = blocks.add_vcc(1)
def _ber_probe():
while True:
val = self.blocks_probe_signal_x_0.level()
try:
self.set_ber(val)
except AttributeError:
pass
time.sleep(1.0 / (0.5))
_ber_thread = threading.Thread(target=_ber_probe)
_ber_thread.daemon = True
_ber_thread.start()
self.analog_noise_source_x_0 = analog.noise_source_c(
analog.GR_GAUSSIAN,
np.sqrt(10**(np.log10(1 / bits_per_symbol) - (mag / 10.0))) * 1,
-1)
self.analog_frequency_modulator_fc_0 = analog.frequency_modulator_fc(
-2.0 / fft_len)
if self.widget_lst:
input_t = bokehgui.bokeh_layout.widgetbox(self.widget_lst)
widgetbox = bokehgui.bokeh_layout.WidgetLayout(input_t)
widgetbox.set_layout(*((0, 0, 2, 1)))
list_obj = [widgetbox] + self.plot_lst
else:
list_obj = self.plot_lst
layout_t = bokehgui.bokeh_layout.create_layout(list_obj,
"stretch_both")
self.doc.add_root(layout_t)
##################################################
# Connections
##################################################
self.msg_connect((self.digital_packet_headerparser_b_0, 'header_data'),
(self.digital_header_payload_demux_0, 'header_data'))
self.msg_connect((self.learning_dl_demod_0, 'losses'),
(self.zeromq_pub_msg_sink_0_0_0, 'in'))
self.msg_connect((self.zeromq_sub_msg_source_0_0, 'out'),
(self.blocks_pdu_to_tagged_stream_0_0, 'pdus'))
self.connect((self.analog_frequency_modulator_fc_0, 0),
(self.blocks_multiply_xx_0, 1))
self.connect((self.analog_noise_source_x_0, 0),
(self.blocks_add_xx_0, 1))
self.connect((self.blocks_add_xx_0, 0),
(self.bokehgui_time_const_x_0, 0))
self.connect((self.blocks_add_xx_0, 0), (self.learning_align_0, 0))
self.connect((self.blocks_char_to_float_0_0, 0),
(self.blocks_sub_xx_0, 0))
self.connect((self.blocks_char_to_float_0_0_0, 0),
(self.blocks_sub_xx_0, 1))
self.connect((self.blocks_delay_0, 0), (self.blocks_multiply_xx_0, 0))
self.connect((self.blocks_multiply_xx_0, 0),
(self.digital_header_payload_demux_0, 0))
self.connect((self.blocks_pdu_to_tagged_stream_0_0, 0),
(self.learning_align_0, 1))
self.connect((self.blocks_repack_bits_bb_0_0_1, 0),
(self.learning_ber_bf_0, 1))
self.connect((self.blocks_repack_bits_bb_0_0_1_1, 0),
(self.learning_ber_bf_0, 0))
self.connect((self.blocks_sub_xx_0, 0),
(self.bokehgui_time_sink_x_0, 0))
self.connect((self.digital_constellation_decoder_cb_0, 0),
(self.digital_packet_headerparser_b_0, 0))
self.connect((self.digital_header_payload_demux_0, 0),
(self.fft_vxx_0, 0))
self.connect((self.digital_header_payload_demux_0, 1),
(self.fft_vxx_1, 0))
self.connect((self.digital_ofdm_chanest_vcvc_0, 0),
(self.digital_ofdm_frame_equalizer_vcvc_0, 0))
self.connect((self.digital_ofdm_frame_equalizer_vcvc_0, 0),
(self.digital_ofdm_serializer_vcc_header, 0))
self.connect((self.digital_ofdm_frame_equalizer_vcvc_1, 0),
(self.digital_ofdm_serializer_vcc_payload, 0))
self.connect((self.digital_ofdm_serializer_vcc_header, 0),
(self.digital_constellation_decoder_cb_0, 0))
self.connect((self.digital_ofdm_serializer_vcc_payload, 0),
(self.blocks_add_xx_0, 0))
self.connect((self.digital_ofdm_sync_sc_cfb_0, 0),
(self.analog_frequency_modulator_fc_0, 0))
self.connect((self.digital_ofdm_sync_sc_cfb_0, 1),
(self.digital_header_payload_demux_0, 1))
self.connect((self.fft_vxx_0, 0),
(self.digital_ofdm_chanest_vcvc_0, 0))
self.connect((self.fft_vxx_1, 0),
(self.digital_ofdm_frame_equalizer_vcvc_1, 0))
self.connect((self.learning_align_0, 1), (self.learning_dl_demod_0, 1))
self.connect((self.learning_align_0, 0), (self.learning_dl_demod_0, 0))
self.connect((self.learning_ber_bf_0, 0),
(self.blocks_probe_signal_x_0, 0))
self.connect((self.learning_dl_demod_0, 1),
(self.blocks_char_to_float_0_0, 0))
self.connect((self.learning_dl_demod_0, 2),
(self.blocks_char_to_float_0_0_0, 0))
self.connect((self.learning_dl_demod_0, 0),
(self.blocks_null_sink_0, 0))
self.connect((self.learning_dl_demod_0, 2),
(self.blocks_repack_bits_bb_0_0_1, 0))
self.connect((self.learning_dl_demod_0, 1),
(self.blocks_repack_bits_bb_0_0_1_1, 0))
self.connect((self.uhd_usrp_source_0, 0), (self.blocks_delay_0, 0))
self.connect((self.uhd_usrp_source_0, 0),
(self.bokehgui_frequency_sink_x_0, 0))
self.connect((self.uhd_usrp_source_0, 0),
(self.digital_ofdm_sync_sc_cfb_0, 0))
try:
from matplotlib import pyplot
except ImportError:
print(
"Error: could not from matplotlib import pyplot (http://matplotlib.sourceforge.net/)"
)
sys.exit(1)
m = 64
# constellation = digital.psk_constellation(m, digital.mod_codes.GRAY_CODE, True)
# (const_points, sym_map) = digital.psk_32_0x0_<permutation>()
# sym_map = [0, 1, 2, 3, 6, 7, 4, 5, 25, 24, 27, 26, 31, 30, 29, 28, 18, 19, 16, 17, 20, 21, 22, 23, 11, 10, 9, 8, 13, 12, 15, 14]
# const_points = [(-7+7j), (-3+7j), (1+7j), (5+7j), (-5+5j), (-1+5j), (3+5j), (7+5j), (-7+3j), (-3+3j), (1+3j), (5+3j), (-5+1j), (-1+1j), (3+1j), (7+1j), (-7-1j), (-3-1j), (1-1j), (5-1j), (-5-3j), (-1-3j), (3-3j), (7-3j), (-7-5j), (-3-5j), (1-5j), (5-5j), (-5-7j), (-1-7j), (3-7j), (7-7j)]
rot_sym = 4
dims = 1
differential = True
mod_code = digital.mod_codes.GRAY_CODE
large_ampls_to_corners = True
# constellation = digital.qam_constellation(const_points, differential, mod_code, large_ampls_to_corners)
constellation = digital.qam_constellation(m, differential, mod_code,
large_ampls_to_corners)
const_points = constellation.points()
# print(len(sym_map))
print(len(const_points))
# print(digital.constellation_calcdist(const_points, sym_map, rot_sym, dims))
def __init__(self):
gr.top_block.__init__(self, "SER Simulation", catch_exceptions=True)
Qt.QWidget.__init__(self)
self.setWindowTitle("SER Simulation")
qtgui.util.check_set_qss()
try:
self.setWindowIcon(Qt.QIcon.fromTheme('gnuradio-grc'))
except:
pass
self.top_scroll_layout = Qt.QVBoxLayout()
self.setLayout(self.top_scroll_layout)
self.top_scroll = Qt.QScrollArea()
self.top_scroll.setFrameStyle(Qt.QFrame.NoFrame)
self.top_scroll_layout.addWidget(self.top_scroll)
self.top_scroll.setWidgetResizable(True)
self.top_widget = Qt.QWidget()
self.top_scroll.setWidget(self.top_widget)
self.top_layout = Qt.QVBoxLayout(self.top_widget)
self.top_grid_layout = Qt.QGridLayout()
self.top_layout.addLayout(self.top_grid_layout)
self.settings = Qt.QSettings("GNU Radio", "ser_simulation")
try:
if StrictVersion(Qt.qVersion()) < StrictVersion("5.0.0"):
self.restoreGeometry(
self.settings.value("geometry").toByteArray())
else:
self.restoreGeometry(self.settings.value("geometry"))
except:
pass
##################################################
# Variables
##################################################
self.const3 = const3 = digital.qam_constellation(16).points()
self.const2 = const2 = digital.constellation_8psk().points()
self.const1 = const1 = digital.constellation_qpsk().points()
self.const0 = const0 = digital.constellation_bpsk().points()
self.samp_rate = samp_rate = 100e3
self.Sps = Sps = 1
self.M3 = M3 = len(const3)
self.M2 = M2 = len(const2)
self.M1 = M1 = len(const1)
self.M0 = M0 = len(const0)
self.run_stop = run_stop = True
self.Rs = Rs = samp_rate / Sps
self.N_snr = N_snr = 128
self.MiconstellationObject3 = MiconstellationObject3 = digital.constellation_calcdist(
const3, np.arange(M3), 4, 1,
digital.constellation.AMPLITUDE_NORMALIZATION).base()
self.MiconstellationObject2 = MiconstellationObject2 = digital.constellation_calcdist(
const2, np.arange(M2), 4, 1,
digital.constellation.AMPLITUDE_NORMALIZATION).base()
self.MiconstellationObject1 = MiconstellationObject1 = digital.constellation_calcdist(
const1, np.arange(M1), 4, 1,
digital.constellation.AMPLITUDE_NORMALIZATION).base()
self.MiconstellationObject0 = MiconstellationObject0 = digital.constellation_calcdist(
const0, np.arange(M0), 4, 1,
digital.constellation.AMPLITUDE_NORMALIZATION).base()
self.MaxErrors = MaxErrors = 1000
self.MaxCount = MaxCount = int(1e7)
self.EsN0min = EsN0min = -5.
self.EsN0max = EsN0max = 25.
##################################################
# Blocks
##################################################
_run_stop_check_box = Qt.QCheckBox('Inicial/Parar')
self._run_stop_choices = {True: True, False: False}
self._run_stop_choices_inv = dict(
(v, k) for k, v in self._run_stop_choices.items())
self._run_stop_callback = lambda i: Qt.QMetaObject.invokeMethod(
_run_stop_check_box, "setChecked",
Qt.Q_ARG("bool", self._run_stop_choices_inv[i]))
self._run_stop_callback(self.run_stop)
_run_stop_check_box.stateChanged.connect(
lambda i: self.set_run_stop(self._run_stop_choices[bool(i)]))
self.top_grid_layout.addWidget(_run_stop_check_box, 0, 0, 1, 1)
for r in range(0, 1):
self.top_grid_layout.setRowStretch(r, 1)
for c in range(0, 1):
self.top_grid_layout.setColumnStretch(c, 1)
self.qtgui_vector_sink_f_0 = qtgui.vector_sink_f(
N_snr,
EsN0min,
(EsN0max - EsN0min) / float(N_snr),
"Es/N0 [dB]",
"logPe",
"Curva de BER",
4, # Number of inputs
None # parent
)
self.qtgui_vector_sink_f_0.set_update_time(0.10)
self.qtgui_vector_sink_f_0.set_y_axis(-8, 0)
self.qtgui_vector_sink_f_0.enable_autoscale(False)
self.qtgui_vector_sink_f_0.enable_grid(False)
self.qtgui_vector_sink_f_0.set_x_axis_units("dB")
self.qtgui_vector_sink_f_0.set_y_axis_units("")
self.qtgui_vector_sink_f_0.set_ref_level(0)
labels = ["BPSK", "QPSK", '8PSK', "16QAM", '', '', '', '', '', '']
widths = [4, 4, 4, 4, 1, 1, 1, 1, 1, 1]
colors = [
"blue", "red", "green", "black", "cyan", "magenta", "yellow",
"dark red", "dark green", "dark blue"
]
alphas = [1, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
for i in range(4):
if len(labels[i]) == 0:
self.qtgui_vector_sink_f_0.set_line_label(
i, "Data {0}".format(i))
else:
self.qtgui_vector_sink_f_0.set_line_label(i, labels[i])
self.qtgui_vector_sink_f_0.set_line_width(i, widths[i])
self.qtgui_vector_sink_f_0.set_line_color(i, colors[i])
self.qtgui_vector_sink_f_0.set_line_alpha(i, alphas[i])
self._qtgui_vector_sink_f_0_win = sip.wrapinstance(
self.qtgui_vector_sink_f_0.pyqwidget(), Qt.QWidget)
self.top_grid_layout.addWidget(self._qtgui_vector_sink_f_0_win, 2, 0,
1, 1)
for r in range(2, 3):
self.top_grid_layout.setRowStretch(r, 1)
for c in range(0, 1):
self.top_grid_layout.setColumnStretch(c, 1)
self.digital_constellation_decoder_cb_0_1 = digital.constellation_decoder_cb(
MiconstellationObject1)
self.digital_constellation_decoder_cb_0_0_0 = digital.constellation_decoder_cb(
MiconstellationObject0)
self.digital_constellation_decoder_cb_0_0 = digital.constellation_decoder_cb(
MiconstellationObject2)
self.digital_constellation_decoder_cb_0 = digital.constellation_decoder_cb(
MiconstellationObject3)
self.digital_chunks_to_symbols_xx_1 = digital.chunks_to_symbols_bc(
const0, 1)
self.digital_chunks_to_symbols_xx_0_0 = digital.chunks_to_symbols_bc(
const1, 1)
self.digital_chunks_to_symbols_xx_0 = digital.chunks_to_symbols_bc(
const3, 1)
self.digital_chunks_to_symbols_xx = digital.chunks_to_symbols_bc(
const2, 1)
self.b_BERTool_0_1 = b_BERTool(
EsN0max=EsN0max,
EsN0min=EsN0min,
N_snr=N_snr,
Rs=Rs,
)
self.b_BERTool_0_0_0 = b_BERTool(
EsN0max=EsN0max,
EsN0min=EsN0min,
N_snr=N_snr,
Rs=Rs,
)
self.b_BERTool_0_0 = b_BERTool(
EsN0max=EsN0max,
EsN0min=EsN0min,
N_snr=N_snr,
Rs=Rs,
)
self.b_BERTool_0 = b_BERTool(
EsN0max=EsN0max,
EsN0min=EsN0min,
N_snr=N_snr,
Rs=Rs,
)
self.analog_random_source_x_1 = blocks.vector_source_b(
list(map(int, numpy.random.randint(0, M0, 10000000))), True)
self.analog_random_source_x_0_0 = blocks.vector_source_b(
list(map(int, numpy.random.randint(0, M1, 10000000))), True)
self.analog_random_source_x_0 = blocks.vector_source_b(
list(map(int, numpy.random.randint(0, M3, 10000000))), True)
self.analog_random_source_x = blocks.vector_source_b(
list(map(int, numpy.random.randint(0, M2, 10000000))), True)
##################################################
# Connections
##################################################
self.connect((self.analog_random_source_x, 0), (self.b_BERTool_0_0, 2))
self.connect((self.analog_random_source_x, 0),
(self.digital_chunks_to_symbols_xx, 0))
self.connect((self.analog_random_source_x_0, 0), (self.b_BERTool_0, 1))
self.connect((self.analog_random_source_x_0, 0),
(self.digital_chunks_to_symbols_xx_0, 0))
self.connect((self.analog_random_source_x_0_0, 0),
(self.b_BERTool_0_1, 2))
self.connect((self.analog_random_source_x_0_0, 0),
(self.digital_chunks_to_symbols_xx_0_0, 0))
self.connect((self.analog_random_source_x_1, 0),
(self.b_BERTool_0_0_0, 2))
self.connect((self.analog_random_source_x_1, 0),
(self.digital_chunks_to_symbols_xx_1, 0))
self.connect((self.b_BERTool_0, 0),
(self.digital_constellation_decoder_cb_0, 0))
self.connect((self.b_BERTool_0, 1), (self.qtgui_vector_sink_f_0, 3))
self.connect((self.b_BERTool_0_0, 0),
(self.digital_constellation_decoder_cb_0_0, 0))
self.connect((self.b_BERTool_0_0, 1), (self.qtgui_vector_sink_f_0, 2))
self.connect((self.b_BERTool_0_0_0, 0),
(self.digital_constellation_decoder_cb_0_0_0, 0))
self.connect((self.b_BERTool_0_0_0, 1),
(self.qtgui_vector_sink_f_0, 0))
self.connect((self.b_BERTool_0_1, 0),
(self.digital_constellation_decoder_cb_0_1, 0))
self.connect((self.b_BERTool_0_1, 1), (self.qtgui_vector_sink_f_0, 1))
self.connect((self.digital_chunks_to_symbols_xx, 0),
(self.b_BERTool_0_0, 0))
self.connect((self.digital_chunks_to_symbols_xx_0, 0),
(self.b_BERTool_0, 0))
self.connect((self.digital_chunks_to_symbols_xx_0_0, 0),
(self.b_BERTool_0_1, 0))
self.connect((self.digital_chunks_to_symbols_xx_1, 0),
(self.b_BERTool_0_0_0, 0))
self.connect((self.digital_constellation_decoder_cb_0, 0),
(self.b_BERTool_0, 2))
self.connect((self.digital_constellation_decoder_cb_0_0, 0),
(self.b_BERTool_0_0, 1))
self.connect((self.digital_constellation_decoder_cb_0_0_0, 0),
(self.b_BERTool_0_0_0, 1))
self.connect((self.digital_constellation_decoder_cb_0_1, 0),
(self.b_BERTool_0_1, 1))
def __init__(self, doc, ip="127.0.0.1"):
gr.top_block.__init__(self, "RL Mod training")
self.doc = doc
self.plot_lst = []
self.widget_lst = []
##################################################
# Parameters
##################################################
self.ip = ip
##################################################
# Variables
##################################################
self.pilot_symbols = pilot_symbols = ((1, 1, 1, -1,),)
self.pilot_carriers = pilot_carriers = ((-21, -7, 7, 21,),)
self.packet_length_tag_key = packet_length_tag_key = "packet_length"
self.occupied_carriers = occupied_carriers = (list(range(-22, -21)) + list(range(-20, -7)) + list(range(-6, 0)) + list(range(1, 7)) + list(range(8, 21)) + list(range(22, 23)),)
self.length_tag_key_0 = length_tag_key_0 = "frame_len"
self.length_tag_key = length_tag_key = "packet_len"
self.header_mod = header_mod = digital.constellation_bpsk()
self.fft_len = fft_len = 64
self.bits_per_symbol = bits_per_symbol = 4
self.variable_0 = variable_0 = len(occupied_carriers[0])
self.tx_lr = tx_lr = -3
self.tx_explo = tx_explo = 0.15
self.training_mod = training_mod = 1
self.timestamp = timestamp = '1'
self.t_state = t_state = 1
self.sync_word2 = sync_word2 = [0, 0, 0, 0, 0, 0, -1, -1, -1, -1, 1, 1, -1, -1, -1, 1, -1, 1, 1, 1, 1, 1, -1, -1, -1, -1, -1, 1, -1, -1, 1, -1, 0, 1, -1, 1, 1, 1, -1, 1, 1, 1, -1, 1, 1, 1, 1, -1, 1, -1, -1, -1, 1, -1, 1, -1, -1, -1, -1, 0, 0, 0, 0, 0]
self.sync_word1 = sync_word1 = [0., 0., 0., 0., 0., 0., 0., 1.41421356, 0., -1.41421356, 0., 1.41421356, 0., -1.41421356, 0., -1.41421356, 0., -1.41421356, 0., 1.41421356, 0., -1.41421356, 0., 1.41421356, 0., -1.41421356, 0., -1.41421356, 0., -1.41421356, 0., -1.41421356, 0., 1.41421356, 0., -1.41421356, 0., 1.41421356, 0., 1.41421356, 0., 1.41421356, 0., -1.41421356, 0., 1.41421356, 0., 1.41421356, 0., 1.41421356, 0., -1.41421356, 0., 1.41421356, 0., 1.41421356, 0., 1.41421356, 0., 0., 0., 0., 0., 0.]
self.samp_rate = samp_rate = 0.3e6
self.rolloff = rolloff = 0
self.puncpat = puncpat = '11'
self.payload_mod = payload_mod = digital.qam_constellation(constellation_points=2**bits_per_symbol)
self.payload_equalizer = payload_equalizer = digital.ofdm_equalizer_static(fft_len, occupied_carriers, pilot_carriers, pilot_symbols, 1)
self.packet_len = packet_len = 400
self.ldpc_enc = ldpc_enc = fec.ldpc_encoder_make(gr.prefix() + "/share/gnuradio/fec/ldpc/" + "n_0100_k_0042_gap_02.alist")
self.header_formatter = header_formatter = digital.packet_header_ofdm(occupied_carriers, n_syms=1, len_tag_key=packet_length_tag_key, frame_len_tag_key=length_tag_key_0, bits_per_header_sym=header_mod.bits_per_symbol(), bits_per_payload_sym=8, scramble_header=False)
self.header_equalizer = header_equalizer = digital.ofdm_equalizer_simpledfe(fft_len, header_mod.base(), occupied_carriers, pilot_carriers, pilot_symbols)
self.hdr_format = hdr_format = digital.header_format_ofdm(occupied_carriers, 1, length_tag_key,)
self.gain = gain = 10
self.freq = freq = 900e6
self.bits_per_symbol_0 = bits_per_symbol_0 = len(occupied_carriers[0])
##################################################
# Blocks
##################################################
self.tx_lr_slider = bokehgui.slider(self.widget_lst, 'TX Learning rate (log)' +":", -5, 0, 0.1, 1, -3)
self.tx_lr_slider.add_callback(lambda attr, old, new: self.set_tx_lr(new))
self.tx_explo_slider = bokehgui.slider(self.widget_lst, 'TX Exploration noise' +":", 0.001, 0.5, 0.001, 1, 0.15)
self.tx_explo_slider.add_callback(lambda attr, old, new: self.set_tx_explo(new))
self.training_mod_textbox = bokehgui.textbox(self.widget_lst, str(1), 'Alternate training every' +": ")
self.training_mod_textbox.add_callback(
lambda attr, old, new: self.set_training_mod(int(new)))
self._t_state_options = [ 1, 0, ]
self._t_state_labels = [ 'On', 'Off', ]
self.t_state_radiobutton = bokehgui.radiobutton(self.widget_lst, None, self._t_state_labels, inline = True)
self.t_state_radiobutton.add_callback(
lambda new: self.set_t_state(int(self._t_state_options[new])))
self.gain_slider = bokehgui.slider(self.widget_lst, 'Amplitude' +":", 0, 90, 0.5, 1, 10)
self.gain_slider.add_callback(lambda attr, old, new: self.set_gain(new))
self.zeromq_sub_msg_source_0_1 = zeromq.sub_msg_source("tcp://"+ip+":50002", 100)
self.zeromq_pub_msg_sink_0_0 = zeromq.pub_msg_sink('tcp://*:50001', 100)
self.uhd_usrp_sink_0 = uhd.usrp_sink(
",".join(("", "")),
uhd.stream_args(
cpu_format="fc32",
args='',
channels=list(range(0,1)),
),
length_tag_key,
)
self.uhd_usrp_sink_0.set_samp_rate(samp_rate)
self.uhd_usrp_sink_0.set_time_now(uhd.time_spec(time.time()), uhd.ALL_MBOARDS)
self.uhd_usrp_sink_0.set_center_freq(freq, 0)
self.uhd_usrp_sink_0.set_antenna('TX/RX', 0)
self.uhd_usrp_sink_0.set_gain(gain, 0)
self.timestamp_textbox = bokehgui.textbox(self.widget_lst, str('1'), 'Load timestamp' +": ")
self.timestamp_textbox.add_callback(
lambda attr, old, new: self.set_timestamp(str(new)))
self.learning_tag_numerotation_0 = learning.tag_numerotation('packet_num', packet_len, 4096, "bb")
self.learning_rl_mod_0 = learning.rl_mod('packet_num', bits_per_symbol, packet_len, 1, training_mod, 10**tx_lr , tx_explo, t_state , '/home/cyrille/Gnu-Radio/modules/gr-learning/examples/saved_models')
self.fft_vxx_0_0 = fft.fft_vcc(fft_len, False, (), True, 1)
self.fec_extended_encoder_1_0_0 = fec.extended_encoder(encoder_obj_list=ldpc_enc, threading= None, puncpat=puncpat)
self.digital_protocol_formatter_bb_0 = digital.protocol_formatter_bb(hdr_format, length_tag_key)
self.digital_ofdm_cyclic_prefixer_0 = digital.ofdm_cyclic_prefixer(
fft_len,
fft_len + fft_len//4,
rolloff,
length_tag_key)
self.digital_ofdm_carrier_allocator_cvc_0 = digital.ofdm_carrier_allocator_cvc( fft_len, occupied_carriers, pilot_carriers, pilot_symbols, (sync_word1, sync_word2), length_tag_key, True)
self.digital_chunks_to_symbols_xx_0 = digital.chunks_to_symbols_bc(header_mod.points(), 1)
self.bokehgui_time_const_x_0 = bokehgui.time_sink_c_proc(1024, samp_rate, "Transmitted noiseless constellation", 1)
self.bokehgui_time_const_x_0_plot = bokehgui.const_sink_c(self.doc, self.plot_lst, self.bokehgui_time_const_x_0, is_message = False)
labels = ['', '', '', '', '',
'', '', '', '', '']
legend_list = []
for i in range( 1 ):
if len(labels[i]) == 0:
if(i % 2 == 0):
legend_list.append("Re{{Data {0}}}".format(i/2))
else:
legend_list.append("Im{{Data {0}}}".format(i/2))
else:
legend_list.append(labels[i])
self.bokehgui_time_const_x_0_plot.initialize(update_time = 500,
legend_list = legend_list)
self.bokehgui_time_const_x_0_plot.set_y_axis([-2, 2])
self.bokehgui_time_const_x_0_plot.set_y_label('Q Channel' + '(' +""+')')
self.bokehgui_time_const_x_0_plot.set_x_label('I Channel' + '(' +""+')')
self.bokehgui_time_const_x_0_plot.enable_tags(-1, False)
self.bokehgui_time_const_x_0_plot.set_trigger_mode(bokehgui.TRIG_MODE_FREE, bokehgui.TRIG_SLOPE_POS, 0.0, 0, 0, "")
self.bokehgui_time_const_x_0_plot.enable_grid(False)
self.bokehgui_time_const_x_0_plot.enable_axis_labels(True)
self.bokehgui_time_const_x_0_plot.disable_legend(not True)
self.bokehgui_time_const_x_0_plot.set_layout(*((0,1,2,2)))
colors = ["blue", "red", "green", "black", "cyan",
"magenta", "yellow", "blue", "blue", "blue"]
widths = [1, 1, 1, 1, 1,
1, 1, 1, 1, 1]
markers = ['o', 'o', 'o', 'o', 'o',
'o', 'o', 'o', 'o', 'o']
alphas = [1.0, 1.0, 1.0, 1.0, 1.0,
1.0, 1.0, 1.0, 1.0, 1.0]
for i in range( 1 ):
self.bokehgui_time_const_x_0_plot.format_line(i, colors[i], widths[i], 'None', markers[i], alphas[i])
self.blocks_throttle_0 = blocks.throttle(gr.sizeof_char*1, samp_rate/6,True)
self.blocks_tagged_stream_to_pdu_0_0_0 = blocks.tagged_stream_to_pdu(blocks.byte_t, 'packet_len')
self.blocks_tagged_stream_mux_0 = blocks.tagged_stream_mux(gr.sizeof_gr_complex*1, length_tag_key, 0)
self.blocks_stream_to_tagged_stream_0_0 = blocks.stream_to_tagged_stream(gr.sizeof_char, 1, packet_len, length_tag_key)
self.blocks_repack_bits_bb_0_0_0 = blocks.repack_bits_bb(8, 1, length_tag_key, False, gr.GR_LSB_FIRST)
self.blocks_repack_bits_bb_0_0 = blocks.repack_bits_bb(1, bits_per_symbol, '', False, gr.GR_LSB_FIRST)
self.blocks_multiply_const_xx_0 = blocks.multiply_const_cc(0.01, 1)
self.analog_random_source_x_0_0 = blocks.vector_source_b(list(map(int, numpy.random.randint(0, 2, 1000))), True)
if self.widget_lst:
input_t = bokehgui.bokeh_layout.widgetbox(self.widget_lst)
widgetbox = bokehgui.bokeh_layout.WidgetLayout(input_t)
widgetbox.set_layout(*((0, 0, 2, 1)))
list_obj = [widgetbox] + self.plot_lst
else:
list_obj = self.plot_lst
layout_t = bokehgui.bokeh_layout.create_layout(list_obj, "fixed")
self.doc.add_root(layout_t)
##################################################
# Connections
##################################################
self.msg_connect((self.blocks_tagged_stream_to_pdu_0_0_0, 'pdus'), (self.zeromq_pub_msg_sink_0_0, 'in'))
self.msg_connect((self.zeromq_sub_msg_source_0_1, 'out'), (self.learning_rl_mod_0, 'losses'))
self.connect((self.analog_random_source_x_0_0, 0), (self.fec_extended_encoder_1_0_0, 0))
self.connect((self.blocks_multiply_const_xx_0, 0), (self.uhd_usrp_sink_0, 0))
self.connect((self.blocks_repack_bits_bb_0_0, 0), (self.blocks_stream_to_tagged_stream_0_0, 0))
self.connect((self.blocks_repack_bits_bb_0_0_0, 0), (self.digital_chunks_to_symbols_xx_0, 0))
self.connect((self.blocks_stream_to_tagged_stream_0_0, 0), (self.digital_protocol_formatter_bb_0, 0))
self.connect((self.blocks_stream_to_tagged_stream_0_0, 0), (self.learning_tag_numerotation_0, 0))
self.connect((self.blocks_tagged_stream_mux_0, 0), (self.digital_ofdm_carrier_allocator_cvc_0, 0))
self.connect((self.blocks_throttle_0, 0), (self.blocks_tagged_stream_to_pdu_0_0_0, 0))
self.connect((self.digital_chunks_to_symbols_xx_0, 0), (self.blocks_tagged_stream_mux_0, 0))
self.connect((self.digital_ofdm_carrier_allocator_cvc_0, 0), (self.fft_vxx_0_0, 0))
self.connect((self.digital_ofdm_cyclic_prefixer_0, 0), (self.blocks_multiply_const_xx_0, 0))
self.connect((self.digital_protocol_formatter_bb_0, 0), (self.blocks_repack_bits_bb_0_0_0, 0))
self.connect((self.fec_extended_encoder_1_0_0, 0), (self.blocks_repack_bits_bb_0_0, 0))
self.connect((self.fft_vxx_0_0, 0), (self.digital_ofdm_cyclic_prefixer_0, 0))
self.connect((self.learning_rl_mod_0, 0), (self.blocks_tagged_stream_mux_0, 1))
self.connect((self.learning_rl_mod_0, 1), (self.bokehgui_time_const_x_0, 0))
self.connect((self.learning_tag_numerotation_0, 0), (self.blocks_throttle_0, 0))
self.connect((self.learning_tag_numerotation_0, 0), (self.learning_rl_mod_0, 0))
def set_bits_per_symbol(self, bits_per_symbol):
self.bits_per_symbol = bits_per_symbol
self.set_payload_mod(digital.qam_constellation(constellation_points=2**self.bits_per_symbol))
self.blocks_repack_bits_bb_0_0.set_k_and_l(1,self.bits_per_symbol)