xref: /honee/src/misc.c (revision e3663b908338ee76924aed100340d66e13adb0df)
1ae2b091fSJames Wright // SPDX-FileCopyrightText: Copyright (c) 2017-2024, HONEE contributors.
2ae2b091fSJames Wright // SPDX-License-Identifier: Apache-2.0 OR BSD-2-Clause
3a515125bSLeila Ghaffari 
4a515125bSLeila Ghaffari /// @file
5a515125bSLeila Ghaffari /// Miscellaneous utility functions
6a515125bSLeila Ghaffari 
7e419654dSJeremy L Thompson #include <ceed.h>
8e419654dSJeremy L Thompson #include <petscdm.h>
9926a6279SJames Wright #include <petscsf.h>
10e419654dSJeremy L Thompson #include <petscts.h>
11e419654dSJeremy L Thompson 
12149fb536SJames Wright #include <navierstokes.h>
139f59f36eSJames Wright #include "../qfunctions/mass.h"
14a515125bSLeila Ghaffari 
15*e3663b90SJames Wright PetscErrorCode ICs_FixMultiplicity(DM dm, Honee honee, Vec Q_loc, Vec Q, CeedScalar time) {
160c373b74SJames Wright   Ceed         ceed = honee->ceed;
17b2948607SJames Wright   CeedVector   mult_vec;
18b2948607SJames Wright   PetscMemType m_mem_type;
19b2948607SJames Wright   Vec          Multiplicity, Multiplicity_loc;
20b2948607SJames Wright 
21a515125bSLeila Ghaffari   PetscFunctionBeginUser;
22*e3663b90SJames Wright   if (honee->phys->ics_time_label) PetscCallCeed(ceed, CeedOperatorSetContextDouble(honee->op_ics_ctx->op, honee->phys->ics_time_label, &time));
23*e3663b90SJames Wright   PetscCall(ApplyCeedOperatorLocalToGlobal(NULL, Q, honee->op_ics_ctx));
24a515125bSLeila Ghaffari 
25*e3663b90SJames Wright   PetscCallCeed(ceed, CeedElemRestrictionCreateVector(honee->elem_restr_q, &mult_vec, NULL));
26a515125bSLeila Ghaffari 
27a515125bSLeila Ghaffari   // -- Get multiplicity
28b2948607SJames Wright   PetscCall(DMGetLocalVector(dm, &Multiplicity_loc));
29a7dac1d5SJames Wright   PetscCall(VecPetscToCeed(Multiplicity_loc, &m_mem_type, mult_vec));
30*e3663b90SJames Wright   PetscCallCeed(ceed, CeedElemRestrictionGetMultiplicity(honee->elem_restr_q, mult_vec));
31a7dac1d5SJames Wright   PetscCall(VecCeedToPetsc(mult_vec, m_mem_type, Multiplicity_loc));
32a515125bSLeila Ghaffari 
33b2948607SJames Wright   PetscCall(DMGetGlobalVector(dm, &Multiplicity));
34b2948607SJames Wright   PetscCall(VecZeroEntries(Multiplicity));
35b2948607SJames Wright   PetscCall(DMLocalToGlobal(dm, Multiplicity_loc, ADD_VALUES, Multiplicity));
36a515125bSLeila Ghaffari 
37a515125bSLeila Ghaffari   // -- Fix multiplicity
38b2948607SJames Wright   PetscCall(VecPointwiseDivide(Q, Q, Multiplicity));
39b2948607SJames Wright   PetscCall(VecPointwiseDivide(Q_loc, Q_loc, Multiplicity_loc));
40a515125bSLeila Ghaffari 
41b2948607SJames Wright   PetscCall(DMRestoreLocalVector(dm, &Multiplicity_loc));
42b2948607SJames Wright   PetscCall(DMRestoreGlobalVector(dm, &Multiplicity));
43b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedVectorDestroy(&mult_vec));
44d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
45a515125bSLeila Ghaffari }
46a515125bSLeila Ghaffari 
47c56e8d5bSJames Wright // Record boundary values from initial condition
48c56e8d5bSJames Wright PetscErrorCode SetBCsFromICs(DM dm, Vec Q, Vec Q_loc) {
49c56e8d5bSJames Wright   PetscFunctionBeginUser;
50313f2f1eSJames Wright   {  // Capture initial condition values in Qbc
51313f2f1eSJames Wright     Vec Qbc;
52313f2f1eSJames Wright 
53c56e8d5bSJames Wright     PetscCall(DMGetNamedLocalVector(dm, "Qbc", &Qbc));
54c56e8d5bSJames Wright     PetscCall(VecCopy(Q_loc, Qbc));
55c56e8d5bSJames Wright     PetscCall(VecZeroEntries(Q_loc));
56c56e8d5bSJames Wright     PetscCall(DMGlobalToLocal(dm, Q, INSERT_VALUES, Q_loc));
57c56e8d5bSJames Wright     PetscCall(VecAXPY(Qbc, -1., Q_loc));
58c56e8d5bSJames Wright     PetscCall(DMRestoreNamedLocalVector(dm, "Qbc", &Qbc));
59313f2f1eSJames Wright   }
60c56e8d5bSJames Wright   PetscCall(PetscObjectComposeFunction((PetscObject)dm, "DMPlexInsertBoundaryValues_C", DMPlexInsertBoundaryValues_FromICs));
61c56e8d5bSJames Wright 
62313f2f1eSJames Wright   {  // Set boundary mask to zero out essential BCs
63313f2f1eSJames Wright     Vec boundary_mask, ones;
64313f2f1eSJames Wright 
65c56e8d5bSJames Wright     PetscCall(DMGetNamedLocalVector(dm, "boundary mask", &boundary_mask));
66313f2f1eSJames Wright     PetscCall(DMGetGlobalVector(dm, &ones));
67c56e8d5bSJames Wright     PetscCall(VecZeroEntries(boundary_mask));
68313f2f1eSJames Wright     PetscCall(VecSet(ones, 1.0));
69313f2f1eSJames Wright     PetscCall(DMGlobalToLocal(dm, ones, INSERT_VALUES, boundary_mask));
70c56e8d5bSJames Wright     PetscCall(DMRestoreNamedLocalVector(dm, "boundary mask", &boundary_mask));
71313f2f1eSJames Wright     PetscCall(DMRestoreGlobalVector(dm, &ones));
72313f2f1eSJames Wright   }
73c56e8d5bSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
74c56e8d5bSJames Wright }
75c56e8d5bSJames Wright 
76c56e8d5bSJames Wright PetscErrorCode DMPlexInsertBoundaryValues_FromICs(DM dm, PetscBool insert_essential, Vec Q_loc, PetscReal time, Vec face_geom_FVM, Vec cell_geom_FVM,
772b916ea7SJeremy L Thompson                                                   Vec grad_FVM) {
789d437337SJames Wright   Vec Qbc, boundary_mask;
79a515125bSLeila Ghaffari 
8006f41313SJames Wright   PetscFunctionBeginUser;
812eb7bf1fSJames Wright   // Mask (zero) Strong BC entries
829d437337SJames Wright   PetscCall(DMGetNamedLocalVector(dm, "boundary mask", &boundary_mask));
839d437337SJames Wright   PetscCall(VecPointwiseMult(Q_loc, Q_loc, boundary_mask));
849d437337SJames Wright   PetscCall(DMRestoreNamedLocalVector(dm, "boundary mask", &boundary_mask));
859d437337SJames Wright 
862b916ea7SJeremy L Thompson   PetscCall(DMGetNamedLocalVector(dm, "Qbc", &Qbc));
872b916ea7SJeremy L Thompson   PetscCall(VecAXPY(Q_loc, 1., Qbc));
882b916ea7SJeremy L Thompson   PetscCall(DMRestoreNamedLocalVector(dm, "Qbc", &Qbc));
89d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
90a515125bSLeila Ghaffari }
91a515125bSLeila Ghaffari 
92a515125bSLeila Ghaffari // Compare reference solution values with current test run for CI
93c56e8d5bSJames Wright PetscErrorCode RegressionTest(AppCtx app_ctx, Vec Q) {
94a515125bSLeila Ghaffari   Vec         Qref;
95a515125bSLeila Ghaffari   PetscViewer viewer;
96a515125bSLeila Ghaffari   PetscReal   error, Qrefnorm;
97e7754af5SKenneth E. Jansen   MPI_Comm    comm = PetscObjectComm((PetscObject)Q);
98a515125bSLeila Ghaffari 
9906f41313SJames Wright   PetscFunctionBeginUser;
100a515125bSLeila Ghaffari   // Read reference file
1012b916ea7SJeremy L Thompson   PetscCall(VecDuplicate(Q, &Qref));
1024c07ec22SJames Wright   PetscCheck(strcmp(app_ctx->test_file_path, "") != 0, comm, PETSC_ERR_FILE_READ, "File for regression test not given");
103e7754af5SKenneth E. Jansen   PetscCall(PetscViewerBinaryOpen(comm, app_ctx->test_file_path, FILE_MODE_READ, &viewer));
104360b37c9SJames Wright   PetscCall(HoneeLoadBinaryVec(viewer, Qref, NULL, NULL));
105a515125bSLeila Ghaffari 
106a515125bSLeila Ghaffari   // Compute error with respect to reference solution
1072b916ea7SJeremy L Thompson   PetscCall(VecAXPY(Q, -1.0, Qref));
1082b916ea7SJeremy L Thompson   PetscCall(VecNorm(Qref, NORM_MAX, &Qrefnorm));
1092b916ea7SJeremy L Thompson   PetscCall(VecScale(Q, 1. / Qrefnorm));
1102b916ea7SJeremy L Thompson   PetscCall(VecNorm(Q, NORM_MAX, &error));
111a515125bSLeila Ghaffari 
112a515125bSLeila Ghaffari   // Check error
113a515125bSLeila Ghaffari   if (error > app_ctx->test_tol) {
1142b916ea7SJeremy L Thompson     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Test failed with error norm %g\n", (double)error));
115a515125bSLeila Ghaffari   }
116a515125bSLeila Ghaffari 
117a515125bSLeila Ghaffari   // Cleanup
1182b916ea7SJeremy L Thompson   PetscCall(PetscViewerDestroy(&viewer));
1192b916ea7SJeremy L Thompson   PetscCall(VecDestroy(&Qref));
120d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
121a515125bSLeila Ghaffari }
122a515125bSLeila Ghaffari 
123a515125bSLeila Ghaffari // Get error for problems with exact solutions
124*e3663b90SJames Wright PetscErrorCode PrintError(DM dm, Honee honee, Vec Q, PetscScalar final_time) {
125a515125bSLeila Ghaffari   PetscInt  loc_nodes;
126a515125bSLeila Ghaffari   Vec       Q_exact, Q_exact_loc;
127a515125bSLeila Ghaffari   PetscReal rel_error, norm_error, norm_exact;
128a515125bSLeila Ghaffari 
12906f41313SJames Wright   PetscFunctionBeginUser;
130a515125bSLeila Ghaffari   // Get exact solution at final time
131b2948607SJames Wright   PetscCall(DMGetGlobalVector(dm, &Q_exact));
1322b916ea7SJeremy L Thompson   PetscCall(DMGetLocalVector(dm, &Q_exact_loc));
1332b916ea7SJeremy L Thompson   PetscCall(VecGetSize(Q_exact_loc, &loc_nodes));
134*e3663b90SJames Wright   PetscCall(ICs_FixMultiplicity(dm, honee, Q_exact_loc, Q_exact, final_time));
135a515125bSLeila Ghaffari 
136a515125bSLeila Ghaffari   // Get |exact solution - obtained solution|
1372b916ea7SJeremy L Thompson   PetscCall(VecNorm(Q_exact, NORM_1, &norm_exact));
1382b916ea7SJeremy L Thompson   PetscCall(VecAXPY(Q, -1.0, Q_exact));
1392b916ea7SJeremy L Thompson   PetscCall(VecNorm(Q, NORM_1, &norm_error));
140a515125bSLeila Ghaffari 
141a515125bSLeila Ghaffari   rel_error = norm_error / norm_exact;
1422b916ea7SJeremy L Thompson   PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Relative Error: %g\n", (double)rel_error));
1432b916ea7SJeremy L Thompson   PetscCall(DMRestoreLocalVector(dm, &Q_exact_loc));
144b2948607SJames Wright   PetscCall(DMRestoreGlobalVector(dm, &Q_exact));
145d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
146a515125bSLeila Ghaffari }
147a515125bSLeila Ghaffari 
148a515125bSLeila Ghaffari // Post-processing
149*e3663b90SJames Wright PetscErrorCode PostProcess(TS ts, DM dm, ProblemData problem, Honee honee, Vec Q, PetscScalar final_time) {
150a515125bSLeila Ghaffari   PetscInt          steps;
151f0784ed3SJed Brown   TSConvergedReason reason;
152a515125bSLeila Ghaffari 
15306f41313SJames Wright   PetscFunctionBeginUser;
154a515125bSLeila Ghaffari   // Print relative error
1550c373b74SJames Wright   if (problem->compute_exact_solution_error && honee->app_ctx->test_type == TESTTYPE_NONE) {
156*e3663b90SJames Wright     PetscCall(PrintError(dm, honee, Q, final_time));
157a515125bSLeila Ghaffari   }
158a515125bSLeila Ghaffari 
159a515125bSLeila Ghaffari   // Print final time and number of steps
1602b916ea7SJeremy L Thompson   PetscCall(TSGetStepNumber(ts, &steps));
161f0784ed3SJed Brown   PetscCall(TSGetConvergedReason(ts, &reason));
1620c373b74SJames Wright   if (honee->app_ctx->test_type == TESTTYPE_NONE) {
163f0784ed3SJed Brown     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Time integrator %s on time step %" PetscInt_FMT " with final time %g\n", TSConvergedReasons[reason],
164f0784ed3SJed Brown                           steps, (double)final_time));
165a515125bSLeila Ghaffari   }
166a515125bSLeila Ghaffari 
167a515125bSLeila Ghaffari   // Output numerical values from command line
1682b916ea7SJeremy L Thompson   PetscCall(VecViewFromOptions(Q, NULL, "-vec_view"));
169a515125bSLeila Ghaffari 
170a515125bSLeila Ghaffari   // Compare reference solution values with current test run for CI
1710c373b74SJames Wright   if (honee->app_ctx->test_type == TESTTYPE_SOLVER) {
1720c373b74SJames Wright     PetscCall(RegressionTest(honee->app_ctx, Q));
173a515125bSLeila Ghaffari   }
174d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
175a515125bSLeila Ghaffari }
176a515125bSLeila Ghaffari 
17715a3537eSJed Brown // Free a plain data context that was allocated using PETSc; returning libCEED error codes
17815a3537eSJed Brown int FreeContextPetsc(void *data) {
1792b916ea7SJeremy L Thompson   if (PetscFree(data)) return CeedError(NULL, CEED_ERROR_ACCESS, "PetscFree failed");
18015a3537eSJed Brown   return CEED_ERROR_SUCCESS;
18115a3537eSJed Brown }
1829f59f36eSJames Wright 
1834cb09fddSJames Wright // @brief Return mass qfunction specification for number of components N
1849f59f36eSJames Wright PetscErrorCode CreateMassQFunction(Ceed ceed, CeedInt N, CeedInt q_data_size, CeedQFunction *qf) {
1859f59f36eSJames Wright   PetscFunctionBeginUser;
1869f59f36eSJames Wright   switch (N) {
1879f59f36eSJames Wright     case 1:
188b4c37c5cSJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_1, Mass_1_loc, qf));
1899f59f36eSJames Wright       break;
19040b78511SJames Wright     case 2:
19140b78511SJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 2, Mass_2, Mass_2_loc, qf));
19240b78511SJames Wright       break;
19340b78511SJames Wright     case 3:
19440b78511SJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 3, Mass_3, Mass_3_loc, qf));
19540b78511SJames Wright       break;
1968c85b835SJames Wright     case 4:
1978c85b835SJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 4, Mass_4, Mass_4_loc, qf));
1988c85b835SJames Wright       break;
1999f59f36eSJames Wright     case 5:
200b4c37c5cSJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_5, Mass_5_loc, qf));
2019f59f36eSJames Wright       break;
202c38c977aSJames Wright     case 7:
203b4c37c5cSJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_7, Mass_7_loc, qf));
204c38c977aSJames Wright       break;
2059f59f36eSJames Wright     case 9:
206b4c37c5cSJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_9, Mass_9_loc, qf));
2079f59f36eSJames Wright       break;
2088c85b835SJames Wright     case 12:
2098c85b835SJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_12, Mass_12_loc, qf));
2108c85b835SJames Wright       break;
2119f59f36eSJames Wright     case 22:
212b4c37c5cSJames Wright       PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Mass_22, Mass_22_loc, qf));
2139f59f36eSJames Wright       break;
2149f59f36eSJames Wright     default:
21540b78511SJames Wright       SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_SUP,
21640b78511SJames Wright               "Mass qfunction of size %" CeedInt_FMT " does not exist.\nA new mass qfunction can be easily added; see source code for pattern.", N);
2179f59f36eSJames Wright   }
2189f59f36eSJames Wright 
219b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddInput(*qf, "u", N, CEED_EVAL_INTERP));
220b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddInput(*qf, "qdata", q_data_size, CEED_EVAL_NONE));
221b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddOutput(*qf, "v", N, CEED_EVAL_INTERP));
2223170c09fSJames Wright   PetscCallCeed(ceed, CeedQFunctionSetUserFlopsEstimate(*qf, N));
223d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
2249f59f36eSJames Wright }
225e5e81594SJames Wright 
226457a5831SJames Wright PetscErrorCode NodalProjectionDataDestroy(NodalProjectionData context) {
227457a5831SJames Wright   PetscFunctionBeginUser;
228d949ddfcSJames Wright   if (context == NULL) PetscFunctionReturn(PETSC_SUCCESS);
229457a5831SJames Wright 
230457a5831SJames Wright   PetscCall(DMDestroy(&context->dm));
231457a5831SJames Wright   PetscCall(KSPDestroy(&context->ksp));
232457a5831SJames Wright 
233457a5831SJames Wright   PetscCall(OperatorApplyContextDestroy(context->l2_rhs_ctx));
234457a5831SJames Wright 
235457a5831SJames Wright   PetscCall(PetscFree(context));
236d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
237457a5831SJames Wright }
238457a5831SJames Wright 
23959572072SJames Wright /**
24059572072SJames Wright    @brief Sets the value of an option if it is not already set
24159572072SJames Wright 
24259572072SJames Wright    @param[in,out] options `PetscOptions` database, or `NULL` for default global database
24359572072SJames Wright    @param[in]     name    Name of the option (with `-` prepended to it)
24459572072SJames Wright    @param[in]     value   Value to set the option to
24559572072SJames Wright **/
24659572072SJames Wright PetscErrorCode HoneeOptionsSetValueDefault(PetscOptions options, const char name[], const char value[]) {
24759572072SJames Wright   PetscBool has_option;
24859572072SJames Wright 
24959572072SJames Wright   PetscFunctionBeginUser;
25059572072SJames Wright   PetscCall(PetscOptionsHasName(options, NULL, name, &has_option));
25159572072SJames Wright   if (!has_option) PetscCall(PetscOptionsSetValue(options, name, value));
25259572072SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
25359572072SJames Wright }
25459572072SJames Wright 
255926a6279SJames Wright // Print information about the given simulation run
2560c373b74SJames Wright PetscErrorCode PrintRunInfo(Honee honee, Physics phys_ctx, ProblemData problem, TS ts) {
2570c373b74SJames Wright   Ceed     ceed = honee->ceed;
2583678fae3SJames Wright   MPI_Comm comm = PetscObjectComm((PetscObject)ts);
2593678fae3SJames Wright 
260926a6279SJames Wright   PetscFunctionBeginUser;
261926a6279SJames Wright   // Header and rank
262926a6279SJames Wright   char        host_name[PETSC_MAX_PATH_LEN];
263926a6279SJames Wright   PetscMPIInt rank, comm_size;
264926a6279SJames Wright   PetscCall(PetscGetHostName(host_name, sizeof host_name));
265926a6279SJames Wright   PetscCallMPI(MPI_Comm_rank(comm, &rank));
266926a6279SJames Wright   PetscCallMPI(MPI_Comm_size(comm, &comm_size));
267926a6279SJames Wright   PetscCall(PetscPrintf(comm,
268926a6279SJames Wright                         "\n-- Navier-Stokes solver - libCEED + PETSc --\n"
269926a6279SJames Wright                         "  MPI:\n"
270926a6279SJames Wright                         "    Host Name                          : %s\n"
271926a6279SJames Wright                         "    Total ranks                        : %d\n",
272926a6279SJames Wright                         host_name, comm_size));
273926a6279SJames Wright 
274926a6279SJames Wright   // Problem specific info
2750c373b74SJames Wright   PetscCall(problem->print_info(honee, problem, honee->app_ctx));
276926a6279SJames Wright 
277926a6279SJames Wright   // libCEED
278926a6279SJames Wright   const char *used_resource;
279926a6279SJames Wright   CeedMemType mem_type_backend;
2800c373b74SJames Wright   PetscCallCeed(ceed, CeedGetResource(honee->ceed, &used_resource));
2810c373b74SJames Wright   PetscCallCeed(ceed, CeedGetPreferredMemType(honee->ceed, &mem_type_backend));
282926a6279SJames Wright   PetscCall(PetscPrintf(comm,
283926a6279SJames Wright                         "  libCEED:\n"
284926a6279SJames Wright                         "    libCEED Backend                    : %s\n"
285926a6279SJames Wright                         "    libCEED Backend MemType            : %s\n",
286926a6279SJames Wright                         used_resource, CeedMemTypes[mem_type_backend]));
287926a6279SJames Wright   // PETSc
28866d54740SJames Wright   {
2893678fae3SJames Wright     VecType  vec_type;
290926a6279SJames Wright     char     box_faces_str[PETSC_MAX_PATH_LEN] = "3,3,3";
29166d54740SJames Wright     PetscInt dim;
29266d54740SJames Wright 
2930c373b74SJames Wright     PetscCall(DMGetDimension(honee->dm, &dim));
29466d54740SJames Wright     if (dim == 2) box_faces_str[3] = '\0';
295926a6279SJames Wright     PetscCall(PetscOptionsGetString(NULL, NULL, "-dm_plex_box_faces", box_faces_str, sizeof(box_faces_str), NULL));
2960c373b74SJames Wright     PetscCall(DMGetVecType(honee->dm, &vec_type));
297926a6279SJames Wright     PetscCall(PetscPrintf(comm,
298926a6279SJames Wright                           "  PETSc:\n"
299926a6279SJames Wright                           "    Box Faces                          : %s\n"
300926a6279SJames Wright                           "    DM VecType                         : %s\n"
301926a6279SJames Wright                           "    Time Stepping Scheme               : %s\n",
3023678fae3SJames Wright                           box_faces_str, vec_type, phys_ctx->implicit ? "implicit" : "explicit"));
30366d54740SJames Wright   }
3043678fae3SJames Wright   {
3053678fae3SJames Wright     char           pmat_type_str[PETSC_MAX_PATH_LEN];
3063678fae3SJames Wright     MatType        amat_type, pmat_type;
3073678fae3SJames Wright     Mat            Amat, Pmat;
3083678fae3SJames Wright     TSIJacobianFn *ijacob_function;
3093678fae3SJames Wright 
3103678fae3SJames Wright     PetscCall(TSGetIJacobian(ts, &Amat, &Pmat, &ijacob_function, NULL));
3113678fae3SJames Wright     PetscCall(MatGetType(Amat, &amat_type));
3123678fae3SJames Wright     PetscCall(MatGetType(Pmat, &pmat_type));
3133678fae3SJames Wright 
3143678fae3SJames Wright     PetscCall(PetscStrncpy(pmat_type_str, pmat_type, sizeof(pmat_type_str)));
3153678fae3SJames Wright     if (!strcmp(pmat_type, MATCEED)) {
3163678fae3SJames Wright       MatType pmat_coo_type;
3173678fae3SJames Wright       char    pmat_coo_type_str[PETSC_MAX_PATH_LEN];
3183678fae3SJames Wright 
3193678fae3SJames Wright       PetscCall(MatCeedGetCOOMatType(Pmat, &pmat_coo_type));
3203678fae3SJames Wright       PetscCall(PetscSNPrintf(pmat_coo_type_str, sizeof(pmat_coo_type_str), " (COO MatType: %s)", pmat_coo_type));
3213678fae3SJames Wright       PetscCall(PetscStrlcat(pmat_type_str, pmat_coo_type_str, sizeof(pmat_type_str)));
3223678fae3SJames Wright     }
3233678fae3SJames Wright     if (ijacob_function) {
3243678fae3SJames Wright       PetscCall(PetscPrintf(comm,
3253678fae3SJames Wright                             "    IJacobian A MatType                : %s\n"
3263678fae3SJames Wright                             "    IJacobian P MatType                : %s\n",
3273678fae3SJames Wright                             amat_type, pmat_type_str));
3283678fae3SJames Wright     }
3293678fae3SJames Wright   }
3300c373b74SJames Wright   if (honee->app_ctx->cont_steps) {
331926a6279SJames Wright     PetscCall(PetscPrintf(comm,
332926a6279SJames Wright                           "  Continue:\n"
333926a6279SJames Wright                           "    Filename:                          : %s\n"
334926a6279SJames Wright                           "    Step:                              : %" PetscInt_FMT "\n"
335926a6279SJames Wright                           "    Time:                              : %g\n",
3360c373b74SJames Wright                           honee->app_ctx->cont_file, honee->app_ctx->cont_steps, honee->app_ctx->cont_time));
337926a6279SJames Wright   }
338926a6279SJames Wright   // Mesh
339926a6279SJames Wright   const PetscInt num_comp_q = 5;
340926a6279SJames Wright   PetscInt       glob_dofs, owned_dofs, local_dofs;
3410c373b74SJames Wright   const CeedInt  num_P = honee->app_ctx->degree + 1, num_Q = num_P + honee->app_ctx->q_extra;
3420c373b74SJames Wright   PetscCall(DMGetGlobalVectorInfo(honee->dm, &owned_dofs, &glob_dofs, NULL));
3430c373b74SJames Wright   PetscCall(DMGetLocalVectorInfo(honee->dm, &local_dofs, NULL, NULL));
344926a6279SJames Wright   PetscCall(PetscPrintf(comm,
345926a6279SJames Wright                         "  Mesh:\n"
346926a6279SJames Wright                         "    Number of 1D Basis Nodes (P)       : %" CeedInt_FMT "\n"
347926a6279SJames Wright                         "    Number of 1D Quadrature Points (Q) : %" CeedInt_FMT "\n"
348926a6279SJames Wright                         "    Global DoFs                        : %" PetscInt_FMT "\n"
349926a6279SJames Wright                         "    DoFs per node                      : %" PetscInt_FMT "\n"
350dfeb939dSJames Wright                         "    Global %" PetscInt_FMT "-DoF nodes                 : %" PetscInt_FMT "\n",
351dfeb939dSJames Wright                         num_P, num_Q, glob_dofs, num_comp_q, num_comp_q, glob_dofs / num_comp_q));
352926a6279SJames Wright   // -- Get Partition Statistics
353926a6279SJames Wright   PetscCall(PetscPrintf(comm, "  Partition:                             (min,max,median,max/median)\n"));
354926a6279SJames Wright   {
355926a6279SJames Wright     PetscInt *gather_buffer = NULL;
356dfeb939dSJames Wright     PetscInt  part_owned_dofs[3], part_local_dofs[3], part_boundary_dofs[3], part_neighbors[3];
357926a6279SJames Wright     PetscInt  median_index = comm_size % 2 ? comm_size / 2 : comm_size / 2 - 1;
358926a6279SJames Wright     if (!rank) PetscCall(PetscMalloc1(comm_size, &gather_buffer));
359926a6279SJames Wright 
360dfeb939dSJames Wright     PetscCallMPI(MPI_Gather(&owned_dofs, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm));
361926a6279SJames Wright     if (!rank) {
362926a6279SJames Wright       PetscCall(PetscSortInt(comm_size, gather_buffer));
363926a6279SJames Wright       part_owned_dofs[0]             = gather_buffer[0];              // min
364926a6279SJames Wright       part_owned_dofs[1]             = gather_buffer[comm_size - 1];  // max
365926a6279SJames Wright       part_owned_dofs[2]             = gather_buffer[median_index];   // median
366926a6279SJames Wright       PetscReal part_owned_dof_ratio = (PetscReal)part_owned_dofs[1] / (PetscReal)part_owned_dofs[2];
367dfeb939dSJames Wright       PetscCall(PetscPrintf(
368dfeb939dSJames Wright           comm, "    Global Vector %" PetscInt_FMT "-DoF nodes          : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n", num_comp_q,
369926a6279SJames Wright           part_owned_dofs[0] / num_comp_q, part_owned_dofs[1] / num_comp_q, part_owned_dofs[2] / num_comp_q, part_owned_dof_ratio));
370926a6279SJames Wright     }
371926a6279SJames Wright 
372dfeb939dSJames Wright     PetscCallMPI(MPI_Gather(&local_dofs, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm));
373926a6279SJames Wright     if (!rank) {
374926a6279SJames Wright       PetscCall(PetscSortInt(comm_size, gather_buffer));
375926a6279SJames Wright       part_local_dofs[0]             = gather_buffer[0];              // min
376926a6279SJames Wright       part_local_dofs[1]             = gather_buffer[comm_size - 1];  // max
377926a6279SJames Wright       part_local_dofs[2]             = gather_buffer[median_index];   // median
378926a6279SJames Wright       PetscReal part_local_dof_ratio = (PetscReal)part_local_dofs[1] / (PetscReal)part_local_dofs[2];
379dfeb939dSJames Wright       PetscCall(PetscPrintf(
380dfeb939dSJames Wright           comm, "    Local Vector %" PetscInt_FMT "-DoF nodes           : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n", num_comp_q,
381926a6279SJames Wright           part_local_dofs[0] / num_comp_q, part_local_dofs[1] / num_comp_q, part_local_dofs[2] / num_comp_q, part_local_dof_ratio));
382926a6279SJames Wright     }
383926a6279SJames Wright 
38445abf96eSJames Wright     if (comm_size != 1) {
385dfeb939dSJames Wright       PetscInt num_remote_roots_total = 0, num_remote_leaves_total = 0, num_ghost_interface_ranks = 0, num_owned_interface_ranks = 0;
386926a6279SJames Wright       {
387926a6279SJames Wright         PetscSF            sf;
3883bbbcc04SJames Wright         PetscMPIInt        nrranks, niranks;
389dfeb939dSJames Wright         const PetscInt    *roffset, *rmine, *rremote, *ioffset, *irootloc;
390dfeb939dSJames Wright         const PetscMPIInt *rranks, *iranks;
3910c373b74SJames Wright         PetscCall(DMGetSectionSF(honee->dm, &sf));
392926a6279SJames Wright         PetscCall(PetscSFGetRootRanks(sf, &nrranks, &rranks, &roffset, &rmine, &rremote));
393dfeb939dSJames Wright         PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, &ioffset, &irootloc));
394926a6279SJames Wright         for (PetscInt i = 0; i < nrranks; i++) {
395926a6279SJames Wright           if (rranks[i] == rank) continue;  // Ignore same-part global->local transfers
396926a6279SJames Wright           num_remote_roots_total += roffset[i + 1] - roffset[i];
397dfeb939dSJames Wright           num_ghost_interface_ranks++;
398dfeb939dSJames Wright         }
399dfeb939dSJames Wright         for (PetscInt i = 0; i < niranks; i++) {
400dfeb939dSJames Wright           if (iranks[i] == rank) continue;
401dfeb939dSJames Wright           num_remote_leaves_total += ioffset[i + 1] - ioffset[i];
402dfeb939dSJames Wright           num_owned_interface_ranks++;
403926a6279SJames Wright         }
404926a6279SJames Wright       }
405dfeb939dSJames Wright       PetscCallMPI(MPI_Gather(&num_remote_roots_total, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm));
406926a6279SJames Wright       if (!rank) {
407926a6279SJames Wright         PetscCall(PetscSortInt(comm_size, gather_buffer));
408dfeb939dSJames Wright         part_boundary_dofs[0]           = gather_buffer[0];              // min
409dfeb939dSJames Wright         part_boundary_dofs[1]           = gather_buffer[comm_size - 1];  // max
410dfeb939dSJames Wright         part_boundary_dofs[2]           = gather_buffer[median_index];   // median
411dfeb939dSJames Wright         PetscReal part_shared_dof_ratio = (PetscReal)part_boundary_dofs[1] / (PetscReal)part_boundary_dofs[2];
412dfeb939dSJames Wright         PetscCall(PetscPrintf(
41345abf96eSJames Wright             comm, "    Ghost Interface %" PetscInt_FMT "-DoF nodes        : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n",
41445abf96eSJames Wright             num_comp_q, part_boundary_dofs[0] / num_comp_q, part_boundary_dofs[1] / num_comp_q, part_boundary_dofs[2] / num_comp_q,
41545abf96eSJames Wright             part_shared_dof_ratio));
416dfeb939dSJames Wright       }
417dfeb939dSJames Wright 
418dfeb939dSJames Wright       PetscCallMPI(MPI_Gather(&num_ghost_interface_ranks, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm));
419dfeb939dSJames Wright       if (!rank) {
420dfeb939dSJames Wright         PetscCall(PetscSortInt(comm_size, gather_buffer));
421dfeb939dSJames Wright         part_neighbors[0]              = gather_buffer[0];              // min
422dfeb939dSJames Wright         part_neighbors[1]              = gather_buffer[comm_size - 1];  // max
423dfeb939dSJames Wright         part_neighbors[2]              = gather_buffer[median_index];   // median
424dfeb939dSJames Wright         PetscReal part_neighbors_ratio = (PetscReal)part_neighbors[1] / (PetscReal)part_neighbors[2];
425dfeb939dSJames Wright         PetscCall(PetscPrintf(comm, "    Ghost Interface Ranks              : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n",
426dfeb939dSJames Wright                               part_neighbors[0], part_neighbors[1], part_neighbors[2], part_neighbors_ratio));
427dfeb939dSJames Wright       }
428dfeb939dSJames Wright 
429dfeb939dSJames Wright       PetscCallMPI(MPI_Gather(&num_remote_leaves_total, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm));
430dfeb939dSJames Wright       if (!rank) {
431dfeb939dSJames Wright         PetscCall(PetscSortInt(comm_size, gather_buffer));
432dfeb939dSJames Wright         part_boundary_dofs[0]           = gather_buffer[0];              // min
433dfeb939dSJames Wright         part_boundary_dofs[1]           = gather_buffer[comm_size - 1];  // max
434dfeb939dSJames Wright         part_boundary_dofs[2]           = gather_buffer[median_index];   // median
435dfeb939dSJames Wright         PetscReal part_shared_dof_ratio = (PetscReal)part_boundary_dofs[1] / (PetscReal)part_boundary_dofs[2];
436dfeb939dSJames Wright         PetscCall(PetscPrintf(
43745abf96eSJames Wright             comm, "    Owned Interface %" PetscInt_FMT "-DoF nodes        : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n",
43845abf96eSJames Wright             num_comp_q, part_boundary_dofs[0] / num_comp_q, part_boundary_dofs[1] / num_comp_q, part_boundary_dofs[2] / num_comp_q,
43945abf96eSJames Wright             part_shared_dof_ratio));
440dfeb939dSJames Wright       }
441dfeb939dSJames Wright 
442dfeb939dSJames Wright       PetscCallMPI(MPI_Gather(&num_owned_interface_ranks, 1, MPIU_INT, gather_buffer, 1, MPIU_INT, 0, comm));
443dfeb939dSJames Wright       if (!rank) {
444dfeb939dSJames Wright         PetscCall(PetscSortInt(comm_size, gather_buffer));
445dfeb939dSJames Wright         part_neighbors[0]              = gather_buffer[0];              // min
446dfeb939dSJames Wright         part_neighbors[1]              = gather_buffer[comm_size - 1];  // max
447dfeb939dSJames Wright         part_neighbors[2]              = gather_buffer[median_index];   // median
448dfeb939dSJames Wright         PetscReal part_neighbors_ratio = (PetscReal)part_neighbors[1] / (PetscReal)part_neighbors[2];
449dfeb939dSJames Wright         PetscCall(PetscPrintf(comm, "    Owned Interface Ranks              : %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ", %f\n",
450dfeb939dSJames Wright                               part_neighbors[0], part_neighbors[1], part_neighbors[2], part_neighbors_ratio));
451926a6279SJames Wright       }
45245abf96eSJames Wright     }
453926a6279SJames Wright 
454926a6279SJames Wright     if (!rank) PetscCall(PetscFree(gather_buffer));
455926a6279SJames Wright   }
456926a6279SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
457926a6279SJames Wright }
458