PICurv 0.1.0
A Parallel Particle-In-Cell Solver for Curvilinear LES
Loading...
Searching...
No Matches
Functions
test_support.c File Reference

Shared C test fixtures, assertions, and PETSc helper utilities. More...

#include "test_support.h"
#include "grid.h"
#include "io.h"
#include "setup.h"
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
Include dependency graph for test_support.c:

Go to the source code of this file.

Functions

static PetscErrorCode DestroyVecIfSet (Vec *vec)
 Destroys a PETSc vector only when the handle is non-null.
 
static PetscErrorCode DestroyDMIfSet (DM *dm)
 Destroys a PETSc DM only when the handle is non-null.
 
static PetscErrorCode DestroyMatIfSet (Mat *mat)
 Destroys a PETSc matrix only when the handle is non-null.
 
static PetscErrorCode DestroyKSPIfSet (KSP *ksp)
 Destroys a PETSc KSP only when the handle is non-null.
 
static PetscErrorCode DestroyNullSpaceIfSet (MatNullSpace *nullsp)
 Destroys a PETSc nullspace only when the handle is non-null.
 
static PetscErrorCode DestroyRandomIfSet (PetscRandom *rand_ctx)
 Destroys a PETSc random generator only when the handle is non-null.
 
static PetscErrorCode RegisterSwarmFieldForTests (DM swarm, const char *field_name, PetscInt field_dim, PetscDataType dtype)
 Registers one DMSwarm field used by the C test fixtures.
 
static PetscErrorCode CreateZeroedGlobalVector (DM dm, Vec *vec)
 Allocates and zeroes a global vector from the provided DM.
 
static PetscErrorCode CreateZeroedLocalVector (DM dm, Vec *vec)
 Allocates and zeroes a local vector from the provided DM.
 
static PetscErrorCode CreateZeroedDuplicate (Vec src, Vec *vec)
 Duplicates and zeroes a vector.
 
PetscErrorCode PicurvRunTests (const char *suite_name, const PicurvTestCase *cases, size_t case_count)
 Runs a named C test suite and prints pass/fail progress markers.
 
PetscErrorCode PicurvEnsureDir (const char *path)
 Ensures a directory exists for test output.
 
PetscErrorCode PicurvMakeTempDir (char *path, size_t path_len)
 Creates a unique temporary directory for one test case.
 
PetscErrorCode PicurvRemoveTempDir (const char *path)
 Recursively removes a temporary directory created by PicurvMakeTempDir.
 
static PetscErrorCode WriteTextFileForTests (const char *path, const char *contents)
 Writes one small temporary text file used by the richer runtime fixtures.
 
static PetscErrorCode PrepareTinyRuntimeConfig (const char *bcs_contents, PetscBool enable_particles, char *tmpdir, size_t tmpdir_len, char *control_path, size_t control_path_len)
 Creates a tiny control-file bundle used by richer runtime fixtures built through the setup path.
 
PetscErrorCode PicurvCreateMinimalContextsWithPeriodicity (SimCtx **simCtx_out, UserCtx **user_out, PetscInt mx, PetscInt my, PetscInt mz, PetscBool x_periodic, PetscBool y_periodic, PetscBool z_periodic)
 Builds minimal SimCtx and UserCtx fixtures for C unit tests with configurable periodicity.
 
PetscErrorCode PicurvCreateMinimalContexts (SimCtx **simCtx_out, UserCtx **user_out, PetscInt mx, PetscInt my, PetscInt mz)
 Builds minimal SimCtx and UserCtx fixtures for C unit tests.
 
PetscErrorCode PicurvPopulateUniformCellCenters (UserCtx *user)
 Populates cell center coordinates for a uniform grid on [0,1]^3.
 
PetscErrorCode PicurvPopulateIdentityMetrics (UserCtx *user)
 Populates identity metric vectors on the minimal grid fixture.
 
PetscErrorCode PicurvCreateSwarmPair (UserCtx *user, PetscInt nlocal, const char *post_field_name)
 Creates matched solver and post-processing swarms for tests.
 
PetscErrorCode PicurvBuildTinyRuntimeContext (const char *bcs_contents, PetscBool enable_particles, SimCtx **simCtx_out, UserCtx **user_out, char *tmpdir, size_t tmpdir_len)
 Builds a tiny runtime context through the real setup path for behavior-level tests.
 
PetscErrorCode PicurvDestroyRuntimeContext (SimCtx **simCtx_ptr)
 Finalizes and frees a runtime context built by PicurvBuildTinyRuntimeContext.
 
PetscErrorCode PicurvDestroyMinimalContexts (SimCtx **simCtx_ptr, UserCtx **user_ptr)
 Destroys minimal SimCtx/UserCtx fixtures and all owned PETSc objects.
 
PetscErrorCode PicurvAssertRealNear (PetscReal expected, PetscReal actual, PetscReal tol, const char *context)
 Asserts that two real values agree within tolerance.
 
PetscErrorCode PicurvAssertIntEqual (PetscInt expected, PetscInt actual, const char *context)
 Asserts that two integer values are equal.
 
PetscErrorCode PicurvAssertBool (PetscBool value, const char *context)
 Asserts that one boolean condition is true.
 
PetscErrorCode PicurvAssertFileExists (const char *path, const char *context)
 Asserts that a filesystem path exists as a readable file.
 
PetscErrorCode PicurvAssertVecConstant (Vec vec, PetscScalar expected, PetscReal tol, const char *context)
 Asserts that a PETSc vector is spatially constant within tolerance.
 

Detailed Description

Shared C test fixtures, assertions, and PETSc helper utilities.

Definition in file test_support.c.

Function Documentation

◆ DestroyVecIfSet()

static PetscErrorCode DestroyVecIfSet ( Vec *  vec)
static

Destroys a PETSc vector only when the handle is non-null.

Definition at line 23 of file test_support.c.

24{
25 PetscFunctionBeginUser;
26 if (vec && *vec) {
27 PetscCall(VecDestroy(vec));
28 }
29 PetscFunctionReturn(0);
30}
Here is the caller graph for this function:

◆ DestroyDMIfSet()

static PetscErrorCode DestroyDMIfSet ( DM *  dm)
static

Destroys a PETSc DM only when the handle is non-null.

Definition at line 35 of file test_support.c.

36{
37 PetscFunctionBeginUser;
38 if (dm && *dm) {
39 PetscCall(DMDestroy(dm));
40 }
41 PetscFunctionReturn(0);
42}
Here is the caller graph for this function:

◆ DestroyMatIfSet()

static PetscErrorCode DestroyMatIfSet ( Mat *  mat)
static

Destroys a PETSc matrix only when the handle is non-null.

Definition at line 47 of file test_support.c.

48{
49 PetscFunctionBeginUser;
50 if (mat && *mat) {
51 PetscCall(MatDestroy(mat));
52 }
53 PetscFunctionReturn(0);
54}
Here is the caller graph for this function:

◆ DestroyKSPIfSet()

static PetscErrorCode DestroyKSPIfSet ( KSP *  ksp)
static

Destroys a PETSc KSP only when the handle is non-null.

Definition at line 59 of file test_support.c.

60{
61 PetscFunctionBeginUser;
62 if (ksp && *ksp) {
63 PetscCall(KSPDestroy(ksp));
64 }
65 PetscFunctionReturn(0);
66}
Here is the caller graph for this function:

◆ DestroyNullSpaceIfSet()

static PetscErrorCode DestroyNullSpaceIfSet ( MatNullSpace *  nullsp)
static

Destroys a PETSc nullspace only when the handle is non-null.

Definition at line 71 of file test_support.c.

72{
73 PetscFunctionBeginUser;
74 if (nullsp && *nullsp) {
75 PetscCall(MatNullSpaceDestroy(nullsp));
76 }
77 PetscFunctionReturn(0);
78}
Here is the caller graph for this function:

◆ DestroyRandomIfSet()

static PetscErrorCode DestroyRandomIfSet ( PetscRandom *  rand_ctx)
static

Destroys a PETSc random generator only when the handle is non-null.

Definition at line 83 of file test_support.c.

84{
85 PetscFunctionBeginUser;
86 if (rand_ctx && *rand_ctx) {
87 PetscCall(PetscRandomDestroy(rand_ctx));
88 }
89 PetscFunctionReturn(0);
90}
Here is the caller graph for this function:

◆ RegisterSwarmFieldForTests()

static PetscErrorCode RegisterSwarmFieldForTests ( DM  swarm,
const char *  field_name,
PetscInt  field_dim,
PetscDataType  dtype 
)
static

Registers one DMSwarm field used by the C test fixtures.

Definition at line 95 of file test_support.c.

96{
97 PetscFunctionBeginUser;
98 PetscCall(DMSwarmRegisterPetscDatatypeField(swarm, field_name, field_dim, dtype));
99 PetscFunctionReturn(0);
100}
Here is the caller graph for this function:

◆ CreateZeroedGlobalVector()

static PetscErrorCode CreateZeroedGlobalVector ( DM  dm,
Vec *  vec 
)
static

Allocates and zeroes a global vector from the provided DM.

Definition at line 105 of file test_support.c.

