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)