Skip to content

Commit

Permalink
add scripts to generate the mask files
Browse files Browse the repository at this point in the history
  • Loading branch information
ziyixi committed Aug 30, 2020
1 parent 4440bf8 commit fc7bbb8
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions seisflow/scripts/xsede/xsede_make_gaussian_mask.py
@@ -0,0 +1,24 @@
import click

from ...slurm.submit_job import submit_job
from ..structure_inversion.make_gaussian_mask import make_gaussian_mask


@click.command()
@click.option('--base_dir', required=True, type=str, help="the simulation directory")
@click.option('--sem_bin_dir', required=True, type=str, help="the sem program directory")
@click.option('--output_dir', required=True, type=str, help="the output directory")
@click.option('--nproc', required=True, type=int, help="nproc in the database directory")
@click.option('--n_node', required=True, type=int, help="number of nodes to use")
@click.option('--n_cores', required=True, type=int, help="number of cores to use")
@click.option('--partition', required=True, type=str, help="the partition to use, eg: skx-normal")
@click.option('--used_time', required=True, type=str, help="the running time")
@click.option('--account', required=True, type=str, help="the account to use")
def main(base_dir, sem_bin_dir, output_dir, nproc, n_node, n_cores, partition, used_time, account):
thecommand = make_gaussian_mask(base_dir, sem_bin_dir, output_dir, nproc)
submit_job("generate mask", thecommand,
n_node, n_cores, partition, used_time, account, "stampede2")


if __name__ == "__main__":
main() # pylint: disable=no-value-for-parameter

0 comments on commit fc7bbb8

Please sign in to comment.