def main(): parser = ArgumentParser() parser.add_argument("--seed", required=True) parser.add_argument("--start", type=timetype) parser.add_argument("--end", type=timetype) parser.add_argument("--delta-hours", type=timedeltatype("hours"), default=timedelta(hours=0)) parser.add_argument("--delta-days", type=timedeltatype("days"), default=timedelta(days=0)) subparsers = parser.add_subparsers(help="sub-command help") for name, project in get_projects(): sub_parser = subparsers.add_parser(name) project.commandline_args(sub_parser) args = parser.parse_args() prj = args.project(args) dataset = netCDF4.Dataset(args.seed, "r") locations = dataset.variables["locations"] read_params(open("tes_run.yaml")) read_params(open("tes_grid.yaml")) prj.setup_tracmass(locations)
def main(): parser = ArgumentParser() parser.add_argument('-g', '--grid-params', type=FileType('r'), required=True) parser.add_argument('--part-quant', default=1) parser.add_argument('--isec', default=4) parser.add_argument('--idir', default=0) parser.add_argument('--nqua', default=1) parser.add_argument('--ist1', default=10) parser.add_argument('--ist2', default=10) parser.add_argument('--jst1', default=50) parser.add_argument('--jst2', default=50) parser.add_argument('--kst1', default=5) parser.add_argument('--kst2', default=5) args = parser.parse_args() read_params(args.grid_params) tm.mod_param.partquant = args.part_quant tm.mod_param.partquant = args.part_quant tm.mod_seed.isec = args.isec tm.mod_seed.idir = args.idir tm.mod_seed.nqua = args.nqua tm.mod_seed.seedtype = 1 tm.mod_seed.ist1 = args.ist1 tm.mod_seed.ist2 = args.ist2 tm.mod_seed.jst1 = args.jst1 tm.mod_seed.jst2 = args.jst2 tm.mod_seed.kst1 = args.kst1 tm.mod_seed.kst2 = args.kst2 tm.init_params2() tm.coordinat() tm.mod_grid.kmt[:] = tm.mod_param.km if tm.mod_seed.nqua == 1: # number of trajectories (per time resolution) # num=NTRACMAX tm.mod_seed.num = tm.mod_param.partquant elif tm.mod_seed.nqua == 2: tm.mod_param.voltr = tm.mod_param.partquant elif tm.mod_seed.nqua == 3: tm.mod_param.voltr = tm.mod_param.partquant tm.mod_traj.trj[:] = 0 tm.mod_traj.nrj[:] = 0 tm.init_seed() tm.mod_seed.seed(0, 0) print tm.mod_seed.seed_ijk print tm.mod_traj.trj
def main(): parser = ArgumentParser() parser.add_argument('-r', '--run-params', type=FileType('r')) parser.add_argument('-g', '--grid-params', type=FileType('r')) parser.add_argument('-s', '--set', nargs=2, action='append') parser.add_argument('--seed') parser.add_argument('--exit') parser.add_argument('--path') parser.add_argument('--start') parser.add_argument('--end') parser.add_argument('--delta-hours') parser.add_argument('--delta-minutes') parser.add_argument('--delta-seconds') parser.add_argument('--print-state', action='store_true') args = parser.parse_args() read_params(args.run_params) read_params(args.grid_params) tm.init_params2() tm.coordinat() setup() tm.mod_grid.kmt[:] = tm.mod_param.km tm.allocate_seed(1) tm.mod_seed.isec = 5 tm.mod_seed.seed_ijk[:,:] = 2 tm.mod_seed.seed_xyz[:,:] = [[9.125, 49.125, 4.5]] tm.mod_seed.seed_set[:] = [tm.mod_seed.isec, tm.mod_seed.idir] if args.print_state: print_state() return print tm.mod_traj.trj.shape print tm.mod_traj.trj[:1,:] def writedata(*args): print tm.mod_traj.trj[:1,:] print tm.writedata2(*args) tm.loop.writedata = tm.writedata2 tm.loop.readfields = tm.tes_readfields tm.loop() print "Done!"