1*9ba83ac0SJeremy L Thompson // Copyright (c) 2017-2026, Lawrence Livermore National Security, LLC and other CEED contributors. 23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3c8817ca9Svaleriabarra // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 5c8817ca9Svaleriabarra // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 7c8817ca9Svaleriabarra 8d7b241e6Sjeremylt // Fortran interface 9d7b241e6Sjeremylt #include <ceed-fortran-name.h> 102b730f8bSJeremy L Thompson #include <ceed-impl.h> 1149aac155SJeremy L Thompson #include <ceed.h> 122b730f8bSJeremy L Thompson #include <ceed/backend.h> 133d576824SJeremy L Thompson #include <stdbool.h> 143d576824SJeremy L Thompson #include <stdint.h> 153d576824SJeremy L Thompson #include <stdio.h> 16d7b241e6Sjeremylt #include <string.h> 17d7b241e6Sjeremylt 18d7b241e6Sjeremylt #define FORTRAN_REQUEST_IMMEDIATE -1 19d7b241e6Sjeremylt #define FORTRAN_REQUEST_ORDERED -2 20d7b241e6Sjeremylt #define FORTRAN_NULL -3 2115910d16Sjeremylt #define FORTRAN_STRIDES_BACKEND -4 2215910d16Sjeremylt #define FORTRAN_VECTOR_ACTIVE -5 2315910d16Sjeremylt #define FORTRAN_VECTOR_NONE -6 2415910d16Sjeremylt #define FORTRAN_ELEMRESTRICTION_NONE -7 25a36217cbSJeremy L Thompson #define FORTRAN_BASIS_NONE -8 2615910d16Sjeremylt #define FORTRAN_QFUNCTION_NONE -9 27d7b241e6Sjeremylt 28d7b241e6Sjeremylt static Ceed *Ceed_dict = NULL; 29d7b241e6Sjeremylt static int Ceed_count = 0; 30d7b241e6Sjeremylt static int Ceed_n = 0; 31d7b241e6Sjeremylt static int Ceed_count_max = 0; 32d7b241e6Sjeremylt 33c2ae719dSJed Brown // This test should actually be for the gfortran version, but we don't currently 34c2ae719dSJed Brown // have a configure system to determine that (TODO). At present, this will use 35c2ae719dSJed Brown // the smaller integer when run with clang+gfortran=8, for example. (That is 36c2ae719dSJed Brown // sketchy, but will likely work for users that don't have huge character 37c2ae719dSJed Brown // strings.) 38c2ae719dSJed Brown #if __GNUC__ >= 8 39c2ae719dSJed Brown typedef size_t fortran_charlen_t; 40c2ae719dSJed Brown #else 41c2ae719dSJed Brown typedef int fortran_charlen_t; 42c2ae719dSJed Brown #endif 43c2ae719dSJed Brown 44c2ae719dSJed Brown #define Splice(a, b) a##b 45c2ae719dSJed Brown 46c2ae719dSJed Brown // Fortran strings are generally unterminated and the length is passed as an 47c2ae719dSJed Brown // extra argument after all the normal arguments. Some compilers (I only know 48c2ae719dSJed Brown // of Windows) place the length argument immediately after the string parameter 49c2ae719dSJed Brown // (TODO). 50c2ae719dSJed Brown // 51c2ae719dSJed Brown // We can't just NULL-terminate the string in-place because that could overwrite 52c2ae719dSJed Brown // other strings or attempt to write to read-only memory. This macro allocates 53c2ae719dSJed Brown // a string to hold the null-terminated version of the string that C expects. 54c2ae719dSJed Brown #define FIX_STRING(stringname) \ 55c2ae719dSJed Brown char Splice(stringname, _c)[1024]; \ 562b730f8bSJeremy L Thompson if (Splice(stringname, _len) > 1023) *err = CeedError(NULL, 1, "Fortran string length too long %zd", (size_t)Splice(stringname, _len)); \ 57c2ae719dSJed Brown strncpy(Splice(stringname, _c), stringname, Splice(stringname, _len)); \ 582b730f8bSJeremy L Thompson Splice(stringname, _c)[Splice(stringname, _len)] = 0; 59c2ae719dSJed Brown 60b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 61b00b43a6SJeremy L Thompson // Ceed 62b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 63d7b241e6Sjeremylt #define fCeedInit FORTRAN_NAME(ceedinit, CEEDINIT) 642b730f8bSJeremy L Thompson CEED_EXTERN void fCeedInit(const char *resource, int *ceed, int *err, fortran_charlen_t resource_len) { 65c2ae719dSJed Brown FIX_STRING(resource); 66d7b241e6Sjeremylt if (Ceed_count == Ceed_count_max) { 67d7b241e6Sjeremylt Ceed_count_max += Ceed_count_max / 2 + 1; 68d7b241e6Sjeremylt CeedRealloc(Ceed_count_max, &Ceed_dict); 69d7b241e6Sjeremylt } 70d7b241e6Sjeremylt 71d7b241e6Sjeremylt Ceed *ceed_ = &Ceed_dict[Ceed_count]; 72c2ae719dSJed Brown *err = CeedInit(resource_c, ceed_); 73d7b241e6Sjeremylt 74d7b241e6Sjeremylt if (*err == 0) { 75d7b241e6Sjeremylt *ceed = Ceed_count++; 76d7b241e6Sjeremylt Ceed_n++; 77d7b241e6Sjeremylt } 78d7b241e6Sjeremylt } 79d7b241e6Sjeremylt 802b730f8bSJeremy L Thompson #define fCeedIsDeterministic FORTRAN_NAME(ceedisdeterministic, CEEDISDETERMINISTIC) 812b730f8bSJeremy L Thompson CEED_EXTERN void fCeedIsDeterministic(int *ceed, int *is_deterministic, int *err) { 82d1d35e2fSjeremylt *err = CeedIsDeterministic(Ceed_dict[*ceed], (bool *)is_deterministic); 839525855cSJeremy L Thompson } 849525855cSJeremy L Thompson 852b730f8bSJeremy L Thompson #define fCeedGetPreferredMemType FORTRAN_NAME(ceedgetpreferredmemtype, CEEDGETPREFERREDMEMTYPE) 862b730f8bSJeremy L Thompson CEED_EXTERN void fCeedGetPreferredMemType(int *ceed, int *type, int *err) { *err = CeedGetPreferredMemType(Ceed_dict[*ceed], (CeedMemType *)type); } 87c907536fSjeremylt 8813a7d540SJeremy L Thompson #define fCeedSetNumViewTabs FORTRAN_NAME(ceedsetnumviewtabs, CEEDSETNUMVIEWTABS) 8913a7d540SJeremy L Thompson CEED_EXTERN void fCeedSetNumViewTabs(int *ceed, int *num_tabs, int *err) { *err = CeedSetNumViewTabs(Ceed_dict[*ceed], *num_tabs); } 9013a7d540SJeremy L Thompson 910a0da059Sjeremylt #define fCeedView FORTRAN_NAME(ceedview, CEEDVIEW) 922b730f8bSJeremy L Thompson CEED_EXTERN void fCeedView(int *ceed, int *err) { *err = CeedView(Ceed_dict[*ceed], stdout); } 930a0da059Sjeremylt 94d7b241e6Sjeremylt #define fCeedDestroy FORTRAN_NAME(ceeddestroy, CEEDDESTROY) 953b271f31SJed Brown CEED_EXTERN void fCeedDestroy(int *ceed, int *err) { 96b22696a8SJeremy L Thompson if (*ceed == FORTRAN_NULL) return; 97d7b241e6Sjeremylt *err = CeedDestroy(&Ceed_dict[*ceed]); 98d7b241e6Sjeremylt 99d7b241e6Sjeremylt if (*err == 0) { 10016595307SJeremy L Thompson *ceed = FORTRAN_NULL; 101d7b241e6Sjeremylt Ceed_n--; 102d7b241e6Sjeremylt if (Ceed_n == 0) { 103d7b241e6Sjeremylt CeedFree(&Ceed_dict); 104d7b241e6Sjeremylt Ceed_count = 0; 105d7b241e6Sjeremylt Ceed_count_max = 0; 106d7b241e6Sjeremylt } 107d7b241e6Sjeremylt } 108d7b241e6Sjeremylt } 109d7b241e6Sjeremylt 110b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 111b00b43a6SJeremy L Thompson // CeedVector 112b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 113d7b241e6Sjeremylt static CeedVector *CeedVector_dict = NULL; 114d7b241e6Sjeremylt static int CeedVector_count = 0; 115d7b241e6Sjeremylt static int CeedVector_n = 0; 116d7b241e6Sjeremylt static int CeedVector_count_max = 0; 117d7b241e6Sjeremylt 118d7b241e6Sjeremylt #define fCeedVectorCreate FORTRAN_NAME(ceedvectorcreate, CEEDVECTORCREATE) 1193b271f31SJed Brown CEED_EXTERN void fCeedVectorCreate(int *ceed, int *length, int *vec, int *err) { 120d7b241e6Sjeremylt if (CeedVector_count == CeedVector_count_max) { 121d7b241e6Sjeremylt CeedVector_count_max += CeedVector_count_max / 2 + 1; 122d7b241e6Sjeremylt CeedRealloc(CeedVector_count_max, &CeedVector_dict); 123d7b241e6Sjeremylt } 124d7b241e6Sjeremylt 125d7b241e6Sjeremylt CeedVector *vec_ = &CeedVector_dict[CeedVector_count]; 126d7b241e6Sjeremylt *err = CeedVectorCreate(Ceed_dict[*ceed], *length, vec_); 127d7b241e6Sjeremylt 128d7b241e6Sjeremylt if (*err == 0) { 129d7b241e6Sjeremylt *vec = CeedVector_count++; 130d7b241e6Sjeremylt CeedVector_n++; 131d7b241e6Sjeremylt } 132d7b241e6Sjeremylt } 133d7b241e6Sjeremylt 134d7b241e6Sjeremylt #define fCeedVectorSetArray FORTRAN_NAME(ceedvectorsetarray, CEEDVECTORSETARRAY) 1352b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorSetArray(int *vec, int *memtype, int *copymode, CeedScalar *array, int64_t *offset, int *err) { 1362b730f8bSJeremy L Thompson *err = CeedVectorSetArray(CeedVector_dict[*vec], (CeedMemType)*memtype, (CeedCopyMode)*copymode, (CeedScalar *)(array + *offset)); 137d7b241e6Sjeremylt } 138d7b241e6Sjeremylt 1396a6c615bSJeremy L Thompson #define fCeedVectorTakeArray FORTRAN_NAME(ceedvectortakearray, CEEDVECTORTAKEARRAY) 1402b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorTakeArray(int *vec, int *memtype, CeedScalar *array, int64_t *offset, int *err) { 1416a6c615bSJeremy L Thompson CeedScalar *b; 1426a6c615bSJeremy L Thompson CeedVector vec_ = CeedVector_dict[*vec]; 1436a6c615bSJeremy L Thompson *err = CeedVectorTakeArray(vec_, (CeedMemType)*memtype, &b); 1446a6c615bSJeremy L Thompson *offset = b - array; 1456a6c615bSJeremy L Thompson } 1466a6c615bSJeremy L Thompson 14754540941SJeremy L Thompson #define fCeedVectorSyncArray FORTRAN_NAME(ceedvectorsyncarray, CEEDVECTORSYNCARRAY) 1482b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorSyncArray(int *vec, int *memtype, int *err) { *err = CeedVectorSyncArray(CeedVector_dict[*vec], (CeedMemType)*memtype); } 14954540941SJeremy L Thompson 150d7b241e6Sjeremylt #define fCeedVectorSetValue FORTRAN_NAME(ceedvectorsetvalue, CEEDVECTORSETVALUE) 1512b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorSetValue(int *vec, CeedScalar *value, int *err) { *err = CeedVectorSetValue(CeedVector_dict[*vec], *value); } 152d7b241e6Sjeremylt 153d7b241e6Sjeremylt #define fCeedVectorGetArray FORTRAN_NAME(ceedvectorgetarray, CEEDVECTORGETARRAY) 1542b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorGetArray(int *vec, int *memtype, CeedScalar *array, int64_t *offset, int *err) { 155d7b241e6Sjeremylt CeedScalar *b; 156d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 157399bb197SJed Brown *err = CeedVectorGetArray(vec_, (CeedMemType)*memtype, &b); 158cbfd6a90SThilina Rathnayake *offset = b - array; 159d7b241e6Sjeremylt } 160d7b241e6Sjeremylt 1612b730f8bSJeremy L Thompson #define fCeedVectorGetArrayRead FORTRAN_NAME(ceedvectorgetarrayread, CEEDVECTORGETARRAYREAD) 1622b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorGetArrayRead(int *vec, int *memtype, CeedScalar *array, int64_t *offset, int *err) { 163d7b241e6Sjeremylt const CeedScalar *b; 164d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 165399bb197SJed Brown *err = CeedVectorGetArrayRead(vec_, (CeedMemType)*memtype, &b); 166cbfd6a90SThilina Rathnayake *offset = b - array; 167d7b241e6Sjeremylt } 168d7b241e6Sjeremylt 1692b730f8bSJeremy L Thompson #define fCeedVectorGetArrayWrite FORTRAN_NAME(ceedvectorgetarraywrite, CEEDVECTORGETARRAYWRITE) 1702b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorGetArrayWrite(int *vec, int *memtype, CeedScalar *array, int64_t *offset, int *err) { 1719c774eddSJeremy L Thompson CeedScalar *b; 1729c774eddSJeremy L Thompson CeedVector vec_ = CeedVector_dict[*vec]; 1739c774eddSJeremy L Thompson *err = CeedVectorGetArrayWrite(vec_, (CeedMemType)*memtype, &b); 1749c774eddSJeremy L Thompson *offset = b - array; 1759c774eddSJeremy L Thompson } 1769c774eddSJeremy L Thompson 1772b730f8bSJeremy L Thompson #define fCeedVectorRestoreArray FORTRAN_NAME(ceedvectorrestorearray, CEEDVECTORRESTOREARRAY) 1782b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorRestoreArray(int *vec, CeedScalar *array, int64_t *offset, int *err) { 1799647a07eSDavid Medina CeedScalar *offsetArray = array + *offset; 1809647a07eSDavid Medina *err = CeedVectorRestoreArray(CeedVector_dict[*vec], &offsetArray); 1816456524eSJed Brown *offset = 0; 182d7b241e6Sjeremylt } 183d7b241e6Sjeremylt 1842b730f8bSJeremy L Thompson #define fCeedVectorRestoreArrayRead FORTRAN_NAME(ceedvectorrestorearrayread, CEEDVECTORRESTOREARRAYREAD) 1852b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorRestoreArrayRead(int *vec, const CeedScalar *array, int64_t *offset, int *err) { 186d7b241e6Sjeremylt *err = CeedVectorRestoreArrayRead(CeedVector_dict[*vec], &array); 1876456524eSJed Brown *offset = 0; 188d7b241e6Sjeremylt } 189d7b241e6Sjeremylt 1902b730f8bSJeremy L Thompson #define fCeedVectorNorm FORTRAN_NAME(ceedvectornorm, CEEDVECTORNORM) 1912b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorNorm(int *vec, int *type, CeedScalar *norm, int *err) { 192399bb197SJed Brown *err = CeedVectorNorm(CeedVector_dict[*vec], (CeedNormType)*type, norm); 193547d9b97Sjeremylt } 194547d9b97Sjeremylt 1952b730f8bSJeremy L Thompson #define fCeedVectorReciprocal FORTRAN_NAME(ceedvectorreciprocal, CEEDVECTORRECIPROCAL) 1962b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorReciprocal(int *vec, int *err) { *err = CeedVectorReciprocal(CeedVector_dict[*vec]); } 197d99fa3c5SJeremy L Thompson 19813a7d540SJeremy L Thompson #define fCeedVectorSetNumViewTabs FORTRAN_NAME(ceedvectorsetnumviewtabs, CEEDVECTORSETNUMVIEWTABS) 19913a7d540SJeremy L Thompson CEED_EXTERN void fCeedVectorSetNumViewTabs(int *vec, int *num_tabs, int *err) { *err = CeedVectorSetNumViewTabs(CeedVector_dict[*vec], *num_tabs); } 20013a7d540SJeremy L Thompson 201d7b241e6Sjeremylt #define fCeedVectorView FORTRAN_NAME(ceedvectorview, CEEDVECTORVIEW) 2022b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorView(int *vec, int *err) { *err = CeedVectorView(CeedVector_dict[*vec], "%12.8f", stdout); } 203d7b241e6Sjeremylt 204d7b241e6Sjeremylt #define fCeedVectorDestroy FORTRAN_NAME(ceedvectordestroy, CEEDVECTORDESTROY) 2053b271f31SJed Brown CEED_EXTERN void fCeedVectorDestroy(int *vec, int *err) { 206b22696a8SJeremy L Thompson if (*vec == FORTRAN_NULL) return; 207d7b241e6Sjeremylt *err = CeedVectorDestroy(&CeedVector_dict[*vec]); 208d7b241e6Sjeremylt 209d7b241e6Sjeremylt if (*err == 0) { 21016595307SJeremy L Thompson *vec = FORTRAN_NULL; 211d7b241e6Sjeremylt CeedVector_n--; 212d7b241e6Sjeremylt if (CeedVector_n == 0) { 213d7b241e6Sjeremylt CeedFree(&CeedVector_dict); 214d7b241e6Sjeremylt CeedVector_count = 0; 215d7b241e6Sjeremylt CeedVector_count_max = 0; 216d7b241e6Sjeremylt } 217d7b241e6Sjeremylt } 218d7b241e6Sjeremylt } 219d7b241e6Sjeremylt 220b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 221b00b43a6SJeremy L Thompson // CeedElemRestriction 222b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 223d7b241e6Sjeremylt static CeedElemRestriction *CeedElemRestriction_dict = NULL; 224d7b241e6Sjeremylt static int CeedElemRestriction_count = 0; 225d7b241e6Sjeremylt static int CeedElemRestriction_n = 0; 226d7b241e6Sjeremylt static int CeedElemRestriction_count_max = 0; 227d7b241e6Sjeremylt 2282b730f8bSJeremy L Thompson #define fCeedElemRestrictionCreate FORTRAN_NAME(ceedelemrestrictioncreate, CEEDELEMRESTRICTIONCREATE) 2292b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionCreate(int *ceed, int *nelements, int *esize, int *num_comp, int *comp_stride, int *lsize, int *memtype, 2302b730f8bSJeremy L Thompson int *copymode, const int *offsets, int *elemrestriction, int *err) { 231d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 232d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 233d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 234d7b241e6Sjeremylt } 235d7b241e6Sjeremylt 236d979a051Sjeremylt const int *offsets_ = offsets; 237d7b241e6Sjeremylt 2382b730f8bSJeremy L Thompson CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 2392b730f8bSJeremy L Thompson *err = CeedElemRestrictionCreate(Ceed_dict[*ceed], *nelements, *esize, *num_comp, *comp_stride, *lsize, (CeedMemType)*memtype, 2402b730f8bSJeremy L Thompson (CeedCopyMode)*copymode, offsets_, elemrestriction_); 241d7b241e6Sjeremylt 242d7b241e6Sjeremylt if (*err == 0) { 243d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 244d7b241e6Sjeremylt CeedElemRestriction_n++; 245d7b241e6Sjeremylt } 246d7b241e6Sjeremylt } 247d7b241e6Sjeremylt 248709403c1SSebastian Grimberg #define fCeedElemRestrictionCreateOriented FORTRAN_NAME(ceedelemrestrictioncreateoriented, CEEDELEMRESTRICTIONCREATEORIENTED) 249709403c1SSebastian Grimberg CEED_EXTERN void fCeedElemRestrictionCreateOriented(int *ceed, int *nelements, int *esize, int *num_comp, int *comp_stride, int *lsize, int *memtype, 250709403c1SSebastian Grimberg int *copymode, const int *offsets, const bool *orients, int *elemrestriction, int *err) { 251709403c1SSebastian Grimberg if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 252709403c1SSebastian Grimberg CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 253709403c1SSebastian Grimberg CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 254709403c1SSebastian Grimberg } 255709403c1SSebastian Grimberg 256709403c1SSebastian Grimberg const int *offsets_ = offsets; 257709403c1SSebastian Grimberg const bool *orients_ = orients; 258709403c1SSebastian Grimberg 259709403c1SSebastian Grimberg CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 260709403c1SSebastian Grimberg *err = CeedElemRestrictionCreateOriented(Ceed_dict[*ceed], *nelements, *esize, *num_comp, *comp_stride, *lsize, (CeedMemType)*memtype, 261709403c1SSebastian Grimberg (CeedCopyMode)*copymode, offsets_, orients_, elemrestriction_); 262709403c1SSebastian Grimberg 263709403c1SSebastian Grimberg if (*err == 0) { 264709403c1SSebastian Grimberg *elemrestriction = CeedElemRestriction_count++; 265709403c1SSebastian Grimberg CeedElemRestriction_n++; 266709403c1SSebastian Grimberg } 267709403c1SSebastian Grimberg } 268709403c1SSebastian Grimberg 269709403c1SSebastian Grimberg #define fCeedElemRestrictionCreateCurlOriented FORTRAN_NAME(ceedelemrestrictioncreatecurloriented, CEEDELEMRESTRICTIONCREATECURLORIENTED) 270709403c1SSebastian Grimberg CEED_EXTERN void fCeedElemRestrictionCreateCurlOriented(int *ceed, int *nelements, int *esize, int *num_comp, int *comp_stride, int *lsize, 271709403c1SSebastian Grimberg int *memtype, int *copymode, const int *offsets, const int8_t *curlorients, 272709403c1SSebastian Grimberg int *elemrestriction, int *err) { 273709403c1SSebastian Grimberg if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 274709403c1SSebastian Grimberg CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 275709403c1SSebastian Grimberg CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 276709403c1SSebastian Grimberg } 277709403c1SSebastian Grimberg 278709403c1SSebastian Grimberg const int *offsets_ = offsets; 279709403c1SSebastian Grimberg const int8_t *curlorients_ = curlorients; 280709403c1SSebastian Grimberg 281709403c1SSebastian Grimberg CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 282709403c1SSebastian Grimberg *err = CeedElemRestrictionCreateCurlOriented(Ceed_dict[*ceed], *nelements, *esize, *num_comp, *comp_stride, *lsize, (CeedMemType)*memtype, 283709403c1SSebastian Grimberg (CeedCopyMode)*copymode, offsets_, curlorients_, elemrestriction_); 284709403c1SSebastian Grimberg 285709403c1SSebastian Grimberg if (*err == 0) { 286709403c1SSebastian Grimberg *elemrestriction = CeedElemRestriction_count++; 287709403c1SSebastian Grimberg CeedElemRestriction_n++; 288709403c1SSebastian Grimberg } 289709403c1SSebastian Grimberg } 290709403c1SSebastian Grimberg 2912b730f8bSJeremy L Thompson #define fCeedElemRestrictionCreateStrided FORTRAN_NAME(ceedelemrestrictioncreatestrided, CEEDELEMRESTRICTIONCREATESTRIDED) 2922b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionCreateStrided(int *ceed, int *nelements, int *esize, int *num_comp, int *lsize, int *strides, 293d7b241e6Sjeremylt int *elemrestriction, int *err) { 294d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 295d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 296d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 297d7b241e6Sjeremylt } 298d7b241e6Sjeremylt 2992b730f8bSJeremy L Thompson CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 3002b730f8bSJeremy L Thompson *err = CeedElemRestrictionCreateStrided(Ceed_dict[*ceed], *nelements, *esize, *num_comp, *lsize, 3012b730f8bSJeremy L Thompson *strides == FORTRAN_STRIDES_BACKEND ? CEED_STRIDES_BACKEND : strides, elemrestriction_); 302709403c1SSebastian Grimberg 303d7b241e6Sjeremylt if (*err == 0) { 304d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 305d7b241e6Sjeremylt CeedElemRestriction_n++; 306d7b241e6Sjeremylt } 307d7b241e6Sjeremylt } 308d7b241e6Sjeremylt 3092b730f8bSJeremy L Thompson #define fCeedElemRestrictionCreateBlocked FORTRAN_NAME(ceedelemrestrictioncreateblocked, CEEDELEMRESTRICTIONCREATEBLOCKED) 3102b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionCreateBlocked(int *ceed, int *nelements, int *esize, int *blocksize, int *num_comp, int *comp_stride, int *lsize, 311709403c1SSebastian Grimberg int *memtype, int *copymode, const int *offsets, int *elemrestriction, int *err) { 312d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 313d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 314d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 315d7b241e6Sjeremylt } 316d7b241e6Sjeremylt 317709403c1SSebastian Grimberg const int *offsets_ = offsets; 318709403c1SSebastian Grimberg 3192b730f8bSJeremy L Thompson CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 320709403c1SSebastian Grimberg *err = CeedElemRestrictionCreateBlocked(Ceed_dict[*ceed], *nelements, *esize, *blocksize, *num_comp, *comp_stride, *lsize, (CeedMemType)*memtype, 321709403c1SSebastian Grimberg (CeedCopyMode)*copymode, offsets_, elemrestriction_); 322709403c1SSebastian Grimberg 323709403c1SSebastian Grimberg if (*err == 0) { 324709403c1SSebastian Grimberg *elemrestriction = CeedElemRestriction_count++; 325709403c1SSebastian Grimberg CeedElemRestriction_n++; 326709403c1SSebastian Grimberg } 327709403c1SSebastian Grimberg } 328709403c1SSebastian Grimberg 329709403c1SSebastian Grimberg #define fCeedElemRestrictionCreateBlockedOriented FORTRAN_NAME(ceedelemrestrictioncreateblockedoriented, CEEDELEMRESTRICTIONCREATEBLOCKEDORIENTED) 330709403c1SSebastian Grimberg CEED_EXTERN void fCeedElemRestrictionCreateBlockedOriented(int *ceed, int *nelements, int *esize, int *blocksize, int *num_comp, int *comp_stride, 331709403c1SSebastian Grimberg int *lsize, int *memtype, int *copymode, const int *offsets, const bool *orients, 332709403c1SSebastian Grimberg int *elemrestriction, int *err) { 333709403c1SSebastian Grimberg if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 334709403c1SSebastian Grimberg CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 335709403c1SSebastian Grimberg CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 336709403c1SSebastian Grimberg } 337709403c1SSebastian Grimberg 338709403c1SSebastian Grimberg const int *offsets_ = offsets; 339709403c1SSebastian Grimberg const bool *orients_ = orients; 340709403c1SSebastian Grimberg 341709403c1SSebastian Grimberg CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 342709403c1SSebastian Grimberg *err = CeedElemRestrictionCreateBlockedOriented(Ceed_dict[*ceed], *nelements, *esize, *blocksize, *num_comp, *comp_stride, *lsize, 343709403c1SSebastian Grimberg (CeedMemType)*memtype, (CeedCopyMode)*copymode, offsets_, orients_, elemrestriction_); 344709403c1SSebastian Grimberg 345709403c1SSebastian Grimberg if (*err == 0) { 346709403c1SSebastian Grimberg *elemrestriction = CeedElemRestriction_count++; 347709403c1SSebastian Grimberg CeedElemRestriction_n++; 348709403c1SSebastian Grimberg } 349709403c1SSebastian Grimberg } 350709403c1SSebastian Grimberg 351709403c1SSebastian Grimberg #define fCeedElemRestrictionCreateBlockedCurlOriented \ 352709403c1SSebastian Grimberg FORTRAN_NAME(ceedelemrestrictioncreateblockedcurloriented, CEEDELEMRESTRICTIONCREATEBLOCKEDCURLORIENTED) 353709403c1SSebastian Grimberg CEED_EXTERN void fCeedElemRestrictionCreateBlockedCurlOriented(int *ceed, int *nelements, int *esize, int *blocksize, int *num_comp, int *comp_stride, 354709403c1SSebastian Grimberg int *lsize, int *memtype, int *copymode, const int *offsets, const int8_t *curlorients, 355709403c1SSebastian Grimberg int *elemrestriction, int *err) { 356709403c1SSebastian Grimberg if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 357709403c1SSebastian Grimberg CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 358709403c1SSebastian Grimberg CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 359709403c1SSebastian Grimberg } 360709403c1SSebastian Grimberg 361709403c1SSebastian Grimberg const int *offsets_ = offsets; 362709403c1SSebastian Grimberg const int8_t *curlorients_ = curlorients; 363709403c1SSebastian Grimberg 364709403c1SSebastian Grimberg CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 365709403c1SSebastian Grimberg *err = CeedElemRestrictionCreateBlockedCurlOriented(Ceed_dict[*ceed], *nelements, *esize, *blocksize, *num_comp, *comp_stride, *lsize, 366709403c1SSebastian Grimberg (CeedMemType)*memtype, (CeedCopyMode)*copymode, offsets_, curlorients_, elemrestriction_); 367d7b241e6Sjeremylt 368d7b241e6Sjeremylt if (*err == 0) { 369d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 370d7b241e6Sjeremylt CeedElemRestriction_n++; 371d7b241e6Sjeremylt } 372d7b241e6Sjeremylt } 373d7b241e6Sjeremylt 3742b730f8bSJeremy L Thompson #define fCeedElemRestrictionCreateBlockedStrided FORTRAN_NAME(ceedelemrestrictioncreateblockedstrided, CEEDELEMRESTRICTIONCREATEBLOCKEDSTRIDED) 3752b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionCreateBlockedStrided(int *ceed, int *nelements, int *esize, int *blk_size, int *num_comp, int *lsize, 3762b730f8bSJeremy L Thompson int *strides, int *elemrestriction, int *err) { 3777509a596Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 3787509a596Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 3797509a596Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 3807509a596Sjeremylt } 3817509a596Sjeremylt 3822b730f8bSJeremy L Thompson CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 3832b730f8bSJeremy L Thompson *err = CeedElemRestrictionCreateBlockedStrided(Ceed_dict[*ceed], *nelements, *esize, *blk_size, *num_comp, *lsize, strides, elemrestriction_); 384709403c1SSebastian Grimberg 3857509a596Sjeremylt if (*err == 0) { 3867509a596Sjeremylt *elemrestriction = CeedElemRestriction_count++; 3877509a596Sjeremylt CeedElemRestriction_n++; 3887509a596Sjeremylt } 3897509a596Sjeremylt } 3907509a596Sjeremylt 391d7b241e6Sjeremylt static CeedRequest *CeedRequest_dict = NULL; 392d7b241e6Sjeremylt static int CeedRequest_count = 0; 393d7b241e6Sjeremylt static int CeedRequest_n = 0; 394d7b241e6Sjeremylt static int CeedRequest_count_max = 0; 395d7b241e6Sjeremylt 3962b730f8bSJeremy L Thompson #define fCeedElemRestrictionApply FORTRAN_NAME(ceedelemrestrictionapply, CEEDELEMRESTRICTIONAPPLY) 3972b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionApply(int *elemr, int *tmode, int *uvec, int *ruvec, int *rqst, int *err) { 398d7b241e6Sjeremylt int createRequest = 1; 399d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 4002b730f8bSJeremy L Thompson if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) createRequest = 0; 401d7b241e6Sjeremylt 402d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 403d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 404d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 405d7b241e6Sjeremylt } 406d7b241e6Sjeremylt 407d7b241e6Sjeremylt CeedRequest *rqst_; 408d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 409d7b241e6Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED) rqst_ = CEED_REQUEST_ORDERED; 410d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 411d7b241e6Sjeremylt 4122b730f8bSJeremy L Thompson *err = 4132b730f8bSJeremy L Thompson CeedElemRestrictionApply(CeedElemRestriction_dict[*elemr], (CeedTransposeMode)*tmode, CeedVector_dict[*uvec], CeedVector_dict[*ruvec], rqst_); 414d7b241e6Sjeremylt 415d7b241e6Sjeremylt if (*err == 0 && createRequest) { 416d7b241e6Sjeremylt *rqst = CeedRequest_count++; 417d7b241e6Sjeremylt CeedRequest_n++; 418d7b241e6Sjeremylt } 419d7b241e6Sjeremylt } 420d7b241e6Sjeremylt 4212b730f8bSJeremy L Thompson #define fCeedElemRestrictionApplyBlock FORTRAN_NAME(ceedelemrestrictionapplyblock, CEEDELEMRESTRICTIONAPPLYBLOCK) 4222b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionApplyBlock(int *elemr, int *block, int *tmode, int *uvec, int *ruvec, int *rqst, int *err) { 423be9261b7Sjeremylt int createRequest = 1; 424be9261b7Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 4252b730f8bSJeremy L Thompson if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) createRequest = 0; 426be9261b7Sjeremylt 427be9261b7Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 428be9261b7Sjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 429be9261b7Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 430be9261b7Sjeremylt } 431be9261b7Sjeremylt 432be9261b7Sjeremylt CeedRequest *rqst_; 433be9261b7Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 434be9261b7Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED) rqst_ = CEED_REQUEST_ORDERED; 435be9261b7Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 436be9261b7Sjeremylt 4372b730f8bSJeremy L Thompson *err = CeedElemRestrictionApplyBlock(CeedElemRestriction_dict[*elemr], *block, (CeedTransposeMode)*tmode, CeedVector_dict[*uvec], 438be9261b7Sjeremylt CeedVector_dict[*ruvec], rqst_); 439be9261b7Sjeremylt 440be9261b7Sjeremylt if (*err == 0 && createRequest) { 441be9261b7Sjeremylt *rqst = CeedRequest_count++; 442be9261b7Sjeremylt CeedRequest_n++; 443be9261b7Sjeremylt } 444be9261b7Sjeremylt } 445be9261b7Sjeremylt 4462b730f8bSJeremy L Thompson #define fCeedElemRestrictionGetMultiplicity FORTRAN_NAME(ceedelemrestrictiongetmultiplicity, CEEDELEMRESTRICTIONGETMULTIPLICITY) 4472b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionGetMultiplicity(int *elemr, int *mult, int *err) { 4482b730f8bSJeremy L Thompson *err = CeedElemRestrictionGetMultiplicity(CeedElemRestriction_dict[*elemr], CeedVector_dict[*mult]); 4491469ee4dSjeremylt } 4501469ee4dSjeremylt 4512b730f8bSJeremy L Thompson #define fCeedElemRestrictionGetELayout FORTRAN_NAME(ceedelemrestrictiongetelayout, CEEDELEMRESTRICTIONGETELAYOUT) 4522b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionGetELayout(int *elemr, int *layout, int *err) { 45349fd234cSJeremy L Thompson CeedInt layout_c[3]; 45456c48462SJeremy L Thompson *err = CeedElemRestrictionGetELayout(CeedElemRestriction_dict[*elemr], layout_c); 4552b730f8bSJeremy L Thompson for (int i = 0; i < 3; i++) layout[i] = layout_c[i]; 45649fd234cSJeremy L Thompson } 45749fd234cSJeremy L Thompson 45813a7d540SJeremy L Thompson #define fCeedElemRestrictionSetNumViewTabs FORTRAN_NAME(ceedelemrestrictionsetnumviewtabs, CEEDELEMRESTRICTIONSETNUMVIEWTABS) 45913a7d540SJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionSetNumViewTabs(int *elemr, int *num_tabs, int *err) { 46013a7d540SJeremy L Thompson *err = CeedElemRestrictionSetNumViewTabs(CeedElemRestriction_dict[*elemr], *num_tabs); 46113a7d540SJeremy L Thompson } 46213a7d540SJeremy L Thompson 4632b730f8bSJeremy L Thompson #define fCeedElemRestrictionView FORTRAN_NAME(ceedelemrestrictionview, CEEDELEMRESTRICTIONVIEW) 4642b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionView(int *elemr, int *err) { *err = CeedElemRestrictionView(CeedElemRestriction_dict[*elemr], stdout); } 4651469ee4dSjeremylt 466d7b241e6Sjeremylt #define fCeedRequestWait FORTRAN_NAME(ceedrequestwait, CEEDREQUESTWAIT) 4673b271f31SJed Brown CEED_EXTERN void fCeedRequestWait(int *rqst, int *err) { 468d7b241e6Sjeremylt // TODO Uncomment this once CeedRequestWait is implemented 469d7b241e6Sjeremylt //*err = CeedRequestWait(&CeedRequest_dict[*rqst]); 470d7b241e6Sjeremylt 471d7b241e6Sjeremylt if (*err == 0) { 472d7b241e6Sjeremylt CeedRequest_n--; 473d7b241e6Sjeremylt if (CeedRequest_n == 0) { 474d7b241e6Sjeremylt CeedFree(&CeedRequest_dict); 475d7b241e6Sjeremylt CeedRequest_count = 0; 476d7b241e6Sjeremylt CeedRequest_count_max = 0; 477d7b241e6Sjeremylt } 478d7b241e6Sjeremylt } 479d7b241e6Sjeremylt } 480d7b241e6Sjeremylt 4812b730f8bSJeremy L Thompson #define fCeedElemRestrictionDestroy FORTRAN_NAME(ceedelemrestrictiondestroy, CEEDELEMRESTRICTIONDESTROY) 4823b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionDestroy(int *elem, int *err) { 483b22696a8SJeremy L Thompson if (*elem == FORTRAN_NULL) return; 484d7b241e6Sjeremylt *err = CeedElemRestrictionDestroy(&CeedElemRestriction_dict[*elem]); 485d7b241e6Sjeremylt 486d7b241e6Sjeremylt if (*err == 0) { 48716595307SJeremy L Thompson *elem = FORTRAN_NULL; 488d7b241e6Sjeremylt CeedElemRestriction_n--; 489d7b241e6Sjeremylt if (CeedElemRestriction_n == 0) { 490d7b241e6Sjeremylt CeedFree(&CeedElemRestriction_dict); 491d7b241e6Sjeremylt CeedElemRestriction_count = 0; 492d7b241e6Sjeremylt CeedElemRestriction_count_max = 0; 493d7b241e6Sjeremylt } 494d7b241e6Sjeremylt } 495d7b241e6Sjeremylt } 496d7b241e6Sjeremylt 497b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 498b00b43a6SJeremy L Thompson // CeedBasis 499b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 500d7b241e6Sjeremylt static CeedBasis *CeedBasis_dict = NULL; 501d7b241e6Sjeremylt static int CeedBasis_count = 0; 502d7b241e6Sjeremylt static int CeedBasis_n = 0; 503d7b241e6Sjeremylt static int CeedBasis_count_max = 0; 504d7b241e6Sjeremylt 5052b730f8bSJeremy L Thompson #define fCeedBasisCreateTensorH1Lagrange FORTRAN_NAME(ceedbasiscreatetensorh1lagrange, CEEDBASISCREATETENSORH1LAGRANGE) 5062b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisCreateTensorH1Lagrange(int *ceed, int *dim, int *num_comp, int *P, int *Q, int *quadmode, int *basis, int *err) { 507d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 508d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max / 2 + 1; 509d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 510d7b241e6Sjeremylt } 511d7b241e6Sjeremylt 5122b730f8bSJeremy L Thompson *err = CeedBasisCreateTensorH1Lagrange(Ceed_dict[*ceed], *dim, *num_comp, *P, *Q, (CeedQuadMode)*quadmode, &CeedBasis_dict[CeedBasis_count]); 513d7b241e6Sjeremylt 514d7b241e6Sjeremylt if (*err == 0) { 515d7b241e6Sjeremylt *basis = CeedBasis_count++; 516d7b241e6Sjeremylt CeedBasis_n++; 517d7b241e6Sjeremylt } 518d7b241e6Sjeremylt } 519d7b241e6Sjeremylt 5202b730f8bSJeremy L Thompson #define fCeedBasisCreateTensorH1 FORTRAN_NAME(ceedbasiscreatetensorh1, CEEDBASISCREATETENSORH1) 5212b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisCreateTensorH1(int *ceed, int *dim, int *num_comp, int *P_1d, int *Q_1d, const CeedScalar *interp_1d, 5222b730f8bSJeremy L Thompson const CeedScalar *grad_1d, const CeedScalar *q_ref_1d, const CeedScalar *q_weight_1d, int *basis, 5238795c945Sjeremylt int *err) { 524d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 525d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max / 2 + 1; 526d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 527d7b241e6Sjeremylt } 528d7b241e6Sjeremylt 5292b730f8bSJeremy L Thompson *err = CeedBasisCreateTensorH1(Ceed_dict[*ceed], *dim, *num_comp, *P_1d, *Q_1d, interp_1d, grad_1d, q_ref_1d, q_weight_1d, 5308795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 531d7b241e6Sjeremylt 532d7b241e6Sjeremylt if (*err == 0) { 533d7b241e6Sjeremylt *basis = CeedBasis_count++; 534d7b241e6Sjeremylt CeedBasis_n++; 535d7b241e6Sjeremylt } 536d7b241e6Sjeremylt } 537d7b241e6Sjeremylt 5382b730f8bSJeremy L Thompson #define fCeedBasisCreateH1 FORTRAN_NAME(ceedbasiscreateh1, CEEDBASISCREATEH1) 5392b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisCreateH1(int *ceed, int *topo, int *num_comp, int *nnodes, int *nqpts, const CeedScalar *interp, const CeedScalar *grad, 5402b730f8bSJeremy L Thompson const CeedScalar *qref, const CeedScalar *qweight, int *basis, int *err) { 541a8de75f0Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 542a8de75f0Sjeremylt CeedBasis_count_max += CeedBasis_count_max / 2 + 1; 543a8de75f0Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 544a8de75f0Sjeremylt } 545a8de75f0Sjeremylt 5462b730f8bSJeremy L Thompson *err = CeedBasisCreateH1(Ceed_dict[*ceed], (CeedElemTopology)*topo, *num_comp, *nnodes, *nqpts, interp, grad, qref, qweight, 5478795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 548a8de75f0Sjeremylt 549a8de75f0Sjeremylt if (*err == 0) { 550a8de75f0Sjeremylt *basis = CeedBasis_count++; 551a8de75f0Sjeremylt CeedBasis_n++; 552a8de75f0Sjeremylt } 553a8de75f0Sjeremylt } 554a8de75f0Sjeremylt 55511b88ddaSSebastian Grimberg #define fCeedBasisCreateHdiv FORTRAN_NAME(ceedbasiscreatehdiv, CEEDBASISCREATEHDIV) 55611b88ddaSSebastian Grimberg CEED_EXTERN void fCeedBasisCreateHdiv(int *ceed, int *topo, int *num_comp, int *nnodes, int *nqpts, const CeedScalar *interp, const CeedScalar *div, 55711b88ddaSSebastian Grimberg const CeedScalar *qref, const CeedScalar *qweight, int *basis, int *err) { 55811b88ddaSSebastian Grimberg if (CeedBasis_count == CeedBasis_count_max) { 55911b88ddaSSebastian Grimberg CeedBasis_count_max += CeedBasis_count_max / 2 + 1; 56011b88ddaSSebastian Grimberg CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 56111b88ddaSSebastian Grimberg } 56211b88ddaSSebastian Grimberg 56311b88ddaSSebastian Grimberg *err = CeedBasisCreateHdiv(Ceed_dict[*ceed], (CeedElemTopology)*topo, *num_comp, *nnodes, *nqpts, interp, div, qref, qweight, 56411b88ddaSSebastian Grimberg &CeedBasis_dict[CeedBasis_count]); 56511b88ddaSSebastian Grimberg 56611b88ddaSSebastian Grimberg if (*err == 0) { 56711b88ddaSSebastian Grimberg *basis = CeedBasis_count++; 56811b88ddaSSebastian Grimberg CeedBasis_n++; 56911b88ddaSSebastian Grimberg } 57011b88ddaSSebastian Grimberg } 57111b88ddaSSebastian Grimberg 57211b88ddaSSebastian Grimberg #define fCeedBasisCreateHcurl FORTRAN_NAME(ceedbasiscreatehcurl, CEEDBASISCREATEHCURL) 57311b88ddaSSebastian Grimberg CEED_EXTERN void fCeedBasisCreateHcurl(int *ceed, int *topo, int *num_comp, int *nnodes, int *nqpts, const CeedScalar *interp, const CeedScalar *curl, 57411b88ddaSSebastian Grimberg const CeedScalar *qref, const CeedScalar *qweight, int *basis, int *err) { 57511b88ddaSSebastian Grimberg if (CeedBasis_count == CeedBasis_count_max) { 57611b88ddaSSebastian Grimberg CeedBasis_count_max += CeedBasis_count_max / 2 + 1; 57711b88ddaSSebastian Grimberg CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 57811b88ddaSSebastian Grimberg } 57911b88ddaSSebastian Grimberg 58011b88ddaSSebastian Grimberg *err = CeedBasisCreateHcurl(Ceed_dict[*ceed], (CeedElemTopology)*topo, *num_comp, *nnodes, *nqpts, interp, curl, qref, qweight, 58111b88ddaSSebastian Grimberg &CeedBasis_dict[CeedBasis_count]); 58211b88ddaSSebastian Grimberg 58311b88ddaSSebastian Grimberg if (*err == 0) { 58411b88ddaSSebastian Grimberg *basis = CeedBasis_count++; 58511b88ddaSSebastian Grimberg CeedBasis_n++; 58611b88ddaSSebastian Grimberg } 58711b88ddaSSebastian Grimberg } 58811b88ddaSSebastian Grimberg 58913a7d540SJeremy L Thompson #define fCeedBasisSetNumViewTabs FORTRAN_NAME(ceedbasissetnumviewtabs, CEEDBASISSETNUMVIEWTABS) 59013a7d540SJeremy L Thompson CEED_EXTERN void fCeedBasisSetNumViewTabs(int *basis, int *num_tabs, int *err) { *err = CeedBasisSetNumViewTabs(CeedBasis_dict[*basis], *num_tabs); } 59113a7d540SJeremy L Thompson 592d7b241e6Sjeremylt #define fCeedBasisView FORTRAN_NAME(ceedbasisview, CEEDBASISVIEW) 5932b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisView(int *basis, int *err) { *err = CeedBasisView(CeedBasis_dict[*basis], stdout); } 594d7b241e6Sjeremylt 5952b730f8bSJeremy L Thompson #define fCeedBasisGetCollocatedGrad FORTRAN_NAME(ceedbasisgetcollocatedgrad, CEEDBASISGETCOLLOCATEDGRAD) 5962b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetCollocatedGrad(int *basis, CeedScalar *colo_grad_1d, int *err) { 597d1d35e2fSjeremylt *err = CeedBasisGetCollocatedGrad(CeedBasis_dict[*basis], colo_grad_1d); 598d7b241e6Sjeremylt } 599d7b241e6Sjeremylt 600d7b241e6Sjeremylt #define fCeedBasisApply FORTRAN_NAME(ceedbasisapply, CEEDBASISAPPLY) 6012b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisApply(int *basis, int *num_elem, int *tmode, int *eval_mode, int *u, int *v, int *err) { 6022b730f8bSJeremy L Thompson *err = CeedBasisApply(CeedBasis_dict[*basis], *num_elem, (CeedTransposeMode)*tmode, (CeedEvalMode)*eval_mode, 6032b730f8bSJeremy L Thompson *u == FORTRAN_VECTOR_NONE ? CEED_VECTOR_NONE : CeedVector_dict[*u], CeedVector_dict[*v]); 604d7b241e6Sjeremylt } 605d7b241e6Sjeremylt 6062b730f8bSJeremy L Thompson #define fCeedBasisGetNumNodes FORTRAN_NAME(ceedbasisgetnumnodes, CEEDBASISGETNUMNODES) 6072b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetNumNodes(int *basis, int *P, int *err) { *err = CeedBasisGetNumNodes(CeedBasis_dict[*basis], P); } 608d7b241e6Sjeremylt 6092b730f8bSJeremy L Thompson #define fCeedBasisGetNumQuadraturePoints FORTRAN_NAME(ceedbasisgetnumquadraturepoints, CEEDBASISGETNUMQUADRATUREPOINTS) 6102b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetNumQuadraturePoints(int *basis, int *Q, int *err) { *err = CeedBasisGetNumQuadraturePoints(CeedBasis_dict[*basis], Q); } 611d7b241e6Sjeremylt 6122b730f8bSJeremy L Thompson #define fCeedBasisGetInterp1D FORTRAN_NAME(ceedbasisgetinterp1d, CEEDBASISGETINTERP1D) 6132b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetInterp1D(int *basis, CeedScalar *interp_1d, int64_t *offset, int *err) { 614d99fa3c5SJeremy L Thompson const CeedScalar *interp1d_; 615d99fa3c5SJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 616d99fa3c5SJeremy L Thompson *err = CeedBasisGetInterp1D(basis_, &interp1d_); 617d1d35e2fSjeremylt *offset = interp1d_ - interp_1d; 618d99fa3c5SJeremy L Thompson } 619d99fa3c5SJeremy L Thompson 6202b730f8bSJeremy L Thompson #define fCeedBasisGetGrad1D FORTRAN_NAME(ceedbasisgetgrad1d, CEEDBASISGETGRAD1D) 6212b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetGrad1D(int *basis, CeedScalar *grad_1d, int64_t *offset, int *err) { 6229ac7b42eSJeremy L Thompson const CeedScalar *grad1d_; 6239ac7b42eSJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 6249ac7b42eSJeremy L Thompson *err = CeedBasisGetGrad1D(basis_, &grad1d_); 6259ac7b42eSJeremy L Thompson *offset = grad1d_ - grad_1d; 6269ac7b42eSJeremy L Thompson } 6279ac7b42eSJeremy L Thompson 6282b730f8bSJeremy L Thompson #define fCeedBasisGetQRef FORTRAN_NAME(ceedbasisgetqref, CEEDBASISGETQREF) 6292b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetQRef(int *basis, CeedScalar *q_ref, int64_t *offset, int *err) { 6309ac7b42eSJeremy L Thompson const CeedScalar *qref_; 6319ac7b42eSJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 6329ac7b42eSJeremy L Thompson *err = CeedBasisGetQRef(basis_, &qref_); 6339ac7b42eSJeremy L Thompson *offset = qref_ - q_ref; 6349ac7b42eSJeremy L Thompson } 6359ac7b42eSJeremy L Thompson 636d7b241e6Sjeremylt #define fCeedBasisDestroy FORTRAN_NAME(ceedbasisdestroy, CEEDBASISDESTROY) 6373b271f31SJed Brown CEED_EXTERN void fCeedBasisDestroy(int *basis, int *err) { 638b22696a8SJeremy L Thompson if (*basis == FORTRAN_NULL) return; 639d7b241e6Sjeremylt *err = CeedBasisDestroy(&CeedBasis_dict[*basis]); 640d7b241e6Sjeremylt 641d7b241e6Sjeremylt if (*err == 0) { 64216595307SJeremy L Thompson *basis = FORTRAN_NULL; 643d7b241e6Sjeremylt CeedBasis_n--; 644d7b241e6Sjeremylt if (CeedBasis_n == 0) { 645d7b241e6Sjeremylt CeedFree(&CeedBasis_dict); 646d7b241e6Sjeremylt CeedBasis_count = 0; 647d7b241e6Sjeremylt CeedBasis_count_max = 0; 648d7b241e6Sjeremylt } 649d7b241e6Sjeremylt } 650d7b241e6Sjeremylt } 651d7b241e6Sjeremylt 652d7b241e6Sjeremylt #define fCeedGaussQuadrature FORTRAN_NAME(ceedgaussquadrature, CEEDGAUSSQUADRATURE) 6532b730f8bSJeremy L Thompson CEED_EXTERN void fCeedGaussQuadrature(int *Q, CeedScalar *q_ref_1d, CeedScalar *q_weight_1d, int *err) { 654d1d35e2fSjeremylt *err = CeedGaussQuadrature(*Q, q_ref_1d, q_weight_1d); 655d7b241e6Sjeremylt } 656d7b241e6Sjeremylt 6572b730f8bSJeremy L Thompson #define fCeedLobattoQuadrature FORTRAN_NAME(ceedlobattoquadrature, CEEDLOBATTOQUADRATURE) 6582b730f8bSJeremy L Thompson CEED_EXTERN void fCeedLobattoQuadrature(int *Q, CeedScalar *q_ref_1d, CeedScalar *q_weight_1d, int *err) { 659d1d35e2fSjeremylt *err = CeedLobattoQuadrature(*Q, q_ref_1d, q_weight_1d); 660d7b241e6Sjeremylt } 661d7b241e6Sjeremylt 662b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 663777ff853SJeremy L Thompson // CeedQFunctionContext 664777ff853SJeremy L Thompson // ----------------------------------------------------------------------------- 665777ff853SJeremy L Thompson static CeedQFunctionContext *CeedQFunctionContext_dict = NULL; 666777ff853SJeremy L Thompson static int CeedQFunctionContext_count = 0; 667777ff853SJeremy L Thompson static int CeedQFunctionContext_n = 0; 668777ff853SJeremy L Thompson static int CeedQFunctionContext_count_max = 0; 669777ff853SJeremy L Thompson 6702b730f8bSJeremy L Thompson #define fCeedQFunctionContextCreate FORTRAN_NAME(ceedqfunctioncontextcreate, CEEDQFUNCTIONCONTEXTCREATE) 6713b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextCreate(int *ceed, int *ctx, int *err) { 672777ff853SJeremy L Thompson if (CeedQFunctionContext_count == CeedQFunctionContext_count_max) { 673777ff853SJeremy L Thompson CeedQFunctionContext_count_max += CeedQFunctionContext_count_max / 2 + 1; 674777ff853SJeremy L Thompson CeedRealloc(CeedQFunctionContext_count_max, &CeedQFunctionContext_dict); 675777ff853SJeremy L Thompson } 676777ff853SJeremy L Thompson 6772b730f8bSJeremy L Thompson CeedQFunctionContext *ctx_ = &CeedQFunctionContext_dict[CeedQFunctionContext_count]; 678777ff853SJeremy L Thompson 679777ff853SJeremy L Thompson *err = CeedQFunctionContextCreate(Ceed_dict[*ceed], ctx_); 680777ff853SJeremy L Thompson if (*err) return; 681777ff853SJeremy L Thompson *ctx = CeedQFunctionContext_count++; 682777ff853SJeremy L Thompson CeedQFunctionContext_n++; 683777ff853SJeremy L Thompson } 684777ff853SJeremy L Thompson 6852b730f8bSJeremy L Thompson #define fCeedQFunctionContextSetData FORTRAN_NAME(ceedqfunctioncontextsetdata, CEEDQFUNCTIONCONTEXTSETDATA) 6862b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextSetData(int *ctx, int *memtype, int *copymode, CeedInt *n, CeedScalar *data, int64_t *offset, int *err) { 687d1d35e2fSjeremylt size_t ctx_size = ((size_t)*n) * sizeof(CeedScalar); 6882b730f8bSJeremy L Thompson *err = CeedQFunctionContextSetData(CeedQFunctionContext_dict[*ctx], (CeedMemType)*memtype, (CeedCopyMode)*copymode, ctx_size, data + *offset); 689777ff853SJeremy L Thompson } 690777ff853SJeremy L Thompson 6912b730f8bSJeremy L Thompson #define fCeedQFunctionContextGetData FORTRAN_NAME(ceedqfunctioncontextgetdata, CEEDQFUNCTIONCONTEXTGETDATA) 6922b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextGetData(int *ctx, int *memtype, CeedScalar *data, int64_t *offset, int *err) { 693777ff853SJeremy L Thompson CeedScalar *b; 694777ff853SJeremy L Thompson CeedQFunctionContext ctx_ = CeedQFunctionContext_dict[*ctx]; 695777ff853SJeremy L Thompson *err = CeedQFunctionContextGetData(ctx_, (CeedMemType)*memtype, &b); 696777ff853SJeremy L Thompson *offset = b - data; 697777ff853SJeremy L Thompson } 698777ff853SJeremy L Thompson 6992b730f8bSJeremy L Thompson #define fCeedQFunctionContextRestoreData FORTRAN_NAME(ceedqfunctioncontextrestoredata, CEEDQFUNCTIONCONTEXTRESTOREDATA) 7002b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextRestoreData(int *ctx, CeedScalar *data, int64_t *offset, int *err) { 7012b730f8bSJeremy L Thompson *err = CeedQFunctionContextRestoreData(CeedQFunctionContext_dict[*ctx], (void **)&data); 702777ff853SJeremy L Thompson *offset = 0; 703777ff853SJeremy L Thompson } 704777ff853SJeremy L Thompson 70513a7d540SJeremy L Thompson #define fCeedQFunctionContextSetNumViewTabs FORTRAN_NAME(ceedqfunctioncontextsetnumviewtabs, CEEDQFUNCTIONCONTEXTSETNUMVIEWTABS) 70613a7d540SJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextSetNumViewTabs(int *ctx, int *num_tabs, int *err) { 70713a7d540SJeremy L Thompson *err = CeedQFunctionContextSetNumViewTabs(CeedQFunctionContext_dict[*ctx], *num_tabs); 70813a7d540SJeremy L Thompson } 70913a7d540SJeremy L Thompson 7102b730f8bSJeremy L Thompson #define fCeedQFunctionContextView FORTRAN_NAME(ceedqfunctioncontextview, CEEDQFUNCTIONCONTEXTVIEW) 7112b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextView(int *ctx, int *err) { *err = CeedQFunctionContextView(CeedQFunctionContext_dict[*ctx], stdout); } 712777ff853SJeremy L Thompson 7132b730f8bSJeremy L Thompson #define fCeedQFunctionContextDestroy FORTRAN_NAME(ceedqfunctioncontextdestroy, CEEDQFUNCTIONCONTEXTDESTROY) 7143b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextDestroy(int *ctx, int *err) { 715777ff853SJeremy L Thompson if (*ctx == FORTRAN_NULL) return; 716777ff853SJeremy L Thompson *err = CeedQFunctionContextDestroy(&CeedQFunctionContext_dict[*ctx]); 717777ff853SJeremy L Thompson 718777ff853SJeremy L Thompson if (*err == 0) { 719777ff853SJeremy L Thompson *ctx = FORTRAN_NULL; 720777ff853SJeremy L Thompson CeedQFunctionContext_n--; 721777ff853SJeremy L Thompson if (CeedQFunctionContext_n == 0) { 722777ff853SJeremy L Thompson CeedFree(&CeedQFunctionContext_dict); 723777ff853SJeremy L Thompson CeedQFunctionContext_count = 0; 724777ff853SJeremy L Thompson CeedQFunctionContext_count_max = 0; 725777ff853SJeremy L Thompson } 726777ff853SJeremy L Thompson } 727777ff853SJeremy L Thompson } 728777ff853SJeremy L Thompson 729777ff853SJeremy L Thompson // ----------------------------------------------------------------------------- 730b00b43a6SJeremy L Thompson // CeedQFunction 731b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 732d7b241e6Sjeremylt static CeedQFunction *CeedQFunction_dict = NULL; 733d7b241e6Sjeremylt static int CeedQFunction_count = 0; 734d7b241e6Sjeremylt static int CeedQFunction_n = 0; 735d7b241e6Sjeremylt static int CeedQFunction_count_max = 0; 736d7b241e6Sjeremylt 7372b730f8bSJeremy L Thompson static int CeedQFunctionFortranStub(void *ctx, int nq, const CeedScalar *const *u, CeedScalar *const *v) { 738777ff853SJeremy L Thompson CeedFortranContext fctx = ctx; 7398cb0412aSJeremy L Thompson CeedQFunctionContext inner_ctx = fctx->inner_ctx; 740d7b241e6Sjeremylt int ierr; 741d7b241e6Sjeremylt 742777ff853SJeremy L Thompson CeedScalar *ctx_ = NULL; 743777ff853SJeremy L Thompson // Note: Device backends are generating their own kernels from 744777ff853SJeremy L Thompson // single source files, so only Host backends need to 745777ff853SJeremy L Thompson // use this Fortran stub. 7468cb0412aSJeremy L Thompson if (inner_ctx) { 7478cb0412aSJeremy L Thompson ierr = CeedQFunctionContextGetData(inner_ctx, CEED_MEM_HOST, &ctx_); 74893f4dbf1SSebastian Grimberg CeedCall(ierr); 749777ff853SJeremy L Thompson } 750777ff853SJeremy L Thompson 7512b730f8bSJeremy L Thompson fctx->f((void *)ctx_, &nq, u[0], u[1], u[2], u[3], u[4], u[5], u[6], u[7], u[8], u[9], u[10], u[11], u[12], u[13], u[14], u[15], v[0], v[1], v[2], 7522b730f8bSJeremy L Thompson v[3], v[4], v[5], v[6], v[7], v[8], v[9], v[10], v[11], v[12], v[13], v[14], v[15], &ierr); 753777ff853SJeremy L Thompson 7548cb0412aSJeremy L Thompson if (inner_ctx) { 7559a25c351SJeremy L Thompson ierr = CeedQFunctionContextRestoreData(inner_ctx, &ctx_); 75693f4dbf1SSebastian Grimberg CeedCall(ierr); 757777ff853SJeremy L Thompson } 758777ff853SJeremy L Thompson 759d7b241e6Sjeremylt return ierr; 760d7b241e6Sjeremylt } 761d7b241e6Sjeremylt 7622b730f8bSJeremy L Thompson #define fCeedQFunctionCreateInterior FORTRAN_NAME(ceedqfunctioncreateinterior, CEEDQFUNCTIONCREATEINTERIOR) 7632b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionCreateInterior( 7642b730f8bSJeremy L Thompson int *ceed, int *vec_length, 7652b730f8bSJeremy L Thompson void (*f)(void *ctx, int *nq, const CeedScalar *u, const CeedScalar *u1, const CeedScalar *u2, const CeedScalar *u3, const CeedScalar *u4, 7662b730f8bSJeremy L Thompson const CeedScalar *u5, const CeedScalar *u6, const CeedScalar *u7, const CeedScalar *u8, const CeedScalar *u9, const CeedScalar *u10, 7672b730f8bSJeremy L Thompson const CeedScalar *u11, const CeedScalar *u12, const CeedScalar *u13, const CeedScalar *u14, const CeedScalar *u15, CeedScalar *v, 7682b730f8bSJeremy L Thompson CeedScalar *v1, CeedScalar *v2, CeedScalar *v3, CeedScalar *v4, CeedScalar *v5, CeedScalar *v6, CeedScalar *v7, CeedScalar *v8, 7692b730f8bSJeremy L Thompson CeedScalar *v9, CeedScalar *v10, CeedScalar *v11, CeedScalar *v12, CeedScalar *v13, CeedScalar *v14, CeedScalar *v15, int *err), 7702b730f8bSJeremy L Thompson const char *source, int *qf, int *err, fortran_charlen_t source_len) { 771288c0443SJeremy L Thompson FIX_STRING(source); 772d7b241e6Sjeremylt if (CeedQFunction_count == CeedQFunction_count_max) { 773d7b241e6Sjeremylt CeedQFunction_count_max += CeedQFunction_count_max / 2 + 1; 774d7b241e6Sjeremylt CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 775d7b241e6Sjeremylt } 776d7b241e6Sjeremylt 777d7b241e6Sjeremylt CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 7782b730f8bSJeremy L Thompson *err = CeedQFunctionCreateInterior(Ceed_dict[*ceed], *vec_length, CeedQFunctionFortranStub, source_c, qf_); 779d7b241e6Sjeremylt 780d7b241e6Sjeremylt if (*err == 0) { 781d7b241e6Sjeremylt *qf = CeedQFunction_count++; 782d7b241e6Sjeremylt CeedQFunction_n++; 783d7b241e6Sjeremylt } 784d7b241e6Sjeremylt 785777ff853SJeremy L Thompson CeedFortranContext fctxdata; 786777ff853SJeremy L Thompson *err = CeedCalloc(1, &fctxdata); 787d7b241e6Sjeremylt if (*err) return; 7882b730f8bSJeremy L Thompson fctxdata->f = f; 7892b730f8bSJeremy L Thompson fctxdata->inner_ctx = NULL; 790777ff853SJeremy L Thompson CeedQFunctionContext fctx; 791777ff853SJeremy L Thompson *err = CeedQFunctionContextCreate(Ceed_dict[*ceed], &fctx); 792777ff853SJeremy L Thompson if (*err) return; 7932b730f8bSJeremy L Thompson *err = CeedQFunctionContextSetData(fctx, CEED_MEM_HOST, CEED_OWN_POINTER, sizeof(*fctxdata), fctxdata); 794777ff853SJeremy L Thompson if (*err) return; 795777ff853SJeremy L Thompson *err = CeedQFunctionSetContext(*qf_, fctx); 796777ff853SJeremy L Thompson if (*err) return; 797777ff853SJeremy L Thompson CeedQFunctionContextDestroy(&fctx); 798777ff853SJeremy L Thompson if (*err) return; 799d7b241e6Sjeremylt 800777ff853SJeremy L Thompson *err = CeedQFunctionSetFortranStatus(*qf_, true); 801d7b241e6Sjeremylt } 802d7b241e6Sjeremylt 8032b730f8bSJeremy L Thompson #define fCeedQFunctionCreateInteriorByName FORTRAN_NAME(ceedqfunctioncreateinteriorbyname, CEEDQFUNCTIONCREATEINTERIORBYNAME) 8042b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionCreateInteriorByName(int *ceed, const char *name, int *qf, int *err, fortran_charlen_t name_len) { 805288c0443SJeremy L Thompson FIX_STRING(name); 806288c0443SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 807288c0443SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max / 2 + 1; 808288c0443SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 809288c0443SJeremy L Thompson } 810288c0443SJeremy L Thompson 811288c0443SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 812288c0443SJeremy L Thompson *err = CeedQFunctionCreateInteriorByName(Ceed_dict[*ceed], name_c, qf_); 813288c0443SJeremy L Thompson 814288c0443SJeremy L Thompson if (*err == 0) { 815288c0443SJeremy L Thompson *qf = CeedQFunction_count++; 816288c0443SJeremy L Thompson CeedQFunction_n++; 817288c0443SJeremy L Thompson } 818288c0443SJeremy L Thompson } 819288c0443SJeremy L Thompson 8202b730f8bSJeremy L Thompson #define fCeedQFunctionCreateIdentity FORTRAN_NAME(ceedqfunctioncreateidentity, CEEDQFUNCTIONCREATEIDENTITY) 8212b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionCreateIdentity(int *ceed, int *size, int *inmode, int *outmode, int *qf, int *err) { 8220219ea01SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 8230219ea01SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max / 2 + 1; 8240219ea01SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 8250219ea01SJeremy L Thompson } 8260219ea01SJeremy L Thompson 8270219ea01SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 8282b730f8bSJeremy L Thompson *err = CeedQFunctionCreateIdentity(Ceed_dict[*ceed], *size, (CeedEvalMode)*inmode, (CeedEvalMode)*outmode, qf_); 8290219ea01SJeremy L Thompson 8300219ea01SJeremy L Thompson if (*err == 0) { 8310219ea01SJeremy L Thompson *qf = CeedQFunction_count++; 8320219ea01SJeremy L Thompson CeedQFunction_n++; 8330219ea01SJeremy L Thompson } 8340219ea01SJeremy L Thompson } 8350219ea01SJeremy L Thompson 8362b730f8bSJeremy L Thompson #define fCeedQFunctionAddInput FORTRAN_NAME(ceedqfunctionaddinput, CEEDQFUNCTIONADDINPUT) 8372b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionAddInput(int *qf, const char *field_name, CeedInt *num_comp, CeedEvalMode *eval_mode, int *err, 838d1d35e2fSjeremylt fortran_charlen_t field_name_len) { 839d1d35e2fSjeremylt FIX_STRING(field_name); 840d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 841d7b241e6Sjeremylt 842d1d35e2fSjeremylt *err = CeedQFunctionAddInput(qf_, field_name_c, *num_comp, *eval_mode); 843d7b241e6Sjeremylt } 844d7b241e6Sjeremylt 8452b730f8bSJeremy L Thompson #define fCeedQFunctionAddOutput FORTRAN_NAME(ceedqfunctionaddoutput, CEEDQFUNCTIONADDOUTPUT) 8462b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionAddOutput(int *qf, const char *field_name, CeedInt *num_comp, CeedEvalMode *eval_mode, int *err, 847d1d35e2fSjeremylt fortran_charlen_t field_name_len) { 848d1d35e2fSjeremylt FIX_STRING(field_name); 849d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 850d7b241e6Sjeremylt 851d1d35e2fSjeremylt *err = CeedQFunctionAddOutput(qf_, field_name_c, *num_comp, *eval_mode); 852d7b241e6Sjeremylt } 853d7b241e6Sjeremylt 8542b730f8bSJeremy L Thompson #define fCeedQFunctionSetContext FORTRAN_NAME(ceedqfunctionsetcontext, CEEDQFUNCTIONSETCONTEXT) 8553b271f31SJed Brown CEED_EXTERN void fCeedQFunctionSetContext(int *qf, int *ctx, int *err) { 856418fb8c2Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 857777ff853SJeremy L Thompson CeedQFunctionContext ctx_ = CeedQFunctionContext_dict[*ctx]; 858418fb8c2Sjeremylt 859777ff853SJeremy L Thompson CeedQFunctionContext fctx; 860777ff853SJeremy L Thompson *err = CeedQFunctionGetContext(qf_, &fctx); 861777ff853SJeremy L Thompson if (*err) return; 862777ff853SJeremy L Thompson CeedFortranContext fctxdata; 863777ff853SJeremy L Thompson *err = CeedQFunctionContextGetData(fctx, CEED_MEM_HOST, &fctxdata); 864777ff853SJeremy L Thompson if (*err) return; 8658cb0412aSJeremy L Thompson fctxdata->inner_ctx = ctx_; 866777ff853SJeremy L Thompson *err = CeedQFunctionContextRestoreData(fctx, (void **)&fctxdata); 8671485364cSJeremy L Thompson if (*err) return; 8681485364cSJeremy L Thompson *err = CeedQFunctionContextDestroy(&fctx); 869418fb8c2Sjeremylt } 870418fb8c2Sjeremylt 87113a7d540SJeremy L Thompson #define fCeedQFunctionSetNumViewTabs FORTRAN_NAME(ceedqfunctionsetnumviewtabs, CEEDQFUNCTIONSETNUMVIEWTABS) 87213a7d540SJeremy L Thompson CEED_EXTERN void fCeedQFunctionSetNumViewTabs(int *qf, int *num_tabs, int *err) { 87313a7d540SJeremy L Thompson *err = CeedQFunctionSetNumViewTabs(CeedQFunction_dict[*qf], *num_tabs); 87413a7d540SJeremy L Thompson } 87513a7d540SJeremy L Thompson 8762b730f8bSJeremy L Thompson #define fCeedQFunctionView FORTRAN_NAME(ceedqfunctionview, CEEDQFUNCTIONVIEW) 8773b271f31SJed Brown CEED_EXTERN void fCeedQFunctionView(int *qf, int *err) { 87875affc3bSjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 87975affc3bSjeremylt 88075affc3bSjeremylt *err = CeedQFunctionView(qf_, stdout); 88175affc3bSjeremylt } 88275affc3bSjeremylt 8832b730f8bSJeremy L Thompson #define fCeedQFunctionApply FORTRAN_NAME(ceedqfunctionapply, CEEDQFUNCTIONAPPLY) 884d7b241e6Sjeremylt // TODO Need Fixing, double pointer 8852b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionApply(int *qf, int *Q, int *u, int *u1, int *u2, int *u3, int *u4, int *u5, int *u6, int *u7, int *u8, int *u9, 8862b730f8bSJeremy L Thompson int *u10, int *u11, int *u12, int *u13, int *u14, int *u15, int *v, int *v1, int *v2, int *v3, int *v4, int *v5, 8872b730f8bSJeremy L Thompson int *v6, int *v7, int *v8, int *v9, int *v10, int *v11, int *v12, int *v13, int *v14, int *v15, int *err) { 888d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 889aedaa0e5Sjeremylt CeedVector *in; 890bf4cb664SJeremy L Thompson *err = CeedCalloc(CEED_FIELD_MAX, &in); 891d7b241e6Sjeremylt if (*err) return; 892aedaa0e5Sjeremylt in[0] = *u == FORTRAN_NULL ? NULL : CeedVector_dict[*u]; 893aedaa0e5Sjeremylt in[1] = *u1 == FORTRAN_NULL ? NULL : CeedVector_dict[*u1]; 894aedaa0e5Sjeremylt in[2] = *u2 == FORTRAN_NULL ? NULL : CeedVector_dict[*u2]; 895aedaa0e5Sjeremylt in[3] = *u3 == FORTRAN_NULL ? NULL : CeedVector_dict[*u3]; 896aedaa0e5Sjeremylt in[4] = *u4 == FORTRAN_NULL ? NULL : CeedVector_dict[*u4]; 897aedaa0e5Sjeremylt in[5] = *u5 == FORTRAN_NULL ? NULL : CeedVector_dict[*u5]; 898aedaa0e5Sjeremylt in[6] = *u6 == FORTRAN_NULL ? NULL : CeedVector_dict[*u6]; 899aedaa0e5Sjeremylt in[7] = *u7 == FORTRAN_NULL ? NULL : CeedVector_dict[*u7]; 900aedaa0e5Sjeremylt in[8] = *u8 == FORTRAN_NULL ? NULL : CeedVector_dict[*u8]; 901aedaa0e5Sjeremylt in[9] = *u9 == FORTRAN_NULL ? NULL : CeedVector_dict[*u9]; 902aedaa0e5Sjeremylt in[10] = *u10 == FORTRAN_NULL ? NULL : CeedVector_dict[*u10]; 903aedaa0e5Sjeremylt in[11] = *u11 == FORTRAN_NULL ? NULL : CeedVector_dict[*u11]; 904aedaa0e5Sjeremylt in[12] = *u12 == FORTRAN_NULL ? NULL : CeedVector_dict[*u12]; 905aedaa0e5Sjeremylt in[13] = *u13 == FORTRAN_NULL ? NULL : CeedVector_dict[*u13]; 906aedaa0e5Sjeremylt in[14] = *u14 == FORTRAN_NULL ? NULL : CeedVector_dict[*u14]; 907aedaa0e5Sjeremylt in[15] = *u15 == FORTRAN_NULL ? NULL : CeedVector_dict[*u15]; 908aedaa0e5Sjeremylt CeedVector *out; 909bf4cb664SJeremy L Thompson *err = CeedCalloc(CEED_FIELD_MAX, &out); 910d7b241e6Sjeremylt if (*err) return; 911aedaa0e5Sjeremylt out[0] = *v == FORTRAN_NULL ? NULL : CeedVector_dict[*v]; 912aedaa0e5Sjeremylt out[1] = *v1 == FORTRAN_NULL ? NULL : CeedVector_dict[*v1]; 913aedaa0e5Sjeremylt out[2] = *v2 == FORTRAN_NULL ? NULL : CeedVector_dict[*v2]; 914aedaa0e5Sjeremylt out[3] = *v3 == FORTRAN_NULL ? NULL : CeedVector_dict[*v3]; 915aedaa0e5Sjeremylt out[4] = *v4 == FORTRAN_NULL ? NULL : CeedVector_dict[*v4]; 916aedaa0e5Sjeremylt out[5] = *v5 == FORTRAN_NULL ? NULL : CeedVector_dict[*v5]; 917aedaa0e5Sjeremylt out[6] = *v6 == FORTRAN_NULL ? NULL : CeedVector_dict[*v6]; 918aedaa0e5Sjeremylt out[7] = *v7 == FORTRAN_NULL ? NULL : CeedVector_dict[*v7]; 919aedaa0e5Sjeremylt out[8] = *v8 == FORTRAN_NULL ? NULL : CeedVector_dict[*v8]; 920aedaa0e5Sjeremylt out[9] = *v9 == FORTRAN_NULL ? NULL : CeedVector_dict[*v9]; 921aedaa0e5Sjeremylt out[10] = *v10 == FORTRAN_NULL ? NULL : CeedVector_dict[*v10]; 922aedaa0e5Sjeremylt out[11] = *v11 == FORTRAN_NULL ? NULL : CeedVector_dict[*v11]; 923aedaa0e5Sjeremylt out[12] = *v12 == FORTRAN_NULL ? NULL : CeedVector_dict[*v12]; 924aedaa0e5Sjeremylt out[13] = *v13 == FORTRAN_NULL ? NULL : CeedVector_dict[*v13]; 925aedaa0e5Sjeremylt out[14] = *v14 == FORTRAN_NULL ? NULL : CeedVector_dict[*v14]; 926aedaa0e5Sjeremylt out[15] = *v15 == FORTRAN_NULL ? NULL : CeedVector_dict[*v15]; 927aedaa0e5Sjeremylt *err = CeedQFunctionApply(qf_, *Q, in, out); 928d7b241e6Sjeremylt if (*err) return; 929d7b241e6Sjeremylt 930d7b241e6Sjeremylt *err = CeedFree(&in); 931d7b241e6Sjeremylt if (*err) return; 932d7b241e6Sjeremylt *err = CeedFree(&out); 933d7b241e6Sjeremylt } 934d7b241e6Sjeremylt 9352b730f8bSJeremy L Thompson #define fCeedQFunctionDestroy FORTRAN_NAME(ceedqfunctiondestroy, CEEDQFUNCTIONDESTROY) 9363b271f31SJed Brown CEED_EXTERN void fCeedQFunctionDestroy(int *qf, int *err) { 937b22696a8SJeremy L Thompson if (*qf == FORTRAN_NULL) return; 938c4b814baSjeremylt 939c4b814baSjeremylt *err = CeedQFunctionDestroy(&CeedQFunction_dict[*qf]); 94016595307SJeremy L Thompson if (*err == 0) { 94116595307SJeremy L Thompson *qf = FORTRAN_NULL; 942d7b241e6Sjeremylt CeedQFunction_n--; 943d7b241e6Sjeremylt if (CeedQFunction_n == 0) { 944d7b241e6Sjeremylt *err = CeedFree(&CeedQFunction_dict); 945d7b241e6Sjeremylt CeedQFunction_count = 0; 946d7b241e6Sjeremylt CeedQFunction_count_max = 0; 947d7b241e6Sjeremylt } 948d7b241e6Sjeremylt } 94916595307SJeremy L Thompson } 950d7b241e6Sjeremylt 951b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 952b00b43a6SJeremy L Thompson // CeedOperator 953b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 954d7b241e6Sjeremylt static CeedOperator *CeedOperator_dict = NULL; 955d7b241e6Sjeremylt static int CeedOperator_count = 0; 956d7b241e6Sjeremylt static int CeedOperator_n = 0; 957d7b241e6Sjeremylt static int CeedOperator_count_max = 0; 958d7b241e6Sjeremylt 9592b730f8bSJeremy L Thompson #define fCeedOperatorCreate FORTRAN_NAME(ceedoperatorcreate, CEEDOPERATORCREATE) 9602b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorCreate(int *ceed, int *qf, int *dqf, int *dqfT, int *op, int *err) { 961777ff853SJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) { 962777ff853SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 963777ff853SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 964777ff853SJeremy L Thompson } 965d7b241e6Sjeremylt 966d7b241e6Sjeremylt CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 967d7b241e6Sjeremylt 968442e7f0bSjeremylt CeedQFunction dqf_ = CEED_QFUNCTION_NONE, dqfT_ = CEED_QFUNCTION_NONE; 969442e7f0bSjeremylt if (*dqf != FORTRAN_QFUNCTION_NONE) dqf_ = CeedQFunction_dict[*dqf]; 970442e7f0bSjeremylt if (*dqfT != FORTRAN_QFUNCTION_NONE) dqfT_ = CeedQFunction_dict[*dqfT]; 971d7b241e6Sjeremylt 9722b730f8bSJeremy L Thompson *err = CeedOperatorCreate(Ceed_dict[*ceed], CeedQFunction_dict[*qf], dqf_, dqfT_, op_); 973d7b241e6Sjeremylt if (*err) return; 974d7b241e6Sjeremylt *op = CeedOperator_count++; 975d7b241e6Sjeremylt CeedOperator_n++; 976d7b241e6Sjeremylt } 977d7b241e6Sjeremylt 978ed094490SJeremy L Thompson #define fCeedOperatorCreateComposite FORTRAN_NAME(ceedoperatorcreatecomposite, CEEDOPERATORCREATECOMPOSITE) 979ed094490SJeremy L Thompson CEED_EXTERN void fCeedOperatorCreateComposite(int *ceed, int *op, int *err) { 980777ff853SJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) { 981777ff853SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 982777ff853SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 983777ff853SJeremy L Thompson } 98452d6035fSJeremy L Thompson 98552d6035fSJeremy L Thompson CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 98652d6035fSJeremy L Thompson 987ed094490SJeremy L Thompson *err = CeedOperatorCreateComposite(Ceed_dict[*ceed], op_); 98852d6035fSJeremy L Thompson if (*err) return; 98952d6035fSJeremy L Thompson *op = CeedOperator_count++; 99052d6035fSJeremy L Thompson CeedOperator_n++; 99152d6035fSJeremy L Thompson } 99252d6035fSJeremy L Thompson 9932b730f8bSJeremy L Thompson #define fCeedOperatorSetField FORTRAN_NAME(ceedoperatorsetfield, CEEDOPERATORSETFIELD) 9942b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorSetField(int *op, const char *field_name, int *r, int *b, int *v, int *err, fortran_charlen_t field_name_len) { 995d1d35e2fSjeremylt FIX_STRING(field_name); 996d7b241e6Sjeremylt CeedElemRestriction r_; 997d7b241e6Sjeremylt CeedBasis b_; 998d7b241e6Sjeremylt CeedVector v_; 999d7b241e6Sjeremylt 1000d7b241e6Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 1001d7b241e6Sjeremylt 1002d7b241e6Sjeremylt if (*r == FORTRAN_NULL) { 1003d7b241e6Sjeremylt r_ = NULL; 100415910d16Sjeremylt } else if (*r == FORTRAN_ELEMRESTRICTION_NONE) { 100515910d16Sjeremylt r_ = CEED_ELEMRESTRICTION_NONE; 1006d7b241e6Sjeremylt } else { 1007d7b241e6Sjeremylt r_ = CeedElemRestriction_dict[*r]; 1008d7b241e6Sjeremylt } 10094dccadb6Sjeremylt 1010d7b241e6Sjeremylt if (*b == FORTRAN_NULL) { 1011d7b241e6Sjeremylt b_ = NULL; 1012a36217cbSJeremy L Thompson } else if (*b == FORTRAN_BASIS_NONE) { 1013356036faSJeremy L Thompson b_ = CEED_BASIS_NONE; 1014d7b241e6Sjeremylt } else { 1015d7b241e6Sjeremylt b_ = CeedBasis_dict[*b]; 1016d7b241e6Sjeremylt } 1017d7b241e6Sjeremylt if (*v == FORTRAN_NULL) { 1018d7b241e6Sjeremylt v_ = NULL; 1019d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_ACTIVE) { 1020d7b241e6Sjeremylt v_ = CEED_VECTOR_ACTIVE; 1021d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_NONE) { 1022d7b241e6Sjeremylt v_ = CEED_VECTOR_NONE; 1023d7b241e6Sjeremylt } else { 1024d7b241e6Sjeremylt v_ = CeedVector_dict[*v]; 1025d7b241e6Sjeremylt } 1026d7b241e6Sjeremylt 1027d1d35e2fSjeremylt *err = CeedOperatorSetField(op_, field_name_c, r_, b_, v_); 1028d7b241e6Sjeremylt } 1029d7b241e6Sjeremylt 1030ed094490SJeremy L Thompson #define fCeedOperatorCompositeAddSub FORTRAN_NAME(ceedoperatorcompositeaddsub, CEEDOPERATORCOMPOSITEADDSUB) 1031ed094490SJeremy L Thompson CEED_EXTERN void fCeedOperatorCompositeAddSub(int *compositeop, int *subop, int *err) { 103252d6035fSJeremy L Thompson CeedOperator compositeop_ = CeedOperator_dict[*compositeop]; 103352d6035fSJeremy L Thompson CeedOperator subop_ = CeedOperator_dict[*subop]; 103452d6035fSJeremy L Thompson 1035ed094490SJeremy L Thompson *err = CeedOperatorCompositeAddSub(compositeop_, subop_); 1036ea6b5821SJeremy L Thompson } 1037ea6b5821SJeremy L Thompson 10382b730f8bSJeremy L Thompson #define fCeedOperatorSetName FORTRAN_NAME(ceedoperatorsetname, CEEDOPERATORSETNAME) 10392b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorSetName(int *op, const char *name, int *err, fortran_charlen_t name_len) { 1040ea6b5821SJeremy L Thompson FIX_STRING(name); 1041ea6b5821SJeremy L Thompson CeedOperator op_ = CeedOperator_dict[*op]; 1042ea6b5821SJeremy L Thompson 1043ea6b5821SJeremy L Thompson *err = CeedOperatorSetName(op_, name_c); 104452d6035fSJeremy L Thompson } 104552d6035fSJeremy L Thompson 10465a526491SJeremy L Thompson #define fCeedOperatorSetNumViewTabs FORTRAN_NAME(ceedoperatorsetnumviewtabs, CEEDOPERATORSETNUMVIEWTABS) 10475a526491SJeremy L Thompson CEED_EXTERN void fCeedOperatorSetNumViewTabs(int *op, int *ntabs, int *err) { 10485a526491SJeremy L Thompson CeedOperator op_ = CeedOperator_dict[*op]; 10495a526491SJeremy L Thompson 10505a526491SJeremy L Thompson *err = CeedOperatorSetNumViewTabs(op_, *ntabs); 10515a526491SJeremy L Thompson } 10525a526491SJeremy L Thompson 10532b730f8bSJeremy L Thompson #define fCeedOperatorLinearAssembleQFunction FORTRAN_NAME(ceedoperatorlinearassembleqfunction, CEEDOPERATORLINEARASSEMBLEQFUNCTION) 10542b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorLinearAssembleQFunction(int *op, int *assembledvec, int *assembledrstr, int *rqst, int *err) { 10551d102b48SJeremy L Thompson // Vector 10561d102b48SJeremy L Thompson if (CeedVector_count == CeedVector_count_max) { 10571d102b48SJeremy L Thompson CeedVector_count_max += CeedVector_count_max / 2 + 1; 10581d102b48SJeremy L Thompson CeedRealloc(CeedVector_count_max, &CeedVector_dict); 10591d102b48SJeremy L Thompson } 10601d102b48SJeremy L Thompson CeedVector *assembledvec_ = &CeedVector_dict[CeedVector_count]; 10611d102b48SJeremy L Thompson 10621d102b48SJeremy L Thompson // Restriction 10631d102b48SJeremy L Thompson if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 10641d102b48SJeremy L Thompson CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 10651d102b48SJeremy L Thompson CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 10661d102b48SJeremy L Thompson } 10672b730f8bSJeremy L Thompson CeedElemRestriction *rstr_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 10681d102b48SJeremy L Thompson 10691d102b48SJeremy L Thompson int createRequest = 1; 10701d102b48SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 10711d102b48SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 10721d102b48SJeremy L Thompson createRequest = 0; 10731d102b48SJeremy L Thompson } 10741d102b48SJeremy L Thompson 10751d102b48SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 10761d102b48SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 10771d102b48SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 10781d102b48SJeremy L Thompson } 10791d102b48SJeremy L Thompson 10801d102b48SJeremy L Thompson CeedRequest *rqst_; 10811d102b48SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 10821d102b48SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 10831d102b48SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 10841d102b48SJeremy L Thompson 10852b730f8bSJeremy L Thompson *err = CeedOperatorLinearAssembleQFunction(CeedOperator_dict[*op], assembledvec_, rstr_, rqst_); 10861d102b48SJeremy L Thompson if (*err) return; 10871d102b48SJeremy L Thompson if (createRequest) { 10881d102b48SJeremy L Thompson *rqst = CeedRequest_count++; 10891d102b48SJeremy L Thompson CeedRequest_n++; 10901d102b48SJeremy L Thompson } 10911d102b48SJeremy L Thompson 10921d102b48SJeremy L Thompson if (*err == 0) { 10931d102b48SJeremy L Thompson *assembledrstr = CeedElemRestriction_count++; 10941d102b48SJeremy L Thompson CeedElemRestriction_n++; 10951d102b48SJeremy L Thompson *assembledvec = CeedVector_count++; 10961d102b48SJeremy L Thompson CeedVector_n++; 10971d102b48SJeremy L Thompson } 10981d102b48SJeremy L Thompson } 10991d102b48SJeremy L Thompson 11002b730f8bSJeremy L Thompson #define fCeedOperatorLinearAssembleDiagonal FORTRAN_NAME(ceedoperatorlinearassemblediagonal, CEEDOPERATORLINEARASSEMBLEDIAGONAL) 11012b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorLinearAssembleDiagonal(int *op, int *assembledvec, int *rqst, int *err) { 1102b7ec98d8SJeremy L Thompson int createRequest = 1; 1103b7ec98d8SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1104b7ec98d8SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 1105b7ec98d8SJeremy L Thompson createRequest = 0; 1106b7ec98d8SJeremy L Thompson } 1107b7ec98d8SJeremy L Thompson 1108b7ec98d8SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1109b7ec98d8SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 1110b7ec98d8SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1111b7ec98d8SJeremy L Thompson } 1112b7ec98d8SJeremy L Thompson 1113b7ec98d8SJeremy L Thompson CeedRequest *rqst_; 1114b7ec98d8SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1115b7ec98d8SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1116b7ec98d8SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1117b7ec98d8SJeremy L Thompson 11182b730f8bSJeremy L Thompson *err = CeedOperatorLinearAssembleDiagonal(CeedOperator_dict[*op], CeedVector_dict[*assembledvec], rqst_); 1119b7ec98d8SJeremy L Thompson if (*err) return; 1120b7ec98d8SJeremy L Thompson if (createRequest) { 1121b7ec98d8SJeremy L Thompson *rqst = CeedRequest_count++; 1122b7ec98d8SJeremy L Thompson CeedRequest_n++; 1123b7ec98d8SJeremy L Thompson } 1124b7ec98d8SJeremy L Thompson } 1125b7ec98d8SJeremy L Thompson 11262b730f8bSJeremy L Thompson #define fCeedOperatorMultigridLevelCreate FORTRAN_NAME(ceedoperatormultigridlevelcreate, CEEDOPERATORMULTIGRIDLEVELCREATE) 11272b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorMultigridLevelCreate(int *opFine, int *pMultFine, int *rstrCoarse, int *basisCoarse, int *opCoarse, int *opProlong, 11282b730f8bSJeremy L Thompson int *opRestrict, int *err) { 1129d99fa3c5SJeremy L Thompson // Operators 1130d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1131d99fa3c5SJeremy L Thompson 1132d99fa3c5SJeremy L Thompson // C interface call 11332b730f8bSJeremy L Thompson *err = CeedOperatorMultigridLevelCreate(CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], CeedElemRestriction_dict[*rstrCoarse], 11342b730f8bSJeremy L Thompson CeedBasis_dict[*basisCoarse], &opCoarse_, &opProlong_, &opRestrict_); 1135d99fa3c5SJeremy L Thompson 1136d99fa3c5SJeremy L Thompson if (*err) return; 1137d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1138d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 1139d99fa3c5SJeremy L Thompson } 1140d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1141d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1142d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1143d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1144d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1145d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1146d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1147d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1148d99fa3c5SJeremy L Thompson } 1149d99fa3c5SJeremy L Thompson 11502b730f8bSJeremy L Thompson #define fCeedOperatorMultigridLevelCreateTensorH1 FORTRAN_NAME(ceedoperatormultigridlevelcreatetensorh1, CEEDOPERATORMULTIGRIDLEVELCREATETENSORH1) 11512b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorMultigridLevelCreateTensorH1(int *opFine, int *pMultFine, int *rstrCoarse, int *basisCoarse, 11522b730f8bSJeremy L Thompson const CeedScalar *interpCtoF, int *opCoarse, int *opProlong, int *opRestrict, int *err) { 11532b730f8bSJeremy L Thompson // Operators 11542b730f8bSJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 11552b730f8bSJeremy L Thompson 11562b730f8bSJeremy L Thompson // C interface call 11572b730f8bSJeremy L Thompson *err = CeedOperatorMultigridLevelCreateTensorH1(CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], CeedElemRestriction_dict[*rstrCoarse], 11582b730f8bSJeremy L Thompson CeedBasis_dict[*basisCoarse], interpCtoF, &opCoarse_, &opProlong_, &opRestrict_); 11592b730f8bSJeremy L Thompson 11602b730f8bSJeremy L Thompson if (*err) return; 11612b730f8bSJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 11622b730f8bSJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 11632b730f8bSJeremy L Thompson } 11642b730f8bSJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 11652b730f8bSJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 11662b730f8bSJeremy L Thompson *opCoarse = CeedOperator_count++; 11672b730f8bSJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 11682b730f8bSJeremy L Thompson *opProlong = CeedOperator_count++; 11692b730f8bSJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 11702b730f8bSJeremy L Thompson *opRestrict = CeedOperator_count++; 11712b730f8bSJeremy L Thompson CeedOperator_n += 3; 11722b730f8bSJeremy L Thompson } 11732b730f8bSJeremy L Thompson 11742b730f8bSJeremy L Thompson #define fCeedOperatorMultigridLevelCreateH1 FORTRAN_NAME(ceedoperatormultigridlevelcreateh1, CEEDOPERATORMULTIGRIDLEVELCREATEH1) 11752b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorMultigridLevelCreateH1(int *opFine, int *pMultFine, int *rstrCoarse, int *basisCoarse, const CeedScalar *interpCtoF, 1176d99fa3c5SJeremy L Thompson int *opCoarse, int *opProlong, int *opRestrict, int *err) { 1177d99fa3c5SJeremy L Thompson // Operators 1178d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1179d99fa3c5SJeremy L Thompson 1180d99fa3c5SJeremy L Thompson // C interface call 11812b730f8bSJeremy L Thompson *err = CeedOperatorMultigridLevelCreateH1(CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], CeedElemRestriction_dict[*rstrCoarse], 11822b730f8bSJeremy L Thompson CeedBasis_dict[*basisCoarse], interpCtoF, &opCoarse_, &opProlong_, &opRestrict_); 1183d99fa3c5SJeremy L Thompson 1184d99fa3c5SJeremy L Thompson if (*err) return; 1185d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1186d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 1187d99fa3c5SJeremy L Thompson } 1188d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1189d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1190d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1191d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1192d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1193d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1194d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1195d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1196d99fa3c5SJeremy L Thompson } 1197d99fa3c5SJeremy L Thompson 11982b730f8bSJeremy L Thompson #define fCeedOperatorView FORTRAN_NAME(ceedoperatorview, CEEDOPERATORVIEW) 11993b271f31SJed Brown CEED_EXTERN void fCeedOperatorView(int *op, int *err) { 12002ebaca42Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 12012ebaca42Sjeremylt 12022ebaca42Sjeremylt *err = CeedOperatorView(op_, stdout); 12032ebaca42Sjeremylt } 12042ebaca42Sjeremylt 12052b730f8bSJeremy L Thompson #define fCeedOperatorCreateFDMElementInverse FORTRAN_NAME(ceedoperatorcreatefdmelementinverse, CEEDOPERATORCREATEFDMELEMENTINVERSE) 12062b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorCreateFDMElementInverse(int *op, int *fdminv, int *rqst, int *err) { 12073bd813ffSjeremylt // Operator 12083bd813ffSjeremylt if (CeedOperator_count == CeedOperator_count_max) { 12093bd813ffSjeremylt CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 12103bd813ffSjeremylt CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 12113bd813ffSjeremylt } 12122b730f8bSJeremy L Thompson CeedOperator *fdminv_ = &CeedOperator_dict[CeedOperator_count]; 12133bd813ffSjeremylt 12143bd813ffSjeremylt int createRequest = 1; 12153bd813ffSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 12163bd813ffSjeremylt if (*rqst == -1 || *rqst == -2) { 12173bd813ffSjeremylt createRequest = 0; 12183bd813ffSjeremylt } 12193bd813ffSjeremylt 12203bd813ffSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 12213bd813ffSjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 12223bd813ffSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 12233bd813ffSjeremylt } 12243bd813ffSjeremylt 12253bd813ffSjeremylt CeedRequest *rqst_; 12263bd813ffSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 12273bd813ffSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 12283bd813ffSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 12293bd813ffSjeremylt 12302b730f8bSJeremy L Thompson *err = CeedOperatorCreateFDMElementInverse(CeedOperator_dict[*op], fdminv_, rqst_); 12313bd813ffSjeremylt if (*err) return; 12323bd813ffSjeremylt if (createRequest) { 12333bd813ffSjeremylt *rqst = CeedRequest_count++; 12343bd813ffSjeremylt CeedRequest_n++; 12353bd813ffSjeremylt } 12363bd813ffSjeremylt 12373bd813ffSjeremylt if (*err == 0) { 12383bd813ffSjeremylt *fdminv = CeedOperator_count++; 12393bd813ffSjeremylt CeedOperator_n++; 12403bd813ffSjeremylt } 12413bd813ffSjeremylt } 12423bd813ffSjeremylt 1243d7b241e6Sjeremylt #define fCeedOperatorApply FORTRAN_NAME(ceedoperatorapply, CEEDOPERATORAPPLY) 12442b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorApply(int *op, int *ustatevec, int *resvec, int *rqst, int *err) { 12452b730f8bSJeremy L Thompson CeedVector ustatevec_ = (*ustatevec == FORTRAN_NULL) ? NULL : (*ustatevec == FORTRAN_VECTOR_NONE ? CEED_VECTOR_NONE : CeedVector_dict[*ustatevec]); 12462b730f8bSJeremy L Thompson CeedVector resvec_ = (*resvec == FORTRAN_NULL) ? NULL : (*resvec == FORTRAN_VECTOR_NONE ? CEED_VECTOR_NONE : CeedVector_dict[*resvec]); 1247d7b241e6Sjeremylt 1248d7b241e6Sjeremylt int createRequest = 1; 1249d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1250d7b241e6Sjeremylt if (*rqst == -1 || *rqst == -2) { 1251d7b241e6Sjeremylt createRequest = 0; 1252d7b241e6Sjeremylt } 1253d7b241e6Sjeremylt 1254d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1255d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 1256d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1257d7b241e6Sjeremylt } 1258d7b241e6Sjeremylt 1259d7b241e6Sjeremylt CeedRequest *rqst_; 1260d7b241e6Sjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1261d7b241e6Sjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1262d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1263d7b241e6Sjeremylt 12642b730f8bSJeremy L Thompson *err = CeedOperatorApply(CeedOperator_dict[*op], ustatevec_, resvec_, rqst_); 1265d7b241e6Sjeremylt if (*err) return; 1266d7b241e6Sjeremylt if (createRequest) { 1267d7b241e6Sjeremylt *rqst = CeedRequest_count++; 1268d7b241e6Sjeremylt CeedRequest_n++; 1269d7b241e6Sjeremylt } 1270d7b241e6Sjeremylt } 1271d7b241e6Sjeremylt 1272cae8b89aSjeremylt #define fCeedOperatorApplyAdd FORTRAN_NAME(ceedoperatorapplyadd, CEEDOPERATORAPPLYADD) 12732b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorApplyAdd(int *op, int *ustatevec, int *resvec, int *rqst, int *err) { 12742b730f8bSJeremy L Thompson CeedVector ustatevec_ = *ustatevec == FORTRAN_NULL ? NULL : CeedVector_dict[*ustatevec]; 12752b730f8bSJeremy L Thompson CeedVector resvec_ = *resvec == FORTRAN_NULL ? NULL : CeedVector_dict[*resvec]; 1276cae8b89aSjeremylt 1277cae8b89aSjeremylt int createRequest = 1; 1278cae8b89aSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1279cae8b89aSjeremylt if (*rqst == -1 || *rqst == -2) { 1280cae8b89aSjeremylt createRequest = 0; 1281cae8b89aSjeremylt } 1282cae8b89aSjeremylt 1283cae8b89aSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1284cae8b89aSjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 1285cae8b89aSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1286cae8b89aSjeremylt } 1287cae8b89aSjeremylt 1288cae8b89aSjeremylt CeedRequest *rqst_; 1289cae8b89aSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1290cae8b89aSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1291cae8b89aSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1292cae8b89aSjeremylt 12932b730f8bSJeremy L Thompson *err = CeedOperatorApplyAdd(CeedOperator_dict[*op], ustatevec_, resvec_, rqst_); 1294cae8b89aSjeremylt if (*err) return; 1295cae8b89aSjeremylt if (createRequest) { 1296cae8b89aSjeremylt *rqst = CeedRequest_count++; 1297cae8b89aSjeremylt CeedRequest_n++; 1298cae8b89aSjeremylt } 1299cae8b89aSjeremylt } 1300cae8b89aSjeremylt 13012b730f8bSJeremy L Thompson #define fCeedOperatorApplyJacobian FORTRAN_NAME(ceedoperatorapplyjacobian, CEEDOPERATORAPPLYJACOBIAN) 13022b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorApplyJacobian(int *op, int *qdatavec, int *ustatevec, int *dustatevec, int *dresvec, int *rqst, int *err) { 1303d7b241e6Sjeremylt // TODO Uncomment this when CeedOperatorApplyJacobian is implemented 1304d7b241e6Sjeremylt // *err = CeedOperatorApplyJacobian(CeedOperator_dict[*op], CeedVector_dict[*qdatavec], 1305d7b241e6Sjeremylt // CeedVector_dict[*ustatevec], CeedVector_dict[*dustatevec], 1306d7b241e6Sjeremylt // CeedVector_dict[*dresvec], &CeedRequest_dict[*rqst]); 1307d7b241e6Sjeremylt } 1308d7b241e6Sjeremylt 13092b730f8bSJeremy L Thompson #define fCeedOperatorDestroy FORTRAN_NAME(ceedoperatordestroy, CEEDOPERATORDESTROY) 13103b271f31SJed Brown CEED_EXTERN void fCeedOperatorDestroy(int *op, int *err) { 1311b22696a8SJeremy L Thompson if (*op == FORTRAN_NULL) return; 1312d7b241e6Sjeremylt *err = CeedOperatorDestroy(&CeedOperator_dict[*op]); 131316595307SJeremy L Thompson if (*err == 0) { 131416595307SJeremy L Thompson *op = FORTRAN_NULL; 1315d7b241e6Sjeremylt CeedOperator_n--; 1316d7b241e6Sjeremylt if (CeedOperator_n == 0) { 1317d7b241e6Sjeremylt *err = CeedFree(&CeedOperator_dict); 1318d7b241e6Sjeremylt CeedOperator_count = 0; 1319d7b241e6Sjeremylt CeedOperator_count_max = 0; 1320d7b241e6Sjeremylt } 1321d7b241e6Sjeremylt } 132216595307SJeremy L Thompson } 1323b00b43a6SJeremy L Thompson 1324b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 1325