1*5aed82e4SJeremy L Thompson // Copyright (c) 2017-2024, 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 14c9c2c079SJeremy 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 // ----------------------------------------------------------------------------- 202b730f8bSJeremy L Thompson CEED_QFUNCTION(SetupMassRhs3)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 21e83e87a5Sjeremylt const CeedScalar *x = in[0], *w = in[1]; 22cb32e2e7SValeria Barra CeedScalar *true_soln = out[0], *rhs = out[1]; 23cb32e2e7SValeria Barra 24cb32e2e7SValeria Barra // Quadrature Point Loop 252b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 26cb32e2e7SValeria Barra // Component 1 27cb32e2e7SValeria 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]); 28cb32e2e7SValeria Barra // Component 2 2982311801Sjeremylt true_soln[i + 1 * Q] = 2 * true_soln[i + 0 * Q]; 30cb32e2e7SValeria Barra // Component 3 3182311801Sjeremylt true_soln[i + 2 * Q] = 3 * true_soln[i + 0 * Q]; 32cb32e2e7SValeria Barra 33cb32e2e7SValeria Barra // Component 1 34e83e87a5Sjeremylt rhs[i + 0 * Q] = w[i] * true_soln[i + 0 * Q]; 35cb32e2e7SValeria Barra // Component 2 3682311801Sjeremylt rhs[i + 1 * Q] = 2 * rhs[i + 0 * Q]; 37cb32e2e7SValeria Barra // Component 3 3882311801Sjeremylt rhs[i + 2 * Q] = 3 * rhs[i + 0 * Q]; 39cb32e2e7SValeria Barra } // End of Quadrature Point Loop 40cb32e2e7SValeria Barra return 0; 41cb32e2e7SValeria Barra } 42cb32e2e7SValeria Barra 43e83e87a5Sjeremylt // ----------------------------------------------------------------------------- 44ed264d09SValeria Barra // This QFunction applies the mass operator for a vector field of 3 components. 45ed264d09SValeria Barra // 46ed264d09SValeria Barra // Inputs: 47ed264d09SValeria Barra // u - Input vector at quadrature points 489b072555Sjeremylt // q_data - Geometric factors 49ed264d09SValeria Barra // 50ed264d09SValeria Barra // Output: 51ed264d09SValeria Barra // v - Output vector (test functions) at quadrature points 52cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 532b730f8bSJeremy L Thompson CEED_QFUNCTION(Mass3)(void *ctx, const CeedInt Q, const CeedScalar *const *in, CeedScalar *const *out) { 549b072555Sjeremylt const CeedScalar *u = in[0], *q_data = in[1]; 55cb32e2e7SValeria Barra CeedScalar *v = out[0]; 56cb32e2e7SValeria Barra 57cb32e2e7SValeria Barra // Quadrature Point Loop 582b730f8bSJeremy L Thompson CeedPragmaSIMD for (CeedInt i = 0; i < Q; i++) { 59cb32e2e7SValeria Barra // Component 1 609b072555Sjeremylt v[i + 0 * Q] = q_data[i] * u[i + 0 * Q]; 61cb32e2e7SValeria Barra // Component 2 629b072555Sjeremylt v[i + 1 * Q] = q_data[i] * u[i + 1 * Q]; 63cb32e2e7SValeria Barra // Component 3 649b072555Sjeremylt v[i + 2 * Q] = q_data[i] * u[i + 2 * Q]; 65cb32e2e7SValeria Barra } // End of Quadrature Point Loop 66cb32e2e7SValeria Barra return 0; 67cb32e2e7SValeria Barra } 68cb32e2e7SValeria Barra // ----------------------------------------------------------------------------- 69f6b55d2cSvaleriabarra 70f6b55d2cSvaleriabarra #endif // bp2_h 71