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/backend.h> 92b730f8bSJeremy L Thompson #include <ceed/ceed.h> 10a0154adeSJed Brown #include <ceed/jit-source/gallery/ceed-vectorpoisson2dapply.h> 112b730f8bSJeremy L Thompson #include <string.h> 126a6224a1SJeremy L Thompson 136a6224a1SJeremy L Thompson /** 14*ea61e9acSJeremy L Thompson @brief Set fields for Ceed QFunction applying the 2D Poisson operator on a vector system with three components 156a6224a1SJeremy L Thompson **/ 162b730f8bSJeremy L Thompson static int CeedQFunctionInit_Vector3Poisson2DApply(Ceed ceed, const char *requested, CeedQFunction qf) { 176a6224a1SJeremy L Thompson // Check QFunction name 18dfb517d7SJeremy L Thompson const char *name = "Vector3Poisson2DApply"; 192b730f8bSJeremy L Thompson if (strcmp(name, requested)) { 206a6224a1SJeremy L Thompson // LCOV_EXCL_START 212b730f8bSJeremy L Thompson return CeedError(ceed, CEED_ERROR_UNSUPPORTED, "QFunction '%s' does not match requested name: %s", name, requested); 226a6224a1SJeremy L Thompson // LCOV_EXCL_STOP 232b730f8bSJeremy L Thompson } 246a6224a1SJeremy L Thompson 256a6224a1SJeremy L Thompson // Add QFunction fields 266a6224a1SJeremy L Thompson const CeedInt dim = 2, num_comp = 3; 272b730f8bSJeremy L Thompson CeedCall(CeedQFunctionAddInput(qf, "du", num_comp * dim, CEED_EVAL_GRAD)); 282b730f8bSJeremy L Thompson CeedCall(CeedQFunctionAddInput(qf, "qdata", dim * (dim + 1) / 2, CEED_EVAL_NONE)); 292b730f8bSJeremy L Thompson CeedCall(CeedQFunctionAddOutput(qf, "dv", num_comp * dim, CEED_EVAL_GRAD)); 306a6224a1SJeremy L Thompson 312b730f8bSJeremy L Thompson CeedCall(CeedQFunctionSetUserFlopsEstimate(qf, num_comp * 6)); 326e15d496SJeremy L Thompson 336a6224a1SJeremy L Thompson return CEED_ERROR_SUCCESS; 346a6224a1SJeremy L Thompson } 356a6224a1SJeremy L Thompson 366a6224a1SJeremy L Thompson /** 37*ea61e9acSJeremy L Thompson @brief Register Ceed QFunction for applying the 2D Poisson operator on a vector system with three components 386a6224a1SJeremy L Thompson **/ 39dfb517d7SJeremy L Thompson CEED_INTERN int CeedQFunctionRegister_Vector3Poisson2DApply(void) { 402b730f8bSJeremy L Thompson return CeedQFunctionRegister("Vector3Poisson2DApply", Vector3Poisson2DApply_loc, 1, Vector3Poisson2DApply, CeedQFunctionInit_Vector3Poisson2DApply); 416a6224a1SJeremy L Thompson } 42