106{
107 PetscFunctionBeginUser;
108 PetscCall(DMCreateGlobalVector(dm, vec));
109 PetscCall(VecSet(*vec, 0.0));
110 PetscFunctionReturn(0);
111}
Here is the caller graph for this function:

◆ CreateZeroedLocalVector()

static PetscErrorCode CreateZeroedLocalVector ( DM  dm,
Vec *  vec 
)
static

Allocates and zeroes a local vector from the provided DM.

Definition at line 116 of file test_support.c.

117{
118 PetscFunctionBeginUser;
119 PetscCall(DMCreateLocalVector(dm, vec));
120 PetscCall(VecSet(*vec, 0.0));
121 PetscFunctionReturn(0);
122}
Here is the caller graph for this function:

◆ CreateZeroedDuplicate()

static PetscErrorCode CreateZeroedDuplicate ( Vec  src,
Vec *  vec 
)
static

Duplicates and zeroes a vector.

Definition at line 127 of file test_support.c.

128{
129 PetscFunctionBeginUser;
130 PetscCall(VecDuplicate(src, vec));
131 PetscCall(VecSet(*vec, 0.0));
132 PetscFunctionReturn(0);
133}
Here is the caller graph for this function:

◆ PicurvRunTests()

PetscErrorCode PicurvRunTests ( const char *  suite_name,
const PicurvTestCase cases,
size_t  case_count 
)

Runs a named C test suite and prints pass/fail progress markers.

Executes a table of unit tests and reports aggregated pass/fail status.

Definition at line 138 of file test_support.c.

139{
140 PetscFunctionBeginUser;
141
142 PetscCall(PetscPrintf(PETSC_COMM_WORLD, "==> Running %s (%zu tests)\n", suite_name, case_count));
143 for (size_t i = 0; i < case_count; ++i) {
144 PetscCall(PetscPrintf(PETSC_COMM_WORLD, " -> %s\n", cases[i].name));
145 PetscCall(cases[i].fn());
146 PetscCall(PetscPrintf(PETSC_COMM_WORLD, " [PASS] %s\n", cases[i].name));
147 }
148
149 PetscCall(PetscPrintf(PETSC_COMM_WORLD, "==> %s complete\n", suite_name));
150 PetscFunctionReturn(0);
151}
Here is the caller graph for this function:

◆ PicurvEnsureDir()

PetscErrorCode PicurvEnsureDir ( const char *  path)

Ensures a directory exists for test output.

Ensures a directory exists, creating it if required.

Definition at line 156 of file test_support.c.

157{
158 PetscFunctionBeginUser;
159 if (mkdir(path, 0777) != 0 && errno != EEXIST) {
160 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_FILE_OPEN, "Failed to create directory '%s': %s", path, strerror(errno));
161 }
162 PetscFunctionReturn(0);
163}
Here is the caller graph for this function:

◆ PicurvMakeTempDir()

PetscErrorCode PicurvMakeTempDir ( char *  path,
size_t  path_len 
)

Creates a unique temporary directory for one test case.

Creates a unique temporary directory path and materializes the directory.

Definition at line 168 of file test_support.c.

169{
170 PetscFunctionBeginUser;
171 if (!path || path_len < 24) {
172 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_NULL, "Temp directory buffer is missing or too small.");
173 }
174
175 PetscCall(PetscSNPrintf(path, path_len, "/tmp/picurv-test-XXXXXX"));
176 if (!mkdtemp(path)) {
177 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_FILE_OPEN, "mkdtemp failed for '%s': %s", path, strerror(errno));
178 }
179 PetscFunctionReturn(0);
180}
Here is the caller graph for this function:

◆ PicurvRemoveTempDir()

PetscErrorCode PicurvRemoveTempDir ( const char *  path)

Recursively removes a temporary directory created by PicurvMakeTempDir.

Definition at line 185 of file test_support.c.

186{
187 char cmd[512];
188
189 PetscFunctionBeginUser;
190 if (!path || path[0] == '\0') PetscFunctionReturn(0);
191 /* Safety: only remove paths under /tmp/picurv-test- */
192 if (strncmp(path, "/tmp/picurv-test-", 17) != 0) {
193 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG,
194 "Refusing to remove path outside /tmp/picurv-test-*: '%s'", path);
195 }
196 PetscCall(PetscSNPrintf(cmd, sizeof(cmd), "rm -rf '%s'", path));
197 if (system(cmd) != 0) {
198 PetscCall(PetscPrintf(PETSC_COMM_SELF, "Warning: failed to remove temp dir '%s'\n", path));
199 }
200 PetscFunctionReturn(0);
201}
Here is the caller graph for this function:

◆ WriteTextFileForTests()

static PetscErrorCode WriteTextFileForTests ( const char *  path,
const char *  contents 
)
static

Writes one small temporary text file used by the richer runtime fixtures.

Definition at line 206 of file test_support.c.

207{
208 FILE *file = NULL;
209
210 PetscFunctionBeginUser;
211 file = fopen(path, "w");
212 PetscCheck(file != NULL, PETSC_COMM_SELF, PETSC_ERR_FILE_OPEN, "Failed to open '%s' for writing.", path);
213 fputs(contents, file);
214 fclose(file);
215 PetscFunctionReturn(0);
216}
Here is the caller graph for this function:

◆ PrepareTinyRuntimeConfig()

static PetscErrorCode PrepareTinyRuntimeConfig ( const char *  bcs_contents,
PetscBool  enable_particles,
char *  tmpdir,
size_t  tmpdir_len,
char *  control_path,
size_t  control_path_len 
)
static

Creates a tiny control-file bundle used by richer runtime fixtures built through the setup path.

Definition at line 221 of file test_support.c.

227{
228 char bcs_path[PETSC_MAX_PATH_LEN];
229 char post_path[PETSC_MAX_PATH_LEN];
230 char output_dir[PETSC_MAX_PATH_LEN];
231 char log_dir[PETSC_MAX_PATH_LEN];
232 char control_buffer[8192];
233 const char *particle_block = NULL;
234 const char *default_bcs =
235 "-Xi WALL noslip\n"
236 "+Xi WALL noslip\n"
237 "-Eta WALL noslip\n"
238 "+Eta WALL noslip\n"
239 "-Zeta INLET constant_velocity vx=0.0 vy=0.0 vz=1.5\n"
240 "+Zeta OUTLET conservation\n";
241
242 PetscFunctionBeginUser;
243 PetscCall(PicurvMakeTempDir(tmpdir, tmpdir_len));
244 PetscCall(PetscSNPrintf(bcs_path, sizeof(bcs_path), "%s/bcs.run", tmpdir));
245 PetscCall(PetscSNPrintf(post_path, sizeof(post_path), "%s/post.run", tmpdir));
246 PetscCall(PetscSNPrintf(output_dir, sizeof(output_dir), "%s/results", tmpdir));
247 PetscCall(PetscSNPrintf(log_dir, sizeof(log_dir), "%s/logs", tmpdir));
248 PetscCall(PetscSNPrintf(control_path, control_path_len, "%s/test.control", tmpdir));
249
250 PetscCall(WriteTextFileForTests(bcs_path, bcs_contents ? bcs_contents : default_bcs));
251 PetscCall(WriteTextFileForTests(
252 post_path,
253 "startTime = 0\n"
254 "endTime = 1\n"
255 "timeStep = 1\n"
256 "output_particles = false\n"));
257
258 if (enable_particles) {
259 particle_block =
260 "-numParticles 8\n"
261 "-pinit 2\n"
262 "-psrc_x 0.5\n"
263 "-psrc_y 0.5\n"
264 "-psrc_z 0.5\n"
265 "-particle_restart_mode init\n";
266 } else {
267 particle_block =
268 "-numParticles 0\n"
269 "-pinit 2\n";
270 }
271
272 PetscCall(PetscSNPrintf(
273 control_buffer,
274 sizeof(control_buffer),
275 "-start_step 0\n"
276 "-totalsteps 2\n"
277 "-ren 100.0\n"
278 "-dt 0.001\n"
279 "-finit 1\n"
280 "-ucont_x 0.0\n"
281 "-ucont_y 0.0\n"
282 "-ucont_z 1.5\n"
283 "-bcs_files %s\n"
284 "-profiling_timestep_mode off\n"
285 "-profiling_final_summary true\n"
286 "-postprocessing_config_file %s\n"
287 "-grid\n"
288 "-im 6\n"
289 "-jm 6\n"
290 "-km 6\n"
291 "-xMins 0.0\n"
292 "-xMaxs 1.0\n"
293 "-yMins 0.0\n"
294 "-yMaxs 1.0\n"
295 "-zMins 0.0\n"
296 "-zMaxs 1.0\n"
297 "-rxs 1.0\n"
298 "-rys 1.0\n"
299 "-rzs 1.0\n"
300 "-cgrids 0\n"
301 "-nblk 1\n"
302 "-euler_field_source solve\n"
303 "-mom_solver_type EXPLICIT_RK\n"
304 "-mg_level 1\n"
305 "-poisson 0\n"
306 "-tio 0\n"
307 "-particle_console_output_freq 0\n"
308 "-logfreq 1\n"
309 "-output_dir %s\n"
310 "-restart_dir %s\n"
311 "-log_dir %s\n"
312 "%s",
313 bcs_path,
314 post_path,
315 output_dir,
316 output_dir,
317 log_dir,
318 particle_block));
319 PetscCall(WriteTextFileForTests(control_path, control_buffer));
320 PetscFunctionReturn(0);
321}
PetscErrorCode PicurvMakeTempDir(char *path, size_t path_len)
Creates a unique temporary directory for one test case.
static PetscErrorCode WriteTextFileForTests(const char *path, const char *contents)
Writes one small temporary text file used by the richer runtime fixtures.
Here is the call graph for this function:
Here is the caller graph for this function:

