1 // SPDX-FileCopyrightText: Copyright (c) 2017-2024, HONEE contributors.
2 // SPDX-License-Identifier: Apache-2.0 OR BSD-2-Clause
3
4 /// @file
5 /// Mass operator for HONEE
6 #include <ceed/types.h>
7
8 // *****************************************************************************
9 // This QFunction applies the mass matrix to N components
10 //
11 // Inputs:
12 // u - Input vector at quadrature points
13 // q_data - Quadrature weights
14 //
15 // Output:
16 // v - Output vector at quadrature points
17 //
18 // *****************************************************************************
Mass_N(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out,const CeedInt N)19 CEED_QFUNCTION_HELPER int Mass_N(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out, const CeedInt N) {
20 const CeedScalar(*u)[CEED_Q_VLA] = (const CeedScalar(*)[CEED_Q_VLA])in[0];
21 const CeedScalar(*q_data) = in[1];
22 CeedScalar(*v)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0];
23
24 CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) CeedPragmaSIMD for (CeedInt j = 0; j < N; j++) v[j][i] = q_data[i] * u[j][i];
25 return 0;
26 }
27
Mass_1(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)28 CEED_QFUNCTION(Mass_1)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 1); }
29
Mass_2(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)30 CEED_QFUNCTION(Mass_2)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 2); }
31
Mass_3(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)32 CEED_QFUNCTION(Mass_3)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 3); }
33
Mass_4(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)34 CEED_QFUNCTION(Mass_4)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 4); }
35
Mass_5(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)36 CEED_QFUNCTION(Mass_5)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 5); }
37
Mass_6(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)38 CEED_QFUNCTION(Mass_6)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 6); }
39
Mass_7(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)40 CEED_QFUNCTION(Mass_7)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 7); }
41
Mass_9(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)42 CEED_QFUNCTION(Mass_9)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 9); }
43
Mass_12(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)44 CEED_QFUNCTION(Mass_12)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 12); }
45
Mass_22(void * ctx,CeedInt Q,const CeedScalar * const * in,CeedScalar * const * out)46 CEED_QFUNCTION(Mass_22)(void *ctx, CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { return Mass_N(ctx, Q, in, out, 22); }
47