def __init__(self, EbN0):
gr.top_block.__init__(self)
self.const = digital.bpsk_constellation()
# Source is N_BITS bits, non-repeated
#print("BITS PER SYMBOL: ", self.const.bits_per_symbol())
#print("arity: ", self.const.arity())
data = map(
int,
numpy.random.randint(0, self.const.arity(),
int(N_BITS / self.const.bits_per_symbol())))
src = blocks.vector_source_b(data, False)
mod = digital.chunks_to_symbols_bc((self.const.points()), 1)
add = blocks.add_vcc()
diff_enc = digital.diff_encoder_bb(2)
diff_dec = digital.diff_decoder_bb(2)
#print("Gaussian amplitude: ", self.EbN0_to_noise_voltage(EbN0))
noise = analog.noise_source_c(analog.GR_GAUSSIAN,
self.EbN0_to_noise_voltage(EbN0),
RAND_SEED)
demod = digital.constellation_decoder_cb(self.const.base())
ber = BitErrors(self.const.bits_per_symbol())
self.sink = blocks.vector_sink_f()
self.connect(src, diff_enc, mod, add, demod, diff_dec, ber, self.sink)
self.connect(noise, (add, 1))
self.connect(src, (ber, 1))
def test_tag(self):
# Send data through bpsk receiver
# followed by qpsk receiver
data = [0.9 + 0j, 0.1 + 0.9j, -1 - 0.1j, -0.1 - 0.6j] * 2
bpsk_data = [1, 1, 0, 0]
qpsk_data = [1, 3, 0, 0]
first_tag = gr.tag_t()
first_tag.key = pmt.intern("set_constellation")
first_tag.value = digital.bpsk_constellation().as_pmt()
first_tag.offset = 0
second_tag = gr.tag_t()
second_tag.key = pmt.intern("set_constellation")
second_tag.value = digital.qpsk_constellation().as_pmt()
second_tag.offset = 4
src = blocks.vector_source_c(data, False, 1, [first_tag, second_tag])
decoder = digital.constellation_receiver_cb(
digital.bpsk_constellation().base(), 0, 0, 0)
snk = blocks.vector_sink_b()
tb = gr.top_block()
tb.connect(src, decoder, snk)
tb.run()
self.assertEqual(list(snk.data()), bpsk_data + qpsk_data)
def test_tag(self):
# Send data through bpsk receiver
# followed by qpsk receiver
data = [0.9+0j, 0.1+0.9j, -1-0.1j, -0.1-0.6j]*2
bpsk_data = [1, 1, 0, 0]
qpsk_data = [1, 3, 0, 0]
first_tag = gr.tag_t()
first_tag.key = pmt.intern("set_constellation")
first_tag.value = digital.bpsk_constellation().as_pmt()
first_tag.offset = 0
second_tag = gr.tag_t()
second_tag.key = pmt.intern("set_constellation")
second_tag.value = digital.qpsk_constellation().as_pmt()
second_tag.offset = 4
src = blocks.vector_source_c(data, False, 1, [first_tag, second_tag])
decoder = digital.constellation_receiver_cb(
digital.bpsk_constellation().base(), 0, 0, 0)
snk = blocks.vector_sink_b()
tb = gr.top_block()
tb.connect(src, decoder, snk)
tb.run()
self.assertEqual(list(snk.data()), bpsk_data+qpsk_data)
def test_two(self):
tb = gr.top_block()
MTU = 4000
tagname = "length"
packets = (
(0, 0, 0, 0)*32,
#(0, 0, 1, 0)*16,
#(1, 1, 1, 0)*4,
)
data, tags = ofdm.utils.packets_to_vectors(packets, tagname)
constellation = digital.bpsk_constellation()
src = gr.vector_source_b(data, tags, False, 1)
tag_scaler1 = ofdm.scale_tags(1, tagname, 0.125)
packer = gr.unpacked_to_packed_bb(1, gr.GR_MSB_FIRST)
crc = ofdm.crc32_bb(False, MTU, tagname)
tag_scaler2 = ofdm.scale_tags(1, tagname, 8)
unpacker = gr.packed_to_unpacked_bb(1, gr.GR_MSB_FIRST)
mod1 = digital.chunks_to_symbols_bc(constellation.points())
mod2 = digital.chunks_to_symbols_bc(constellation.points())
occupied_carriers = ((1, 2, 4, 5),)
fft_len = 16
header = ofdm.ofdm_header_bb(len(occupied_carriers[0]))
mux1 = ofdm.tagged_stream_mux(gr.sizeof_gr_complex, 2, tagname, MTU)
mux2 = ofdm.tagged_stream_mux(gr.sizeof_gr_complex*fft_len, 2, tagname, MTU)
ifft = fft.fft_vcc(fft_len, forward=False, window=[1]*fft_len)
cp_len = 4
rolloff_len = 2
cp = digital.ofdm_cyclic_prefixer(fft_len, fft_len+cp_len, rolloff_len, tagname, MTU)
pilot_symbols = ((1j,),)
pilot_carriers = ((3,),)
alloc = ofdm.carrier_allocator_cvc(fft_len,
occupied_carriers,
pilot_carriers,
pilot_symbols,
tagname)
snk = gr.vector_sink_c()
sync_packet = [0, 1, 0, 1, 0, 1] + [0]*10
sync_data, sync_tags = ofdm.utils.packets_to_vectors([sync_packet], tagname, vlen=fft_len)
sync_header = gr.vector_source_c(sync_data, sync_tags, True, fft_len)
tb.connect(src, tag_scaler1, packer, crc, tag_scaler2, unpacker, mod1, (mux1, 1))
tb.connect(crc, header, mod2, (mux1, 0))
tb.connect(mux1, alloc, (mux2, 1))
tb.connect(sync_header, (mux2, 0))
tb.connect(mux2, ifft, cp, snk)
#tb.connect(src, tag_scaler1, packer, crc, tag_scaler2, unpacker, mod1, alloc, snk)
tb.run()
def __init__(self, EbN0):
gr.top_block.__init__(self)
# self.const = digital.qpsk_constellation()
self.const = digital.bpsk_constellation()
# Source is N_BITS bits, non-repeated
data = list(map(int, numpy.random.randint(0, self.const.arity(), int(N_BITS / self.const.bits_per_symbol()))))
src = blocks.vector_source_b(data, False)
mod = digital.chunks_to_symbols_bc((self.const.points()), 1)
add = blocks.add_vcc()
noise = analog.noise_source_c(analog.GR_GAUSSIAN,
self.EbN0_to_noise_voltage(EbN0),
RAND_SEED)
demod = digital.constellation_decoder_cb(self.const.base())
ber = BitErrors(self.const.bits_per_symbol())
self.sink = blocks.vector_sink_f()
self.connect(src, mod, add, demod, ber, self.sink)
self.connect(noise, (add, 1))
self.connect(src, (ber, 1))
def atest_one(self):
tb = gr.top_block()
MTU = 4000
tagname = "_length"
packets = (
(0, 0, 0, 0)*32,
(0, 0, 1, 0)*16,
(1, 1, 1, 0)*4,
)
data, tags = ofdm.utils.packets_to_vectors(packets, tagname)
constellation = digital.bpsk_constellation()
src = gr.vector_source_b(data, tags, False, 1)
tag_scaler1 = ofdm.scale_tags(1, tagname, 0.125)
packer = gr.unpacked_to_packed_bb(1, gr.GR_MSB_FIRST)
crc1 = ofdm.crc32_bb(False, MTU, tagname)
crc2 = ofdm.crc32_bb(True, MTU, tagname)
tag_scaler2 = ofdm.scale_tags(1, tagname, 8)
unpacker = gr.packed_to_unpacked_bb(1, gr.GR_MSB_FIRST)
chunks2symbols = digital.chunks_to_symbols_bc(constellation.points())
snk = gr.vector_sink_b()
#tb.connect(src, packer, crc, tag_scaler, unpacker, chunks2symbols, snk)
tb.connect(src, tag_scaler1, packer, crc1, crc2, tag_scaler2, unpacker, snk)
tb.run()