◆ PicurvCreateMinimalContextsWithPeriodicity()

PetscErrorCode PicurvCreateMinimalContextsWithPeriodicity ( SimCtx **  simCtx_out,
UserCtx **  user_out,
PetscInt  mx,
PetscInt  my,
PetscInt  mz,
PetscBool  x_periodic,
PetscBool  y_periodic,
PetscBool  z_periodic 
)

Builds minimal SimCtx and UserCtx fixtures for C unit tests with configurable periodicity.

Creates minimal PETSc/solver contexts with optional periodic DM boundaries.

Definition at line 325 of file test_support.c.

333{
334 SimCtx *simCtx = NULL;
335 UserCtx *user = NULL;
336 BoundingBox *boxes = NULL;
337 PetscInt da_mx = mx + 1;
338 PetscInt da_my = my + 1;
339 PetscInt da_mz = mz + 1;
340 DMBoundaryType x_boundary = x_periodic ? DM_BOUNDARY_PERIODIC : DM_BOUNDARY_NONE;
341 DMBoundaryType y_boundary = y_periodic ? DM_BOUNDARY_PERIODIC : DM_BOUNDARY_NONE;
342 DMBoundaryType z_boundary = z_periodic ? DM_BOUNDARY_PERIODIC : DM_BOUNDARY_NONE;
343 PetscInt stencil_width = (x_periodic || y_periodic || z_periodic) ? 3 : 1;
344
345 PetscFunctionBeginUser;
346 if (!simCtx_out || !user_out) {
347 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_NULL, "Output pointers cannot be NULL.");
348 }
349
350 PetscCall(PetscCalloc1(1, &simCtx));
351 PetscCall(PetscCalloc1(1, &user));
352
353 PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &simCtx->rank));
354 PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &simCtx->size));
355 simCtx->block_number = 1;
356 simCtx->dt = 0.1;
357 simCtx->tiout = 2;
358 simCtx->forceScalingFactor = 1.0;
359 simCtx->LoggingFrequency = 1;
360 simCtx->exec_mode = EXEC_MODE_SOLVER;
362 simCtx->poisson = 0;
363 simCtx->ren = 1.0;
364 simCtx->schmidt_number = 1.0;
365 simCtx->StartStep = 0;
366 simCtx->StepsToRun = 1;
367 simCtx->step = 1;
368 simCtx->np = 0;
369 simCtx->i_periodic = x_periodic ? 1 : 0;
370 simCtx->j_periodic = y_periodic ? 1 : 0;
371 simCtx->k_periodic = z_periodic ? 1 : 0;
372 PetscCall(PetscStrncpy(simCtx->euler_subdir, "euler", sizeof(simCtx->euler_subdir)));
373 PetscCall(PetscStrncpy(simCtx->particle_subdir, "particles", sizeof(simCtx->particle_subdir)));
374 PetscCall(PetscStrncpy(simCtx->output_dir, "/tmp", sizeof(simCtx->output_dir)));
375 PetscCall(PetscStrncpy(simCtx->restart_dir, "/tmp", sizeof(simCtx->restart_dir)));
376 simCtx->mglevels = 1;
377 simCtx->usermg.mglevels = 1;
378 PetscCall(PetscCalloc1(1, &simCtx->usermg.mgctx));
379 simCtx->usermg.mgctx[0].thislevel = 0;
380 PetscCall(PetscCalloc1(simCtx->size, &simCtx->bboxlist));
381 PetscCall(PetscRandomCreate(PETSC_COMM_WORLD, &simCtx->BrownianMotionRNG));
382 PetscCall(PetscRandomSetType(simCtx->BrownianMotionRNG, PETSCRAND48));
383 PetscCall(PetscRandomSetInterval(simCtx->BrownianMotionRNG, 0.0, 1.0));
384 PetscCall(PetscRandomSetSeed(simCtx->BrownianMotionRNG, 12345));
385 PetscCall(PetscRandomSeed(simCtx->BrownianMotionRNG));
386
387 user->simCtx = simCtx;
388 user->_this = 0;
389 user->thislevel = 0;
390 user->mglevels = 1;
391 user->IM = mx;
392 user->JM = my;
393 user->KM = mz;
394 user->Min_X = 0.0;
395 user->Min_Y = 0.0;
396 user->Min_Z = 0.0;
397 user->Max_X = 1.0;
398 user->Max_Y = 1.0;
399 user->Max_Z = 1.0;
400 user->rx = 1.0;
401 user->ry = 1.0;
402 user->rz = 1.0;
403 user->bbox.min_coords.x = 0.0;
404 user->bbox.min_coords.y = 0.0;
405 user->bbox.min_coords.z = 0.0;
406 user->bbox.max_coords.x = 1.0;
407 user->bbox.max_coords.y = 1.0;
408 user->bbox.max_coords.z = 1.0;
409 simCtx->usermg.mgctx[0].user = user;
410 for (PetscMPIInt rank_idx = 0; rank_idx < simCtx->size; ++rank_idx) {
411 simCtx->bboxlist[rank_idx].min_coords.x = 0.0;
412 simCtx->bboxlist[rank_idx].min_coords.y = 0.0;
413 simCtx->bboxlist[rank_idx].min_coords.z = 0.0;
414 simCtx->bboxlist[rank_idx].max_coords.x = 1.0;
415 simCtx->bboxlist[rank_idx].max_coords.y = 1.0;
416 simCtx->bboxlist[rank_idx].max_coords.z = 1.0;
417 }
418
419 PetscCall(DMDACreate3d(PETSC_COMM_WORLD,
420 x_boundary, y_boundary, z_boundary,
421 DMDA_STENCIL_BOX,
422 da_mx, da_my, da_mz,
423 PETSC_DECIDE, PETSC_DECIDE, PETSC_DECIDE,
424 1, stencil_width,
425 NULL, NULL, NULL,
426 &user->da));
427 PetscCall(DMSetUp(user->da));
428 PetscCall(DMGetCoordinateDM(user->da, &user->fda));
429 PetscCall(PetscObjectReference((PetscObject)user->fda));
430 PetscCall(DMDASetUniformCoordinates(user->da, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0));
431 PetscCall(DMDAGetLocalInfo(user->da, &user->info));
432 PetscCall(ComputeLocalBoundingBox(user, &user->bbox));
433 PetscCall(GatherAllBoundingBoxes(user, &boxes));
434 PetscCall(BroadcastAllBoundingBoxes(user, &boxes));
435 for (PetscMPIInt rank_idx = 0; rank_idx < simCtx->size; ++rank_idx) {
436 simCtx->bboxlist[rank_idx] = boxes[rank_idx];
437 }
438 free(boxes);
439
440 PetscCall(CreateZeroedGlobalVector(user->da, &user->P));
441 PetscCall(CreateZeroedLocalVector(user->da, &user->lP));
442 PetscCall(CreateZeroedGlobalVector(user->da, &user->Phi));
443 PetscCall(CreateZeroedLocalVector(user->da, &user->lPhi));
444 PetscCall(CreateZeroedGlobalVector(user->da, &user->Nvert));
445 PetscCall(CreateZeroedLocalVector(user->da, &user->lNvert));
446 PetscCall(CreateZeroedGlobalVector(user->da, &user->ParticleCount));
447 PetscCall(CreateZeroedLocalVector(user->da, &user->lParticleCount));
448 PetscCall(CreateZeroedGlobalVector(user->da, &user->Psi));
449 PetscCall(CreateZeroedLocalVector(user->da, &user->lPsi));
450 PetscCall(CreateZeroedGlobalVector(user->da, &user->Qcrit));
451 PetscCall(CreateZeroedGlobalVector(user->da, &user->P_nodal));
452 PetscCall(CreateZeroedGlobalVector(user->da, &user->Psi_nodal));
453 PetscCall(CreateZeroedGlobalVector(user->da, &user->Aj));
454 PetscCall(CreateZeroedLocalVector(user->da, &user->lAj));
455 PetscCall(CreateZeroedGlobalVector(user->da, &user->Diffusivity));
456 PetscCall(CreateZeroedLocalVector(user->da, &user->lDiffusivity));
457 PetscCall(CreateZeroedGlobalVector(user->da, &user->B));
458 PetscCall(CreateZeroedGlobalVector(user->da, &user->R));
459
460 PetscCall(CreateZeroedGlobalVector(user->fda, &user->Ucat));
461 PetscCall(CreateZeroedLocalVector(user->fda, &user->lUcat));
462 PetscCall(CreateZeroedGlobalVector(user->fda, &user->Ucont));
463 PetscCall(CreateZeroedLocalVector(user->fda, &user->lUcont));
464 PetscCall(CreateZeroedGlobalVector(user->fda, &user->Csi));
465 PetscCall(CreateZeroedLocalVector(user->fda, &user->lCsi));
466 PetscCall(CreateZeroedDuplicate(user->Csi, &user->Eta));
467 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lEta));
468 PetscCall(CreateZeroedDuplicate(user->Csi, &user->Zet));
469 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lZet));
470 PetscCall(CreateZeroedGlobalVector(user->fda, &user->Cent));
471 PetscCall(CreateZeroedLocalVector(user->fda, &user->lCent));
472 PetscCall(CreateZeroedGlobalVector(user->fda, &user->GridSpace));
473 PetscCall(CreateZeroedLocalVector(user->fda, &user->lGridSpace));
474 PetscCall(CreateZeroedLocalVector(user->fda, &user->Centx));
475 PetscCall(CreateZeroedLocalVector(user->fda, &user->Centy));
476 PetscCall(CreateZeroedLocalVector(user->fda, &user->Centz));
477 PetscCall(CreateZeroedGlobalVector(user->fda, &user->Ucat_nodal));
478 PetscCall(CreateZeroedGlobalVector(user->fda, &user->DiffusivityGradient));
479 PetscCall(CreateZeroedLocalVector(user->fda, &user->lDiffusivityGradient));
480 PetscCall(CreateZeroedGlobalVector(user->fda, &user->Rhs));
481 PetscCall(CreateZeroedGlobalVector(user->fda, &user->dUcont));
482 PetscCall(CreateZeroedGlobalVector(user->fda, &user->pUcont));
483 PetscCall(CreateZeroedGlobalVector(user->fda, &user->Bcs.Ubcs));
484 PetscCall(CreateZeroedGlobalVector(user->fda, &user->Bcs.Uch));
485
486 PetscCall(CreateZeroedDuplicate(user->Ucont, &user->Ucont_o));
487 PetscCall(CreateZeroedDuplicate(user->lUcont, &user->lUcont_o));
488 PetscCall(CreateZeroedDuplicate(user->Ucont, &user->Ucont_rm1));
489 PetscCall(CreateZeroedDuplicate(user->lUcont, &user->lUcont_rm1));
490 PetscCall(CreateZeroedDuplicate(user->Ucat, &user->Ucat_o));
491 PetscCall(CreateZeroedDuplicate(user->P, &user->P_o));
492 PetscCall(CreateZeroedDuplicate(user->Nvert, &user->Nvert_o));
493 PetscCall(CreateZeroedLocalVector(user->da, &user->lNvert_o));
494 PetscCall(CreateZeroedDuplicate(user->Csi, &user->ICsi));
495 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lICsi));
496 PetscCall(CreateZeroedDuplicate(user->Csi, &user->IEta));
497 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lIEta));
498 PetscCall(CreateZeroedDuplicate(user->Csi, &user->IZet));
499 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lIZet));
500 PetscCall(CreateZeroedDuplicate(user->Csi, &user->JCsi));
501 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lJCsi));
502 PetscCall(CreateZeroedDuplicate(user->Csi, &user->JEta));
503 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lJEta));
504 PetscCall(CreateZeroedDuplicate(user->Csi, &user->JZet));
505 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lJZet));
506 PetscCall(CreateZeroedDuplicate(user->Csi, &user->KCsi));
507 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lKCsi));
508 PetscCall(CreateZeroedDuplicate(user->Csi, &user->KEta));
509 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lKEta));
510 PetscCall(CreateZeroedDuplicate(user->Csi, &user->KZet));
511 PetscCall(CreateZeroedDuplicate(user->lCsi, &user->lKZet));
512 PetscCall(CreateZeroedDuplicate(user->Aj, &user->IAj));
513 PetscCall(CreateZeroedDuplicate(user->lAj, &user->lIAj));
514 PetscCall(CreateZeroedDuplicate(user->Aj, &user->JAj));
515 PetscCall(CreateZeroedDuplicate(user->lAj, &user->lJAj));
516 PetscCall(CreateZeroedDuplicate(user->Aj, &user->KAj));
517 PetscCall(CreateZeroedDuplicate(user->lAj, &user->lKAj));
518
519 PetscCall(PicurvPopulateUniformCellCenters(user));
520 PetscCall(PicurvPopulateIdentityMetrics(user));
521
522 *simCtx_out = simCtx;
523 *user_out = user;
524 PetscFunctionReturn(0);
525}
PetscErrorCode BroadcastAllBoundingBoxes(UserCtx *user, BoundingBox **bboxlist)
Broadcasts the bounding box information collected on rank 0 to all other ranks.
Definition grid.c:883
PetscErrorCode ComputeLocalBoundingBox(UserCtx *user, BoundingBox *localBBox)
Computes the local bounding box of the grid on the current process.
Definition grid.c:679
PetscErrorCode GatherAllBoundingBoxes(UserCtx *user, BoundingBox **allBBoxes)
Gathers local bounding boxes from all MPI processes to rank 0.
Definition grid.c:821
static PetscErrorCode CreateZeroedLocalVector(DM dm, Vec *vec)
Allocates and zeroes a local vector from the provided DM.
static PetscErrorCode CreateZeroedDuplicate(Vec src, Vec *vec)
Duplicates and zeroes a vector.
PetscErrorCode PicurvPopulateUniformCellCenters(UserCtx *user)
Populates cell center coordinates for a uniform grid on [0,1]^3.
static PetscErrorCode CreateZeroedGlobalVector(DM dm, Vec *vec)
Allocates and zeroes a global vector from the provided DM.
PetscErrorCode PicurvPopulateIdentityMetrics(UserCtx *user)
Populates identity metric vectors on the minimal grid fixture.
Vec lDiffusivityGradient
Definition variables.h:841
Vec lCent
Definition variables.h:858
Vec GridSpace
Definition variables.h:858
Vec P_nodal
Definition variables.h:887
Vec JCsi
Definition variables.h:861
Vec KAj
Definition variables.h:862
UserCtx * user
Definition variables.h:528
Vec JEta
Definition variables.h:861
Vec Zet
Definition variables.h:858
Vec Rhs
Definition variables.h:845
PetscReal schmidt_number
Definition variables.h:709
PetscMPIInt rank
Definition variables.h:646
PetscInt mglevels
Definition variables.h:874
PetscInt block_number
Definition variables.h:712
Vec lIEta
Definition variables.h:860
Vec lIZet
Definition variables.h:860
Vec lNvert
Definition variables.h:837
Vec Phi
Definition variables.h:837
char euler_subdir[PETSC_MAX_PATH_LEN]
Definition variables.h:666
PetscReal forceScalingFactor
Definition variables.h:723
SimCtx * simCtx
Back-pointer to the master simulation context.
Definition variables.h:814
Vec IZet
Definition variables.h:860
Vec Centz
Definition variables.h:859
Vec IEta
Definition variables.h:860
PetscReal Min_X
Definition variables.h:821
PetscInt KM
Definition variables.h:820
PetscInt tiout
Definition variables.h:655
Vec lZet
Definition variables.h:858
UserMG usermg
Definition variables.h:764
Vec Csi
Definition variables.h:858
PetscReal ren
Definition variables.h:691
Vec lUcont_rm1
Definition variables.h:845
Vec lIAj
Definition variables.h:860
Cmpnts max_coords
Maximum x, y, z coordinates of the bounding box.
Definition variables.h:156
PetscInt _this
Definition variables.h:824
Vec lKEta
Definition variables.h:862
char output_dir[PETSC_MAX_PATH_LEN]
Definition variables.h:665
PetscReal dt
Definition variables.h:658
PetscReal ry
Definition variables.h:825
PetscInt StepsToRun
Definition variables.h:654
PetscInt k_periodic
Definition variables.h:713
Vec Ucat_nodal
Definition variables.h:888
Vec lPsi
Definition variables.h:883
PetscInt np
Definition variables.h:739
PetscReal Max_Y
Definition variables.h:821
PetscInt thislevel
Definition variables.h:529
Vec DiffusivityGradient
Definition variables.h:841
Vec lJCsi
Definition variables.h:861
Vec Ucont
Definition variables.h:837
PetscInt StartStep
Definition variables.h:653
Cmpnts min_coords
Minimum x, y, z coordinates of the bounding box.
Definition variables.h:155
Vec Ubcs
Physical Cartesian velocity at boundary faces. Full 3D array but only boundary-face entries are meani...
Definition variables.h:121
@ MOMENTUM_SOLVER_EXPLICIT_RK
Definition variables.h:503
Vec Qcrit
Definition variables.h:889
PetscScalar x
Definition variables.h:101
Vec JZet
Definition variables.h:861
Vec Centx
Definition variables.h:859
BCS Bcs
Definition variables.h:832
Vec lPhi
Definition variables.h:837
Vec lParticleCount
Definition variables.h:882
Vec lUcont_o
Definition variables.h:844
Vec Ucat_o
Definition variables.h:844
PetscInt poisson
Definition variables.h:687
char particle_subdir[PETSC_MAX_PATH_LEN]
Definition variables.h:667
BoundingBox * bboxlist
Definition variables.h:742
Vec lKZet
Definition variables.h:862
Vec Eta
Definition variables.h:858
PetscReal rz
Definition variables.h:825
Vec lJEta
Definition variables.h:861
Vec lCsi
Definition variables.h:858
Vec lGridSpace
Definition variables.h:858
PetscInt thislevel
Definition variables.h:874
Vec ICsi
Definition variables.h:860
PetscScalar z
Definition variables.h:101
Vec pUcont
Definition variables.h:845
Vec lKCsi
Definition variables.h:862
Vec Ucat
Definition variables.h:837
Vec ParticleCount
Definition variables.h:882
Vec Ucont_o
Definition variables.h:844
PetscInt JM
Definition variables.h:820
PetscInt mglevels
Definition variables.h:535
PetscReal Min_Z
Definition variables.h:821
PetscInt mglevels
Definition variables.h:686
Vec lJZet
Definition variables.h:861
Vec Nvert_o
Definition variables.h:844
Vec IAj
Definition variables.h:860
Vec Psi_nodal
Definition variables.h:890
Vec JAj
Definition variables.h:861
Vec KEta
Definition variables.h:862
PetscReal Max_X
Definition variables.h:821
Vec Ucont_rm1
Definition variables.h:845
PetscReal Min_Y
Definition variables.h:821
PetscInt i_periodic
Definition variables.h:713
Vec lUcont
Definition variables.h:837
PetscInt step
Definition variables.h:651
Vec Diffusivity
Definition variables.h:840
Vec lAj
Definition variables.h:858
PetscRandom BrownianMotionRNG
Definition variables.h:753
Vec lICsi
Definition variables.h:860
DMDALocalInfo info
Definition variables.h:818
Vec dUcont
Definition variables.h:845
Vec lUcat
Definition variables.h:837
PetscScalar y
Definition variables.h:101
PetscMPIInt size
Definition variables.h:647
@ EXEC_MODE_SOLVER
Definition variables.h:616
PetscInt IM
Definition variables.h:820
Vec lEta
Definition variables.h:858
Vec KZet
Definition variables.h:862
Vec Cent
Definition variables.h:858
Vec Nvert
Definition variables.h:837
Vec KCsi
Definition variables.h:862
MGCtx * mgctx
Definition variables.h:538
Vec lDiffusivity
Definition variables.h:840
Vec lNvert_o
Definition variables.h:844
Vec Centy
Definition variables.h:859
PetscReal rx
Definition variables.h:825
ExecutionMode exec_mode
Definition variables.h:662
Vec lJAj
Definition variables.h:861
BoundingBox bbox
Definition variables.h:822
PetscReal Max_Z
Definition variables.h:821
MomentumSolverType mom_solver_type
Definition variables.h:683
Vec lKAj
Definition variables.h:862
char restart_dir[PETSC_MAX_PATH_LEN]
Definition variables.h:664
PetscInt LoggingFrequency
Definition variables.h:769
Vec Psi
Definition variables.h:883
Vec P_o
Definition variables.h:844
Vec Uch
Characteristic velocity for boundary conditions.
Definition variables.h:122
PetscInt j_periodic
Definition variables.h:713
Defines a 3D axis-aligned bounding box.
Definition variables.h:154
The master context for the entire simulation.
Definition variables.h:643
User-defined context containing data specific to a single computational grid level.
Definition variables.h:811
Here is the call graph for this function:
Here is the caller graph for this function:

