xref: /honee/qfunctions/monitor_cfl.h (revision 87fd7f33e5ff541d346273b55e9eb051d4de0b43)
1 // SPDX-FileCopyrightText: Copyright (c) 2017-2024, HONEE contributors.
2 // SPDX-License-Identifier: Apache-2.0 OR BSD-2-Clause
3 
4 #include <ceed/types.h>
5 #include "newtonian_state.h"
6 
7 CEED_QFUNCTION_HELPER int MonitorCFL(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out, StateVariable state_var,
8                                      CeedInt dim) {
9   const NewtonianIdealGasContext gas = (const NewtonianIdealGasContext)ctx;
10   const CeedScalar(*q)[CEED_Q_VLA]   = (const CeedScalar(*)[CEED_Q_VLA])in[0];
11   const CeedScalar(*q_data)          = in[1];
12   CeedScalar(*v)                     = out[0];
13 
14   CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) {
15     const CeedScalar qi[5] = {q[0][i], q[1][i], q[2][i], q[3][i], q[4][i]};
16     const State      s     = StateFromQ(gas, qi, state_var);
17     CeedScalar       wdetJ, dXdx[9], gij_uj[3] = {0.}, gijd_mat[9] = {0.};
18 
19     QdataUnpack_ND(dim, Q, i, q_data, &wdetJ, dXdx);
20     MatMatN(dXdx, dXdx, dim, CEED_TRANSPOSE, CEED_NOTRANSPOSE, gijd_mat);
21     // (1/2)^2 to account for reference element size; for length 1 square/cube element, gij should be identity matrix
22     ScaleN((CeedScalar *)gijd_mat, 0.25, Square(dim));
23     // u_i g_ij u_j
24     MatVecNM(gijd_mat, s.Y.velocity, dim, dim, CEED_NOTRANSPOSE, gij_uj);
25     v[i] = sqrt(Dot3(s.Y.velocity, gij_uj));
26   }
27   return 0;
28 }
29 
30 CEED_QFUNCTION(MonitorCFL_3D_Conserv)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
31   return MonitorCFL(ctx, Q, in, out, STATEVAR_CONSERVATIVE, 3);
32 }
33 
34 CEED_QFUNCTION(MonitorCFL_3D_Prim)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
35   return MonitorCFL(ctx, Q, in, out, STATEVAR_PRIMITIVE, 3);
36 }
37 
38 CEED_QFUNCTION(MonitorCFL_3D_Entropy)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
39   return MonitorCFL(ctx, Q, in, out, STATEVAR_ENTROPY, 3);
40 }
41 
42 CEED_QFUNCTION(MonitorCFL_2D_Conserv)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
43   return MonitorCFL(ctx, Q, in, out, STATEVAR_CONSERVATIVE, 2);
44 }
45 
46 CEED_QFUNCTION(MonitorCFL_2D_Prim)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
47   return MonitorCFL(ctx, Q, in, out, STATEVAR_PRIMITIVE, 2);
48 }
49 
50 CEED_QFUNCTION(MonitorCFL_2D_Entropy)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
51   return MonitorCFL(ctx, Q, in, out, STATEVAR_ENTROPY, 2);
52 }
53