Lines Matching refs:user
19 PetscErrorCode UpdateBoundaryValues(User user, Vec Q_loc, PetscReal t) { in UpdateBoundaryValues() argument
21 if (user->time_bc_set != t) { in UpdateBoundaryValues()
22 PetscCall(DMPlexInsertBoundaryValues(user->dm, PETSC_TRUE, Q_loc, t, NULL, NULL, NULL)); in UpdateBoundaryValues()
23 user->time_bc_set = t; in UpdateBoundaryValues()
32 User user = *(User *)user_data; in RHS_NS() local
33 Ceed ceed = user->ceed; in RHS_NS()
35 Vec Q_loc = user->Q_loc; in RHS_NS()
40 PetscCall(UpdateBoundaryValues(user, Q_loc, t)); in RHS_NS()
41 …if (user->phys->solution_time_label) PetscCallCeed(ceed, CeedOperatorSetContextDouble(user->op_rhs… in RHS_NS()
43 …if (user->phys->timestep_size_label) PetscCallCeed(ceed, CeedOperatorSetContextDouble(user->op_rhs… in RHS_NS()
45 PetscCall(ApplyCeedOperatorGlobalToGlobal(Q, G, user->op_rhs_ctx)); in RHS_NS()
47 PetscCall(VecReadPetscToCeed(Q_loc, &q_mem_type, user->q_ceed)); in RHS_NS()
50 PetscCall(KSPSolve(user->mass_ksp, G, G)); in RHS_NS()
52 PetscCall(VecReadCeedToPetsc(user->q_ceed, q_mem_type, Q_loc)); in RHS_NS()
104 User user = *(User *)user_data; in IFunction_NS() local
105 Ceed ceed = user->ceed; in IFunction_NS()
107 Vec Q_loc = user->Q_loc, Q_dot_loc = user->Q_dot_loc, G_loc; in IFunction_NS()
112 PetscCall(DMGetNamedLocalVector(user->dm, "ResidualLocal", &G_loc)); in IFunction_NS()
115 PetscCall(UpdateBoundaryValues(user, Q_loc, t)); in IFunction_NS()
116 …if (user->phys->solution_time_label) PetscCallCeed(ceed, CeedOperatorSetContextDouble(user->op_ifu… in IFunction_NS()
118 …if (user->phys->timestep_size_label) PetscCallCeed(ceed, CeedOperatorSetContextDouble(user->op_ifu… in IFunction_NS()
121 PetscCall(DMGlobalToLocalBegin(user->dm, Q, INSERT_VALUES, Q_loc)); in IFunction_NS()
122 PetscCall(DMGlobalToLocalBegin(user->dm, Q_dot, INSERT_VALUES, Q_dot_loc)); in IFunction_NS()
123 PetscCall(DMGlobalToLocalEnd(user->dm, Q, INSERT_VALUES, Q_loc)); in IFunction_NS()
124 PetscCall(DMGlobalToLocalEnd(user->dm, Q_dot, INSERT_VALUES, Q_dot_loc)); in IFunction_NS()
127 PetscCall(VecReadPetscToCeed(Q_loc, &q_mem_type, user->q_ceed)); in IFunction_NS()
128 PetscCall(VecReadPetscToCeed(Q_dot_loc, &q_dot_mem_type, user->q_dot_ceed)); in IFunction_NS()
129 PetscCall(VecPetscToCeed(G_loc, &g_mem_type, user->g_ceed)); in IFunction_NS()
134 …PetscCallCeed(user->ceed, CeedOperatorApply(user->op_ifunction, user->q_ceed, user->g_ceed, CEED_R… in IFunction_NS()
139 PetscCall(VecReadCeedToPetsc(user->q_ceed, q_mem_type, Q_loc)); in IFunction_NS()
140 PetscCall(VecReadCeedToPetsc(user->q_dot_ceed, q_dot_mem_type, Q_dot_loc)); in IFunction_NS()
141 PetscCall(VecCeedToPetsc(user->g_ceed, g_mem_type, G_loc)); in IFunction_NS()
145 PetscCall(DMLocalToGlobal(user->dm, G_loc, ADD_VALUES, G)); in IFunction_NS()
148 PetscCall(DMRestoreNamedLocalVector(user->dm, "ResidualLocal", &G_loc)); in IFunction_NS()
153 User user = *(User *)user_data; in FormIJacobian_NS() local
162 PetscCall(MatCeedSetContextReal(user->mat_ijacobian, "ijacobian time shift", shift)); in FormIJacobian_NS()
167 } else PetscCall(MatCeedAssembleCOO(user->mat_ijacobian, J)); in FormIJacobian_NS()
173 PetscCall(MatCeedAssembleCOO(user->mat_ijacobian, J_pre)); in FormIJacobian_NS()
178 PetscErrorCode WriteOutput(User user, Vec Q, PetscInt step_no, PetscScalar time) { in WriteOutput() argument
184 if (user->app_ctx->checkpoint_vtk) { in WriteOutput()
186 PetscCall(DMGetLocalVector(user->dm, &Q_loc)); in WriteOutput()
189 PetscCall(DMGlobalToLocal(user->dm, Q, INSERT_VALUES, Q_loc)); in WriteOutput()
192 …PetscCall(PetscSNPrintf(file_path, sizeof file_path, "%s/ns-%03" PetscInt_FMT ".vtu", user->app_ct… in WriteOutput()
197 if (user->dm_viz) { in WriteOutput()
202 PetscCall(DMGetGlobalVector(user->dm_viz, &Q_refined)); in WriteOutput()
203 PetscCall(DMGetLocalVector(user->dm_viz, &Q_refined_loc)); in WriteOutput()
206 PetscCall(MatInterpolate(user->interp_viz, Q, Q_refined)); in WriteOutput()
208 PetscCall(DMGlobalToLocal(user->dm_viz, Q_refined, INSERT_VALUES, Q_refined_loc)); in WriteOutput()
210 …efined, sizeof file_path_refined, "%s/nsrefined-%03" PetscInt_FMT ".vtu", user->app_ctx->output_di… in WriteOutput()
215 PetscCall(DMRestoreLocalVector(user->dm_viz, &Q_refined_loc)); in WriteOutput()
216 PetscCall(DMRestoreGlobalVector(user->dm_viz, &Q_refined)); in WriteOutput()
219 PetscCall(DMRestoreLocalVector(user->dm, &Q_loc)); in WriteOutput()
223 if (user->app_ctx->add_stepnum2bin) { in WriteOutput()
224 …intf(file_path, sizeof file_path, "%s/ns-solution-%" PetscInt_FMT ".bin", user->app_ctx->output_di… in WriteOutput()
226 …PetscCall(PetscSNPrintf(file_path, sizeof file_path, "%s/ns-solution.bin", user->app_ctx->output_d… in WriteOutput()
228 PetscCall(PetscViewerBinaryOpen(user->comm, file_path, FILE_MODE_WRITE, &viewer)); in WriteOutput()
233 time /= user->units->second; // Dimensionalize time back in WriteOutput()
242 User user = ctx; in TSMonitor_WallForce() local
244 PetscInt num_wall = user->app_ctx->wall_forces.num_wall, dim = 3; in TSMonitor_WallForce()
245 const PetscInt *walls = user->app_ctx->wall_forces.walls; in TSMonitor_WallForce()
246 PetscViewer viewer = user->app_ctx->wall_forces.viewer; in TSMonitor_WallForce()
247 PetscViewerFormat format = user->app_ctx->wall_forces.viewer_format; in TSMonitor_WallForce()
253 PetscCall(DMGetNamedLocalVector(user->dm, "ResidualLocal", &G_loc)); in TSMonitor_WallForce()
255 PetscCall(Surface_Forces_NS(user->dm, G_loc, num_wall, walls, reaction_force)); in TSMonitor_WallForce()
256 PetscCall(DMRestoreNamedLocalVector(user->dm, "ResidualLocal", &G_loc)); in TSMonitor_WallForce()
261 if (format == PETSC_VIEWER_ASCII_CSV && !user->app_ctx->wall_forces.header_written) { in TSMonitor_WallForce()
263 user->app_ctx->wall_forces.header_written = PETSC_TRUE; in TSMonitor_WallForce()
283 User user = ctx; in TSMonitor_NS() local
287 …if (user->app_ctx->checkpoint_interval <= 0 || step_no % user->app_ctx->checkpoint_interval != 0 || in TSMonitor_NS()
288 (user->app_ctx->cont_steps == step_no && step_no != 0)) { in TSMonitor_NS()
292 PetscCall(WriteOutput(user, Q, step_no, time)); in TSMonitor_NS()
297 PetscErrorCode TSSolve_NS(DM dm, User user, AppCtx app_ctx, Physics phys, ProblemData problem, Vec … in TSSolve_NS() argument
298 MPI_Comm comm = user->comm; in TSSolve_NS()
305 PetscCall(TSSetApplicationContext(*ts, user)); in TSSolve_NS()
308 if (user->op_ifunction) { in TSSolve_NS()
309 PetscCall(TSSetIFunction(*ts, NULL, IFunction_NS, &user)); in TSSolve_NS()
311 PetscCall(TSSetRHSFunction(*ts, NULL, RHS_NS, &user)); in TSSolve_NS()
313 if (user->mat_ijacobian) { in TSSolve_NS()
314 PetscCall(DMTSSetIJacobian(dm, FormIJacobian_NS, &user)); in TSSolve_NS()
317 PetscCheck(user->op_rhs_ctx, comm, PETSC_ERR_ARG_NULL, "Problem does not provide RHSFunction"); in TSSolve_NS()
320 PetscCall(TSSetRHSFunction(*ts, NULL, RHS_NS, &user)); in TSSolve_NS()
322 PetscCall(TSSetMaxTime(*ts, 500. * user->units->second)); in TSSolve_NS()
325 PetscCall(TSSetTimeStep(*ts, 1.e-2 * user->units->second)); in TSSolve_NS()
327 PetscCall(TSAdaptSetStepLimits(adapt, 1.e-12 * user->units->second, 1.e2 * user->units->second)); in TSSolve_NS()
329 if (user->mat_ijacobian) { in TSSolve_NS()
337 … PetscCall(CreateSolveOperatorsFromMatCeed(ksp, user->mat_ijacobian, PETSC_FALSE, &Amat, &Pmat)); in TSSolve_NS()
338 PetscCall(TSSetIJacobian(*ts, user->mat_ijacobian, Pmat, NULL, NULL)); in TSSolve_NS()
343 user->time_bc_set = -1.0; // require all BCs be updated in TSSolve_NS()
355 PetscCall(TSSetTime(*ts, app_ctx->cont_time * user->units->second)); in TSSolve_NS()
359 PetscCall(TSMonitorSet(*ts, TSMonitor_NS, user, NULL)); in TSSolve_NS()
362 PetscCall(TSMonitorSet(*ts, TSMonitor_WallForce, user, NULL)); in TSSolve_NS()
365 PetscCall(TSMonitorSet(*ts, TSMonitor_TurbulenceStatistics, user, NULL)); in TSSolve_NS()
366 CeedScalar previous_time = app_ctx->cont_time * user->units->second; in TSSolve_NS()
367 PetscCallCeed(user->ceed, in TSSolve_NS()
368 …CeedOperatorSetContextDouble(user->spanstats.op_stats_collect_ctx->op, user->spanstats.previous_ti… in TSSolve_NS()
370 …if (app_ctx->diff_filter_monitor) PetscCall(TSMonitorSet(*ts, TSMonitor_DifferentialFilter, user, … in TSSolve_NS()
372 if (app_ctx->test_type == TESTTYPE_NONE) PetscCall(PrintRunInfo(user, user->phys, problem, *ts)); in TSSolve_NS()
410 if (user->app_ctx->checkpoint_interval > 0 || user->app_ctx->checkpoint_interval == -1) { in TSSolve_NS()
411 PetscCall(WriteOutput(user, *Q, step_no, final_time)); in TSSolve_NS()