◆ PicurvCreateMinimalContexts()

PetscErrorCode PicurvCreateMinimalContexts ( SimCtx **  simCtx_out,
UserCtx **  user_out,
PetscInt  mx,
PetscInt  my,
PetscInt  mz 
)

Builds minimal SimCtx and UserCtx fixtures for C unit tests.

Creates minimal PETSc/solver contexts used by isolated kernel tests.

Definition at line 530 of file test_support.c.

531{
532 PetscFunctionBeginUser;
533 PetscCall(PicurvCreateMinimalContextsWithPeriodicity(simCtx_out, user_out, mx, my, mz, PETSC_FALSE, PETSC_FALSE, PETSC_FALSE));
534 PetscFunctionReturn(0);
535}
PetscErrorCode PicurvCreateMinimalContextsWithPeriodicity(SimCtx **simCtx_out, UserCtx **user_out, PetscInt mx, PetscInt my, PetscInt mz, PetscBool x_periodic, PetscBool y_periodic, PetscBool z_periodic)
Builds minimal SimCtx and UserCtx fixtures for C unit tests with configurable periodicity.
Here is the call graph for this function:
Here is the caller graph for this function:

◆ PicurvPopulateUniformCellCenters()

PetscErrorCode PicurvPopulateUniformCellCenters ( UserCtx user)

