xref: /libCEED/interface/ceed-operator.c (revision 5107b09fcb4710dffb8bf2363d6d7d4be3d24cc9)
1d7b241e6Sjeremylt // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at
2d7b241e6Sjeremylt // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights
3d7b241e6Sjeremylt // reserved. See files LICENSE and NOTICE for details.
4d7b241e6Sjeremylt //
5d7b241e6Sjeremylt // This file is part of CEED, a collection of benchmarks, miniapps, software
6d7b241e6Sjeremylt // libraries and APIs for efficient high-order finite element and spectral
7d7b241e6Sjeremylt // element discretizations for exascale applications. For more information and
8d7b241e6Sjeremylt // source code availability see http://github.com/ceed.
9d7b241e6Sjeremylt //
10d7b241e6Sjeremylt // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
11d7b241e6Sjeremylt // a collaborative effort of two U.S. Department of Energy organizations (Office
12d7b241e6Sjeremylt // of Science and the National Nuclear Security Administration) responsible for
13d7b241e6Sjeremylt // the planning and preparation of a capable exascale ecosystem, including
14d7b241e6Sjeremylt // software, applications, hardware, advanced system engineering and early
15d7b241e6Sjeremylt // testbed platforms, in support of the nation's exascale computing imperative.
16d7b241e6Sjeremylt 
17d7b241e6Sjeremylt #include <ceed-impl.h>
18d863ab9bSjeremylt #include <ceed-backend.h>
19d7b241e6Sjeremylt #include <string.h>
20d7b241e6Sjeremylt 
21dfdf5a53Sjeremylt /// @file
22dfdf5a53Sjeremylt /// Implementation of public CeedOperator interfaces
23dfdf5a53Sjeremylt ///
24dfdf5a53Sjeremylt /// @addtogroup CeedOperator
25dfdf5a53Sjeremylt ///   @{
26d7b241e6Sjeremylt 
27d7b241e6Sjeremylt /**
280219ea01SJeremy L Thompson   @brief Create a CeedOperator and associate a CeedQFunction. A CeedBasis and
290219ea01SJeremy L Thompson            CeedElemRestriction can be associated with CeedQFunction fields with
300219ea01SJeremy L Thompson            \ref CeedOperatorSetField.
31d7b241e6Sjeremylt 
32b11c1e72Sjeremylt   @param ceed    A Ceed object where the CeedOperator will be created
33d7b241e6Sjeremylt   @param qf      QFunction defining the action of the operator at quadrature points
34d7b241e6Sjeremylt   @param dqf     QFunction defining the action of the Jacobian of @a qf (or NULL)
35d7b241e6Sjeremylt   @param dqfT    QFunction defining the action of the transpose of the Jacobian
36d7b241e6Sjeremylt                    of @a qf (or NULL)
37b11c1e72Sjeremylt   @param[out] op Address of the variable where the newly created
38b11c1e72Sjeremylt                      CeedOperator will be stored
39b11c1e72Sjeremylt 
40b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
41dfdf5a53Sjeremylt 
42dfdf5a53Sjeremylt   @ref Basic
43d7b241e6Sjeremylt  */
44d7b241e6Sjeremylt int CeedOperatorCreate(Ceed ceed, CeedQFunction qf, CeedQFunction dqf,
45d7b241e6Sjeremylt                        CeedQFunction dqfT, CeedOperator *op) {
46d7b241e6Sjeremylt   int ierr;
47d7b241e6Sjeremylt 
485fe0d4faSjeremylt   if (!ceed->OperatorCreate) {
495fe0d4faSjeremylt     Ceed delegate;
50aefd8378Sjeremylt     ierr = CeedGetObjectDelegate(ceed, &delegate, "Operator"); CeedChk(ierr);
515fe0d4faSjeremylt 
525fe0d4faSjeremylt     if (!delegate)
53c042f62fSJeremy L Thompson       // LCOV_EXCL_START
545fe0d4faSjeremylt       return CeedError(ceed, 1, "Backend does not support OperatorCreate");
55c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
565fe0d4faSjeremylt 
575fe0d4faSjeremylt     ierr = CeedOperatorCreate(delegate, qf, dqf, dqfT, op); CeedChk(ierr);
585fe0d4faSjeremylt     return 0;
595fe0d4faSjeremylt   }
605fe0d4faSjeremylt 
61d7b241e6Sjeremylt   ierr = CeedCalloc(1, op); CeedChk(ierr);
62d7b241e6Sjeremylt   (*op)->ceed = ceed;
63d7b241e6Sjeremylt   ceed->refcount++;
64d7b241e6Sjeremylt   (*op)->refcount = 1;
65442e7f0bSjeremylt   if (qf == CEED_QFUNCTION_NONE)
66442e7f0bSjeremylt     // LCOV_EXCL_START
67442e7f0bSjeremylt     return CeedError(ceed, 1, "Operator must have a valid QFunction.");
68442e7f0bSjeremylt   // LCOV_EXCL_STOP
69d7b241e6Sjeremylt   (*op)->qf = qf;
70d7b241e6Sjeremylt   qf->refcount++;
71442e7f0bSjeremylt   if (dqf && dqf != CEED_QFUNCTION_NONE) {
72d7b241e6Sjeremylt     (*op)->dqf = dqf;
73442e7f0bSjeremylt     dqf->refcount++;
74442e7f0bSjeremylt   }
75442e7f0bSjeremylt   if (dqfT && dqfT != CEED_QFUNCTION_NONE) {
76d7b241e6Sjeremylt     (*op)->dqfT = dqfT;
77442e7f0bSjeremylt     dqfT->refcount++;
78442e7f0bSjeremylt   }
79fe2413ffSjeremylt   ierr = CeedCalloc(16, &(*op)->inputfields); CeedChk(ierr);
80fe2413ffSjeremylt   ierr = CeedCalloc(16, &(*op)->outputfields); CeedChk(ierr);
81d7b241e6Sjeremylt   ierr = ceed->OperatorCreate(*op); CeedChk(ierr);
82d7b241e6Sjeremylt   return 0;
83d7b241e6Sjeremylt }
84d7b241e6Sjeremylt 
85d7b241e6Sjeremylt /**
8652d6035fSJeremy L Thompson   @brief Create an operator that composes the action of several operators
8752d6035fSJeremy L Thompson 
8852d6035fSJeremy L Thompson   @param ceed    A Ceed object where the CeedOperator will be created
8952d6035fSJeremy L Thompson   @param[out] op Address of the variable where the newly created
9052d6035fSJeremy L Thompson                      Composite CeedOperator will be stored
9152d6035fSJeremy L Thompson 
9252d6035fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
9352d6035fSJeremy L Thompson 
9452d6035fSJeremy L Thompson   @ref Basic
9552d6035fSJeremy L Thompson  */
9652d6035fSJeremy L Thompson int CeedCompositeOperatorCreate(Ceed ceed, CeedOperator *op) {
9752d6035fSJeremy L Thompson   int ierr;
9852d6035fSJeremy L Thompson 
9952d6035fSJeremy L Thompson   if (!ceed->CompositeOperatorCreate) {
10052d6035fSJeremy L Thompson     Ceed delegate;
101aefd8378Sjeremylt     ierr = CeedGetObjectDelegate(ceed, &delegate, "Operator"); CeedChk(ierr);
10252d6035fSJeremy L Thompson 
10352d6035fSJeremy L Thompson     if (!delegate)
104c042f62fSJeremy L Thompson       // LCOV_EXCL_START
1051d102b48SJeremy L Thompson       return CeedError(ceed, 1, "Backend does not support "
1061d102b48SJeremy L Thompson                        "CompositeOperatorCreate");
107c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
10852d6035fSJeremy L Thompson 
10952d6035fSJeremy L Thompson     ierr = CeedCompositeOperatorCreate(delegate, op); CeedChk(ierr);
11052d6035fSJeremy L Thompson     return 0;
11152d6035fSJeremy L Thompson   }
11252d6035fSJeremy L Thompson 
11352d6035fSJeremy L Thompson   ierr = CeedCalloc(1,op); CeedChk(ierr);
11452d6035fSJeremy L Thompson   (*op)->ceed = ceed;
11552d6035fSJeremy L Thompson   ceed->refcount++;
11652d6035fSJeremy L Thompson   (*op)->composite = true;
11752d6035fSJeremy L Thompson   ierr = CeedCalloc(16, &(*op)->suboperators); CeedChk(ierr);
11852d6035fSJeremy L Thompson   ierr = ceed->CompositeOperatorCreate(*op); CeedChk(ierr);
11952d6035fSJeremy L Thompson   return 0;
12052d6035fSJeremy L Thompson }
12152d6035fSJeremy L Thompson 
12252d6035fSJeremy L Thompson /**
123b11c1e72Sjeremylt   @brief Provide a field to a CeedOperator for use by its CeedQFunction
124d7b241e6Sjeremylt 
125d7b241e6Sjeremylt   This function is used to specify both active and passive fields to a
126d7b241e6Sjeremylt   CeedOperator.  For passive fields, a vector @arg v must be provided.  Passive
127d7b241e6Sjeremylt   fields can inputs or outputs (updated in-place when operator is applied).
128d7b241e6Sjeremylt 
129d7b241e6Sjeremylt   Active fields must be specified using this function, but their data (in a
130d7b241e6Sjeremylt   CeedVector) is passed in CeedOperatorApply().  There can be at most one active
131d7b241e6Sjeremylt   input and at most one active output.
132d7b241e6Sjeremylt 
1338c91a0c9SJeremy L Thompson   @param op         CeedOperator on which to provide the field
1348795c945Sjeremylt   @param fieldname  Name of the field (to be matched with the name used by
1358795c945Sjeremylt                       CeedQFunction)
136b11c1e72Sjeremylt   @param r          CeedElemRestriction
137b0e29e78Sjeremylt   @param lmode      CeedTransposeMode which specifies the ordering of the
138b0e29e78Sjeremylt                       components of the l-vector used by this CeedOperatorField,
139b0e29e78Sjeremylt                       CEED_NOTRANSPOSE indicates the component is the
140b0e29e78Sjeremylt                       outermost index and CEED_TRANSPOSE indicates the component
141b0e29e78Sjeremylt                       is the innermost index in ordering of the l-vector
142783c99b3SValeria Barra   @param b          CeedBasis in which the field resides or CEED_BASIS_COLLOCATED
143b11c1e72Sjeremylt                       if collocated with quadrature points
144b11c1e72Sjeremylt   @param v          CeedVector to be used by CeedOperator or CEED_VECTOR_ACTIVE
145b11c1e72Sjeremylt                       if field is active or CEED_VECTOR_NONE if using
1468c91a0c9SJeremy L Thompson                       CEED_EVAL_WEIGHT in the QFunction
147b11c1e72Sjeremylt 
148b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
149dfdf5a53Sjeremylt 
150dfdf5a53Sjeremylt   @ref Basic
151b11c1e72Sjeremylt **/
152d7b241e6Sjeremylt int CeedOperatorSetField(CeedOperator op, const char *fieldname,
1534dccadb6Sjeremylt                          CeedElemRestriction r, CeedTransposeMode lmode,
1544dccadb6Sjeremylt                          CeedBasis b, CeedVector v) {
155d7b241e6Sjeremylt   int ierr;
15652d6035fSJeremy L Thompson   if (op->composite)
157c042f62fSJeremy L Thompson     // LCOV_EXCL_START
15852d6035fSJeremy L Thompson     return CeedError(op->ceed, 1, "Cannot add field to composite operator.");
159c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
1608b067b84SJed Brown   if (!r)
161c042f62fSJeremy L Thompson     // LCOV_EXCL_START
162c042f62fSJeremy L Thompson     return CeedError(op->ceed, 1,
163c042f62fSJeremy L Thompson                      "ElemRestriction r for field \"%s\" must be non-NULL.",
164c042f62fSJeremy L Thompson                      fieldname);
165c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
1668b067b84SJed Brown   if (!b)
167c042f62fSJeremy L Thompson     // LCOV_EXCL_START
168c042f62fSJeremy L Thompson     return CeedError(op->ceed, 1, "Basis b for field \"%s\" must be non-NULL.",
169c042f62fSJeremy L Thompson                      fieldname);
170c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
1718b067b84SJed Brown   if (!v)
172c042f62fSJeremy L Thompson     // LCOV_EXCL_START
173c042f62fSJeremy L Thompson     return CeedError(op->ceed, 1, "Vector v for field \"%s\" must be non-NULL.",
174c042f62fSJeremy L Thompson                      fieldname);
175c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
17652d6035fSJeremy L Thompson 
177d7b241e6Sjeremylt   CeedInt numelements;
178d7b241e6Sjeremylt   ierr = CeedElemRestrictionGetNumElements(r, &numelements); CeedChk(ierr);
1792cb0afc5Sjeremylt   if (op->hasrestriction && op->numelements != numelements)
180c042f62fSJeremy L Thompson     // LCOV_EXCL_START
181d7b241e6Sjeremylt     return CeedError(op->ceed, 1,
1821d102b48SJeremy L Thompson                      "ElemRestriction with %d elements incompatible with prior "
1831d102b48SJeremy L Thompson                      "%d elements", numelements, op->numelements);
184c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
185d7b241e6Sjeremylt   op->numelements = numelements;
1862cb0afc5Sjeremylt   op->hasrestriction = true; // Restriction set, but numelements may be 0
187d7b241e6Sjeremylt 
188783c99b3SValeria Barra   if (b != CEED_BASIS_COLLOCATED) {
189d7b241e6Sjeremylt     CeedInt numqpoints;
190d7b241e6Sjeremylt     ierr = CeedBasisGetNumQuadraturePoints(b, &numqpoints); CeedChk(ierr);
191d7b241e6Sjeremylt     if (op->numqpoints && op->numqpoints != numqpoints)
192c042f62fSJeremy L Thompson       // LCOV_EXCL_START
1931d102b48SJeremy L Thompson       return CeedError(op->ceed, 1, "Basis with %d quadrature points "
1941d102b48SJeremy L Thompson                        "incompatible with prior %d points", numqpoints,
1951d102b48SJeremy L Thompson                        op->numqpoints);
196c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
197d7b241e6Sjeremylt     op->numqpoints = numqpoints;
198d7b241e6Sjeremylt   }
199d1bcdac9Sjeremylt   CeedOperatorField *ofield;
200d7b241e6Sjeremylt   for (CeedInt i=0; i<op->qf->numinputfields; i++) {
201fe2413ffSjeremylt     if (!strcmp(fieldname, (*op->qf->inputfields[i]).fieldname)) {
202d7b241e6Sjeremylt       ofield = &op->inputfields[i];
203d7b241e6Sjeremylt       goto found;
204d7b241e6Sjeremylt     }
205d7b241e6Sjeremylt   }
206d7b241e6Sjeremylt   for (CeedInt i=0; i<op->qf->numoutputfields; i++) {
207fe2413ffSjeremylt     if (!strcmp(fieldname, (*op->qf->outputfields[i]).fieldname)) {
208d7b241e6Sjeremylt       ofield = &op->outputfields[i];
209d7b241e6Sjeremylt       goto found;
210d7b241e6Sjeremylt     }
211d7b241e6Sjeremylt   }
212c042f62fSJeremy L Thompson   // LCOV_EXCL_START
213d7b241e6Sjeremylt   return CeedError(op->ceed, 1, "QFunction has no knowledge of field '%s'",
214d7b241e6Sjeremylt                    fieldname);
215c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
216d7b241e6Sjeremylt found:
217fe2413ffSjeremylt   ierr = CeedCalloc(1, ofield); CeedChk(ierr);
218fe2413ffSjeremylt   (*ofield)->Erestrict = r;
21971352a93Sjeremylt   r->refcount += 1;
220fe2413ffSjeremylt   (*ofield)->lmode = lmode;
221fe2413ffSjeremylt   (*ofield)->basis = b;
22271352a93Sjeremylt   if (b != CEED_BASIS_COLLOCATED)
22371352a93Sjeremylt     b->refcount += 1;
224fe2413ffSjeremylt   (*ofield)->vec = v;
22571352a93Sjeremylt   if (v != CEED_VECTOR_ACTIVE && v != CEED_VECTOR_NONE)
22671352a93Sjeremylt     v->refcount += 1;
227d7b241e6Sjeremylt   op->nfields += 1;
228d7b241e6Sjeremylt   return 0;
229d7b241e6Sjeremylt }
230d7b241e6Sjeremylt 
231d7b241e6Sjeremylt /**
23252d6035fSJeremy L Thompson   @brief Add a sub-operator to a composite CeedOperator
233288c0443SJeremy L Thompson 
234288c0443SJeremy L Thompson   @param[out] compositeop Address of the composite CeedOperator
235288c0443SJeremy L Thompson   @param      subop       Address of the sub-operator CeedOperator
23652d6035fSJeremy L Thompson 
23752d6035fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
23852d6035fSJeremy L Thompson 
23952d6035fSJeremy L Thompson   @ref Basic
24052d6035fSJeremy L Thompson  */
241692c2638Sjeremylt int CeedCompositeOperatorAddSub(CeedOperator compositeop, CeedOperator subop) {
24252d6035fSJeremy L Thompson   if (!compositeop->composite)
243c042f62fSJeremy L Thompson     // LCOV_EXCL_START
2441d102b48SJeremy L Thompson     return CeedError(compositeop->ceed, 1, "CeedOperator is not a composite "
2451d102b48SJeremy L Thompson                      "operator");
246c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
24752d6035fSJeremy L Thompson 
24852d6035fSJeremy L Thompson   if (compositeop->numsub == CEED_COMPOSITE_MAX)
249c042f62fSJeremy L Thompson     // LCOV_EXCL_START
2501d102b48SJeremy L Thompson     return CeedError(compositeop->ceed, 1, "Cannot add additional suboperators");
251c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
25252d6035fSJeremy L Thompson 
25352d6035fSJeremy L Thompson   compositeop->suboperators[compositeop->numsub] = subop;
25452d6035fSJeremy L Thompson   subop->refcount++;
25552d6035fSJeremy L Thompson   compositeop->numsub++;
25652d6035fSJeremy L Thompson   return 0;
25752d6035fSJeremy L Thompson }
25852d6035fSJeremy L Thompson 
25952d6035fSJeremy L Thompson /**
260*5107b09fSJeremy L Thompson   @brief Duplicate a CeedOperator with a reference Ceed to fallback for advanced
261*5107b09fSJeremy L Thompson            CeedOperator functionality
262*5107b09fSJeremy L Thompson 
263*5107b09fSJeremy L Thompson   @param op           CeedOperator to create fallback for
264*5107b09fSJeremy L Thompson 
265*5107b09fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
266*5107b09fSJeremy L Thompson 
267*5107b09fSJeremy L Thompson   @ref Developer
268*5107b09fSJeremy L Thompson **/
269*5107b09fSJeremy L Thompson int CeedOperatorCreateFallback(CeedOperator op) {
270*5107b09fSJeremy L Thompson   int ierr;
271*5107b09fSJeremy L Thompson 
272*5107b09fSJeremy L Thompson   // Fallback Ceed
273*5107b09fSJeremy L Thompson   const char *resource, *fallbackresource;
274*5107b09fSJeremy L Thompson   ierr = CeedGetResource(op->ceed, &resource); CeedChk(ierr);
275*5107b09fSJeremy L Thompson   ierr = CeedGetOperatorFallbackResource(op->ceed, &fallbackresource);
276*5107b09fSJeremy L Thompson   CeedChk(ierr);
277*5107b09fSJeremy L Thompson   if (!strcmp(resource, fallbackresource))
278*5107b09fSJeremy L Thompson     // LCOV_EXCL_START
279*5107b09fSJeremy L Thompson     return CeedError(op->ceed, 1, "Backend %s cannot create an operator"
280*5107b09fSJeremy L Thompson                      "fallback to resource %s", resource, fallbackresource);
281*5107b09fSJeremy L Thompson   // LCOV_EXCL_STOP
282*5107b09fSJeremy L Thompson 
283*5107b09fSJeremy L Thompson   Ceed ceedref;
284*5107b09fSJeremy L Thompson   ierr = CeedInit(fallbackresource, &ceedref); CeedChk(ierr);
285*5107b09fSJeremy L Thompson   ceedref->opfallbackparent = op->ceed;
286*5107b09fSJeremy L Thompson   op->ceed->opfallbackceed = ceedref;
287*5107b09fSJeremy L Thompson 
288*5107b09fSJeremy L Thompson   // Clone Op
289*5107b09fSJeremy L Thompson   CeedOperator opref;
290*5107b09fSJeremy L Thompson   ierr = CeedCalloc(1, &opref); CeedChk(ierr);
291*5107b09fSJeremy L Thompson   memcpy(opref, op, sizeof(*opref)); CeedChk(ierr);
292*5107b09fSJeremy L Thompson   opref->data = NULL;
293*5107b09fSJeremy L Thompson   opref->setupdone = 0;
294*5107b09fSJeremy L Thompson   opref->ceed = ceedref;
295*5107b09fSJeremy L Thompson   ierr = ceedref->OperatorCreate(opref); CeedChk(ierr);
296*5107b09fSJeremy L Thompson   op->opfallback = opref;
297*5107b09fSJeremy L Thompson 
298*5107b09fSJeremy L Thompson   // Clone QF
299*5107b09fSJeremy L Thompson   CeedQFunction qfref;
300*5107b09fSJeremy L Thompson   ierr = CeedCalloc(1, &qfref); CeedChk(ierr);
301*5107b09fSJeremy L Thompson   memcpy(qfref, (op->qf), sizeof(*qfref)); CeedChk(ierr);
302*5107b09fSJeremy L Thompson   qfref->data = NULL;
303*5107b09fSJeremy L Thompson   qfref->ceed = ceedref;
304*5107b09fSJeremy L Thompson   ierr = ceedref->QFunctionCreate(qfref); CeedChk(ierr);
305*5107b09fSJeremy L Thompson   opref->qf = qfref;
306*5107b09fSJeremy L Thompson   op->qffallback = qfref;
307*5107b09fSJeremy L Thompson 
308*5107b09fSJeremy L Thompson   return 0;
309*5107b09fSJeremy L Thompson }
310*5107b09fSJeremy L Thompson 
311*5107b09fSJeremy L Thompson /**
3121d102b48SJeremy L Thompson   @brief Assemble a linear CeedQFunction associated with a CeedOperator
3131d102b48SJeremy L Thompson 
3141d102b48SJeremy L Thompson   This returns a CeedVector containing a matrix at each quadrature point
3151d102b48SJeremy L Thompson     providing the action of the CeedQFunction associated with the CeedOperator.
3161d102b48SJeremy L Thompson     The vector 'assembled' is of shape
3171d102b48SJeremy L Thompson       [num_elements, num_input_fields, num_output_fields, num_quad_points]
3181d102b48SJeremy L Thompson     and contains column-major matrices representing the action of the
3191d102b48SJeremy L Thompson     CeedQFunction for a corresponding quadrature point on an element. Inputs and
3201d102b48SJeremy L Thompson     outputs are in the order provided by the user when adding CeedOperator fields.
3211d102b48SJeremy L Thompson     For example, a QFunction with inputs 'u' and 'gradu' and outputs 'gradv' and
3221d102b48SJeremy L Thompson     'v', provided in that order, would result in an assembled QFunction that
3231d102b48SJeremy L Thompson     consists of (1 + dim) x (dim + 1) matrices at each quadrature point acting
3241d102b48SJeremy L Thompson     on the input [u, du_0, du_1] and producing the output [dv_0, dv_1, v].
3251d102b48SJeremy L Thompson 
3261d102b48SJeremy L Thompson   @param op             CeedOperator to assemble CeedQFunction
3271d102b48SJeremy L Thompson   @param[out] assembled CeedVector to store assembled CeedQFunction at
3281d102b48SJeremy L Thompson                           quadrature points
3291d102b48SJeremy L Thompson   @param[out] rstr      CeedElemRestriction for CeedVector containing assembled
3301d102b48SJeremy L Thompson                           CeedQFunction
3311d102b48SJeremy L Thompson   @param request        Address of CeedRequest for non-blocking completion, else
3321d102b48SJeremy L Thompson                           CEED_REQUEST_IMMEDIATE
3331d102b48SJeremy L Thompson 
3341d102b48SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
3351d102b48SJeremy L Thompson 
336b7ec98d8SJeremy L Thompson   @ref Basic
3371d102b48SJeremy L Thompson **/
3381d102b48SJeremy L Thompson int CeedOperatorAssembleLinearQFunction(CeedOperator op, CeedVector *assembled,
3397f823360Sjeremylt                                         CeedElemRestriction *rstr,
3407f823360Sjeremylt                                         CeedRequest *request) {
3411d102b48SJeremy L Thompson   int ierr;
3421d102b48SJeremy L Thompson   Ceed ceed = op->ceed;
3431d102b48SJeremy L Thompson   CeedQFunction qf = op->qf;
3441d102b48SJeremy L Thompson 
3451d102b48SJeremy L Thompson   if (op->composite) {
346138d4072Sjeremylt     // LCOV_EXCL_START
3471d102b48SJeremy L Thompson     return CeedError(ceed, 1, "Cannot assemble QFunction for composite operator");
348138d4072Sjeremylt     // LCOV_EXCL_STOP
3491d102b48SJeremy L Thompson   } else {
3501d102b48SJeremy L Thompson     if (op->nfields == 0)
351138d4072Sjeremylt       // LCOV_EXCL_START
3521d102b48SJeremy L Thompson       return CeedError(ceed, 1, "No operator fields set");
353138d4072Sjeremylt     // LCOV_EXCL_STOP
3541d102b48SJeremy L Thompson     if (op->nfields < qf->numinputfields + qf->numoutputfields)
355138d4072Sjeremylt       // LCOV_EXCL_START
3561d102b48SJeremy L Thompson       return CeedError( ceed, 1, "Not all operator fields set");
357138d4072Sjeremylt     // LCOV_EXCL_STOP
3582cb0afc5Sjeremylt     if (!op->hasrestriction)
359138d4072Sjeremylt       // LCOV_EXCL_START
3601d102b48SJeremy L Thompson       return CeedError(ceed, 1, "At least one restriction required");
361138d4072Sjeremylt     // LCOV_EXCL_STOP
3621d102b48SJeremy L Thompson     if (op->numqpoints == 0)
363138d4072Sjeremylt       // LCOV_EXCL_START
3641d102b48SJeremy L Thompson       return CeedError(ceed, 1, "At least one non-collocated basis required");
365138d4072Sjeremylt     // LCOV_EXCL_STOP
3661d102b48SJeremy L Thompson   }
367*5107b09fSJeremy L Thompson   if (op->AssembleLinearQFunction) {
3681d102b48SJeremy L Thompson     ierr = op->AssembleLinearQFunction(op, assembled, rstr, request);
3691d102b48SJeremy L Thompson     CeedChk(ierr);
370*5107b09fSJeremy L Thompson   } else {
371*5107b09fSJeremy L Thompson     // Fallback to reference Ceed
372*5107b09fSJeremy L Thompson     if (!op->opfallback) {
373*5107b09fSJeremy L Thompson       ierr = CeedOperatorCreateFallback(op); CeedChk(ierr);
374*5107b09fSJeremy L Thompson     }
375*5107b09fSJeremy L Thompson     // Assemble
376*5107b09fSJeremy L Thompson     ierr = op->opfallback->AssembleLinearQFunction(op->opfallback, assembled,
377*5107b09fSJeremy L Thompson            rstr, request); CeedChk(ierr);
378*5107b09fSJeremy L Thompson   }
3791d102b48SJeremy L Thompson   return 0;
3801d102b48SJeremy L Thompson }
3811d102b48SJeremy L Thompson 
3821d102b48SJeremy L Thompson /**
383b7ec98d8SJeremy L Thompson   @brief Assemble the diagonal of a square linear Operator
384b7ec98d8SJeremy L Thompson 
385b7ec98d8SJeremy L Thompson   This returns a CeedVector containing the diagonal of a linear CeedOperator.
386b7ec98d8SJeremy L Thompson 
387b7ec98d8SJeremy L Thompson   @param op             CeedOperator to assemble CeedQFunction
388b7ec98d8SJeremy L Thompson   @param[out] assembled CeedVector to store assembled CeedOperator diagonal
389b7ec98d8SJeremy L Thompson   @param request        Address of CeedRequest for non-blocking completion, else
390b7ec98d8SJeremy L Thompson                           CEED_REQUEST_IMMEDIATE
391b7ec98d8SJeremy L Thompson 
392b7ec98d8SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
393b7ec98d8SJeremy L Thompson 
394b7ec98d8SJeremy L Thompson   @ref Basic
395b7ec98d8SJeremy L Thompson **/
3967f823360Sjeremylt int CeedOperatorAssembleLinearDiagonal(CeedOperator op, CeedVector *assembled,
3977f823360Sjeremylt                                        CeedRequest *request) {
398b7ec98d8SJeremy L Thompson   int ierr;
399b7ec98d8SJeremy L Thompson   Ceed ceed = op->ceed;
400b7ec98d8SJeremy L Thompson   CeedQFunction qf = op->qf;
401b7ec98d8SJeremy L Thompson 
402b7ec98d8SJeremy L Thompson   if (op->composite) {
403b7ec98d8SJeremy L Thompson     // LCOV_EXCL_START
404b7ec98d8SJeremy L Thompson     return CeedError(ceed, 1, "Cannot assemble QFunction for composite operator");
405b7ec98d8SJeremy L Thompson     // LCOV_EXCL_STOP
406b7ec98d8SJeremy L Thompson   } else {
407b7ec98d8SJeremy L Thompson     if (op->nfields == 0)
408b7ec98d8SJeremy L Thompson       // LCOV_EXCL_START
409b7ec98d8SJeremy L Thompson       return CeedError(ceed, 1, "No operator fields set");
410b7ec98d8SJeremy L Thompson     // LCOV_EXCL_STOP
411b7ec98d8SJeremy L Thompson     if (op->nfields < qf->numinputfields + qf->numoutputfields)
412b7ec98d8SJeremy L Thompson       // LCOV_EXCL_START
413b7ec98d8SJeremy L Thompson       return CeedError( ceed, 1, "Not all operator fields set");
414b7ec98d8SJeremy L Thompson     // LCOV_EXCL_STOP
4152cb0afc5Sjeremylt     if (!op->hasrestriction)
416b7ec98d8SJeremy L Thompson       // LCOV_EXCL_START
417b7ec98d8SJeremy L Thompson       return CeedError(ceed, 1, "At least one restriction required");
418b7ec98d8SJeremy L Thompson     // LCOV_EXCL_STOP
419b7ec98d8SJeremy L Thompson     if (op->numqpoints == 0)
420b7ec98d8SJeremy L Thompson       // LCOV_EXCL_START
421b7ec98d8SJeremy L Thompson       return CeedError(ceed, 1, "At least one non-collocated basis required");
422b7ec98d8SJeremy L Thompson     // LCOV_EXCL_STOP
423b7ec98d8SJeremy L Thompson   }
424b7ec98d8SJeremy L Thompson 
425b7ec98d8SJeremy L Thompson   // Use backend version, if available
426b7ec98d8SJeremy L Thompson   if (op->AssembleLinearDiagonal) {
427b7ec98d8SJeremy L Thompson     ierr = op->AssembleLinearDiagonal(op, assembled, request); CeedChk(ierr);
428b7ec98d8SJeremy L Thompson     return 0;
429b7ec98d8SJeremy L Thompson   }
430b7ec98d8SJeremy L Thompson 
431b7ec98d8SJeremy L Thompson   // Assemble QFunction
432b7ec98d8SJeremy L Thompson   CeedVector assembledqf;
433b7ec98d8SJeremy L Thompson   CeedElemRestriction rstr;
434b7ec98d8SJeremy L Thompson   ierr = CeedOperatorAssembleLinearQFunction(op,  &assembledqf, &rstr, request);
435b7ec98d8SJeremy L Thompson   CeedChk(ierr);
436b7ec98d8SJeremy L Thompson   ierr = CeedElemRestrictionDestroy(&rstr); CeedChk(ierr);
437b7ec98d8SJeremy L Thompson 
438b7ec98d8SJeremy L Thompson   // Determine active input basis
439112e3f70Sjeremylt   CeedInt numemodein = 0, ncomp, dim = 1;
440b7ec98d8SJeremy L Thompson   CeedEvalMode *emodein = NULL;
441112e3f70Sjeremylt   CeedBasis basisin = NULL;
442112e3f70Sjeremylt   CeedElemRestriction rstrin = NULL;
443b7ec98d8SJeremy L Thompson   for (CeedInt i=0; i<qf->numinputfields; i++)
444b7ec98d8SJeremy L Thompson     if (op->inputfields[i]->vec == CEED_VECTOR_ACTIVE) {
445b7ec98d8SJeremy L Thompson       ierr = CeedOperatorFieldGetBasis(op->inputfields[i], &basisin);
446b7ec98d8SJeremy L Thompson       CeedChk(ierr);
447b7ec98d8SJeremy L Thompson       ierr = CeedBasisGetNumComponents(basisin, &ncomp); CeedChk(ierr);
448b7ec98d8SJeremy L Thompson       ierr = CeedBasisGetDimension(basisin, &dim); CeedChk(ierr);
449b7ec98d8SJeremy L Thompson       ierr = CeedOperatorFieldGetElemRestriction(op->inputfields[i], &rstrin);
450b7ec98d8SJeremy L Thompson       CeedChk(ierr);
451b7ec98d8SJeremy L Thompson       CeedEvalMode emode;
452b7ec98d8SJeremy L Thompson       ierr = CeedQFunctionFieldGetEvalMode(qf->inputfields[i], &emode);
453b7ec98d8SJeremy L Thompson       CeedChk(ierr);
454b7ec98d8SJeremy L Thompson       switch (emode) {
455b7ec98d8SJeremy L Thompson       case CEED_EVAL_NONE:
456b7ec98d8SJeremy L Thompson       case CEED_EVAL_INTERP:
457b7ec98d8SJeremy L Thompson         ierr = CeedRealloc(numemodein + 1, &emodein); CeedChk(ierr);
458b7ec98d8SJeremy L Thompson         emodein[numemodein] = emode;
459b7ec98d8SJeremy L Thompson         numemodein += 1;
460b7ec98d8SJeremy L Thompson         break;
461b7ec98d8SJeremy L Thompson       case CEED_EVAL_GRAD:
462b7ec98d8SJeremy L Thompson         ierr = CeedRealloc(numemodein + dim, &emodein); CeedChk(ierr);
463b7ec98d8SJeremy L Thompson         for (CeedInt d=0; d<dim; d++)
464b7ec98d8SJeremy L Thompson           emodein[numemodein+d] = emode;
465b7ec98d8SJeremy L Thompson         numemodein += dim;
466b7ec98d8SJeremy L Thompson         break;
467b7ec98d8SJeremy L Thompson       case CEED_EVAL_WEIGHT:
468b7ec98d8SJeremy L Thompson       case CEED_EVAL_DIV:
469b7ec98d8SJeremy L Thompson       case CEED_EVAL_CURL:
470b7ec98d8SJeremy L Thompson         break; // Caught by QF Assembly
471b7ec98d8SJeremy L Thompson       }
472b7ec98d8SJeremy L Thompson     }
473b7ec98d8SJeremy L Thompson 
474b7ec98d8SJeremy L Thompson   // Determine active output basis
475b7ec98d8SJeremy L Thompson   CeedInt numemodeout = 0;
476b7ec98d8SJeremy L Thompson   CeedEvalMode *emodeout = NULL;
477112e3f70Sjeremylt   CeedBasis basisout = NULL;
478112e3f70Sjeremylt   CeedElemRestriction rstrout = NULL;
479112e3f70Sjeremylt   CeedTransposeMode lmodeout = CEED_NOTRANSPOSE;
480b7ec98d8SJeremy L Thompson   for (CeedInt i=0; i<qf->numoutputfields; i++)
481b7ec98d8SJeremy L Thompson     if (op->outputfields[i]->vec == CEED_VECTOR_ACTIVE) {
482b7ec98d8SJeremy L Thompson       ierr = CeedOperatorFieldGetBasis(op->outputfields[i], &basisout);
483b7ec98d8SJeremy L Thompson       CeedChk(ierr);
484b7ec98d8SJeremy L Thompson       ierr = CeedOperatorFieldGetElemRestriction(op->outputfields[i], &rstrout);
485b7ec98d8SJeremy L Thompson       CeedChk(ierr);
486b7ec98d8SJeremy L Thompson       ierr = CeedOperatorFieldGetLMode(op->outputfields[i], &lmodeout);
487b7ec98d8SJeremy L Thompson       CeedChk(ierr);
488b7ec98d8SJeremy L Thompson       CeedEvalMode emode;
489b7ec98d8SJeremy L Thompson       ierr = CeedQFunctionFieldGetEvalMode(qf->outputfields[i], &emode);
490b7ec98d8SJeremy L Thompson       CeedChk(ierr);
491b7ec98d8SJeremy L Thompson       switch (emode) {
492b7ec98d8SJeremy L Thompson       case CEED_EVAL_NONE:
493b7ec98d8SJeremy L Thompson       case CEED_EVAL_INTERP:
494b7ec98d8SJeremy L Thompson         ierr = CeedRealloc(numemodeout + 1, &emodeout); CeedChk(ierr);
495b7ec98d8SJeremy L Thompson         emodeout[numemodeout] = emode;
496b7ec98d8SJeremy L Thompson         numemodeout += 1;
497b7ec98d8SJeremy L Thompson         break;
498b7ec98d8SJeremy L Thompson       case CEED_EVAL_GRAD:
499b7ec98d8SJeremy L Thompson         ierr = CeedRealloc(numemodeout + dim, &emodeout); CeedChk(ierr);
500b7ec98d8SJeremy L Thompson         for (CeedInt d=0; d<dim; d++)
501b7ec98d8SJeremy L Thompson           emodeout[numemodeout+d] = emode;
502b7ec98d8SJeremy L Thompson         numemodeout += dim;
503b7ec98d8SJeremy L Thompson         break;
504b7ec98d8SJeremy L Thompson       case CEED_EVAL_WEIGHT:
505b7ec98d8SJeremy L Thompson       case CEED_EVAL_DIV:
506b7ec98d8SJeremy L Thompson       case CEED_EVAL_CURL:
507b7ec98d8SJeremy L Thompson         break; // Caught by QF Assembly
508b7ec98d8SJeremy L Thompson       }
509b7ec98d8SJeremy L Thompson     }
510b7ec98d8SJeremy L Thompson 
511b7ec98d8SJeremy L Thompson   // Create diagonal vector
512b7ec98d8SJeremy L Thompson   CeedVector elemdiag;
513b7ec98d8SJeremy L Thompson   ierr = CeedElemRestrictionCreateVector(rstrin, assembled, &elemdiag);
514b7ec98d8SJeremy L Thompson   CeedChk(ierr);
515b7ec98d8SJeremy L Thompson 
516b7ec98d8SJeremy L Thompson   // Assemble element operator diagonals
517b7ec98d8SJeremy L Thompson   CeedScalar *elemdiagarray, *assembledqfarray;
518b7ec98d8SJeremy L Thompson   ierr = CeedVectorSetValue(elemdiag, 0.0); CeedChk(ierr);
519b7ec98d8SJeremy L Thompson   ierr = CeedVectorGetArray(elemdiag, CEED_MEM_HOST, &elemdiagarray);
520b7ec98d8SJeremy L Thompson   CeedChk(ierr);
521b7ec98d8SJeremy L Thompson   ierr = CeedVectorGetArray(assembledqf, CEED_MEM_HOST, &assembledqfarray);
522b7ec98d8SJeremy L Thompson   CeedChk(ierr);
523b7ec98d8SJeremy L Thompson   CeedInt nelem, nnodes, nqpts;
524b7ec98d8SJeremy L Thompson   ierr = CeedElemRestrictionGetNumElements(rstrin, &nelem); CeedChk(ierr);
525b7ec98d8SJeremy L Thompson   ierr = CeedBasisGetNumNodes(basisin, &nnodes); CeedChk(ierr);
526b7ec98d8SJeremy L Thompson   ierr = CeedBasisGetNumQuadraturePoints(basisin, &nqpts); CeedChk(ierr);
527b7ec98d8SJeremy L Thompson   // Compute the diagonal of B^T D B
528b7ec98d8SJeremy L Thompson   // Each node, qpt pair
529b7ec98d8SJeremy L Thompson   for (CeedInt n=0; n<nnodes; n++)
530b7ec98d8SJeremy L Thompson     for (CeedInt q=0; q<nqpts; q++) {
531b7ec98d8SJeremy L Thompson       CeedInt dout = -1;
532b7ec98d8SJeremy L Thompson       // Each basis eval mode pair
533b7ec98d8SJeremy L Thompson       for (CeedInt eout=0; eout<numemodeout; eout++) {
534b7ec98d8SJeremy L Thompson         CeedScalar bt = 1.0;
535b7ec98d8SJeremy L Thompson         if (emodeout[eout] == CEED_EVAL_GRAD)
536b7ec98d8SJeremy L Thompson           dout += 1;
537b7ec98d8SJeremy L Thompson         ierr = CeedBasisGetValue(basisout, emodeout[eout], q, n, dout, &bt);
538b7ec98d8SJeremy L Thompson         CeedChk(ierr);
539b7ec98d8SJeremy L Thompson         CeedInt din = -1;
540b7ec98d8SJeremy L Thompson         for (CeedInt ein=0; ein<numemodein; ein++) {
541b7ec98d8SJeremy L Thompson           CeedScalar b = 0.0;
542b7ec98d8SJeremy L Thompson           if (emodein[ein] == CEED_EVAL_GRAD)
543b7ec98d8SJeremy L Thompson             din += 1;
544b7ec98d8SJeremy L Thompson           ierr = CeedBasisGetValue(basisin, emodein[ein], q, n, din, &b);
545b7ec98d8SJeremy L Thompson           CeedChk(ierr);
546b7ec98d8SJeremy L Thompson           // Each element and component
547b7ec98d8SJeremy L Thompson           for (CeedInt e=0; e<nelem; e++)
548b7ec98d8SJeremy L Thompson             for (CeedInt cout=0; cout<ncomp; cout++) {
549b7ec98d8SJeremy L Thompson               CeedScalar db = 0.0;
550b7ec98d8SJeremy L Thompson               for (CeedInt cin=0; cin<ncomp; cin++)
551b7ec98d8SJeremy L Thompson                 db += assembledqfarray[((((e*numemodein+ein)*ncomp+cin)*
552b7ec98d8SJeremy L Thompson                                          numemodeout+eout)*ncomp+cout)*nqpts+q]*b;
553b7ec98d8SJeremy L Thompson               elemdiagarray[(e*ncomp+cout)*nnodes+n] += bt * db;
554b7ec98d8SJeremy L Thompson             }
555b7ec98d8SJeremy L Thompson         }
556b7ec98d8SJeremy L Thompson       }
557b7ec98d8SJeremy L Thompson     }
558b7ec98d8SJeremy L Thompson 
559b7ec98d8SJeremy L Thompson   ierr = CeedVectorRestoreArray(elemdiag, &elemdiagarray); CeedChk(ierr);
560b7ec98d8SJeremy L Thompson   ierr = CeedVectorRestoreArray(assembledqf, &assembledqfarray); CeedChk(ierr);
561b7ec98d8SJeremy L Thompson 
562b7ec98d8SJeremy L Thompson   // Assemble local operator diagonal
563b7ec98d8SJeremy L Thompson   ierr = CeedVectorSetValue(*assembled, 0.0); CeedChk(ierr);
564b7ec98d8SJeremy L Thompson   ierr = CeedElemRestrictionApply(rstrout, CEED_TRANSPOSE, lmodeout, elemdiag,
565b7ec98d8SJeremy L Thompson                                   *assembled, request); CeedChk(ierr);
566b7ec98d8SJeremy L Thompson 
567b7ec98d8SJeremy L Thompson   // Cleanup
568b7ec98d8SJeremy L Thompson   ierr = CeedVectorDestroy(&assembledqf); CeedChk(ierr);
569b7ec98d8SJeremy L Thompson   ierr = CeedVectorDestroy(&elemdiag); CeedChk(ierr);
570b7ec98d8SJeremy L Thompson   ierr = CeedFree(&emodein); CeedChk(ierr);
571b7ec98d8SJeremy L Thompson   ierr = CeedFree(&emodeout); CeedChk(ierr);
572b7ec98d8SJeremy L Thompson 
573b7ec98d8SJeremy L Thompson   return 0;
574b7ec98d8SJeremy L Thompson }
575b7ec98d8SJeremy L Thompson 
576b7ec98d8SJeremy L Thompson /**
577b11c1e72Sjeremylt   @brief Apply CeedOperator to a vector
578d7b241e6Sjeremylt 
579d7b241e6Sjeremylt   This computes the action of the operator on the specified (active) input,
580d7b241e6Sjeremylt   yielding its (active) output.  All inputs and outputs must be specified using
581d7b241e6Sjeremylt   CeedOperatorSetField().
582d7b241e6Sjeremylt 
583d7b241e6Sjeremylt   @param op        CeedOperator to apply
584b11c1e72Sjeremylt   @param[in] in    CeedVector containing input state or NULL if there are no
585b11c1e72Sjeremylt                      active inputs
586b11c1e72Sjeremylt   @param[out] out  CeedVector to store result of applying operator (must be
587d7b241e6Sjeremylt                      distinct from @a in) or NULL if there are no active outputs
588d7b241e6Sjeremylt   @param request   Address of CeedRequest for non-blocking completion, else
589d7b241e6Sjeremylt                      CEED_REQUEST_IMMEDIATE
590b11c1e72Sjeremylt 
591b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
592dfdf5a53Sjeremylt 
593dfdf5a53Sjeremylt   @ref Basic
594b11c1e72Sjeremylt **/
595692c2638Sjeremylt int CeedOperatorApply(CeedOperator op, CeedVector in, CeedVector out,
596692c2638Sjeremylt                       CeedRequest *request) {
597d7b241e6Sjeremylt   int ierr;
598d7b241e6Sjeremylt   Ceed ceed = op->ceed;
599d7b241e6Sjeremylt   CeedQFunction qf = op->qf;
600d7b241e6Sjeremylt 
60152d6035fSJeremy L Thompson   if (op->composite) {
602c042f62fSJeremy L Thompson     if (!op->numsub)
603c042f62fSJeremy L Thompson       // LCOV_EXCL_START
604c042f62fSJeremy L Thompson       return CeedError(ceed, 1, "No suboperators set");
605c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
60652d6035fSJeremy L Thompson   } else {
607c042f62fSJeremy L Thompson     if (op->nfields == 0)
608c042f62fSJeremy L Thompson       // LCOV_EXCL_START
609c042f62fSJeremy L Thompson       return CeedError(ceed, 1, "No operator fields set");
610c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
611c042f62fSJeremy L Thompson     if (op->nfields < qf->numinputfields + qf->numoutputfields)
612c042f62fSJeremy L Thompson       // LCOV_EXCL_START
613c042f62fSJeremy L Thompson       return CeedError(ceed, 1, "Not all operator fields set");
614c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
6152cb0afc5Sjeremylt     if (!op->hasrestriction)
616c042f62fSJeremy L Thompson       // LCOV_EXCL_START
617c042f62fSJeremy L Thompson       return CeedError(ceed, 1,"At least one restriction required");
618c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
619c042f62fSJeremy L Thompson     if (op->numqpoints == 0)
620c042f62fSJeremy L Thompson       // LCOV_EXCL_START
621c042f62fSJeremy L Thompson       return CeedError(ceed, 1,"At least one non-collocated basis required");
622c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
62352d6035fSJeremy L Thompson   }
6241d7d2407SJeremy L Thompson   if (op->numelements || op->composite) {
625e97ff134Sjeremylt     ierr = op->Apply(op, in != CEED_VECTOR_NONE ? in : NULL,
626e97ff134Sjeremylt                      out != CEED_VECTOR_NONE ? out : NULL, request);
627e97ff134Sjeremylt     CeedChk(ierr);
6281d7d2407SJeremy L Thompson   }
629d7b241e6Sjeremylt   return 0;
630d7b241e6Sjeremylt }
631d7b241e6Sjeremylt 
632d7b241e6Sjeremylt /**
6334ce2993fSjeremylt   @brief Get the Ceed associated with a CeedOperator
6344ce2993fSjeremylt 
6354ce2993fSjeremylt   @param op              CeedOperator
6364ce2993fSjeremylt   @param[out] ceed       Variable to store Ceed
6374ce2993fSjeremylt 
6384ce2993fSjeremylt   @return An error code: 0 - success, otherwise - failure
6394ce2993fSjeremylt 
64023617272Sjeremylt   @ref Advanced
6414ce2993fSjeremylt **/
6424ce2993fSjeremylt 
6434ce2993fSjeremylt int CeedOperatorGetCeed(CeedOperator op, Ceed *ceed) {
6444ce2993fSjeremylt   *ceed = op->ceed;
6454ce2993fSjeremylt   return 0;
6464ce2993fSjeremylt }
6474ce2993fSjeremylt 
6484ce2993fSjeremylt /**
6494ce2993fSjeremylt   @brief Get the number of elements associated with a CeedOperator
6504ce2993fSjeremylt 
6514ce2993fSjeremylt   @param op              CeedOperator
6524ce2993fSjeremylt   @param[out] numelem    Variable to store number of elements
6534ce2993fSjeremylt 
6544ce2993fSjeremylt   @return An error code: 0 - success, otherwise - failure
6554ce2993fSjeremylt 
65623617272Sjeremylt   @ref Advanced
6574ce2993fSjeremylt **/
6584ce2993fSjeremylt 
6594ce2993fSjeremylt int CeedOperatorGetNumElements(CeedOperator op, CeedInt *numelem) {
66052d6035fSJeremy L Thompson   if (op->composite)
661c042f62fSJeremy L Thompson     // LCOV_EXCL_START
66252d6035fSJeremy L Thompson     return CeedError(op->ceed, 1, "Not defined for composite operator");
663c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
66452d6035fSJeremy L Thompson 
6654ce2993fSjeremylt   *numelem = op->numelements;
6664ce2993fSjeremylt   return 0;
6674ce2993fSjeremylt }
6684ce2993fSjeremylt 
6694ce2993fSjeremylt /**
6704ce2993fSjeremylt   @brief Get the number of quadrature points associated with a CeedOperator
6714ce2993fSjeremylt 
6724ce2993fSjeremylt   @param op              CeedOperator
6734ce2993fSjeremylt   @param[out] numqpts    Variable to store vector number of quadrature points
6744ce2993fSjeremylt 
6754ce2993fSjeremylt   @return An error code: 0 - success, otherwise - failure
6764ce2993fSjeremylt 
67723617272Sjeremylt   @ref Advanced
6784ce2993fSjeremylt **/
6794ce2993fSjeremylt 
6804ce2993fSjeremylt int CeedOperatorGetNumQuadraturePoints(CeedOperator op, CeedInt *numqpts) {
68152d6035fSJeremy L Thompson   if (op->composite)
682c042f62fSJeremy L Thompson     // LCOV_EXCL_START
68352d6035fSJeremy L Thompson     return CeedError(op->ceed, 1, "Not defined for composite operator");
684c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
68552d6035fSJeremy L Thompson 
6864ce2993fSjeremylt   *numqpts = op->numqpoints;
6874ce2993fSjeremylt   return 0;
6884ce2993fSjeremylt }
6894ce2993fSjeremylt 
6904ce2993fSjeremylt /**
6914ce2993fSjeremylt   @brief Get the number of arguments associated with a CeedOperator
6924ce2993fSjeremylt 
6934ce2993fSjeremylt   @param op              CeedOperator
6944ce2993fSjeremylt   @param[out] numargs    Variable to store vector number of arguments
6954ce2993fSjeremylt 
6964ce2993fSjeremylt   @return An error code: 0 - success, otherwise - failure
6974ce2993fSjeremylt 
69823617272Sjeremylt   @ref Advanced
6994ce2993fSjeremylt **/
7004ce2993fSjeremylt 
7014ce2993fSjeremylt int CeedOperatorGetNumArgs(CeedOperator op, CeedInt *numargs) {
70252d6035fSJeremy L Thompson   if (op->composite)
703c042f62fSJeremy L Thompson     // LCOV_EXCL_START
70452d6035fSJeremy L Thompson     return CeedError(op->ceed, 1, "Not defined for composite operators");
705c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
706c042f62fSJeremy L Thompson 
7074ce2993fSjeremylt   *numargs = op->nfields;
7084ce2993fSjeremylt   return 0;
7094ce2993fSjeremylt }
7104ce2993fSjeremylt 
7114ce2993fSjeremylt /**
7124ce2993fSjeremylt   @brief Get the setup status of a CeedOperator
7134ce2993fSjeremylt 
7144ce2993fSjeremylt   @param op             CeedOperator
715288c0443SJeremy L Thompson   @param[out] setupdone Variable to store setup status
7164ce2993fSjeremylt 
7174ce2993fSjeremylt   @return An error code: 0 - success, otherwise - failure
7184ce2993fSjeremylt 
71923617272Sjeremylt   @ref Advanced
7204ce2993fSjeremylt **/
7214ce2993fSjeremylt 
7224ce2993fSjeremylt int CeedOperatorGetSetupStatus(CeedOperator op, bool *setupdone) {
7234ce2993fSjeremylt   *setupdone = op->setupdone;
7244ce2993fSjeremylt   return 0;
7254ce2993fSjeremylt }
7264ce2993fSjeremylt 
7274ce2993fSjeremylt /**
7284ce2993fSjeremylt   @brief Get the QFunction associated with a CeedOperator
7294ce2993fSjeremylt 
7304ce2993fSjeremylt   @param op              CeedOperator
7318c91a0c9SJeremy L Thompson   @param[out] qf         Variable to store QFunction
7324ce2993fSjeremylt 
7334ce2993fSjeremylt   @return An error code: 0 - success, otherwise - failure
7344ce2993fSjeremylt 
73523617272Sjeremylt   @ref Advanced
7364ce2993fSjeremylt **/
7374ce2993fSjeremylt 
7384ce2993fSjeremylt int CeedOperatorGetQFunction(CeedOperator op, CeedQFunction *qf) {
73952d6035fSJeremy L Thompson   if (op->composite)
740c042f62fSJeremy L Thompson     // LCOV_EXCL_START
74152d6035fSJeremy L Thompson     return CeedError(op->ceed, 1, "Not defined for composite operator");
742c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
74352d6035fSJeremy L Thompson 
7444ce2993fSjeremylt   *qf = op->qf;
7454ce2993fSjeremylt   return 0;
7464ce2993fSjeremylt }
7474ce2993fSjeremylt 
7484ce2993fSjeremylt /**
74952d6035fSJeremy L Thompson   @brief Get the number of suboperators associated with a CeedOperator
75052d6035fSJeremy L Thompson 
75152d6035fSJeremy L Thompson   @param op              CeedOperator
75252d6035fSJeremy L Thompson   @param[out] numsub     Variable to store number of suboperators
75352d6035fSJeremy L Thompson 
75452d6035fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
75552d6035fSJeremy L Thompson 
75652d6035fSJeremy L Thompson   @ref Advanced
75752d6035fSJeremy L Thompson **/
75852d6035fSJeremy L Thompson 
75952d6035fSJeremy L Thompson int CeedOperatorGetNumSub(CeedOperator op, CeedInt *numsub) {
760c042f62fSJeremy L Thompson   if (!op->composite)
761c042f62fSJeremy L Thompson     // LCOV_EXCL_START
762c042f62fSJeremy L Thompson     return CeedError(op->ceed, 1, "Not a composite operator");
763c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
76452d6035fSJeremy L Thompson 
76552d6035fSJeremy L Thompson   *numsub = op->numsub;
76652d6035fSJeremy L Thompson   return 0;
76752d6035fSJeremy L Thompson }
76852d6035fSJeremy L Thompson 
76952d6035fSJeremy L Thompson /**
77052d6035fSJeremy L Thompson   @brief Get the list of suboperators associated with a CeedOperator
77152d6035fSJeremy L Thompson 
77252d6035fSJeremy L Thompson   @param op                CeedOperator
77352d6035fSJeremy L Thompson   @param[out] suboperators Variable to store list of suboperators
77452d6035fSJeremy L Thompson 
77552d6035fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
77652d6035fSJeremy L Thompson 
77752d6035fSJeremy L Thompson   @ref Advanced
77852d6035fSJeremy L Thompson **/
77952d6035fSJeremy L Thompson 
78052d6035fSJeremy L Thompson int CeedOperatorGetSubList(CeedOperator op, CeedOperator **suboperators) {
781c042f62fSJeremy L Thompson   if (!op->composite)
782c042f62fSJeremy L Thompson     // LCOV_EXCL_START
783c042f62fSJeremy L Thompson     return CeedError(op->ceed, 1, "Not a composite operator");
784c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
78552d6035fSJeremy L Thompson 
78652d6035fSJeremy L Thompson   *suboperators = op->suboperators;
78752d6035fSJeremy L Thompson   return 0;
78852d6035fSJeremy L Thompson }
78952d6035fSJeremy L Thompson 
79052d6035fSJeremy L Thompson /**
791fe2413ffSjeremylt   @brief Set the backend data of a CeedOperator
792fe2413ffSjeremylt 
793fe2413ffSjeremylt   @param[out] op         CeedOperator
794fe2413ffSjeremylt   @param data            Data to set
795fe2413ffSjeremylt 
796fe2413ffSjeremylt   @return An error code: 0 - success, otherwise - failure
797fe2413ffSjeremylt 
798fe2413ffSjeremylt   @ref Advanced
799fe2413ffSjeremylt **/
800fe2413ffSjeremylt 
801fe2413ffSjeremylt int CeedOperatorSetData(CeedOperator op, void **data) {
802fe2413ffSjeremylt   op->data = *data;
803fe2413ffSjeremylt   return 0;
804fe2413ffSjeremylt }
805fe2413ffSjeremylt 
806fe2413ffSjeremylt /**
8074ce2993fSjeremylt   @brief Get the backend data of a CeedOperator
8084ce2993fSjeremylt 
8094ce2993fSjeremylt   @param op              CeedOperator
8104ce2993fSjeremylt   @param[out] data       Variable to store data
8114ce2993fSjeremylt 
8124ce2993fSjeremylt   @return An error code: 0 - success, otherwise - failure
8134ce2993fSjeremylt 
81423617272Sjeremylt   @ref Advanced
8154ce2993fSjeremylt **/
8164ce2993fSjeremylt 
8174ce2993fSjeremylt int CeedOperatorGetData(CeedOperator op, void **data) {
8184ce2993fSjeremylt   *data = op->data;
8194ce2993fSjeremylt   return 0;
8204ce2993fSjeremylt }
8214ce2993fSjeremylt 
8224ce2993fSjeremylt /**
8234ce2993fSjeremylt   @brief Set the setup flag of a CeedOperator to True
8244ce2993fSjeremylt 
8254ce2993fSjeremylt   @param op              CeedOperator
8264ce2993fSjeremylt 
8274ce2993fSjeremylt   @return An error code: 0 - success, otherwise - failure
8284ce2993fSjeremylt 
82923617272Sjeremylt   @ref Advanced
8304ce2993fSjeremylt **/
8314ce2993fSjeremylt 
8324ce2993fSjeremylt int CeedOperatorSetSetupDone(CeedOperator op) {
8334ce2993fSjeremylt   op->setupdone = 1;
8344ce2993fSjeremylt   return 0;
8354ce2993fSjeremylt }
8364ce2993fSjeremylt 
8374ce2993fSjeremylt /**
8382ebaca42Sjeremylt   @brief View a field of a CeedOperator
8392ebaca42Sjeremylt 
8402ebaca42Sjeremylt   @param[in] field       Operator field to view
8412ebaca42Sjeremylt   @param[in] fieldnumber Number of field being viewed
8422ebaca42Sjeremylt   @param[in] stream      Stream to view to, e.g., stdout
8432ebaca42Sjeremylt 
8442ebaca42Sjeremylt   @return An error code: 0 - success, otherwise - failure
8452ebaca42Sjeremylt 
8462ebaca42Sjeremylt   @ref Utility
8472ebaca42Sjeremylt **/
8482ebaca42Sjeremylt static int CeedOperatorFieldView(CeedOperatorField field,
8492ebaca42Sjeremylt                                  CeedQFunctionField qffield,
8502da88da5Sjeremylt                                  CeedInt fieldnumber, bool sub, bool in,
8512da88da5Sjeremylt                                  FILE *stream) {
8522ebaca42Sjeremylt   const char *pre = sub ? "  " : "";
8532da88da5Sjeremylt   const char *inout = in ? "Input" : "Output";
8542ebaca42Sjeremylt 
8552da88da5Sjeremylt   fprintf(stream, "%s    %s Field [%d]:\n"
8562da88da5Sjeremylt           "%s      Name: \"%s\"\n"
8572da88da5Sjeremylt           "%s      Lmode: \"%s\"\n",
8582da88da5Sjeremylt           pre, inout, fieldnumber, pre, qffield->fieldname,
8592ebaca42Sjeremylt           pre, CeedTransposeModes[field->lmode]);
8602ebaca42Sjeremylt 
8612ebaca42Sjeremylt   if (field->basis == CEED_BASIS_COLLOCATED)
8622ebaca42Sjeremylt     fprintf(stream, "%s      Collocated basis\n", pre);
8632ebaca42Sjeremylt 
8642ebaca42Sjeremylt   if (field->vec == CEED_VECTOR_ACTIVE)
8652ebaca42Sjeremylt     fprintf(stream, "%s      Active vector\n", pre);
8662ebaca42Sjeremylt   else if (field->vec == CEED_VECTOR_NONE)
8672ebaca42Sjeremylt     fprintf(stream, "%s      No vector\n", pre);
8682ebaca42Sjeremylt 
8692ebaca42Sjeremylt   return 0;
8702ebaca42Sjeremylt }
8712ebaca42Sjeremylt 
8722ebaca42Sjeremylt /**
8732ebaca42Sjeremylt   @brief View a single CeedOperator
8742ebaca42Sjeremylt 
8752ebaca42Sjeremylt   @param[in] op     CeedOperator to view
8762ebaca42Sjeremylt   @param[in] stream Stream to write; typically stdout/stderr or a file
8772ebaca42Sjeremylt 
8782ebaca42Sjeremylt   @return Error code: 0 - success, otherwise - failure
8792ebaca42Sjeremylt 
8802ebaca42Sjeremylt   @ref Utility
8812ebaca42Sjeremylt **/
8822ebaca42Sjeremylt int CeedOperatorSingleView(CeedOperator op, bool sub, FILE *stream) {
8832ebaca42Sjeremylt   int ierr;
8842ebaca42Sjeremylt   const char *pre = sub ? "  " : "";
8852ebaca42Sjeremylt 
8862ebaca42Sjeremylt   CeedInt totalfields;
8872ebaca42Sjeremylt   ierr = CeedOperatorGetNumArgs(op, &totalfields); CeedChk(ierr);
8882da88da5Sjeremylt 
8892ebaca42Sjeremylt   fprintf(stream, "%s  %d Field%s\n", pre, totalfields, totalfields>1 ? "s" : "");
8902ebaca42Sjeremylt 
8912da88da5Sjeremylt   fprintf(stream, "%s  %d Input Field%s:\n", pre, op->qf->numinputfields,
8922ebaca42Sjeremylt           op->qf->numinputfields>1 ? "s" : "");
8932ebaca42Sjeremylt   for (CeedInt i=0; i<op->qf->numinputfields; i++) {
8942ebaca42Sjeremylt     ierr = CeedOperatorFieldView(op->inputfields[i], op->qf->inputfields[i],
8952da88da5Sjeremylt                                  i, sub, 1, stream); CeedChk(ierr);
8962ebaca42Sjeremylt   }
8972ebaca42Sjeremylt 
8982da88da5Sjeremylt   fprintf(stream, "%s  %d Output Field%s:\n", pre, op->qf->numoutputfields,
8992ebaca42Sjeremylt           op->qf->numoutputfields>1 ? "s" : "");
9002ebaca42Sjeremylt   for (CeedInt i=0; i<op->qf->numoutputfields; i++) {
9012ebaca42Sjeremylt     ierr = CeedOperatorFieldView(op->outputfields[i], op->qf->inputfields[i],
9022da88da5Sjeremylt                                  i, sub, 0, stream); CeedChk(ierr);
9032ebaca42Sjeremylt   }
9042ebaca42Sjeremylt 
9052ebaca42Sjeremylt   return 0;
9062ebaca42Sjeremylt }
9072ebaca42Sjeremylt 
9082ebaca42Sjeremylt /**
9092ebaca42Sjeremylt   @brief View a CeedOperator
9102ebaca42Sjeremylt 
9112ebaca42Sjeremylt   @param[in] op     CeedOperator to view
9122ebaca42Sjeremylt   @param[in] stream Stream to write; typically stdout/stderr or a file
9132ebaca42Sjeremylt 
9142ebaca42Sjeremylt   @return Error code: 0 - success, otherwise - failure
9152ebaca42Sjeremylt 
9162ebaca42Sjeremylt   @ref Utility
9172ebaca42Sjeremylt **/
9182ebaca42Sjeremylt int CeedOperatorView(CeedOperator op, FILE *stream) {
9192ebaca42Sjeremylt   int ierr;
9202ebaca42Sjeremylt 
9212ebaca42Sjeremylt   if (op->composite) {
9222ebaca42Sjeremylt     fprintf(stream, "Composite CeedOperator\n");
9232ebaca42Sjeremylt 
9242ebaca42Sjeremylt     for (CeedInt i=0; i<op->numsub; i++) {
9252da88da5Sjeremylt       fprintf(stream, "  SubOperator [%d]:\n", i);
9262ebaca42Sjeremylt       ierr = CeedOperatorSingleView(op->suboperators[i], 1, stream);
9272ebaca42Sjeremylt       CeedChk(ierr);
9282ebaca42Sjeremylt     }
9292ebaca42Sjeremylt   } else {
9302ebaca42Sjeremylt     fprintf(stream, "CeedOperator\n");
9312ebaca42Sjeremylt     ierr = CeedOperatorSingleView(op, 0, stream); CeedChk(ierr);
9322ebaca42Sjeremylt   }
9332ebaca42Sjeremylt 
9342ebaca42Sjeremylt   return 0;
9352ebaca42Sjeremylt }
9362ebaca42Sjeremylt 
9372ebaca42Sjeremylt /**
938d1bcdac9Sjeremylt   @brief Get the CeedOperatorFields of a CeedOperator
939d1bcdac9Sjeremylt 
940d1bcdac9Sjeremylt   @param op                 CeedOperator
941d1bcdac9Sjeremylt   @param[out] inputfields   Variable to store inputfields
942d1bcdac9Sjeremylt   @param[out] outputfields  Variable to store outputfields
943d1bcdac9Sjeremylt 
944d1bcdac9Sjeremylt   @return An error code: 0 - success, otherwise - failure
945d1bcdac9Sjeremylt 
946d1bcdac9Sjeremylt   @ref Advanced
947d1bcdac9Sjeremylt **/
948d1bcdac9Sjeremylt 
949692c2638Sjeremylt int CeedOperatorGetFields(CeedOperator op, CeedOperatorField **inputfields,
950d1bcdac9Sjeremylt                           CeedOperatorField **outputfields) {
95152d6035fSJeremy L Thompson   if (op->composite)
952c042f62fSJeremy L Thompson     // LCOV_EXCL_START
95352d6035fSJeremy L Thompson     return CeedError(op->ceed, 1, "Not defined for composite operator");
954c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
95552d6035fSJeremy L Thompson 
956d1bcdac9Sjeremylt   if (inputfields) *inputfields = op->inputfields;
957d1bcdac9Sjeremylt   if (outputfields) *outputfields = op->outputfields;
958d1bcdac9Sjeremylt   return 0;
959d1bcdac9Sjeremylt }
960d1bcdac9Sjeremylt 
961d1bcdac9Sjeremylt /**
9624dccadb6Sjeremylt   @brief Get the L vector CeedTransposeMode of a CeedOperatorField
9634dccadb6Sjeremylt 
9644dccadb6Sjeremylt   @param opfield         CeedOperatorField
9654dccadb6Sjeremylt   @param[out] lmode      Variable to store CeedTransposeMode
9664dccadb6Sjeremylt 
9674dccadb6Sjeremylt   @return An error code: 0 - success, otherwise - failure
9684dccadb6Sjeremylt 
9694dccadb6Sjeremylt   @ref Advanced
9704dccadb6Sjeremylt **/
9714dccadb6Sjeremylt 
9724dccadb6Sjeremylt int CeedOperatorFieldGetLMode(CeedOperatorField opfield,
9734dccadb6Sjeremylt                               CeedTransposeMode *lmode) {
974fe2413ffSjeremylt   *lmode = opfield->lmode;
9754dccadb6Sjeremylt   return 0;
9762b8e417aSjeremylt }
9772b8e417aSjeremylt 
9782b8e417aSjeremylt /**
979d1bcdac9Sjeremylt   @brief Get the CeedElemRestriction of a CeedOperatorField
980d1bcdac9Sjeremylt 
981d1bcdac9Sjeremylt   @param opfield         CeedOperatorField
982d1bcdac9Sjeremylt   @param[out] rstr       Variable to store CeedElemRestriction
983d1bcdac9Sjeremylt 
984d1bcdac9Sjeremylt   @return An error code: 0 - success, otherwise - failure
985d1bcdac9Sjeremylt 
986d1bcdac9Sjeremylt   @ref Advanced
987d1bcdac9Sjeremylt **/
988d1bcdac9Sjeremylt 
989d1bcdac9Sjeremylt int CeedOperatorFieldGetElemRestriction(CeedOperatorField opfield,
990d1bcdac9Sjeremylt                                         CeedElemRestriction *rstr) {
991fe2413ffSjeremylt   *rstr = opfield->Erestrict;
992d1bcdac9Sjeremylt   return 0;
993d1bcdac9Sjeremylt }
994d1bcdac9Sjeremylt 
995d1bcdac9Sjeremylt /**
996d1bcdac9Sjeremylt   @brief Get the CeedBasis of a CeedOperatorField
997d1bcdac9Sjeremylt 
998d1bcdac9Sjeremylt   @param opfield         CeedOperatorField
999d1bcdac9Sjeremylt   @param[out] basis      Variable to store CeedBasis
1000d1bcdac9Sjeremylt 
1001d1bcdac9Sjeremylt   @return An error code: 0 - success, otherwise - failure
1002d1bcdac9Sjeremylt 
1003d1bcdac9Sjeremylt   @ref Advanced
1004d1bcdac9Sjeremylt **/
1005d1bcdac9Sjeremylt 
1006692c2638Sjeremylt int CeedOperatorFieldGetBasis(CeedOperatorField opfield, CeedBasis *basis) {
1007fe2413ffSjeremylt   *basis = opfield->basis;
1008d1bcdac9Sjeremylt   return 0;
1009d1bcdac9Sjeremylt }
1010d1bcdac9Sjeremylt 
1011d1bcdac9Sjeremylt /**
1012d1bcdac9Sjeremylt   @brief Get the CeedVector of a CeedOperatorField
1013d1bcdac9Sjeremylt 
1014d1bcdac9Sjeremylt   @param opfield         CeedOperatorField
1015d1bcdac9Sjeremylt   @param[out] vec        Variable to store CeedVector
1016d1bcdac9Sjeremylt 
1017d1bcdac9Sjeremylt   @return An error code: 0 - success, otherwise - failure
1018d1bcdac9Sjeremylt 
1019d1bcdac9Sjeremylt   @ref Advanced
1020d1bcdac9Sjeremylt **/
1021d1bcdac9Sjeremylt 
1022692c2638Sjeremylt int CeedOperatorFieldGetVector(CeedOperatorField opfield, CeedVector *vec) {
1023fe2413ffSjeremylt   *vec = opfield->vec;
1024d1bcdac9Sjeremylt   return 0;
1025d1bcdac9Sjeremylt }
1026d1bcdac9Sjeremylt 
1027d1bcdac9Sjeremylt /**
1028b11c1e72Sjeremylt   @brief Destroy a CeedOperator
1029d7b241e6Sjeremylt 
1030d7b241e6Sjeremylt   @param op CeedOperator to destroy
1031b11c1e72Sjeremylt 
1032b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
1033dfdf5a53Sjeremylt 
1034dfdf5a53Sjeremylt   @ref Basic
1035b11c1e72Sjeremylt **/
1036d7b241e6Sjeremylt int CeedOperatorDestroy(CeedOperator *op) {
1037d7b241e6Sjeremylt   int ierr;
1038d7b241e6Sjeremylt 
1039d7b241e6Sjeremylt   if (!*op || --(*op)->refcount > 0) return 0;
1040d7b241e6Sjeremylt   if ((*op)->Destroy) {
1041d7b241e6Sjeremylt     ierr = (*op)->Destroy(*op); CeedChk(ierr);
1042d7b241e6Sjeremylt   }
1043fe2413ffSjeremylt   ierr = CeedDestroy(&(*op)->ceed); CeedChk(ierr);
1044fe2413ffSjeremylt   // Free fields
10451d102b48SJeremy L Thompson   for (int i=0; i<(*op)->nfields; i++)
104652d6035fSJeremy L Thompson     if ((*op)->inputfields[i]) {
104771352a93Sjeremylt       ierr = CeedElemRestrictionDestroy(&(*op)->inputfields[i]->Erestrict);
104871352a93Sjeremylt       CeedChk(ierr);
104971352a93Sjeremylt       if ((*op)->inputfields[i]->basis != CEED_BASIS_COLLOCATED) {
105071352a93Sjeremylt         ierr = CeedBasisDestroy(&(*op)->inputfields[i]->basis); CeedChk(ierr);
105171352a93Sjeremylt       }
105271352a93Sjeremylt       if ((*op)->inputfields[i]->vec != CEED_VECTOR_ACTIVE &&
105371352a93Sjeremylt           (*op)->inputfields[i]->vec != CEED_VECTOR_NONE ) {
105471352a93Sjeremylt         ierr = CeedVectorDestroy(&(*op)->inputfields[i]->vec); CeedChk(ierr);
105571352a93Sjeremylt       }
1056fe2413ffSjeremylt       ierr = CeedFree(&(*op)->inputfields[i]); CeedChk(ierr);
1057fe2413ffSjeremylt     }
10581d102b48SJeremy L Thompson   for (int i=0; i<(*op)->nfields; i++)
1059d0eb30a5Sjeremylt     if ((*op)->outputfields[i]) {
106071352a93Sjeremylt       ierr = CeedElemRestrictionDestroy(&(*op)->outputfields[i]->Erestrict);
106171352a93Sjeremylt       CeedChk(ierr);
106271352a93Sjeremylt       if ((*op)->outputfields[i]->basis != CEED_BASIS_COLLOCATED) {
106371352a93Sjeremylt         ierr = CeedBasisDestroy(&(*op)->outputfields[i]->basis); CeedChk(ierr);
106471352a93Sjeremylt       }
106571352a93Sjeremylt       if ((*op)->outputfields[i]->vec != CEED_VECTOR_ACTIVE &&
106671352a93Sjeremylt           (*op)->outputfields[i]->vec != CEED_VECTOR_NONE ) {
106771352a93Sjeremylt         ierr = CeedVectorDestroy(&(*op)->outputfields[i]->vec); CeedChk(ierr);
106871352a93Sjeremylt       }
1069fe2413ffSjeremylt       ierr = CeedFree(&(*op)->outputfields[i]); CeedChk(ierr);
1070fe2413ffSjeremylt     }
107152d6035fSJeremy L Thompson   // Destroy suboperators
10721d102b48SJeremy L Thompson   for (int i=0; i<(*op)->numsub; i++)
107352d6035fSJeremy L Thompson     if ((*op)->suboperators[i]) {
107452d6035fSJeremy L Thompson       ierr = CeedOperatorDestroy(&(*op)->suboperators[i]); CeedChk(ierr);
107552d6035fSJeremy L Thompson     }
1076d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->qf); CeedChk(ierr);
1077d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->dqf); CeedChk(ierr);
1078d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->dqfT); CeedChk(ierr);
1079fe2413ffSjeremylt 
1080*5107b09fSJeremy L Thompson   // Destroy fallback
1081*5107b09fSJeremy L Thompson   if ((*op)->opfallback) {
1082*5107b09fSJeremy L Thompson     ierr = (*op)->qffallback->Destroy((*op)->qffallback); CeedChk(ierr);
1083*5107b09fSJeremy L Thompson     ierr = CeedFree(&(*op)->qffallback); CeedChk(ierr);
1084*5107b09fSJeremy L Thompson     ierr = (*op)->opfallback->Destroy((*op)->opfallback); CeedChk(ierr);
1085*5107b09fSJeremy L Thompson     ierr = CeedFree(&(*op)->opfallback); CeedChk(ierr);
1086*5107b09fSJeremy L Thompson   }
1087*5107b09fSJeremy L Thompson 
1088fe2413ffSjeremylt   ierr = CeedFree(&(*op)->inputfields); CeedChk(ierr);
1089fe2413ffSjeremylt   ierr = CeedFree(&(*op)->outputfields); CeedChk(ierr);
109052d6035fSJeremy L Thompson   ierr = CeedFree(&(*op)->suboperators); CeedChk(ierr);
1091d7b241e6Sjeremylt   ierr = CeedFree(op); CeedChk(ierr);
1092d7b241e6Sjeremylt   return 0;
1093d7b241e6Sjeremylt }
1094d7b241e6Sjeremylt 
1095d7b241e6Sjeremylt /// @}
1096