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