1*5aed82e4SJeremy 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 8e3bad73bSvaleriabarra #ifndef ex1_volume_h 9e3bad73bSvaleriabarra #define ex1_volume_h 10e3bad73bSvaleriabarra 11c9c2c079SJeremy L Thompson #include <ceed.h> 12c9c2c079SJeremy L Thompson 1366087c08SValeria Barra /// A structure used to pass additional data to f_build_mass 142b730f8bSJeremy L Thompson struct BuildContext { 152b730f8bSJeremy L Thompson CeedInt dim, space_dim; 162b730f8bSJeremy L Thompson }; 1766087c08SValeria Barra 1866087c08SValeria Barra /// libCEED Q-function for building quadrature data for a mass operator 19d37d859eSJeremy L Thompson CEED_QFUNCTION(build_mass)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 2066087c08SValeria Barra // in[0] is Jacobians with shape [dim, nc=dim, Q] 2166087c08SValeria Barra // in[1] is quadrature weights, size (Q) 22d37d859eSJeremy L Thompson struct BuildContext *build_data = (struct BuildContext *)ctx; 2366087c08SValeria Barra const CeedScalar *J = in[0], *w = in[1]; 24d1d35e2fSjeremylt CeedScalar *q_data = out[0]; 2566087c08SValeria Barra 26d37d859eSJeremy L Thompson switch (build_data->dim + 10 * build_data->space_dim) { 2766087c08SValeria Barra case 11: 2866087c08SValeria Barra // Quadrature Point Loop 292b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { q_data[i] = J[i] * w[i]; } // End of Quadrature Point Loop 3066087c08SValeria Barra break; 3166087c08SValeria Barra case 22: 3266087c08SValeria Barra // Quadrature Point Loop 332b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 3466087c08SValeria Barra // 0 2 3566087c08SValeria Barra // 1 3 36d1d35e2fSjeremylt q_data[i] = (J[i + Q * 0] * J[i + Q * 3] - J[i + Q * 1] * J[i + Q * 2]) * w[i]; 3766087c08SValeria Barra } // End of Quadrature Point Loop 3866087c08SValeria Barra break; 3966087c08SValeria Barra case 33: 4066087c08SValeria Barra // Quadrature Point Loop 412b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 4266087c08SValeria Barra // 0 3 6 4366087c08SValeria Barra // 1 4 7 4466087c08SValeria Barra // 2 5 8 45d1d35e2fSjeremylt q_data[i] = (J[i + Q * 0] * (J[i + Q * 4] * J[i + Q * 8] - J[i + Q * 5] * J[i + Q * 7]) - 4666087c08SValeria Barra J[i + Q * 1] * (J[i + Q * 3] * J[i + Q * 8] - J[i + Q * 5] * J[i + Q * 6]) + 472b730f8bSJeremy L Thompson J[i + Q * 2] * (J[i + Q * 3] * J[i + Q * 7] - J[i + Q * 4] * J[i + Q * 6])) * 482b730f8bSJeremy L Thompson w[i]; 4966087c08SValeria Barra } // End of Quadrature Point Loop 5066087c08SValeria Barra break; 5166087c08SValeria Barra } 5266087c08SValeria Barra return 0; 5366087c08SValeria Barra } 5466087c08SValeria Barra 5566087c08SValeria Barra /// libCEED Q-function for applying a mass operator 56d37d859eSJeremy L Thompson CEED_QFUNCTION(apply_mass)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 57d1d35e2fSjeremylt const CeedScalar *u = in[0], *q_data = in[1]; 5866087c08SValeria Barra CeedScalar *v = out[0]; 5966087c08SValeria Barra 6066087c08SValeria Barra // Quadrature Point Loop 612b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { v[i] = q_data[i] * u[i]; } // End of Quadrature Point Loop 6266087c08SValeria Barra return 0; 6366087c08SValeria Barra } 64e3bad73bSvaleriabarra 65e3bad73bSvaleriabarra #endif // ex1_volume_h 66