diff options
Diffstat (limited to 'gnuradio-runtime/examples/mp-sched/affinity_set.py')
| -rw-r--r-- | gnuradio-runtime/examples/mp-sched/affinity_set.py | 24 |
1 files changed, 15 insertions, 9 deletions
diff --git a/gnuradio-runtime/examples/mp-sched/affinity_set.py b/gnuradio-runtime/examples/mp-sched/affinity_set.py index 4516b96f54..530742ef22 100644 --- a/gnuradio-runtime/examples/mp-sched/affinity_set.py +++ b/gnuradio-runtime/examples/mp-sched/affinity_set.py @@ -13,6 +13,7 @@ from gnuradio.filter import firdes from argparse import ArgumentParser import sys + class affinity_set(gr.top_block): def __init__(self): @@ -27,19 +28,23 @@ class affinity_set(gr.top_block): # Blocks ################################################## vec_len = 1 - self.blocks_throttle_0 = blocks.throttle(gr.sizeof_gr_complex*vec_len, samp_rate) - self.blocks_null_source_0 = blocks.null_source(gr.sizeof_gr_complex*vec_len) - self.blocks_null_sink_0 = blocks.null_sink(gr.sizeof_gr_complex*vec_len) - self.filter_filt_0 = filter.fir_filter_ccc(1, 40000*[0.2+0.3j,]) - self.filter_filt_1 = filter.fir_filter_ccc(1, 40000*[0.2+0.3j,]) + self.blocks_throttle_0 = blocks.throttle( + gr.sizeof_gr_complex * vec_len, samp_rate) + self.blocks_null_source_0 = blocks.null_source( + gr.sizeof_gr_complex * vec_len) + self.blocks_null_sink_0 = blocks.null_sink( + gr.sizeof_gr_complex * vec_len) + self.filter_filt_0 = filter.fir_filter_ccc(1, 40000 * [0.2 + 0.3j, ]) + self.filter_filt_1 = filter.fir_filter_ccc(1, 40000 * [0.2 + 0.3j, ]) - self.filter_filt_0.set_processor_affinity([0,]) - self.filter_filt_1.set_processor_affinity([0,1]) + self.filter_filt_0.set_processor_affinity([0, ]) + self.filter_filt_1.set_processor_affinity([0, 1]) ################################################## # Connections ################################################## - self.connect((self.blocks_null_source_0, 0), (self.blocks_throttle_0, 0)) + self.connect((self.blocks_null_source_0, 0), + (self.blocks_throttle_0, 0)) self.connect((self.blocks_throttle_0, 0), (self.filter_filt_0, 0)) self.connect((self.filter_filt_0, 0), (self.filter_filt_1, 0)) self.connect((self.filter_filt_1, 0), (self.blocks_null_sink_0, 0)) @@ -50,6 +55,7 @@ class affinity_set(gr.top_block): def set_samp_rate(self, samp_rate): self.samp_rate = samp_rate + if __name__ == '__main__': parser = ArgumentParser() args = parser.parse_args() @@ -69,4 +75,4 @@ if __name__ == '__main__': except ValueError: print("Invalid number") else: - tb.filter_filt_0.set_processor_affinity([n,]) + tb.filter_filt_0.set_processor_affinity([n, ]) |