()
| 37 | return args; |
| 38 | |
| 39 | def main(): |
| 40 | args = parse_args(); |
| 41 | side_lengths = np.ones(args.dim) * args.size; |
| 42 | if (args.X is not None): |
| 43 | side_lengths[0] = args.X |
| 44 | if (args.Y is not None): |
| 45 | side_lengths[1] = args.Y |
| 46 | if (args.dim == 3 and args.Z is not None): |
| 47 | side_lengths[2] = args.Z |
| 48 | |
| 49 | num_samples = np.ones(args.dim, dtype=int) * args.num_samples; |
| 50 | if args.num_samples_X is not None: |
| 51 | num_samples[0] = args.num_samples_X; |
| 52 | if args.num_samples_Y is not None: |
| 53 | num_samples[1] = args.num_samples_Y; |
| 54 | if args.dim == 3 and args.num_samples_Z is not None: |
| 55 | num_samples[2] = args.num_samples_Z; |
| 56 | |
| 57 | using_simplex = not(args.with_quad or args.with_hex); |
| 58 | mesh = generate_box_mesh(-0.5 * side_lengths, 0.5 * side_lengths, |
| 59 | num_samples, args.symmetric, args.subdiv, using_simplex); |
| 60 | save_mesh(args.output, mesh, "cell_index"); |
| 61 | |
| 62 | if __name__ == "__main__": |
| 63 | main(); |
no test coverage detected