1 // Copyright (c) 2017-2022, 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/backend.h> 9 #include <ceed/ceed.h> 10 #include <ceed/jit-source/gallery/ceed-vectorpoisson3dapply.h> 11 #include <string.h> 12 13 /** 14 @brief Set fields for Ceed QFunction applying the 3D Poisson operator 15 on a vector system with three components 16 **/ 17 static int CeedQFunctionInit_Vector3Poisson3DApply(Ceed ceed, const char *requested, CeedQFunction qf) { 18 // Check QFunction name 19 const char *name = "Vector3Poisson3DApply"; 20 if (strcmp(name, requested)) { 21 // LCOV_EXCL_START 22 return CeedError(ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested); 23 // LCOV_EXCL_STOP 24 } 25 26 // Add QFunction fields 27 const CeedInt dim = 3, num_comp = 3; 28 CeedCall(CeedQFunctionAddInput(qf, "du", num_comp * dim, CEED_EVAL_GRAD)); 29 CeedCall(CeedQFunctionAddInput(qf, "qdata", dim * (dim + 1) / 2, CEED_EVAL_NONE)); 30 CeedCall(CeedQFunctionAddOutput(qf, "dv", num_comp * dim, CEED_EVAL_GRAD)); 31 32 CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, num_comp * 15)); 33 34 return CEED_ERROR_SUCCESS; 35 } 36 37 /** 38 @brief Register Ceed QFunction for applying the 3D Poisson operator 39 on a vector system with three components 40 **/ 41 CEED_INTERN int CeedQFunctionRegister_Vector3Poisson3DApply(void) { 42 return CeedQFunctionRegister("Vector3Poisson3DApply", Vector3Poisson3DApply_loc, 1, Vector3Poisson3DApply, CeedQFunctionInit_Vector3Poisson3DApply); 43 } 44