From 741f2deb4f6e26f5d55ffe0291e7368379e24c82 Mon Sep 17 00:00:00 2001 From: Sakimori Date: Thu, 31 Jul 2025 12:21:43 -0400 Subject: [PATCH] added output directory option --- signalgen.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/signalgen.py b/signalgen.py index 1777952..58903ab 100755 --- a/signalgen.py +++ b/signalgen.py @@ -236,6 +236,9 @@ if __name__ == "__main__": parser.add_argument( "-D", "--directory", dest="fileDir", type=str, help="Directory containing filterbank files." ) + parser.add_argument( + "-o", "--output", dest="output", type=str, help="Set output directory." + ) parser.set_defaults(dm=250.0) parser.set_defaults(nsamp=int(3e5)) parser.set_defaults(listfile=None) @@ -243,10 +246,14 @@ if __name__ == "__main__": parser.set_defaults(fileDir=None) parser.set_defaults(plot=False) parser.set_defaults(rsamp=False) + parser.set_defaults(output=None) values = parser.parse_args() - #set working directory to ignored directory - outdir = path.join(getcwd(),"out","") + #set working directory to ignored directory or set output + if values.output is None: + outdir = path.join(getcwd(),"out","") + else: + outdir = values.output if not path.isdir(outdir): makedirs(outdir) chdir(outdir)