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 14493642f1SJames Wright #include <ceed.h> 15d0cce58aSJeremy L Thompson #include <math.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 // ***************************************************************************** 28*2b916ea7SJeremy L Thompson CEED_QFUNCTION(Mass)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 29a515125bSLeila Ghaffari // *INDENT-OFF* 30a515125bSLeila Ghaffari // Inputs 31*2b916ea7SJeremy L Thompson const CeedScalar(*u)[CEED_Q_VLA] = (const CeedScalar(*)[CEED_Q_VLA])in[0], (*q_data) = in[1]; 32a515125bSLeila Ghaffari 33a515125bSLeila Ghaffari // Outputs 34a515125bSLeila Ghaffari CeedScalar(*v)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0]; 35a515125bSLeila Ghaffari // *INDENT-ON* 36a515125bSLeila Ghaffari 37*2b916ea7SJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 38a515125bSLeila Ghaffari v[0][i] = q_data[i] * u[0][i]; 39a515125bSLeila Ghaffari v[1][i] = q_data[i] * u[1][i]; 40a515125bSLeila Ghaffari v[2][i] = q_data[i] * u[2][i]; 41a515125bSLeila Ghaffari v[3][i] = q_data[i] * u[3][i]; 42a515125bSLeila Ghaffari v[4][i] = q_data[i] * u[4][i]; 43a515125bSLeila Ghaffari } 44a515125bSLeila Ghaffari return 0; 45a515125bSLeila Ghaffari } 46a515125bSLeila Ghaffari 47a515125bSLeila Ghaffari // ***************************************************************************** 48a515125bSLeila Ghaffari 49a515125bSLeila Ghaffari #endif // mass_h 50