1 // Copyright (c) 2017-2024, 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(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