xref: /honee/qfunctions/mass.h (revision 727da7e70824329c32c70307643ee0d0901369c8)
1*727da7e7SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2*727da7e7SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3a515125bSLeila Ghaffari //
4*727da7e7SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
5a515125bSLeila Ghaffari //
6*727da7e7SJeremy L Thompson // This file is part of CEED:  http://github.com/ceed
7a515125bSLeila Ghaffari 
8a515125bSLeila Ghaffari /// @file
9a515125bSLeila Ghaffari /// Mass operator for Navier-Stokes example using PETSc
10a515125bSLeila Ghaffari 
11a515125bSLeila Ghaffari #ifndef mass_h
12a515125bSLeila Ghaffari #define mass_h
13a515125bSLeila Ghaffari 
14a515125bSLeila Ghaffari #include <math.h>
15a515125bSLeila Ghaffari 
16a515125bSLeila Ghaffari // *****************************************************************************
17a515125bSLeila Ghaffari // This QFunction applies the mass matrix to five interlaced fields.
18a515125bSLeila Ghaffari //
19a515125bSLeila Ghaffari // Inputs:
20a515125bSLeila Ghaffari //   u     - Input vector at quadrature points
21a515125bSLeila Ghaffari //   q_data - Quadrature weights
22a515125bSLeila Ghaffari //
23a515125bSLeila Ghaffari // Output:
24a515125bSLeila Ghaffari //   v - Output vector at quadrature points
25a515125bSLeila Ghaffari //
26a515125bSLeila Ghaffari // *****************************************************************************
27a515125bSLeila Ghaffari CEED_QFUNCTION(Mass)(void *ctx, CeedInt Q,
28a515125bSLeila Ghaffari                      const CeedScalar *const *in, CeedScalar *const *out) {
29a515125bSLeila Ghaffari   // *INDENT-OFF*
30a515125bSLeila Ghaffari   // Inputs
31a515125bSLeila Ghaffari   const CeedScalar (*u)[CEED_Q_VLA] = (const CeedScalar(*)[CEED_Q_VLA])in[0],
32a515125bSLeila Ghaffari                    (*q_data) = in[1];
33a515125bSLeila Ghaffari 
34a515125bSLeila Ghaffari   // Outputs
35a515125bSLeila Ghaffari   CeedScalar (*v)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0];
36a515125bSLeila Ghaffari   // *INDENT-ON*
37a515125bSLeila Ghaffari 
38a515125bSLeila Ghaffari   CeedPragmaSIMD
39a515125bSLeila Ghaffari   for (CeedInt i=0; i<Q; i++) {
40a515125bSLeila Ghaffari     v[0][i] = q_data[i] * u[0][i];
41a515125bSLeila Ghaffari     v[1][i] = q_data[i] * u[1][i];
42a515125bSLeila Ghaffari     v[2][i] = q_data[i] * u[2][i];
43a515125bSLeila Ghaffari     v[3][i] = q_data[i] * u[3][i];
44a515125bSLeila Ghaffari     v[4][i] = q_data[i] * u[4][i];
45a515125bSLeila Ghaffari   }
46a515125bSLeila Ghaffari   return 0;
47a515125bSLeila Ghaffari }
48a515125bSLeila Ghaffari 
49a515125bSLeila Ghaffari // *****************************************************************************
50a515125bSLeila Ghaffari 
51a515125bSLeila Ghaffari #endif // mass_h
52