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!"
