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 276*3668ca4bSJeremy 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, 285*3668ca4bSJeremy L Thompson CeedContextFieldLabel field_label, CeedContextFieldType field_type, 286*3668ca4bSJeremy L Thompson void *value) { 287d8dd9a91SJeremy L Thompson int ierr; 288d8dd9a91SJeremy L Thompson 289*3668ca4bSJeremy L Thompson if (!field_label) 290*3668ca4bSJeremy L Thompson // LCOV_EXCL_START 291*3668ca4bSJeremy L Thompson return CeedError(op->ceed, CEED_ERROR_UNSUPPORTED, 292*3668ca4bSJeremy L Thompson "Invalid field label"); 293*3668ca4bSJeremy L Thompson // LCOV_EXCL_STOP 294*3668ca4bSJeremy L Thompson 295*3668ca4bSJeremy 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); 303*3668ca4bSJeremy L Thompson if (num_sub != field_label->num_sub_labels) 304*3668ca4bSJeremy L Thompson // LCOV_EXCL_START 305*3668ca4bSJeremy L Thompson return CeedError(op->ceed, CEED_ERROR_UNSUPPORTED, 306*3668ca4bSJeremy L Thompson "ContextLabel does not correspond to composite operator.\n" 307*3668ca4bSJeremy L Thompson "Use CeedOperatorGetContextFieldLabel()."); 308*3668ca4bSJeremy 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 312*3668ca4bSJeremy L Thompson if (field_label->sub_labels[i] && sub_operators[i]->qf->ctx) { 313*3668ca4bSJeremy L Thompson ierr = CeedQFunctionContextSetGeneric(sub_operators[i]->qf->ctx, 314*3668ca4bSJeremy L Thompson field_label->sub_labels[i], 315*3668ca4bSJeremy 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 325*3668ca4bSJeremy L Thompson ierr = CeedQFunctionContextSetGeneric(op->qf->ctx, field_label, 326*3668ca4bSJeremy 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 /** 1096*3668ca4bSJeremy L Thompson @brief Get label for a registered QFunctionContext field, or `NULL` if no 1097*3668ca4bSJeremy L Thompson field has been registered with this `field_name`. 1098*3668ca4bSJeremy L Thompson 1099*3668ca4bSJeremy L Thompson @param[in] op CeedOperator 1100*3668ca4bSJeremy L Thompson @param[in] field_name Name of field to retrieve label 1101*3668ca4bSJeremy L Thompson @param[out] field_label Variable to field label 1102*3668ca4bSJeremy L Thompson 1103*3668ca4bSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1104*3668ca4bSJeremy L Thompson 1105*3668ca4bSJeremy L Thompson @ref User 1106*3668ca4bSJeremy L Thompson **/ 1107*3668ca4bSJeremy L Thompson int CeedOperatorContextGetFieldLabel(CeedOperator op, 1108*3668ca4bSJeremy L Thompson const char *field_name, 1109*3668ca4bSJeremy L Thompson CeedContextFieldLabel *field_label) { 1110*3668ca4bSJeremy L Thompson int ierr; 1111*3668ca4bSJeremy L Thompson 1112*3668ca4bSJeremy L Thompson bool is_composite; 1113*3668ca4bSJeremy L Thompson ierr = CeedOperatorIsComposite(op, &is_composite); CeedChk(ierr); 1114*3668ca4bSJeremy L Thompson if (is_composite) { 1115*3668ca4bSJeremy L Thompson // Check if composite label already created 1116*3668ca4bSJeremy L Thompson for (CeedInt i=0; i<op->num_context_labels; i++) { 1117*3668ca4bSJeremy L Thompson if (!strcmp(op->context_labels[i]->name, field_name)) { 1118*3668ca4bSJeremy L Thompson *field_label = op->context_labels[i]; 1119*3668ca4bSJeremy L Thompson return CEED_ERROR_SUCCESS; 1120*3668ca4bSJeremy L Thompson } 1121*3668ca4bSJeremy L Thompson } 1122*3668ca4bSJeremy L Thompson 1123*3668ca4bSJeremy L Thompson // Create composite label if needed 1124*3668ca4bSJeremy L Thompson CeedInt num_sub; 1125*3668ca4bSJeremy L Thompson CeedOperator *sub_operators; 1126*3668ca4bSJeremy L Thompson CeedContextFieldLabel new_field_label; 1127*3668ca4bSJeremy L Thompson 1128*3668ca4bSJeremy L Thompson ierr = CeedCalloc(1, &new_field_label); CeedChk(ierr); 1129*3668ca4bSJeremy L Thompson ierr = CeedOperatorGetNumSub(op, &num_sub); CeedChk(ierr); 1130*3668ca4bSJeremy L Thompson ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr); 1131*3668ca4bSJeremy L Thompson ierr = CeedCalloc(num_sub, &new_field_label->sub_labels); CeedChk(ierr); 1132*3668ca4bSJeremy L Thompson new_field_label->num_sub_labels = num_sub; 1133*3668ca4bSJeremy L Thompson 1134*3668ca4bSJeremy L Thompson bool label_found = false; 1135*3668ca4bSJeremy L Thompson for (CeedInt i=0; i<num_sub; i++) { 1136*3668ca4bSJeremy L Thompson if (sub_operators[i]->qf->ctx) { 1137*3668ca4bSJeremy L Thompson CeedContextFieldLabel new_field_label_i; 1138*3668ca4bSJeremy L Thompson ierr = CeedQFunctionContextGetFieldLabel(sub_operators[i]->qf->ctx, field_name, 1139*3668ca4bSJeremy L Thompson &new_field_label_i); CeedChk(ierr); 1140*3668ca4bSJeremy L Thompson if (new_field_label_i) { 1141*3668ca4bSJeremy L Thompson label_found = true; 1142*3668ca4bSJeremy L Thompson new_field_label->sub_labels[i] = new_field_label_i; 1143*3668ca4bSJeremy L Thompson new_field_label->name = new_field_label_i->name; 1144*3668ca4bSJeremy L Thompson new_field_label->description = new_field_label_i->description; 1145*3668ca4bSJeremy L Thompson } 1146*3668ca4bSJeremy L Thompson } 1147*3668ca4bSJeremy L Thompson } 1148*3668ca4bSJeremy L Thompson if (!label_found) { 1149*3668ca4bSJeremy L Thompson // LCOV_EXCL_START 1150*3668ca4bSJeremy L Thompson ierr = CeedFree(&new_field_label); CeedChk(ierr); 1151*3668ca4bSJeremy L Thompson *field_label = NULL; 1152*3668ca4bSJeremy L Thompson // LCOV_EXCL_STOP 1153*3668ca4bSJeremy L Thompson } else { 1154*3668ca4bSJeremy L Thompson // Move new composite label to operator 1155*3668ca4bSJeremy L Thompson if (op->num_context_labels == 0) { 1156*3668ca4bSJeremy L Thompson ierr = CeedCalloc(1, &op->context_labels); CeedChk(ierr); 1157*3668ca4bSJeremy L Thompson op->max_context_labels = 1; 1158*3668ca4bSJeremy L Thompson } else if (op->num_context_labels == op->max_context_labels) { 1159*3668ca4bSJeremy L Thompson ierr = CeedRealloc(2*op->num_context_labels, &op->context_labels); 1160*3668ca4bSJeremy L Thompson CeedChk(ierr); 1161*3668ca4bSJeremy L Thompson op->max_context_labels *= 2; 1162*3668ca4bSJeremy L Thompson } 1163*3668ca4bSJeremy L Thompson op->context_labels[op->num_context_labels] = new_field_label; 1164*3668ca4bSJeremy L Thompson *field_label = new_field_label; 1165*3668ca4bSJeremy L Thompson op->num_context_labels++; 1166*3668ca4bSJeremy L Thompson } 1167*3668ca4bSJeremy L Thompson 1168*3668ca4bSJeremy L Thompson return CEED_ERROR_SUCCESS; 1169*3668ca4bSJeremy L Thompson } else { 1170*3668ca4bSJeremy L Thompson return CeedQFunctionContextGetFieldLabel(op->qf->ctx, field_name, field_label); 1171*3668ca4bSJeremy L Thompson } 1172*3668ca4bSJeremy L Thompson } 1173*3668ca4bSJeremy L Thompson 1174*3668ca4bSJeremy L Thompson /** 1175d8dd9a91SJeremy L Thompson @brief Set QFunctionContext field holding a double precision value. 1176d8dd9a91SJeremy L Thompson For composite operators, the value is set in all 1177d8dd9a91SJeremy L Thompson sub-operator QFunctionContexts that have a matching `field_name`. 1178d8dd9a91SJeremy L Thompson 1179d8dd9a91SJeremy L Thompson @param op CeedOperator 1180*3668ca4bSJeremy L Thompson @param field_label Label of field to register 1181d8dd9a91SJeremy L Thompson @param value Value to set 1182d8dd9a91SJeremy L Thompson 1183d8dd9a91SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1184d8dd9a91SJeremy L Thompson 1185d8dd9a91SJeremy L Thompson @ref User 1186d8dd9a91SJeremy L Thompson **/ 1187*3668ca4bSJeremy L Thompson int CeedOperatorContextSetDouble(CeedOperator op, 1188*3668ca4bSJeremy L Thompson CeedContextFieldLabel field_label, 1189d8dd9a91SJeremy L Thompson double value) { 1190*3668ca4bSJeremy L Thompson return CeedOperatorContextSetGeneric(op, field_label, CEED_CONTEXT_FIELD_DOUBLE, 1191d8dd9a91SJeremy L Thompson &value); 1192d8dd9a91SJeremy L Thompson } 1193d8dd9a91SJeremy L Thompson 1194d8dd9a91SJeremy L Thompson /** 1195d8dd9a91SJeremy L Thompson @brief Set QFunctionContext field holding an int32 value. 1196d8dd9a91SJeremy L Thompson For composite operators, the value is set in all 1197d8dd9a91SJeremy L Thompson sub-operator QFunctionContexts that have a matching `field_name`. 1198d8dd9a91SJeremy L Thompson 1199d8dd9a91SJeremy L Thompson @param op CeedOperator 1200*3668ca4bSJeremy L Thompson @param field_label Label of field to set 1201d8dd9a91SJeremy L Thompson @param value Value to set 1202d8dd9a91SJeremy L Thompson 1203d8dd9a91SJeremy L Thompson @return An error code: 0 - success, otherwise - failure 1204d8dd9a91SJeremy L Thompson 1205d8dd9a91SJeremy L Thompson @ref User 1206d8dd9a91SJeremy L Thompson **/ 1207*3668ca4bSJeremy L Thompson int CeedOperatorContextSetInt32(CeedOperator op, 1208*3668ca4bSJeremy L Thompson CeedContextFieldLabel field_label, 1209d8dd9a91SJeremy L Thompson int value) { 1210*3668ca4bSJeremy L Thompson return CeedOperatorContextSetGeneric(op, field_label, CEED_CONTEXT_FIELD_INT32, 1211d8dd9a91SJeremy L Thompson &value); 1212d8dd9a91SJeremy L Thompson } 1213d8dd9a91SJeremy L Thompson 1214d8dd9a91SJeremy L Thompson /** 12153bd813ffSjeremylt @brief Apply CeedOperator to a vector 1216d7b241e6Sjeremylt 1217d7b241e6Sjeremylt This computes the action of the operator on the specified (active) input, 1218d7b241e6Sjeremylt yielding its (active) output. All inputs and outputs must be specified using 1219d7b241e6Sjeremylt CeedOperatorSetField(). 1220d7b241e6Sjeremylt 1221f04ea552SJeremy L Thompson Note: Calling this function asserts that setup is complete 1222f04ea552SJeremy L Thompson and sets the CeedOperator as immutable. 1223f04ea552SJeremy L Thompson 1224d7b241e6Sjeremylt @param op CeedOperator to apply 12254cc79fe7SJed Brown @param[in] in CeedVector containing input state or @ref CEED_VECTOR_NONE if 122634138859Sjeremylt there are no active inputs 1227b11c1e72Sjeremylt @param[out] out CeedVector to store result of applying operator (must be 12284cc79fe7SJed Brown distinct from @a in) or @ref CEED_VECTOR_NONE if there are no 122934138859Sjeremylt active outputs 1230d7b241e6Sjeremylt @param request Address of CeedRequest for non-blocking completion, else 12314cc79fe7SJed Brown @ref CEED_REQUEST_IMMEDIATE 1232b11c1e72Sjeremylt 1233b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 1234dfdf5a53Sjeremylt 12357a982d89SJeremy L. Thompson @ref User 1236b11c1e72Sjeremylt **/ 1237692c2638Sjeremylt int CeedOperatorApply(CeedOperator op, CeedVector in, CeedVector out, 1238692c2638Sjeremylt CeedRequest *request) { 1239d7b241e6Sjeremylt int ierr; 1240e2f04181SAndrew T. Barker ierr = CeedOperatorCheckReady(op); CeedChk(ierr); 1241d7b241e6Sjeremylt 1242d1d35e2fSjeremylt if (op->num_elem) { 1243250756a7Sjeremylt // Standard Operator 1244cae8b89aSjeremylt if (op->Apply) { 1245250756a7Sjeremylt ierr = op->Apply(op, in, out, request); CeedChk(ierr); 1246cae8b89aSjeremylt } else { 1247cae8b89aSjeremylt // Zero all output vectors 1248250756a7Sjeremylt CeedQFunction qf = op->qf; 1249d1d35e2fSjeremylt for (CeedInt i=0; i<qf->num_output_fields; i++) { 1250d1d35e2fSjeremylt CeedVector vec = op->output_fields[i]->vec; 1251cae8b89aSjeremylt if (vec == CEED_VECTOR_ACTIVE) 1252cae8b89aSjeremylt vec = out; 1253cae8b89aSjeremylt if (vec != CEED_VECTOR_NONE) { 1254cae8b89aSjeremylt ierr = CeedVectorSetValue(vec, 0.0); CeedChk(ierr); 1255cae8b89aSjeremylt } 1256cae8b89aSjeremylt } 1257250756a7Sjeremylt // Apply 1258250756a7Sjeremylt ierr = op->ApplyAdd(op, in, out, request); CeedChk(ierr); 1259250756a7Sjeremylt } 1260f04ea552SJeremy L Thompson } else if (op->is_composite) { 1261250756a7Sjeremylt // Composite Operator 1262250756a7Sjeremylt if (op->ApplyComposite) { 1263250756a7Sjeremylt ierr = op->ApplyComposite(op, in, out, request); CeedChk(ierr); 1264250756a7Sjeremylt } else { 1265d1d35e2fSjeremylt CeedInt num_suboperators; 1266d1d35e2fSjeremylt ierr = CeedOperatorGetNumSub(op, &num_suboperators); CeedChk(ierr); 1267d1d35e2fSjeremylt CeedOperator *sub_operators; 1268d1d35e2fSjeremylt ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr); 1269250756a7Sjeremylt 1270250756a7Sjeremylt // Zero all output vectors 1271250756a7Sjeremylt if (out != CEED_VECTOR_NONE) { 1272cae8b89aSjeremylt ierr = CeedVectorSetValue(out, 0.0); CeedChk(ierr); 1273cae8b89aSjeremylt } 1274d1d35e2fSjeremylt for (CeedInt i=0; i<num_suboperators; i++) { 1275d1d35e2fSjeremylt for (CeedInt j=0; j<sub_operators[i]->qf->num_output_fields; j++) { 1276d1d35e2fSjeremylt CeedVector vec = sub_operators[i]->output_fields[j]->vec; 1277250756a7Sjeremylt if (vec != CEED_VECTOR_ACTIVE && vec != CEED_VECTOR_NONE) { 1278250756a7Sjeremylt ierr = CeedVectorSetValue(vec, 0.0); CeedChk(ierr); 1279250756a7Sjeremylt } 1280250756a7Sjeremylt } 1281250756a7Sjeremylt } 1282250756a7Sjeremylt // Apply 1283d1d35e2fSjeremylt for (CeedInt i=0; i<op->num_suboperators; i++) { 1284d1d35e2fSjeremylt ierr = CeedOperatorApplyAdd(op->sub_operators[i], in, out, request); 1285cae8b89aSjeremylt CeedChk(ierr); 1286cae8b89aSjeremylt } 1287cae8b89aSjeremylt } 1288250756a7Sjeremylt } 1289e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1290cae8b89aSjeremylt } 1291cae8b89aSjeremylt 1292cae8b89aSjeremylt /** 1293cae8b89aSjeremylt @brief Apply CeedOperator to a vector and add result to output vector 1294cae8b89aSjeremylt 1295cae8b89aSjeremylt This computes the action of the operator on the specified (active) input, 1296cae8b89aSjeremylt yielding its (active) output. All inputs and outputs must be specified using 1297cae8b89aSjeremylt CeedOperatorSetField(). 1298cae8b89aSjeremylt 1299cae8b89aSjeremylt @param op CeedOperator to apply 1300cae8b89aSjeremylt @param[in] in CeedVector containing input state or NULL if there are no 1301cae8b89aSjeremylt active inputs 1302cae8b89aSjeremylt @param[out] out CeedVector to sum in result of applying operator (must be 1303cae8b89aSjeremylt distinct from @a in) or NULL if there are no active outputs 1304cae8b89aSjeremylt @param request Address of CeedRequest for non-blocking completion, else 13054cc79fe7SJed Brown @ref CEED_REQUEST_IMMEDIATE 1306cae8b89aSjeremylt 1307cae8b89aSjeremylt @return An error code: 0 - success, otherwise - failure 1308cae8b89aSjeremylt 13097a982d89SJeremy L. Thompson @ref User 1310cae8b89aSjeremylt **/ 1311cae8b89aSjeremylt int CeedOperatorApplyAdd(CeedOperator op, CeedVector in, CeedVector out, 1312cae8b89aSjeremylt CeedRequest *request) { 1313cae8b89aSjeremylt int ierr; 1314e2f04181SAndrew T. Barker ierr = CeedOperatorCheckReady(op); CeedChk(ierr); 1315cae8b89aSjeremylt 1316d1d35e2fSjeremylt if (op->num_elem) { 1317250756a7Sjeremylt // Standard Operator 1318250756a7Sjeremylt ierr = op->ApplyAdd(op, in, out, request); CeedChk(ierr); 1319f04ea552SJeremy L Thompson } else if (op->is_composite) { 1320250756a7Sjeremylt // Composite Operator 1321250756a7Sjeremylt if (op->ApplyAddComposite) { 1322250756a7Sjeremylt ierr = op->ApplyAddComposite(op, in, out, request); CeedChk(ierr); 1323cae8b89aSjeremylt } else { 1324d1d35e2fSjeremylt CeedInt num_suboperators; 1325d1d35e2fSjeremylt ierr = CeedOperatorGetNumSub(op, &num_suboperators); CeedChk(ierr); 1326d1d35e2fSjeremylt CeedOperator *sub_operators; 1327d1d35e2fSjeremylt ierr = CeedOperatorGetSubList(op, &sub_operators); CeedChk(ierr); 1328250756a7Sjeremylt 1329d1d35e2fSjeremylt for (CeedInt i=0; i<num_suboperators; i++) { 1330d1d35e2fSjeremylt ierr = CeedOperatorApplyAdd(sub_operators[i], in, out, request); 1331cae8b89aSjeremylt CeedChk(ierr); 13321d7d2407SJeremy L Thompson } 1333250756a7Sjeremylt } 1334250756a7Sjeremylt } 1335e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1336d7b241e6Sjeremylt } 1337d7b241e6Sjeremylt 1338d7b241e6Sjeremylt /** 1339b11c1e72Sjeremylt @brief Destroy a CeedOperator 1340d7b241e6Sjeremylt 1341d7b241e6Sjeremylt @param op CeedOperator to destroy 1342b11c1e72Sjeremylt 1343b11c1e72Sjeremylt @return An error code: 0 - success, otherwise - failure 1344dfdf5a53Sjeremylt 13457a982d89SJeremy L. Thompson @ref User 1346b11c1e72Sjeremylt **/ 1347d7b241e6Sjeremylt int CeedOperatorDestroy(CeedOperator *op) { 1348d7b241e6Sjeremylt int ierr; 1349d7b241e6Sjeremylt 1350d1d35e2fSjeremylt if (!*op || --(*op)->ref_count > 0) return CEED_ERROR_SUCCESS; 1351d7b241e6Sjeremylt if ((*op)->Destroy) { 1352d7b241e6Sjeremylt ierr = (*op)->Destroy(*op); CeedChk(ierr); 1353d7b241e6Sjeremylt } 1354fe2413ffSjeremylt ierr = CeedDestroy(&(*op)->ceed); CeedChk(ierr); 1355fe2413ffSjeremylt // Free fields 1356*3668ca4bSJeremy L Thompson for (CeedInt i=0; i<(*op)->num_fields; i++) 1357d1d35e2fSjeremylt if ((*op)->input_fields[i]) { 1358d1d35e2fSjeremylt if ((*op)->input_fields[i]->elem_restr != CEED_ELEMRESTRICTION_NONE) { 1359d1d35e2fSjeremylt ierr = CeedElemRestrictionDestroy(&(*op)->input_fields[i]->elem_restr); 136071352a93Sjeremylt CeedChk(ierr); 136115910d16Sjeremylt } 1362d1d35e2fSjeremylt if ((*op)->input_fields[i]->basis != CEED_BASIS_COLLOCATED) { 1363d1d35e2fSjeremylt ierr = CeedBasisDestroy(&(*op)->input_fields[i]->basis); CeedChk(ierr); 136471352a93Sjeremylt } 1365d1d35e2fSjeremylt if ((*op)->input_fields[i]->vec != CEED_VECTOR_ACTIVE && 1366d1d35e2fSjeremylt (*op)->input_fields[i]->vec != CEED_VECTOR_NONE ) { 1367d1d35e2fSjeremylt ierr = CeedVectorDestroy(&(*op)->input_fields[i]->vec); CeedChk(ierr); 136871352a93Sjeremylt } 1369d1d35e2fSjeremylt ierr = CeedFree(&(*op)->input_fields[i]->field_name); CeedChk(ierr); 1370d1d35e2fSjeremylt ierr = CeedFree(&(*op)->input_fields[i]); CeedChk(ierr); 1371fe2413ffSjeremylt } 1372*3668ca4bSJeremy L Thompson for (CeedInt i=0; i<(*op)->num_fields; i++) 1373d1d35e2fSjeremylt if ((*op)->output_fields[i]) { 1374d1d35e2fSjeremylt ierr = CeedElemRestrictionDestroy(&(*op)->output_fields[i]->elem_restr); 137571352a93Sjeremylt CeedChk(ierr); 1376d1d35e2fSjeremylt if ((*op)->output_fields[i]->basis != CEED_BASIS_COLLOCATED) { 1377d1d35e2fSjeremylt ierr = CeedBasisDestroy(&(*op)->output_fields[i]->basis); CeedChk(ierr); 137871352a93Sjeremylt } 1379d1d35e2fSjeremylt if ((*op)->output_fields[i]->vec != CEED_VECTOR_ACTIVE && 1380d1d35e2fSjeremylt (*op)->output_fields[i]->vec != CEED_VECTOR_NONE ) { 1381d1d35e2fSjeremylt ierr = CeedVectorDestroy(&(*op)->output_fields[i]->vec); CeedChk(ierr); 138271352a93Sjeremylt } 1383d1d35e2fSjeremylt ierr = CeedFree(&(*op)->output_fields[i]->field_name); CeedChk(ierr); 1384d1d35e2fSjeremylt ierr = CeedFree(&(*op)->output_fields[i]); CeedChk(ierr); 1385fe2413ffSjeremylt } 1386d1d35e2fSjeremylt // Destroy sub_operators 1387*3668ca4bSJeremy L Thompson for (CeedInt i=0; i<(*op)->num_suboperators; i++) 1388d1d35e2fSjeremylt if ((*op)->sub_operators[i]) { 1389d1d35e2fSjeremylt ierr = CeedOperatorDestroy(&(*op)->sub_operators[i]); CeedChk(ierr); 139052d6035fSJeremy L Thompson } 1391d7b241e6Sjeremylt ierr = CeedQFunctionDestroy(&(*op)->qf); CeedChk(ierr); 1392d7b241e6Sjeremylt ierr = CeedQFunctionDestroy(&(*op)->dqf); CeedChk(ierr); 1393d7b241e6Sjeremylt ierr = CeedQFunctionDestroy(&(*op)->dqfT); CeedChk(ierr); 1394*3668ca4bSJeremy L Thompson // Destroy any composite labels 1395*3668ca4bSJeremy L Thompson for (CeedInt i=0; i<(*op)->num_context_labels; i++) { 1396*3668ca4bSJeremy L Thompson ierr = CeedFree(&(*op)->context_labels[i]->sub_labels); CeedChk(ierr); 1397*3668ca4bSJeremy L Thompson ierr = CeedFree(&(*op)->context_labels[i]); CeedChk(ierr); 1398*3668ca4bSJeremy L Thompson } 1399*3668ca4bSJeremy L Thompson ierr = CeedFree(&(*op)->context_labels); CeedChk(ierr); 1400fe2413ffSjeremylt 14015107b09fSJeremy L Thompson // Destroy fallback 1402d1d35e2fSjeremylt if ((*op)->op_fallback) { 1403d1d35e2fSjeremylt ierr = (*op)->qf_fallback->Destroy((*op)->qf_fallback); CeedChk(ierr); 1404d1d35e2fSjeremylt ierr = CeedFree(&(*op)->qf_fallback); CeedChk(ierr); 140570a7ffb3SJeremy L Thompson ierr = CeedVectorDestroy(&(*op)->op_fallback->qf_assembled); CeedChk(ierr); 140670a7ffb3SJeremy L Thompson ierr = CeedElemRestrictionDestroy(&(*op)->op_fallback->qf_assembled_rstr); 140770a7ffb3SJeremy L Thompson CeedChk(ierr); 1408d1d35e2fSjeremylt ierr = (*op)->op_fallback->Destroy((*op)->op_fallback); CeedChk(ierr); 1409d1d35e2fSjeremylt ierr = CeedFree(&(*op)->op_fallback); CeedChk(ierr); 14105107b09fSJeremy L Thompson } 14115107b09fSJeremy L Thompson 141270a7ffb3SJeremy L Thompson // Destroy QF assembly cache 141370a7ffb3SJeremy L Thompson ierr = CeedVectorDestroy(&(*op)->qf_assembled); CeedChk(ierr); 141470a7ffb3SJeremy L Thompson ierr = CeedElemRestrictionDestroy(&(*op)->qf_assembled_rstr); CeedChk(ierr); 141570a7ffb3SJeremy L Thompson 1416d1d35e2fSjeremylt ierr = CeedFree(&(*op)->input_fields); CeedChk(ierr); 1417d1d35e2fSjeremylt ierr = CeedFree(&(*op)->output_fields); CeedChk(ierr); 1418d1d35e2fSjeremylt ierr = CeedFree(&(*op)->sub_operators); CeedChk(ierr); 1419d7b241e6Sjeremylt ierr = CeedFree(op); CeedChk(ierr); 1420e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 1421d7b241e6Sjeremylt } 1422d7b241e6Sjeremylt 1423d7b241e6Sjeremylt /// @} 1424