def test_001_setup(self):
multiple = 6
overlap = 1 # must be 1 in this test!
L = 4
taps = np.array([0, 1, 2, 1])
taps = np.append(taps, [0.0, ])
data = np.arange(1, multiple * L + 1, dtype=np.complex)
try: # make sure ctor checks work correctly!
dummy = fbmc.tx_sdft_vcc([1, 1, 1], L)
except RuntimeError as e:
s = str(e)
pos = s.find("number of filter taps must be equal to")
if pos < 0:
raise
tx_sdft = fbmc.tx_sdft_vcc(taps, L)
src = blocks.vector_source_c(data, vlen=L)
snk = blocks.vector_sink_c(vlen=1)
# set up fg
self.tb.connect(src, tx_sdft, snk)
self.tb.run()
# check data
print "L: ", tx_sdft.L()
print "overlap: ", tx_sdft.overlap()
print "taps: ", tx_sdft.taps()
res = snk.data()
print res
def test_001_setup(self):
multiple = 6
overlap = 1 # must be 1 in this test!
L = 4
taps = np.array([0, 1, 2, 1])
taps = np.append(taps, [
0.0,
])
data = np.arange(1, multiple * L + 1, dtype=np.complex)
try: # make sure ctor checks work correctly!
dummy = fbmc.tx_sdft_vcc([1, 1, 1], L)
except RuntimeError as e:
s = str(e)
pos = s.find("number of filter taps must be equal to")
if pos < 0:
raise
tx_sdft = fbmc.tx_sdft_vcc(taps, L)
src = blocks.vector_source_c(data, vlen=L)
snk = blocks.vector_sink_c(vlen=1)
# set up fg
self.tb.connect(src, tx_sdft, snk)
self.tb.run()
# check data
print "L: ", tx_sdft.L()
print "overlap: ", tx_sdft.overlap()
print "taps: ", tx_sdft.taps()
res = snk.data()
print res
def test_004_config_frame_mod(self):
num_frames = 10
cfg = fbmc.fbmc_config(num_used_subcarriers=20, num_payload_sym=16, num_overlap_sym=4, modulation="QPSK",
preamble="IAM", samp_rate=250000)
total_subcarriers = cfg.num_total_subcarriers() # 8
used_subcarriers = cfg.num_used_subcarriers() # 4
channel_map = cfg.channel_map() # ft.get_channel_map(used_subcarriers, total_subcarriers)
payload_symbols = cfg.num_payload_sym() # 8
overlap = cfg.num_overlap_sym() # 4
taps = cfg.phydyas_impulse_taps(cfg.num_total_subcarriers(), cfg.num_overlap_sym()) # ft.generate_phydyas_filter(total_subcarriers, overlap)
preamble = ft.get_preamble(total_subcarriers)
num_preamble_symbols = len(preamble) // total_subcarriers
payload = ft.get_payload(payload_symbols, used_subcarriers)
payload = np.tile(payload, num_frames).flatten()
src = blocks.vector_source_b(payload, repeat=False)
framer = fbmc.frame_generator_bvc(used_subcarriers, total_subcarriers, payload_symbols, overlap, channel_map, preamble)
snk_frame = blocks.vector_sink_c(total_subcarriers) # a debug output
mod = fbmc.tx_sdft_vcc(taps, total_subcarriers)
demod = fbmc.rx_sdft_cvc(taps, total_subcarriers)
skipper = blocks.skiphead(8 * total_subcarriers, 4)
snk_rx = blocks.vector_sink_c(total_subcarriers)
deframer = fbmc.deframer_vcb(used_subcarriers, total_subcarriers, num_preamble_symbols, payload_symbols, overlap, channel_map)
snk = blocks.vector_sink_b(1)
self.tb.connect(src, framer, mod, demod, skipper, deframer, snk)
self.tb.connect(framer, snk_frame)
self.tb.connect(skipper, snk_rx)
self.tb.run()
res = np.array(snk.data())
print res
print payload
moddata = np.array(snk_frame.data())
print "len(moddata) = ", len(moddata)
rxdata = np.array(snk_rx.data())
print "len(rxdata) = ", len(rxdata)
# plt.plot(rxdata.real * 0.03)
# for i in range(len(moddata)):
# if (i + 1) % total_subcarriers == 0:
# plt.axvline(i)
# plt.plot(moddata.real)
# plt.grid()
# plt.show()
print "len(payload) = ", len(payload)
print "len(result) = ", len(res)
self.assertTupleEqual(tuple(payload[:len(res)]), tuple(res))
def test_003_small_frame_mod(self):
num_frames = 300
total_subcarriers = 8
used_subcarriers = 4
channel_map = np.array((0, 0, 1, 1, 1, 1, 0, 0)) #ft.get_channel_map(used_subcarriers, total_subcarriers)
payload_symbols = 8
overlap = 4
taps = ft.generate_phydyas_filter(total_subcarriers, overlap)
preamble = ft.get_preamble(total_subcarriers)
num_preamble_symbols = len(preamble) // total_subcarriers
payload = ft.get_payload(payload_symbols, used_subcarriers)
payload = np.tile(payload, num_frames).flatten()
src = blocks.vector_source_b(payload, repeat=False)
framer = fbmc.frame_generator_bvc(used_subcarriers, total_subcarriers, payload_symbols, overlap, channel_map, preamble)
snk_frame = blocks.vector_sink_c(total_subcarriers) # a debug output
mod = fbmc.tx_sdft_vcc(taps, total_subcarriers)
demod = fbmc.rx_sdft_cvc(taps, total_subcarriers)
skipper = blocks.skiphead(8 * total_subcarriers, 4)
snk_rx = blocks.vector_sink_c(total_subcarriers)
deframer = fbmc.deframer_vcb(used_subcarriers, total_subcarriers, num_preamble_symbols, payload_symbols, overlap, channel_map)
snk = blocks.vector_sink_b(1)
self.tb.connect(src, framer, mod, demod, skipper, deframer, snk)
self.tb.connect(framer, snk_frame)
self.tb.connect(skipper, snk_rx)
self.tb.run()
res = np.array(snk.data())
print "len(res) = ", len(res), ", len(payload) = ", len(payload)
print "ref: ", payload
print "res: ", res
moddata = np.array(snk_frame.data())
print "len(moddata) = ", len(moddata)
rxdata = np.array(snk_rx.data())
print "len(rxdata) = ", len(rxdata), " diff: ", len(moddata) - len(rxdata)
# plt.plot(rxdata.real * 0.03)
# for i in range(len(moddata)):
# if (i + 1) % total_subcarriers == 0:
# plt.axvline(i)
# plt.plot(moddata.real)
# plt.grid()
# plt.show()
print "len(payload) = ", len(payload)
print "len(result ) = ", len(res)
self.assertTupleEqual(tuple(payload[:len(res)]), tuple(res))
def test_002_crunch(self):
print "\n\n\ntest_002"
multiple = 6
overlap = 1
L = 4
taps = np.array([
1,
1,
1,
1,
2,
2,
2,
2,
])
taps = np.append(taps, [
0.0,
])
data = np.arange(1, multiple * L + 1, dtype=np.complex)
tx_sdft = fbmc.tx_sdft_vcc(taps, L)
src = blocks.vector_source_c(data, vlen=L)
snk0 = blocks.vector_sink_c(vlen=1)
ft = fft.fft_vcc(L, False, (), False, 1)
pfb = fbmc.polyphase_filterbank_vcvc(L, taps)
comm = fbmc.output_commutator_vcc(L=L)
snk1 = blocks.vector_sink_c(vlen=1)
self.tb.connect(src, tx_sdft, snk0)
self.tb.connect(src, ft, pfb, comm, snk1)
self.tb.run()
r0 = np.array(snk0.data()[L // 2:-L // 2])
r1 = np.array(snk1.data())
print "sdft overlap = ", tx_sdft.overlap()
print "sdft history = ", tx_sdft.history()
print "\ntx_sdft"
print r0.reshape((-1, L)).T
print "\npolyphase"
print r1.reshape((-1, L)).T
tmp = np.concatenate((r0, r1)).reshape((2, -1)).T
for x, y in tmp:
print x, y, "\tis equal", x == y
self.assertComplexTuplesAlmostEqual(r0, r1[0:len(r0)])
def test_003_go_big(self):
print "\n\n\ntest_003"
multiple = 2000
overlap = 4
# get test data, taps, config, etc.
self.cfg = fbmc.fbmc_config(num_used_subcarriers=20,
num_payload_sym=16,
num_overlap_sym=overlap,
modulation="QPSK",
preamble="IAM")
L = self.cfg.num_total_subcarriers()
taps = np.array(self.cfg.prototype_taps_float(), dtype=np.float)
taps[-1] = 0
data = np.arange(1, L + 1, dtype=np.complex)
data = np.repeat(data, multiple)
# set up fg
tx_sdft = fbmc.tx_sdft_vcc(taps, L)
src = blocks.vector_source_c(data, vlen=L)
snk0 = blocks.vector_sink_c(vlen=1)
self.tb.connect(src, tx_sdft, snk0)
ft = fft.fft_vcc(L, False, (), False, 1)
pfb = fbmc.polyphase_filterbank_vcvc(L, taps)
comm = fbmc.output_commutator_vcc(L=L)
snk1 = blocks.vector_sink_c(vlen=1)
self.tb.connect(src, ft, pfb, comm, snk1)
self.tb.run()
r0 = np.array(snk0.data()[L:])
r1 = np.array(snk1.data()[:-L])
print "\ntx_sdft"
print r0.reshape((-1, L)).T
print "\npolyphase"
print r1.reshape((-1, L)).T
self.assertComplexTuplesAlmostEqual(r0, r1, 4)
def test_002_modulation(self):
total_subcarriers = 8
overlap = 4
taps = ft.generate_phydyas_filter(total_subcarriers, overlap)
zvals = np.zeros(overlap * total_subcarriers, dtype=complex)
rvals = np.arange(1, total_subcarriers + 1, dtype=complex)
dummy_frame = np.concatenate((zvals, rvals))
data = np.tile(dummy_frame, 4).flatten()
src = blocks.vector_source_c(data, repeat=False, vlen=total_subcarriers)
mod = fbmc.tx_sdft_vcc(taps, total_subcarriers)
demod = fbmc.rx_sdft_cvc(taps, total_subcarriers)
snk = blocks.vector_sink_c(total_subcarriers)
self.tb.connect(src, mod, demod, snk)
self.tb.run()
res = np.array(snk.data())
def test_002_crunch(self):
print "\n\n\ntest_002"
multiple = 6
overlap = 1
L = 4
taps = np.array([1, 1, 1, 1, 2, 2, 2, 2, ])
taps = np.append(taps, [0.0, ])
data = np.arange(1, multiple * L + 1, dtype=np.complex)
tx_sdft = fbmc.tx_sdft_vcc(taps, L)
src = blocks.vector_source_c(data, vlen=L)
snk0 = blocks.vector_sink_c(vlen=1)
ft = fft.fft_vcc(L, False, (), False, 1)
pfb = fbmc.polyphase_filterbank_vcvc(L, taps)
comm = fbmc.output_commutator_vcc(L=L)
snk1 = blocks.vector_sink_c(vlen=1)
self.tb.connect(src, tx_sdft, snk0)
self.tb.connect(src, ft, pfb, comm, snk1)
self.tb.run()
r0 = np.array(snk0.data()[L//2:-L//2])
r1 = np.array(snk1.data())
print "sdft overlap = ", tx_sdft.overlap()
print "sdft history = ", tx_sdft.history()
print "\ntx_sdft"
print r0.reshape((-1, L)).T
print "\npolyphase"
print r1.reshape((-1, L)).T
tmp = np.concatenate((r0, r1)).reshape((2, -1)).T
for x, y in tmp:
print x, y, "\tis equal", x == y
self.assertComplexTuplesAlmostEqual(r0, r1[0:len(r0)])
def test_003_go_big(self):
print "\n\n\ntest_003"
multiple = 2000
overlap = 4
# get test data, taps, config, etc.
self.cfg = fbmc.fbmc_config(num_used_subcarriers=20, num_payload_sym=16, num_overlap_sym=overlap,
modulation="QPSK", preamble="IAM")
L = self.cfg.num_total_subcarriers()
taps = np.array(self.cfg.prototype_taps_float(), dtype=np.float)
taps[-1] = 0
data = np.arange(1, L + 1, dtype=np.complex)
data = np.repeat(data, multiple)
# set up fg
tx_sdft = fbmc.tx_sdft_vcc(taps, L)
src = blocks.vector_source_c(data, vlen=L)
snk0 = blocks.vector_sink_c(vlen=1)
self.tb.connect(src, tx_sdft, snk0)
ft = fft.fft_vcc(L, False, (), False, 1)
pfb = fbmc.polyphase_filterbank_vcvc(L, taps)
comm = fbmc.output_commutator_vcc(L=L)
snk1 = blocks.vector_sink_c(vlen=1)
self.tb.connect(src, ft, pfb, comm, snk1)
self.tb.run()
r0 = np.array(snk0.data()[L:])
r1 = np.array(snk1.data()[:-L])
print "\ntx_sdft"
print r0.reshape((-1, L)).T
print "\npolyphase"
print r1.reshape((-1, L)).T
self.assertComplexTuplesAlmostEqual(r0, r1, 4)