1*3d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 2*3d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3cb32e2e7SValeria Barra // 4*3d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 5cb32e2e7SValeria Barra // 6*3d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 7cb32e2e7SValeria Barra 8cb32e2e7SValeria Barra /// @file 9cb32e2e7SValeria Barra /// libCEED QFunctions for mass operator example using PETSc 10cb32e2e7SValeria Barra 11f6b55d2cSvaleriabarra #ifndef bp2_h 12f6b55d2cSvaleriabarra #define bp2_h 13f6b55d2cSvaleriabarra 14cb32e2e7SValeria Barra #include <math.h> 15cb32e2e7SValeria Barra 16e83e87a5Sjeremylt // ----------------------------------------------------------------------------- 17ed264d09SValeria Barra // This QFunction sets up the rhs and true solution for the problem 18cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 19cb32e2e7SValeria Barra CEED_QFUNCTION(SetupMassRhs3)(void *ctx, const CeedInt Q, 20cb32e2e7SValeria Barra const CeedScalar *const *in, 21cb32e2e7SValeria Barra CeedScalar *const *out) { 22e83e87a5Sjeremylt const CeedScalar *x = in[0], *w = in[1]; 23cb32e2e7SValeria Barra CeedScalar *true_soln = out[0], *rhs = out[1]; 24cb32e2e7SValeria Barra 25cb32e2e7SValeria Barra // Quadrature Point Loop 26cb32e2e7SValeria Barra CeedPragmaSIMD 27cb32e2e7SValeria Barra for (CeedInt i=0; i<Q; i++) { 28cb32e2e7SValeria Barra // Component 1 29cb32e2e7SValeria Barra true_soln[i+0*Q] = sqrt(x[i]*x[i] + x[i+Q]*x[i+Q] + x[i+2*Q]*x[i+2*Q]); 30cb32e2e7SValeria Barra // Component 2 3182311801Sjeremylt true_soln[i+1*Q] = 2 * true_soln[i+0*Q]; 32cb32e2e7SValeria Barra // Component 3 3382311801Sjeremylt true_soln[i+2*Q] = 3 * true_soln[i+0*Q]; 34cb32e2e7SValeria Barra 35cb32e2e7SValeria Barra // Component 1 36e83e87a5Sjeremylt rhs[i+0*Q] = w[i] * true_soln[i+0*Q]; 37cb32e2e7SValeria Barra // Component 2 3882311801Sjeremylt rhs[i+1*Q] = 2 * rhs[i+0*Q]; 39cb32e2e7SValeria Barra // Component 3 4082311801Sjeremylt rhs[i+2*Q] = 3 * rhs[i+0*Q]; 41cb32e2e7SValeria Barra } // End of Quadrature Point Loop 42cb32e2e7SValeria Barra return 0; 43cb32e2e7SValeria Barra } 44cb32e2e7SValeria Barra 45e83e87a5Sjeremylt // ----------------------------------------------------------------------------- 46ed264d09SValeria Barra // This QFunction applies the mass operator for a vector field of 3 components. 47ed264d09SValeria Barra // 48ed264d09SValeria Barra // Inputs: 49ed264d09SValeria Barra // u - Input vector at quadrature points 509b072555Sjeremylt // q_data - Geometric factors 51ed264d09SValeria Barra // 52ed264d09SValeria Barra // Output: 53ed264d09SValeria Barra // v - Output vector (test functions) at quadrature points 54ed264d09SValeria Barra // 55cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 56cb32e2e7SValeria Barra CEED_QFUNCTION(Mass3)(void *ctx, const CeedInt Q, 57cb32e2e7SValeria Barra const CeedScalar *const *in, CeedScalar *const *out) { 589b072555Sjeremylt const CeedScalar *u = in[0], *q_data = in[1]; 59cb32e2e7SValeria Barra CeedScalar *v = out[0]; 60cb32e2e7SValeria Barra 61cb32e2e7SValeria Barra // Quadrature Point Loop 62cb32e2e7SValeria Barra CeedPragmaSIMD 63cb32e2e7SValeria Barra for (CeedInt i=0; i<Q; i++) { 64cb32e2e7SValeria Barra // Component 1 659b072555Sjeremylt v[i+0*Q] = q_data[i] * u[i+0*Q]; 66cb32e2e7SValeria Barra // Component 2 679b072555Sjeremylt v[i+1*Q] = q_data[i] * u[i+1*Q]; 68cb32e2e7SValeria Barra // Component 3 699b072555Sjeremylt v[i+2*Q] = q_data[i] * u[i+2*Q]; 70cb32e2e7SValeria Barra } // End of Quadrature Point Loop 71cb32e2e7SValeria Barra return 0; 72cb32e2e7SValeria Barra } 73cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 74f6b55d2cSvaleriabarra 75f6b55d2cSvaleriabarra #endif // bp2_h 76