xref: /honee/problems/stg_shur14.c (revision b4c37c5c26bd208d7f474cbd9aa0850e82e1a854)
1493642f1SJames Wright // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2493642f1SJames Wright // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3493642f1SJames Wright //
4493642f1SJames Wright // SPDX-License-Identifier: BSD-2-Clause
5493642f1SJames Wright //
6493642f1SJames Wright // This file is part of CEED:  http://github.com/ceed
7493642f1SJames Wright 
8493642f1SJames Wright /// @file
9493642f1SJames Wright /// Implementation of the Synthetic Turbulence Generation (STG) algorithm
10493642f1SJames Wright /// presented in Shur et al. 2014
11493642f1SJames Wright 
122b916ea7SJeremy L Thompson #include "stg_shur14.h"
132b916ea7SJeremy L Thompson 
14e419654dSJeremy L Thompson #include <ceed.h>
15493642f1SJames Wright #include <math.h>
16e419654dSJeremy L Thompson #include <petscdm.h>
172b916ea7SJeremy L Thompson #include <stdlib.h>
182b916ea7SJeremy L Thompson 
19493642f1SJames Wright #include "../navierstokes.h"
20493642f1SJames Wright #include "../qfunctions/stg_shur14.h"
21493642f1SJames Wright 
228085925cSJames Wright STGShur14Context global_stg_ctx;
238085925cSJames Wright 
24493642f1SJames Wright /*
25493642f1SJames Wright  * @brief Perform Cholesky decomposition on array of symmetric 3x3 matrices
26493642f1SJames Wright  *
2704e40bb6SJeremy L Thompson  * This assumes the input matrices are in order [11,22,33,12,13,23].
2804e40bb6SJeremy L Thompson  * This format is also used for the output.
29493642f1SJames Wright  *
30493642f1SJames Wright  * @param[in]  comm   MPI_Comm
31493642f1SJames Wright  * @param[in]  nprofs Number of matrices in Rij
32493642f1SJames Wright  * @param[in]  Rij    Array of the symmetric matrices [6,nprofs]
33493642f1SJames Wright  * @param[out] Cij    Array of the Cholesky Decomposition matrices, [6,nprofs]
34493642f1SJames Wright  */
352b916ea7SJeremy L Thompson PetscErrorCode CalcCholeskyDecomp(MPI_Comm comm, PetscInt nprofs, const CeedScalar Rij[6][nprofs], CeedScalar Cij[6][nprofs]) {
36493642f1SJames Wright   PetscFunctionBeginUser;
37493642f1SJames Wright   for (PetscInt i = 0; i < nprofs; i++) {
38493642f1SJames Wright     Cij[0][i] = sqrt(Rij[0][i]);
39493642f1SJames Wright     Cij[3][i] = Rij[3][i] / Cij[0][i];
40493642f1SJames Wright     Cij[1][i] = sqrt(Rij[1][i] - pow(Cij[3][i], 2));
41493642f1SJames Wright     Cij[4][i] = Rij[4][i] / Cij[0][i];
42493642f1SJames Wright     Cij[5][i] = (Rij[5][i] - Cij[3][i] * Cij[4][i]) / Cij[1][i];
43493642f1SJames Wright     Cij[2][i] = sqrt(Rij[2][i] - pow(Cij[4][i], 2) - pow(Cij[5][i], 2));
44493642f1SJames Wright 
455d28dccaSJames Wright     PetscCheck(!isnan(Cij[0][i]) && !isnan(Cij[1][i]) && !isnan(Cij[2][i]), comm, PETSC_ERR_FP,
465d28dccaSJames Wright                "Cholesky decomposition failed at profile point %" PetscInt_FMT ". Either STGInflow has non-SPD matrix or contains nan.", i + 1);
47493642f1SJames Wright   }
48d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
49493642f1SJames Wright }
50493642f1SJames Wright 
51493642f1SJames Wright /*
52493642f1SJames Wright  * @brief Read the STGInflow file and load the contents into stg_ctx
53493642f1SJames Wright  *
5404e40bb6SJeremy L Thompson  * Assumes that the first line of the file has the number of rows and columns as the only two entries, separated by a single space.
5504e40bb6SJeremy L Thompson  * Assumes there are 14 columns in the file.
56493642f1SJames Wright  *
5704e40bb6SJeremy L Thompson  * Function calculates the Cholesky decomposition from the Reynolds stress profile found in the file.
58493642f1SJames Wright  *
59493642f1SJames Wright  * @param[in]     comm    MPI_Comm for the program
60493642f1SJames Wright  * @param[in]     path    Path to the STGInflow.dat file
6104e40bb6SJeremy L Thompson  * @param[in,out] stg_ctx STGShur14Context where the data will be loaded into
62493642f1SJames Wright  */
632b916ea7SJeremy L Thompson static PetscErrorCode ReadSTGInflow(const MPI_Comm comm, const char path[PETSC_MAX_PATH_LEN], STGShur14Context stg_ctx) {
64defe8520SJames Wright   PetscInt       dims[2];
65defe8520SJames Wright   int            ndims;
66493642f1SJames Wright   FILE          *fp;
67493642f1SJames Wright   const PetscInt char_array_len = 512;
68493642f1SJames Wright   char           line[char_array_len];
69493642f1SJames Wright   char         **array;
70493642f1SJames Wright 
71493642f1SJames Wright   PetscFunctionBeginUser;
72493642f1SJames Wright 
73676080b4SJames Wright   PetscCall(PHASTADatFileOpen(comm, path, char_array_len, dims, &fp));
74493642f1SJames Wright 
75493642f1SJames Wright   CeedScalar  rij[6][stg_ctx->nprofs];
76c77f3192SJames Wright   CeedScalar *wall_dist              = &stg_ctx->data[stg_ctx->offsets.wall_dist];
77493642f1SJames Wright   CeedScalar *eps                    = &stg_ctx->data[stg_ctx->offsets.eps];
78493642f1SJames Wright   CeedScalar *lt                     = &stg_ctx->data[stg_ctx->offsets.lt];
792b916ea7SJeremy L Thompson   CeedScalar(*ubar)[stg_ctx->nprofs] = (CeedScalar(*)[stg_ctx->nprofs]) & stg_ctx->data[stg_ctx->offsets.ubar];
80493642f1SJames Wright 
81493642f1SJames Wright   for (PetscInt i = 0; i < stg_ctx->nprofs; i++) {
8234b5deb1SJames Wright     PetscCall(PetscSynchronizedFGets(comm, fp, char_array_len, line));
8334b5deb1SJames Wright     PetscCall(PetscStrToArray(line, ' ', &ndims, &array));
845d28dccaSJames Wright     PetscCheck(ndims == dims[1], comm, PETSC_ERR_FILE_UNEXPECTED,
85defe8520SJames Wright                "Line %" PetscInt_FMT " of %s does not have correct number of columns (%d instead of %" PetscInt_FMT ")", i, path, ndims, dims[1]);
86493642f1SJames Wright 
87c77f3192SJames Wright     wall_dist[i] = (CeedScalar)atof(array[0]);
88493642f1SJames Wright     ubar[0][i]   = (CeedScalar)atof(array[1]);
89493642f1SJames Wright     ubar[1][i]   = (CeedScalar)atof(array[2]);
90493642f1SJames Wright     ubar[2][i]   = (CeedScalar)atof(array[3]);
91493642f1SJames Wright     rij[0][i]    = (CeedScalar)atof(array[4]);
92493642f1SJames Wright     rij[1][i]    = (CeedScalar)atof(array[5]);
93493642f1SJames Wright     rij[2][i]    = (CeedScalar)atof(array[6]);
94493642f1SJames Wright     rij[3][i]    = (CeedScalar)atof(array[7]);
95493642f1SJames Wright     rij[4][i]    = (CeedScalar)atof(array[8]);
96493642f1SJames Wright     rij[5][i]    = (CeedScalar)atof(array[9]);
97493642f1SJames Wright     lt[i]        = (CeedScalar)atof(array[12]);
98493642f1SJames Wright     eps[i]       = (CeedScalar)atof(array[13]);
99493642f1SJames Wright 
1005d28dccaSJames Wright     PetscCheck(wall_dist[i] >= 0, comm, PETSC_ERR_FILE_UNEXPECTED, "Distance to wall in %s cannot be negative", path);
1015d28dccaSJames Wright     PetscCheck(lt[i] >= 0, comm, PETSC_ERR_FILE_UNEXPECTED, "Turbulent length scale in %s cannot be negative", path);
1025d28dccaSJames Wright     PetscCheck(eps[i] >= 0, comm, PETSC_ERR_FILE_UNEXPECTED, "Turbulent dissipation in %s cannot be negative", path);
103493642f1SJames Wright   }
1042b916ea7SJeremy L Thompson   CeedScalar(*cij)[stg_ctx->nprofs] = (CeedScalar(*)[stg_ctx->nprofs]) & stg_ctx->data[stg_ctx->offsets.cij];
10534b5deb1SJames Wright   PetscCall(CalcCholeskyDecomp(comm, stg_ctx->nprofs, rij, cij));
10634b5deb1SJames Wright   PetscCall(PetscFClose(comm, fp));
1072b916ea7SJeremy L Thompson 
108d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
109493642f1SJames Wright }
110493642f1SJames Wright 
111493642f1SJames Wright /*
112493642f1SJames Wright  * @brief Read the STGRand file and load the contents into stg_ctx
113493642f1SJames Wright  *
11404e40bb6SJeremy L Thompson  * Assumes that the first line of the file has the number of rows and columns as the only two entries, separated by a single space.
11504e40bb6SJeremy L Thompson  * Assumes there are 7 columns in the file.
116493642f1SJames Wright  *
117493642f1SJames Wright  * @param[in]     comm    MPI_Comm for the program
118493642f1SJames Wright  * @param[in]     path    Path to the STGRand.dat file
11904e40bb6SJeremy L Thompson  * @param[in,out] stg_ctx STGShur14Context where the data will be loaded into
120493642f1SJames Wright  */
1212b916ea7SJeremy L Thompson static PetscErrorCode ReadSTGRand(const MPI_Comm comm, const char path[PETSC_MAX_PATH_LEN], STGShur14Context stg_ctx) {
122defe8520SJames Wright   PetscInt       dims[2];
123defe8520SJames Wright   int            ndims;
124493642f1SJames Wright   FILE          *fp;
125493642f1SJames Wright   const PetscInt char_array_len = 512;
126493642f1SJames Wright   char           line[char_array_len];
127493642f1SJames Wright   char         **array;
128493642f1SJames Wright 
129493642f1SJames Wright   PetscFunctionBeginUser;
130676080b4SJames Wright   PetscCall(PHASTADatFileOpen(comm, path, char_array_len, dims, &fp));
131493642f1SJames Wright 
132493642f1SJames Wright   CeedScalar *phi                     = &stg_ctx->data[stg_ctx->offsets.phi];
1332b916ea7SJeremy L Thompson   CeedScalar(*d)[stg_ctx->nmodes]     = (CeedScalar(*)[stg_ctx->nmodes]) & stg_ctx->data[stg_ctx->offsets.d];
1342b916ea7SJeremy L Thompson   CeedScalar(*sigma)[stg_ctx->nmodes] = (CeedScalar(*)[stg_ctx->nmodes]) & stg_ctx->data[stg_ctx->offsets.sigma];
135493642f1SJames Wright 
136493642f1SJames Wright   for (PetscInt i = 0; i < stg_ctx->nmodes; i++) {
13734b5deb1SJames Wright     PetscCall(PetscSynchronizedFGets(comm, fp, char_array_len, line));
13834b5deb1SJames Wright     PetscCall(PetscStrToArray(line, ' ', &ndims, &array));
1395d28dccaSJames Wright     PetscCheck(ndims == dims[1], comm, PETSC_ERR_FILE_UNEXPECTED,
140defe8520SJames Wright                "Line %" PetscInt_FMT " of %s does not have correct number of columns (%d instead of %" PetscInt_FMT ")", i, path, ndims, dims[1]);
141493642f1SJames Wright 
142493642f1SJames Wright     d[0][i]     = (CeedScalar)atof(array[0]);
143493642f1SJames Wright     d[1][i]     = (CeedScalar)atof(array[1]);
144493642f1SJames Wright     d[2][i]     = (CeedScalar)atof(array[2]);
145493642f1SJames Wright     phi[i]      = (CeedScalar)atof(array[3]);
146493642f1SJames Wright     sigma[0][i] = (CeedScalar)atof(array[4]);
147493642f1SJames Wright     sigma[1][i] = (CeedScalar)atof(array[5]);
148493642f1SJames Wright     sigma[2][i] = (CeedScalar)atof(array[6]);
149493642f1SJames Wright   }
15034b5deb1SJames Wright   PetscCall(PetscFClose(comm, fp));
1512b916ea7SJeremy L Thompson 
152d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
153493642f1SJames Wright }
154493642f1SJames Wright 
155493642f1SJames Wright /*
156493642f1SJames Wright  * @brief Read STG data from input paths and put in STGShur14Context
157493642f1SJames Wright  *
158493642f1SJames Wright  * Reads data from input paths and puts them into a STGShur14Context object.
15968d8e747SJames Wright  * Data stored initially in `*stg_ctx` will be copied over to the new STGShur14Context instance.
160493642f1SJames Wright  *
161493642f1SJames Wright  * @param[in]     comm            MPI_Comm for the program
162493642f1SJames Wright  * @param[in]     dm              DM for the program
163493642f1SJames Wright  * @param[in]     stg_inflow_path Path to STGInflow.dat file
164493642f1SJames Wright  * @param[in]     stg_rand_path   Path to STGRand.dat file
16568d8e747SJames Wright  * @param[in,out] stg_ctx         Pointer to STGShur14Context where the data will be loaded into
166493642f1SJames Wright  */
1672b916ea7SJeremy L Thompson PetscErrorCode GetSTGContextData(const MPI_Comm comm, const DM dm, char stg_inflow_path[PETSC_MAX_PATH_LEN], char stg_rand_path[PETSC_MAX_PATH_LEN],
16868d8e747SJames Wright                                  STGShur14Context *stg_ctx, const CeedScalar ynodes[]) {
169493642f1SJames Wright   PetscInt nmodes, nprofs;
170493642f1SJames Wright   PetscFunctionBeginUser;
171493642f1SJames Wright 
172493642f1SJames Wright   // Get options
173676080b4SJames Wright   PetscCall(PHASTADatFileGetNRows(comm, stg_rand_path, &nmodes));
174676080b4SJames Wright   PetscCall(PHASTADatFileGetNRows(comm, stg_inflow_path, &nprofs));
1755d28dccaSJames Wright   PetscCheck(nmodes < STG_NMODES_MAX, comm, PETSC_ERR_SUP,
176defe8520SJames Wright              "Number of wavemodes in %s (%" PetscInt_FMT ") exceeds STG_NMODES_MAX (%d). Change size of STG_NMODES_MAX and recompile", stg_rand_path,
177defe8520SJames Wright              nmodes, STG_NMODES_MAX);
178493642f1SJames Wright 
179493642f1SJames Wright   {
18068d8e747SJames Wright     STGShur14Context temp_ctx;
18168d8e747SJames Wright     PetscCall(PetscCalloc1(1, &temp_ctx));
18268d8e747SJames Wright     *temp_ctx                   = **stg_ctx;
18368d8e747SJames Wright     temp_ctx->nmodes            = nmodes;
18468d8e747SJames Wright     temp_ctx->nprofs            = nprofs;
18568d8e747SJames Wright     temp_ctx->offsets.sigma     = 0;
18668d8e747SJames Wright     temp_ctx->offsets.d         = nmodes * 3;
18768d8e747SJames Wright     temp_ctx->offsets.phi       = temp_ctx->offsets.d + nmodes * 3;
18868d8e747SJames Wright     temp_ctx->offsets.kappa     = temp_ctx->offsets.phi + nmodes;
18968d8e747SJames Wright     temp_ctx->offsets.wall_dist = temp_ctx->offsets.kappa + nmodes;
19068d8e747SJames Wright     temp_ctx->offsets.ubar      = temp_ctx->offsets.wall_dist + nprofs;
19168d8e747SJames Wright     temp_ctx->offsets.cij       = temp_ctx->offsets.ubar + nprofs * 3;
19268d8e747SJames Wright     temp_ctx->offsets.eps       = temp_ctx->offsets.cij + nprofs * 6;
19368d8e747SJames Wright     temp_ctx->offsets.lt        = temp_ctx->offsets.eps + nprofs;
19468d8e747SJames Wright     temp_ctx->offsets.ynodes    = temp_ctx->offsets.lt + nprofs;
19568d8e747SJames Wright     PetscInt total_num_scalars  = temp_ctx->offsets.ynodes + temp_ctx->nynodes;
19668d8e747SJames Wright     temp_ctx->total_bytes       = sizeof(*temp_ctx) + total_num_scalars * sizeof(temp_ctx->data[0]);
19768d8e747SJames Wright     PetscCall(PetscFree(*stg_ctx));
19868d8e747SJames Wright     PetscCall(PetscMalloc(temp_ctx->total_bytes, stg_ctx));
19968d8e747SJames Wright     **stg_ctx = *temp_ctx;
20068d8e747SJames Wright     PetscCall(PetscFree(temp_ctx));
201493642f1SJames Wright   }
202493642f1SJames Wright 
20368d8e747SJames Wright   PetscCall(ReadSTGInflow(comm, stg_inflow_path, *stg_ctx));
20468d8e747SJames Wright   PetscCall(ReadSTGRand(comm, stg_rand_path, *stg_ctx));
205493642f1SJames Wright 
20668d8e747SJames Wright   if ((*stg_ctx)->nynodes > 0) {
20768d8e747SJames Wright     CeedScalar *ynodes_ctx = &(*stg_ctx)->data[(*stg_ctx)->offsets.ynodes];
20868d8e747SJames Wright     for (PetscInt i = 0; i < (*stg_ctx)->nynodes; i++) ynodes_ctx[i] = ynodes[i];
209e6098bcdSJames Wright   }
210e6098bcdSJames Wright 
21168d8e747SJames Wright   {  // -- Calculate kappa
21268d8e747SJames Wright     CeedScalar *kappa     = &(*stg_ctx)->data[(*stg_ctx)->offsets.kappa];
21368d8e747SJames Wright     CeedScalar *wall_dist = &(*stg_ctx)->data[(*stg_ctx)->offsets.wall_dist];
21468d8e747SJames Wright     CeedScalar *lt        = &(*stg_ctx)->data[(*stg_ctx)->offsets.lt];
215493642f1SJames Wright     CeedScalar  le, le_max = 0;
216493642f1SJames Wright 
21768d8e747SJames Wright     CeedPragmaSIMD for (PetscInt i = 0; i < (*stg_ctx)->nprofs; i++) {
218c77f3192SJames Wright       le = PetscMin(2 * wall_dist[i], 3 * lt[i]);
219493642f1SJames Wright       if (le_max < le) le_max = le;
220493642f1SJames Wright     }
221493642f1SJames Wright     CeedScalar kmin = M_PI / le_max;
222493642f1SJames Wright 
22368d8e747SJames Wright     CeedPragmaSIMD for (PetscInt i = 0; i < (*stg_ctx)->nmodes; i++) { kappa[i] = kmin * pow((*stg_ctx)->alpha, i); }
22468d8e747SJames Wright   }
225493642f1SJames Wright 
226d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
227493642f1SJames Wright }
228493642f1SJames Wright 
2292b916ea7SJeremy L Thompson PetscErrorCode SetupSTG(const MPI_Comm comm, const DM dm, ProblemData *problem, User user, const bool prescribe_T, const CeedScalar theta0,
2302b916ea7SJeremy L Thompson                         const CeedScalar P0, const CeedScalar ynodes[], const CeedInt nynodes) {
231*b4c37c5cSJames Wright   Ceed                     ceed                                = user->ceed;
232493642f1SJames Wright   char                     stg_inflow_path[PETSC_MAX_PATH_LEN] = "./STGInflow.dat";
233493642f1SJames Wright   char                     stg_rand_path[PETSC_MAX_PATH_LEN]   = "./STGRand.dat";
2342b916ea7SJeremy L Thompson   PetscBool                mean_only = PETSC_FALSE, use_stgstrong = PETSC_FALSE, use_fluctuating_IC = PETSC_FALSE;
2352b916ea7SJeremy L Thompson   CeedScalar               u0 = 0.0, alpha = 1.01;
236493642f1SJames Wright   CeedQFunctionContext     stg_context;
237493642f1SJames Wright   NewtonianIdealGasContext newtonian_ig_ctx;
238493642f1SJames Wright   PetscFunctionBeginUser;
239493642f1SJames Wright 
240493642f1SJames Wright   // Get options
241493642f1SJames Wright   PetscOptionsBegin(comm, NULL, "STG Boundary Condition Options", NULL);
2422b916ea7SJeremy L Thompson   PetscCall(PetscOptionsString("-stg_inflow_path", "Path to STGInflow.dat", NULL, stg_inflow_path, stg_inflow_path, sizeof(stg_inflow_path), NULL));
2432b916ea7SJeremy L Thompson   PetscCall(PetscOptionsString("-stg_rand_path", "Path to STGInflow.dat", NULL, stg_rand_path, stg_rand_path, sizeof(stg_rand_path), NULL));
2442b916ea7SJeremy L Thompson   PetscCall(PetscOptionsReal("-stg_alpha", "Growth rate of the wavemodes", NULL, alpha, &alpha, NULL));
2452b916ea7SJeremy L Thompson   PetscCall(PetscOptionsReal("-stg_u0", "Advective velocity for the fluctuations", NULL, u0, &u0, NULL));
2462b916ea7SJeremy L Thompson   PetscCall(PetscOptionsBool("-stg_mean_only", "Only apply mean profile", NULL, mean_only, &mean_only, NULL));
2472b916ea7SJeremy L Thompson   PetscCall(PetscOptionsBool("-stg_strong", "Enforce STG inflow strongly", NULL, use_stgstrong, &use_stgstrong, NULL));
2482b916ea7SJeremy L Thompson   PetscCall(PetscOptionsBool("-stg_fluctuating_IC", "\"Extrude\" the fluctuations through the domain as an initial condition", NULL,
2492b916ea7SJeremy L Thompson                              use_fluctuating_IC, &use_fluctuating_IC, NULL));
250493642f1SJames Wright   PetscOptionsEnd();
251493642f1SJames Wright 
25234b5deb1SJames Wright   PetscCall(PetscCalloc1(1, &global_stg_ctx));
2538085925cSJames Wright   global_stg_ctx->alpha              = alpha;
2548085925cSJames Wright   global_stg_ctx->u0                 = u0;
2558085925cSJames Wright   global_stg_ctx->is_implicit        = user->phys->implicit;
2568085925cSJames Wright   global_stg_ctx->prescribe_T        = prescribe_T;
2578085925cSJames Wright   global_stg_ctx->mean_only          = mean_only;
258d4e0f297SJames Wright   global_stg_ctx->use_fluctuating_IC = use_fluctuating_IC;
2598085925cSJames Wright   global_stg_ctx->theta0             = theta0;
2608085925cSJames Wright   global_stg_ctx->P0                 = P0;
2618085925cSJames Wright   global_stg_ctx->nynodes            = nynodes;
262493642f1SJames Wright 
263493642f1SJames Wright   {
264493642f1SJames Wright     // Calculate dx assuming constant spacing
265493642f1SJames Wright     PetscReal domain_min[3], domain_max[3], domain_size[3];
26634b5deb1SJames Wright     PetscCall(DMGetBoundingBox(dm, domain_min, domain_max));
267493642f1SJames Wright     for (PetscInt i = 0; i < 3; i++) domain_size[i] = domain_max[i] - domain_min[i];
268493642f1SJames Wright 
269493642f1SJames Wright     PetscInt nmax = 3, faces[3];
2702b916ea7SJeremy L Thompson     PetscCall(PetscOptionsGetIntArray(NULL, NULL, "-dm_plex_box_faces", faces, &nmax, NULL));
2718085925cSJames Wright     global_stg_ctx->dx = domain_size[0] / faces[0];
2728085925cSJames Wright     global_stg_ctx->dz = domain_size[2] / faces[2];
273493642f1SJames Wright   }
274493642f1SJames Wright 
275*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionContextGetData(problem->apply_vol_rhs.qfunction_context, CEED_MEM_HOST, &newtonian_ig_ctx));
2768085925cSJames Wright   global_stg_ctx->newtonian_ctx = *newtonian_ig_ctx;
277*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionContextRestoreData(problem->apply_vol_rhs.qfunction_context, &newtonian_ig_ctx));
278493642f1SJames Wright 
2792b916ea7SJeremy L Thompson   PetscCall(GetSTGContextData(comm, dm, stg_inflow_path, stg_rand_path, &global_stg_ctx, ynodes));
280493642f1SJames Wright 
281*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionContextCreate(user->ceed, &stg_context));
282*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionContextSetData(stg_context, CEED_MEM_HOST, CEED_USE_POINTER, global_stg_ctx->total_bytes, global_stg_ctx));
283*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionContextSetDataDestroy(stg_context, CEED_MEM_HOST, FreeContextPetsc));
284*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionContextRegisterDouble(stg_context, "solution time", offsetof(struct STGShur14Context_, time), 1,
285*b4c37c5cSJames Wright                                                          "Physical time of the solution"));
286493642f1SJames Wright 
287*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionContextDestroy(&problem->ics.qfunction_context));
28843bff553SJames Wright   problem->ics.qfunction         = ICsSTG;
28943bff553SJames Wright   problem->ics.qfunction_loc     = ICsSTG_loc;
29043bff553SJames Wright   problem->ics.qfunction_context = stg_context;
29143bff553SJames Wright 
292e6098bcdSJames Wright   if (use_stgstrong) {
2938085925cSJames Wright     // Use default boundary integral QF (BoundaryIntegral) in newtonian.h
2942eb7bf1fSJames Wright     problem->use_strong_bc_ceed = PETSC_TRUE;
295d7244455SJames Wright     problem->bc_from_ics        = PETSC_FALSE;
296e6098bcdSJames Wright   } else {
297493642f1SJames Wright     problem->apply_inflow.qfunction              = STGShur14_Inflow;
298493642f1SJames Wright     problem->apply_inflow.qfunction_loc          = STGShur14_Inflow_loc;
299a6e8f989SJames Wright     problem->apply_inflow_jacobian.qfunction     = STGShur14_Inflow_Jacobian;
300a6e8f989SJames Wright     problem->apply_inflow_jacobian.qfunction_loc = STGShur14_Inflow_Jacobian_loc;
301*b4c37c5cSJames Wright     PetscCallCeed(ceed, CeedQFunctionContextReferenceCopy(stg_context, &problem->apply_inflow.qfunction_context));
302*b4c37c5cSJames Wright     PetscCallCeed(ceed, CeedQFunctionContextReferenceCopy(stg_context, &problem->apply_inflow_jacobian.qfunction_context));
303d7244455SJames Wright     problem->bc_from_ics = PETSC_TRUE;
304e6098bcdSJames Wright   }
305493642f1SJames Wright 
306d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
307493642f1SJames Wright }
308e6098bcdSJames Wright 
3092b916ea7SJeremy L Thompson static inline PetscScalar FindDy(const PetscScalar ynodes[], const PetscInt nynodes, const PetscScalar y) {
310e6098bcdSJames Wright   const PetscScalar half_mindy = 0.5 * (ynodes[1] - ynodes[0]);
311e6098bcdSJames Wright   // ^^assuming min(dy) is first element off the wall
312e6098bcdSJames Wright   PetscInt idx = -1;  // Index
313e6098bcdSJames Wright 
314e6098bcdSJames Wright   for (PetscInt i = 0; i < nynodes; i++) {
315e6098bcdSJames Wright     if (y < ynodes[i] + half_mindy) {
3162b916ea7SJeremy L Thompson       idx = i;
3172b916ea7SJeremy L Thompson       break;
318e6098bcdSJames Wright     }
319e6098bcdSJames Wright   }
320e6098bcdSJames Wright   if (idx == 0) return ynodes[1] - ynodes[0];
321e6098bcdSJames Wright   else if (idx == nynodes - 1) return ynodes[nynodes - 2] - ynodes[nynodes - 1];
322e6098bcdSJames Wright   else return 0.5 * (ynodes[idx + 1] - ynodes[idx - 1]);
323e6098bcdSJames Wright }
324e6098bcdSJames Wright 
325e6098bcdSJames Wright // Function passed to DMAddBoundary
3266d0190e2SJames Wright // NOTE: Not used in favor of QFunction-based method
3272b916ea7SJeremy L Thompson PetscErrorCode StrongSTGbcFunc(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nc, PetscScalar bcval[], void *ctx) {
328e6098bcdSJames Wright   PetscFunctionBeginUser;
3292b916ea7SJeremy L Thompson 
330e6098bcdSJames Wright   const STGShur14Context stg_ctx = (STGShur14Context)ctx;
331e6098bcdSJames Wright   PetscScalar            qn[stg_ctx->nmodes], u[3], ubar[3], cij[6], eps, lt;
332e6098bcdSJames Wright   const bool             mean_only = stg_ctx->mean_only;
333e6098bcdSJames Wright   const PetscScalar      dx        = stg_ctx->dx;
334e6098bcdSJames Wright   const PetscScalar      dz        = stg_ctx->dz;
335e6098bcdSJames Wright   const PetscScalar      mu        = stg_ctx->newtonian_ctx.mu;
336e6098bcdSJames Wright   const PetscScalar      theta0    = stg_ctx->theta0;
337e6098bcdSJames Wright   const PetscScalar      P0        = stg_ctx->P0;
338e6098bcdSJames Wright   const PetscScalar      cv        = stg_ctx->newtonian_ctx.cv;
339e6098bcdSJames Wright   const PetscScalar      cp        = stg_ctx->newtonian_ctx.cp;
340e6098bcdSJames Wright   const PetscScalar      Rd        = cp - cv;
341e6098bcdSJames Wright 
342e6098bcdSJames Wright   const CeedScalar rho = P0 / (Rd * theta0);
343e6098bcdSJames Wright   InterpolateProfile(x[1], ubar, cij, &eps, &lt, stg_ctx);
344e6098bcdSJames Wright   if (!mean_only) {
345e6098bcdSJames Wright     const PetscInt     nynodes = stg_ctx->nynodes;
346e6098bcdSJames Wright     const PetscScalar *ynodes  = &stg_ctx->data[stg_ctx->offsets.ynodes];
347e6098bcdSJames Wright     const PetscScalar  h[3]    = {dx, FindDy(ynodes, nynodes, x[1]), dz};
348e6098bcdSJames Wright     CalcSpectrum(x[1], eps, lt, h, mu / rho, qn, stg_ctx);
349e6098bcdSJames Wright     STGShur14_Calc(x, time, ubar, cij, qn, u, stg_ctx);
350e6098bcdSJames Wright   } else {
351e6098bcdSJames Wright     for (CeedInt j = 0; j < 3; j++) u[j] = ubar[j];
352e6098bcdSJames Wright   }
353e6098bcdSJames Wright 
354e6098bcdSJames Wright   bcval[0] = rho;
355e6098bcdSJames Wright   bcval[1] = rho * u[0];
356e6098bcdSJames Wright   bcval[2] = rho * u[1];
357e6098bcdSJames Wright   bcval[3] = rho * u[2];
358d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
359e6098bcdSJames Wright }
360e6098bcdSJames Wright 
3612b916ea7SJeremy L Thompson PetscErrorCode SetupStrongSTG(DM dm, SimpleBC bc, ProblemData *problem, Physics phys) {
362e6098bcdSJames Wright   DMLabel label;
363e6098bcdSJames Wright   PetscFunctionBeginUser;
364e6098bcdSJames Wright 
365d374fb47SJames Wright   PetscInt comps[5], num_comps = 4;
3663636f6a4SJames Wright   switch (phys->state_var) {
3673636f6a4SJames Wright     case STATEVAR_CONSERVATIVE:
368d374fb47SJames Wright       // {0,1,2,3} for rho, rho*u, rho*v, rho*w
369d374fb47SJames Wright       for (int i = 0; i < 4; i++) comps[i] = i;
3703636f6a4SJames Wright       break;
3713636f6a4SJames Wright 
3723636f6a4SJames Wright     case STATEVAR_PRIMITIVE:
3733636f6a4SJames Wright       // {1,2,3,4} for u, v, w, T
3743636f6a4SJames Wright       for (int i = 0; i < 4; i++) comps[i] = i + 1;
3753636f6a4SJames Wright       break;
376d374fb47SJames Wright   }
377d374fb47SJames Wright 
37834b5deb1SJames Wright   PetscCall(DMGetLabel(dm, "Face Sets", &label));
379e6098bcdSJames Wright   // Set wall BCs
380e6098bcdSJames Wright   if (bc->num_inflow > 0) {
3812b916ea7SJeremy L Thompson     PetscCall(DMAddBoundary(dm, DM_BC_ESSENTIAL, "STG", label, bc->num_inflow, bc->inflows, 0, num_comps, comps, (void (*)(void))StrongSTGbcFunc,
38234b5deb1SJames Wright                             NULL, global_stg_ctx, NULL));
383e6098bcdSJames Wright   }
384e6098bcdSJames Wright 
385d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
386e6098bcdSJames Wright }
3876d0190e2SJames Wright 
3882b916ea7SJeremy L Thompson PetscErrorCode SetupStrongSTG_QF(Ceed ceed, ProblemData *problem, CeedInt num_comp_x, CeedInt num_comp_q, CeedInt stg_data_size,
38968d8e747SJames Wright                                  CeedInt q_data_size_sur, CeedQFunction *qf_strongbc) {
3906d0190e2SJames Wright   PetscFunctionBeginUser;
391*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, STGShur14_Inflow_StrongQF, STGShur14_Inflow_StrongQF_loc, qf_strongbc));
392*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddInput(*qf_strongbc, "surface qdata", q_data_size_sur, CEED_EVAL_NONE));
393*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddInput(*qf_strongbc, "x", num_comp_x, CEED_EVAL_NONE));
394*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddInput(*qf_strongbc, "scale", 1, CEED_EVAL_NONE));
395*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddInput(*qf_strongbc, "stg data", stg_data_size, CEED_EVAL_NONE));
396*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddOutput(*qf_strongbc, "q", num_comp_q, CEED_EVAL_NONE));
3976d0190e2SJames Wright 
398*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionSetContext(*qf_strongbc, problem->ics.qfunction_context));
399d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
4006d0190e2SJames Wright }
4019eeef72bSJames Wright 
4022b916ea7SJeremy L Thompson PetscErrorCode SetupStrongSTG_PreProcessing(Ceed ceed, ProblemData *problem, CeedInt num_comp_x, CeedInt stg_data_size, CeedInt q_data_size_sur,
40368d8e747SJames Wright                                             CeedQFunction *qf_strongbc) {
4049eeef72bSJames Wright   PetscFunctionBeginUser;
405*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, Preprocess_STGShur14, Preprocess_STGShur14_loc, qf_strongbc));
406*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddInput(*qf_strongbc, "surface qdata", q_data_size_sur, CEED_EVAL_NONE));
407*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddInput(*qf_strongbc, "x", num_comp_x, CEED_EVAL_NONE));
408*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionAddOutput(*qf_strongbc, "stg data", stg_data_size, CEED_EVAL_NONE));
4099eeef72bSJames Wright 
410*b4c37c5cSJames Wright   PetscCallCeed(ceed, CeedQFunctionSetContext(*qf_strongbc, problem->ics.qfunction_context));
411d949ddfcSJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
4129eeef72bSJames Wright }
413