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. 36a6224a1SJeremy L Thompson // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 56a6224a1SJeremy L Thompson // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 76a6224a1SJeremy L Thompson 86a6224a1SJeremy L Thompson #include <ceed/ceed.h> 96a6224a1SJeremy L Thompson #include <ceed/backend.h> 106a6224a1SJeremy L Thompson #include <string.h> 11*a0154adeSJed Brown #include <ceed/jit-source/gallery/ceed-poisson3dapply.h> 126a6224a1SJeremy L Thompson 136a6224a1SJeremy L Thompson /** 146a6224a1SJeremy L Thompson @brief Set fields for Ceed QFunction applying the 3D Poisson operator 156a6224a1SJeremy L Thompson **/ 166a6224a1SJeremy L Thompson static int CeedQFunctionInit_Poisson3DApply(Ceed ceed, const char *requested, 176a6224a1SJeremy L Thompson CeedQFunction qf) { 186a6224a1SJeremy L Thompson int ierr; 196a6224a1SJeremy L Thompson 206a6224a1SJeremy L Thompson // Check QFunction name 216a6224a1SJeremy L Thompson const char *name = "Poisson3DApply"; 226a6224a1SJeremy L Thompson if (strcmp(name, requested)) 236a6224a1SJeremy L Thompson // LCOV_EXCL_START 246a6224a1SJeremy L Thompson return CeedError(ceed, CEED_ERROR_UNSUPPORTED, 256a6224a1SJeremy L Thompson "QFunction '%s' does not match requested name: %s", 266a6224a1SJeremy L Thompson name, requested); 276a6224a1SJeremy L Thompson // LCOV_EXCL_STOP 286a6224a1SJeremy L Thompson 296a6224a1SJeremy L Thompson // Add QFunction fields 306a6224a1SJeremy L Thompson const CeedInt dim = 3; 316a6224a1SJeremy L Thompson ierr = CeedQFunctionAddInput(qf, "du", dim, CEED_EVAL_GRAD); CeedChk(ierr); 326a6224a1SJeremy L Thompson ierr = CeedQFunctionAddInput(qf, "qdata", dim*(dim+1)/2, CEED_EVAL_NONE); 336a6224a1SJeremy L Thompson CeedChk(ierr); 346a6224a1SJeremy L Thompson ierr = CeedQFunctionAddOutput(qf, "dv", dim, CEED_EVAL_GRAD); CeedChk(ierr); 356a6224a1SJeremy L Thompson 366e15d496SJeremy L Thompson ierr = CeedQFunctionSetUserFlopsEstimate(qf, 15); CeedChk(ierr); 376e15d496SJeremy L Thompson 386a6224a1SJeremy L Thompson return CEED_ERROR_SUCCESS; 396a6224a1SJeremy L Thompson } 406a6224a1SJeremy L Thompson 416a6224a1SJeremy L Thompson /** 426a6224a1SJeremy L Thompson @brief Register Ceed QFunction for applying the 3D Poisson operator 436a6224a1SJeremy L Thompson **/ 446a6224a1SJeremy L Thompson CEED_INTERN int CeedQFunctionRegister_Poisson3DApply(void) { 456a6224a1SJeremy L Thompson return CeedQFunctionRegister("Poisson3DApply", Poisson3DApply_loc, 1, 466a6224a1SJeremy L Thompson Poisson3DApply, CeedQFunctionInit_Poisson3DApply); 476a6224a1SJeremy L Thompson } 48