added output directory option
This commit is contained in:
parent
2ef005ca70
commit
741f2deb4f
|
@ -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
|
||||
#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)
|
||||
|
|
Loading…
Reference in a new issue