xref: /libCEED/rust/libceed-sys/c-src/gallery/mass-vector/ceed-vectormassapply.c (revision a0154adecfab8547cdc0febbbf40ac009dbe9d1d)
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-vectormassapply.h>
126a6224a1SJeremy L Thompson 
136a6224a1SJeremy L Thompson /**
146a6224a1SJeremy L Thompson   @brief Set fields for Ceed QFunction for applying the mass matrix
156a6224a1SJeremy L Thompson            on a vector system with three components
166a6224a1SJeremy L Thompson **/
17dfb517d7SJeremy L Thompson static int CeedQFunctionInit_Vector3MassApply(Ceed ceed, const char *requested,
186a6224a1SJeremy L Thompson     CeedQFunction qf) {
196a6224a1SJeremy L Thompson   int ierr;
206a6224a1SJeremy L Thompson 
216a6224a1SJeremy L Thompson   // Check QFunction name
22dfb517d7SJeremy L Thompson   const char *name = "Vector3MassApply";
236a6224a1SJeremy L Thompson   if (strcmp(name, requested))
246a6224a1SJeremy L Thompson     // LCOV_EXCL_START
256a6224a1SJeremy L Thompson     return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
266a6224a1SJeremy L Thompson                      "QFunction '%s' does not match requested name: %s",
276a6224a1SJeremy L Thompson                      name, requested);
286a6224a1SJeremy L Thompson   // LCOV_EXCL_STOP
296a6224a1SJeremy L Thompson 
306a6224a1SJeremy L Thompson   // Add QFunction fields
316a6224a1SJeremy L Thompson   const CeedInt num_comp = 3;
326a6224a1SJeremy L Thompson   ierr = CeedQFunctionAddInput(qf, "u", num_comp, CEED_EVAL_INTERP);
336a6224a1SJeremy L Thompson   CeedChk(ierr);
346a6224a1SJeremy L Thompson   ierr = CeedQFunctionAddInput(qf, "qdata", 1, CEED_EVAL_NONE); CeedChk(ierr);
356a6224a1SJeremy L Thompson   ierr = CeedQFunctionAddOutput(qf, "v", num_comp, CEED_EVAL_INTERP);
366a6224a1SJeremy L Thompson   CeedChk(ierr);
376a6224a1SJeremy L Thompson 
386e15d496SJeremy L Thompson   ierr = CeedQFunctionSetUserFlopsEstimate(qf, num_comp); CeedChk(ierr);
396e15d496SJeremy L Thompson 
406a6224a1SJeremy L Thompson   return CEED_ERROR_SUCCESS;
416a6224a1SJeremy L Thompson }
426a6224a1SJeremy L Thompson 
436a6224a1SJeremy L Thompson /**
446a6224a1SJeremy L Thompson   @brief Register Ceed QFunction for applying the mass matrix on a vector system
456a6224a1SJeremy L Thompson            with three components
466a6224a1SJeremy L Thompson **/
47dfb517d7SJeremy L Thompson CEED_INTERN int CeedQFunctionRegister_Vector3MassApply(void) {
48dfb517d7SJeremy L Thompson   return CeedQFunctionRegister("Vector3MassApply", Vector3MassApply_loc, 1,
49dfb517d7SJeremy L Thompson                                Vector3MassApply, CeedQFunctionInit_Vector3MassApply);
506a6224a1SJeremy L Thompson }
51