13d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, 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. 3cb32e2e7SValeria Barra // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 5cb32e2e7SValeria Barra // 63d8e8822SJeremy 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 14*c9c2c079SJeremy L Thompson #include <ceed.h> 15cb32e2e7SValeria Barra #include <math.h> 16cb32e2e7SValeria Barra 17e83e87a5Sjeremylt // ----------------------------------------------------------------------------- 18ed264d09SValeria Barra // This QFunction sets up the rhs and true solution for the problem 19cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 20cb32e2e7SValeria Barra CEED_QFUNCTION(SetupMassRhs3)(void *ctx, const CeedInt Q, 21cb32e2e7SValeria Barra const CeedScalar *const *in, 22cb32e2e7SValeria Barra CeedScalar *const *out) { 23e83e87a5Sjeremylt const CeedScalar *x = in[0], *w = in[1]; 24cb32e2e7SValeria Barra CeedScalar *true_soln = out[0], *rhs = out[1]; 25cb32e2e7SValeria Barra 26cb32e2e7SValeria Barra // Quadrature Point Loop 27cb32e2e7SValeria Barra CeedPragmaSIMD 28cb32e2e7SValeria Barra for (CeedInt i=0; i<Q; i++) { 29cb32e2e7SValeria Barra // Component 1 30cb32e2e7SValeria 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]); 31cb32e2e7SValeria Barra // Component 2 3282311801Sjeremylt true_soln[i+1*Q] = 2 * true_soln[i+0*Q]; 33cb32e2e7SValeria Barra // Component 3 3482311801Sjeremylt true_soln[i+2*Q] = 3 * true_soln[i+0*Q]; 35cb32e2e7SValeria Barra 36cb32e2e7SValeria Barra // Component 1 37e83e87a5Sjeremylt rhs[i+0*Q] = w[i] * true_soln[i+0*Q]; 38cb32e2e7SValeria Barra // Component 2 3982311801Sjeremylt rhs[i+1*Q] = 2 * rhs[i+0*Q]; 40cb32e2e7SValeria Barra // Component 3 4182311801Sjeremylt rhs[i+2*Q] = 3 * rhs[i+0*Q]; 42cb32e2e7SValeria Barra } // End of Quadrature Point Loop 43cb32e2e7SValeria Barra return 0; 44cb32e2e7SValeria Barra } 45cb32e2e7SValeria Barra 46e83e87a5Sjeremylt // ----------------------------------------------------------------------------- 47ed264d09SValeria Barra // This QFunction applies the mass operator for a vector field of 3 components. 48ed264d09SValeria Barra // 49ed264d09SValeria Barra // Inputs: 50ed264d09SValeria Barra // u - Input vector at quadrature points 519b072555Sjeremylt // q_data - Geometric factors 52ed264d09SValeria Barra // 53ed264d09SValeria Barra // Output: 54ed264d09SValeria Barra // v - Output vector (test functions) at quadrature points 55ed264d09SValeria Barra // 56cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 57cb32e2e7SValeria Barra CEED_QFUNCTION(Mass3)(void *ctx, const CeedInt Q, 58cb32e2e7SValeria Barra const CeedScalar *const *in, CeedScalar *const *out) { 599b072555Sjeremylt const CeedScalar *u = in[0], *q_data = in[1]; 60cb32e2e7SValeria Barra CeedScalar *v = out[0]; 61cb32e2e7SValeria Barra 62cb32e2e7SValeria Barra // Quadrature Point Loop 63cb32e2e7SValeria Barra CeedPragmaSIMD 64cb32e2e7SValeria Barra for (CeedInt i=0; i<Q; i++) { 65cb32e2e7SValeria Barra // Component 1 669b072555Sjeremylt v[i+0*Q] = q_data[i] * u[i+0*Q]; 67cb32e2e7SValeria Barra // Component 2 689b072555Sjeremylt v[i+1*Q] = q_data[i] * u[i+1*Q]; 69cb32e2e7SValeria Barra // Component 3 709b072555Sjeremylt v[i+2*Q] = q_data[i] * u[i+2*Q]; 71cb32e2e7SValeria Barra } // End of Quadrature Point Loop 72cb32e2e7SValeria Barra return 0; 73cb32e2e7SValeria Barra } 74cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 75f6b55d2cSvaleriabarra 76f6b55d2cSvaleriabarra #endif // bp2_h 77