def test_004_connect (self): """ Advanced test: - Allocator -> IFFT -> Frequency offset -> FFT -> Serializer - FFT does shift (moves DC to middle) - Make sure input == output - Frequency offset is -2 carriers """ fft_len = 8 n_syms = 1 carr_offset = -2 freq_offset = 1.0 / fft_len * carr_offset # Normalized frequency occupied_carriers = ((-2, -1, 1, 2),) pilot_carriers = ((-3,),(3,)) pilot_symbols = ((1j,),(-1j,)) tx_data = (1, 2, 3, 4) tag_name = "len" tag = gr.tag_t() tag.offset = 0 tag.key = pmt.string_to_symbol(tag_name) tag.value = pmt.from_long(len(tx_data)) offsettag = gr.tag_t() offsettag.offset = 0 offsettag.key = pmt.string_to_symbol("ofdm_sync_carr_offset") offsettag.value = pmt.from_long(carr_offset) src = blocks.vector_source_c(tx_data, False, 1, (tag, offsettag)) alloc = digital.ofdm_carrier_allocator_cvc(fft_len, occupied_carriers, pilot_carriers, pilot_symbols, (), tag_name) tx_ifft = fft.fft_vcc(fft_len, False, (1.0/fft_len,)*fft_len, True) oscillator = analog.sig_source_c(1.0, analog.GR_COS_WAVE, freq_offset, 1.0/fft_len) mixer = blocks.multiply_cc() rx_fft = fft.fft_vcc(fft_len, True, (), True) sink2 = blocks.vector_sink_c(fft_len) self.tb.connect(rx_fft, sink2) serializer = digital.ofdm_serializer_vcc( alloc, "", 0, "ofdm_sync_carr_offset", True ) sink = blocks.vector_sink_c() self.tb.connect( src, alloc, tx_ifft, blocks.vector_to_stream(gr.sizeof_gr_complex, fft_len), (mixer, 0), blocks.stream_to_vector(gr.sizeof_gr_complex, fft_len), rx_fft, serializer, sink ) self.tb.connect(oscillator, (mixer, 1)) self.tb.run () self.assertComplexTuplesAlmostEqual(sink.data()[-len(occupied_carriers[0]):], tx_data, places=4)
def test_004_connect (self): """ Advanced test: - Allocator -> IFFT -> Frequency offset -> FFT -> Serializer - FFT does shift (moves DC to middle) - Make sure input == output - Frequency offset is -2 carriers """ fft_len = 8 n_syms = 2 carr_offset = -2 freq_offset = 2 * numpy.pi * carr_offset / fft_len # If the sampling rate == 1 occupied_carriers = ((1, 2, -2, -1),) pilot_carriers = ((3,),(5,)) pilot_symbols = ((1j,),(-1j,)) tx_data = tuple([numpy.random.randint(0, 10) for x in range(4 * n_syms)]) #tx_data = (1,) * occupied_carriers[0] * n_syms tag_name = "len" tag = gr.gr_tag_t() tag.offset = 0 tag.key = pmt.pmt_string_to_symbol(tag_name) tag.value = pmt.pmt_from_long(len(tx_data)) offsettag = gr.gr_tag_t() offsettag.offset = 0 offsettag.key = pmt.pmt_string_to_symbol("ofdm_sync_carr_offset") offsettag.value = pmt.pmt_from_long(carr_offset) src = gr.vector_source_c(tx_data, False, 1, (tag, offsettag)) alloc = digital.ofdm_carrier_allocator_cvc(fft_len, occupied_carriers, pilot_carriers, pilot_symbols, tag_name) tx_ifft = fft.fft_vcc(fft_len, False, ()) offset_sig = gr.sig_source_c(1.0, gr.GR_COS_WAVE, freq_offset, 1.0) mixer = gr.multiply_cc() rx_fft = fft.fft_vcc(fft_len, True, (), True) serializer = digital.ofdm_serializer_vcc(alloc) sink = gr.vector_sink_c() self.tb.connect( src, alloc, tx_ifft, gr.vector_to_stream(gr.sizeof_gr_complex, fft_len), (mixer, 0), gr.stream_to_vector(gr.sizeof_gr_complex, fft_len), rx_fft, serializer, sink ) self.tb.connect(offset_sig, (mixer, 1)) self.tb.run ()
def test_003(self): # Same test as above, only use 2 threads tb = gr.top_block() fft_size = 32 tmp_data = ( (4377 + 4516j), (-1706.1268310546875 + 1638.4256591796875j), (-915.2083740234375 + 660.69427490234375j), (-660.370361328125 + 381.59600830078125j), (-499.96044921875 + 238.41630554199219j), (-462.26748657226562 + 152.88948059082031j), (-377.98440551757812 + 77.5928955078125j), (-346.85821533203125 + 47.152004241943359j), (-295 + 20j), (-286.33609008789062 - 22.257017135620117j), (-271.52999877929688 - 33.081821441650391j), (-224.6358642578125 - 67.019538879394531j), (-244.24473571777344 - 91.524826049804688j), (-203.09068298339844 - 108.54627227783203j), (-198.45195007324219 - 115.90768432617188j), (-182.97744750976562 - 128.12318420410156j), (-167 - 180j), (-130.33688354492188 - 173.83778381347656j), (-141.19784545898438 - 190.28807067871094j), (-111.09677124023438 - 214.48896789550781j), (-70.039543151855469 - 242.41630554199219j), (-68.960540771484375 - 228.30015563964844j), (-53.049201965332031 - 291.47097778320312j), (-28.695289611816406 - 317.64553833007812j), (57 - 300j), (45.301143646240234 - 335.69509887695312j), (91.936195373535156 - 373.32437133789062j), (172.09465026855469 - 439.275146484375j), (242.24473571777344 - 504.47515869140625j), (387.81732177734375 - 666.6788330078125j), (689.48553466796875 - 918.2142333984375j), (1646.539306640625 - 1694.1956787109375j), ) src_data = tuple([x / fft_size for x in tmp_data]) expected_result = tuple([complex(primes[2 * i], primes[2 * i + 1]) for i in range(fft_size)]) nthreads = 2 src = blocks.vector_source_c(src_data) s2v = blocks.stream_to_vector(gr.sizeof_gr_complex, fft_size) op = fft.fft_vcc(fft_size, False, [], False, nthreads) v2s = blocks.vector_to_stream(gr.sizeof_gr_complex, fft_size) dst = blocks.vector_sink_c() tb.connect(src, s2v, op, v2s, dst) tb.run() result_data = dst.data() self.assert_fft_ok2(expected_result, result_data)
def test_003(self): # Same test as above, only use 2 threads tb = gr.top_block() fft_size = 32 tmp_data = ( (4377 + 4516j), (-1706.1268310546875 + 1638.4256591796875j), (-915.2083740234375 + 660.69427490234375j), (-660.370361328125 + 381.59600830078125j), (-499.96044921875 + 238.41630554199219j), (-462.26748657226562 + 152.88948059082031j), (-377.98440551757812 + 77.5928955078125j), (-346.85821533203125 + 47.152004241943359j), (-295 + 20j), (-286.33609008789062 - 22.257017135620117j), (-271.52999877929688 - 33.081821441650391j), (-224.6358642578125 - 67.019538879394531j), (-244.24473571777344 - 91.524826049804688j), (-203.09068298339844 - 108.54627227783203j), (-198.45195007324219 - 115.90768432617188j), (-182.97744750976562 - 128.12318420410156j), (-167 - 180j), (-130.33688354492188 - 173.83778381347656j), (-141.19784545898438 - 190.28807067871094j), (-111.09677124023438 - 214.48896789550781j), (-70.039543151855469 - 242.41630554199219j), (-68.960540771484375 - 228.30015563964844j), (-53.049201965332031 - 291.47097778320312j), (-28.695289611816406 - 317.64553833007812j), (57 - 300j), (45.301143646240234 - 335.69509887695312j), (91.936195373535156 - 373.32437133789062j), (172.09465026855469 - 439.275146484375j), (242.24473571777344 - 504.47515869140625j), (387.81732177734375 - 666.6788330078125j), (689.48553466796875 - 918.2142333984375j), (1646.539306640625 - 1694.1956787109375j)) src_data = tuple([x / fft_size for x in tmp_data]) expected_result = tuple([ complex(primes[2 * i], primes[2 * i + 1]) for i in range(fft_size) ]) nthreads = 2 src = blocks.vector_source_c(src_data) s2v = blocks.stream_to_vector(gr.sizeof_gr_complex, fft_size) op = fft.fft_vcc(fft_size, False, [], False, nthreads) v2s = blocks.vector_to_stream(gr.sizeof_gr_complex, fft_size) dst = blocks.vector_sink_c() tb.connect(src, s2v, op, v2s, dst) tb.run() result_data = dst.data() self.assert_fft_ok2(expected_result, result_data)