Skip to content
Snippets Groups Projects
Commit 68030e1d authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Make output file optional in command line tool

parent e4a7515d
No related branches found
No related tags found
No related merge requests found
...@@ -50,8 +50,9 @@ def main(): ...@@ -50,8 +50,9 @@ def main():
parser = ArgumentParser(description="Stand-alone loopy frontend") parser = ArgumentParser(description="Stand-alone loopy frontend")
parser.add_argument("infile") parser.add_argument("infile", metavar="INPUT_FILE")
parser.add_argument("outfile") parser.add_argument("outfile", default="-", metavar="OUTPUT_FILE",
help="Defaults to stdout ('-').", nargs='?')
parser.add_argument("--lang", metavar="LANGUAGE", help="loopy|fortran") parser.add_argument("--lang", metavar="LANGUAGE", help="loopy|fortran")
parser.add_argument("--target") parser.add_argument("--target")
parser.add_argument("--name") parser.add_argument("--name")
...@@ -185,10 +186,15 @@ def main(): ...@@ -185,10 +186,15 @@ def main():
code, impl_arg_info = generate_code(kernel) code, impl_arg_info = generate_code(kernel)
codes.append(code) codes.append(code)
if args.outfile == "-": if args.outfile:
outfile, = args.outfile
else:
outfile = "-"
if outfile == "-":
sys.stdout.write("\n\n".join(codes)) sys.stdout.write("\n\n".join(codes))
else: else:
with open(args.outfile, "w") as outfile_fd: with open(outfile, "w") as outfile_fd:
outfile_fd.write("\n\n".join(codes)) outfile_fd.write("\n\n".join(codes))
......
#! /bin/sh
NAME="$1"
shift
python $(which loopy) --lang=fpp "$NAME" - "$@"
#! /bin/sh
python $(which loopy) --lang=loopy "$NAME" - "$@"
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment