1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#
# Copyright 2013 Free Software Foundation, Inc.
#
# This file is part of GNU Radio.
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
###############################################################################
# Imports
###############################################################################
from gnuradio import zeromq
#import zeromq
from gnuradio import gr
from gnuradio import blocks
from gnuradio import analog
from gnuradio import eng_notation
from gnuradio.eng_arg import eng_float, intx
from argparse import ArgumentParser
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://*:6667"
probe_adr = "tcp://*:5557"
source_adr = "tcp://" + self.options.servername + ":5555"
# blocks
#self.zmq_source = zeromq.req_source(gr.sizeof_float, 1, source_adr)
#self.zmq_source = zeromq.pull_source(gr.sizeof_float, 1, source_adr)
self.zmq_source = zeromq.sub_source(gr.sizeof_float, 1, source_adr)
#self.zmq_probe = zeromq.push_sink(gr.sizeof_float, 1, probe_adr)
self.zmq_probe = zeromq.pub_sink(gr.sizeof_float, 1, probe_adr)
# connects
self.connect(self.zmq_source, 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.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()
###############################################################################
# Options Parser
###############################################################################
def parse_args():
"""Argument parser."""
parser = ArgumentParser()
parser.add_argument("-s", "--servername", default="localhost",
help="Server hostname")
args = parser.parse_args()
return args
###############################################################################
# Waiter Thread
###############################################################################
class waiter(Thread):
""" To keep the program alive when flowgraph is stopped. """
def run(self):
while keep_running:
time.sleep(1)
###############################################################################
# Main
###############################################################################
if __name__ == "__main__":
args = parse_args()
tb = top_block(args)
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
|
