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)