1 // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at 2 // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights 3 // reserved. See files LICENSE and NOTICE for details. 4 // 5 // This file is part of CEED, a collection of benchmarks, miniapps, software 6 // libraries and APIs for efficient high-order finite element and spectral 7 // element discretizations for exascale applications. For more information and 8 // source code availability see http://github.com/ceed. 9 // 10 // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC, 11 // a collaborative effort of two U.S. Department of Energy organizations (Office 12 // of Science and the National Nuclear Security Administration) responsible for 13 // the planning and preparation of a capable exascale ecosystem, including 14 // software, applications, hardware, advanced system engineering and early 15 // testbed platforms, in support of the nation's exascale computing imperative. 16 17 #include <ceed/ceed.h> 18 #include <ceed/backend.h> 19 #include <ceed-impl.h> 20 #include <limits.h> 21 #include <stdbool.h> 22 #include <stdio.h> 23 #include <string.h> 24 25 /// @file 26 /// Implementation of public CeedQFunction interfaces 27 28 /// @cond DOXYGEN_SKIP 29 static struct CeedQFunction_private ceed_qfunction_none; 30 /// @endcond 31 32 /// @addtogroup CeedQFunctionUser 33 /// @{ 34 35 // Indicate that no QFunction is provided by the user 36 const CeedQFunction CEED_QFUNCTION_NONE = &ceed_qfunction_none; 37 38 /// @} 39 40 /// @cond DOXYGEN_SKIP 41 static struct { 42 char name[CEED_MAX_RESOURCE_LEN]; 43 char source[CEED_MAX_RESOURCE_LEN]; 44 CeedInt vec_length; 45 CeedQFunctionUser f; 46 int (*init)(Ceed ceed, const char *name, CeedQFunction qf); 47 } gallery_qfunctions[1024]; 48 static size_t num_qfunctions; 49 /// @endcond 50 51 /// ---------------------------------------------------------------------------- 52 /// CeedQFunction Library Internal Functions 53 /// ---------------------------------------------------------------------------- 54 /// @addtogroup CeedQFunctionDeveloper 55 /// @{ 56 57 /** 58 @brief Register a gallery QFunction 59 60 @param name Name for this backend to respond to 61 @param source Absolute path to source of QFunction, 62 "\path\CEED_DIR\gallery\folder\file.h:function_name" 63 @param vec_length Vector length. Caller must ensure that number of quadrature 64 points is a multiple of vec_length. 65 @param f Function pointer to evaluate action at quadrature points. 66 See \ref CeedQFunctionUser. 67 @param init Initialization function called by CeedQFunctionInit() when the 68 QFunction is selected. 69 70 @return An error code: 0 - success, otherwise - failure 71 72 @ref Developer 73 **/ 74 int CeedQFunctionRegister(const char *name, const char *source, 75 CeedInt vec_length, CeedQFunctionUser f, 76 int (*init)(Ceed, const char *, CeedQFunction)) { 77 if (num_qfunctions >= sizeof(gallery_qfunctions) / sizeof( 78 gallery_qfunctions[0])) 79 // LCOV_EXCL_START 80 return CeedError(NULL, CEED_ERROR_MAJOR, "Too many gallery QFunctions"); 81 // LCOV_EXCL_STOP 82 83 strncpy(gallery_qfunctions[num_qfunctions].name, name, CEED_MAX_RESOURCE_LEN); 84 gallery_qfunctions[num_qfunctions].name[CEED_MAX_RESOURCE_LEN-1] = 0; 85 strncpy(gallery_qfunctions[num_qfunctions].source, source, 86 CEED_MAX_RESOURCE_LEN); 87 gallery_qfunctions[num_qfunctions].source[CEED_MAX_RESOURCE_LEN-1] = 0; 88 gallery_qfunctions[num_qfunctions].vec_length = vec_length; 89 gallery_qfunctions[num_qfunctions].f = f; 90 gallery_qfunctions[num_qfunctions].init = init; 91 num_qfunctions++; 92 return CEED_ERROR_SUCCESS; 93 } 94 95 /** 96 @brief Set a CeedQFunction field, used by CeedQFunctionAddInput/Output 97 98 @param f CeedQFunctionField 99 @param field_name Name of QFunction field 100 @param size Size of QFunction field, (num_comp * dim) for @ref CEED_EVAL_GRAD or 101 (num_comp * 1) for @ref CEED_EVAL_NONE, @ref CEED_EVAL_INTERP, and @ref CEED_EVAL_WEIGHT 102 @param eval_mode \ref CEED_EVAL_NONE to use values directly, 103 \ref CEED_EVAL_INTERP to use interpolated values, 104 \ref CEED_EVAL_GRAD to use gradients, 105 \ref CEED_EVAL_WEIGHT to use quadrature weights. 106 107 @return An error code: 0 - success, otherwise - failure 108 109 @ref Developer 110 **/ 111 static int CeedQFunctionFieldSet(CeedQFunctionField *f,const char *field_name, 112 CeedInt size, CeedEvalMode eval_mode) { 113 size_t len = strlen(field_name); 114 char *tmp; 115 int ierr; 116 117 ierr = CeedCalloc(1, f); CeedChk(ierr); 118 ierr = CeedCalloc(len+1, &tmp); CeedChk(ierr); 119 memcpy(tmp, field_name, len+1); 120 (*f)->field_name = tmp; 121 (*f)->size = size; 122 (*f)->eval_mode = eval_mode; 123 return CEED_ERROR_SUCCESS; 124 } 125 126 /** 127 @brief View a field of a CeedQFunction 128 129 @param[in] field QFunction field to view 130 @param[in] field_number Number of field being viewed 131 @param[in] in true for input field, false for output 132 @param[in] stream Stream to view to, e.g., stdout 133 134 @return An error code: 0 - success, otherwise - failure 135 136 @ref Utility 137 **/ 138 static int CeedQFunctionFieldView(CeedQFunctionField field, 139 CeedInt field_number, 140 bool in, FILE *stream) { 141 int ierr; 142 const char *inout = in ? "Input" : "Output"; 143 char *field_name; 144 ierr = CeedQFunctionFieldGetName(field, &field_name); CeedChk(ierr); 145 CeedInt size; 146 ierr = CeedQFunctionFieldGetSize(field, &size); CeedChk(ierr); 147 CeedEvalMode eval_mode; 148 ierr = CeedQFunctionFieldGetEvalMode(field, &eval_mode); CeedChk(ierr); 149 fprintf(stream, " %s Field [%d]:\n" 150 " Name: \"%s\"\n" 151 " Size: %d\n" 152 " EvalMode: \"%s\"\n", 153 inout, field_number, field_name, size, CeedEvalModes[eval_mode]); 154 return CEED_ERROR_SUCCESS; 155 } 156 157 /** 158 @brief Set flag to determine if Fortran interface is used 159 160 @param qf CeedQFunction 161 @param status Boolean value to set as Fortran status 162 163 @return An error code: 0 - success, otherwise - failure 164 165 @ref Backend 166 **/ 167 int CeedQFunctionSetFortranStatus(CeedQFunction qf, bool status) { 168 qf->fortran_status = status; 169 return CEED_ERROR_SUCCESS; 170 } 171 172 /// @} 173 174 /// ---------------------------------------------------------------------------- 175 /// CeedQFunction Backend API 176 /// ---------------------------------------------------------------------------- 177 /// @addtogroup CeedQFunctionBackend 178 /// @{ 179 180 /** 181 @brief Get the Ceed associated with a CeedQFunction 182 183 @param qf CeedQFunction 184 @param[out] ceed Variable to store Ceed 185 186 @return An error code: 0 - success, otherwise - failure 187 188 @ref Backend 189 **/ 190 int CeedQFunctionGetCeed(CeedQFunction qf, Ceed *ceed) { 191 *ceed = qf->ceed; 192 return CEED_ERROR_SUCCESS; 193 } 194 195 /** 196 @brief Get the vector length of a CeedQFunction 197 198 @param qf CeedQFunction 199 @param[out] vec_length Variable to store vector length 200 201 @return An error code: 0 - success, otherwise - failure 202 203 @ref Backend 204 **/ 205 int CeedQFunctionGetVectorLength(CeedQFunction qf, CeedInt *vec_length) { 206 *vec_length = qf->vec_length; 207 return CEED_ERROR_SUCCESS; 208 } 209 210 /** 211 @brief Get the number of inputs and outputs to a CeedQFunction 212 213 @param qf CeedQFunction 214 @param[out] num_input Variable to store number of input fields 215 @param[out] num_output Variable to store number of output fields 216 217 @return An error code: 0 - success, otherwise - failure 218 219 @ref Backend 220 **/ 221 int CeedQFunctionGetNumArgs(CeedQFunction qf, CeedInt *num_input, 222 CeedInt *num_output) { 223 if (num_input) *num_input = qf->num_input_fields; 224 if (num_output) *num_output = qf->num_output_fields; 225 return CEED_ERROR_SUCCESS; 226 } 227 228 /** 229 @brief Get the source path string for a CeedQFunction 230 231 @param qf CeedQFunction 232 @param[out] source Variable to store source path string 233 234 @return An error code: 0 - success, otherwise - failure 235 236 @ref Backend 237 **/ 238 int CeedQFunctionGetSourcePath(CeedQFunction qf, char **source) { 239 *source = (char *) qf->source_path; 240 return CEED_ERROR_SUCCESS; 241 } 242 243 /** 244 @brief Get the User Function for a CeedQFunction 245 246 @param qf CeedQFunction 247 @param[out] f Variable to store user function 248 249 @return An error code: 0 - success, otherwise - failure 250 251 @ref Backend 252 **/ 253 int CeedQFunctionGetUserFunction(CeedQFunction qf, CeedQFunctionUser *f) { 254 *f = qf->function; 255 return CEED_ERROR_SUCCESS; 256 } 257 258 /** 259 @brief Get global context for a CeedQFunction. 260 Note: For QFunctions from the Fortran interface, this 261 function will return the Fortran context 262 CeedQFunctionContext. 263 264 @param qf CeedQFunction 265 @param[out] ctx Variable to store CeedQFunctionContext 266 267 @return An error code: 0 - success, otherwise - failure 268 269 @ref Backend 270 **/ 271 int CeedQFunctionGetContext(CeedQFunction qf, CeedQFunctionContext *ctx) { 272 *ctx = qf->ctx; 273 return CEED_ERROR_SUCCESS; 274 } 275 276 /** 277 @brief Get true user context for a CeedQFunction 278 Note: For all QFunctions this function will return the user 279 CeedQFunctionContext and not interface context 280 CeedQFunctionContext, if any such object exists. 281 282 @param qf CeedQFunction 283 @param[out] ctx Variable to store CeedQFunctionContext 284 285 @return An error code: 0 - success, otherwise - failure 286 @ref Backend 287 **/ 288 int CeedQFunctionGetInnerContext(CeedQFunction qf, CeedQFunctionContext *ctx) { 289 int ierr; 290 if (qf->fortran_status) { 291 CeedFortranContext fortran_ctx = NULL; 292 ierr = CeedQFunctionContextGetData(qf->ctx, CEED_MEM_HOST, &fortran_ctx); 293 CeedChk(ierr); 294 *ctx = fortran_ctx->innerctx; 295 ierr = CeedQFunctionContextRestoreData(qf->ctx, (void *)&fortran_ctx); 296 CeedChk(ierr); 297 } else { 298 *ctx = qf->ctx; 299 } 300 return CEED_ERROR_SUCCESS; 301 } 302 303 /** 304 @brief Determine if QFunction is identity 305 306 @param qf CeedQFunction 307 @param[out] is_identity Variable to store identity status 308 309 @return An error code: 0 - success, otherwise - failure 310 311 @ref Backend 312 **/ 313 int CeedQFunctionIsIdentity(CeedQFunction qf, bool *is_identity) { 314 *is_identity = qf->identity; 315 return CEED_ERROR_SUCCESS; 316 } 317 318 /** 319 @brief Get backend data of a CeedQFunction 320 321 @param qf CeedQFunction 322 @param[out] data Variable to store data 323 324 @return An error code: 0 - success, otherwise - failure 325 326 @ref Backend 327 **/ 328 int CeedQFunctionGetData(CeedQFunction qf, void *data) { 329 *(void **)data = qf->data; 330 return CEED_ERROR_SUCCESS; 331 } 332 333 /** 334 @brief Set backend data of a CeedQFunction 335 336 @param[out] qf CeedQFunction 337 @param data Data to set 338 339 @return An error code: 0 - success, otherwise - failure 340 341 @ref Backend 342 **/ 343 int CeedQFunctionSetData(CeedQFunction qf, void *data) { 344 qf->data = data; 345 return CEED_ERROR_SUCCESS; 346 } 347 348 /** 349 @brief Increment the reference counter for a CeedQFunction 350 351 @param qf CeedQFunction to increment the reference counter 352 353 @return An error code: 0 - success, otherwise - failure 354 355 @ref Backend 356 **/ 357 int CeedQFunctionReference(CeedQFunction qf) { 358 qf->ref_count++; 359 return CEED_ERROR_SUCCESS; 360 } 361 362 /** 363 @brief Get the CeedQFunctionFields of a CeedQFunction 364 365 @param qf CeedQFunction 366 @param[out] input_fields Variable to store input_fields 367 @param[out] output_fields Variable to store output_fields 368 369 @return An error code: 0 - success, otherwise - failure 370 371 @ref Backend 372 **/ 373 int CeedQFunctionGetFields(CeedQFunction qf, CeedQFunctionField **input_fields, 374 CeedQFunctionField **output_fields) { 375 if (input_fields) *input_fields = qf->input_fields; 376 if (output_fields) *output_fields = qf->output_fields; 377 return CEED_ERROR_SUCCESS; 378 } 379 380 /** 381 @brief Get the name of a CeedQFunctionField 382 383 @param qf_field CeedQFunctionField 384 @param[out] field_name Variable to store the field name 385 386 @return An error code: 0 - success, otherwise - failure 387 388 @ref Backend 389 **/ 390 int CeedQFunctionFieldGetName(CeedQFunctionField qf_field, char **field_name) { 391 *field_name = (char *)qf_field->field_name; 392 return CEED_ERROR_SUCCESS; 393 } 394 395 /** 396 @brief Get the number of components of a CeedQFunctionField 397 398 @param qf_field CeedQFunctionField 399 @param[out] size Variable to store the size of the field 400 401 @return An error code: 0 - success, otherwise - failure 402 403 @ref Backend 404 **/ 405 int CeedQFunctionFieldGetSize(CeedQFunctionField qf_field, CeedInt *size) { 406 *size = qf_field->size; 407 return CEED_ERROR_SUCCESS; 408 } 409 410 /** 411 @brief Get the CeedEvalMode of a CeedQFunctionField 412 413 @param qf_field CeedQFunctionField 414 @param[out] eval_mode Variable to store the field evaluation mode 415 416 @return An error code: 0 - success, otherwise - failure 417 418 @ref Backend 419 **/ 420 int CeedQFunctionFieldGetEvalMode(CeedQFunctionField qf_field, 421 CeedEvalMode *eval_mode) { 422 *eval_mode = qf_field->eval_mode; 423 return CEED_ERROR_SUCCESS; 424 } 425 426 /// @} 427 428 /// ---------------------------------------------------------------------------- 429 /// CeedQFunction Public API 430 /// ---------------------------------------------------------------------------- 431 /// @addtogroup CeedQFunctionUser 432 /// @{ 433 434 /** 435 @brief Create a CeedQFunction for evaluating interior (volumetric) terms. 436 437 @param ceed A Ceed object where the CeedQFunction will be created 438 @param vec_length Vector length. Caller must ensure that number of quadrature 439 points is a multiple of vec_length. 440 @param f Function pointer to evaluate action at quadrature points. 441 See \ref CeedQFunctionUser. 442 @param source Absolute path to source of QFunction, 443 "\abs_path\file.h:function_name". 444 For support across all backends, this source must only 445 contain constructs supported by C99, C++11, and CUDA. 446 @param[out] qf Address of the variable where the newly created 447 CeedQFunction will be stored 448 449 @return An error code: 0 - success, otherwise - failure 450 451 See \ref CeedQFunctionUser for details on the call-back function @a f's 452 arguments. 453 454 @ref User 455 **/ 456 int CeedQFunctionCreateInterior(Ceed ceed, CeedInt vec_length, 457 CeedQFunctionUser f, 458 const char *source, CeedQFunction *qf) { 459 int ierr; 460 char *source_copy; 461 462 if (!ceed->QFunctionCreate) { 463 Ceed delegate; 464 ierr = CeedGetObjectDelegate(ceed, &delegate, "QFunction"); CeedChk(ierr); 465 466 if (!delegate) 467 // LCOV_EXCL_START 468 return CeedError(ceed, CEED_ERROR_UNSUPPORTED, 469 "Backend does not support QFunctionCreate"); 470 // LCOV_EXCL_STOP 471 472 ierr = CeedQFunctionCreateInterior(delegate, vec_length, f, source, qf); 473 CeedChk(ierr); 474 return CEED_ERROR_SUCCESS; 475 } 476 477 ierr = CeedCalloc(1, qf); CeedChk(ierr); 478 (*qf)->ceed = ceed; 479 ierr = CeedReference(ceed); CeedChk(ierr); 480 (*qf)->ref_count = 1; 481 (*qf)->vec_length = vec_length; 482 (*qf)->identity = 0; 483 (*qf)->function = f; 484 size_t slen = strlen(source) + 1; 485 ierr = CeedMalloc(slen, &source_copy); CeedChk(ierr); 486 memcpy(source_copy, source, slen); 487 (*qf)->source_path = source_copy; 488 ierr = CeedCalloc(16, &(*qf)->input_fields); CeedChk(ierr); 489 ierr = CeedCalloc(16, &(*qf)->output_fields); CeedChk(ierr); 490 ierr = ceed->QFunctionCreate(*qf); CeedChk(ierr); 491 return CEED_ERROR_SUCCESS; 492 } 493 494 /** 495 @brief Create a CeedQFunction for evaluating interior (volumetric) terms by name. 496 497 @param ceed A Ceed object where the CeedQFunction will be created 498 @param name Name of QFunction to use from gallery 499 @param[out] qf Address of the variable where the newly created 500 CeedQFunction will be stored 501 502 @return An error code: 0 - success, otherwise - failure 503 504 @ref User 505 **/ 506 int CeedQFunctionCreateInteriorByName(Ceed ceed, const char *name, 507 CeedQFunction *qf) { 508 int ierr; 509 size_t match_len = 0, match_idx = UINT_MAX; 510 char *name_copy; 511 512 ierr = CeedQFunctionRegisterAll(); CeedChk(ierr); 513 // Find matching backend 514 if (!name) return CeedError(ceed, CEED_ERROR_INCOMPLETE, 515 "No QFunction name provided"); 516 for (size_t i=0; i<num_qfunctions; i++) { 517 size_t n; 518 const char *curr_name = gallery_qfunctions[i].name; 519 for (n = 0; curr_name[n] && curr_name[n] == name[n]; n++) {} 520 if (n > match_len) { 521 match_len = n; 522 match_idx = i; 523 } 524 } 525 if (!match_len) 526 // LCOV_EXCL_START 527 return CeedError(ceed, CEED_ERROR_UNSUPPORTED, "No suitable gallery QFunction"); 528 // LCOV_EXCL_STOP 529 530 // Create QFunction 531 ierr = CeedQFunctionCreateInterior(ceed, 532 gallery_qfunctions[match_idx].vec_length, 533 gallery_qfunctions[match_idx].f, 534 gallery_qfunctions[match_idx].source, qf); 535 CeedChk(ierr); 536 537 // QFunction specific setup 538 ierr = gallery_qfunctions[match_idx].init(ceed, name, *qf); CeedChk(ierr); 539 540 // Copy name 541 size_t slen = strlen(name) + 1; 542 ierr = CeedMalloc(slen, &name_copy); CeedChk(ierr); 543 memcpy(name_copy, name, slen); 544 (*qf)->qf_name = name_copy; 545 return CEED_ERROR_SUCCESS; 546 } 547 548 /** 549 @brief Create an identity CeedQFunction. Inputs are written into outputs in 550 the order given. This is useful for CeedOperators that can be 551 represented with only the action of a CeedRestriction and CeedBasis, 552 such as restriction and prolongation operators for p-multigrid. 553 Backends may optimize CeedOperators with this CeedQFunction to avoid 554 the copy of input data to output fields by using the same memory 555 location for both. 556 557 @param ceed A Ceed object where the CeedQFunction will be created 558 @param[in] size Size of the QFunction fields 559 @param[in] in_mode CeedEvalMode for input to CeedQFunction 560 @param[in] out_mode CeedEvalMode for output to CeedQFunction 561 @param[out] qf Address of the variable where the newly created 562 CeedQFunction will be stored 563 564 @return An error code: 0 - success, otherwise - failure 565 566 @ref User 567 **/ 568 int CeedQFunctionCreateIdentity(Ceed ceed, CeedInt size, CeedEvalMode in_mode, 569 CeedEvalMode out_mode, CeedQFunction *qf) { 570 int ierr; 571 572 ierr = CeedQFunctionCreateInteriorByName(ceed, "Identity", qf); CeedChk(ierr); 573 ierr = CeedQFunctionAddInput(*qf, "input", size, in_mode); CeedChk(ierr); 574 ierr = CeedQFunctionAddOutput(*qf, "output", size, out_mode); CeedChk(ierr); 575 576 (*qf)->identity = 1; 577 CeedInt *size_data; 578 ierr = CeedCalloc(1, &size_data); CeedChk(ierr); 579 size_data[0] = size; 580 CeedQFunctionContext ctx; 581 ierr = CeedQFunctionContextCreate(ceed, &ctx); CeedChk(ierr); 582 ierr = CeedQFunctionContextSetData(ctx, CEED_MEM_HOST, CEED_OWN_POINTER, 583 sizeof(*size_data), (void *)size_data); 584 CeedChk(ierr); 585 ierr = CeedQFunctionSetContext(*qf, ctx); CeedChk(ierr); 586 ierr = CeedQFunctionContextDestroy(&ctx); CeedChk(ierr); 587 return CEED_ERROR_SUCCESS; 588 } 589 590 /** 591 @brief Copy the pointer to a CeedQFunction. Both pointers should 592 be destroyed with `CeedQFunctionDestroy()`; 593 Note: If `*qf_copy` is non-NULL, then it is assumed that 594 `*qf_copy` is a pointer to a CeedQFunction. This 595 CeedQFunction will be destroyed if `*qf_copy` is the only 596 reference to this CeedQFunction. 597 598 @param qf CeedQFunction to copy reference to 599 @param[out] qf_copy Variable to store copied reference 600 601 @return An error code: 0 - success, otherwise - failure 602 603 @ref User 604 **/ 605 int CeedQFunctionReferenceCopy(CeedQFunction qf, CeedQFunction *qf_copy) { 606 int ierr; 607 608 ierr = CeedQFunctionReference(qf); CeedChk(ierr); 609 ierr = CeedQFunctionDestroy(qf_copy); CeedChk(ierr); 610 *qf_copy = qf; 611 return CEED_ERROR_SUCCESS; 612 } 613 614 /** 615 @brief Add a CeedQFunction input 616 617 @param qf CeedQFunction 618 @param field_name Name of QFunction field 619 @param size Size of QFunction field, (num_comp * dim) for @ref CEED_EVAL_GRAD or 620 (num_comp * 1) for @ref CEED_EVAL_NONE and @ref CEED_EVAL_INTERP 621 @param eval_mode \ref CEED_EVAL_NONE to use values directly, 622 \ref CEED_EVAL_INTERP to use interpolated values, 623 \ref CEED_EVAL_GRAD to use gradients. 624 625 @return An error code: 0 - success, otherwise - failure 626 627 @ref User 628 **/ 629 int CeedQFunctionAddInput(CeedQFunction qf, const char *field_name, 630 CeedInt size, 631 CeedEvalMode eval_mode) { 632 if (qf->operators_set) 633 // LCOV_EXCL_START 634 return CeedError(qf->ceed, CEED_ERROR_MAJOR, 635 "QFunction cannot be changed when in use by an operator"); 636 // LCOV_EXCL_STOP 637 if ((eval_mode == CEED_EVAL_WEIGHT) && (size != 1)) 638 // LCOV_EXCL_START 639 return CeedError(qf->ceed, CEED_ERROR_DIMENSION, 640 "CEED_EVAL_WEIGHT should have size 1"); 641 // LCOV_EXCL_STOP 642 int ierr = CeedQFunctionFieldSet(&qf->input_fields[qf->num_input_fields], 643 field_name, size, eval_mode); 644 CeedChk(ierr); 645 qf->num_input_fields++; 646 return CEED_ERROR_SUCCESS; 647 } 648 649 /** 650 @brief Add a CeedQFunction output 651 652 @param qf CeedQFunction 653 @param field_name Name of QFunction field 654 @param size Size of QFunction field, (num_comp * dim) for @ref CEED_EVAL_GRAD or 655 (num_comp * 1) for @ref CEED_EVAL_NONE and @ref CEED_EVAL_INTERP 656 @param eval_mode \ref CEED_EVAL_NONE to use values directly, 657 \ref CEED_EVAL_INTERP to use interpolated values, 658 \ref CEED_EVAL_GRAD to use gradients. 659 660 @return An error code: 0 - success, otherwise - failure 661 662 @ref User 663 **/ 664 int CeedQFunctionAddOutput(CeedQFunction qf, const char *field_name, 665 CeedInt size, CeedEvalMode eval_mode) { 666 if (qf->operators_set) 667 // LCOV_EXCL_START 668 return CeedError(qf->ceed, CEED_ERROR_MAJOR, 669 "QFunction cannot be changed when in use by an operator"); 670 // LCOV_EXCL_STOP 671 if (eval_mode == CEED_EVAL_WEIGHT) 672 // LCOV_EXCL_START 673 return CeedError(qf->ceed, CEED_ERROR_DIMENSION, 674 "Cannot create QFunction output with " 675 "CEED_EVAL_WEIGHT"); 676 // LCOV_EXCL_STOP 677 int ierr = CeedQFunctionFieldSet(&qf->output_fields[qf->num_output_fields], 678 field_name, size, eval_mode); 679 CeedChk(ierr); 680 qf->num_output_fields++; 681 return CEED_ERROR_SUCCESS; 682 } 683 684 /** 685 @brief Set global context for a CeedQFunction 686 687 @param qf CeedQFunction 688 @param ctx Context data to set 689 690 @return An error code: 0 - success, otherwise - failure 691 692 @ref User 693 **/ 694 int CeedQFunctionSetContext(CeedQFunction qf, CeedQFunctionContext ctx) { 695 int ierr; 696 qf->ctx = ctx; 697 ierr = CeedQFunctionContextReference(ctx); CeedChk(ierr); 698 return CEED_ERROR_SUCCESS; 699 } 700 701 /** 702 @brief View a CeedQFunction 703 704 @param[in] qf CeedQFunction to view 705 @param[in] stream Stream to write; typically stdout/stderr or a file 706 707 @return Error code: 0 - success, otherwise - failure 708 709 @ref User 710 **/ 711 int CeedQFunctionView(CeedQFunction qf, FILE *stream) { 712 int ierr; 713 714 fprintf(stream, "%sCeedQFunction %s\n", 715 qf->qf_name ? "Gallery " : "User ", qf->qf_name ? qf->qf_name : ""); 716 717 fprintf(stream, " %d Input Field%s:\n", qf->num_input_fields, 718 qf->num_input_fields>1 ? "s" : ""); 719 for (CeedInt i=0; i<qf->num_input_fields; i++) { 720 ierr = CeedQFunctionFieldView(qf->input_fields[i], i, 1, stream); 721 CeedChk(ierr); 722 } 723 724 fprintf(stream, " %d Output Field%s:\n", qf->num_output_fields, 725 qf->num_output_fields>1 ? "s" : ""); 726 for (CeedInt i=0; i<qf->num_output_fields; i++) { 727 ierr = CeedQFunctionFieldView(qf->output_fields[i], i, 0, stream); 728 CeedChk(ierr); 729 } 730 return CEED_ERROR_SUCCESS; 731 } 732 733 /** 734 @brief Apply the action of a CeedQFunction 735 736 @param qf CeedQFunction 737 @param Q Number of quadrature points 738 @param[in] u Array of input CeedVectors 739 @param[out] v Array of output CeedVectors 740 741 @return An error code: 0 - success, otherwise - failure 742 743 @ref User 744 **/ 745 int CeedQFunctionApply(CeedQFunction qf, CeedInt Q, 746 CeedVector *u, CeedVector *v) { 747 int ierr; 748 if (!qf->Apply) 749 // LCOV_EXCL_START 750 return CeedError(qf->ceed, CEED_ERROR_UNSUPPORTED, 751 "Backend does not support QFunctionApply"); 752 // LCOV_EXCL_STOP 753 if (Q % qf->vec_length) 754 // LCOV_EXCL_START 755 return CeedError(qf->ceed, CEED_ERROR_DIMENSION, 756 "Number of quadrature points %d must be a " 757 "multiple of %d", Q, qf->vec_length); 758 // LCOV_EXCL_STOP 759 ierr = qf->Apply(qf, Q, u, v); CeedChk(ierr); 760 return CEED_ERROR_SUCCESS; 761 } 762 763 /** 764 @brief Destroy a CeedQFunction 765 766 @param qf CeedQFunction to destroy 767 768 @return An error code: 0 - success, otherwise - failure 769 770 @ref User 771 **/ 772 int CeedQFunctionDestroy(CeedQFunction *qf) { 773 int ierr; 774 775 if (!*qf || --(*qf)->ref_count > 0) return CEED_ERROR_SUCCESS; 776 // Backend destroy 777 if ((*qf)->Destroy) { 778 ierr = (*qf)->Destroy(*qf); CeedChk(ierr); 779 } 780 // Free fields 781 for (int i=0; i<(*qf)->num_input_fields; i++) { 782 ierr = CeedFree(&(*(*qf)->input_fields[i]).field_name); CeedChk(ierr); 783 ierr = CeedFree(&(*qf)->input_fields[i]); CeedChk(ierr); 784 } 785 for (int i=0; i<(*qf)->num_output_fields; i++) { 786 ierr = CeedFree(&(*(*qf)->output_fields[i]).field_name); CeedChk(ierr); 787 ierr = CeedFree(&(*qf)->output_fields[i]); CeedChk(ierr); 788 } 789 ierr = CeedFree(&(*qf)->input_fields); CeedChk(ierr); 790 ierr = CeedFree(&(*qf)->output_fields); CeedChk(ierr); 791 792 // User context data object 793 ierr = CeedQFunctionContextDestroy(&(*qf)->ctx); CeedChk(ierr); 794 795 ierr = CeedFree(&(*qf)->source_path); CeedChk(ierr); 796 ierr = CeedFree(&(*qf)->qf_name); CeedChk(ierr); 797 ierr = CeedDestroy(&(*qf)->ceed); CeedChk(ierr); 798 ierr = CeedFree(qf); CeedChk(ierr); 799 return CEED_ERROR_SUCCESS; 800 } 801 802 /// @} 803