Populates cell center coordinates for a uniform grid on [0,1]^3.

Uses the shifted-index convention: cell (i,j,k) is stored at array index (i+1, j+1, k+1). For mx cells on [0,1], cell i has center at (i+0.5)/mx.

Definition at line 543 of file test_support.c.

544{
545 Cmpnts ***cent = NULL;
546 PetscInt mx = user->IM;
547 PetscInt my = user->JM;
548 PetscInt mz = user->KM;
549
550 PetscFunctionBeginUser;
551 PetscCall(DMDAVecGetArray(user->fda, user->Cent, &cent));
552 for (PetscInt k = user->info.zs; k < user->info.zs + user->info.zm; k++) {
553 for (PetscInt j = user->info.ys; j < user->info.ys + user->info.ym; j++) {
554 for (PetscInt i = user->info.xs; i < user->info.xs + user->info.xm; i++) {
555 cent[k][j][i].x = (i - 0.5) / (PetscReal)mx;
556 cent[k][j][i].y = (j - 0.5) / (PetscReal)my;
557 cent[k][j][i].z = (k - 0.5) / (PetscReal)mz;
558 }
559 }
560 }
561 PetscCall(DMDAVecRestoreArray(user->fda, user->Cent, &cent));
562 PetscCall(DMGlobalToLocalBegin(user->fda, user->Cent, INSERT_VALUES, user->lCent));
563 PetscCall(DMGlobalToLocalEnd(user->fda, user->Cent, INSERT_VALUES, user->lCent));
564 PetscFunctionReturn(0);
565}
A 3D point or vector with PetscScalar components.
Definition variables.h:100
Here is the caller graph for this function:

◆ PicurvPopulateIdentityMetrics()

PetscErrorCode PicurvPopulateIdentityMetrics ( UserCtx user)

Populates identity metric vectors on the minimal grid fixture.

Fills metric vectors with identity metrics for Cartesian-reference tests.

Definition at line 571 of file test_support.c.

