xref: /libCEED/rust/libceed-sys/c-src/interface/ceed-operator.c (revision 480fae850cb0898970f2dd40c5ea1abc8a8b0488)
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 
17ec3da8bcSJed Brown #include <ceed/ceed.h>
18ec3da8bcSJed Brown #include <ceed/backend.h>
193d576824SJeremy L Thompson #include <ceed-impl.h>
203d576824SJeremy L Thompson #include <stdbool.h>
213d576824SJeremy L Thompson #include <stdio.h>
223d576824SJeremy L Thompson #include <string.h>
23d7b241e6Sjeremylt 
24dfdf5a53Sjeremylt /// @file
257a982d89SJeremy L. Thompson /// Implementation of CeedOperator interfaces
267a982d89SJeremy L. Thompson 
277a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
287a982d89SJeremy L. Thompson /// CeedOperator Library Internal Functions
297a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
307a982d89SJeremy L. Thompson /// @addtogroup CeedOperatorDeveloper
317a982d89SJeremy L. Thompson /// @{
327a982d89SJeremy L. Thompson 
337a982d89SJeremy L. Thompson /**
34e15f9bd0SJeremy L Thompson   @brief Check if a CeedOperator Field matches the QFunction Field
35e15f9bd0SJeremy L Thompson 
36e15f9bd0SJeremy L Thompson   @param[in] ceed      Ceed object for error handling
37d1d35e2fSjeremylt   @param[in] qf_field  QFunction Field matching Operator Field
38e15f9bd0SJeremy L Thompson   @param[in] r         Operator Field ElemRestriction
39e15f9bd0SJeremy L Thompson   @param[in] b         Operator Field Basis
40e15f9bd0SJeremy L Thompson 
41e15f9bd0SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
42e15f9bd0SJeremy L Thompson 
43e15f9bd0SJeremy L Thompson   @ref Developer
44e15f9bd0SJeremy L Thompson **/
45d1d35e2fSjeremylt static int CeedOperatorCheckField(Ceed ceed, CeedQFunctionField qf_field,
46e15f9bd0SJeremy L Thompson                                   CeedElemRestriction r, CeedBasis b) {
47e15f9bd0SJeremy L Thompson   int ierr;
48d1d35e2fSjeremylt   CeedEvalMode eval_mode = qf_field->eval_mode;
49d1d35e2fSjeremylt   CeedInt dim = 1, num_comp = 1, restr_num_comp = 1, size = qf_field->size;
50e15f9bd0SJeremy L Thompson   // Restriction
51e15f9bd0SJeremy L Thompson   if (r != CEED_ELEMRESTRICTION_NONE) {
52d1d35e2fSjeremylt     if (eval_mode == CEED_EVAL_WEIGHT) {
53e15f9bd0SJeremy L Thompson       // LCOV_EXCL_START
54e1e9e29dSJed Brown       return CeedError(ceed, CEED_ERROR_INCOMPATIBLE,
55e1e9e29dSJed Brown                        "CEED_ELEMRESTRICTION_NONE should be used "
56e15f9bd0SJeremy L Thompson                        "for a field with eval mode CEED_EVAL_WEIGHT");
57e15f9bd0SJeremy L Thompson       // LCOV_EXCL_STOP
58e15f9bd0SJeremy L Thompson     }
59d1d35e2fSjeremylt     ierr = CeedElemRestrictionGetNumComponents(r, &restr_num_comp);
6078464608Sjeremylt     CeedChk(ierr);
61e1e9e29dSJed Brown   }
62d1d35e2fSjeremylt   if ((r == CEED_ELEMRESTRICTION_NONE) != (eval_mode == CEED_EVAL_WEIGHT)) {
63e1e9e29dSJed Brown     // LCOV_EXCL_START
64e1e9e29dSJed Brown     return CeedError(ceed, CEED_ERROR_INCOMPATIBLE,
65e1e9e29dSJed Brown                      "CEED_ELEMRESTRICTION_NONE and CEED_EVAL_WEIGHT "
66e1e9e29dSJed Brown                      "must be used together.");
67e1e9e29dSJed Brown     // LCOV_EXCL_STOP
68e1e9e29dSJed Brown   }
69e15f9bd0SJeremy L Thompson   // Basis
70e15f9bd0SJeremy L Thompson   if (b != CEED_BASIS_COLLOCATED) {
71d1d35e2fSjeremylt     if (eval_mode == CEED_EVAL_NONE)
728229195eSjeremylt       // LCOV_EXCL_START
73e1e9e29dSJed Brown       return CeedError(ceed, CEED_ERROR_INCOMPATIBLE,
74d1d35e2fSjeremylt                        "Field '%s' configured with CEED_EVAL_NONE must "
75d1d35e2fSjeremylt                        "be used with CEED_BASIS_COLLOCATED",
768229195eSjeremylt                        // LCOV_EXCL_STOP
77d1d35e2fSjeremylt                        qf_field->field_name);
78e15f9bd0SJeremy L Thompson     ierr = CeedBasisGetDimension(b, &dim); CeedChk(ierr);
79d1d35e2fSjeremylt     ierr = CeedBasisGetNumComponents(b, &num_comp); CeedChk(ierr);
80d1d35e2fSjeremylt     if (r != CEED_ELEMRESTRICTION_NONE && restr_num_comp != num_comp) {
81e15f9bd0SJeremy L Thompson       // LCOV_EXCL_START
82e15f9bd0SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_DIMENSION,
83d1d35e2fSjeremylt                        "Field '%s' of size %d and EvalMode %s: ElemRestriction "
84d1d35e2fSjeremylt                        "has %d components, but Basis has %d components",
85d1d35e2fSjeremylt                        qf_field->field_name, qf_field->size, CeedEvalModes[qf_field->eval_mode],
86d1d35e2fSjeremylt                        restr_num_comp,
87d1d35e2fSjeremylt                        num_comp);
88e15f9bd0SJeremy L Thompson       // LCOV_EXCL_STOP
89e15f9bd0SJeremy L Thompson     }
90e15f9bd0SJeremy L Thompson   }
91e15f9bd0SJeremy L Thompson   // Field size
92d1d35e2fSjeremylt   switch(eval_mode) {
93e15f9bd0SJeremy L Thompson   case CEED_EVAL_NONE:
94d1d35e2fSjeremylt     if (size != restr_num_comp)
95e15f9bd0SJeremy L Thompson       // LCOV_EXCL_START
96e15f9bd0SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_DIMENSION,
97e1e9e29dSJed Brown                        "Field '%s' of size %d and EvalMode %s: ElemRestriction has %d components",
98d1d35e2fSjeremylt                        qf_field->field_name, qf_field->size, CeedEvalModes[qf_field->eval_mode],
99d1d35e2fSjeremylt                        restr_num_comp);
100e15f9bd0SJeremy L Thompson     // LCOV_EXCL_STOP
101e15f9bd0SJeremy L Thompson     break;
102e15f9bd0SJeremy L Thompson   case CEED_EVAL_INTERP:
103d1d35e2fSjeremylt     if (size != num_comp)
104e15f9bd0SJeremy L Thompson       // LCOV_EXCL_START
105e15f9bd0SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_DIMENSION,
106e1e9e29dSJed Brown                        "Field '%s' of size %d and EvalMode %s: ElemRestriction/Basis has %d components",
107d1d35e2fSjeremylt                        qf_field->field_name, qf_field->size, CeedEvalModes[qf_field->eval_mode],
108d1d35e2fSjeremylt                        num_comp);
109e15f9bd0SJeremy L Thompson     // LCOV_EXCL_STOP
110e15f9bd0SJeremy L Thompson     break;
111e15f9bd0SJeremy L Thompson   case CEED_EVAL_GRAD:
112d1d35e2fSjeremylt     if (size != num_comp * dim)
113e15f9bd0SJeremy L Thompson       // LCOV_EXCL_START
114e15f9bd0SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_DIMENSION,
115d1d35e2fSjeremylt                        "Field '%s' of size %d and EvalMode %s in %d dimensions: "
116d1d35e2fSjeremylt                        "ElemRestriction/Basis has %d components",
117d1d35e2fSjeremylt                        qf_field->field_name, qf_field->size, CeedEvalModes[qf_field->eval_mode], dim,
118d1d35e2fSjeremylt                        num_comp);
119e15f9bd0SJeremy L Thompson     // LCOV_EXCL_STOP
120e15f9bd0SJeremy L Thompson     break;
121e15f9bd0SJeremy L Thompson   case CEED_EVAL_WEIGHT:
122d1d35e2fSjeremylt     // No additional checks required
123e15f9bd0SJeremy L Thompson     break;
124e15f9bd0SJeremy L Thompson   case CEED_EVAL_DIV:
125e15f9bd0SJeremy L Thompson     // Not implemented
126e15f9bd0SJeremy L Thompson     break;
127e15f9bd0SJeremy L Thompson   case CEED_EVAL_CURL:
128e15f9bd0SJeremy L Thompson     // Not implemented
129e15f9bd0SJeremy L Thompson     break;
130e15f9bd0SJeremy L Thompson   }
131e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1327a982d89SJeremy L. Thompson }
1337a982d89SJeremy L. Thompson 
1347a982d89SJeremy L. Thompson /**
1357a982d89SJeremy L. Thompson   @brief View a field of a CeedOperator
1367a982d89SJeremy L. Thompson 
1377a982d89SJeremy L. Thompson   @param[in] field         Operator field to view
138d1d35e2fSjeremylt   @param[in] qf_field      QFunction field (carries field name)
139d1d35e2fSjeremylt   @param[in] field_number  Number of field being viewed
1404c4400c7SValeria Barra   @param[in] sub           true indicates sub-operator, which increases indentation; false for top-level operator
141d1d35e2fSjeremylt   @param[in] input         true for an input field; false for output field
1427a982d89SJeremy L. Thompson   @param[in] stream        Stream to view to, e.g., stdout
1437a982d89SJeremy L. Thompson 
1447a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
1457a982d89SJeremy L. Thompson 
1467a982d89SJeremy L. Thompson   @ref Utility
1477a982d89SJeremy L. Thompson **/
1487a982d89SJeremy L. Thompson static int CeedOperatorFieldView(CeedOperatorField field,
149d1d35e2fSjeremylt                                  CeedQFunctionField qf_field,
150d1d35e2fSjeremylt                                  CeedInt field_number, bool sub, bool input,
1517a982d89SJeremy L. Thompson                                  FILE *stream) {
1527a982d89SJeremy L. Thompson   const char *pre = sub ? "  " : "";
153d1d35e2fSjeremylt   const char *in_out = input ? "Input" : "Output";
1547a982d89SJeremy L. Thompson 
1557a982d89SJeremy L. Thompson   fprintf(stream, "%s    %s Field [%d]:\n"
1567a982d89SJeremy L. Thompson           "%s      Name: \"%s\"\n",
157d1d35e2fSjeremylt           pre, in_out, field_number, pre, qf_field->field_name);
1587a982d89SJeremy L. Thompson 
1597a982d89SJeremy L. Thompson   if (field->basis == CEED_BASIS_COLLOCATED)
1607a982d89SJeremy L. Thompson     fprintf(stream, "%s      Collocated basis\n", pre);
1617a982d89SJeremy L. Thompson 
1627a982d89SJeremy L. Thompson   if (field->vec == CEED_VECTOR_ACTIVE)
1637a982d89SJeremy L. Thompson     fprintf(stream, "%s      Active vector\n", pre);
1647a982d89SJeremy L. Thompson   else if (field->vec == CEED_VECTOR_NONE)
1657a982d89SJeremy L. Thompson     fprintf(stream, "%s      No vector\n", pre);
166e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1677a982d89SJeremy L. Thompson }
1687a982d89SJeremy L. Thompson 
1697a982d89SJeremy L. Thompson /**
1707a982d89SJeremy L. Thompson   @brief View a single CeedOperator
1717a982d89SJeremy L. Thompson 
1727a982d89SJeremy L. Thompson   @param[in] op      CeedOperator to view
1737a982d89SJeremy L. Thompson   @param[in] sub     Boolean flag for sub-operator
1747a982d89SJeremy L. Thompson   @param[in] stream  Stream to write; typically stdout/stderr or a file
1757a982d89SJeremy L. Thompson 
1767a982d89SJeremy L. Thompson   @return Error code: 0 - success, otherwise - failure
1777a982d89SJeremy L. Thompson 
1787a982d89SJeremy L. Thompson   @ref Utility
1797a982d89SJeremy L. Thompson **/
1807a982d89SJeremy L. Thompson int CeedOperatorSingleView(CeedOperator op, bool sub, FILE *stream) {
1817a982d89SJeremy L. Thompson   int ierr;
1827a982d89SJeremy L. Thompson   const char *pre = sub ? "  " : "";
1837a982d89SJeremy L. Thompson 
18478464608Sjeremylt   CeedInt total_fields = 0;
18578464608Sjeremylt   ierr = CeedOperatorGetNumArgs(op, &total_fields); CeedChk(ierr);
1867a982d89SJeremy L. Thompson 
18778464608Sjeremylt   fprintf(stream, "%s  %d Field%s\n", pre, total_fields,
18878464608Sjeremylt           total_fields>1 ? "s" : "");
1897a982d89SJeremy L. Thompson 
190d1d35e2fSjeremylt   fprintf(stream, "%s  %d Input Field%s:\n", pre, op->qf->num_input_fields,
191d1d35e2fSjeremylt           op->qf->num_input_fields>1 ? "s" : "");
192d1d35e2fSjeremylt   for (CeedInt i=0; i<op->qf->num_input_fields; i++) {
193d1d35e2fSjeremylt     ierr = CeedOperatorFieldView(op->input_fields[i], op->qf->input_fields[i],
1947a982d89SJeremy L. Thompson                                  i, sub, 1, stream); CeedChk(ierr);
1957a982d89SJeremy L. Thompson   }
1967a982d89SJeremy L. Thompson 
197d1d35e2fSjeremylt   fprintf(stream, "%s  %d Output Field%s:\n", pre, op->qf->num_output_fields,
198d1d35e2fSjeremylt           op->qf->num_output_fields>1 ? "s" : "");
199d1d35e2fSjeremylt   for (CeedInt i=0; i<op->qf->num_output_fields; i++) {
200d1d35e2fSjeremylt     ierr = CeedOperatorFieldView(op->output_fields[i], op->qf->output_fields[i],
2017a982d89SJeremy L. Thompson                                  i, sub, 0, stream); CeedChk(ierr);
2027a982d89SJeremy L. Thompson   }
203e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2047a982d89SJeremy L. Thompson }
2057a982d89SJeremy L. Thompson 
206d99fa3c5SJeremy L Thompson /**
207eaf62fffSJeremy L Thompson   @brief Find the active vector basis for a CeedOperator
208eaf62fffSJeremy L Thompson 
209eaf62fffSJeremy L Thompson   @param[in] op             CeedOperator to find active basis for
210eaf62fffSJeremy L Thompson   @param[out] active_basis  Basis for active input vector
211eaf62fffSJeremy L Thompson 
212eaf62fffSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
213eaf62fffSJeremy L Thompson 
214eaf62fffSJeremy L Thompson   @ ref Developer
215eaf62fffSJeremy L Thompson **/
216eaf62fffSJeremy L Thompson int CeedOperatorGetActiveBasis(CeedOperator op, CeedBasis *active_basis) {
217eaf62fffSJeremy L Thompson   *active_basis = NULL;
218eaf62fffSJeremy L Thompson   for (int i = 0; i < op->qf->num_input_fields; i++)
219eaf62fffSJeremy L Thompson     if (op->input_fields[i]->vec == CEED_VECTOR_ACTIVE) {
220eaf62fffSJeremy L Thompson       *active_basis = op->input_fields[i]->basis;
221eaf62fffSJeremy L Thompson       break;
222eaf62fffSJeremy L Thompson     }
223eaf62fffSJeremy L Thompson 
224eaf62fffSJeremy L Thompson   if (!*active_basis) {
225eaf62fffSJeremy L Thompson     // LCOV_EXCL_START
226eaf62fffSJeremy L Thompson     int ierr;
227eaf62fffSJeremy L Thompson     Ceed ceed;
228eaf62fffSJeremy L Thompson     ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
229eaf62fffSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_MINOR,
230eaf62fffSJeremy L Thompson                      "No active CeedBasis found");
231eaf62fffSJeremy L Thompson     // LCOV_EXCL_STOP
232eaf62fffSJeremy L Thompson   }
233eaf62fffSJeremy L Thompson   return CEED_ERROR_SUCCESS;
234eaf62fffSJeremy L Thompson }
235eaf62fffSJeremy L Thompson 
236eaf62fffSJeremy L Thompson /**
237e2f04181SAndrew T. Barker   @brief Find the active vector ElemRestriction for a CeedOperator
238e2f04181SAndrew T. Barker 
239e2f04181SAndrew T. Barker   @param[in] op            CeedOperator to find active basis for
240d1d35e2fSjeremylt   @param[out] active_rstr  ElemRestriction for active input vector
241e2f04181SAndrew T. Barker 
242e2f04181SAndrew T. Barker   @return An error code: 0 - success, otherwise - failure
243e2f04181SAndrew T. Barker 
244e2f04181SAndrew T. Barker   @ref Utility
245e2f04181SAndrew T. Barker **/
246eaf62fffSJeremy L Thompson int CeedOperatorGetActiveElemRestriction(CeedOperator op,
247d1d35e2fSjeremylt     CeedElemRestriction *active_rstr) {
248d1d35e2fSjeremylt   *active_rstr = NULL;
249d1d35e2fSjeremylt   for (int i = 0; i < op->qf->num_input_fields; i++)
250d1d35e2fSjeremylt     if (op->input_fields[i]->vec == CEED_VECTOR_ACTIVE) {
251d1d35e2fSjeremylt       *active_rstr = op->input_fields[i]->elem_restr;
252e2f04181SAndrew T. Barker       break;
253e2f04181SAndrew T. Barker     }
254e2f04181SAndrew T. Barker 
255d1d35e2fSjeremylt   if (!*active_rstr) {
256e2f04181SAndrew T. Barker     // LCOV_EXCL_START
257e2f04181SAndrew T. Barker     int ierr;
258e2f04181SAndrew T. Barker     Ceed ceed;
259e2f04181SAndrew T. Barker     ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
260e2f04181SAndrew T. Barker     return CeedError(ceed, CEED_ERROR_INCOMPLETE,
261eaf62fffSJeremy L Thompson                      "No active CeedElemRestriction found");
262e2f04181SAndrew T. Barker     // LCOV_EXCL_STOP
263e2f04181SAndrew T. Barker   }
264e2f04181SAndrew T. Barker   return CEED_ERROR_SUCCESS;
265e2f04181SAndrew T. Barker }
266e2f04181SAndrew T. Barker 
267d8dd9a91SJeremy L Thompson /**
268d8dd9a91SJeremy L Thompson   @brief Set QFunctionContext field value of the specified type.
269d8dd9a91SJeremy L Thompson            For composite operators, the value is set in all
270d8dd9a91SJeremy L Thompson            sub-operator QFunctionContexts that have a matching `field_name`.
271d8dd9a91SJeremy L Thompson            A non-zero error code is returned for single operators
272d8dd9a91SJeremy L Thompson            that do not have a matching field of the same type or composite
273d8dd9a91SJeremy L Thompson            operators that do not have any field of a matching type.
274d8dd9a91SJeremy L Thompson 
275d8dd9a91SJeremy L Thompson   @param op          CeedOperator
2763668ca4bSJeremy L Thompson   @param field_label Label of field to set
277d8dd9a91SJeremy L Thompson   @param field_type  Type of field to set
278d8dd9a91SJeremy L Thompson   @param value       Value to set
279d8dd9a91SJeremy L Thompson 
280d8dd9a91SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
281d8dd9a91SJeremy L Thompson 
282d8dd9a91SJeremy L Thompson   @ref User
283d8dd9a91SJeremy L Thompson **/
284d8dd9a91SJeremy L Thompson static int CeedOperatorContextSetGeneric(CeedOperator op,
2853668ca4bSJeremy L Thompson     CeedContextFieldLabel field_label, CeedContextFieldType field_type,
2863668ca4bSJeremy L Thompson     void *value) {
287d8dd9a91SJeremy L Thompson   int ierr;
288d8dd9a91SJeremy L Thompson 
2893668ca4bSJeremy L Thompson   if (!field_label)
2903668ca4bSJeremy L Thompson     // LCOV_EXCL_START
2913668ca4bSJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_UNSUPPORTED,
2923668ca4bSJeremy L Thompson                      "Invalid field label");
2933668ca4bSJeremy L Thompson   // LCOV_EXCL_STOP
2943668ca4bSJeremy L Thompson 
2953668ca4bSJeremy L Thompson   bool is_composite = false;
296d8dd9a91SJeremy L Thompson   ierr = CeedOperatorIsComposite(op, &is_composite); CeedChk(ierr);
297d8dd9a91SJeremy L Thompson   if (is_composite) {
298d8dd9a91SJeremy L Thompson     CeedInt num_sub;
299d8dd9a91SJeremy L Thompson     CeedOperator *sub_operators;
300d8dd9a91SJeremy L Thompson 
301d8dd9a91SJeremy L Thompson     ierr = CeedOperatorGetNumSub(op, &num_sub); CeedChk(ierr);
302d8dd9a91SJeremy L Thompson     ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr);
3033668ca4bSJeremy L Thompson     if (num_sub != field_label->num_sub_labels)
3043668ca4bSJeremy L Thompson       // LCOV_EXCL_START
3053668ca4bSJeremy L Thompson       return CeedError(op->ceed, CEED_ERROR_UNSUPPORTED,
3063668ca4bSJeremy L Thompson                        "ContextLabel does not correspond to composite operator.\n"
3073668ca4bSJeremy L Thompson                        "Use CeedOperatorGetContextFieldLabel().");
3083668ca4bSJeremy L Thompson     // LCOV_EXCL_STOP
309d8dd9a91SJeremy L Thompson 
310d8dd9a91SJeremy L Thompson     for (CeedInt i = 0; i < num_sub; i++) {
311d8dd9a91SJeremy L Thompson       // Try every sub-operator, ok if some sub-operators do not have field
3123668ca4bSJeremy L Thompson       if (field_label->sub_labels[i] && sub_operators[i]->qf->ctx) {
3133668ca4bSJeremy L Thompson         ierr = CeedQFunctionContextSetGeneric(sub_operators[i]->qf->ctx,
3143668ca4bSJeremy L Thompson                                               field_label->sub_labels[i],
3153668ca4bSJeremy L Thompson                                               field_type, value); CeedChk(ierr);
316d8dd9a91SJeremy L Thompson       }
317d8dd9a91SJeremy L Thompson     }
318d8dd9a91SJeremy L Thompson   } else {
319d8dd9a91SJeremy L Thompson     if (!op->qf->ctx)
320d8dd9a91SJeremy L Thompson       // LCOV_EXCL_START
321d8dd9a91SJeremy L Thompson       return CeedError(op->ceed, CEED_ERROR_UNSUPPORTED,
322d8dd9a91SJeremy L Thompson                        "QFunction does not have context data");
323d8dd9a91SJeremy L Thompson     // LCOV_EXCL_STOP
324d8dd9a91SJeremy L Thompson 
3253668ca4bSJeremy L Thompson     ierr = CeedQFunctionContextSetGeneric(op->qf->ctx, field_label,
3263668ca4bSJeremy L Thompson                                           field_type, value); CeedChk(ierr);
327d8dd9a91SJeremy L Thompson   }
328d8dd9a91SJeremy L Thompson 
329d8dd9a91SJeremy L Thompson   return CEED_ERROR_SUCCESS;
330d8dd9a91SJeremy L Thompson }
331d8dd9a91SJeremy L Thompson 
3327a982d89SJeremy L. Thompson /// @}
3337a982d89SJeremy L. Thompson 
3347a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
3357a982d89SJeremy L. Thompson /// CeedOperator Backend API
3367a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
3377a982d89SJeremy L. Thompson /// @addtogroup CeedOperatorBackend
3387a982d89SJeremy L. Thompson /// @{
3397a982d89SJeremy L. Thompson 
3407a982d89SJeremy L. Thompson /**
3417a982d89SJeremy L. Thompson   @brief Get the number of arguments associated with a CeedOperator
3427a982d89SJeremy L. Thompson 
3437a982d89SJeremy L. Thompson   @param op             CeedOperator
344d1d35e2fSjeremylt   @param[out] num_args  Variable to store vector number of arguments
3457a982d89SJeremy L. Thompson 
3467a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
3477a982d89SJeremy L. Thompson 
3487a982d89SJeremy L. Thompson   @ref Backend
3497a982d89SJeremy L. Thompson **/
3507a982d89SJeremy L. Thompson 
351d1d35e2fSjeremylt int CeedOperatorGetNumArgs(CeedOperator op, CeedInt *num_args) {
352f04ea552SJeremy L Thompson   if (op->is_composite)
3537a982d89SJeremy L. Thompson     // LCOV_EXCL_START
354e15f9bd0SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR,
355e15f9bd0SJeremy L Thompson                      "Not defined for composite operators");
3567a982d89SJeremy L. Thompson   // LCOV_EXCL_STOP
3577a982d89SJeremy L. Thompson 
358d1d35e2fSjeremylt   *num_args = op->num_fields;
359e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3607a982d89SJeremy L. Thompson }
3617a982d89SJeremy L. Thompson 
3627a982d89SJeremy L. Thompson /**
3637a982d89SJeremy L. Thompson   @brief Get the setup status of a CeedOperator
3647a982d89SJeremy L. Thompson 
3657a982d89SJeremy L. Thompson   @param op                  CeedOperator
366d1d35e2fSjeremylt   @param[out] is_setup_done  Variable to store setup status
3677a982d89SJeremy L. Thompson 
3687a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
3697a982d89SJeremy L. Thompson 
3707a982d89SJeremy L. Thompson   @ref Backend
3717a982d89SJeremy L. Thompson **/
3727a982d89SJeremy L. Thompson 
373d1d35e2fSjeremylt int CeedOperatorIsSetupDone(CeedOperator op, bool *is_setup_done) {
374f04ea552SJeremy L Thompson   *is_setup_done = op->is_backend_setup;
375e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3767a982d89SJeremy L. Thompson }
3777a982d89SJeremy L. Thompson 
3787a982d89SJeremy L. Thompson /**
3797a982d89SJeremy L. Thompson   @brief Get the QFunction associated with a CeedOperator
3807a982d89SJeremy L. Thompson 
3817a982d89SJeremy L. Thompson   @param op       CeedOperator
3827a982d89SJeremy L. Thompson   @param[out] qf  Variable to store QFunction
3837a982d89SJeremy L. Thompson 
3847a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
3857a982d89SJeremy L. Thompson 
3867a982d89SJeremy L. Thompson   @ref Backend
3877a982d89SJeremy L. Thompson **/
3887a982d89SJeremy L. Thompson 
3897a982d89SJeremy L. Thompson int CeedOperatorGetQFunction(CeedOperator op, CeedQFunction *qf) {
390f04ea552SJeremy L Thompson   if (op->is_composite)
3917a982d89SJeremy L. Thompson     // LCOV_EXCL_START
392e15f9bd0SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR,
393e15f9bd0SJeremy L Thompson                      "Not defined for composite operator");
3947a982d89SJeremy L. Thompson   // LCOV_EXCL_STOP
3957a982d89SJeremy L. Thompson 
3967a982d89SJeremy L. Thompson   *qf = op->qf;
397e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3987a982d89SJeremy L. Thompson }
3997a982d89SJeremy L. Thompson 
4007a982d89SJeremy L. Thompson /**
401c04a41a7SJeremy L Thompson   @brief Get a boolean value indicating if the CeedOperator is composite
402c04a41a7SJeremy L Thompson 
403c04a41a7SJeremy L Thompson   @param op                 CeedOperator
404d1d35e2fSjeremylt   @param[out] is_composite  Variable to store composite status
405c04a41a7SJeremy L Thompson 
406c04a41a7SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
407c04a41a7SJeremy L Thompson 
408c04a41a7SJeremy L Thompson   @ref Backend
409c04a41a7SJeremy L Thompson **/
410c04a41a7SJeremy L Thompson 
411d1d35e2fSjeremylt int CeedOperatorIsComposite(CeedOperator op, bool *is_composite) {
412f04ea552SJeremy L Thompson   *is_composite = op->is_composite;
413e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
414c04a41a7SJeremy L Thompson }
415c04a41a7SJeremy L Thompson 
416c04a41a7SJeremy L Thompson /**
417d1d35e2fSjeremylt   @brief Get the number of sub_operators associated with a CeedOperator
4187a982d89SJeremy L. Thompson 
4197a982d89SJeremy L. Thompson   @param op                     CeedOperator
420d1d35e2fSjeremylt   @param[out] num_suboperators  Variable to store number of sub_operators
4217a982d89SJeremy L. Thompson 
4227a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
4237a982d89SJeremy L. Thompson 
4247a982d89SJeremy L. Thompson   @ref Backend
4257a982d89SJeremy L. Thompson **/
4267a982d89SJeremy L. Thompson 
427d1d35e2fSjeremylt int CeedOperatorGetNumSub(CeedOperator op, CeedInt *num_suboperators) {
428f04ea552SJeremy L Thompson   if (!op->is_composite)
4297a982d89SJeremy L. Thompson     // LCOV_EXCL_START
430e15f9bd0SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR, "Not a composite operator");
4317a982d89SJeremy L. Thompson   // LCOV_EXCL_STOP
4327a982d89SJeremy L. Thompson 
433d1d35e2fSjeremylt   *num_suboperators = op->num_suboperators;
434e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4357a982d89SJeremy L. Thompson }
4367a982d89SJeremy L. Thompson 
4377a982d89SJeremy L. Thompson /**
438d1d35e2fSjeremylt   @brief Get the list of sub_operators associated with a CeedOperator
4397a982d89SJeremy L. Thompson 
4407a982d89SJeremy L. Thompson   @param op                  CeedOperator
441d1d35e2fSjeremylt   @param[out] sub_operators  Variable to store list of sub_operators
4427a982d89SJeremy L. Thompson 
4437a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
4447a982d89SJeremy L. Thompson 
4457a982d89SJeremy L. Thompson   @ref Backend
4467a982d89SJeremy L. Thompson **/
4477a982d89SJeremy L. Thompson 
448d1d35e2fSjeremylt int CeedOperatorGetSubList(CeedOperator op, CeedOperator **sub_operators) {
449f04ea552SJeremy L Thompson   if (!op->is_composite)
4507a982d89SJeremy L. Thompson     // LCOV_EXCL_START
451e15f9bd0SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR, "Not a composite operator");
4527a982d89SJeremy L. Thompson   // LCOV_EXCL_STOP
4537a982d89SJeremy L. Thompson 
454d1d35e2fSjeremylt   *sub_operators = op->sub_operators;
455e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4567a982d89SJeremy L. Thompson }
4577a982d89SJeremy L. Thompson 
4587a982d89SJeremy L. Thompson /**
4597a982d89SJeremy L. Thompson   @brief Get the backend data of a CeedOperator
4607a982d89SJeremy L. Thompson 
4617a982d89SJeremy L. Thompson   @param op         CeedOperator
4627a982d89SJeremy L. Thompson   @param[out] data  Variable to store data
4637a982d89SJeremy L. Thompson 
4647a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
4657a982d89SJeremy L. Thompson 
4667a982d89SJeremy L. Thompson   @ref Backend
4677a982d89SJeremy L. Thompson **/
4687a982d89SJeremy L. Thompson 
469777ff853SJeremy L Thompson int CeedOperatorGetData(CeedOperator op, void *data) {
470777ff853SJeremy L Thompson   *(void **)data = op->data;
471e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4727a982d89SJeremy L. Thompson }
4737a982d89SJeremy L. Thompson 
4747a982d89SJeremy L. Thompson /**
4757a982d89SJeremy L. Thompson   @brief Set the backend data of a CeedOperator
4767a982d89SJeremy L. Thompson 
4777a982d89SJeremy L. Thompson   @param[out] op  CeedOperator
4787a982d89SJeremy L. Thompson   @param data     Data to set
4797a982d89SJeremy L. Thompson 
4807a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
4817a982d89SJeremy L. Thompson 
4827a982d89SJeremy L. Thompson   @ref Backend
4837a982d89SJeremy L. Thompson **/
4847a982d89SJeremy L. Thompson 
485777ff853SJeremy L Thompson int CeedOperatorSetData(CeedOperator op, void *data) {
486777ff853SJeremy L Thompson   op->data = data;
487e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4887a982d89SJeremy L. Thompson }
4897a982d89SJeremy L. Thompson 
4907a982d89SJeremy L. Thompson /**
49134359f16Sjeremylt   @brief Increment the reference counter for a CeedOperator
49234359f16Sjeremylt 
49334359f16Sjeremylt   @param op  CeedOperator to increment the reference counter
49434359f16Sjeremylt 
49534359f16Sjeremylt   @return An error code: 0 - success, otherwise - failure
49634359f16Sjeremylt 
49734359f16Sjeremylt   @ref Backend
49834359f16Sjeremylt **/
4999560d06aSjeremylt int CeedOperatorReference(CeedOperator op) {
50034359f16Sjeremylt   op->ref_count++;
50134359f16Sjeremylt   return CEED_ERROR_SUCCESS;
50234359f16Sjeremylt }
50334359f16Sjeremylt 
50434359f16Sjeremylt /**
5057a982d89SJeremy L. Thompson   @brief Set the setup flag of a CeedOperator to True
5067a982d89SJeremy L. Thompson 
5077a982d89SJeremy L. Thompson   @param op  CeedOperator
5087a982d89SJeremy L. Thompson 
5097a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
5107a982d89SJeremy L. Thompson 
5117a982d89SJeremy L. Thompson   @ref Backend
5127a982d89SJeremy L. Thompson **/
5137a982d89SJeremy L. Thompson 
5147a982d89SJeremy L. Thompson int CeedOperatorSetSetupDone(CeedOperator op) {
515f04ea552SJeremy L Thompson   op->is_backend_setup = true;
516e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
5177a982d89SJeremy L. Thompson }
5187a982d89SJeremy L. Thompson 
5197a982d89SJeremy L. Thompson /// @}
5207a982d89SJeremy L. Thompson 
5217a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
5227a982d89SJeremy L. Thompson /// CeedOperator Public API
5237a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
5247a982d89SJeremy L. Thompson /// @addtogroup CeedOperatorUser
525dfdf5a53Sjeremylt /// @{
526d7b241e6Sjeremylt 
527d7b241e6Sjeremylt /**
5280219ea01SJeremy L Thompson   @brief Create a CeedOperator and associate a CeedQFunction. A CeedBasis and
5290219ea01SJeremy L Thompson            CeedElemRestriction can be associated with CeedQFunction fields with
5300219ea01SJeremy L Thompson            \ref CeedOperatorSetField.
531d7b241e6Sjeremylt 
532b11c1e72Sjeremylt   @param ceed     A Ceed object where the CeedOperator will be created
533d7b241e6Sjeremylt   @param qf       QFunction defining the action of the operator at quadrature points
53434138859Sjeremylt   @param dqf      QFunction defining the action of the Jacobian of @a qf (or
5354cc79fe7SJed Brown                     @ref CEED_QFUNCTION_NONE)
536d7b241e6Sjeremylt   @param dqfT     QFunction defining the action of the transpose of the Jacobian
5374cc79fe7SJed Brown                     of @a qf (or @ref CEED_QFUNCTION_NONE)
538b11c1e72Sjeremylt   @param[out] op  Address of the variable where the newly created
539b11c1e72Sjeremylt                     CeedOperator will be stored
540b11c1e72Sjeremylt 
541b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
542dfdf5a53Sjeremylt 
5437a982d89SJeremy L. Thompson   @ref User
544d7b241e6Sjeremylt  */
545d7b241e6Sjeremylt int CeedOperatorCreate(Ceed ceed, CeedQFunction qf, CeedQFunction dqf,
546d7b241e6Sjeremylt                        CeedQFunction dqfT, CeedOperator *op) {
547d7b241e6Sjeremylt   int ierr;
548d7b241e6Sjeremylt 
5495fe0d4faSjeremylt   if (!ceed->OperatorCreate) {
5505fe0d4faSjeremylt     Ceed delegate;
551aefd8378Sjeremylt     ierr = CeedGetObjectDelegate(ceed, &delegate, "Operator"); CeedChk(ierr);
5525fe0d4faSjeremylt 
5535fe0d4faSjeremylt     if (!delegate)
554c042f62fSJeremy L Thompson       // LCOV_EXCL_START
555e15f9bd0SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
556e15f9bd0SJeremy L Thompson                        "Backend does not support OperatorCreate");
557c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
5585fe0d4faSjeremylt 
5595fe0d4faSjeremylt     ierr = CeedOperatorCreate(delegate, qf, dqf, dqfT, op); CeedChk(ierr);
560e15f9bd0SJeremy L Thompson     return CEED_ERROR_SUCCESS;
5615fe0d4faSjeremylt   }
5625fe0d4faSjeremylt 
563b3b7035fSJeremy L Thompson   if (!qf || qf == CEED_QFUNCTION_NONE)
564b3b7035fSJeremy L Thompson     // LCOV_EXCL_START
565e15f9bd0SJeremy L Thompson     return CeedError(ceed, CEED_ERROR_MINOR,
566e15f9bd0SJeremy L Thompson                      "Operator must have a valid QFunction.");
567b3b7035fSJeremy L Thompson   // LCOV_EXCL_STOP
568d7b241e6Sjeremylt   ierr = CeedCalloc(1, op); CeedChk(ierr);
569d7b241e6Sjeremylt   (*op)->ceed = ceed;
5709560d06aSjeremylt   ierr = CeedReference(ceed); CeedChk(ierr);
571d1d35e2fSjeremylt   (*op)->ref_count = 1;
572d7b241e6Sjeremylt   (*op)->qf = qf;
5739560d06aSjeremylt   ierr = CeedQFunctionReference(qf); CeedChk(ierr);
574442e7f0bSjeremylt   if (dqf && dqf != CEED_QFUNCTION_NONE) {
575d7b241e6Sjeremylt     (*op)->dqf = dqf;
5769560d06aSjeremylt     ierr = CeedQFunctionReference(dqf); CeedChk(ierr);
577442e7f0bSjeremylt   }
578442e7f0bSjeremylt   if (dqfT && dqfT != CEED_QFUNCTION_NONE) {
579d7b241e6Sjeremylt     (*op)->dqfT = dqfT;
5809560d06aSjeremylt     ierr = CeedQFunctionReference(dqfT); CeedChk(ierr);
581442e7f0bSjeremylt   }
582*480fae85SJeremy L Thompson   ierr = CeedQFunctionAssemblyDataCreate(ceed, &(*op)->qf_assembled);
583*480fae85SJeremy L Thompson   CeedChk(ierr);
584bf4cb664SJeremy L Thompson   ierr = CeedCalloc(CEED_FIELD_MAX, &(*op)->input_fields); CeedChk(ierr);
585bf4cb664SJeremy L Thompson   ierr = CeedCalloc(CEED_FIELD_MAX, &(*op)->output_fields); CeedChk(ierr);
586d7b241e6Sjeremylt   ierr = ceed->OperatorCreate(*op); CeedChk(ierr);
587e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
588d7b241e6Sjeremylt }
589d7b241e6Sjeremylt 
590d7b241e6Sjeremylt /**
59152d6035fSJeremy L Thompson   @brief Create an operator that composes the action of several operators
59252d6035fSJeremy L Thompson 
59352d6035fSJeremy L Thompson   @param ceed     A Ceed object where the CeedOperator will be created
59452d6035fSJeremy L Thompson   @param[out] op  Address of the variable where the newly created
59552d6035fSJeremy L Thompson                     Composite CeedOperator will be stored
59652d6035fSJeremy L Thompson 
59752d6035fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
59852d6035fSJeremy L Thompson 
5997a982d89SJeremy L. Thompson   @ref User
60052d6035fSJeremy L Thompson  */
60152d6035fSJeremy L Thompson int CeedCompositeOperatorCreate(Ceed ceed, CeedOperator *op) {
60252d6035fSJeremy L Thompson   int ierr;
60352d6035fSJeremy L Thompson 
60452d6035fSJeremy L Thompson   if (!ceed->CompositeOperatorCreate) {
60552d6035fSJeremy L Thompson     Ceed delegate;
606aefd8378Sjeremylt     ierr = CeedGetObjectDelegate(ceed, &delegate, "Operator"); CeedChk(ierr);
60752d6035fSJeremy L Thompson 
608250756a7Sjeremylt     if (delegate) {
60952d6035fSJeremy L Thompson       ierr = CeedCompositeOperatorCreate(delegate, op); CeedChk(ierr);
610e15f9bd0SJeremy L Thompson       return CEED_ERROR_SUCCESS;
61152d6035fSJeremy L Thompson     }
612250756a7Sjeremylt   }
61352d6035fSJeremy L Thompson 
61452d6035fSJeremy L Thompson   ierr = CeedCalloc(1, op); CeedChk(ierr);
61552d6035fSJeremy L Thompson   (*op)->ceed = ceed;
6169560d06aSjeremylt   ierr = CeedReference(ceed); CeedChk(ierr);
617f04ea552SJeremy L Thompson   (*op)->is_composite = true;
618bf4cb664SJeremy L Thompson   ierr = CeedCalloc(CEED_COMPOSITE_MAX, &(*op)->sub_operators); CeedChk(ierr);
619250756a7Sjeremylt 
620250756a7Sjeremylt   if (ceed->CompositeOperatorCreate) {
62152d6035fSJeremy L Thompson     ierr = ceed->CompositeOperatorCreate(*op); CeedChk(ierr);
622250756a7Sjeremylt   }
623e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
62452d6035fSJeremy L Thompson }
62552d6035fSJeremy L Thompson 
62652d6035fSJeremy L Thompson /**
6279560d06aSjeremylt   @brief Copy the pointer to a CeedOperator. Both pointers should
6289560d06aSjeremylt            be destroyed with `CeedOperatorDestroy()`;
6299560d06aSjeremylt            Note: If `*op_copy` is non-NULL, then it is assumed that
6309560d06aSjeremylt            `*op_copy` is a pointer to a CeedOperator. This
6319560d06aSjeremylt            CeedOperator will be destroyed if `*op_copy` is the only
6329560d06aSjeremylt            reference to this CeedOperator.
6339560d06aSjeremylt 
6349560d06aSjeremylt   @param op            CeedOperator to copy reference to
6359560d06aSjeremylt   @param[out] op_copy  Variable to store copied reference
6369560d06aSjeremylt 
6379560d06aSjeremylt   @return An error code: 0 - success, otherwise - failure
6389560d06aSjeremylt 
6399560d06aSjeremylt   @ref User
6409560d06aSjeremylt **/
6419560d06aSjeremylt int CeedOperatorReferenceCopy(CeedOperator op, CeedOperator *op_copy) {
6429560d06aSjeremylt   int ierr;
6439560d06aSjeremylt 
6449560d06aSjeremylt   ierr = CeedOperatorReference(op); CeedChk(ierr);
6459560d06aSjeremylt   ierr = CeedOperatorDestroy(op_copy); CeedChk(ierr);
6469560d06aSjeremylt   *op_copy = op;
6479560d06aSjeremylt   return CEED_ERROR_SUCCESS;
6489560d06aSjeremylt }
6499560d06aSjeremylt 
6509560d06aSjeremylt /**
651b11c1e72Sjeremylt   @brief Provide a field to a CeedOperator for use by its CeedQFunction
652d7b241e6Sjeremylt 
653d7b241e6Sjeremylt   This function is used to specify both active and passive fields to a
654d7b241e6Sjeremylt   CeedOperator.  For passive fields, a vector @arg v must be provided.  Passive
655d7b241e6Sjeremylt   fields can inputs or outputs (updated in-place when operator is applied).
656d7b241e6Sjeremylt 
657d7b241e6Sjeremylt   Active fields must be specified using this function, but their data (in a
658d7b241e6Sjeremylt   CeedVector) is passed in CeedOperatorApply().  There can be at most one active
659d7b241e6Sjeremylt   input and at most one active output.
660d7b241e6Sjeremylt 
6618c91a0c9SJeremy L Thompson   @param op          CeedOperator on which to provide the field
662d1d35e2fSjeremylt   @param field_name  Name of the field (to be matched with the name used by
6638795c945Sjeremylt                        CeedQFunction)
664b11c1e72Sjeremylt   @param r           CeedElemRestriction
6654cc79fe7SJed Brown   @param b           CeedBasis in which the field resides or @ref CEED_BASIS_COLLOCATED
666b11c1e72Sjeremylt                        if collocated with quadrature points
6674cc79fe7SJed Brown   @param v           CeedVector to be used by CeedOperator or @ref CEED_VECTOR_ACTIVE
6684cc79fe7SJed Brown                        if field is active or @ref CEED_VECTOR_NONE if using
6694cc79fe7SJed Brown                        @ref CEED_EVAL_WEIGHT in the QFunction
670b11c1e72Sjeremylt 
671b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
672dfdf5a53Sjeremylt 
6737a982d89SJeremy L. Thompson   @ref User
674b11c1e72Sjeremylt **/
675d1d35e2fSjeremylt int CeedOperatorSetField(CeedOperator op, const char *field_name,
676a8d32208Sjeremylt                          CeedElemRestriction r, CeedBasis b, CeedVector v) {
677d7b241e6Sjeremylt   int ierr;
678f04ea552SJeremy L Thompson   if (op->is_composite)
679c042f62fSJeremy L Thompson     // LCOV_EXCL_START
680e1e9e29dSJed Brown     return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
681e15f9bd0SJeremy L Thompson                      "Cannot add field to composite operator.");
682c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
683f04ea552SJeremy L Thompson   if (op->is_immutable)
684f04ea552SJeremy L Thompson     // LCOV_EXCL_START
685f04ea552SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MAJOR,
686f04ea552SJeremy L Thompson                      "Operator cannot be changed after set as immutable");
687f04ea552SJeremy L Thompson   // LCOV_EXCL_STOP
6888b067b84SJed Brown   if (!r)
689c042f62fSJeremy L Thompson     // LCOV_EXCL_START
690e1e9e29dSJed Brown     return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
691c042f62fSJeremy L Thompson                      "ElemRestriction r for field \"%s\" must be non-NULL.",
692d1d35e2fSjeremylt                      field_name);
693c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
6948b067b84SJed Brown   if (!b)
695c042f62fSJeremy L Thompson     // LCOV_EXCL_START
696e1e9e29dSJed Brown     return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
697e15f9bd0SJeremy L Thompson                      "Basis b for field \"%s\" must be non-NULL.",
698d1d35e2fSjeremylt                      field_name);
699c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
7008b067b84SJed Brown   if (!v)
701c042f62fSJeremy L Thompson     // LCOV_EXCL_START
702e1e9e29dSJed Brown     return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
703e15f9bd0SJeremy L Thompson                      "Vector v for field \"%s\" must be non-NULL.",
704d1d35e2fSjeremylt                      field_name);
705c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
70652d6035fSJeremy L Thompson 
707d1d35e2fSjeremylt   CeedInt num_elem;
708d1d35e2fSjeremylt   ierr = CeedElemRestrictionGetNumElements(r, &num_elem); CeedChk(ierr);
709d1d35e2fSjeremylt   if (r != CEED_ELEMRESTRICTION_NONE && op->has_restriction &&
710d1d35e2fSjeremylt       op->num_elem != num_elem)
711c042f62fSJeremy L Thompson     // LCOV_EXCL_START
712e15f9bd0SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_DIMENSION,
7131d102b48SJeremy L Thompson                      "ElemRestriction with %d elements incompatible with prior "
714d1d35e2fSjeremylt                      "%d elements", num_elem, op->num_elem);
715c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
716d7b241e6Sjeremylt 
71778464608Sjeremylt   CeedInt num_qpts = 0;
718e15f9bd0SJeremy L Thompson   if (b != CEED_BASIS_COLLOCATED) {
719d1d35e2fSjeremylt     ierr = CeedBasisGetNumQuadraturePoints(b, &num_qpts); CeedChk(ierr);
720d1d35e2fSjeremylt     if (op->num_qpts && op->num_qpts != num_qpts)
721c042f62fSJeremy L Thompson       // LCOV_EXCL_START
722e15f9bd0SJeremy L Thompson       return CeedError(op->ceed, CEED_ERROR_DIMENSION,
723e15f9bd0SJeremy L Thompson                        "Basis with %d quadrature points "
724d1d35e2fSjeremylt                        "incompatible with prior %d points", num_qpts,
725d1d35e2fSjeremylt                        op->num_qpts);
726c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
727d7b241e6Sjeremylt   }
728d1d35e2fSjeremylt   CeedQFunctionField qf_field;
729d1d35e2fSjeremylt   CeedOperatorField *op_field;
730d1d35e2fSjeremylt   for (CeedInt i=0; i<op->qf->num_input_fields; i++) {
731d1d35e2fSjeremylt     if (!strcmp(field_name, (*op->qf->input_fields[i]).field_name)) {
732d1d35e2fSjeremylt       qf_field = op->qf->input_fields[i];
733d1d35e2fSjeremylt       op_field = &op->input_fields[i];
734d7b241e6Sjeremylt       goto found;
735d7b241e6Sjeremylt     }
736d7b241e6Sjeremylt   }
737d1d35e2fSjeremylt   for (CeedInt i=0; i<op->qf->num_output_fields; i++) {
738d1d35e2fSjeremylt     if (!strcmp(field_name, (*op->qf->output_fields[i]).field_name)) {
739d1d35e2fSjeremylt       qf_field = op->qf->output_fields[i];
740d1d35e2fSjeremylt       op_field = &op->output_fields[i];
741d7b241e6Sjeremylt       goto found;
742d7b241e6Sjeremylt     }
743d7b241e6Sjeremylt   }
744c042f62fSJeremy L Thompson   // LCOV_EXCL_START
745e15f9bd0SJeremy L Thompson   return CeedError(op->ceed, CEED_ERROR_INCOMPLETE,
746e15f9bd0SJeremy L Thompson                    "QFunction has no knowledge of field '%s'",
747d1d35e2fSjeremylt                    field_name);
748c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
749d7b241e6Sjeremylt found:
750d1d35e2fSjeremylt   ierr = CeedOperatorCheckField(op->ceed, qf_field, r, b); CeedChk(ierr);
751d1d35e2fSjeremylt   ierr = CeedCalloc(1, op_field); CeedChk(ierr);
752e15f9bd0SJeremy L Thompson 
753d1d35e2fSjeremylt   (*op_field)->vec = v;
754e15f9bd0SJeremy L Thompson   if (v != CEED_VECTOR_ACTIVE && v != CEED_VECTOR_NONE) {
7559560d06aSjeremylt     ierr = CeedVectorReference(v); CeedChk(ierr);
756e15f9bd0SJeremy L Thompson   }
757e15f9bd0SJeremy L Thompson 
758d1d35e2fSjeremylt   (*op_field)->elem_restr = r;
7599560d06aSjeremylt   ierr = CeedElemRestrictionReference(r); CeedChk(ierr);
760e15f9bd0SJeremy L Thompson   if (r != CEED_ELEMRESTRICTION_NONE) {
761d1d35e2fSjeremylt     op->num_elem = num_elem;
762d1d35e2fSjeremylt     op->has_restriction = true; // Restriction set, but num_elem may be 0
763e15f9bd0SJeremy L Thompson   }
764d99fa3c5SJeremy L Thompson 
765d1d35e2fSjeremylt   (*op_field)->basis = b;
766e15f9bd0SJeremy L Thompson   if (b != CEED_BASIS_COLLOCATED) {
767cd4dfc48Sjeremylt     if (!op->num_qpts) {
768cd4dfc48Sjeremylt       ierr = CeedOperatorSetNumQuadraturePoints(op, num_qpts); CeedChk(ierr);
769cd4dfc48Sjeremylt     }
7709560d06aSjeremylt     ierr = CeedBasisReference(b); CeedChk(ierr);
771e15f9bd0SJeremy L Thompson   }
772e15f9bd0SJeremy L Thompson 
773d1d35e2fSjeremylt   op->num_fields += 1;
774f7e22acaSJeremy L Thompson   ierr = CeedStringAllocCopy(field_name, (char **)&(*op_field)->field_name);
775f7e22acaSJeremy L Thompson   CeedChk(ierr);
776e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
777d7b241e6Sjeremylt }
778d7b241e6Sjeremylt 
779d7b241e6Sjeremylt /**
78043bbe138SJeremy L Thompson   @brief Get the CeedOperatorFields of a CeedOperator
78143bbe138SJeremy L Thompson 
782f04ea552SJeremy L Thompson   Note: Calling this function asserts that setup is complete
783f04ea552SJeremy L Thompson           and sets the CeedOperator as immutable.
784f04ea552SJeremy L Thompson 
78543bbe138SJeremy L Thompson   @param op                      CeedOperator
786f74ec584SJeremy L Thompson   @param[out] num_input_fields   Variable to store number of input fields
78743bbe138SJeremy L Thompson   @param[out] input_fields       Variable to store input_fields
788f74ec584SJeremy L Thompson   @param[out] num_output_fields  Variable to store number of output fields
78943bbe138SJeremy L Thompson   @param[out] output_fields      Variable to store output_fields
79043bbe138SJeremy L Thompson 
79143bbe138SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
79243bbe138SJeremy L Thompson 
793e9b533fbSJeremy L Thompson   @ref Advanced
79443bbe138SJeremy L Thompson **/
79543bbe138SJeremy L Thompson int CeedOperatorGetFields(CeedOperator op, CeedInt *num_input_fields,
79643bbe138SJeremy L Thompson                           CeedOperatorField **input_fields,
79743bbe138SJeremy L Thompson                           CeedInt *num_output_fields,
79843bbe138SJeremy L Thompson                           CeedOperatorField **output_fields) {
799f04ea552SJeremy L Thompson   int ierr;
800f04ea552SJeremy L Thompson 
801f04ea552SJeremy L Thompson   if (op->is_composite)
80243bbe138SJeremy L Thompson     // LCOV_EXCL_START
80343bbe138SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR,
80443bbe138SJeremy L Thompson                      "Not defined for composite operator");
80543bbe138SJeremy L Thompson   // LCOV_EXCL_STOP
806f04ea552SJeremy L Thompson   ierr = CeedOperatorCheckReady(op); CeedChk(ierr);
80743bbe138SJeremy L Thompson 
80843bbe138SJeremy L Thompson   if (num_input_fields) *num_input_fields = op->qf->num_input_fields;
80943bbe138SJeremy L Thompson   if (input_fields) *input_fields = op->input_fields;
81043bbe138SJeremy L Thompson   if (num_output_fields) *num_output_fields = op->qf->num_output_fields;
81143bbe138SJeremy L Thompson   if (output_fields) *output_fields = op->output_fields;
81243bbe138SJeremy L Thompson   return CEED_ERROR_SUCCESS;
81343bbe138SJeremy L Thompson }
81443bbe138SJeremy L Thompson 
81543bbe138SJeremy L Thompson /**
81628567f8fSJeremy L Thompson   @brief Get the name of a CeedOperatorField
81728567f8fSJeremy L Thompson 
81828567f8fSJeremy L Thompson   @param op_field         CeedOperatorField
81928567f8fSJeremy L Thompson   @param[out] field_name  Variable to store the field name
82028567f8fSJeremy L Thompson 
82128567f8fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
82228567f8fSJeremy L Thompson 
823e9b533fbSJeremy L Thompson   @ref Advanced
82428567f8fSJeremy L Thompson **/
82528567f8fSJeremy L Thompson int CeedOperatorFieldGetName(CeedOperatorField op_field, char **field_name) {
82628567f8fSJeremy L Thompson   *field_name = (char *)op_field->field_name;
82728567f8fSJeremy L Thompson   return CEED_ERROR_SUCCESS;
82828567f8fSJeremy L Thompson }
82928567f8fSJeremy L Thompson 
83028567f8fSJeremy L Thompson /**
83143bbe138SJeremy L Thompson   @brief Get the CeedElemRestriction of a CeedOperatorField
83243bbe138SJeremy L Thompson 
83343bbe138SJeremy L Thompson   @param op_field   CeedOperatorField
83443bbe138SJeremy L Thompson   @param[out] rstr  Variable to store CeedElemRestriction
83543bbe138SJeremy L Thompson 
83643bbe138SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
83743bbe138SJeremy L Thompson 
838e9b533fbSJeremy L Thompson   @ref Advanced
83943bbe138SJeremy L Thompson **/
84043bbe138SJeremy L Thompson int CeedOperatorFieldGetElemRestriction(CeedOperatorField op_field,
84143bbe138SJeremy L Thompson                                         CeedElemRestriction *rstr) {
84243bbe138SJeremy L Thompson   *rstr = op_field->elem_restr;
84343bbe138SJeremy L Thompson   return CEED_ERROR_SUCCESS;
84443bbe138SJeremy L Thompson }
84543bbe138SJeremy L Thompson 
84643bbe138SJeremy L Thompson /**
84743bbe138SJeremy L Thompson   @brief Get the CeedBasis of a CeedOperatorField
84843bbe138SJeremy L Thompson 
84943bbe138SJeremy L Thompson   @param op_field    CeedOperatorField
85043bbe138SJeremy L Thompson   @param[out] basis  Variable to store CeedBasis
85143bbe138SJeremy L Thompson 
85243bbe138SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
85343bbe138SJeremy L Thompson 
854e9b533fbSJeremy L Thompson   @ref Advanced
85543bbe138SJeremy L Thompson **/
85643bbe138SJeremy L Thompson int CeedOperatorFieldGetBasis(CeedOperatorField op_field, CeedBasis *basis) {
85743bbe138SJeremy L Thompson   *basis = op_field->basis;
85843bbe138SJeremy L Thompson   return CEED_ERROR_SUCCESS;
85943bbe138SJeremy L Thompson }
86043bbe138SJeremy L Thompson 
86143bbe138SJeremy L Thompson /**
86243bbe138SJeremy L Thompson   @brief Get the CeedVector of a CeedOperatorField
86343bbe138SJeremy L Thompson 
86443bbe138SJeremy L Thompson   @param op_field  CeedOperatorField
86543bbe138SJeremy L Thompson   @param[out] vec  Variable to store CeedVector
86643bbe138SJeremy L Thompson 
86743bbe138SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
86843bbe138SJeremy L Thompson 
869e9b533fbSJeremy L Thompson   @ref Advanced
87043bbe138SJeremy L Thompson **/
87143bbe138SJeremy L Thompson int CeedOperatorFieldGetVector(CeedOperatorField op_field, CeedVector *vec) {
87243bbe138SJeremy L Thompson   *vec = op_field->vec;
87343bbe138SJeremy L Thompson   return CEED_ERROR_SUCCESS;
87443bbe138SJeremy L Thompson }
87543bbe138SJeremy L Thompson 
87643bbe138SJeremy L Thompson /**
87752d6035fSJeremy L Thompson   @brief Add a sub-operator to a composite CeedOperator
878288c0443SJeremy L Thompson 
879d1d35e2fSjeremylt   @param[out] composite_op  Composite CeedOperator
880d1d35e2fSjeremylt   @param      sub_op        Sub-operator CeedOperator
88152d6035fSJeremy L Thompson 
88252d6035fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
88352d6035fSJeremy L Thompson 
8847a982d89SJeremy L. Thompson   @ref User
88552d6035fSJeremy L Thompson  */
886d1d35e2fSjeremylt int CeedCompositeOperatorAddSub(CeedOperator composite_op,
887d1d35e2fSjeremylt                                 CeedOperator sub_op) {
88834359f16Sjeremylt   int ierr;
88934359f16Sjeremylt 
890f04ea552SJeremy L Thompson   if (!composite_op->is_composite)
891c042f62fSJeremy L Thompson     // LCOV_EXCL_START
892d1d35e2fSjeremylt     return CeedError(composite_op->ceed, CEED_ERROR_MINOR,
893e2f04181SAndrew T. Barker                      "CeedOperator is not a composite operator");
894c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
89552d6035fSJeremy L Thompson 
896d1d35e2fSjeremylt   if (composite_op->num_suboperators == CEED_COMPOSITE_MAX)
897c042f62fSJeremy L Thompson     // LCOV_EXCL_START
898d1d35e2fSjeremylt     return CeedError(composite_op->ceed, CEED_ERROR_UNSUPPORTED,
899d1d35e2fSjeremylt                      "Cannot add additional sub_operators");
900c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
901f04ea552SJeremy L Thompson   if (composite_op->is_immutable)
902f04ea552SJeremy L Thompson     // LCOV_EXCL_START
903f04ea552SJeremy L Thompson     return CeedError(composite_op->ceed, CEED_ERROR_MAJOR,
904f04ea552SJeremy L Thompson                      "Operator cannot be changed after set as immutable");
905f04ea552SJeremy L Thompson   // LCOV_EXCL_STOP
90652d6035fSJeremy L Thompson 
907d1d35e2fSjeremylt   composite_op->sub_operators[composite_op->num_suboperators] = sub_op;
9089560d06aSjeremylt   ierr = CeedOperatorReference(sub_op); CeedChk(ierr);
909d1d35e2fSjeremylt   composite_op->num_suboperators++;
910e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
91152d6035fSJeremy L Thompson }
91252d6035fSJeremy L Thompson 
91352d6035fSJeremy L Thompson /**
9144db537f9SJeremy L Thompson   @brief Check if a CeedOperator is ready to be used.
9154db537f9SJeremy L Thompson 
9164db537f9SJeremy L Thompson   @param[in] op  CeedOperator to check
9174db537f9SJeremy L Thompson 
9184db537f9SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
9194db537f9SJeremy L Thompson 
9204db537f9SJeremy L Thompson   @ref User
9214db537f9SJeremy L Thompson **/
9224db537f9SJeremy L Thompson int CeedOperatorCheckReady(CeedOperator op) {
9234db537f9SJeremy L Thompson   int ierr;
9244db537f9SJeremy L Thompson   Ceed ceed;
9254db537f9SJeremy L Thompson   ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
9264db537f9SJeremy L Thompson 
9274db537f9SJeremy L Thompson   if (op->is_interface_setup)
9284db537f9SJeremy L Thompson     return CEED_ERROR_SUCCESS;
9294db537f9SJeremy L Thompson 
9304db537f9SJeremy L Thompson   CeedQFunction qf = op->qf;
9314db537f9SJeremy L Thompson   if (op->is_composite) {
9324db537f9SJeremy L Thompson     if (!op->num_suboperators)
9334db537f9SJeremy L Thompson       // LCOV_EXCL_START
9344db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE, "No sub_operators set");
9354db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9364db537f9SJeremy L Thompson     for (CeedInt i = 0; i < op->num_suboperators; i++) {
9374db537f9SJeremy L Thompson       ierr = CeedOperatorCheckReady(op->sub_operators[i]); CeedChk(ierr);
9384db537f9SJeremy L Thompson     }
9394db537f9SJeremy L Thompson   } else {
9404db537f9SJeremy L Thompson     if (op->num_fields == 0)
9414db537f9SJeremy L Thompson       // LCOV_EXCL_START
9424db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE, "No operator fields set");
9434db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9444db537f9SJeremy L Thompson     if (op->num_fields < qf->num_input_fields + qf->num_output_fields)
9454db537f9SJeremy L Thompson       // LCOV_EXCL_START
9464db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE, "Not all operator fields set");
9474db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9484db537f9SJeremy L Thompson     if (!op->has_restriction)
9494db537f9SJeremy L Thompson       // LCOV_EXCL_START
9504db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE,
9514db537f9SJeremy L Thompson                        "At least one restriction required");
9524db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9534db537f9SJeremy L Thompson     if (op->num_qpts == 0)
9544db537f9SJeremy L Thompson       // LCOV_EXCL_START
9554db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE,
9564db537f9SJeremy L Thompson                        "At least one non-collocated basis is required "
9574db537f9SJeremy L Thompson                        "or the number of quadrature points must be set");
9584db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9594db537f9SJeremy L Thompson   }
9604db537f9SJeremy L Thompson 
9614db537f9SJeremy L Thompson   // Flag as immutable and ready
9624db537f9SJeremy L Thompson   op->is_interface_setup = true;
9634db537f9SJeremy L Thompson   if (op->qf && op->qf != CEED_QFUNCTION_NONE)
9644db537f9SJeremy L Thompson     // LCOV_EXCL_START
9654db537f9SJeremy L Thompson     op->qf->is_immutable = true;
9664db537f9SJeremy L Thompson   // LCOV_EXCL_STOP
9674db537f9SJeremy L Thompson   if (op->dqf && op->dqf != CEED_QFUNCTION_NONE)
9684db537f9SJeremy L Thompson     // LCOV_EXCL_START
9694db537f9SJeremy L Thompson     op->dqf->is_immutable = true;
9704db537f9SJeremy L Thompson   // LCOV_EXCL_STOP
9714db537f9SJeremy L Thompson   if (op->dqfT && op->dqfT != CEED_QFUNCTION_NONE)
9724db537f9SJeremy L Thompson     // LCOV_EXCL_START
9734db537f9SJeremy L Thompson     op->dqfT->is_immutable = true;
9744db537f9SJeremy L Thompson   // LCOV_EXCL_STOP
9754db537f9SJeremy L Thompson   return CEED_ERROR_SUCCESS;
9764db537f9SJeremy L Thompson }
9774db537f9SJeremy L Thompson 
9784db537f9SJeremy L Thompson /**
979cd4dfc48Sjeremylt   @brief Set the number of quadrature points associated with a CeedOperator.
980cd4dfc48Sjeremylt            This should be used when creating a CeedOperator where every
981cd4dfc48Sjeremylt            field has a collocated basis. This function cannot be used for
982cd4dfc48Sjeremylt            composite CeedOperators.
983cd4dfc48Sjeremylt 
984cd4dfc48Sjeremylt   @param op        CeedOperator
985cd4dfc48Sjeremylt   @param num_qpts  Number of quadrature points to set
986cd4dfc48Sjeremylt 
987cd4dfc48Sjeremylt   @return An error code: 0 - success, otherwise - failure
988cd4dfc48Sjeremylt 
989e9b533fbSJeremy L Thompson   @ref Advanced
990cd4dfc48Sjeremylt **/
991cd4dfc48Sjeremylt int CeedOperatorSetNumQuadraturePoints(CeedOperator op, CeedInt num_qpts) {
992f04ea552SJeremy L Thompson   if (op->is_composite)
993cd4dfc48Sjeremylt     // LCOV_EXCL_START
994cd4dfc48Sjeremylt     return CeedError(op->ceed, CEED_ERROR_MINOR,
995cd4dfc48Sjeremylt                      "Not defined for composite operator");
996cd4dfc48Sjeremylt   // LCOV_EXCL_STOP
997cd4dfc48Sjeremylt   if (op->num_qpts)
998cd4dfc48Sjeremylt     // LCOV_EXCL_START
999cd4dfc48Sjeremylt     return CeedError(op->ceed, CEED_ERROR_MINOR,
1000cd4dfc48Sjeremylt                      "Number of quadrature points already defined");
1001cd4dfc48Sjeremylt   // LCOV_EXCL_STOP
1002f04ea552SJeremy L Thompson   if (op->is_immutable)
1003f04ea552SJeremy L Thompson     // LCOV_EXCL_START
1004f04ea552SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MAJOR,
1005f04ea552SJeremy L Thompson                      "Operator cannot be changed after set as immutable");
1006f04ea552SJeremy L Thompson   // LCOV_EXCL_STOP
1007cd4dfc48Sjeremylt 
1008cd4dfc48Sjeremylt   op->num_qpts = num_qpts;
1009cd4dfc48Sjeremylt   return CEED_ERROR_SUCCESS;
1010cd4dfc48Sjeremylt }
1011cd4dfc48Sjeremylt 
1012cd4dfc48Sjeremylt /**
10137a982d89SJeremy L. Thompson   @brief View a CeedOperator
10147a982d89SJeremy L. Thompson 
10157a982d89SJeremy L. Thompson   @param[in] op      CeedOperator to view
10167a982d89SJeremy L. Thompson   @param[in] stream  Stream to write; typically stdout/stderr or a file
10177a982d89SJeremy L. Thompson 
10187a982d89SJeremy L. Thompson   @return Error code: 0 - success, otherwise - failure
10197a982d89SJeremy L. Thompson 
10207a982d89SJeremy L. Thompson   @ref User
10217a982d89SJeremy L. Thompson **/
10227a982d89SJeremy L. Thompson int CeedOperatorView(CeedOperator op, FILE *stream) {
10237a982d89SJeremy L. Thompson   int ierr;
10247a982d89SJeremy L. Thompson 
1025f04ea552SJeremy L Thompson   if (op->is_composite) {
10267a982d89SJeremy L. Thompson     fprintf(stream, "Composite CeedOperator\n");
10277a982d89SJeremy L. Thompson 
1028d1d35e2fSjeremylt     for (CeedInt i=0; i<op->num_suboperators; i++) {
10297a982d89SJeremy L. Thompson       fprintf(stream, "  SubOperator [%d]:\n", i);
1030d1d35e2fSjeremylt       ierr = CeedOperatorSingleView(op->sub_operators[i], 1, stream);
10317a982d89SJeremy L. Thompson       CeedChk(ierr);
10327a982d89SJeremy L. Thompson     }
10337a982d89SJeremy L. Thompson   } else {
10347a982d89SJeremy L. Thompson     fprintf(stream, "CeedOperator\n");
10357a982d89SJeremy L. Thompson     ierr = CeedOperatorSingleView(op, 0, stream); CeedChk(ierr);
10367a982d89SJeremy L. Thompson   }
1037e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
10387a982d89SJeremy L. Thompson }
10393bd813ffSjeremylt 
10403bd813ffSjeremylt /**
1041b7c9bbdaSJeremy L Thompson   @brief Get the Ceed associated with a CeedOperator
1042b7c9bbdaSJeremy L Thompson 
1043b7c9bbdaSJeremy L Thompson   @param op         CeedOperator
1044b7c9bbdaSJeremy L Thompson   @param[out] ceed  Variable to store Ceed
1045b7c9bbdaSJeremy L Thompson 
1046b7c9bbdaSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1047b7c9bbdaSJeremy L Thompson 
1048b7c9bbdaSJeremy L Thompson   @ref Advanced
1049b7c9bbdaSJeremy L Thompson **/
1050b7c9bbdaSJeremy L Thompson int CeedOperatorGetCeed(CeedOperator op, Ceed *ceed) {
1051b7c9bbdaSJeremy L Thompson   *ceed = op->ceed;
1052b7c9bbdaSJeremy L Thompson   return CEED_ERROR_SUCCESS;
1053b7c9bbdaSJeremy L Thompson }
1054b7c9bbdaSJeremy L Thompson 
1055b7c9bbdaSJeremy L Thompson /**
1056b7c9bbdaSJeremy L Thompson   @brief Get the number of elements associated with a CeedOperator
1057b7c9bbdaSJeremy L Thompson 
1058b7c9bbdaSJeremy L Thompson   @param op             CeedOperator
1059b7c9bbdaSJeremy L Thompson   @param[out] num_elem  Variable to store number of elements
1060b7c9bbdaSJeremy L Thompson 
1061b7c9bbdaSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1062b7c9bbdaSJeremy L Thompson 
1063b7c9bbdaSJeremy L Thompson   @ref Advanced
1064b7c9bbdaSJeremy L Thompson **/
1065b7c9bbdaSJeremy L Thompson int CeedOperatorGetNumElements(CeedOperator op, CeedInt *num_elem) {
1066b7c9bbdaSJeremy L Thompson   if (op->is_composite)
1067b7c9bbdaSJeremy L Thompson     // LCOV_EXCL_START
1068b7c9bbdaSJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR,
1069b7c9bbdaSJeremy L Thompson                      "Not defined for composite operator");
1070b7c9bbdaSJeremy L Thompson   // LCOV_EXCL_STOP
1071b7c9bbdaSJeremy L Thompson 
1072b7c9bbdaSJeremy L Thompson   *num_elem = op->num_elem;
1073b7c9bbdaSJeremy L Thompson   return CEED_ERROR_SUCCESS;
1074b7c9bbdaSJeremy L Thompson }
1075b7c9bbdaSJeremy L Thompson 
1076b7c9bbdaSJeremy L Thompson /**
1077b7c9bbdaSJeremy L Thompson   @brief Get the number of quadrature points associated with a CeedOperator
1078b7c9bbdaSJeremy L Thompson 
1079b7c9bbdaSJeremy L Thompson   @param op             CeedOperator
1080b7c9bbdaSJeremy L Thompson   @param[out] num_qpts  Variable to store vector number of quadrature points
1081b7c9bbdaSJeremy L Thompson 
1082b7c9bbdaSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1083b7c9bbdaSJeremy L Thompson 
1084b7c9bbdaSJeremy L Thompson   @ref Advanced
1085b7c9bbdaSJeremy L Thompson **/
1086b7c9bbdaSJeremy L Thompson int CeedOperatorGetNumQuadraturePoints(CeedOperator op, CeedInt *num_qpts) {
1087b7c9bbdaSJeremy L Thompson   if (op->is_composite)
1088b7c9bbdaSJeremy L Thompson     // LCOV_EXCL_START
1089b7c9bbdaSJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR,
1090b7c9bbdaSJeremy L Thompson                      "Not defined for composite operator");
1091b7c9bbdaSJeremy L Thompson   // LCOV_EXCL_STOP
1092b7c9bbdaSJeremy L Thompson 
1093b7c9bbdaSJeremy L Thompson   *num_qpts = op->num_qpts;
1094b7c9bbdaSJeremy L Thompson   return CEED_ERROR_SUCCESS;
1095b7c9bbdaSJeremy L Thompson }
1096b7c9bbdaSJeremy L Thompson 
1097b7c9bbdaSJeremy L Thompson /**
10983668ca4bSJeremy L Thompson   @brief Get label for a registered QFunctionContext field, or `NULL` if no
10993668ca4bSJeremy L Thompson            field has been registered with this `field_name`.
11003668ca4bSJeremy L Thompson 
11013668ca4bSJeremy L Thompson   @param[in] op            CeedOperator
11023668ca4bSJeremy L Thompson   @param[in] field_name    Name of field to retrieve label
11033668ca4bSJeremy L Thompson   @param[out] field_label  Variable to field label
11043668ca4bSJeremy L Thompson 
11053668ca4bSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
11063668ca4bSJeremy L Thompson 
11073668ca4bSJeremy L Thompson   @ref User
11083668ca4bSJeremy L Thompson **/
11093668ca4bSJeremy L Thompson int CeedOperatorContextGetFieldLabel(CeedOperator op,
11103668ca4bSJeremy L Thompson                                      const char *field_name,
11113668ca4bSJeremy L Thompson                                      CeedContextFieldLabel *field_label) {
11123668ca4bSJeremy L Thompson   int ierr;
11133668ca4bSJeremy L Thompson 
11143668ca4bSJeremy L Thompson   bool is_composite;
11153668ca4bSJeremy L Thompson   ierr = CeedOperatorIsComposite(op, &is_composite); CeedChk(ierr);
11163668ca4bSJeremy L Thompson   if (is_composite) {
11173668ca4bSJeremy L Thompson     // Check if composite label already created
11183668ca4bSJeremy L Thompson     for (CeedInt i=0; i<op->num_context_labels; i++) {
11193668ca4bSJeremy L Thompson       if (!strcmp(op->context_labels[i]->name, field_name)) {
11203668ca4bSJeremy L Thompson         *field_label = op->context_labels[i];
11213668ca4bSJeremy L Thompson         return CEED_ERROR_SUCCESS;
11223668ca4bSJeremy L Thompson       }
11233668ca4bSJeremy L Thompson     }
11243668ca4bSJeremy L Thompson 
11253668ca4bSJeremy L Thompson     // Create composite label if needed
11263668ca4bSJeremy L Thompson     CeedInt num_sub;
11273668ca4bSJeremy L Thompson     CeedOperator *sub_operators;
11283668ca4bSJeremy L Thompson     CeedContextFieldLabel new_field_label;
11293668ca4bSJeremy L Thompson 
11303668ca4bSJeremy L Thompson     ierr = CeedCalloc(1, &new_field_label); CeedChk(ierr);
11313668ca4bSJeremy L Thompson     ierr = CeedOperatorGetNumSub(op, &num_sub); CeedChk(ierr);
11323668ca4bSJeremy L Thompson     ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr);
11333668ca4bSJeremy L Thompson     ierr = CeedCalloc(num_sub, &new_field_label->sub_labels); CeedChk(ierr);
11343668ca4bSJeremy L Thompson     new_field_label->num_sub_labels = num_sub;
11353668ca4bSJeremy L Thompson 
11363668ca4bSJeremy L Thompson     bool label_found = false;
11373668ca4bSJeremy L Thompson     for (CeedInt i=0; i<num_sub; i++) {
11383668ca4bSJeremy L Thompson       if (sub_operators[i]->qf->ctx) {
11393668ca4bSJeremy L Thompson         CeedContextFieldLabel new_field_label_i;
11403668ca4bSJeremy L Thompson         ierr = CeedQFunctionContextGetFieldLabel(sub_operators[i]->qf->ctx, field_name,
11413668ca4bSJeremy L Thompson                &new_field_label_i); CeedChk(ierr);
11423668ca4bSJeremy L Thompson         if (new_field_label_i) {
11433668ca4bSJeremy L Thompson           label_found = true;
11443668ca4bSJeremy L Thompson           new_field_label->sub_labels[i] = new_field_label_i;
11453668ca4bSJeremy L Thompson           new_field_label->name = new_field_label_i->name;
11463668ca4bSJeremy L Thompson           new_field_label->description = new_field_label_i->description;
11477bfe0f0eSJeremy L Thompson           if (new_field_label->type &&
11487bfe0f0eSJeremy L Thompson               new_field_label->type != new_field_label_i->type) {
11497bfe0f0eSJeremy L Thompson             // LCOV_EXCL_START
11507bfe0f0eSJeremy L Thompson             ierr = CeedFree(&new_field_label); CeedChk(ierr);
11517bfe0f0eSJeremy L Thompson             return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
11527bfe0f0eSJeremy L Thompson                              "Incompatible field types on sub-operator contexts. "
11537bfe0f0eSJeremy L Thompson                              "%s != %s",
11547bfe0f0eSJeremy L Thompson                              CeedContextFieldTypes[new_field_label->type],
11557bfe0f0eSJeremy L Thompson                              CeedContextFieldTypes[new_field_label_i->type]);
11567bfe0f0eSJeremy L Thompson             // LCOV_EXCL_STOP
11577bfe0f0eSJeremy L Thompson           } else {
11587bfe0f0eSJeremy L Thompson             new_field_label->type = new_field_label_i->type;
11597bfe0f0eSJeremy L Thompson           }
11607bfe0f0eSJeremy L Thompson           if (new_field_label->num_values != 0 &&
11617bfe0f0eSJeremy L Thompson               new_field_label->num_values != new_field_label_i->num_values) {
11627bfe0f0eSJeremy L Thompson             // LCOV_EXCL_START
11637bfe0f0eSJeremy L Thompson             ierr = CeedFree(&new_field_label); CeedChk(ierr);
11647bfe0f0eSJeremy L Thompson             return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
11657bfe0f0eSJeremy L Thompson                              "Incompatible field number of values on sub-operator"
11667bfe0f0eSJeremy L Thompson                              " contexts. %ld != %ld",
11677bfe0f0eSJeremy L Thompson                              new_field_label->num_values, new_field_label_i->num_values);
11687bfe0f0eSJeremy L Thompson             // LCOV_EXCL_STOP
11697bfe0f0eSJeremy L Thompson           } else {
11707bfe0f0eSJeremy L Thompson             new_field_label->num_values = new_field_label_i->num_values;
11717bfe0f0eSJeremy L Thompson           }
11723668ca4bSJeremy L Thompson         }
11733668ca4bSJeremy L Thompson       }
11743668ca4bSJeremy L Thompson     }
11753668ca4bSJeremy L Thompson     if (!label_found) {
11763668ca4bSJeremy L Thompson       // LCOV_EXCL_START
1177a48e5f43SJeremy L Thompson       ierr = CeedFree(&new_field_label->sub_labels); CeedChk(ierr);
11783668ca4bSJeremy L Thompson       ierr = CeedFree(&new_field_label); CeedChk(ierr);
11793668ca4bSJeremy L Thompson       *field_label = NULL;
11803668ca4bSJeremy L Thompson       // LCOV_EXCL_STOP
11813668ca4bSJeremy L Thompson     } else {
11823668ca4bSJeremy L Thompson       // Move new composite label to operator
11833668ca4bSJeremy L Thompson       if (op->num_context_labels == 0) {
11843668ca4bSJeremy L Thompson         ierr = CeedCalloc(1, &op->context_labels); CeedChk(ierr);
11853668ca4bSJeremy L Thompson         op->max_context_labels = 1;
11863668ca4bSJeremy L Thompson       } else if (op->num_context_labels == op->max_context_labels) {
11873668ca4bSJeremy L Thompson         ierr = CeedRealloc(2*op->num_context_labels, &op->context_labels);
11883668ca4bSJeremy L Thompson         CeedChk(ierr);
11893668ca4bSJeremy L Thompson         op->max_context_labels *= 2;
11903668ca4bSJeremy L Thompson       }
11913668ca4bSJeremy L Thompson       op->context_labels[op->num_context_labels] = new_field_label;
11923668ca4bSJeremy L Thompson       *field_label = new_field_label;
11933668ca4bSJeremy L Thompson       op->num_context_labels++;
11943668ca4bSJeremy L Thompson     }
11953668ca4bSJeremy L Thompson 
11963668ca4bSJeremy L Thompson     return CEED_ERROR_SUCCESS;
11973668ca4bSJeremy L Thompson   } else {
11983668ca4bSJeremy L Thompson     return CeedQFunctionContextGetFieldLabel(op->qf->ctx, field_name, field_label);
11993668ca4bSJeremy L Thompson   }
12003668ca4bSJeremy L Thompson }
12013668ca4bSJeremy L Thompson 
12023668ca4bSJeremy L Thompson /**
1203d8dd9a91SJeremy L Thompson   @brief Set QFunctionContext field holding a double precision value.
1204d8dd9a91SJeremy L Thompson            For composite operators, the value is set in all
1205d8dd9a91SJeremy L Thompson            sub-operator QFunctionContexts that have a matching `field_name`.
1206d8dd9a91SJeremy L Thompson 
1207d8dd9a91SJeremy L Thompson   @param op          CeedOperator
12083668ca4bSJeremy L Thompson   @param field_label Label of field to register
12097bfe0f0eSJeremy L Thompson   @param values      Values to set
1210d8dd9a91SJeremy L Thompson 
1211d8dd9a91SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1212d8dd9a91SJeremy L Thompson 
1213d8dd9a91SJeremy L Thompson   @ref User
1214d8dd9a91SJeremy L Thompson **/
12153668ca4bSJeremy L Thompson int CeedOperatorContextSetDouble(CeedOperator op,
12163668ca4bSJeremy L Thompson                                  CeedContextFieldLabel field_label,
12177bfe0f0eSJeremy L Thompson                                  double *values) {
12183668ca4bSJeremy L Thompson   return CeedOperatorContextSetGeneric(op, field_label, CEED_CONTEXT_FIELD_DOUBLE,
12197bfe0f0eSJeremy L Thompson                                        values);
1220d8dd9a91SJeremy L Thompson }
1221d8dd9a91SJeremy L Thompson 
1222d8dd9a91SJeremy L Thompson /**
1223d8dd9a91SJeremy L Thompson   @brief Set QFunctionContext field holding an int32 value.
1224d8dd9a91SJeremy L Thompson            For composite operators, the value is set in all
1225d8dd9a91SJeremy L Thompson            sub-operator QFunctionContexts that have a matching `field_name`.
1226d8dd9a91SJeremy L Thompson 
1227d8dd9a91SJeremy L Thompson   @param op          CeedOperator
12283668ca4bSJeremy L Thompson   @param field_label Label of field to set
12297bfe0f0eSJeremy L Thompson   @param values      Values to set
1230d8dd9a91SJeremy L Thompson 
1231d8dd9a91SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1232d8dd9a91SJeremy L Thompson 
1233d8dd9a91SJeremy L Thompson   @ref User
1234d8dd9a91SJeremy L Thompson **/
12353668ca4bSJeremy L Thompson int CeedOperatorContextSetInt32(CeedOperator op,
12363668ca4bSJeremy L Thompson                                 CeedContextFieldLabel field_label,
12377bfe0f0eSJeremy L Thompson                                 int *values) {
12383668ca4bSJeremy L Thompson   return CeedOperatorContextSetGeneric(op, field_label, CEED_CONTEXT_FIELD_INT32,
12397bfe0f0eSJeremy L Thompson                                        values);
1240d8dd9a91SJeremy L Thompson }
1241d8dd9a91SJeremy L Thompson 
1242d8dd9a91SJeremy L Thompson /**
12433bd813ffSjeremylt   @brief Apply CeedOperator to a vector
1244d7b241e6Sjeremylt 
1245d7b241e6Sjeremylt   This computes the action of the operator on the specified (active) input,
1246d7b241e6Sjeremylt   yielding its (active) output.  All inputs and outputs must be specified using
1247d7b241e6Sjeremylt   CeedOperatorSetField().
1248d7b241e6Sjeremylt 
1249f04ea552SJeremy L Thompson   Note: Calling this function asserts that setup is complete
1250f04ea552SJeremy L Thompson           and sets the CeedOperator as immutable.
1251f04ea552SJeremy L Thompson 
1252d7b241e6Sjeremylt   @param op        CeedOperator to apply
12534cc79fe7SJed Brown   @param[in] in    CeedVector containing input state or @ref CEED_VECTOR_NONE if
125434138859Sjeremylt                      there are no active inputs
1255b11c1e72Sjeremylt   @param[out] out  CeedVector to store result of applying operator (must be
12564cc79fe7SJed Brown                      distinct from @a in) or @ref CEED_VECTOR_NONE if there are no
125734138859Sjeremylt                      active outputs
1258d7b241e6Sjeremylt   @param request   Address of CeedRequest for non-blocking completion, else
12594cc79fe7SJed Brown                      @ref CEED_REQUEST_IMMEDIATE
1260b11c1e72Sjeremylt 
1261b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
1262dfdf5a53Sjeremylt 
12637a982d89SJeremy L. Thompson   @ref User
1264b11c1e72Sjeremylt **/
1265692c2638Sjeremylt int CeedOperatorApply(CeedOperator op, CeedVector in, CeedVector out,
1266692c2638Sjeremylt                       CeedRequest *request) {
1267d7b241e6Sjeremylt   int ierr;
1268e2f04181SAndrew T. Barker   ierr = CeedOperatorCheckReady(op); CeedChk(ierr);
1269d7b241e6Sjeremylt 
1270d1d35e2fSjeremylt   if (op->num_elem)  {
1271250756a7Sjeremylt     // Standard Operator
1272cae8b89aSjeremylt     if (op->Apply) {
1273250756a7Sjeremylt       ierr = op->Apply(op, in, out, request); CeedChk(ierr);
1274cae8b89aSjeremylt     } else {
1275cae8b89aSjeremylt       // Zero all output vectors
1276250756a7Sjeremylt       CeedQFunction qf = op->qf;
1277d1d35e2fSjeremylt       for (CeedInt i=0; i<qf->num_output_fields; i++) {
1278d1d35e2fSjeremylt         CeedVector vec = op->output_fields[i]->vec;
1279cae8b89aSjeremylt         if (vec == CEED_VECTOR_ACTIVE)
1280cae8b89aSjeremylt           vec = out;
1281cae8b89aSjeremylt         if (vec != CEED_VECTOR_NONE) {
1282cae8b89aSjeremylt           ierr = CeedVectorSetValue(vec, 0.0); CeedChk(ierr);
1283cae8b89aSjeremylt         }
1284cae8b89aSjeremylt       }
1285250756a7Sjeremylt       // Apply
1286250756a7Sjeremylt       ierr = op->ApplyAdd(op, in, out, request); CeedChk(ierr);
1287250756a7Sjeremylt     }
1288f04ea552SJeremy L Thompson   } else if (op->is_composite) {
1289250756a7Sjeremylt     // Composite Operator
1290250756a7Sjeremylt     if (op->ApplyComposite) {
1291250756a7Sjeremylt       ierr = op->ApplyComposite(op, in, out, request); CeedChk(ierr);
1292250756a7Sjeremylt     } else {
1293d1d35e2fSjeremylt       CeedInt num_suboperators;
1294d1d35e2fSjeremylt       ierr = CeedOperatorGetNumSub(op, &num_suboperators); CeedChk(ierr);
1295d1d35e2fSjeremylt       CeedOperator *sub_operators;
1296d1d35e2fSjeremylt       ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr);
1297250756a7Sjeremylt 
1298250756a7Sjeremylt       // Zero all output vectors
1299250756a7Sjeremylt       if (out != CEED_VECTOR_NONE) {
1300cae8b89aSjeremylt         ierr = CeedVectorSetValue(out, 0.0); CeedChk(ierr);
1301cae8b89aSjeremylt       }
1302d1d35e2fSjeremylt       for (CeedInt i=0; i<num_suboperators; i++) {
1303d1d35e2fSjeremylt         for (CeedInt j=0; j<sub_operators[i]->qf->num_output_fields; j++) {
1304d1d35e2fSjeremylt           CeedVector vec = sub_operators[i]->output_fields[j]->vec;
1305250756a7Sjeremylt           if (vec != CEED_VECTOR_ACTIVE && vec != CEED_VECTOR_NONE) {
1306250756a7Sjeremylt             ierr = CeedVectorSetValue(vec, 0.0); CeedChk(ierr);
1307250756a7Sjeremylt           }
1308250756a7Sjeremylt         }
1309250756a7Sjeremylt       }
1310250756a7Sjeremylt       // Apply
1311d1d35e2fSjeremylt       for (CeedInt i=0; i<op->num_suboperators; i++) {
1312d1d35e2fSjeremylt         ierr = CeedOperatorApplyAdd(op->sub_operators[i], in, out, request);
1313cae8b89aSjeremylt         CeedChk(ierr);
1314cae8b89aSjeremylt       }
1315cae8b89aSjeremylt     }
1316250756a7Sjeremylt   }
1317e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1318cae8b89aSjeremylt }
1319cae8b89aSjeremylt 
1320cae8b89aSjeremylt /**
1321cae8b89aSjeremylt   @brief Apply CeedOperator to a vector and add result to output vector
1322cae8b89aSjeremylt 
1323cae8b89aSjeremylt   This computes the action of the operator on the specified (active) input,
1324cae8b89aSjeremylt   yielding its (active) output.  All inputs and outputs must be specified using
1325cae8b89aSjeremylt   CeedOperatorSetField().
1326cae8b89aSjeremylt 
1327cae8b89aSjeremylt   @param op        CeedOperator to apply
1328cae8b89aSjeremylt   @param[in] in    CeedVector containing input state or NULL if there are no
1329cae8b89aSjeremylt                      active inputs
1330cae8b89aSjeremylt   @param[out] out  CeedVector to sum in result of applying operator (must be
1331cae8b89aSjeremylt                      distinct from @a in) or NULL if there are no active outputs
1332cae8b89aSjeremylt   @param request   Address of CeedRequest for non-blocking completion, else
13334cc79fe7SJed Brown                      @ref CEED_REQUEST_IMMEDIATE
1334cae8b89aSjeremylt 
1335cae8b89aSjeremylt   @return An error code: 0 - success, otherwise - failure
1336cae8b89aSjeremylt 
13377a982d89SJeremy L. Thompson   @ref User
1338cae8b89aSjeremylt **/
1339cae8b89aSjeremylt int CeedOperatorApplyAdd(CeedOperator op, CeedVector in, CeedVector out,
1340cae8b89aSjeremylt                          CeedRequest *request) {
1341cae8b89aSjeremylt   int ierr;
1342e2f04181SAndrew T. Barker   ierr = CeedOperatorCheckReady(op); CeedChk(ierr);
1343cae8b89aSjeremylt 
1344d1d35e2fSjeremylt   if (op->num_elem)  {
1345250756a7Sjeremylt     // Standard Operator
1346250756a7Sjeremylt     ierr = op->ApplyAdd(op, in, out, request); CeedChk(ierr);
1347f04ea552SJeremy L Thompson   } else if (op->is_composite) {
1348250756a7Sjeremylt     // Composite Operator
1349250756a7Sjeremylt     if (op->ApplyAddComposite) {
1350250756a7Sjeremylt       ierr = op->ApplyAddComposite(op, in, out, request); CeedChk(ierr);
1351cae8b89aSjeremylt     } else {
1352d1d35e2fSjeremylt       CeedInt num_suboperators;
1353d1d35e2fSjeremylt       ierr = CeedOperatorGetNumSub(op, &num_suboperators); CeedChk(ierr);
1354d1d35e2fSjeremylt       CeedOperator *sub_operators;
1355d1d35e2fSjeremylt       ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr);
1356250756a7Sjeremylt 
1357d1d35e2fSjeremylt       for (CeedInt i=0; i<num_suboperators; i++) {
1358d1d35e2fSjeremylt         ierr = CeedOperatorApplyAdd(sub_operators[i], in, out, request);
1359cae8b89aSjeremylt         CeedChk(ierr);
13601d7d2407SJeremy L Thompson       }
1361250756a7Sjeremylt     }
1362250756a7Sjeremylt   }
1363e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1364d7b241e6Sjeremylt }
1365d7b241e6Sjeremylt 
1366d7b241e6Sjeremylt /**
1367b11c1e72Sjeremylt   @brief Destroy a CeedOperator
1368d7b241e6Sjeremylt 
1369d7b241e6Sjeremylt   @param op  CeedOperator to destroy
1370b11c1e72Sjeremylt 
1371b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
1372dfdf5a53Sjeremylt 
13737a982d89SJeremy L. Thompson   @ref User
1374b11c1e72Sjeremylt **/
1375d7b241e6Sjeremylt int CeedOperatorDestroy(CeedOperator *op) {
1376d7b241e6Sjeremylt   int ierr;
1377d7b241e6Sjeremylt 
1378d1d35e2fSjeremylt   if (!*op || --(*op)->ref_count > 0) return CEED_ERROR_SUCCESS;
1379d7b241e6Sjeremylt   if ((*op)->Destroy) {
1380d7b241e6Sjeremylt     ierr = (*op)->Destroy(*op); CeedChk(ierr);
1381d7b241e6Sjeremylt   }
1382fe2413ffSjeremylt   ierr = CeedDestroy(&(*op)->ceed); CeedChk(ierr);
1383fe2413ffSjeremylt   // Free fields
13843668ca4bSJeremy L Thompson   for (CeedInt i=0; i<(*op)->num_fields; i++)
1385d1d35e2fSjeremylt     if ((*op)->input_fields[i]) {
1386d1d35e2fSjeremylt       if ((*op)->input_fields[i]->elem_restr != CEED_ELEMRESTRICTION_NONE) {
1387d1d35e2fSjeremylt         ierr = CeedElemRestrictionDestroy(&(*op)->input_fields[i]->elem_restr);
138871352a93Sjeremylt         CeedChk(ierr);
138915910d16Sjeremylt       }
1390d1d35e2fSjeremylt       if ((*op)->input_fields[i]->basis != CEED_BASIS_COLLOCATED) {
1391d1d35e2fSjeremylt         ierr = CeedBasisDestroy(&(*op)->input_fields[i]->basis); CeedChk(ierr);
139271352a93Sjeremylt       }
1393d1d35e2fSjeremylt       if ((*op)->input_fields[i]->vec != CEED_VECTOR_ACTIVE &&
1394d1d35e2fSjeremylt           (*op)->input_fields[i]->vec != CEED_VECTOR_NONE ) {
1395d1d35e2fSjeremylt         ierr = CeedVectorDestroy(&(*op)->input_fields[i]->vec); CeedChk(ierr);
139671352a93Sjeremylt       }
1397d1d35e2fSjeremylt       ierr = CeedFree(&(*op)->input_fields[i]->field_name); CeedChk(ierr);
1398d1d35e2fSjeremylt       ierr = CeedFree(&(*op)->input_fields[i]); CeedChk(ierr);
1399fe2413ffSjeremylt     }
14003668ca4bSJeremy L Thompson   for (CeedInt i=0; i<(*op)->num_fields; i++)
1401d1d35e2fSjeremylt     if ((*op)->output_fields[i]) {
1402d1d35e2fSjeremylt       ierr = CeedElemRestrictionDestroy(&(*op)->output_fields[i]->elem_restr);
140371352a93Sjeremylt       CeedChk(ierr);
1404d1d35e2fSjeremylt       if ((*op)->output_fields[i]->basis != CEED_BASIS_COLLOCATED) {
1405d1d35e2fSjeremylt         ierr = CeedBasisDestroy(&(*op)->output_fields[i]->basis); CeedChk(ierr);
140671352a93Sjeremylt       }
1407d1d35e2fSjeremylt       if ((*op)->output_fields[i]->vec != CEED_VECTOR_ACTIVE &&
1408d1d35e2fSjeremylt           (*op)->output_fields[i]->vec != CEED_VECTOR_NONE ) {
1409d1d35e2fSjeremylt         ierr = CeedVectorDestroy(&(*op)->output_fields[i]->vec); CeedChk(ierr);
141071352a93Sjeremylt       }
1411d1d35e2fSjeremylt       ierr = CeedFree(&(*op)->output_fields[i]->field_name); CeedChk(ierr);
1412d1d35e2fSjeremylt       ierr = CeedFree(&(*op)->output_fields[i]); CeedChk(ierr);
1413fe2413ffSjeremylt     }
1414d1d35e2fSjeremylt   // Destroy sub_operators
14153668ca4bSJeremy L Thompson   for (CeedInt i=0; i<(*op)->num_suboperators; i++)
1416d1d35e2fSjeremylt     if ((*op)->sub_operators[i]) {
1417d1d35e2fSjeremylt       ierr = CeedOperatorDestroy(&(*op)->sub_operators[i]); CeedChk(ierr);
141852d6035fSJeremy L Thompson     }
1419d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->qf); CeedChk(ierr);
1420d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->dqf); CeedChk(ierr);
1421d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->dqfT); CeedChk(ierr);
14223668ca4bSJeremy L Thompson   // Destroy any composite labels
14233668ca4bSJeremy L Thompson   for (CeedInt i=0; i<(*op)->num_context_labels; i++) {
14243668ca4bSJeremy L Thompson     ierr = CeedFree(&(*op)->context_labels[i]->sub_labels); CeedChk(ierr);
14253668ca4bSJeremy L Thompson     ierr = CeedFree(&(*op)->context_labels[i]); CeedChk(ierr);
14263668ca4bSJeremy L Thompson   }
14273668ca4bSJeremy L Thompson   ierr = CeedFree(&(*op)->context_labels); CeedChk(ierr);
1428fe2413ffSjeremylt 
14295107b09fSJeremy L Thompson   // Destroy fallback
1430d1d35e2fSjeremylt   if ((*op)->op_fallback) {
1431d1d35e2fSjeremylt     ierr = (*op)->qf_fallback->Destroy((*op)->qf_fallback); CeedChk(ierr);
1432d1d35e2fSjeremylt     ierr = CeedFree(&(*op)->qf_fallback); CeedChk(ierr);
1433d1d35e2fSjeremylt     ierr = (*op)->op_fallback->Destroy((*op)->op_fallback); CeedChk(ierr);
1434d1d35e2fSjeremylt     ierr = CeedFree(&(*op)->op_fallback); CeedChk(ierr);
14355107b09fSJeremy L Thompson   }
14365107b09fSJeremy L Thompson 
143770a7ffb3SJeremy L Thompson   // Destroy QF assembly cache
1438*480fae85SJeremy L Thompson   ierr = CeedQFunctionAssemblyDataDestroy(&(*op)->qf_assembled); CeedChk(ierr);
143970a7ffb3SJeremy L Thompson 
1440d1d35e2fSjeremylt   ierr = CeedFree(&(*op)->input_fields); CeedChk(ierr);
1441d1d35e2fSjeremylt   ierr = CeedFree(&(*op)->output_fields); CeedChk(ierr);
1442d1d35e2fSjeremylt   ierr = CeedFree(&(*op)->sub_operators); CeedChk(ierr);
1443d7b241e6Sjeremylt   ierr = CeedFree(op); CeedChk(ierr);
1444e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1445d7b241e6Sjeremylt }
1446d7b241e6Sjeremylt 
1447d7b241e6Sjeremylt /// @}
1448