zhetadelta/greenburstAux
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

added output directory option

Browse source
This commit is contained in:
Sakimori 2025-07-31 12:21:43 -04:00
parent 2ef005ca70
commit 741f2deb4f
No known key found for this signature in database
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
signalgen.py
View file

@ -236,6 +236,9 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
"-D", "--directory", dest="fileDir", type=str, help="Directory containing filterbank files." "-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(dm=250.0)
parser.set_defaults(nsamp=int(3e5)) parser.set_defaults(nsamp=int(3e5))
parser.set_defaults(listfile=None) parser.set_defaults(listfile=None)
@ -243,10 +246,14 @@ if __name__ == "__main__":
parser.set_defaults(fileDir=None) parser.set_defaults(fileDir=None)
parser.set_defaults(plot=False) parser.set_defaults(plot=False)
parser.set_defaults(rsamp=False) parser.set_defaults(rsamp=False)
parser.set_defaults(output=None)
values = parser.parse_args() values = parser.parse_args()
#set working directory to ignored directory #set working directory to ignored directory or set output
outdir = path.join(getcwd(),"out","") if values.output is None:
outdir = path.join(getcwd(),"out","")
else:
outdir = values.output
if not path.isdir(outdir): if not path.isdir(outdir):
makedirs(outdir) makedirs(outdir)
chdir(outdir) chdir(outdir)