diff options
| author | Johnathan Corgan <johnathan@corganlabs.com> | 2014-05-06 13:40:15 -0700 |
|---|---|---|
| committer | Johnathan Corgan <johnathan@corganlabs.com> | 2014-05-06 13:40:15 -0700 |
| commit | ddb4c1d744b417e79cf2d263d6cd8db37c4dbcb2 (patch) | |
| tree | 3855d9b0d071ab07246c6779c0c5554ba2d943b9 /gr-zeromq/examples/python/server.py | |
| parent | 73a6083b84dd3506bf4fdfbcb15bfd9a7033c98c (diff) | |
zeromq: cleanup docs and examples
Diffstat (limited to 'gr-zeromq/examples/python/server.py')
| -rwxr-xr-x | gr-zeromq/examples/python/server.py | 125 |
1 files changed, 125 insertions, 0 deletions
diff --git a/gr-zeromq/examples/python/server.py b/gr-zeromq/examples/python/server.py new file mode 100755 index 0000000000..28b732ed3e --- /dev/null +++ b/gr-zeromq/examples/python/server.py @@ -0,0 +1,125 @@ +# +# Copyright 2013 Free Software Foundation, Inc. +# +# This file is part of GNU Radio. +# +# This is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 3, or (at your option) +# any later version. +# +# This software is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this software; see the file COPYING. If not, write to +# the Free Software Foundation, Inc., 51 Franklin Street, +# Boston, MA 02110-1301, USA. +# + +############################################################################### +# Imports +############################################################################### +from gnuradio import zeromq +from gnuradio import gr +from gnuradio import blocks +from gnuradio import analog +from gnuradio import eng_notation +from gnuradio.eng_option import eng_option +from optparse import OptionParser +import numpy +import sys +from threading import Thread +import time + + +############################################################################### +# GNU Radio top_block +############################################################################### +class top_block(gr.top_block): + def __init__(self, options): + gr.top_block.__init__(self) + + self.options = options + + # socket addresses + rpc_adr = "tcp://*:6666" + probe_adr = "tcp://*:5556" + sink_adr = "tcp://*:5555" + + # the strange sampling rate gives a nice movement in the plot :P + self.samp_rate = samp_rate = 48200 + + # blocks + self.gr_sig_source = analog.sig_source_f(samp_rate, analog.GR_SIN_WAVE , 1000, 1, 0) + self.throttle = blocks.throttle(gr.sizeof_float, samp_rate) + self.mult = blocks.multiply_const_ff(1) + self.zmq_sink = zeromq.rep_sink(gr.sizeof_float, 1, sink_adr) + #self.zmq_sink = zeromq.push_sink(gr.sizeof_float, 1, sink_adr) + #self.zmq_probe = zeromq.push_sink(gr.sizeof_float, probe_adr) + self.zmq_probe = zeromq.pub_sink(gr.sizeof_float, probe_adr) + #self.null_sink = blocks.null_sink(gr.sizeof_float) + + # connects + self.connect(self.gr_sig_source, self.mult, self.throttle, self.zmq_sink) + self.connect(self.throttle, self.zmq_probe) + + # ZeroMQ + self.rpc_manager = zeromq.rpc_manager() + self.rpc_manager.set_reply_socket(rpc_adr) + self.rpc_manager.add_interface("start_fg",self.start_fg) + self.rpc_manager.add_interface("stop_fg",self.stop_fg) + self.rpc_manager.add_interface("set_waveform",self.set_waveform) + self.rpc_manager.add_interface("set_k",self.mult.set_k) + self.rpc_manager.add_interface("get_sample_rate",self.throttle.sample_rate) + self.rpc_manager.start_watcher() + + def start_fg(self): + print "Start Flowgraph" + try: + self.start() + except RuntimeError: + print "Can't start, flowgraph already running!" + + def stop_fg(self): + print "Stop Flowgraph" + self.stop() + self.wait() + + def set_waveform(self, waveform_str): + waveform = {'Constant' : analog.GR_CONST_WAVE, + 'Sine' : analog.GR_SIN_WAVE, + 'Cosine' : analog.GR_COS_WAVE, + 'Square' : analog.GR_SQR_WAVE, + 'Triangle' : analog.GR_TRI_WAVE, + 'Saw Tooth' : analog.GR_SAW_WAVE}[waveform_str] + self.gr_sig_source.set_waveform(waveform) + +############################################################################### +# Options Parser +############################################################################### +def parse_options(): + """ Options parser. """ + parser = OptionParser(option_class=eng_option, usage="%prog: [options]") + (options, args) = parser.parse_args() + return options + +############################################################################### +# Main +############################################################################### +if __name__ == "__main__": + options = parse_options() + tb = top_block(options) + try: + # keep the program running when flowgraph is stopped + while True: + time.sleep(1) + except KeyboardInterrupt: + pass + print "Shutting down flowgraph." + tb.rpc_manager.stop_watcher() + tb.stop() + tb.wait() + tb = None |