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 #include <ceed.h> 9 10 CEED_QFUNCTION(setup)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 11 const CeedScalar *weight = in[0], *dxdX = in[1]; 12 CeedScalar *rho = out[0]; 13 14 for (CeedInt i = 0; i < Q; i++) { 15 rho[i] = weight[i] * dxdX[i]; 16 } 17 return 0; 18 } 19 20 CEED_QFUNCTION(mass)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 21 const CeedScalar *rho = in[0], (*u)[CEED_Q_VLA] = (const CeedScalar(*)[CEED_Q_VLA])in[1]; 22 CeedScalar(*v)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0]; 23 24 for (CeedInt i = 0; i < Q; i++) { 25 v[0][i] = rho[i] * u[0][i]; 26 v[1][i] = rho[i] * u[1][i]; 27 } 28 return 0; 29 } 30