572{
573 Cmpnts ***csi = NULL;
574 Cmpnts ***eta = NULL;
575 Cmpnts ***zet = NULL;
576 Cmpnts ***icsi = NULL;
577 Cmpnts ***ieta = NULL;
578 Cmpnts ***izet = NULL;
579 Cmpnts ***jcsi = NULL;
580 Cmpnts ***jeta = NULL;
581 Cmpnts ***jzet = NULL;
582 Cmpnts ***kcsi = NULL;
583 Cmpnts ***keta = NULL;
584 Cmpnts ***kzet = NULL;
585 PetscReal ***aj = NULL;
586 PetscReal ***iaj = NULL;
587 PetscReal ***jaj = NULL;
588 PetscReal ***kaj = NULL;
589
590 PetscFunctionBeginUser;
591 PetscCall(DMDAVecGetArray(user->fda, user->Csi, &csi));
592 PetscCall(DMDAVecGetArray(user->fda, user->Eta, &eta));
593 PetscCall(DMDAVecGetArray(user->fda, user->Zet, &zet));
594 PetscCall(DMDAVecGetArray(user->fda, user->ICsi, &icsi));
595 PetscCall(DMDAVecGetArray(user->fda, user->IEta, &ieta));
596 PetscCall(DMDAVecGetArray(user->fda, user->IZet, &izet));
597 PetscCall(DMDAVecGetArray(user->fda, user->JCsi, &jcsi));
598 PetscCall(DMDAVecGetArray(user->fda, user->JEta, &jeta));
599 PetscCall(DMDAVecGetArray(user->fda, user->JZet, &jzet));
600 PetscCall(DMDAVecGetArray(user->fda, user->KCsi, &kcsi));
601 PetscCall(DMDAVecGetArray(user->fda, user->KEta, &keta));
602 PetscCall(DMDAVecGetArray(user->fda, user->KZet, &kzet));
603 PetscCall(DMDAVecGetArray(user->da, user->Aj, &aj));
604 PetscCall(DMDAVecGetArray(user->da, user->IAj, &iaj));
605 PetscCall(DMDAVecGetArray(user->da, user->JAj, &jaj));
606 PetscCall(DMDAVecGetArray(user->da, user->KAj, &kaj));
607
608 for (PetscInt k = user->info.zs; k < user->info.zs + user->info.zm; ++k) {
609 for (PetscInt j = user->info.ys; j < user->info.ys + user->info.ym; ++j) {
610 for (PetscInt i = user->info.xs; i < user->info.xs + user->info.xm; ++i) {
611 csi[k][j][i].x = 1.0; csi[k][j][i].y = 0.0; csi[k][j][i].z = 0.0;
612 eta[k][j][i].x = 0.0; eta[k][j][i].y = 1.0; eta[k][j][i].z = 0.0;
613 zet[k][j][i].x = 0.0; zet[k][j][i].y = 0.0; zet[k][j][i].z = 1.0;
614 icsi[k][j][i] = csi[k][j][i];
615 ieta[k][j][i] = eta[k][j][i];
616 izet[k][j][i] = zet[k][j][i];
617 jcsi[k][j][i] = csi[k][j][i];
618 jeta[k][j][i] = eta[k][j][i];
619 jzet[k][j][i] = zet[k][j][i];
620 kcsi[k][j][i] = csi[k][j][i];
621 keta[k][j][i] = eta[k][j][i];
622 kzet[k][j][i] = zet[k][j][i];
623 aj[k][j][i] = 1.0;
624 iaj[k][j][i] = 1.0;
625 jaj[k][j][i] = 1.0;
626 kaj[k][j][i] = 1.0;
627 }
628 }
629 }
630
631 PetscCall(DMDAVecRestoreArray(user->fda, user->Csi, &csi));
632 PetscCall(DMDAVecRestoreArray(user->fda, user->Eta, &eta));
633 PetscCall(DMDAVecRestoreArray(user->fda, user->Zet, &zet));
634 PetscCall(DMDAVecRestoreArray(user->fda, user->ICsi, &icsi));
635 PetscCall(DMDAVecRestoreArray(user->fda, user->IEta, &ieta));
636 PetscCall(DMDAVecRestoreArray(user->fda, user->IZet, &izet));
637 PetscCall(DMDAVecRestoreArray(user->fda, user->JCsi, &jcsi));
638 PetscCall(DMDAVecRestoreArray(user->fda, user->JEta, &jeta));
639 PetscCall(DMDAVecRestoreArray(user->fda, user->JZet, &jzet));
640 PetscCall(DMDAVecRestoreArray(user->fda, user->KCsi, &kcsi));
641 PetscCall(DMDAVecRestoreArray(user->fda, user->KEta, &keta));
642 PetscCall(DMDAVecRestoreArray(user->fda, user->KZet, &kzet));
643 PetscCall(DMDAVecRestoreArray(user->da, user->Aj, &aj));
644 PetscCall(DMDAVecRestoreArray(user->da, user->IAj, &iaj));
645 PetscCall(DMDAVecRestoreArray(user->da, user->JAj, &jaj));
646 PetscCall(DMDAVecRestoreArray(user->da, user->KAj, &kaj));
647
648 PetscCall(DMGlobalToLocalBegin(user->fda, user->Csi, INSERT_VALUES, user->lCsi));
649 PetscCall(DMGlobalToLocalEnd(user->fda, user->Csi, INSERT_VALUES, user->lCsi));
650 PetscCall(DMGlobalToLocalBegin(user->fda, user->Eta, INSERT_VALUES, user->lEta));
651 PetscCall(DMGlobalToLocalEnd(user->fda, user->Eta, INSERT_VALUES, user->lEta));
652 PetscCall(DMGlobalToLocalBegin(user->fda, user->Zet, INSERT_VALUES, user->lZet));
653 PetscCall(DMGlobalToLocalEnd(user->fda, user->Zet, INSERT_VALUES, user->lZet));
654 PetscCall(DMGlobalToLocalBegin(user->fda, user->ICsi, INSERT_VALUES, user->lICsi));
655 PetscCall(DMGlobalToLocalEnd(user->fda, user->ICsi, INSERT_VALUES, user->lICsi));
656 PetscCall(DMGlobalToLocalBegin(user->fda, user->IEta, INSERT_VALUES, user->lIEta));
657 PetscCall(DMGlobalToLocalEnd(user->fda, user->IEta, INSERT_VALUES, user->lIEta));
658 PetscCall(DMGlobalToLocalBegin(user->fda, user->IZet, INSERT_VALUES, user->lIZet));
659 PetscCall(DMGlobalToLocalEnd(user->fda, user->IZet, INSERT_VALUES, user->lIZet));
660 PetscCall(DMGlobalToLocalBegin(user->fda, user->JCsi, INSERT_VALUES, user->lJCsi));
661 PetscCall(DMGlobalToLocalEnd(user->fda, user->JCsi, INSERT_VALUES, user->lJCsi));
662 PetscCall(DMGlobalToLocalBegin(user->fda, user->JEta, INSERT_VALUES, user->lJEta));
663 PetscCall(DMGlobalToLocalEnd(user->fda, user->JEta, INSERT_VALUES, user->lJEta));
664 PetscCall(DMGlobalToLocalBegin(user->fda, user->JZet, INSERT_VALUES, user->lJZet));
665 PetscCall(DMGlobalToLocalEnd(user->fda, user->JZet, INSERT_VALUES, user->lJZet));
666 PetscCall(DMGlobalToLocalBegin(user->fda, user->KCsi, INSERT_VALUES, user->lKCsi));
667 PetscCall(DMGlobalToLocalEnd(user->fda, user->KCsi, INSERT_VALUES, user->lKCsi));
668 PetscCall(DMGlobalToLocalBegin(user->fda, user->KEta, INSERT_VALUES, user->lKEta));
669 PetscCall(DMGlobalToLocalEnd(user->fda, user->KEta, INSERT_VALUES, user->lKEta));
670 PetscCall(DMGlobalToLocalBegin(user->fda, user->KZet, INSERT_VALUES, user->lKZet));
671 PetscCall(DMGlobalToLocalEnd(user->fda, user->KZet, INSERT_VALUES, user->lKZet));
672 PetscCall(DMGlobalToLocalBegin(user->da, user->Nvert, INSERT_VALUES, user->lNvert));
673 PetscCall(DMGlobalToLocalEnd(user->da, user->Nvert, INSERT_VALUES, user->lNvert));
674 PetscCall(DMGlobalToLocalBegin(user->da, user->Aj, INSERT_VALUES, user->lAj));
675 PetscCall(DMGlobalToLocalEnd(user->da, user->Aj, INSERT_VALUES, user->lAj));
676 PetscCall(DMGlobalToLocalBegin(user->da, user->IAj, INSERT_VALUES, user->lIAj));
677 PetscCall(DMGlobalToLocalEnd(user->da, user->IAj, INSERT_VALUES, user->lIAj));
678 PetscCall(DMGlobalToLocalBegin(user->da, user->JAj, INSERT_VALUES, user->lJAj));
679 PetscCall(DMGlobalToLocalEnd(user->da, user->JAj, INSERT_VALUES, user->lJAj));
680 PetscCall(DMGlobalToLocalBegin(user->da, user->KAj, INSERT_VALUES, user->lKAj));
681 PetscCall(DMGlobalToLocalEnd(user->da, user->KAj, INSERT_VALUES, user->lKAj));
682 PetscCall(DMGlobalToLocalBegin(user->da, user->P, INSERT_VALUES, user->lP));
683 PetscCall(DMGlobalToLocalEnd(user->da, user->P, INSERT_VALUES, user->lP));
684 PetscCall(DMGlobalToLocalBegin(user->da, user->Psi, INSERT_VALUES, user->lPsi));
685 PetscCall(DMGlobalToLocalEnd(user->da, user->Psi, INSERT_VALUES, user->lPsi));
686 PetscCall(DMGlobalToLocalBegin(user->fda, user->Ucat, INSERT_VALUES, user->lUcat));
687 PetscCall(DMGlobalToLocalEnd(user->fda, user->Ucat, INSERT_VALUES, user->lUcat));
688 PetscCall(DMGlobalToLocalBegin(user->fda, user->Ucont, INSERT_VALUES, user->lUcont));
689 PetscCall(DMGlobalToLocalEnd(user->fda, user->Ucont, INSERT_VALUES, user->lUcont));
690 PetscCall(DMGlobalToLocalBegin(user->fda, user->Cent, INSERT_VALUES, user->lCent));
691 PetscCall(DMGlobalToLocalEnd(user->fda, user->Cent, INSERT_VALUES, user->lCent));
692 PetscFunctionReturn(0);
693}
Here is the caller graph for this function:

◆ PicurvCreateSwarmPair()

PetscErrorCode PicurvCreateSwarmPair ( UserCtx user,
PetscInt  nlocal,
const char *  post_field_name 
)

Creates matched solver and post-processing swarms for tests.

Creates paired runtime/post swarms with optional extra post-processing field registration.

Definition at line 698 of file test_support.c.

