xref: /libCEED/gallery/poisson-vector/ceed-vectorpoisson3dapply.c (revision 7ed3e4cde27d28430628eaa24b22da48dc51cc32)
1 // Copyright (c) 2017-2018, Lawrence Livermore National Security, LLC.
2 // Produced at the Lawrence Livermore National Laboratory. LLNL-CODE-734707.
3 // All Rights reserved. See files LICENSE and NOTICE for details.
4 //
5 // This file is part of CEED, a collection of benchmarks, miniapps, software
6 // libraries and APIs for efficient high-order finite element and spectral
7 // element discretizations for exascale applications. For more information and
8 // source code availability see http://github.com/ceed.
9 //
10 // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
11 // a collaborative effort of two U.S. Department of Energy organizations (Office
12 // of Science and the National Nuclear Security Administration) responsible for
13 // the planning and preparation of a capable exascale ecosystem, including
14 // software, applications, hardware, advanced system engineering and early
15 // testbed platforms, in support of the nation's exascale computing imperative.
16 
17 #include <ceed/ceed.h>
18 #include <ceed/backend.h>
19 #include <string.h>
20 #include "ceed-vectorpoisson3dapply.h"
21 
22 /**
23   @brief Set fields for Ceed QFunction applying the 3D Poisson operator
24            on a vector system with three components
25 **/
26 static int CeedQFunctionInit_Vector3Poisson3DApply(Ceed ceed,
27     const char *requested,
28     CeedQFunction qf) {
29   int ierr;
30 
31   // Check QFunction name
32   const char *name = "Vector3Poisson3DApply";
33   if (strcmp(name, requested))
34     // LCOV_EXCL_START
35     return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
36                      "QFunction '%s' does not match requested name: %s",
37                      name, requested);
38   // LCOV_EXCL_STOP
39 
40   // Add QFunction fields
41   const CeedInt dim = 3, num_comp = 3;
42   ierr = CeedQFunctionAddInput(qf, "du", num_comp*dim, CEED_EVAL_GRAD);
43   CeedChk(ierr);
44   ierr = CeedQFunctionAddInput(qf, "qdata", dim*(dim+1)/2, CEED_EVAL_NONE);
45   CeedChk(ierr);
46   ierr = CeedQFunctionAddOutput(qf, "dv", num_comp*dim, CEED_EVAL_GRAD);
47   CeedChk(ierr);
48 
49   return CEED_ERROR_SUCCESS;
50 }
51 
52 /**
53   @brief Register Ceed QFunction for applying the 3D Poisson operator
54            on a vector system with three components
55 **/
56 CEED_INTERN int CeedQFunctionRegister_Vector3Poisson3DApply(void) {
57   return CeedQFunctionRegister("Vector3Poisson3DApply", Vector3Poisson3DApply_loc,
58                                1, Vector3Poisson3DApply,
59                                CeedQFunctionInit_Vector3Poisson3DApply);
60 }
61