xref: /libCEED/rust/libceed-sys/c-src/interface/ceed-operator.c (revision a48e5f439fbaa8d62ee0101c527193505fde197d)
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   }
582bf4cb664SJeremy L Thompson   ierr = CeedCalloc(CEED_FIELD_MAX, &(*op)->input_fields); CeedChk(ierr);
583bf4cb664SJeremy L Thompson   ierr = CeedCalloc(CEED_FIELD_MAX, &(*op)->output_fields); CeedChk(ierr);
584d7b241e6Sjeremylt   ierr = ceed->OperatorCreate(*op); CeedChk(ierr);
585e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
586d7b241e6Sjeremylt }
587d7b241e6Sjeremylt 
588d7b241e6Sjeremylt /**
58952d6035fSJeremy L Thompson   @brief Create an operator that composes the action of several operators
59052d6035fSJeremy L Thompson 
59152d6035fSJeremy L Thompson   @param ceed     A Ceed object where the CeedOperator will be created
59252d6035fSJeremy L Thompson   @param[out] op  Address of the variable where the newly created
59352d6035fSJeremy L Thompson                     Composite CeedOperator will be stored
59452d6035fSJeremy L Thompson 
59552d6035fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
59652d6035fSJeremy L Thompson 
5977a982d89SJeremy L. Thompson   @ref User
59852d6035fSJeremy L Thompson  */
59952d6035fSJeremy L Thompson int CeedCompositeOperatorCreate(Ceed ceed, CeedOperator *op) {
60052d6035fSJeremy L Thompson   int ierr;
60152d6035fSJeremy L Thompson 
60252d6035fSJeremy L Thompson   if (!ceed->CompositeOperatorCreate) {
60352d6035fSJeremy L Thompson     Ceed delegate;
604aefd8378Sjeremylt     ierr = CeedGetObjectDelegate(ceed, &delegate, "Operator"); CeedChk(ierr);
60552d6035fSJeremy L Thompson 
606250756a7Sjeremylt     if (delegate) {
60752d6035fSJeremy L Thompson       ierr = CeedCompositeOperatorCreate(delegate, op); CeedChk(ierr);
608e15f9bd0SJeremy L Thompson       return CEED_ERROR_SUCCESS;
60952d6035fSJeremy L Thompson     }
610250756a7Sjeremylt   }
61152d6035fSJeremy L Thompson 
61252d6035fSJeremy L Thompson   ierr = CeedCalloc(1, op); CeedChk(ierr);
61352d6035fSJeremy L Thompson   (*op)->ceed = ceed;
6149560d06aSjeremylt   ierr = CeedReference(ceed); CeedChk(ierr);
615f04ea552SJeremy L Thompson   (*op)->is_composite = true;
616bf4cb664SJeremy L Thompson   ierr = CeedCalloc(CEED_COMPOSITE_MAX, &(*op)->sub_operators); CeedChk(ierr);
617250756a7Sjeremylt 
618250756a7Sjeremylt   if (ceed->CompositeOperatorCreate) {
61952d6035fSJeremy L Thompson     ierr = ceed->CompositeOperatorCreate(*op); CeedChk(ierr);
620250756a7Sjeremylt   }
621e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
62252d6035fSJeremy L Thompson }
62352d6035fSJeremy L Thompson 
62452d6035fSJeremy L Thompson /**
6259560d06aSjeremylt   @brief Copy the pointer to a CeedOperator. Both pointers should
6269560d06aSjeremylt            be destroyed with `CeedOperatorDestroy()`;
6279560d06aSjeremylt            Note: If `*op_copy` is non-NULL, then it is assumed that
6289560d06aSjeremylt            `*op_copy` is a pointer to a CeedOperator. This
6299560d06aSjeremylt            CeedOperator will be destroyed if `*op_copy` is the only
6309560d06aSjeremylt            reference to this CeedOperator.
6319560d06aSjeremylt 
6329560d06aSjeremylt   @param op            CeedOperator to copy reference to
6339560d06aSjeremylt   @param[out] op_copy  Variable to store copied reference
6349560d06aSjeremylt 
6359560d06aSjeremylt   @return An error code: 0 - success, otherwise - failure
6369560d06aSjeremylt 
6379560d06aSjeremylt   @ref User
6389560d06aSjeremylt **/
6399560d06aSjeremylt int CeedOperatorReferenceCopy(CeedOperator op, CeedOperator *op_copy) {
6409560d06aSjeremylt   int ierr;
6419560d06aSjeremylt 
6429560d06aSjeremylt   ierr = CeedOperatorReference(op); CeedChk(ierr);
6439560d06aSjeremylt   ierr = CeedOperatorDestroy(op_copy); CeedChk(ierr);
6449560d06aSjeremylt   *op_copy = op;
6459560d06aSjeremylt   return CEED_ERROR_SUCCESS;
6469560d06aSjeremylt }
6479560d06aSjeremylt 
6489560d06aSjeremylt /**
649b11c1e72Sjeremylt   @brief Provide a field to a CeedOperator for use by its CeedQFunction
650d7b241e6Sjeremylt 
651d7b241e6Sjeremylt   This function is used to specify both active and passive fields to a
652d7b241e6Sjeremylt   CeedOperator.  For passive fields, a vector @arg v must be provided.  Passive
653d7b241e6Sjeremylt   fields can inputs or outputs (updated in-place when operator is applied).
654d7b241e6Sjeremylt 
655d7b241e6Sjeremylt   Active fields must be specified using this function, but their data (in a
656d7b241e6Sjeremylt   CeedVector) is passed in CeedOperatorApply().  There can be at most one active
657d7b241e6Sjeremylt   input and at most one active output.
658d7b241e6Sjeremylt 
6598c91a0c9SJeremy L Thompson   @param op          CeedOperator on which to provide the field
660d1d35e2fSjeremylt   @param field_name  Name of the field (to be matched with the name used by
6618795c945Sjeremylt                        CeedQFunction)
662b11c1e72Sjeremylt   @param r           CeedElemRestriction
6634cc79fe7SJed Brown   @param b           CeedBasis in which the field resides or @ref CEED_BASIS_COLLOCATED
664b11c1e72Sjeremylt                        if collocated with quadrature points
6654cc79fe7SJed Brown   @param v           CeedVector to be used by CeedOperator or @ref CEED_VECTOR_ACTIVE
6664cc79fe7SJed Brown                        if field is active or @ref CEED_VECTOR_NONE if using
6674cc79fe7SJed Brown                        @ref CEED_EVAL_WEIGHT in the QFunction
668b11c1e72Sjeremylt 
669b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
670dfdf5a53Sjeremylt 
6717a982d89SJeremy L. Thompson   @ref User
672b11c1e72Sjeremylt **/
673d1d35e2fSjeremylt int CeedOperatorSetField(CeedOperator op, const char *field_name,
674a8d32208Sjeremylt                          CeedElemRestriction r, CeedBasis b, CeedVector v) {
675d7b241e6Sjeremylt   int ierr;
676f04ea552SJeremy L Thompson   if (op->is_composite)
677c042f62fSJeremy L Thompson     // LCOV_EXCL_START
678e1e9e29dSJed Brown     return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
679e15f9bd0SJeremy L Thompson                      "Cannot add field to composite operator.");
680c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
681f04ea552SJeremy L Thompson   if (op->is_immutable)
682f04ea552SJeremy L Thompson     // LCOV_EXCL_START
683f04ea552SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MAJOR,
684f04ea552SJeremy L Thompson                      "Operator cannot be changed after set as immutable");
685f04ea552SJeremy L Thompson   // LCOV_EXCL_STOP
6868b067b84SJed Brown   if (!r)
687c042f62fSJeremy L Thompson     // LCOV_EXCL_START
688e1e9e29dSJed Brown     return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
689c042f62fSJeremy L Thompson                      "ElemRestriction r for field \"%s\" must be non-NULL.",
690d1d35e2fSjeremylt                      field_name);
691c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
6928b067b84SJed Brown   if (!b)
693c042f62fSJeremy L Thompson     // LCOV_EXCL_START
694e1e9e29dSJed Brown     return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
695e15f9bd0SJeremy L Thompson                      "Basis b for field \"%s\" must be non-NULL.",
696d1d35e2fSjeremylt                      field_name);
697c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
6988b067b84SJed Brown   if (!v)
699c042f62fSJeremy L Thompson     // LCOV_EXCL_START
700e1e9e29dSJed Brown     return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
701e15f9bd0SJeremy L Thompson                      "Vector v for field \"%s\" must be non-NULL.",
702d1d35e2fSjeremylt                      field_name);
703c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
70452d6035fSJeremy L Thompson 
705d1d35e2fSjeremylt   CeedInt num_elem;
706d1d35e2fSjeremylt   ierr = CeedElemRestrictionGetNumElements(r, &num_elem); CeedChk(ierr);
707d1d35e2fSjeremylt   if (r != CEED_ELEMRESTRICTION_NONE && op->has_restriction &&
708d1d35e2fSjeremylt       op->num_elem != num_elem)
709c042f62fSJeremy L Thompson     // LCOV_EXCL_START
710e15f9bd0SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_DIMENSION,
7111d102b48SJeremy L Thompson                      "ElemRestriction with %d elements incompatible with prior "
712d1d35e2fSjeremylt                      "%d elements", num_elem, op->num_elem);
713c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
714d7b241e6Sjeremylt 
71578464608Sjeremylt   CeedInt num_qpts = 0;
716e15f9bd0SJeremy L Thompson   if (b != CEED_BASIS_COLLOCATED) {
717d1d35e2fSjeremylt     ierr = CeedBasisGetNumQuadraturePoints(b, &num_qpts); CeedChk(ierr);
718d1d35e2fSjeremylt     if (op->num_qpts && op->num_qpts != num_qpts)
719c042f62fSJeremy L Thompson       // LCOV_EXCL_START
720e15f9bd0SJeremy L Thompson       return CeedError(op->ceed, CEED_ERROR_DIMENSION,
721e15f9bd0SJeremy L Thompson                        "Basis with %d quadrature points "
722d1d35e2fSjeremylt                        "incompatible with prior %d points", num_qpts,
723d1d35e2fSjeremylt                        op->num_qpts);
724c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
725d7b241e6Sjeremylt   }
726d1d35e2fSjeremylt   CeedQFunctionField qf_field;
727d1d35e2fSjeremylt   CeedOperatorField *op_field;
728d1d35e2fSjeremylt   for (CeedInt i=0; i<op->qf->num_input_fields; i++) {
729d1d35e2fSjeremylt     if (!strcmp(field_name, (*op->qf->input_fields[i]).field_name)) {
730d1d35e2fSjeremylt       qf_field = op->qf->input_fields[i];
731d1d35e2fSjeremylt       op_field = &op->input_fields[i];
732d7b241e6Sjeremylt       goto found;
733d7b241e6Sjeremylt     }
734d7b241e6Sjeremylt   }
735d1d35e2fSjeremylt   for (CeedInt i=0; i<op->qf->num_output_fields; i++) {
736d1d35e2fSjeremylt     if (!strcmp(field_name, (*op->qf->output_fields[i]).field_name)) {
737d1d35e2fSjeremylt       qf_field = op->qf->output_fields[i];
738d1d35e2fSjeremylt       op_field = &op->output_fields[i];
739d7b241e6Sjeremylt       goto found;
740d7b241e6Sjeremylt     }
741d7b241e6Sjeremylt   }
742c042f62fSJeremy L Thompson   // LCOV_EXCL_START
743e15f9bd0SJeremy L Thompson   return CeedError(op->ceed, CEED_ERROR_INCOMPLETE,
744e15f9bd0SJeremy L Thompson                    "QFunction has no knowledge of field '%s'",
745d1d35e2fSjeremylt                    field_name);
746c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
747d7b241e6Sjeremylt found:
748d1d35e2fSjeremylt   ierr = CeedOperatorCheckField(op->ceed, qf_field, r, b); CeedChk(ierr);
749d1d35e2fSjeremylt   ierr = CeedCalloc(1, op_field); CeedChk(ierr);
750e15f9bd0SJeremy L Thompson 
751d1d35e2fSjeremylt   (*op_field)->vec = v;
752e15f9bd0SJeremy L Thompson   if (v != CEED_VECTOR_ACTIVE && v != CEED_VECTOR_NONE) {
7539560d06aSjeremylt     ierr = CeedVectorReference(v); CeedChk(ierr);
754e15f9bd0SJeremy L Thompson   }
755e15f9bd0SJeremy L Thompson 
756d1d35e2fSjeremylt   (*op_field)->elem_restr = r;
7579560d06aSjeremylt   ierr = CeedElemRestrictionReference(r); CeedChk(ierr);
758e15f9bd0SJeremy L Thompson   if (r != CEED_ELEMRESTRICTION_NONE) {
759d1d35e2fSjeremylt     op->num_elem = num_elem;
760d1d35e2fSjeremylt     op->has_restriction = true; // Restriction set, but num_elem may be 0
761e15f9bd0SJeremy L Thompson   }
762d99fa3c5SJeremy L Thompson 
763d1d35e2fSjeremylt   (*op_field)->basis = b;
764e15f9bd0SJeremy L Thompson   if (b != CEED_BASIS_COLLOCATED) {
765cd4dfc48Sjeremylt     if (!op->num_qpts) {
766cd4dfc48Sjeremylt       ierr = CeedOperatorSetNumQuadraturePoints(op, num_qpts); CeedChk(ierr);
767cd4dfc48Sjeremylt     }
7689560d06aSjeremylt     ierr = CeedBasisReference(b); CeedChk(ierr);
769e15f9bd0SJeremy L Thompson   }
770e15f9bd0SJeremy L Thompson 
771d1d35e2fSjeremylt   op->num_fields += 1;
772f7e22acaSJeremy L Thompson   ierr = CeedStringAllocCopy(field_name, (char **)&(*op_field)->field_name);
773f7e22acaSJeremy L Thompson   CeedChk(ierr);
774e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
775d7b241e6Sjeremylt }
776d7b241e6Sjeremylt 
777d7b241e6Sjeremylt /**
77843bbe138SJeremy L Thompson   @brief Get the CeedOperatorFields of a CeedOperator
77943bbe138SJeremy L Thompson 
780f04ea552SJeremy L Thompson   Note: Calling this function asserts that setup is complete
781f04ea552SJeremy L Thompson           and sets the CeedOperator as immutable.
782f04ea552SJeremy L Thompson 
78343bbe138SJeremy L Thompson   @param op                      CeedOperator
784f74ec584SJeremy L Thompson   @param[out] num_input_fields   Variable to store number of input fields
78543bbe138SJeremy L Thompson   @param[out] input_fields       Variable to store input_fields
786f74ec584SJeremy L Thompson   @param[out] num_output_fields  Variable to store number of output fields
78743bbe138SJeremy L Thompson   @param[out] output_fields      Variable to store output_fields
78843bbe138SJeremy L Thompson 
78943bbe138SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
79043bbe138SJeremy L Thompson 
791e9b533fbSJeremy L Thompson   @ref Advanced
79243bbe138SJeremy L Thompson **/
79343bbe138SJeremy L Thompson int CeedOperatorGetFields(CeedOperator op, CeedInt *num_input_fields,
79443bbe138SJeremy L Thompson                           CeedOperatorField **input_fields,
79543bbe138SJeremy L Thompson                           CeedInt *num_output_fields,
79643bbe138SJeremy L Thompson                           CeedOperatorField **output_fields) {
797f04ea552SJeremy L Thompson   int ierr;
798f04ea552SJeremy L Thompson 
799f04ea552SJeremy L Thompson   if (op->is_composite)
80043bbe138SJeremy L Thompson     // LCOV_EXCL_START
80143bbe138SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR,
80243bbe138SJeremy L Thompson                      "Not defined for composite operator");
80343bbe138SJeremy L Thompson   // LCOV_EXCL_STOP
804f04ea552SJeremy L Thompson   ierr = CeedOperatorCheckReady(op); CeedChk(ierr);
80543bbe138SJeremy L Thompson 
80643bbe138SJeremy L Thompson   if (num_input_fields) *num_input_fields = op->qf->num_input_fields;
80743bbe138SJeremy L Thompson   if (input_fields) *input_fields = op->input_fields;
80843bbe138SJeremy L Thompson   if (num_output_fields) *num_output_fields = op->qf->num_output_fields;
80943bbe138SJeremy L Thompson   if (output_fields) *output_fields = op->output_fields;
81043bbe138SJeremy L Thompson   return CEED_ERROR_SUCCESS;
81143bbe138SJeremy L Thompson }
81243bbe138SJeremy L Thompson 
81343bbe138SJeremy L Thompson /**
81428567f8fSJeremy L Thompson   @brief Get the name of a CeedOperatorField
81528567f8fSJeremy L Thompson 
81628567f8fSJeremy L Thompson   @param op_field         CeedOperatorField
81728567f8fSJeremy L Thompson   @param[out] field_name  Variable to store the field name
81828567f8fSJeremy L Thompson 
81928567f8fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
82028567f8fSJeremy L Thompson 
821e9b533fbSJeremy L Thompson   @ref Advanced
82228567f8fSJeremy L Thompson **/
82328567f8fSJeremy L Thompson int CeedOperatorFieldGetName(CeedOperatorField op_field, char **field_name) {
82428567f8fSJeremy L Thompson   *field_name = (char *)op_field->field_name;
82528567f8fSJeremy L Thompson   return CEED_ERROR_SUCCESS;
82628567f8fSJeremy L Thompson }
82728567f8fSJeremy L Thompson 
82828567f8fSJeremy L Thompson /**
82943bbe138SJeremy L Thompson   @brief Get the CeedElemRestriction of a CeedOperatorField
83043bbe138SJeremy L Thompson 
83143bbe138SJeremy L Thompson   @param op_field   CeedOperatorField
83243bbe138SJeremy L Thompson   @param[out] rstr  Variable to store CeedElemRestriction
83343bbe138SJeremy L Thompson 
83443bbe138SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
83543bbe138SJeremy L Thompson 
836e9b533fbSJeremy L Thompson   @ref Advanced
83743bbe138SJeremy L Thompson **/
83843bbe138SJeremy L Thompson int CeedOperatorFieldGetElemRestriction(CeedOperatorField op_field,
83943bbe138SJeremy L Thompson                                         CeedElemRestriction *rstr) {
84043bbe138SJeremy L Thompson   *rstr = op_field->elem_restr;
84143bbe138SJeremy L Thompson   return CEED_ERROR_SUCCESS;
84243bbe138SJeremy L Thompson }
84343bbe138SJeremy L Thompson 
84443bbe138SJeremy L Thompson /**
84543bbe138SJeremy L Thompson   @brief Get the CeedBasis of a CeedOperatorField
84643bbe138SJeremy L Thompson 
84743bbe138SJeremy L Thompson   @param op_field    CeedOperatorField
84843bbe138SJeremy L Thompson   @param[out] basis  Variable to store CeedBasis
84943bbe138SJeremy L Thompson 
85043bbe138SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
85143bbe138SJeremy L Thompson 
852e9b533fbSJeremy L Thompson   @ref Advanced
85343bbe138SJeremy L Thompson **/
85443bbe138SJeremy L Thompson int CeedOperatorFieldGetBasis(CeedOperatorField op_field, CeedBasis *basis) {
85543bbe138SJeremy L Thompson   *basis = op_field->basis;
85643bbe138SJeremy L Thompson   return CEED_ERROR_SUCCESS;
85743bbe138SJeremy L Thompson }
85843bbe138SJeremy L Thompson 
85943bbe138SJeremy L Thompson /**
86043bbe138SJeremy L Thompson   @brief Get the CeedVector of a CeedOperatorField
86143bbe138SJeremy L Thompson 
86243bbe138SJeremy L Thompson   @param op_field  CeedOperatorField
86343bbe138SJeremy L Thompson   @param[out] vec  Variable to store CeedVector
86443bbe138SJeremy L Thompson 
86543bbe138SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
86643bbe138SJeremy L Thompson 
867e9b533fbSJeremy L Thompson   @ref Advanced
86843bbe138SJeremy L Thompson **/
86943bbe138SJeremy L Thompson int CeedOperatorFieldGetVector(CeedOperatorField op_field, CeedVector *vec) {
87043bbe138SJeremy L Thompson   *vec = op_field->vec;
87143bbe138SJeremy L Thompson   return CEED_ERROR_SUCCESS;
87243bbe138SJeremy L Thompson }
87343bbe138SJeremy L Thompson 
87443bbe138SJeremy L Thompson /**
87552d6035fSJeremy L Thompson   @brief Add a sub-operator to a composite CeedOperator
876288c0443SJeremy L Thompson 
877d1d35e2fSjeremylt   @param[out] composite_op  Composite CeedOperator
878d1d35e2fSjeremylt   @param      sub_op        Sub-operator CeedOperator
87952d6035fSJeremy L Thompson 
88052d6035fSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
88152d6035fSJeremy L Thompson 
8827a982d89SJeremy L. Thompson   @ref User
88352d6035fSJeremy L Thompson  */
884d1d35e2fSjeremylt int CeedCompositeOperatorAddSub(CeedOperator composite_op,
885d1d35e2fSjeremylt                                 CeedOperator sub_op) {
88634359f16Sjeremylt   int ierr;
88734359f16Sjeremylt 
888f04ea552SJeremy L Thompson   if (!composite_op->is_composite)
889c042f62fSJeremy L Thompson     // LCOV_EXCL_START
890d1d35e2fSjeremylt     return CeedError(composite_op->ceed, CEED_ERROR_MINOR,
891e2f04181SAndrew T. Barker                      "CeedOperator is not a composite operator");
892c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
89352d6035fSJeremy L Thompson 
894d1d35e2fSjeremylt   if (composite_op->num_suboperators == CEED_COMPOSITE_MAX)
895c042f62fSJeremy L Thompson     // LCOV_EXCL_START
896d1d35e2fSjeremylt     return CeedError(composite_op->ceed, CEED_ERROR_UNSUPPORTED,
897d1d35e2fSjeremylt                      "Cannot add additional sub_operators");
898c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
899f04ea552SJeremy L Thompson   if (composite_op->is_immutable)
900f04ea552SJeremy L Thompson     // LCOV_EXCL_START
901f04ea552SJeremy L Thompson     return CeedError(composite_op->ceed, CEED_ERROR_MAJOR,
902f04ea552SJeremy L Thompson                      "Operator cannot be changed after set as immutable");
903f04ea552SJeremy L Thompson   // LCOV_EXCL_STOP
90452d6035fSJeremy L Thompson 
905d1d35e2fSjeremylt   composite_op->sub_operators[composite_op->num_suboperators] = sub_op;
9069560d06aSjeremylt   ierr = CeedOperatorReference(sub_op); CeedChk(ierr);
907d1d35e2fSjeremylt   composite_op->num_suboperators++;
908e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
90952d6035fSJeremy L Thompson }
91052d6035fSJeremy L Thompson 
91152d6035fSJeremy L Thompson /**
9124db537f9SJeremy L Thompson   @brief Check if a CeedOperator is ready to be used.
9134db537f9SJeremy L Thompson 
9144db537f9SJeremy L Thompson   @param[in] op  CeedOperator to check
9154db537f9SJeremy L Thompson 
9164db537f9SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
9174db537f9SJeremy L Thompson 
9184db537f9SJeremy L Thompson   @ref User
9194db537f9SJeremy L Thompson **/
9204db537f9SJeremy L Thompson int CeedOperatorCheckReady(CeedOperator op) {
9214db537f9SJeremy L Thompson   int ierr;
9224db537f9SJeremy L Thompson   Ceed ceed;
9234db537f9SJeremy L Thompson   ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
9244db537f9SJeremy L Thompson 
9254db537f9SJeremy L Thompson   if (op->is_interface_setup)
9264db537f9SJeremy L Thompson     return CEED_ERROR_SUCCESS;
9274db537f9SJeremy L Thompson 
9284db537f9SJeremy L Thompson   CeedQFunction qf = op->qf;
9294db537f9SJeremy L Thompson   if (op->is_composite) {
9304db537f9SJeremy L Thompson     if (!op->num_suboperators)
9314db537f9SJeremy L Thompson       // LCOV_EXCL_START
9324db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE, "No sub_operators set");
9334db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9344db537f9SJeremy L Thompson     for (CeedInt i = 0; i < op->num_suboperators; i++) {
9354db537f9SJeremy L Thompson       ierr = CeedOperatorCheckReady(op->sub_operators[i]); CeedChk(ierr);
9364db537f9SJeremy L Thompson     }
9374db537f9SJeremy L Thompson   } else {
9384db537f9SJeremy L Thompson     if (op->num_fields == 0)
9394db537f9SJeremy L Thompson       // LCOV_EXCL_START
9404db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE, "No operator fields set");
9414db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9424db537f9SJeremy L Thompson     if (op->num_fields < qf->num_input_fields + qf->num_output_fields)
9434db537f9SJeremy L Thompson       // LCOV_EXCL_START
9444db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE, "Not all operator fields set");
9454db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9464db537f9SJeremy L Thompson     if (!op->has_restriction)
9474db537f9SJeremy L Thompson       // LCOV_EXCL_START
9484db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE,
9494db537f9SJeremy L Thompson                        "At least one restriction required");
9504db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9514db537f9SJeremy L Thompson     if (op->num_qpts == 0)
9524db537f9SJeremy L Thompson       // LCOV_EXCL_START
9534db537f9SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_INCOMPLETE,
9544db537f9SJeremy L Thompson                        "At least one non-collocated basis is required "
9554db537f9SJeremy L Thompson                        "or the number of quadrature points must be set");
9564db537f9SJeremy L Thompson     // LCOV_EXCL_STOP
9574db537f9SJeremy L Thompson   }
9584db537f9SJeremy L Thompson 
9594db537f9SJeremy L Thompson   // Flag as immutable and ready
9604db537f9SJeremy L Thompson   op->is_interface_setup = true;
9614db537f9SJeremy L Thompson   if (op->qf && op->qf != CEED_QFUNCTION_NONE)
9624db537f9SJeremy L Thompson     // LCOV_EXCL_START
9634db537f9SJeremy L Thompson     op->qf->is_immutable = true;
9644db537f9SJeremy L Thompson   // LCOV_EXCL_STOP
9654db537f9SJeremy L Thompson   if (op->dqf && op->dqf != CEED_QFUNCTION_NONE)
9664db537f9SJeremy L Thompson     // LCOV_EXCL_START
9674db537f9SJeremy L Thompson     op->dqf->is_immutable = true;
9684db537f9SJeremy L Thompson   // LCOV_EXCL_STOP
9694db537f9SJeremy L Thompson   if (op->dqfT && op->dqfT != CEED_QFUNCTION_NONE)
9704db537f9SJeremy L Thompson     // LCOV_EXCL_START
9714db537f9SJeremy L Thompson     op->dqfT->is_immutable = true;
9724db537f9SJeremy L Thompson   // LCOV_EXCL_STOP
9734db537f9SJeremy L Thompson   return CEED_ERROR_SUCCESS;
9744db537f9SJeremy L Thompson }
9754db537f9SJeremy L Thompson 
9764db537f9SJeremy L Thompson /**
977cd4dfc48Sjeremylt   @brief Set the number of quadrature points associated with a CeedOperator.
978cd4dfc48Sjeremylt            This should be used when creating a CeedOperator where every
979cd4dfc48Sjeremylt            field has a collocated basis. This function cannot be used for
980cd4dfc48Sjeremylt            composite CeedOperators.
981cd4dfc48Sjeremylt 
982cd4dfc48Sjeremylt   @param op        CeedOperator
983cd4dfc48Sjeremylt   @param num_qpts  Number of quadrature points to set
984cd4dfc48Sjeremylt 
985cd4dfc48Sjeremylt   @return An error code: 0 - success, otherwise - failure
986cd4dfc48Sjeremylt 
987e9b533fbSJeremy L Thompson   @ref Advanced
988cd4dfc48Sjeremylt **/
989cd4dfc48Sjeremylt int CeedOperatorSetNumQuadraturePoints(CeedOperator op, CeedInt num_qpts) {
990f04ea552SJeremy L Thompson   if (op->is_composite)
991cd4dfc48Sjeremylt     // LCOV_EXCL_START
992cd4dfc48Sjeremylt     return CeedError(op->ceed, CEED_ERROR_MINOR,
993cd4dfc48Sjeremylt                      "Not defined for composite operator");
994cd4dfc48Sjeremylt   // LCOV_EXCL_STOP
995cd4dfc48Sjeremylt   if (op->num_qpts)
996cd4dfc48Sjeremylt     // LCOV_EXCL_START
997cd4dfc48Sjeremylt     return CeedError(op->ceed, CEED_ERROR_MINOR,
998cd4dfc48Sjeremylt                      "Number of quadrature points already defined");
999cd4dfc48Sjeremylt   // LCOV_EXCL_STOP
1000f04ea552SJeremy L Thompson   if (op->is_immutable)
1001f04ea552SJeremy L Thompson     // LCOV_EXCL_START
1002f04ea552SJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MAJOR,
1003f04ea552SJeremy L Thompson                      "Operator cannot be changed after set as immutable");
1004f04ea552SJeremy L Thompson   // LCOV_EXCL_STOP
1005cd4dfc48Sjeremylt 
1006cd4dfc48Sjeremylt   op->num_qpts = num_qpts;
1007cd4dfc48Sjeremylt   return CEED_ERROR_SUCCESS;
1008cd4dfc48Sjeremylt }
1009cd4dfc48Sjeremylt 
1010cd4dfc48Sjeremylt /**
10117a982d89SJeremy L. Thompson   @brief View a CeedOperator
10127a982d89SJeremy L. Thompson 
10137a982d89SJeremy L. Thompson   @param[in] op      CeedOperator to view
10147a982d89SJeremy L. Thompson   @param[in] stream  Stream to write; typically stdout/stderr or a file
10157a982d89SJeremy L. Thompson 
10167a982d89SJeremy L. Thompson   @return Error code: 0 - success, otherwise - failure
10177a982d89SJeremy L. Thompson 
10187a982d89SJeremy L. Thompson   @ref User
10197a982d89SJeremy L. Thompson **/
10207a982d89SJeremy L. Thompson int CeedOperatorView(CeedOperator op, FILE *stream) {
10217a982d89SJeremy L. Thompson   int ierr;
10227a982d89SJeremy L. Thompson 
1023f04ea552SJeremy L Thompson   if (op->is_composite) {
10247a982d89SJeremy L. Thompson     fprintf(stream, "Composite CeedOperator\n");
10257a982d89SJeremy L. Thompson 
1026d1d35e2fSjeremylt     for (CeedInt i=0; i<op->num_suboperators; i++) {
10277a982d89SJeremy L. Thompson       fprintf(stream, "  SubOperator [%d]:\n", i);
1028d1d35e2fSjeremylt       ierr = CeedOperatorSingleView(op->sub_operators[i], 1, stream);
10297a982d89SJeremy L. Thompson       CeedChk(ierr);
10307a982d89SJeremy L. Thompson     }
10317a982d89SJeremy L. Thompson   } else {
10327a982d89SJeremy L. Thompson     fprintf(stream, "CeedOperator\n");
10337a982d89SJeremy L. Thompson     ierr = CeedOperatorSingleView(op, 0, stream); CeedChk(ierr);
10347a982d89SJeremy L. Thompson   }
1035e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
10367a982d89SJeremy L. Thompson }
10373bd813ffSjeremylt 
10383bd813ffSjeremylt /**
1039b7c9bbdaSJeremy L Thompson   @brief Get the Ceed associated with a CeedOperator
1040b7c9bbdaSJeremy L Thompson 
1041b7c9bbdaSJeremy L Thompson   @param op         CeedOperator
1042b7c9bbdaSJeremy L Thompson   @param[out] ceed  Variable to store Ceed
1043b7c9bbdaSJeremy L Thompson 
1044b7c9bbdaSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1045b7c9bbdaSJeremy L Thompson 
1046b7c9bbdaSJeremy L Thompson   @ref Advanced
1047b7c9bbdaSJeremy L Thompson **/
1048b7c9bbdaSJeremy L Thompson int CeedOperatorGetCeed(CeedOperator op, Ceed *ceed) {
1049b7c9bbdaSJeremy L Thompson   *ceed = op->ceed;
1050b7c9bbdaSJeremy L Thompson   return CEED_ERROR_SUCCESS;
1051b7c9bbdaSJeremy L Thompson }
1052b7c9bbdaSJeremy L Thompson 
1053b7c9bbdaSJeremy L Thompson /**
1054b7c9bbdaSJeremy L Thompson   @brief Get the number of elements associated with a CeedOperator
1055b7c9bbdaSJeremy L Thompson 
1056b7c9bbdaSJeremy L Thompson   @param op             CeedOperator
1057b7c9bbdaSJeremy L Thompson   @param[out] num_elem  Variable to store number of elements
1058b7c9bbdaSJeremy L Thompson 
1059b7c9bbdaSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1060b7c9bbdaSJeremy L Thompson 
1061b7c9bbdaSJeremy L Thompson   @ref Advanced
1062b7c9bbdaSJeremy L Thompson **/
1063b7c9bbdaSJeremy L Thompson int CeedOperatorGetNumElements(CeedOperator op, CeedInt *num_elem) {
1064b7c9bbdaSJeremy L Thompson   if (op->is_composite)
1065b7c9bbdaSJeremy L Thompson     // LCOV_EXCL_START
1066b7c9bbdaSJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR,
1067b7c9bbdaSJeremy L Thompson                      "Not defined for composite operator");
1068b7c9bbdaSJeremy L Thompson   // LCOV_EXCL_STOP
1069b7c9bbdaSJeremy L Thompson 
1070b7c9bbdaSJeremy L Thompson   *num_elem = op->num_elem;
1071b7c9bbdaSJeremy L Thompson   return CEED_ERROR_SUCCESS;
1072b7c9bbdaSJeremy L Thompson }
1073b7c9bbdaSJeremy L Thompson 
1074b7c9bbdaSJeremy L Thompson /**
1075b7c9bbdaSJeremy L Thompson   @brief Get the number of quadrature points associated with a CeedOperator
1076b7c9bbdaSJeremy L Thompson 
1077b7c9bbdaSJeremy L Thompson   @param op             CeedOperator
1078b7c9bbdaSJeremy L Thompson   @param[out] num_qpts  Variable to store vector number of quadrature points
1079b7c9bbdaSJeremy L Thompson 
1080b7c9bbdaSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1081b7c9bbdaSJeremy L Thompson 
1082b7c9bbdaSJeremy L Thompson   @ref Advanced
1083b7c9bbdaSJeremy L Thompson **/
1084b7c9bbdaSJeremy L Thompson int CeedOperatorGetNumQuadraturePoints(CeedOperator op, CeedInt *num_qpts) {
1085b7c9bbdaSJeremy L Thompson   if (op->is_composite)
1086b7c9bbdaSJeremy L Thompson     // LCOV_EXCL_START
1087b7c9bbdaSJeremy L Thompson     return CeedError(op->ceed, CEED_ERROR_MINOR,
1088b7c9bbdaSJeremy L Thompson                      "Not defined for composite operator");
1089b7c9bbdaSJeremy L Thompson   // LCOV_EXCL_STOP
1090b7c9bbdaSJeremy L Thompson 
1091b7c9bbdaSJeremy L Thompson   *num_qpts = op->num_qpts;
1092b7c9bbdaSJeremy L Thompson   return CEED_ERROR_SUCCESS;
1093b7c9bbdaSJeremy L Thompson }
1094b7c9bbdaSJeremy L Thompson 
1095b7c9bbdaSJeremy L Thompson /**
10963668ca4bSJeremy L Thompson   @brief Get label for a registered QFunctionContext field, or `NULL` if no
10973668ca4bSJeremy L Thompson            field has been registered with this `field_name`.
10983668ca4bSJeremy L Thompson 
10993668ca4bSJeremy L Thompson   @param[in] op            CeedOperator
11003668ca4bSJeremy L Thompson   @param[in] field_name    Name of field to retrieve label
11013668ca4bSJeremy L Thompson   @param[out] field_label  Variable to field label
11023668ca4bSJeremy L Thompson 
11033668ca4bSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
11043668ca4bSJeremy L Thompson 
11053668ca4bSJeremy L Thompson   @ref User
11063668ca4bSJeremy L Thompson **/
11073668ca4bSJeremy L Thompson int CeedOperatorContextGetFieldLabel(CeedOperator op,
11083668ca4bSJeremy L Thompson                                      const char *field_name,
11093668ca4bSJeremy L Thompson                                      CeedContextFieldLabel *field_label) {
11103668ca4bSJeremy L Thompson   int ierr;
11113668ca4bSJeremy L Thompson 
11123668ca4bSJeremy L Thompson   bool is_composite;
11133668ca4bSJeremy L Thompson   ierr = CeedOperatorIsComposite(op, &is_composite); CeedChk(ierr);
11143668ca4bSJeremy L Thompson   if (is_composite) {
11153668ca4bSJeremy L Thompson     // Check if composite label already created
11163668ca4bSJeremy L Thompson     for (CeedInt i=0; i<op->num_context_labels; i++) {
11173668ca4bSJeremy L Thompson       if (!strcmp(op->context_labels[i]->name, field_name)) {
11183668ca4bSJeremy L Thompson         *field_label = op->context_labels[i];
11193668ca4bSJeremy L Thompson         return CEED_ERROR_SUCCESS;
11203668ca4bSJeremy L Thompson       }
11213668ca4bSJeremy L Thompson     }
11223668ca4bSJeremy L Thompson 
11233668ca4bSJeremy L Thompson     // Create composite label if needed
11243668ca4bSJeremy L Thompson     CeedInt num_sub;
11253668ca4bSJeremy L Thompson     CeedOperator *sub_operators;
11263668ca4bSJeremy L Thompson     CeedContextFieldLabel new_field_label;
11273668ca4bSJeremy L Thompson 
11283668ca4bSJeremy L Thompson     ierr = CeedCalloc(1, &new_field_label); CeedChk(ierr);
11293668ca4bSJeremy L Thompson     ierr = CeedOperatorGetNumSub(op, &num_sub); CeedChk(ierr);
11303668ca4bSJeremy L Thompson     ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr);
11313668ca4bSJeremy L Thompson     ierr = CeedCalloc(num_sub, &new_field_label->sub_labels); CeedChk(ierr);
11323668ca4bSJeremy L Thompson     new_field_label->num_sub_labels = num_sub;
11333668ca4bSJeremy L Thompson 
11343668ca4bSJeremy L Thompson     bool label_found = false;
11353668ca4bSJeremy L Thompson     for (CeedInt i=0; i<num_sub; i++) {
11363668ca4bSJeremy L Thompson       if (sub_operators[i]->qf->ctx) {
11373668ca4bSJeremy L Thompson         CeedContextFieldLabel new_field_label_i;
11383668ca4bSJeremy L Thompson         ierr = CeedQFunctionContextGetFieldLabel(sub_operators[i]->qf->ctx, field_name,
11393668ca4bSJeremy L Thompson                &new_field_label_i); CeedChk(ierr);
11403668ca4bSJeremy L Thompson         if (new_field_label_i) {
11413668ca4bSJeremy L Thompson           label_found = true;
11423668ca4bSJeremy L Thompson           new_field_label->sub_labels[i] = new_field_label_i;
11433668ca4bSJeremy L Thompson           new_field_label->name = new_field_label_i->name;
11443668ca4bSJeremy L Thompson           new_field_label->description = new_field_label_i->description;
11457bfe0f0eSJeremy L Thompson           if (new_field_label->type &&
11467bfe0f0eSJeremy L Thompson               new_field_label->type != new_field_label_i->type) {
11477bfe0f0eSJeremy L Thompson             // LCOV_EXCL_START
11487bfe0f0eSJeremy L Thompson             ierr = CeedFree(&new_field_label); CeedChk(ierr);
11497bfe0f0eSJeremy L Thompson             return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
11507bfe0f0eSJeremy L Thompson                              "Incompatible field types on sub-operator contexts. "
11517bfe0f0eSJeremy L Thompson                              "%s != %s",
11527bfe0f0eSJeremy L Thompson                              CeedContextFieldTypes[new_field_label->type],
11537bfe0f0eSJeremy L Thompson                              CeedContextFieldTypes[new_field_label_i->type]);
11547bfe0f0eSJeremy L Thompson             // LCOV_EXCL_STOP
11557bfe0f0eSJeremy L Thompson           } else {
11567bfe0f0eSJeremy L Thompson             new_field_label->type = new_field_label_i->type;
11577bfe0f0eSJeremy L Thompson           }
11587bfe0f0eSJeremy L Thompson           if (new_field_label->num_values != 0 &&
11597bfe0f0eSJeremy L Thompson               new_field_label->num_values != new_field_label_i->num_values) {
11607bfe0f0eSJeremy L Thompson             // LCOV_EXCL_START
11617bfe0f0eSJeremy L Thompson             ierr = CeedFree(&new_field_label); CeedChk(ierr);
11627bfe0f0eSJeremy L Thompson             return CeedError(op->ceed, CEED_ERROR_INCOMPATIBLE,
11637bfe0f0eSJeremy L Thompson                              "Incompatible field number of values on sub-operator"
11647bfe0f0eSJeremy L Thompson                              " contexts. %ld != %ld",
11657bfe0f0eSJeremy L Thompson                              new_field_label->num_values, new_field_label_i->num_values);
11667bfe0f0eSJeremy L Thompson             // LCOV_EXCL_STOP
11677bfe0f0eSJeremy L Thompson           } else {
11687bfe0f0eSJeremy L Thompson             new_field_label->num_values = new_field_label_i->num_values;
11697bfe0f0eSJeremy L Thompson           }
11703668ca4bSJeremy L Thompson         }
11713668ca4bSJeremy L Thompson       }
11723668ca4bSJeremy L Thompson     }
11733668ca4bSJeremy L Thompson     if (!label_found) {
11743668ca4bSJeremy L Thompson       // LCOV_EXCL_START
1175*a48e5f43SJeremy L Thompson       ierr = CeedFree(&new_field_label->sub_labels); CeedChk(ierr);
11763668ca4bSJeremy L Thompson       ierr = CeedFree(&new_field_label); CeedChk(ierr);
11773668ca4bSJeremy L Thompson       *field_label = NULL;
11783668ca4bSJeremy L Thompson       // LCOV_EXCL_STOP
11793668ca4bSJeremy L Thompson     } else {
11803668ca4bSJeremy L Thompson       // Move new composite label to operator
11813668ca4bSJeremy L Thompson       if (op->num_context_labels == 0) {
11823668ca4bSJeremy L Thompson         ierr = CeedCalloc(1, &op->context_labels); CeedChk(ierr);
11833668ca4bSJeremy L Thompson         op->max_context_labels = 1;
11843668ca4bSJeremy L Thompson       } else if (op->num_context_labels == op->max_context_labels) {
11853668ca4bSJeremy L Thompson         ierr = CeedRealloc(2*op->num_context_labels, &op->context_labels);
11863668ca4bSJeremy L Thompson         CeedChk(ierr);
11873668ca4bSJeremy L Thompson         op->max_context_labels *= 2;
11883668ca4bSJeremy L Thompson       }
11893668ca4bSJeremy L Thompson       op->context_labels[op->num_context_labels] = new_field_label;
11903668ca4bSJeremy L Thompson       *field_label = new_field_label;
11913668ca4bSJeremy L Thompson       op->num_context_labels++;
11923668ca4bSJeremy L Thompson     }
11933668ca4bSJeremy L Thompson 
11943668ca4bSJeremy L Thompson     return CEED_ERROR_SUCCESS;
11953668ca4bSJeremy L Thompson   } else {
11963668ca4bSJeremy L Thompson     return CeedQFunctionContextGetFieldLabel(op->qf->ctx, field_name, field_label);
11973668ca4bSJeremy L Thompson   }
11983668ca4bSJeremy L Thompson }
11993668ca4bSJeremy L Thompson 
12003668ca4bSJeremy L Thompson /**
1201d8dd9a91SJeremy L Thompson   @brief Set QFunctionContext field holding a double precision value.
1202d8dd9a91SJeremy L Thompson            For composite operators, the value is set in all
1203d8dd9a91SJeremy L Thompson            sub-operator QFunctionContexts that have a matching `field_name`.
1204d8dd9a91SJeremy L Thompson 
1205d8dd9a91SJeremy L Thompson   @param op          CeedOperator
12063668ca4bSJeremy L Thompson   @param field_label Label of field to register
12077bfe0f0eSJeremy L Thompson   @param values      Values to set
1208d8dd9a91SJeremy L Thompson 
1209d8dd9a91SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1210d8dd9a91SJeremy L Thompson 
1211d8dd9a91SJeremy L Thompson   @ref User
1212d8dd9a91SJeremy L Thompson **/
12133668ca4bSJeremy L Thompson int CeedOperatorContextSetDouble(CeedOperator op,
12143668ca4bSJeremy L Thompson                                  CeedContextFieldLabel field_label,
12157bfe0f0eSJeremy L Thompson                                  double *values) {
12163668ca4bSJeremy L Thompson   return CeedOperatorContextSetGeneric(op, field_label, CEED_CONTEXT_FIELD_DOUBLE,
12177bfe0f0eSJeremy L Thompson                                        values);
1218d8dd9a91SJeremy L Thompson }
1219d8dd9a91SJeremy L Thompson 
1220d8dd9a91SJeremy L Thompson /**
1221d8dd9a91SJeremy L Thompson   @brief Set QFunctionContext field holding an int32 value.
1222d8dd9a91SJeremy L Thompson            For composite operators, the value is set in all
1223d8dd9a91SJeremy L Thompson            sub-operator QFunctionContexts that have a matching `field_name`.
1224d8dd9a91SJeremy L Thompson 
1225d8dd9a91SJeremy L Thompson   @param op          CeedOperator
12263668ca4bSJeremy L Thompson   @param field_label Label of field to set
12277bfe0f0eSJeremy L Thompson   @param values      Values to set
1228d8dd9a91SJeremy L Thompson 
1229d8dd9a91SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1230d8dd9a91SJeremy L Thompson 
1231d8dd9a91SJeremy L Thompson   @ref User
1232d8dd9a91SJeremy L Thompson **/
12333668ca4bSJeremy L Thompson int CeedOperatorContextSetInt32(CeedOperator op,
12343668ca4bSJeremy L Thompson                                 CeedContextFieldLabel field_label,
12357bfe0f0eSJeremy L Thompson                                 int *values) {
12363668ca4bSJeremy L Thompson   return CeedOperatorContextSetGeneric(op, field_label, CEED_CONTEXT_FIELD_INT32,
12377bfe0f0eSJeremy L Thompson                                        values);
1238d8dd9a91SJeremy L Thompson }
1239d8dd9a91SJeremy L Thompson 
1240d8dd9a91SJeremy L Thompson /**
12413bd813ffSjeremylt   @brief Apply CeedOperator to a vector
1242d7b241e6Sjeremylt 
1243d7b241e6Sjeremylt   This computes the action of the operator on the specified (active) input,
1244d7b241e6Sjeremylt   yielding its (active) output.  All inputs and outputs must be specified using
1245d7b241e6Sjeremylt   CeedOperatorSetField().
1246d7b241e6Sjeremylt 
1247f04ea552SJeremy L Thompson   Note: Calling this function asserts that setup is complete
1248f04ea552SJeremy L Thompson           and sets the CeedOperator as immutable.
1249f04ea552SJeremy L Thompson 
1250d7b241e6Sjeremylt   @param op        CeedOperator to apply
12514cc79fe7SJed Brown   @param[in] in    CeedVector containing input state or @ref CEED_VECTOR_NONE if
125234138859Sjeremylt                      there are no active inputs
1253b11c1e72Sjeremylt   @param[out] out  CeedVector to store result of applying operator (must be
12544cc79fe7SJed Brown                      distinct from @a in) or @ref CEED_VECTOR_NONE if there are no
125534138859Sjeremylt                      active outputs
1256d7b241e6Sjeremylt   @param request   Address of CeedRequest for non-blocking completion, else
12574cc79fe7SJed Brown                      @ref CEED_REQUEST_IMMEDIATE
1258b11c1e72Sjeremylt 
1259b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
1260dfdf5a53Sjeremylt 
12617a982d89SJeremy L. Thompson   @ref User
1262b11c1e72Sjeremylt **/
1263692c2638Sjeremylt int CeedOperatorApply(CeedOperator op, CeedVector in, CeedVector out,
1264692c2638Sjeremylt                       CeedRequest *request) {
1265d7b241e6Sjeremylt   int ierr;
1266e2f04181SAndrew T. Barker   ierr = CeedOperatorCheckReady(op); CeedChk(ierr);
1267d7b241e6Sjeremylt 
1268d1d35e2fSjeremylt   if (op->num_elem)  {
1269250756a7Sjeremylt     // Standard Operator
1270cae8b89aSjeremylt     if (op->Apply) {
1271250756a7Sjeremylt       ierr = op->Apply(op, in, out, request); CeedChk(ierr);
1272cae8b89aSjeremylt     } else {
1273cae8b89aSjeremylt       // Zero all output vectors
1274250756a7Sjeremylt       CeedQFunction qf = op->qf;
1275d1d35e2fSjeremylt       for (CeedInt i=0; i<qf->num_output_fields; i++) {
1276d1d35e2fSjeremylt         CeedVector vec = op->output_fields[i]->vec;
1277cae8b89aSjeremylt         if (vec == CEED_VECTOR_ACTIVE)
1278cae8b89aSjeremylt           vec = out;
1279cae8b89aSjeremylt         if (vec != CEED_VECTOR_NONE) {
1280cae8b89aSjeremylt           ierr = CeedVectorSetValue(vec, 0.0); CeedChk(ierr);
1281cae8b89aSjeremylt         }
1282cae8b89aSjeremylt       }
1283250756a7Sjeremylt       // Apply
1284250756a7Sjeremylt       ierr = op->ApplyAdd(op, in, out, request); CeedChk(ierr);
1285250756a7Sjeremylt     }
1286f04ea552SJeremy L Thompson   } else if (op->is_composite) {
1287250756a7Sjeremylt     // Composite Operator
1288250756a7Sjeremylt     if (op->ApplyComposite) {
1289250756a7Sjeremylt       ierr = op->ApplyComposite(op, in, out, request); CeedChk(ierr);
1290250756a7Sjeremylt     } else {
1291d1d35e2fSjeremylt       CeedInt num_suboperators;
1292d1d35e2fSjeremylt       ierr = CeedOperatorGetNumSub(op, &num_suboperators); CeedChk(ierr);
1293d1d35e2fSjeremylt       CeedOperator *sub_operators;
1294d1d35e2fSjeremylt       ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr);
1295250756a7Sjeremylt 
1296250756a7Sjeremylt       // Zero all output vectors
1297250756a7Sjeremylt       if (out != CEED_VECTOR_NONE) {
1298cae8b89aSjeremylt         ierr = CeedVectorSetValue(out, 0.0); CeedChk(ierr);
1299cae8b89aSjeremylt       }
1300d1d35e2fSjeremylt       for (CeedInt i=0; i<num_suboperators; i++) {
1301d1d35e2fSjeremylt         for (CeedInt j=0; j<sub_operators[i]->qf->num_output_fields; j++) {
1302d1d35e2fSjeremylt           CeedVector vec = sub_operators[i]->output_fields[j]->vec;
1303250756a7Sjeremylt           if (vec != CEED_VECTOR_ACTIVE && vec != CEED_VECTOR_NONE) {
1304250756a7Sjeremylt             ierr = CeedVectorSetValue(vec, 0.0); CeedChk(ierr);
1305250756a7Sjeremylt           }
1306250756a7Sjeremylt         }
1307250756a7Sjeremylt       }
1308250756a7Sjeremylt       // Apply
1309d1d35e2fSjeremylt       for (CeedInt i=0; i<op->num_suboperators; i++) {
1310d1d35e2fSjeremylt         ierr = CeedOperatorApplyAdd(op->sub_operators[i], in, out, request);
1311cae8b89aSjeremylt         CeedChk(ierr);
1312cae8b89aSjeremylt       }
1313cae8b89aSjeremylt     }
1314250756a7Sjeremylt   }
1315e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1316cae8b89aSjeremylt }
1317cae8b89aSjeremylt 
1318cae8b89aSjeremylt /**
1319cae8b89aSjeremylt   @brief Apply CeedOperator to a vector and add result to output vector
1320cae8b89aSjeremylt 
1321cae8b89aSjeremylt   This computes the action of the operator on the specified (active) input,
1322cae8b89aSjeremylt   yielding its (active) output.  All inputs and outputs must be specified using
1323cae8b89aSjeremylt   CeedOperatorSetField().
1324cae8b89aSjeremylt 
1325cae8b89aSjeremylt   @param op        CeedOperator to apply
1326cae8b89aSjeremylt   @param[in] in    CeedVector containing input state or NULL if there are no
1327cae8b89aSjeremylt                      active inputs
1328cae8b89aSjeremylt   @param[out] out  CeedVector to sum in result of applying operator (must be
1329cae8b89aSjeremylt                      distinct from @a in) or NULL if there are no active outputs
1330cae8b89aSjeremylt   @param request   Address of CeedRequest for non-blocking completion, else
13314cc79fe7SJed Brown                      @ref CEED_REQUEST_IMMEDIATE
1332cae8b89aSjeremylt 
1333cae8b89aSjeremylt   @return An error code: 0 - success, otherwise - failure
1334cae8b89aSjeremylt 
13357a982d89SJeremy L. Thompson   @ref User
1336cae8b89aSjeremylt **/
1337cae8b89aSjeremylt int CeedOperatorApplyAdd(CeedOperator op, CeedVector in, CeedVector out,
1338cae8b89aSjeremylt                          CeedRequest *request) {
1339cae8b89aSjeremylt   int ierr;
1340e2f04181SAndrew T. Barker   ierr = CeedOperatorCheckReady(op); CeedChk(ierr);
1341cae8b89aSjeremylt 
1342d1d35e2fSjeremylt   if (op->num_elem)  {
1343250756a7Sjeremylt     // Standard Operator
1344250756a7Sjeremylt     ierr = op->ApplyAdd(op, in, out, request); CeedChk(ierr);
1345f04ea552SJeremy L Thompson   } else if (op->is_composite) {
1346250756a7Sjeremylt     // Composite Operator
1347250756a7Sjeremylt     if (op->ApplyAddComposite) {
1348250756a7Sjeremylt       ierr = op->ApplyAddComposite(op, in, out, request); CeedChk(ierr);
1349cae8b89aSjeremylt     } else {
1350d1d35e2fSjeremylt       CeedInt num_suboperators;
1351d1d35e2fSjeremylt       ierr = CeedOperatorGetNumSub(op, &num_suboperators); CeedChk(ierr);
1352d1d35e2fSjeremylt       CeedOperator *sub_operators;
1353d1d35e2fSjeremylt       ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr);
1354250756a7Sjeremylt 
1355d1d35e2fSjeremylt       for (CeedInt i=0; i<num_suboperators; i++) {
1356d1d35e2fSjeremylt         ierr = CeedOperatorApplyAdd(sub_operators[i], in, out, request);
1357cae8b89aSjeremylt         CeedChk(ierr);
13581d7d2407SJeremy L Thompson       }
1359250756a7Sjeremylt     }
1360250756a7Sjeremylt   }
1361e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1362d7b241e6Sjeremylt }
1363d7b241e6Sjeremylt 
1364d7b241e6Sjeremylt /**
1365b11c1e72Sjeremylt   @brief Destroy a CeedOperator
1366d7b241e6Sjeremylt 
1367d7b241e6Sjeremylt   @param op  CeedOperator to destroy
1368b11c1e72Sjeremylt 
1369b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
1370dfdf5a53Sjeremylt 
13717a982d89SJeremy L. Thompson   @ref User
1372b11c1e72Sjeremylt **/
1373d7b241e6Sjeremylt int CeedOperatorDestroy(CeedOperator *op) {
1374d7b241e6Sjeremylt   int ierr;
1375d7b241e6Sjeremylt 
1376d1d35e2fSjeremylt   if (!*op || --(*op)->ref_count > 0) return CEED_ERROR_SUCCESS;
1377d7b241e6Sjeremylt   if ((*op)->Destroy) {
1378d7b241e6Sjeremylt     ierr = (*op)->Destroy(*op); CeedChk(ierr);
1379d7b241e6Sjeremylt   }
1380fe2413ffSjeremylt   ierr = CeedDestroy(&(*op)->ceed); CeedChk(ierr);
1381fe2413ffSjeremylt   // Free fields
13823668ca4bSJeremy L Thompson   for (CeedInt i=0; i<(*op)->num_fields; i++)
1383d1d35e2fSjeremylt     if ((*op)->input_fields[i]) {
1384d1d35e2fSjeremylt       if ((*op)->input_fields[i]->elem_restr != CEED_ELEMRESTRICTION_NONE) {
1385d1d35e2fSjeremylt         ierr = CeedElemRestrictionDestroy(&(*op)->input_fields[i]->elem_restr);
138671352a93Sjeremylt         CeedChk(ierr);
138715910d16Sjeremylt       }
1388d1d35e2fSjeremylt       if ((*op)->input_fields[i]->basis != CEED_BASIS_COLLOCATED) {
1389d1d35e2fSjeremylt         ierr = CeedBasisDestroy(&(*op)->input_fields[i]->basis); CeedChk(ierr);
139071352a93Sjeremylt       }
1391d1d35e2fSjeremylt       if ((*op)->input_fields[i]->vec != CEED_VECTOR_ACTIVE &&
1392d1d35e2fSjeremylt           (*op)->input_fields[i]->vec != CEED_VECTOR_NONE ) {
1393d1d35e2fSjeremylt         ierr = CeedVectorDestroy(&(*op)->input_fields[i]->vec); CeedChk(ierr);
139471352a93Sjeremylt       }
1395d1d35e2fSjeremylt       ierr = CeedFree(&(*op)->input_fields[i]->field_name); CeedChk(ierr);
1396d1d35e2fSjeremylt       ierr = CeedFree(&(*op)->input_fields[i]); CeedChk(ierr);
1397fe2413ffSjeremylt     }
13983668ca4bSJeremy L Thompson   for (CeedInt i=0; i<(*op)->num_fields; i++)
1399d1d35e2fSjeremylt     if ((*op)->output_fields[i]) {
1400d1d35e2fSjeremylt       ierr = CeedElemRestrictionDestroy(&(*op)->output_fields[i]->elem_restr);
140171352a93Sjeremylt       CeedChk(ierr);
1402d1d35e2fSjeremylt       if ((*op)->output_fields[i]->basis != CEED_BASIS_COLLOCATED) {
1403d1d35e2fSjeremylt         ierr = CeedBasisDestroy(&(*op)->output_fields[i]->basis); CeedChk(ierr);
140471352a93Sjeremylt       }
1405d1d35e2fSjeremylt       if ((*op)->output_fields[i]->vec != CEED_VECTOR_ACTIVE &&
1406d1d35e2fSjeremylt           (*op)->output_fields[i]->vec != CEED_VECTOR_NONE ) {
1407d1d35e2fSjeremylt         ierr = CeedVectorDestroy(&(*op)->output_fields[i]->vec); CeedChk(ierr);
140871352a93Sjeremylt       }
1409d1d35e2fSjeremylt       ierr = CeedFree(&(*op)->output_fields[i]->field_name); CeedChk(ierr);
1410d1d35e2fSjeremylt       ierr = CeedFree(&(*op)->output_fields[i]); CeedChk(ierr);
1411fe2413ffSjeremylt     }
1412d1d35e2fSjeremylt   // Destroy sub_operators
14133668ca4bSJeremy L Thompson   for (CeedInt i=0; i<(*op)->num_suboperators; i++)
1414d1d35e2fSjeremylt     if ((*op)->sub_operators[i]) {
1415d1d35e2fSjeremylt       ierr = CeedOperatorDestroy(&(*op)->sub_operators[i]); CeedChk(ierr);
141652d6035fSJeremy L Thompson     }
1417d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->qf); CeedChk(ierr);
1418d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->dqf); CeedChk(ierr);
1419d7b241e6Sjeremylt   ierr = CeedQFunctionDestroy(&(*op)->dqfT); CeedChk(ierr);
14203668ca4bSJeremy L Thompson   // Destroy any composite labels
14213668ca4bSJeremy L Thompson   for (CeedInt i=0; i<(*op)->num_context_labels; i++) {
14223668ca4bSJeremy L Thompson     ierr = CeedFree(&(*op)->context_labels[i]->sub_labels); CeedChk(ierr);
14233668ca4bSJeremy L Thompson     ierr = CeedFree(&(*op)->context_labels[i]); CeedChk(ierr);
14243668ca4bSJeremy L Thompson   }
14253668ca4bSJeremy L Thompson   ierr = CeedFree(&(*op)->context_labels); CeedChk(ierr);
1426fe2413ffSjeremylt 
14275107b09fSJeremy L Thompson   // Destroy fallback
1428d1d35e2fSjeremylt   if ((*op)->op_fallback) {
1429d1d35e2fSjeremylt     ierr = (*op)->qf_fallback->Destroy((*op)->qf_fallback); CeedChk(ierr);
1430d1d35e2fSjeremylt     ierr = CeedFree(&(*op)->qf_fallback); CeedChk(ierr);
143170a7ffb3SJeremy L Thompson     ierr = CeedVectorDestroy(&(*op)->op_fallback->qf_assembled); CeedChk(ierr);
143270a7ffb3SJeremy L Thompson     ierr = CeedElemRestrictionDestroy(&(*op)->op_fallback->qf_assembled_rstr);
143370a7ffb3SJeremy L Thompson     CeedChk(ierr);
1434d1d35e2fSjeremylt     ierr = (*op)->op_fallback->Destroy((*op)->op_fallback); CeedChk(ierr);
1435d1d35e2fSjeremylt     ierr = CeedFree(&(*op)->op_fallback); CeedChk(ierr);
14365107b09fSJeremy L Thompson   }
14375107b09fSJeremy L Thompson 
143870a7ffb3SJeremy L Thompson   // Destroy QF assembly cache
143970a7ffb3SJeremy L Thompson   ierr = CeedVectorDestroy(&(*op)->qf_assembled); CeedChk(ierr);
144070a7ffb3SJeremy L Thompson   ierr = CeedElemRestrictionDestroy(&(*op)->qf_assembled_rstr); CeedChk(ierr);
144170a7ffb3SJeremy L Thompson 
1442d1d35e2fSjeremylt   ierr = CeedFree(&(*op)->input_fields); CeedChk(ierr);
1443d1d35e2fSjeremylt   ierr = CeedFree(&(*op)->output_fields); CeedChk(ierr);
1444d1d35e2fSjeremylt   ierr = CeedFree(&(*op)->sub_operators); CeedChk(ierr);
1445d7b241e6Sjeremylt   ierr = CeedFree(op); CeedChk(ierr);
1446e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1447d7b241e6Sjeremylt }
1448d7b241e6Sjeremylt 
1449d7b241e6Sjeremylt /// @}
1450