1 // Copyright (c) 2017-2026, 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/types.h>
9
apply(void * ctx,const CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)10 CEED_QFUNCTION(apply)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) {
11 // in[0] is gradient u_0, shape [2, num_comp=2, Q]
12 // in[1] is mass quadrature data, size (Q)
13 // in[2] is Poisson quadrature data, size (Q)
14 // in[3] is u_0, size (Q)
15 // in[4] is u_1, size (Q)
16 const CeedScalar *du_0 = in[0], *qd_mass = in[1], *qd_diff = in[2], *u_0 = in[3], *u_1 = in[4];
17
18 // out[0] is output to multiply against v_0, size (Q)
19 // out[1] is output to multiply against v_1, size (Q)
20 // out[2] is output to multiply against gradient v_0, shape [2, num_comp=2, Q]
21 CeedScalar *v_0 = out[0], *v_1 = out[1], *dv_0 = out[2];
22
23 const CeedInt num_comp_0 = 2;
24
25 // Quadrature point loop
26 for (CeedInt i = 0; i < Q; i++) {
27 for (CeedInt c = 0; c < num_comp_0; c++) {
28 // Mass
29 v_0[i + Q * c] = qd_mass[i] * (c + 1) * u_0[i + Q * c];
30 // Diff
31 dv_0[i + Q * (0 * num_comp_0 + c)] =
32 qd_diff[i + Q * 0] * (c + 1) * du_0[i + Q * (0 * num_comp_0 + c)] + qd_diff[i + Q * 2] * du_0[i + Q * (1 * num_comp_0 + c)];
33 dv_0[i + Q * (1 * num_comp_0 + c)] =
34 qd_diff[i + Q * 2] * (c + 1) * du_0[i + Q * (0 * num_comp_0 + c)] + qd_diff[i + Q * 1] * du_0[i + Q * (1 * num_comp_0 + c)];
35 }
36 // Mass
37 v_1[i] = qd_mass[i] * u_1[i];
38 }
39
40 return 0;
41 }
42