699{
700 PetscFunctionBeginUser;
701 if (!user) {
702 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_NULL, "UserCtx cannot be NULL.");
703 }
704
705 PetscCall(DMCreate(PETSC_COMM_WORLD, &user->swarm));
706 PetscCall(DMSetType(user->swarm, DMSWARM));
707 PetscCall(DMSetDimension(user->swarm, 3));
708 PetscCall(DMSwarmSetType(user->swarm, DMSWARM_BASIC));
709 PetscCall(DMSwarmSetCellDM(user->swarm, user->da));
710 PetscCall(RegisterSwarmFieldForTests(user->swarm, "position", 3, PETSC_REAL));
711 PetscCall(RegisterSwarmFieldForTests(user->swarm, "velocity", 3, PETSC_REAL));
712 PetscCall(RegisterSwarmFieldForTests(user->swarm, "DMSwarm_CellID", 3, PETSC_INT));
713 PetscCall(RegisterSwarmFieldForTests(user->swarm, "weight", 3, PETSC_REAL));
714 PetscCall(RegisterSwarmFieldForTests(user->swarm, "Diffusivity", 1, PETSC_REAL));
715 PetscCall(RegisterSwarmFieldForTests(user->swarm, "DiffusivityGradient", 3, PETSC_REAL));
716 PetscCall(RegisterSwarmFieldForTests(user->swarm, "Psi", 1, PETSC_REAL));
717 PetscCall(RegisterSwarmFieldForTests(user->swarm, "DMSwarm_location_status", 1, PETSC_INT));
718 PetscCall(DMSwarmFinalizeFieldRegister(user->swarm));
719 PetscCall(DMSwarmSetLocalSizes(user->swarm, nlocal, 0));
720
721 PetscCall(DMCreate(PETSC_COMM_WORLD, &user->post_swarm));
722 PetscCall(DMSetType(user->post_swarm, DMSWARM));
723 PetscCall(DMSetDimension(user->post_swarm, 3));
724 PetscCall(DMSwarmSetType(user->post_swarm, DMSWARM_BASIC));
725 PetscCall(DMSwarmSetCellDM(user->post_swarm, user->da));
726 PetscCall(RegisterSwarmFieldForTests(user->post_swarm, post_field_name, 1, PETSC_REAL));
727 PetscCall(DMSwarmFinalizeFieldRegister(user->post_swarm));
728 PetscCall(DMSwarmSetLocalSizes(user->post_swarm, nlocal, 0));
729 PetscFunctionReturn(0);
730}
static PetscErrorCode RegisterSwarmFieldForTests(DM swarm, const char *field_name, PetscInt field_dim, PetscDataType dtype)
Registers one DMSwarm field used by the C test fixtures.
DM post_swarm
Definition variables.h:886
Here is the call graph for this function:
Here is the caller graph for this function:

◆ PicurvBuildTinyRuntimeContext()

PetscErrorCode PicurvBuildTinyRuntimeContext ( const char *  bcs_contents,
PetscBool  enable_particles,
SimCtx **  simCtx_out,
UserCtx **  user_out,
char *  tmpdir,
size_t  tmpdir_len 
)

Builds a tiny runtime context through the real setup path for behavior-level tests.

Definition at line 735 of file test_support.c.

741{
742 char control_path[PETSC_MAX_PATH_LEN];
743 SimCtx *simCtx = NULL;
744
745 PetscFunctionBeginUser;
746 PetscCheck(simCtx_out != NULL, PETSC_COMM_SELF, PETSC_ERR_ARG_NULL, "SimCtx output cannot be NULL.");
747
748 PetscCall(PetscOptionsClear(NULL));
749 PetscCall(PrepareTinyRuntimeConfig(bcs_contents, enable_particles, tmpdir, tmpdir_len, control_path, sizeof(control_path)));
750 PetscCall(PetscOptionsSetValue(NULL, "-control_file", control_path));
751 PetscCall(CreateSimulationContext(0, NULL, &simCtx));
752 simCtx->exec_mode = EXEC_MODE_SOLVER;
753 PetscCall(SetupSimulationEnvironment(simCtx));
754 PetscCall(SetupGridAndSolvers(simCtx));
755 PetscCall(SetupBoundaryConditions(simCtx));
756 PetscCall(SetupDomainRankInfo(simCtx));
757
758 *simCtx_out = simCtx;
759 if (user_out) {
760 *user_out = simCtx->usermg.mgctx[simCtx->usermg.mglevels - 1].user;
761 }
762 PetscFunctionReturn(0);
763}
PetscErrorCode SetupDomainRankInfo(SimCtx *simCtx)
Sets up the full rank communication infrastructure, including neighbor ranks and bounding box exchang...
Definition setup.c:2174
PetscErrorCode SetupGridAndSolvers(SimCtx *simCtx)
The main orchestrator for setting up all grid-related components.
Definition setup.c:1132
PetscErrorCode SetupSimulationEnvironment(SimCtx *simCtx)
Verifies and prepares the complete I/O environment for a simulation run.
Definition setup.c:813
PetscErrorCode CreateSimulationContext(int argc, char **argv, SimCtx **p_simCtx)
Allocates and populates the master SimulationContext object.
Definition setup.c:50
PetscErrorCode SetupBoundaryConditions(SimCtx *simCtx)
(Orchestrator) Sets up all boundary conditions for the simulation.
Definition setup.c:1633
static PetscErrorCode PrepareTinyRuntimeConfig(const char *bcs_contents, PetscBool enable_particles, char *tmpdir, size_t tmpdir_len, char *control_path, size_t control_path_len)
Creates a tiny control-file bundle used by richer runtime fixtures built through the setup path.
Here is the call graph for this function:
Here is the caller graph for this function:

◆ PicurvDestroyRuntimeContext()

PetscErrorCode PicurvDestroyRuntimeContext ( SimCtx **  simCtx_ptr)

Finalizes and frees a runtime context built by PicurvBuildTinyRuntimeContext.

Definition at line 768 of file test_support.c.

769{
770 PetscFunctionBeginUser;
771 if (simCtx_ptr && *simCtx_ptr) {
772 PetscCall(FinalizeSimulation(*simCtx_ptr));
773 PetscCall(PetscFree(*simCtx_ptr));
774 *simCtx_ptr = NULL;
775 }
776 PetscCall(PetscOptionsClear(NULL));
777 PetscFunctionReturn(0);
778}
PetscErrorCode FinalizeSimulation(SimCtx *simCtx)
Main cleanup function for the entire simulation context.
Definition setup.c:3188
Here is the call graph for this function:
Here is the caller graph for this function:

◆ PicurvDestroyMinimalContexts()

PetscErrorCode PicurvDestroyMinimalContexts ( SimCtx **  simCtx_ptr,
UserCtx **  user_ptr 
)

Destroys minimal SimCtx/UserCtx fixtures and all owned PETSc objects.

Destroys contexts previously created by PicurvCreateMinimalContexts.

Definition at line 783 of file test_support.c.

