1 // SPDX-FileCopyrightText: Copyright (c) 2017-2024, HONEE contributors. 2 // SPDX-License-Identifier: Apache-2.0 OR BSD-2-Clause 3 4 /// @file 5 /// Command line option processing for HONEE 6 7 #include <ctype.h> 8 #include <petscdevice.h> 9 #include <petscsys.h> 10 11 #include <navierstokes.h> 12 13 // Register problems to be available on the command line 14 static PetscErrorCode RegisterProblems_NS(AppCtx app_ctx) { 15 app_ctx->problems = NULL; 16 17 PetscFunctionBeginUser; 18 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "density_current", NS_DENSITY_CURRENT)); 19 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "euler_vortex", NS_EULER_VORTEX)); 20 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "shocktube", NS_SHOCKTUBE)); 21 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "advection", NS_ADVECTION)); 22 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "blasius", NS_BLASIUS)); 23 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "channel", NS_CHANNEL)); 24 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "gaussian_wave", NS_GAUSSIAN_WAVE)); 25 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "newtonian", NS_NEWTONIAN_IG)); 26 PetscCall(PetscFunctionListAdd(&app_ctx->problems, "taylor_green", NS_TAYLOR_GREEN)); 27 PetscFunctionReturn(PETSC_SUCCESS); 28 } 29 30 /** 31 @brief Convert ISO 8601 time string to duration in seconds 32 33 Accepted formats are 'hh', 'hh:mm', and 'hh:mm:ss'. 34 35 @param[in] comm MPI_Comm for error handling 36 @param[in] string string of the ISO 8601 duration 37 @param[out] duration Duration in number of seconds 38 **/ 39 PetscErrorCode ISO8601TimeDurationToSeconds(MPI_Comm comm, const char *string, time_t *duration) { 40 int num_items; 41 char **entries; 42 43 PetscFunctionBeginUser; 44 if (string[0] == '\0') { 45 *duration = 0; 46 PetscFunctionReturn(PETSC_SUCCESS); 47 } 48 for (PetscInt i = 0; i < strlen(string); i++) { 49 PetscCheck(isdigit(string[i]) || string[i] == ':', comm, PETSC_ERR_SUP, "Time duration may only include digits and ':' separator, found '%c'", 50 string[i]); 51 } 52 PetscCall(PetscStrToArray(string, ':', &num_items, &entries)); 53 switch (num_items) { 54 case 1: // Only hours 55 *duration = 60 * 60 * atoi(entries[0]); 56 break; 57 case 2: // Hours and Minutes 58 *duration = 60 * 60 * atoi(entries[0]) + 60 * atoi(entries[1]); 59 break; 60 case 3: // Hours, Minutes, and Seconds 61 *duration = 60 * 60 * atoi(entries[0]) + 60 * atoi(entries[1]) + atoi(entries[2]); 62 break; 63 default: 64 SETERRQ(comm, PETSC_ERR_SUP, "Recieved %d ':' delimited entries, expect either 1, 2, or 3", num_items); 65 } 66 PetscCall(PetscStrToArrayDestroy(num_items, entries)); 67 PetscFunctionReturn(PETSC_SUCCESS); 68 } 69 70 // Process general command line options 71 PetscErrorCode ProcessCommandLineOptions(Honee honee, SimpleBC bc) { 72 MPI_Comm comm = honee->comm; 73 AppCtx app_ctx = honee->app_ctx; 74 PetscBool ceed_flag = PETSC_FALSE; 75 PetscBool problem_flag = PETSC_FALSE; 76 PetscBool option_set = PETSC_FALSE; 77 78 PetscFunctionBeginUser; 79 { 80 PetscInt num_options; 81 82 PetscCall(PetscOptionsLeftGet(NULL, &num_options, NULL, NULL)); 83 PetscCheck(num_options > 0, comm, PETSC_ERR_USER_INPUT, 84 "Command line options required." 85 " Please consult the documentation to see which options are required."); 86 PetscCall(PetscOptionsLeftRestore(NULL, &num_options, NULL, NULL)); 87 } 88 89 PetscCall(RegisterProblems_NS(app_ctx)); 90 PetscOptionsBegin(comm, NULL, "HONEE - High-Order Navier-stokes Equation Evaluator", NULL); 91 92 PetscCall(PetscOptionsString("-ceed", "CEED resource specifier", NULL, app_ctx->ceed_resource, app_ctx->ceed_resource, 93 sizeof(app_ctx->ceed_resource), &ceed_flag)); 94 95 app_ctx->test_type = TESTTYPE_NONE; 96 PetscCall(PetscOptionsEnum("-test_type", "Type of test to run", NULL, TestTypes, (PetscEnum)(app_ctx->test_type), (PetscEnum *)&app_ctx->test_type, 97 NULL)); 98 99 app_ctx->test_tol = 1E-11; 100 PetscCall(PetscOptionsScalar("-compare_final_state_atol", "Test absolute tolerance", NULL, app_ctx->test_tol, &app_ctx->test_tol, NULL)); 101 102 PetscCall(PetscOptionsString("-compare_final_state_filename", "Test filename", NULL, app_ctx->test_file_path, app_ctx->test_file_path, 103 sizeof(app_ctx->test_file_path), NULL)); 104 105 PetscCall(PetscOptionsFList("-problem", "Problem to solve", NULL, app_ctx->problems, app_ctx->problem_name, app_ctx->problem_name, 106 sizeof(app_ctx->problem_name), &problem_flag)); 107 108 app_ctx->viz_refine = 0; 109 PetscCall(PetscOptionsInt("-viz_refine", "Regular refinement levels for visualization", NULL, app_ctx->viz_refine, &app_ctx->viz_refine, NULL)); 110 111 app_ctx->checkpoint_interval = 0; 112 app_ctx->checkpoint_vtk = PETSC_FALSE; 113 PetscCall(PetscOptionsDeprecated("-output_freq", "-checkpoint_interval", "libCEED 0.11.1", "Use -checkpoint_vtk true to include VTK output")); 114 PetscCall(PetscOptionsInt("-output_freq", "Frequency of output, in number of steps", NULL, app_ctx->checkpoint_interval, 115 &app_ctx->checkpoint_interval, &option_set)); 116 if (option_set) app_ctx->checkpoint_vtk = PETSC_TRUE; 117 PetscCall(PetscOptionsInt("-checkpoint_interval", "Frequency of output, in number of steps", NULL, app_ctx->checkpoint_interval, 118 &app_ctx->checkpoint_interval, NULL)); 119 PetscCall(PetscOptionsBool("-checkpoint_vtk", "Include VTK (*.vtu) output at each binary checkpoint", NULL, app_ctx->checkpoint_vtk, 120 &app_ctx->checkpoint_vtk, NULL)); 121 122 PetscCall(PetscOptionsBool("-output_add_stepnum2bin", "Add step number to the binary outputs", NULL, app_ctx->add_stepnum2bin, 123 &app_ctx->add_stepnum2bin, NULL)); 124 125 PetscCall(PetscStrncpy(app_ctx->output_dir, ".", 2)); 126 PetscCall(PetscOptionsString("-output_dir", "Output directory", NULL, app_ctx->output_dir, app_ctx->output_dir, sizeof(app_ctx->output_dir), NULL)); 127 PetscMPIInt rank; 128 MPI_Comm_rank(comm, &rank); 129 if (!rank) PetscCall(PetscMkdir(app_ctx->output_dir)); 130 131 PetscCall(PetscOptionsString("-continue_filename", "Filename to get initial condition from", NULL, app_ctx->cont_file, app_ctx->cont_file, 132 sizeof(app_ctx->cont_file), NULL)); 133 if (app_ctx->cont_file[0] != '\0') app_ctx->use_continue_file = PETSC_TRUE; 134 135 PetscCall( 136 PetscOptionsDeprecated("-continue", NULL, "HONEE 0.0.0", "Set -continue_filename to non-empty string to continue from previous solution")); 137 PetscCall( 138 PetscOptionsDeprecated("-continue_time_filename", NULL, "HONEE 0.0.0", "HONEE no longer supports reading in solution times from binary file")); 139 140 app_ctx->degree = 1; 141 PetscCall(PetscOptionsInt("-degree", "Polynomial degree of finite elements", NULL, app_ctx->degree, &app_ctx->degree, NULL)); 142 143 app_ctx->q_extra = 0; 144 PetscCall(PetscOptionsInt("-q_extra", "Number of extra quadrature points", NULL, app_ctx->q_extra, &app_ctx->q_extra, NULL)); 145 146 { 147 PetscBool option_set; 148 char amat_type[256] = ""; 149 PetscCall(PetscOptionsFList("-amat_type", "Set the type of Amat distinct from Pmat (-dm_mat_type)", NULL, MatList, amat_type, amat_type, 150 sizeof(amat_type), &option_set)); 151 if (option_set) PetscCall(PetscStrallocpy(amat_type, (char **)&app_ctx->amat_type)); 152 } 153 { 154 PetscBool option_set; 155 PetscCall(PetscOptionsHasName(NULL, NULL, "-pmat_pbdiagonal", &option_set)); 156 if (option_set) PetscCall(PetscPrintf(comm, "Warning! -pmat_pbdiagonal no longer used. Pmat assembly determined from -pc_type setting\n")); 157 } 158 159 // Provide default ceed resource if not specified 160 if (!ceed_flag) { 161 const char *ceed_resource = "/cpu/self"; 162 strncpy(app_ctx->ceed_resource, ceed_resource, 10); 163 } 164 // If we request a GPU, make sure PETSc has initialized its device (which is 165 // MPI-aware in case multiple devices are available) before CeedInit so that 166 // PETSc and libCEED agree about which device to use. 167 if (strncmp(app_ctx->ceed_resource, "/gpu", 4) == 0) PetscCall(PetscDeviceInitialize(PETSC_DEVICE_DEFAULT())); 168 169 // Provide default problem if not specified 170 if (!problem_flag) { 171 const char *problem_name = "density_current"; 172 strncpy(app_ctx->problem_name, problem_name, 16); 173 } 174 175 // Statistics Options 176 app_ctx->turb_spanstats_collect_interval = 1; 177 PetscCall(PetscOptionsInt("-ts_monitor_turbulence_spanstats_collect_interval", "Number of timesteps between statistics collection", NULL, 178 app_ctx->turb_spanstats_collect_interval, &app_ctx->turb_spanstats_collect_interval, NULL)); 179 180 app_ctx->turb_spanstats_viewer_interval = -1; 181 PetscCall(PetscOptionsInt("-ts_monitor_turbulence_spanstats_viewer_interval", "Number of timesteps between statistics viewer writing", NULL, 182 app_ctx->turb_spanstats_viewer_interval, &app_ctx->turb_spanstats_viewer_interval, NULL)); 183 184 PetscCall(PetscOptionsViewer("-ts_monitor_turbulence_spanstats_viewer", "Viewer for the statistics", NULL, &app_ctx->turb_spanstats_viewer, 185 &app_ctx->turb_spanstats_viewer_format, &app_ctx->turb_spanstats_enable)); 186 187 PetscCall(PetscOptionsViewer("-ts_monitor_wall_force", "Viewer for force on each (no-slip) wall", NULL, &app_ctx->wall_forces.viewer, 188 &app_ctx->wall_forces.viewer_format, NULL)); 189 190 // SGS Model Options 191 app_ctx->sgs_model_type = SGS_MODEL_NONE; 192 PetscCall(PetscOptionsEnum("-sgs_model_type", "Subgrid Stress Model type", NULL, SGSModelTypes, (PetscEnum)app_ctx->sgs_model_type, 193 (PetscEnum *)&app_ctx->sgs_model_type, NULL)); 194 195 PetscCall(PetscOptionsBool("-diff_filter_monitor", "Enable differential filtering TSMonitor", NULL, app_ctx->diff_filter_monitor, 196 &app_ctx->diff_filter_monitor, NULL)); 197 198 // Mesh Transformation Options 199 app_ctx->mesh_transform_type = MESH_TRANSFORM_NONE; 200 PetscCall(PetscOptionsEnum("-mesh_transform", "Mesh transform to perform", NULL, MeshTransformTypes, (PetscEnum)app_ctx->mesh_transform_type, 201 (PetscEnum *)&app_ctx->mesh_transform_type, NULL)); 202 203 PetscCall( 204 PetscOptionsBool("-sgs_train_enable", "Enable Data-Driven SGS training", NULL, app_ctx->sgs_train_enable, &app_ctx->sgs_train_enable, NULL)); 205 if (app_ctx->sgs_train_enable) honee->set_poststep = PETSC_TRUE; 206 207 PetscCall(PetscOptionsEnum("-div_diff_flux_projection_method", "Method of divergence of diffusive flux projection", NULL, 208 DivDiffFluxProjectionMethods, (PetscEnum)(app_ctx->divFdiffproj_method), (PetscEnum *)&app_ctx->divFdiffproj_method, 209 NULL)); 210 211 app_ctx->check_step_interval = -1; 212 PetscCall(PetscOptionsDeprecated("-ts_monitor_nan_interval", "-honee_check_step_interval", "HONEE 0.0", NULL)); 213 PetscCall(PetscOptionsInt("-honee_check_step_interval", "Number of timesteps between verifying the validity of the solution", NULL, 214 app_ctx->check_step_interval, &app_ctx->check_step_interval, NULL)); 215 if (app_ctx->check_step_interval > 0) honee->set_poststep = PETSC_TRUE; 216 217 { 218 char buffer[2048] = "0"; 219 time_t max_wall_time_buffer, max_wall_time_duration; 220 PetscCall(PetscOptionsString("-honee_max_wall_time_duration", "Maximum wall time duration HONEE should wait before stopping TSSolve", NULL, 221 buffer, buffer, sizeof(buffer), NULL)); 222 PetscCall(ISO8601TimeDurationToSeconds(comm, buffer, &max_wall_time_duration)); 223 PetscCall(PetscStrncpy(buffer, "0:1", sizeof(buffer))); // Default 1 minute buffer 224 PetscCall(PetscOptionsString("-honee_max_wall_time_buffer", 225 "Time before max_wall_time_duration when TSSolve should be stopped and checkpoint files written", NULL, buffer, 226 buffer, sizeof(buffer), NULL)); 227 PetscCall(ISO8601TimeDurationToSeconds(comm, buffer, &max_wall_time_buffer)); 228 if (max_wall_time_duration == 0) honee->max_wall_time = -1; 229 else { 230 honee->set_poststep = PETSC_TRUE; 231 honee->max_wall_time = honee->start_time + max_wall_time_duration - max_wall_time_buffer; 232 } 233 234 honee->max_wall_time_interval = 1; 235 PetscCall(PetscOptionsInt("-honee_max_wall_time_interval", 236 "Number of timesteps between checking whether TSSolve should be stopped due to max_wall_time", NULL, 237 honee->max_wall_time_interval, &honee->max_wall_time_interval, NULL)); 238 } 239 240 PetscOptionsEnd(); 241 PetscFunctionReturn(PETSC_SUCCESS); 242 } 243