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. 366087c08SValeria Barra // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 566087c08SValeria Barra // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 766087c08SValeria Barra 8*c0b5abf0SJeremy L Thompson #include <ceed/types.h> 9c9c2c079SJeremy L Thompson 1066087c08SValeria Barra /// 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 }; 1466087c08SValeria Barra 1566087c08SValeria Barra /// libCEED Q-function for building quadrature data for a mass operator 16d37d859eSJeremy L Thompson CEED_QFUNCTION(build_mass)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 1766087c08SValeria Barra // in[0] is Jacobians with shape [dim, nc=dim, Q] 1866087c08SValeria Barra // in[1] is quadrature weights, size (Q) 19d37d859eSJeremy L Thompson struct BuildContext *build_data = (struct BuildContext *)ctx; 2066087c08SValeria Barra const CeedScalar *J = in[0], *w = in[1]; 21d1d35e2fSjeremylt CeedScalar *q_data = out[0]; 2266087c08SValeria Barra 23d37d859eSJeremy L Thompson switch (build_data->dim + 10 * build_data->space_dim) { 2466087c08SValeria Barra case 11: 2566087c08SValeria Barra // Quadrature Point Loop 262b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { q_data[i] = J[i] * w[i]; } // End of Quadrature Point Loop 2766087c08SValeria Barra break; 2866087c08SValeria Barra case 22: 2966087c08SValeria Barra // Quadrature Point Loop 302b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 3166087c08SValeria Barra // 0 2 3266087c08SValeria Barra // 1 3 33d1d35e2fSjeremylt q_data[i] = (J[i + Q * 0] * J[i + Q * 3] - J[i + Q * 1] * J[i + Q * 2]) * w[i]; 3466087c08SValeria Barra } // End of Quadrature Point Loop 3566087c08SValeria Barra break; 3666087c08SValeria Barra case 33: 3766087c08SValeria Barra // Quadrature Point Loop 382b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 3966087c08SValeria Barra // 0 3 6 4066087c08SValeria Barra // 1 4 7 4166087c08SValeria Barra // 2 5 8 42d1d35e2fSjeremylt q_data[i] = (J[i + Q * 0] * (J[i + Q * 4] * J[i + Q * 8] - J[i + Q * 5] * J[i + Q * 7]) - 4366087c08SValeria Barra 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]; 4666087c08SValeria Barra } // End of Quadrature Point Loop 4766087c08SValeria Barra break; 4866087c08SValeria Barra } 4966087c08SValeria Barra return 0; 5066087c08SValeria Barra } 5166087c08SValeria Barra 5266087c08SValeria Barra /// libCEED Q-function for applying a mass operator 53d37d859eSJeremy L Thompson CEED_QFUNCTION(apply_mass)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 54d1d35e2fSjeremylt const CeedScalar *u = in[0], *q_data = in[1]; 5566087c08SValeria Barra CeedScalar *v = out[0]; 5666087c08SValeria Barra 5766087c08SValeria Barra // Quadrature Point Loop 582b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { v[i] = q_data[i] * u[i]; } // End of Quadrature Point Loop 5966087c08SValeria Barra return 0; 6066087c08SValeria Barra } 61