def main():
parser = ArgumentParser()
parser.add_argument('--seed')
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)
particle_locations = prj.seed(args)
if particle_locations.shape[1] == 2:
locations = np.zeros([particle_locations.shape[0], 3], dtype='float64')
locations[:, :2] = particle_locations
locations[:, 2] = 4.5
else:
locations = particle_locations
dataset = netCDF4.Dataset(args.seed, 'w')
dataset.createDimension('particle', locations.shape[0])
dataset.createDimension('loc', 3)
nc_locations = dataset.createVariable('locations', 'f8', ('particle', 'loc'))
nc_locations[:] = locations
dataset.close()
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)