784{
785 UserCtx *user = NULL;
786 SimCtx *simCtx = NULL;
787
788 PetscFunctionBeginUser;
789 if (user_ptr) {
790 user = *user_ptr;
791 }
792 if (simCtx_ptr) {
793 simCtx = *simCtx_ptr;
794 }
795
796 if (user) {
797 PetscCall(DestroyDMIfSet(&user->swarm));
798 PetscCall(DestroyDMIfSet(&user->post_swarm));
799
800 PetscCall(DestroyVecIfSet(&user->P));
801 PetscCall(DestroyVecIfSet(&user->lP));
802 PetscCall(DestroyVecIfSet(&user->Phi));
803 PetscCall(DestroyVecIfSet(&user->lPhi));
804 PetscCall(DestroyVecIfSet(&user->Nvert));
805 PetscCall(DestroyVecIfSet(&user->lNvert));
806 PetscCall(DestroyVecIfSet(&user->ParticleCount));
807 PetscCall(DestroyVecIfSet(&user->lParticleCount));
808 PetscCall(DestroyVecIfSet(&user->Psi));
809 PetscCall(DestroyVecIfSet(&user->lPsi));
810 PetscCall(DestroyVecIfSet(&user->Qcrit));
811 PetscCall(DestroyVecIfSet(&user->P_nodal));
812 PetscCall(DestroyVecIfSet(&user->Psi_nodal));
813 PetscCall(DestroyVecIfSet(&user->Ucat));
814 PetscCall(DestroyVecIfSet(&user->lUcat));
815 PetscCall(DestroyVecIfSet(&user->Ucont));
816 PetscCall(DestroyVecIfSet(&user->lUcont));
817 PetscCall(DestroyVecIfSet(&user->Csi));
818 PetscCall(DestroyVecIfSet(&user->lCsi));
819 PetscCall(DestroyVecIfSet(&user->Eta));
820 PetscCall(DestroyVecIfSet(&user->lEta));
821 PetscCall(DestroyVecIfSet(&user->Zet));
822 PetscCall(DestroyVecIfSet(&user->lZet));
823 PetscCall(DestroyVecIfSet(&user->Aj));
824 PetscCall(DestroyVecIfSet(&user->lAj));
825 PetscCall(DestroyVecIfSet(&user->IAj));
826 PetscCall(DestroyVecIfSet(&user->lIAj));
827 PetscCall(DestroyVecIfSet(&user->JAj));
828 PetscCall(DestroyVecIfSet(&user->lJAj));
829 PetscCall(DestroyVecIfSet(&user->KAj));
830 PetscCall(DestroyVecIfSet(&user->lKAj));
831 PetscCall(DestroyVecIfSet(&user->Diffusivity));
832 PetscCall(DestroyVecIfSet(&user->lDiffusivity));
833 PetscCall(DestroyVecIfSet(&user->DiffusivityGradient));
834 PetscCall(DestroyVecIfSet(&user->lDiffusivityGradient));
835 PetscCall(DestroyVecIfSet(&user->CS));
836 PetscCall(DestroyVecIfSet(&user->lCs));
837 PetscCall(DestroyVecIfSet(&user->Nu_t));
838 PetscCall(DestroyVecIfSet(&user->lNu_t));
839 PetscCall(DestroyVecIfSet(&user->Ucont_o));
840 PetscCall(DestroyVecIfSet(&user->lUcont_o));
841 PetscCall(DestroyVecIfSet(&user->Ucat_o));
842 PetscCall(DestroyVecIfSet(&user->P_o));
843 PetscCall(DestroyVecIfSet(&user->Nvert_o));
844 PetscCall(DestroyVecIfSet(&user->lNvert_o));
845 PetscCall(DestroyVecIfSet(&user->Ucont_rm1));
846 PetscCall(DestroyVecIfSet(&user->lUcont_rm1));
847 PetscCall(DestroyVecIfSet(&user->Rhs));
848 PetscCall(DestroyVecIfSet(&user->dUcont));
849 PetscCall(DestroyVecIfSet(&user->pUcont));
850 PetscCall(DestroyVecIfSet(&user->CellFieldAtCorner));
851 PetscCall(DestroyVecIfSet(&user->lCellFieldAtCorner));
852 PetscCall(DestroyVecIfSet(&user->B));
853 PetscCall(DestroyVecIfSet(&user->R));
854 PetscCall(DestroyVecIfSet(&user->Cent));
855 PetscCall(DestroyVecIfSet(&user->lCent));
856 PetscCall(DestroyVecIfSet(&user->GridSpace));
857 PetscCall(DestroyVecIfSet(&user->lGridSpace));
858 PetscCall(DestroyVecIfSet(&user->Centx));
859 PetscCall(DestroyVecIfSet(&user->Centy));
860 PetscCall(DestroyVecIfSet(&user->Centz));
861 PetscCall(DestroyVecIfSet(&user->Ucat_nodal));
862 PetscCall(DestroyVecIfSet(&user->Bcs.Ubcs));
863 PetscCall(DestroyVecIfSet(&user->Bcs.Uch));
864 PetscCall(DestroyMatIfSet(&user->A));
865 PetscCall(DestroyMatIfSet(&user->C));
866 PetscCall(DestroyKSPIfSet(&user->ksp));
867 PetscCall(DestroyNullSpaceIfSet(&user->nullsp));
868 PetscCall(PetscFree(user->RankCellInfoMap));
869 user->RankCellInfoMap = NULL;
870 for (PetscInt face = BC_FACE_NEG_X; face <= BC_FACE_POS_Z; ++face) {
871 if (user->boundary_faces[face].params) {
873 user->boundary_faces[face].params = NULL;
874 }
875 }
876
877 PetscCall(DestroyDMIfSet(&user->fda));
878 PetscCall(DestroyDMIfSet(&user->da));
879 PetscCall(PetscFree(user));
880 if (user_ptr) {
881 *user_ptr = NULL;
882 }
883 }
884
885 if (simCtx) {
886 PetscCall(DestroyRandomIfSet(&simCtx->BrownianMotionRNG));
887 PetscCall(PetscFree(simCtx->bboxlist));
888 PetscCall(PetscFree(simCtx->usermg.mgctx));
889 PetscCall(PetscFree(simCtx->pps));
890 PetscCall(PetscFree(simCtx));
891 if (simCtx_ptr) {
892 *simCtx_ptr = NULL;
893 }
894 }
895
896 PetscFunctionReturn(0);
897}
void FreeBC_ParamList(BC_Param *head)
Frees an entire linked list of boundary-condition parameters.
Definition io.c:302
static PetscErrorCode DestroyKSPIfSet(KSP *ksp)
Destroys a PETSc KSP only when the handle is non-null.
static PetscErrorCode DestroyDMIfSet(DM *dm)
Destroys a PETSc DM only when the handle is non-null.
static PetscErrorCode DestroyNullSpaceIfSet(MatNullSpace *nullsp)
Destroys a PETSc nullspace only when the handle is non-null.
static PetscErrorCode DestroyRandomIfSet(PetscRandom *rand_ctx)
Destroys a PETSc random generator only when the handle is non-null.
static PetscErrorCode DestroyVecIfSet(Vec *vec)
Destroys a PETSc vector only when the handle is non-null.
static PetscErrorCode DestroyMatIfSet(Mat *mat)
Destroys a PETSc matrix only when the handle is non-null.
BoundaryFaceConfig boundary_faces[6]
Definition variables.h:829
MatNullSpace nullsp
Definition variables.h:849
Vec lCellFieldAtCorner
Definition variables.h:846
Vec lCs
Definition variables.h:865
RankCellInfo * RankCellInfoMap
Definition variables.h:881
Vec lNu_t
Definition variables.h:865
Vec Nu_t
Definition variables.h:865
BC_Param * params
Definition variables.h:338
Vec CellFieldAtCorner
Definition variables.h:846
PostProcessParams * pps
Definition variables.h:798
KSP ksp
Definition variables.h:849
@ BC_FACE_NEG_X
Definition variables.h:245
@ BC_FACE_POS_Z
Definition variables.h:247
Here is the call graph for this function:
Here is the caller graph for this function:

◆ PicurvAssertRealNear()

PetscErrorCode PicurvAssertRealNear ( PetscReal  expected,
PetscReal  actual,
PetscReal  tol,
const char *  context 
)

Asserts that two real values agree within tolerance.

Asserts two real values are within tolerance for test validation.

Definition at line 902 of file test_support.c.

903{
904 PetscFunctionBeginUser;
905 if (PetscAbsReal(expected - actual) > tol) {
906 PetscCall(PetscPrintf(PETSC_COMM_WORLD,
907 "[FAIL] %s | expected=%0.12e actual=%0.12e tol=%0.12e\n",
908 context, (double)expected, (double)actual, (double)tol));
909 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Assertion failed.");
910 }
911 PetscFunctionReturn(0);
912}
Here is the caller graph for this function:

◆ PicurvAssertIntEqual()

PetscErrorCode PicurvAssertIntEqual ( PetscInt  expected,
PetscInt  actual,
const char *  context 
)

Asserts that two integer values are equal.

Asserts two integer values are exactly equal for test validation.

Definition at line 917 of file test_support.c.

918{
919 PetscFunctionBeginUser;
920 if (expected != actual) {
921 PetscCall(PetscPrintf(PETSC_COMM_WORLD,
922 "[FAIL] %s | expected=%" PetscInt_FMT " actual=%" PetscInt_FMT "\n",
923 context, expected, actual));
924 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Assertion failed.");
925 }
926 PetscFunctionReturn(0);
927}
Here is the caller graph for this function:

◆ PicurvAssertBool()

PetscErrorCode PicurvAssertBool ( PetscBool  value,
const char *  context 
)

Asserts that one boolean condition is true.

Asserts a boolean condition is true for test validation.

Definition at line 932 of file test_support.c.

933{
934 PetscFunctionBeginUser;
935 if (!value) {
936 PetscCall(PetscPrintf(PETSC_COMM_WORLD, "[FAIL] %s\n", context));
937 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Assertion failed.");
938 }
939 PetscFunctionReturn(0);
940}
Here is the caller graph for this function:

◆ PicurvAssertFileExists()

PetscErrorCode PicurvAssertFileExists ( const char *  path,
const char *  context 
)

Asserts that a filesystem path exists as a readable file.

Asserts a filesystem path exists.

Definition at line 945 of file test_support.c.

946{
947 PetscBool exists = PETSC_FALSE;
948
949 PetscFunctionBeginUser;
950 PetscCall(PetscTestFile(path, 'r', &exists));
951 if (!exists) {
952 PetscCall(PetscPrintf(PETSC_COMM_WORLD, "[FAIL] %s | missing file: %s\n", context, path));
953 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_FILE_OPEN, "Expected file is missing.");
954 }
955 PetscFunctionReturn(0);
956}
Here is the caller graph for this function:

◆ PicurvAssertVecConstant()

PetscErrorCode PicurvAssertVecConstant ( Vec  vec,
PetscScalar  expected,
PetscReal  tol,
const char *  context 
)

Asserts that a PETSc vector is spatially constant within tolerance.

Asserts every entry of a vector equals a constant within tolerance.

Definition at line 961 of file test_support.c.

962{
963 PetscReal vmin = 0.0;
964 PetscReal vmax = 0.0;
965
966 PetscFunctionBeginUser;
967 PetscCall(VecMin(vec, NULL, &vmin));
968 PetscCall(VecMax(vec, NULL, &vmax));
969 PetscCall(PicurvAssertRealNear((PetscReal)expected, vmin, tol, context));
970 PetscCall(PicurvAssertRealNear((PetscReal)expected, vmax, tol, context));
971 PetscFunctionReturn(0);
972}
PetscErrorCode PicurvAssertRealNear(PetscReal expected, PetscReal actual, PetscReal tol, const char *context)
Asserts that two real values agree within tolerance.
Here is the call graph for this function:
Here is the caller graph for this function: