From 3388c012cfb07b959c3cde93ea0fd07bb0fb776f Mon Sep 17 00:00:00 2001 From: Sebastian Koslowski <koslowski@kit.edu> Date: Fri, 23 Sep 2016 21:13:36 +0200 Subject: grc: rewrite grcc --- grc/compiler.py | 73 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 73 insertions(+) create mode 100755 grc/compiler.py (limited to 'grc/compiler.py') diff --git a/grc/compiler.py b/grc/compiler.py new file mode 100755 index 0000000000..3be98def75 --- /dev/null +++ b/grc/compiler.py @@ -0,0 +1,73 @@ +# Copyright 2016 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. + +from __future__ import print_function, absolute_import + +import argparse +import os +import subprocess + +from gnuradio import gr + +from .core import Messages +from .core.Platform import Platform + + +def argument_parser(): + parser = argparse.ArgumentParser(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.add_argument("-d", "--directory", dest='out_dir', default='', + help="Specify the directory to output the compiled program " + "(default is the hier_block library)") + parser.add_argument("-e", "--execute", action="store_true", default=False, + help="Run the program after compiling [default=%(default)s]") + parser.add_argument(metavar="GRC_FILE", dest='grc_files', nargs='+', + help=".grc file to compile") + return parser + + +def main(args=None): + args = args or argument_parser().parse_args() + + platform = Platform( + name='GNU Radio Companion Compiler', + prefs_file=gr.prefs(), + version=gr.version(), + version_parts=(gr.major_version(), gr.api_version(), gr.minor_version()) + ) + out_dir = args.out_dir or platform.config.hier_block_lib_dir + if not os.path.exists(out_dir): + exit('Error: Invalid output directory.') + + Messages.send_init(platform) + flow_graph = file_path = None + for grc_file in args.grc_files: + os.path.exists(grc_file) or exit('Error: missing ' + grc_file) + Messages.send('\n') + + flow_graph, file_path = platform.load_and_generate_flow_graph( + os.path.abspath(grc_file), os.path.abspath(out_dir)) + if not file_path: + exit('Compilation error. Aborting.') + if file_path and args.execute: + run_command_args = flow_graph.get_run_command(file_path, split=True) + subprocess.call(run_command_args) -- cgit v1.2.3 From da8429ffcb28bd7a0edc634bc4e1c53fbbef6b56 Mon Sep 17 00:00:00 2001 From: Sebastian Koslowski <koslowski@kit.edu> Date: Wed, 28 Sep 2016 09:28:39 +0200 Subject: grc: changes to grcc CLI --- grc/compiler.py | 27 +++++++++++++++------------ grc/tests/test_compiler.py | 8 ++++++-- 2 files changed, 21 insertions(+), 14 deletions(-) (limited to 'grc/compiler.py') diff --git a/grc/compiler.py b/grc/compiler.py index 3be98def75..0cda0d946d 100755 --- a/grc/compiler.py +++ b/grc/compiler.py @@ -31,14 +31,13 @@ from .core.Platform import Platform def argument_parser(): parser = argparse.ArgumentParser(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." + "Compile a GRC file (.grc) into a GNU Radio Python program and run it." )) - parser.add_argument("-d", "--directory", dest='out_dir', default='', - help="Specify the directory to output the compiled program " - "(default is the hier_block library)") - parser.add_argument("-e", "--execute", action="store_true", default=False, + parser.add_argument("-o", "--output", metavar='DIR', default='.', + help="Output directory for compiled program [default=%(default)s]") + parser.add_argument("-u", "--user-lib-dir", action='store_true', default=False, + help="Output to default hier_block library (overwrites -o)") + parser.add_argument("-r", "--run", action="store_true", default=False, help="Run the program after compiling [default=%(default)s]") parser.add_argument(metavar="GRC_FILE", dest='grc_files', nargs='+', help=".grc file to compile") @@ -54,9 +53,13 @@ def main(args=None): version=gr.version(), version_parts=(gr.major_version(), gr.api_version(), gr.minor_version()) ) - out_dir = args.out_dir or platform.config.hier_block_lib_dir - if not os.path.exists(out_dir): - exit('Error: Invalid output directory.') + out_dir = args.output if not args.user_lib_dir else platform.config.hier_block_lib_dir + if os.path.exists(out_dir): + pass # all is well + elif args.save_to_lib: + os.mkdir(out_dir) # create missing hier_block lib directory + else: + exit('Error: Invalid output directory') Messages.send_init(platform) flow_graph = file_path = None @@ -67,7 +70,7 @@ def main(args=None): flow_graph, file_path = platform.load_and_generate_flow_graph( os.path.abspath(grc_file), os.path.abspath(out_dir)) if not file_path: - exit('Compilation error. Aborting.') - if file_path and args.execute: + exit('Compilation error') + if file_path and args.run: run_command_args = flow_graph.get_run_command(file_path, split=True) subprocess.call(run_command_args) diff --git a/grc/tests/test_compiler.py b/grc/tests/test_compiler.py index 1ae3db8603..27b5670871 100644 --- a/grc/tests/test_compiler.py +++ b/grc/tests/test_compiler.py @@ -25,8 +25,12 @@ from grc.compiler import main def test_compiler(capsys): - file_path = path.join(path.dirname(__file__), 'resources', 'test_compiler.grc') - args = Namespace(out_dir=tempfile.gettempdir(), grc_files=[file_path], execute=True) + args = Namespace( + output=tempfile.gettempdir(), + user_lib_dir=False, + grc_files=[path.join(path.dirname(__file__), 'resources', 'test_compiler.grc')], + run=True + ) main(args) -- cgit v1.2.3