xref: /libCEED/examples/fluids/navierstokes.h (revision 8146534cc2e0b7c8bbf1d8cfbea1d0cb64188b7c)
1 // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3 //
4 // SPDX-License-Identifier: BSD-2-Clause
5 //
6 // This file is part of CEED:  http://github.com/ceed
7 
8 #ifndef libceed_fluids_examples_navier_stokes_h
9 #define libceed_fluids_examples_navier_stokes_h
10 
11 #include <ceed.h>
12 #include <petscts.h>
13 #include <stdbool.h>
14 
15 #include "./include/petsc_ops.h"
16 #include "qfunctions/newtonian_types.h"
17 #include "qfunctions/stabilization_types.h"
18 
19 #if PETSC_VERSION_LT(3, 20, 0)
20 #error "PETSc v3.20 or later is required"
21 #endif
22 
23 #if PETSC_VERSION_LT(3, 21, 0)
24 #define DMSetCoordinateDisc(a, b, c) DMProjectCoordinates(a, b)
25 #endif
26 
27 #define PetscCallCeed(ceed, ...)                                    \
28   do {                                                              \
29     int ierr = __VA_ARGS__;                                         \
30     if (ierr != CEED_ERROR_SUCCESS) {                               \
31       const char *error_message;                                    \
32       CeedGetErrorMessage(ceed, &error_message);                    \
33       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "%s", error_message); \
34     }                                                               \
35   } while (0)
36 
37 // -----------------------------------------------------------------------------
38 // Enums
39 // -----------------------------------------------------------------------------
40 // Translate PetscMemType to CeedMemType
41 static inline CeedMemType MemTypeP2C(PetscMemType mem_type) { return PetscMemTypeDevice(mem_type) ? CEED_MEM_DEVICE : CEED_MEM_HOST; }
42 
43 // Euler - test cases
44 typedef enum {
45   EULER_TEST_ISENTROPIC_VORTEX = 0,
46   EULER_TEST_1                 = 1,
47   EULER_TEST_2                 = 2,
48   EULER_TEST_3                 = 3,
49   EULER_TEST_4                 = 4,
50   EULER_TEST_5                 = 5,
51 } EulerTestType;
52 static const char *const EulerTestTypes[] = {"isentropic_vortex", "test_1",      "test_2", "test_3", "test_4", "test_5",
53                                              "EulerTestType",     "EULER_TEST_", NULL};
54 
55 // Advection - Wind types
56 static const char *const WindTypes[] = {"rotation", "translation", "WindType", "WIND_", NULL};
57 
58 // Advection - Bubble Types
59 static const char *const BubbleTypes[] = {"sphere", "cylinder", "BubbleType", "BUBBLE_", NULL};
60 
61 // Advection - Bubble Continuity Types
62 static const char *const BubbleContinuityTypes[] = {"smooth", "back_sharp", "thick", "BubbleContinuityType", "BUBBLE_CONTINUITY_", NULL};
63 
64 // Stabilization methods
65 static const char *const StabilizationTypes[] = {"none", "SU", "SUPG", "StabilizationType", "STAB_", NULL};
66 
67 // Test mode type
68 typedef enum {
69   TESTTYPE_NONE           = 0,
70   TESTTYPE_SOLVER         = 1,
71   TESTTYPE_TURB_SPANSTATS = 2,
72   TESTTYPE_DIFF_FILTER    = 3,
73 } TestType;
74 static const char *const TestTypes[] = {"none", "solver", "turb_spanstats", "diff_filter", "TestType", "TESTTYPE_", NULL};
75 
76 // Subgrid-Stress mode type
77 typedef enum {
78   SGS_MODEL_NONE        = 0,
79   SGS_MODEL_DATA_DRIVEN = 1,
80 } SGSModelType;
81 static const char *const SGSModelTypes[] = {"none", "data_driven", "SGSModelType", "SGS_MODEL_", NULL};
82 
83 // Mesh transformation type
84 typedef enum {
85   MESH_TRANSFORM_NONE      = 0,
86   MESH_TRANSFORM_PLATEMESH = 1,
87 } MeshTransformType;
88 static const char *const MeshTransformTypes[] = {"none", "platemesh", "MeshTransformType", "MESH_TRANSFORM_", NULL};
89 
90 static const char *const DifferentialFilterDampingFunctions[] = {
91     "none", "van_driest", "mms", "DifferentialFilterDampingFunction", "DIFF_FILTER_DAMP_", NULL};
92 
93 // -----------------------------------------------------------------------------
94 // Log Events
95 // -----------------------------------------------------------------------------
96 extern PetscLogEvent FLUIDS_CeedOperatorApply;
97 extern PetscLogEvent FLUIDS_CeedOperatorAssemble;
98 extern PetscLogEvent FLUIDS_CeedOperatorAssembleDiagonal;
99 extern PetscLogEvent FLUIDS_CeedOperatorAssemblePointBlockDiagonal;
100 PetscErrorCode       RegisterLogEvents();
101 
102 // -----------------------------------------------------------------------------
103 // Structs
104 // -----------------------------------------------------------------------------
105 // Structs declarations
106 typedef struct AppCtx_private   *AppCtx;
107 typedef struct CeedData_private *CeedData;
108 typedef struct User_private     *User;
109 typedef struct Units_private    *Units;
110 typedef struct SimpleBC_private *SimpleBC;
111 typedef struct Physics_private  *Physics;
112 
113 // Application context from user command line options
114 struct AppCtx_private {
115   // libCEED arguments
116   char     ceed_resource[PETSC_MAX_PATH_LEN];  // libCEED backend
117   PetscInt degree;
118   PetscInt q_extra;
119   // Solver arguments
120   MatType   amat_type;
121   PetscBool pmat_pbdiagonal;
122   // Post-processing arguments
123   PetscInt  checkpoint_interval;
124   PetscInt  viz_refine;
125   PetscInt  cont_steps;
126   PetscReal cont_time;
127   char      cont_file[PETSC_MAX_PATH_LEN];
128   char      cont_time_file[PETSC_MAX_PATH_LEN];
129   char      output_dir[PETSC_MAX_PATH_LEN];
130   PetscBool add_stepnum2bin;
131   PetscBool checkpoint_vtk;
132   // Problem type arguments
133   PetscFunctionList problems;
134   char              problem_name[PETSC_MAX_PATH_LEN];
135   // Test mode arguments
136   TestType    test_type;
137   PetscScalar test_tol;
138   char        test_file_path[PETSC_MAX_PATH_LEN];
139   // Turbulent spanwise statistics
140   PetscBool         turb_spanstats_enable;
141   PetscInt          turb_spanstats_collect_interval;
142   PetscInt          turb_spanstats_viewer_interval;
143   PetscViewer       turb_spanstats_viewer;
144   PetscViewerFormat turb_spanstats_viewer_format;
145   // Wall forces
146   struct {
147     PetscInt          num_wall;
148     PetscInt         *walls;
149     PetscViewer       viewer;
150     PetscViewerFormat viewer_format;
151     PetscBool         header_written;
152   } wall_forces;
153   // Subgrid Stress Model
154   SGSModelType sgs_model_type;
155   // Differential Filtering
156   PetscBool         diff_filter_monitor;
157   MeshTransformType mesh_transform_type;
158 };
159 
160 // libCEED data struct
161 struct CeedData_private {
162   CeedVector           x_coord, q_data;
163   CeedBasis            basis_x, basis_xc, basis_q, basis_x_sur, basis_q_sur, basis_xc_sur;
164   CeedElemRestriction  elem_restr_x, elem_restr_q, elem_restr_qd_i;
165   CeedOperator         op_setup_vol;
166   OperatorApplyContext op_ics_ctx;
167   CeedQFunction        qf_setup_vol, qf_ics, qf_rhs_vol, qf_ifunction_vol, qf_setup_sur, qf_apply_inflow, qf_apply_inflow_jacobian, qf_apply_outflow,
168       qf_apply_outflow_jacobian, qf_apply_freestream, qf_apply_freestream_jacobian;
169 };
170 
171 typedef struct {
172   DM                    dm;
173   PetscSF               sf;  // For communicating child data to parents
174   OperatorApplyContext  op_stats_collect_ctx, op_proj_rhs_ctx;
175   PetscInt              num_comp_stats;
176   Vec                   Child_Stats_loc, Parent_Stats_loc;
177   KSP                   ksp;         // For the L^2 projection solve
178   CeedScalar            span_width;  // spanwise width of the child domain
179   PetscBool             do_mms_test;
180   OperatorApplyContext  mms_error_ctx;
181   CeedContextFieldLabel solution_time_label, previous_time_label;
182 } SpanStatsData;
183 
184 typedef struct {
185   DM                   dm;
186   PetscInt             num_comp;
187   OperatorApplyContext l2_rhs_ctx;
188   KSP                  ksp;
189 } *NodalProjectionData;
190 
191 typedef struct {
192   DM                   dm_sgs;
193   PetscInt             num_comp_sgs;
194   OperatorApplyContext op_nodal_evaluation_ctx, op_sgs_apply_ctx;
195   CeedVector           sgs_nodal_ceed;
196 } *SgsDDData;
197 
198 typedef struct {
199   DM                   dm_filter;
200   PetscInt             num_filtered_fields;
201   CeedInt             *num_field_components;
202   PetscInt             field_prim_state, field_velo_prod;
203   OperatorApplyContext op_rhs_ctx;
204   KSP                  ksp;
205   PetscBool            do_mms_test;
206 } *DiffFilterData;
207 
208 typedef struct {
209   void    *client;
210   char     rank_id_name[16];
211   PetscInt collocated_database_num_ranks;
212 } *SmartSimData;
213 
214 // PETSc user data
215 struct User_private {
216   MPI_Comm             comm;
217   DM                   dm;
218   DM                   dm_viz;
219   Mat                  interp_viz;
220   Ceed                 ceed;
221   Units                units;
222   Vec                  M_inv, Q_loc, Q_dot_loc;
223   Physics              phys;
224   AppCtx               app_ctx;
225   CeedVector           q_ceed, q_dot_ceed, g_ceed, coo_values_amat, coo_values_pmat, x_ceed;
226   CeedOperator         op_rhs_vol, op_ifunction_vol, op_ifunction, op_ijacobian;
227   OperatorApplyContext op_rhs_ctx, op_strong_bc_ctx;
228   bool                 matrices_set_up;
229   CeedScalar           time_bc_set;
230   SpanStatsData        spanstats;
231   NodalProjectionData  grad_velo_proj;
232   SgsDDData            sgs_dd_data;
233   DiffFilterData       diff_filter;
234   SmartSimData         smartsim;
235 };
236 
237 // Units
238 struct Units_private {
239   // fundamental units
240   PetscScalar meter;
241   PetscScalar kilogram;
242   PetscScalar second;
243   PetscScalar Kelvin;
244   // derived units
245   PetscScalar Pascal;
246   PetscScalar J_per_kg_K;
247   PetscScalar m_per_squared_s;
248   PetscScalar W_per_m_K;
249   PetscScalar Joule;
250 };
251 
252 // Boundary conditions
253 struct SimpleBC_private {
254   PetscInt num_wall,  // Number of faces with wall BCs
255       wall_comps[5],  // An array of constrained component numbers
256       num_comps,
257       num_slip[3],  // Number of faces with slip BCs
258       num_inflow, num_outflow, num_freestream;
259   PetscInt  walls[16], slips[3][16], inflows[16], outflows[16], freestreams[16];
260   PetscBool user_bc;
261 };
262 
263 // Struct that contains all enums and structs used for the physics of all problems
264 struct Physics_private {
265   PetscBool             implicit;
266   StateVariable         state_var;
267   CeedContextFieldLabel solution_time_label;
268   CeedContextFieldLabel stg_solution_time_label;
269   CeedContextFieldLabel timestep_size_label;
270   CeedContextFieldLabel ics_time_label;
271   CeedContextFieldLabel ijacobian_time_shift_label;
272 };
273 
274 typedef struct {
275   CeedQFunctionUser    qfunction;
276   const char          *qfunction_loc;
277   CeedQFunctionContext qfunction_context;
278 } ProblemQFunctionSpec;
279 
280 // Problem specific data
281 typedef struct ProblemData_private ProblemData;
282 struct ProblemData_private {
283   CeedInt              dim, q_data_size_vol, q_data_size_sur, jac_data_size_sur;
284   CeedScalar           dm_scale;
285   ProblemQFunctionSpec setup_vol, setup_sur, ics, apply_vol_rhs, apply_vol_ifunction, apply_vol_ijacobian, apply_inflow, apply_outflow,
286       apply_freestream, apply_inflow_jacobian, apply_outflow_jacobian, apply_freestream_jacobian;
287   bool      non_zero_time;
288   PetscBool bc_from_ics, use_strong_bc_ceed;
289   PetscErrorCode (*print_info)(User, ProblemData *, AppCtx);
290 };
291 
292 extern int FreeContextPetsc(void *);
293 
294 // -----------------------------------------------------------------------------
295 // Set up problems
296 // -----------------------------------------------------------------------------
297 // Set up function for each problem
298 extern PetscErrorCode NS_TAYLOR_GREEN(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
299 extern PetscErrorCode NS_GAUSSIAN_WAVE(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
300 extern PetscErrorCode NS_CHANNEL(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
301 extern PetscErrorCode NS_BLASIUS(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
302 extern PetscErrorCode NS_NEWTONIAN_IG(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
303 extern PetscErrorCode NS_DENSITY_CURRENT(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
304 extern PetscErrorCode NS_EULER_VORTEX(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
305 extern PetscErrorCode NS_SHOCKTUBE(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
306 extern PetscErrorCode NS_ADVECTION(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
307 extern PetscErrorCode NS_ADVECTION2D(ProblemData *problem, DM dm, void *ctx, SimpleBC bc);
308 
309 // Print function for each problem
310 extern PetscErrorCode PRINT_NEWTONIAN(User user, ProblemData *problem, AppCtx app_ctx);
311 
312 extern PetscErrorCode PRINT_EULER_VORTEX(User user, ProblemData *problem, AppCtx app_ctx);
313 
314 extern PetscErrorCode PRINT_SHOCKTUBE(User user, ProblemData *problem, AppCtx app_ctx);
315 
316 extern PetscErrorCode PRINT_ADVECTION(User user, ProblemData *problem, AppCtx app_ctx);
317 
318 extern PetscErrorCode PRINT_ADVECTION2D(User user, ProblemData *problem, AppCtx app_ctx);
319 
320 PetscErrorCode PrintRunInfo(User user, Physics phys_ctx, ProblemData *problem, MPI_Comm comm);
321 
322 // -----------------------------------------------------------------------------
323 // libCEED functions
324 // -----------------------------------------------------------------------------
325 // Utility function to create local CEED restriction
326 PetscErrorCode CreateRestrictionFromPlex(Ceed ceed, DM dm, CeedInt height, DMLabel domain_label, CeedInt label_value, PetscInt dm_field,
327                                          CeedElemRestriction *elem_restr);
328 
329 PetscErrorCode DMPlexCeedElemRestrictionCreate(Ceed ceed, DM dm, DMLabel domain_label, PetscInt label_value, PetscInt height, PetscInt dm_field,
330                                                CeedElemRestriction *restriction);
331 PetscErrorCode DMPlexCeedElemRestrictionCoordinateCreate(Ceed ceed, DM dm, DMLabel domain_label, PetscInt label_value, PetscInt height,
332                                                          CeedElemRestriction *restriction);
333 PetscErrorCode DMPlexCeedElemRestrictionQDataCreate(Ceed ceed, DM dm, DMLabel domain_label, PetscInt label_value, PetscInt height,
334                                                     PetscInt q_data_size, CeedElemRestriction *restriction);
335 PetscErrorCode DMPlexCeedElemRestrictionCollocatedCreate(Ceed ceed, DM dm, DMLabel domain_label, PetscInt label_value, PetscInt height,
336                                                          PetscInt q_data_size, CeedElemRestriction *restriction);
337 
338 PetscErrorCode CreateBasisFromPlex(Ceed ceed, DM dm, DMLabel domain_label, CeedInt label_value, CeedInt height, CeedInt dm_field, CeedBasis *basis);
339 
340 // Utility function to create CEED Composite Operator for the entire domain
341 PetscErrorCode CreateOperatorForDomain(Ceed ceed, DM dm, SimpleBC bc, CeedData ceed_data, Physics phys, CeedOperator op_apply_vol,
342                                        CeedOperator op_apply_ijacobian_vol, CeedInt height, CeedInt P_sur, CeedInt Q_sur, CeedInt q_data_size_sur,
343                                        CeedInt jac_data_size_sur, CeedOperator *op_apply, CeedOperator *op_apply_ijacobian);
344 
345 PetscErrorCode SetupLibceed(Ceed ceed, CeedData ceed_data, DM dm, User user, AppCtx app_ctx, ProblemData *problem, SimpleBC bc);
346 
347 // -----------------------------------------------------------------------------
348 // Time-stepping functions
349 // -----------------------------------------------------------------------------
350 // Compute mass matrix for explicit scheme
351 PetscErrorCode ComputeLumpedMassMatrix(Ceed ceed, DM dm, CeedData ceed_data, Vec M);
352 
353 // RHS (Explicit time-stepper) function setup
354 PetscErrorCode RHS_NS(TS ts, PetscReal t, Vec Q, Vec G, void *user_data);
355 
356 // Implicit time-stepper function setup
357 PetscErrorCode IFunction_NS(TS ts, PetscReal t, Vec Q, Vec Q_dot, Vec G, void *user_data);
358 
359 // User provided TS Monitor
360 PetscErrorCode TSMonitor_NS(TS ts, PetscInt step_no, PetscReal time, Vec Q, void *ctx);
361 
362 // TS: Create, setup, and solve
363 PetscErrorCode TSSolve_NS(DM dm, User user, AppCtx app_ctx, Physics phys, Vec *Q, PetscScalar *f_time, TS *ts);
364 
365 // Update Boundary Values when time has changed
366 PetscErrorCode UpdateBoundaryValues(User user, Vec Q_loc, PetscReal t);
367 
368 // -----------------------------------------------------------------------------
369 // Setup DM
370 // -----------------------------------------------------------------------------
371 // Create mesh
372 PetscErrorCode CreateDM(MPI_Comm comm, ProblemData *problem, MatType, VecType, DM *dm);
373 
374 // Set up DM
375 PetscErrorCode SetUpDM(DM dm, ProblemData *problem, PetscInt degree, PetscInt q_extra, SimpleBC bc, Physics phys);
376 PetscErrorCode DMSetupByOrderBegin_FEM(PetscBool setup_faces, PetscBool setup_coords, PetscInt degree, PetscInt coord_order, PetscInt q_extra,
377                                        PetscInt num_fields, const PetscInt *field_sizes, DM dm);
378 PetscErrorCode DMSetupByOrderEnd_FEM(PetscBool setup_coords, DM dm);
379 PetscErrorCode DMSetupByOrder_FEM(PetscBool setup_faces, PetscBool setup_coords, PetscInt degree, PetscInt coord_order, PetscInt q_extra,
380                                   PetscInt num_fields, const PetscInt *field_sizes, DM dm);
381 
382 // Refine DM for high-order viz
383 PetscErrorCode VizRefineDM(DM dm, User user, ProblemData *problem, SimpleBC bc, Physics phys);
384 
385 // -----------------------------------------------------------------------------
386 // Process command line options
387 // -----------------------------------------------------------------------------
388 // Register problems to be available on the command line
389 PetscErrorCode RegisterProblems_NS(AppCtx app_ctx);
390 
391 // Process general command line options
392 PetscErrorCode ProcessCommandLineOptions(MPI_Comm comm, AppCtx app_ctx, SimpleBC bc);
393 
394 // -----------------------------------------------------------------------------
395 // Miscellaneous utility functions
396 // -----------------------------------------------------------------------------
397 PetscErrorCode ICs_FixMultiplicity(DM dm, CeedData ceed_data, User user, Vec Q_loc, Vec Q, CeedScalar time);
398 
399 PetscErrorCode DMPlexInsertBoundaryValues_FromICs(DM dm, PetscBool insert_essential, Vec Q_loc, PetscReal time, Vec face_geom_FVM, Vec cell_geom_FVM,
400                                                   Vec grad_FVM);
401 
402 // Compare reference solution values with current test run for CI
403 PetscErrorCode RegressionTest(AppCtx app_ctx, Vec Q);
404 
405 // Get error for problems with exact solutions
406 PetscErrorCode PrintError(CeedData ceed_data, DM dm, User user, Vec Q, PetscScalar final_time);
407 
408 // Post-processing
409 PetscErrorCode PostProcess(TS ts, CeedData ceed_data, DM dm, ProblemData *problem, User user, Vec Q, PetscScalar final_time);
410 
411 // -- Gather initial Q values in case of continuation of simulation
412 PetscErrorCode SetupICsFromBinary(MPI_Comm comm, AppCtx app_ctx, Vec Q);
413 
414 // Record boundary values from initial condition
415 PetscErrorCode SetBCsFromICs(DM dm, Vec Q, Vec Q_loc);
416 
417 // Versioning token for binary checkpoints
418 extern const PetscInt32 FLUIDS_FILE_TOKEN;  // for backwards compatibility
419 extern const PetscInt32 FLUIDS_FILE_TOKEN_32;
420 extern const PetscInt32 FLUIDS_FILE_TOKEN_64;
421 
422 // Create appropriate mass qfunction based on number of components N
423 PetscErrorCode CreateMassQFunction(Ceed ceed, CeedInt N, CeedInt q_data_size, CeedQFunction *qf);
424 
425 PetscErrorCode NodalProjectionDataDestroy(NodalProjectionData context);
426 
427 PetscErrorCode PhastaDatFileOpen(const MPI_Comm comm, const char path[PETSC_MAX_PATH_LEN], const PetscInt char_array_len, PetscInt dims[2],
428                                  FILE **fp);
429 
430 PetscErrorCode PhastaDatFileGetNRows(const MPI_Comm comm, const char path[PETSC_MAX_PATH_LEN], PetscInt *nrows);
431 
432 PetscErrorCode PhastaDatFileReadToArrayReal(const MPI_Comm comm, const char path[PETSC_MAX_PATH_LEN], PetscReal array[]);
433 
434 PetscErrorCode IntArrayC2P(PetscInt num_entries, CeedInt **array_ceed, PetscInt **array_petsc);
435 PetscErrorCode IntArrayP2C(PetscInt num_entries, PetscInt **array_petsc, CeedInt **array_ceed);
436 
437 // -----------------------------------------------------------------------------
438 // Turbulence Statistics Collection Functions
439 // -----------------------------------------------------------------------------
440 
441 PetscErrorCode TurbulenceStatisticsSetup(Ceed ceed, User user, CeedData ceed_data, ProblemData *problem);
442 PetscErrorCode TSMonitor_TurbulenceStatistics(TS ts, PetscInt steps, PetscReal solution_time, Vec Q, void *ctx);
443 PetscErrorCode TurbulenceStatisticsDestroy(User user, CeedData ceed_data);
444 
445 // -----------------------------------------------------------------------------
446 // Data-Driven Subgrid Stress (DD-SGS) Modeling Functions
447 // -----------------------------------------------------------------------------
448 
449 PetscErrorCode SgsDDModelSetup(Ceed ceed, User user, CeedData ceed_data, ProblemData *problem);
450 PetscErrorCode SgsDDDataDestroy(SgsDDData sgs_dd_data);
451 PetscErrorCode SgsDDModelApplyIFunction(User user, const Vec Q_loc, Vec G_loc);
452 PetscErrorCode VelocityGradientProjectionSetup(Ceed ceed, User user, CeedData ceed_data, ProblemData *problem, StateVariable state_var_input,
453                                                CeedElemRestriction elem_restr_input, CeedBasis basis_input, NodalProjectionData *pgrad_velo_proj);
454 PetscErrorCode VelocityGradientProjectionApply(NodalProjectionData grad_velo_proj, Vec Q_loc, Vec VelocityGradient);
455 PetscErrorCode GridAnisotropyTensorProjectionSetupApply(Ceed ceed, User user, CeedData ceed_data, CeedElemRestriction *elem_restr_grid_aniso,
456                                                         CeedVector *grid_aniso_vector);
457 PetscErrorCode GridAnisotropyTensorCalculateCollocatedVector(Ceed ceed, User user, CeedData ceed_data, CeedElemRestriction *elem_restr_grid_aniso,
458                                                              CeedVector *aniso_colloc_ceed, PetscInt *num_comp_aniso);
459 
460 // -----------------------------------------------------------------------------
461 // Boundary Condition Related Functions
462 // -----------------------------------------------------------------------------
463 
464 // Setup StrongBCs that use QFunctions
465 PetscErrorCode SetupStrongBC_Ceed(Ceed ceed, CeedData ceed_data, DM dm, User user, ProblemData *problem, SimpleBC bc);
466 
467 PetscErrorCode FreestreamBCSetup(ProblemData *problem, DM dm, void *ctx, NewtonianIdealGasContext newtonian_ig_ctx, const StatePrimitive *reference);
468 PetscErrorCode OutflowBCSetup(ProblemData *problem, DM dm, void *ctx, NewtonianIdealGasContext newtonian_ig_ctx, const StatePrimitive *reference);
469 
470 // -----------------------------------------------------------------------------
471 // Differential Filtering Functions
472 // -----------------------------------------------------------------------------
473 
474 PetscErrorCode DifferentialFilterSetup(Ceed ceed, User user, CeedData ceed_data, ProblemData *problem);
475 PetscErrorCode DifferentialFilterDataDestroy(DiffFilterData diff_filter);
476 PetscErrorCode TSMonitor_DifferentialFilter(TS ts, PetscInt steps, PetscReal solution_time, Vec Q, void *ctx);
477 PetscErrorCode DifferentialFilterApply(User user, const PetscReal solution_time, const Vec Q, Vec Filtered_Solution);
478 PetscErrorCode DifferentialFilterMmsICSetup(ProblemData *problem);
479 
480 // -----------------------------------------------------------------------------
481 // SGS Data-Driven Training via SmartSim
482 // -----------------------------------------------------------------------------
483 PetscErrorCode SmartSimSetup(User user);
484 
485 #endif  // libceed_fluids_examples_navier_stokes_h
486