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
|
#!/usr/bin/env python
#
# Copyright 2012 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
# GNU Radio 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.
#
# GNU Radio 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 GNU Radio; see the file COPYING. If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street,
# Boston, MA 02110-1301, USA.
#
import os
import sys
from argparse import ArgumentParser
import warnings
warnings.simplefilter('ignore')
from gnuradio import gr
try:
from grc.core.Platform import Platform
except ImportError:
from gnuradio.grc.core.Platform import Platform
class GRCC:
def __init__(self, grcfile, out_dir):
self.out_dir = out_dir
self.platform = Platform(
prefs_file=gr.prefs(),
version=gr.version(),
version_parts=(gr.major_version(), gr.api_version(), gr.minor_version())
)
data = self.platform.parse_flow_graph(grcfile)
self.fg = self.platform.get_new_flow_graph()
self.fg.import_data(data)
self.fg.grc_file_path = os.path.abspath(grcfile)
self.fg.validate()
if not self.fg.is_valid():
raise StandardError("\n\n".join(
["Validation failed:"] + self.fg.get_error_messages()
))
self.gen = self.platform.Generator(self.fg, out_dir)
self.gen.write()
def exec_program(self):
progname = self.fg.get_option('id')
os.system("{0}/{1}.py".format(self.out_dir, progname))
def main():
description = "Compiles a GRC file (.grc) into a GNU Radio Python program. The program is stored in ~/.grc_gnuradio by default, but this location can be changed with the -d option."
parser = ArgumentParser(description=description)
parser.add_argument("-d", "--directory",
default='{0}/.grc_gnuradio/'.format(os.environ["HOME"]),
help="Specify the directory to output the compile program [default=%(default)s]")
parser.add_argument("-e", "--execute", action="store_true", default=False,
help="Run the program after compiling [default=%(default)s]")
parser.add_argument('grc_file', metavar="GRC_FILE", help=".grc file to compile")
args = parser.parse_args()
try:
g = GRCC(args.grc_file, args.directory + "/")
except Exception as e:
sys.stderr.write(str(e) + "\n")
sys.stderr.write("Error during file compilation.\n")
sys.exit(1)
if args.execute:
g.exec_program()
if __name__ == "__main__":
main()
|
