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 /// @file 9 /// Mass operator for Navier-Stokes example using PETSc 10 11 #ifndef mass_h 12 #define mass_h 13 14 #include <math.h> 15 16 // ***************************************************************************** 17 // This QFunction applies the mass matrix to five interlaced fields. 18 // 19 // Inputs: 20 // u - Input vector at quadrature points 21 // q_data - Quadrature weights 22 // 23 // Output: 24 // v - Output vector at quadrature points 25 // 26 // ***************************************************************************** 27 CEED_QFUNCTION(Mass)(void *ctx, CeedInt Q, 28 const CeedScalar *const *in, CeedScalar *const *out) { 29 // *INDENT-OFF* 30 // Inputs 31 const CeedScalar (*u)[CEED_Q_VLA] = (const CeedScalar(*)[CEED_Q_VLA])in[0], 32 (*q_data) = in[1]; 33 34 // Outputs 35 CeedScalar (*v)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0]; 36 // *INDENT-ON* 37 38 CeedPragmaSIMD 39 for (CeedInt i=0; i<Q; i++) { 40 v[0][i] = q_data[i] * u[0][i]; 41 v[1][i] = q_data[i] * u[1][i]; 42 v[2][i] = q_data[i] * u[2][i]; 43 v[3][i] = q_data[i] * u[3][i]; 44 v[4][i] = q_data[i] * u[4][i]; 45 } 46 return 0; 47 } 48 49 // ***************************************************************************** 50 51 #endif // mass_h 52