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/ceed.h> 9 #include <ceed/backend.h> 10 #include <string.h> 11 #include <ceed/jit-source/gallery/ceed-poisson2dbuild.h> 12 13 /** 14 @brief Set fields for Ceed QFunction building the geometric data for the 2D 15 Poisson operator 16 **/ 17 static int CeedQFunctionInit_Poisson2DBuild(Ceed ceed, const char *requested, 18 CeedQFunction qf) { 19 int ierr; 20 21 // Check QFunction name 22 const char *name = "Poisson2DBuild"; 23 if (strcmp(name, requested)) 24 // LCOV_EXCL_START 25 return CeedError(ceed, CEED_ERROR_UNSUPPORTED, 26 "QFunction '%s' does not match requested name: %s", 27 name, requested); 28 // LCOV_EXCL_STOP 29 30 // Add QFunction fields 31 const CeedInt dim = 2; 32 ierr = CeedQFunctionAddInput(qf, "dx", dim*dim, CEED_EVAL_GRAD); 33 CeedChk(ierr); 34 ierr = CeedQFunctionAddInput(qf, "weights", 1, CEED_EVAL_WEIGHT); 35 CeedChk(ierr); 36 ierr = CeedQFunctionAddOutput(qf, "qdata", dim*(dim+1)/2, CEED_EVAL_NONE); 37 CeedChk(ierr); 38 39 ierr = CeedQFunctionSetUserFlopsEstimate(qf, 17); CeedChk(ierr); 40 41 return CEED_ERROR_SUCCESS; 42 } 43 44 /** 45 @brief Register Ceed QFunction for building the geometric data for the 2D 46 Poisson operator 47 **/ 48 CEED_INTERN int CeedQFunctionRegister_Poisson2DBuild(void) { 49 return CeedQFunctionRegister("Poisson2DBuild", Poisson2DBuild_loc, 1, 50 Poisson2DBuild, CeedQFunctionInit_Poisson2DBuild); 51 } 52