15aed82e4SJeremy L Thompson // Copyright (c) 2017-2024, Lawrence Livermore National Security, LLC and other CEED contributors. 23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 34d537eeaSYohann // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 54d537eeaSYohann // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 74d537eeaSYohann 8*c0b5abf0SJeremy L Thompson #include <ceed/types.h> 9c9c2c079SJeremy L Thompson 104d537eeaSYohann /// A structure used to pass additional data to f_build_mass 112b730f8bSJeremy L Thompson struct BuildContext { 122b730f8bSJeremy L Thompson CeedInt dim, space_dim; 132b730f8bSJeremy L Thompson }; 144d537eeaSYohann 154d537eeaSYohann /// libCEED Q-function for building quadrature data for a mass operator 162b730f8bSJeremy L Thompson CEED_QFUNCTION(f_build_mass)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 174d537eeaSYohann // in[0] is Jacobians with shape [dim, nc=dim, Q] 184d537eeaSYohann // in[1] is quadrature weights, size (Q) 194d537eeaSYohann BuildContext *bc = (BuildContext *)ctx; 20a2fa7910SValeria Barra const CeedScalar *J = in[0], *w = in[1]; 21a2fa7910SValeria Barra CeedScalar *qdata = out[0]; 22ee07ded2SValeria Barra 234d537eeaSYohann switch (bc->dim + 10 * bc->space_dim) { 244d537eeaSYohann case 11: 25ee07ded2SValeria Barra // Quadrature Point Loop 262b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { qdata[i] = J[i] * w[i]; } 274d537eeaSYohann break; 284d537eeaSYohann case 22: 29ee07ded2SValeria Barra // Quadrature Point Loop 302b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 314d537eeaSYohann // 0 2 324d537eeaSYohann // 1 3 33a2fa7910SValeria Barra qdata[i] = (J[i + Q * 0] * J[i + Q * 3] - J[i + Q * 1] * J[i + Q * 2]) * w[i]; 344d537eeaSYohann } 354d537eeaSYohann break; 364d537eeaSYohann case 33: 37ee07ded2SValeria Barra // Quadrature Point Loop 382b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 394d537eeaSYohann // 0 3 6 404d537eeaSYohann // 1 4 7 414d537eeaSYohann // 2 5 8 42a2fa7910SValeria Barra qdata[i] = (J[i + Q * 0] * (J[i + Q * 4] * J[i + Q * 8] - J[i + Q * 5] * J[i + Q * 7]) - 434d537eeaSYohann J[i + Q * 1] * (J[i + Q * 3] * J[i + Q * 8] - J[i + Q * 5] * J[i + Q * 6]) + 442b730f8bSJeremy L Thompson J[i + Q * 2] * (J[i + Q * 3] * J[i + Q * 7] - J[i + Q * 4] * J[i + Q * 6])) * 452b730f8bSJeremy L Thompson w[i]; 464d537eeaSYohann } 474d537eeaSYohann break; 484d537eeaSYohann } 494d537eeaSYohann return 0; 504d537eeaSYohann } 514d537eeaSYohann 524d537eeaSYohann /// libCEED Q-function for applying a mass operator 532b730f8bSJeremy L Thompson CEED_QFUNCTION(f_apply_mass)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 54a2fa7910SValeria Barra const CeedScalar *u = in[0], *qdata = in[1]; 554d537eeaSYohann CeedScalar *v = out[0]; 56ee07ded2SValeria Barra 57ee07ded2SValeria Barra // Quadrature Point Loop 582b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { v[i] = qdata[i] * u[i]; } 594d537eeaSYohann return 0; 604d537eeaSYohann } 61