git.gag.com Git - debian/gnuradio/blob - gnuradio-core/src/python/gnuradio/gr/benchmark_filters.py

   1 #!/usr/bin/env python
   2
   3 import time
   4 import random
   5 from optparse import OptionParser
   6 from gnuradio import gr
   7 from gnuradio.eng_option import eng_option
   8
   9 def make_random_complex_tuple(L):
  10     result = []
  11     for x in range(L):
  12         result.append(complex(random.uniform(-1000,1000),
  13                               random.uniform(-1000,1000)))
  14     return tuple(result)
  15
  16 def benchmark(name, creator, dec, ntaps, total_test_size, block_size):
  17     block_size = 32768
  18
  19     fg = gr.flow_graph()
  20     taps = make_random_complex_tuple(ntaps)
  21     src = gr.vector_source_c(make_random_complex_tuple(block_size), True)
  22     head = gr.head(gr.sizeof_gr_complex, int(total_test_size))
  23     op = creator(dec, taps)
  24     dst = gr.null_sink(gr.sizeof_gr_complex)
  25     fg.connect(src, head, op, dst)
  26     start = time.time()
  27     fg.run()
  28     stop = time.time()
  29     delta = stop - start
  30     print "%16s: taps: %4d  input: %4g, time: %6.3f  taps/sec: %10.4g" % (
  31         name, ntaps, total_test_size, delta, ntaps*total_test_size/delta)
  32
  33 def main():
  34     parser = OptionParser(option_class=eng_option)
  35     parser.add_option("-n", "--ntaps", type="int", default=256)
  36     parser.add_option("-t", "--total-input-size", type="eng_float", default=40e6)
  37     parser.add_option("-b", "--block-size", type="intx", default=50000)
  38     parser.add_option("-d", "--decimation", type="int", default=1)
  39     (options, args) = parser.parse_args()
  40     if len(args) != 0:
  41         parser.print_help()
  42         sys.exit(1)
  43
  44     ntaps = options.ntaps
  45     total_input_size = options.total_input_size
  46     block_size = options.block_size
  47     dec = options.decimation
  48
  49     benchmark("gr.fir_filter_ccc", gr.fir_filter_ccc,
  50               dec, ntaps, total_input_size, block_size)
  51     benchmark("gr.fft_filter_ccc", gr.fft_filter_ccc,
  52               dec, ntaps, total_input_size, block_size)
  53
  54 if __name__ == '__main__':
  55     main()