1727da7e7SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 2727da7e7SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3a515125bSLeila Ghaffari // 4727da7e7SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 5a515125bSLeila Ghaffari // 6727da7e7SJeremy 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> 15*493642f1SJames Wright #include <ceed.h> 16a515125bSLeila Ghaffari 17a515125bSLeila Ghaffari // ***************************************************************************** 18a515125bSLeila Ghaffari // This QFunction applies the mass matrix to five interlaced fields. 19a515125bSLeila Ghaffari // 20a515125bSLeila Ghaffari // Inputs: 21a515125bSLeila Ghaffari // u - Input vector at quadrature points 22a515125bSLeila Ghaffari // q_data - Quadrature weights 23a515125bSLeila Ghaffari // 24a515125bSLeila Ghaffari // Output: 25a515125bSLeila Ghaffari // v - Output vector at quadrature points 26a515125bSLeila Ghaffari // 27a515125bSLeila Ghaffari // ***************************************************************************** 28a515125bSLeila Ghaffari CEED_QFUNCTION(Mass)(void *ctx, CeedInt Q, 29a515125bSLeila Ghaffari const CeedScalar *const *in, CeedScalar *const *out) { 30a515125bSLeila Ghaffari // *INDENT-OFF* 31a515125bSLeila Ghaffari // Inputs 32a515125bSLeila Ghaffari const CeedScalar (*u)[CEED_Q_VLA] = (const CeedScalar(*)[CEED_Q_VLA])in[0], 33a515125bSLeila Ghaffari (*q_data) = in[1]; 34a515125bSLeila Ghaffari 35a515125bSLeila Ghaffari // Outputs 36a515125bSLeila Ghaffari CeedScalar (*v)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0]; 37a515125bSLeila Ghaffari // *INDENT-ON* 38a515125bSLeila Ghaffari 39a515125bSLeila Ghaffari CeedPragmaSIMD 40a515125bSLeila Ghaffari for (CeedInt i=0; i<Q; i++) { 41a515125bSLeila Ghaffari v[0][i] = q_data[i] * u[0][i]; 42a515125bSLeila Ghaffari v[1][i] = q_data[i] * u[1][i]; 43a515125bSLeila Ghaffari v[2][i] = q_data[i] * u[2][i]; 44a515125bSLeila Ghaffari v[3][i] = q_data[i] * u[3][i]; 45a515125bSLeila Ghaffari v[4][i] = q_data[i] * u[4][i]; 46a515125bSLeila Ghaffari } 47a515125bSLeila Ghaffari return 0; 48a515125bSLeila Ghaffari } 49a515125bSLeila Ghaffari 50a515125bSLeila Ghaffari // ***************************************************************************** 51a515125bSLeila Ghaffari 52a515125bSLeila Ghaffari #endif // mass_h 53