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*9f59f36eSJames Wright CEED_QFUNCTION_HELPER int Mass_N(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out, const CeedInt N) { 29a515125bSLeila Ghaffari // Inputs 303d65b166SJames Wright const CeedScalar(*u)[CEED_Q_VLA] = (const CeedScalar(*)[CEED_Q_VLA])in[0]; 313d65b166SJames Wright const CeedScalar(*q_data) = in[1]; 32a515125bSLeila Ghaffari 33a515125bSLeila Ghaffari // Outputs 34a515125bSLeila Ghaffari CeedScalar(*v)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0]; 35a515125bSLeila Ghaffari 362b916ea7SJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 37*9f59f36eSJames Wright CeedPragmaSIMD for (CeedInt j = 0; j < N; j++) { v[j][i] = q_data[i] * u[j][i]; } 38a515125bSLeila Ghaffari } 39a515125bSLeila Ghaffari return 0; 40a515125bSLeila Ghaffari } 41a515125bSLeila Ghaffari 42*9f59f36eSJames Wright CEED_QFUNCTION(Mass_1)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 1); } 43*9f59f36eSJames Wright 44*9f59f36eSJames Wright CEED_QFUNCTION(Mass_5)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 5); } 45*9f59f36eSJames Wright 46*9f59f36eSJames Wright CEED_QFUNCTION(Mass_9)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 9); } 47*9f59f36eSJames Wright 48*9f59f36eSJames Wright CEED_QFUNCTION(Mass_22)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 22); } 49a515125bSLeila Ghaffari // ***************************************************************************** 50a515125bSLeila Ghaffari 51a515125bSLeila Ghaffari #endif // mass_h 52