13d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, 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> 10*2b730f8bSJeremy L Thompson #include <ceed-impl.h> 11*2b730f8bSJeremy L Thompson #include <ceed/backend.h> 12*2b730f8bSJeremy L Thompson #include <ceed/ceed.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 2515910d16Sjeremylt #define FORTRAN_BASIS_COLLOCATED -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]; \ 56*2b730f8bSJeremy 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)); \ 58*2b730f8bSJeremy 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) 64*2b730f8bSJeremy 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 80*2b730f8bSJeremy L Thompson #define fCeedIsDeterministic FORTRAN_NAME(ceedisdeterministic, CEEDISDETERMINISTIC) 81*2b730f8bSJeremy 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 85*2b730f8bSJeremy L Thompson #define fCeedGetPreferredMemType FORTRAN_NAME(ceedgetpreferredmemtype, CEEDGETPREFERREDMEMTYPE) 86*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedGetPreferredMemType(int *ceed, int *type, int *err) { *err = CeedGetPreferredMemType(Ceed_dict[*ceed], (CeedMemType *)type); } 87c907536fSjeremylt 880a0da059Sjeremylt #define fCeedView FORTRAN_NAME(ceedview, CEEDVIEW) 89*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedView(int *ceed, int *err) { *err = CeedView(Ceed_dict[*ceed], stdout); } 900a0da059Sjeremylt 91d7b241e6Sjeremylt #define fCeedDestroy FORTRAN_NAME(ceeddestroy, CEEDDESTROY) 923b271f31SJed Brown CEED_EXTERN void fCeedDestroy(int *ceed, int *err) { 93b22696a8SJeremy L Thompson if (*ceed == FORTRAN_NULL) return; 94d7b241e6Sjeremylt *err = CeedDestroy(&Ceed_dict[*ceed]); 95d7b241e6Sjeremylt 96d7b241e6Sjeremylt if (*err == 0) { 9716595307SJeremy L Thompson *ceed = FORTRAN_NULL; 98d7b241e6Sjeremylt Ceed_n--; 99d7b241e6Sjeremylt if (Ceed_n == 0) { 100d7b241e6Sjeremylt CeedFree(&Ceed_dict); 101d7b241e6Sjeremylt Ceed_count = 0; 102d7b241e6Sjeremylt Ceed_count_max = 0; 103d7b241e6Sjeremylt } 104d7b241e6Sjeremylt } 105d7b241e6Sjeremylt } 106d7b241e6Sjeremylt 107b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 108b00b43a6SJeremy L Thompson // CeedVector 109b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 110d7b241e6Sjeremylt static CeedVector *CeedVector_dict = NULL; 111d7b241e6Sjeremylt static int CeedVector_count = 0; 112d7b241e6Sjeremylt static int CeedVector_n = 0; 113d7b241e6Sjeremylt static int CeedVector_count_max = 0; 114d7b241e6Sjeremylt 115d7b241e6Sjeremylt #define fCeedVectorCreate FORTRAN_NAME(ceedvectorcreate, CEEDVECTORCREATE) 1163b271f31SJed Brown CEED_EXTERN void fCeedVectorCreate(int *ceed, int *length, int *vec, int *err) { 117d7b241e6Sjeremylt if (CeedVector_count == CeedVector_count_max) { 118d7b241e6Sjeremylt CeedVector_count_max += CeedVector_count_max / 2 + 1; 119d7b241e6Sjeremylt CeedRealloc(CeedVector_count_max, &CeedVector_dict); 120d7b241e6Sjeremylt } 121d7b241e6Sjeremylt 122d7b241e6Sjeremylt CeedVector *vec_ = &CeedVector_dict[CeedVector_count]; 123d7b241e6Sjeremylt *err = CeedVectorCreate(Ceed_dict[*ceed], *length, vec_); 124d7b241e6Sjeremylt 125d7b241e6Sjeremylt if (*err == 0) { 126d7b241e6Sjeremylt *vec = CeedVector_count++; 127d7b241e6Sjeremylt CeedVector_n++; 128d7b241e6Sjeremylt } 129d7b241e6Sjeremylt } 130d7b241e6Sjeremylt 131d7b241e6Sjeremylt #define fCeedVectorSetArray FORTRAN_NAME(ceedvectorsetarray, CEEDVECTORSETARRAY) 132*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorSetArray(int *vec, int *memtype, int *copymode, CeedScalar *array, int64_t *offset, int *err) { 133*2b730f8bSJeremy L Thompson *err = CeedVectorSetArray(CeedVector_dict[*vec], (CeedMemType)*memtype, (CeedCopyMode)*copymode, (CeedScalar *)(array + *offset)); 134d7b241e6Sjeremylt } 135d7b241e6Sjeremylt 1366a6c615bSJeremy L Thompson #define fCeedVectorTakeArray FORTRAN_NAME(ceedvectortakearray, CEEDVECTORTAKEARRAY) 137*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorTakeArray(int *vec, int *memtype, CeedScalar *array, int64_t *offset, int *err) { 1386a6c615bSJeremy L Thompson CeedScalar *b; 1396a6c615bSJeremy L Thompson CeedVector vec_ = CeedVector_dict[*vec]; 1406a6c615bSJeremy L Thompson *err = CeedVectorTakeArray(vec_, (CeedMemType)*memtype, &b); 1416a6c615bSJeremy L Thompson *offset = b - array; 1426a6c615bSJeremy L Thompson } 1436a6c615bSJeremy L Thompson 14454540941SJeremy L Thompson #define fCeedVectorSyncArray FORTRAN_NAME(ceedvectorsyncarray, CEEDVECTORSYNCARRAY) 145*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorSyncArray(int *vec, int *memtype, int *err) { *err = CeedVectorSyncArray(CeedVector_dict[*vec], (CeedMemType)*memtype); } 14654540941SJeremy L Thompson 147d7b241e6Sjeremylt #define fCeedVectorSetValue FORTRAN_NAME(ceedvectorsetvalue, CEEDVECTORSETVALUE) 148*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorSetValue(int *vec, CeedScalar *value, int *err) { *err = CeedVectorSetValue(CeedVector_dict[*vec], *value); } 149d7b241e6Sjeremylt 150d7b241e6Sjeremylt #define fCeedVectorGetArray FORTRAN_NAME(ceedvectorgetarray, CEEDVECTORGETARRAY) 151*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorGetArray(int *vec, int *memtype, CeedScalar *array, int64_t *offset, int *err) { 152d7b241e6Sjeremylt CeedScalar *b; 153d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 154399bb197SJed Brown *err = CeedVectorGetArray(vec_, (CeedMemType)*memtype, &b); 155cbfd6a90SThilina Rathnayake *offset = b - array; 156d7b241e6Sjeremylt } 157d7b241e6Sjeremylt 158*2b730f8bSJeremy L Thompson #define fCeedVectorGetArrayRead FORTRAN_NAME(ceedvectorgetarrayread, CEEDVECTORGETARRAYREAD) 159*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorGetArrayRead(int *vec, int *memtype, CeedScalar *array, int64_t *offset, int *err) { 160d7b241e6Sjeremylt const CeedScalar *b; 161d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 162399bb197SJed Brown *err = CeedVectorGetArrayRead(vec_, (CeedMemType)*memtype, &b); 163cbfd6a90SThilina Rathnayake *offset = b - array; 164d7b241e6Sjeremylt } 165d7b241e6Sjeremylt 166*2b730f8bSJeremy L Thompson #define fCeedVectorGetArrayWrite FORTRAN_NAME(ceedvectorgetarraywrite, CEEDVECTORGETARRAYWRITE) 167*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorGetArrayWrite(int *vec, int *memtype, CeedScalar *array, int64_t *offset, int *err) { 1689c774eddSJeremy L Thompson CeedScalar *b; 1699c774eddSJeremy L Thompson CeedVector vec_ = CeedVector_dict[*vec]; 1709c774eddSJeremy L Thompson *err = CeedVectorGetArrayWrite(vec_, (CeedMemType)*memtype, &b); 1719c774eddSJeremy L Thompson *offset = b - array; 1729c774eddSJeremy L Thompson } 1739c774eddSJeremy L Thompson 174*2b730f8bSJeremy L Thompson #define fCeedVectorRestoreArray FORTRAN_NAME(ceedvectorrestorearray, CEEDVECTORRESTOREARRAY) 175*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorRestoreArray(int *vec, CeedScalar *array, int64_t *offset, int *err) { 1769647a07eSDavid Medina CeedScalar *offsetArray = array + *offset; 1779647a07eSDavid Medina *err = CeedVectorRestoreArray(CeedVector_dict[*vec], &offsetArray); 1786456524eSJed Brown *offset = 0; 179d7b241e6Sjeremylt } 180d7b241e6Sjeremylt 181*2b730f8bSJeremy L Thompson #define fCeedVectorRestoreArrayRead FORTRAN_NAME(ceedvectorrestorearrayread, CEEDVECTORRESTOREARRAYREAD) 182*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorRestoreArrayRead(int *vec, const CeedScalar *array, int64_t *offset, int *err) { 183d7b241e6Sjeremylt *err = CeedVectorRestoreArrayRead(CeedVector_dict[*vec], &array); 1846456524eSJed Brown *offset = 0; 185d7b241e6Sjeremylt } 186d7b241e6Sjeremylt 187*2b730f8bSJeremy L Thompson #define fCeedVectorNorm FORTRAN_NAME(ceedvectornorm, CEEDVECTORNORM) 188*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorNorm(int *vec, int *type, CeedScalar *norm, int *err) { 189399bb197SJed Brown *err = CeedVectorNorm(CeedVector_dict[*vec], (CeedNormType)*type, norm); 190547d9b97Sjeremylt } 191547d9b97Sjeremylt 192*2b730f8bSJeremy L Thompson #define fCeedVectorReciprocal FORTRAN_NAME(ceedvectorreciprocal, CEEDVECTORRECIPROCAL) 193*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorReciprocal(int *vec, int *err) { *err = CeedVectorReciprocal(CeedVector_dict[*vec]); } 194d99fa3c5SJeremy L Thompson 195d7b241e6Sjeremylt #define fCeedVectorView FORTRAN_NAME(ceedvectorview, CEEDVECTORVIEW) 196*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedVectorView(int *vec, int *err) { *err = CeedVectorView(CeedVector_dict[*vec], "%12.8f", stdout); } 197d7b241e6Sjeremylt 198d7b241e6Sjeremylt #define fCeedVectorDestroy FORTRAN_NAME(ceedvectordestroy, CEEDVECTORDESTROY) 1993b271f31SJed Brown CEED_EXTERN void fCeedVectorDestroy(int *vec, int *err) { 200b22696a8SJeremy L Thompson if (*vec == FORTRAN_NULL) return; 201d7b241e6Sjeremylt *err = CeedVectorDestroy(&CeedVector_dict[*vec]); 202d7b241e6Sjeremylt 203d7b241e6Sjeremylt if (*err == 0) { 20416595307SJeremy L Thompson *vec = FORTRAN_NULL; 205d7b241e6Sjeremylt CeedVector_n--; 206d7b241e6Sjeremylt if (CeedVector_n == 0) { 207d7b241e6Sjeremylt CeedFree(&CeedVector_dict); 208d7b241e6Sjeremylt CeedVector_count = 0; 209d7b241e6Sjeremylt CeedVector_count_max = 0; 210d7b241e6Sjeremylt } 211d7b241e6Sjeremylt } 212d7b241e6Sjeremylt } 213d7b241e6Sjeremylt 214b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 215b00b43a6SJeremy L Thompson // CeedElemRestriction 216b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 217d7b241e6Sjeremylt static CeedElemRestriction *CeedElemRestriction_dict = NULL; 218d7b241e6Sjeremylt static int CeedElemRestriction_count = 0; 219d7b241e6Sjeremylt static int CeedElemRestriction_n = 0; 220d7b241e6Sjeremylt static int CeedElemRestriction_count_max = 0; 221d7b241e6Sjeremylt 222*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionCreate FORTRAN_NAME(ceedelemrestrictioncreate, CEEDELEMRESTRICTIONCREATE) 223*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionCreate(int *ceed, int *nelements, int *esize, int *num_comp, int *comp_stride, int *lsize, int *memtype, 224*2b730f8bSJeremy L Thompson int *copymode, const int *offsets, int *elemrestriction, int *err) { 225d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 226d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 227d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 228d7b241e6Sjeremylt } 229d7b241e6Sjeremylt 230d979a051Sjeremylt const int *offsets_ = offsets; 231d7b241e6Sjeremylt 232*2b730f8bSJeremy L Thompson CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 233*2b730f8bSJeremy L Thompson *err = CeedElemRestrictionCreate(Ceed_dict[*ceed], *nelements, *esize, *num_comp, *comp_stride, *lsize, (CeedMemType)*memtype, 234*2b730f8bSJeremy L Thompson (CeedCopyMode)*copymode, offsets_, elemrestriction_); 235d7b241e6Sjeremylt 236d7b241e6Sjeremylt if (*err == 0) { 237d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 238d7b241e6Sjeremylt CeedElemRestriction_n++; 239d7b241e6Sjeremylt } 240d7b241e6Sjeremylt } 241d7b241e6Sjeremylt 242*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionCreateStrided FORTRAN_NAME(ceedelemrestrictioncreatestrided, CEEDELEMRESTRICTIONCREATESTRIDED) 243*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionCreateStrided(int *ceed, int *nelements, int *esize, int *num_comp, int *lsize, int *strides, 244d7b241e6Sjeremylt int *elemrestriction, int *err) { 245d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 246d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 247d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 248d7b241e6Sjeremylt } 249d7b241e6Sjeremylt 250*2b730f8bSJeremy L Thompson CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 251*2b730f8bSJeremy L Thompson *err = CeedElemRestrictionCreateStrided(Ceed_dict[*ceed], *nelements, *esize, *num_comp, *lsize, 252*2b730f8bSJeremy L Thompson *strides == FORTRAN_STRIDES_BACKEND ? CEED_STRIDES_BACKEND : strides, elemrestriction_); 253d7b241e6Sjeremylt if (*err == 0) { 254d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 255d7b241e6Sjeremylt CeedElemRestriction_n++; 256d7b241e6Sjeremylt } 257d7b241e6Sjeremylt } 258d7b241e6Sjeremylt 259*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionCreateBlocked FORTRAN_NAME(ceedelemrestrictioncreateblocked, CEEDELEMRESTRICTIONCREATEBLOCKED) 260*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionCreateBlocked(int *ceed, int *nelements, int *esize, int *blocksize, int *num_comp, int *comp_stride, int *lsize, 261*2b730f8bSJeremy L Thompson int *mtype, int *cmode, int *blkindices, int *elemrestriction, int *err) { 262d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 263d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 264d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 265d7b241e6Sjeremylt } 266d7b241e6Sjeremylt 267*2b730f8bSJeremy L Thompson CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 268*2b730f8bSJeremy L Thompson *err = CeedElemRestrictionCreateBlocked(Ceed_dict[*ceed], *nelements, *esize, *blocksize, *num_comp, *comp_stride, *lsize, (CeedMemType)*mtype, 269*2b730f8bSJeremy L Thompson (CeedCopyMode)*cmode, blkindices, elemrestriction_); 270d7b241e6Sjeremylt 271d7b241e6Sjeremylt if (*err == 0) { 272d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 273d7b241e6Sjeremylt CeedElemRestriction_n++; 274d7b241e6Sjeremylt } 275d7b241e6Sjeremylt } 276d7b241e6Sjeremylt 277*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionCreateBlockedStrided FORTRAN_NAME(ceedelemrestrictioncreateblockedstrided, CEEDELEMRESTRICTIONCREATEBLOCKEDSTRIDED) 278*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionCreateBlockedStrided(int *ceed, int *nelements, int *esize, int *blk_size, int *num_comp, int *lsize, 279*2b730f8bSJeremy L Thompson int *strides, int *elemrestriction, int *err) { 2807509a596Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 2817509a596Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 2827509a596Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 2837509a596Sjeremylt } 2847509a596Sjeremylt 285*2b730f8bSJeremy L Thompson CeedElemRestriction *elemrestriction_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 286*2b730f8bSJeremy L Thompson *err = CeedElemRestrictionCreateBlockedStrided(Ceed_dict[*ceed], *nelements, *esize, *blk_size, *num_comp, *lsize, strides, elemrestriction_); 2877509a596Sjeremylt if (*err == 0) { 2887509a596Sjeremylt *elemrestriction = CeedElemRestriction_count++; 2897509a596Sjeremylt CeedElemRestriction_n++; 2907509a596Sjeremylt } 2917509a596Sjeremylt } 2927509a596Sjeremylt 293d7b241e6Sjeremylt static CeedRequest *CeedRequest_dict = NULL; 294d7b241e6Sjeremylt static int CeedRequest_count = 0; 295d7b241e6Sjeremylt static int CeedRequest_n = 0; 296d7b241e6Sjeremylt static int CeedRequest_count_max = 0; 297d7b241e6Sjeremylt 298*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionApply FORTRAN_NAME(ceedelemrestrictionapply, CEEDELEMRESTRICTIONAPPLY) 299*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionApply(int *elemr, int *tmode, int *uvec, int *ruvec, int *rqst, int *err) { 300d7b241e6Sjeremylt int createRequest = 1; 301d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 302*2b730f8bSJeremy L Thompson if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) createRequest = 0; 303d7b241e6Sjeremylt 304d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 305d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 306d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 307d7b241e6Sjeremylt } 308d7b241e6Sjeremylt 309d7b241e6Sjeremylt CeedRequest *rqst_; 310d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 311d7b241e6Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED) rqst_ = CEED_REQUEST_ORDERED; 312d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 313d7b241e6Sjeremylt 314*2b730f8bSJeremy L Thompson *err = 315*2b730f8bSJeremy L Thompson CeedElemRestrictionApply(CeedElemRestriction_dict[*elemr], (CeedTransposeMode)*tmode, CeedVector_dict[*uvec], CeedVector_dict[*ruvec], rqst_); 316d7b241e6Sjeremylt 317d7b241e6Sjeremylt if (*err == 0 && createRequest) { 318d7b241e6Sjeremylt *rqst = CeedRequest_count++; 319d7b241e6Sjeremylt CeedRequest_n++; 320d7b241e6Sjeremylt } 321d7b241e6Sjeremylt } 322d7b241e6Sjeremylt 323*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionApplyBlock FORTRAN_NAME(ceedelemrestrictionapplyblock, CEEDELEMRESTRICTIONAPPLYBLOCK) 324*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionApplyBlock(int *elemr, int *block, int *tmode, int *uvec, int *ruvec, int *rqst, int *err) { 325be9261b7Sjeremylt int createRequest = 1; 326be9261b7Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 327*2b730f8bSJeremy L Thompson if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) createRequest = 0; 328be9261b7Sjeremylt 329be9261b7Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 330be9261b7Sjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 331be9261b7Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 332be9261b7Sjeremylt } 333be9261b7Sjeremylt 334be9261b7Sjeremylt CeedRequest *rqst_; 335be9261b7Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 336be9261b7Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED) rqst_ = CEED_REQUEST_ORDERED; 337be9261b7Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 338be9261b7Sjeremylt 339*2b730f8bSJeremy L Thompson *err = CeedElemRestrictionApplyBlock(CeedElemRestriction_dict[*elemr], *block, (CeedTransposeMode)*tmode, CeedVector_dict[*uvec], 340be9261b7Sjeremylt CeedVector_dict[*ruvec], rqst_); 341be9261b7Sjeremylt 342be9261b7Sjeremylt if (*err == 0 && createRequest) { 343be9261b7Sjeremylt *rqst = CeedRequest_count++; 344be9261b7Sjeremylt CeedRequest_n++; 345be9261b7Sjeremylt } 346be9261b7Sjeremylt } 347be9261b7Sjeremylt 348*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionGetMultiplicity FORTRAN_NAME(ceedelemrestrictiongetmultiplicity, CEEDELEMRESTRICTIONGETMULTIPLICITY) 349*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionGetMultiplicity(int *elemr, int *mult, int *err) { 350*2b730f8bSJeremy L Thompson *err = CeedElemRestrictionGetMultiplicity(CeedElemRestriction_dict[*elemr], CeedVector_dict[*mult]); 3511469ee4dSjeremylt } 3521469ee4dSjeremylt 353*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionGetELayout FORTRAN_NAME(ceedelemrestrictiongetelayout, CEEDELEMRESTRICTIONGETELAYOUT) 354*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionGetELayout(int *elemr, int *layout, int *err) { 35549fd234cSJeremy L Thompson CeedInt layout_c[3]; 356*2b730f8bSJeremy L Thompson *err = CeedElemRestrictionGetELayout(CeedElemRestriction_dict[*elemr], &layout_c); 357*2b730f8bSJeremy L Thompson for (int i = 0; i < 3; i++) layout[i] = layout_c[i]; 35849fd234cSJeremy L Thompson } 35949fd234cSJeremy L Thompson 360*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionView FORTRAN_NAME(ceedelemrestrictionview, CEEDELEMRESTRICTIONVIEW) 361*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedElemRestrictionView(int *elemr, int *err) { *err = CeedElemRestrictionView(CeedElemRestriction_dict[*elemr], stdout); } 3621469ee4dSjeremylt 363d7b241e6Sjeremylt #define fCeedRequestWait FORTRAN_NAME(ceedrequestwait, CEEDREQUESTWAIT) 3643b271f31SJed Brown CEED_EXTERN void fCeedRequestWait(int *rqst, int *err) { 365d7b241e6Sjeremylt // TODO Uncomment this once CeedRequestWait is implemented 366d7b241e6Sjeremylt //*err = CeedRequestWait(&CeedRequest_dict[*rqst]); 367d7b241e6Sjeremylt 368d7b241e6Sjeremylt if (*err == 0) { 369d7b241e6Sjeremylt CeedRequest_n--; 370d7b241e6Sjeremylt if (CeedRequest_n == 0) { 371d7b241e6Sjeremylt CeedFree(&CeedRequest_dict); 372d7b241e6Sjeremylt CeedRequest_count = 0; 373d7b241e6Sjeremylt CeedRequest_count_max = 0; 374d7b241e6Sjeremylt } 375d7b241e6Sjeremylt } 376d7b241e6Sjeremylt } 377d7b241e6Sjeremylt 378*2b730f8bSJeremy L Thompson #define fCeedElemRestrictionDestroy FORTRAN_NAME(ceedelemrestrictiondestroy, CEEDELEMRESTRICTIONDESTROY) 3793b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionDestroy(int *elem, int *err) { 380b22696a8SJeremy L Thompson if (*elem == FORTRAN_NULL) return; 381d7b241e6Sjeremylt *err = CeedElemRestrictionDestroy(&CeedElemRestriction_dict[*elem]); 382d7b241e6Sjeremylt 383d7b241e6Sjeremylt if (*err == 0) { 38416595307SJeremy L Thompson *elem = FORTRAN_NULL; 385d7b241e6Sjeremylt CeedElemRestriction_n--; 386d7b241e6Sjeremylt if (CeedElemRestriction_n == 0) { 387d7b241e6Sjeremylt CeedFree(&CeedElemRestriction_dict); 388d7b241e6Sjeremylt CeedElemRestriction_count = 0; 389d7b241e6Sjeremylt CeedElemRestriction_count_max = 0; 390d7b241e6Sjeremylt } 391d7b241e6Sjeremylt } 392d7b241e6Sjeremylt } 393d7b241e6Sjeremylt 394b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 395b00b43a6SJeremy L Thompson // CeedBasis 396b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 397d7b241e6Sjeremylt static CeedBasis *CeedBasis_dict = NULL; 398d7b241e6Sjeremylt static int CeedBasis_count = 0; 399d7b241e6Sjeremylt static int CeedBasis_n = 0; 400d7b241e6Sjeremylt static int CeedBasis_count_max = 0; 401d7b241e6Sjeremylt 402*2b730f8bSJeremy L Thompson #define fCeedBasisCreateTensorH1Lagrange FORTRAN_NAME(ceedbasiscreatetensorh1lagrange, CEEDBASISCREATETENSORH1LAGRANGE) 403*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisCreateTensorH1Lagrange(int *ceed, int *dim, int *num_comp, int *P, int *Q, int *quadmode, int *basis, int *err) { 404d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 405d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max / 2 + 1; 406d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 407d7b241e6Sjeremylt } 408d7b241e6Sjeremylt 409*2b730f8bSJeremy L Thompson *err = CeedBasisCreateTensorH1Lagrange(Ceed_dict[*ceed], *dim, *num_comp, *P, *Q, (CeedQuadMode)*quadmode, &CeedBasis_dict[CeedBasis_count]); 410d7b241e6Sjeremylt 411d7b241e6Sjeremylt if (*err == 0) { 412d7b241e6Sjeremylt *basis = CeedBasis_count++; 413d7b241e6Sjeremylt CeedBasis_n++; 414d7b241e6Sjeremylt } 415d7b241e6Sjeremylt } 416d7b241e6Sjeremylt 417*2b730f8bSJeremy L Thompson #define fCeedBasisCreateTensorH1 FORTRAN_NAME(ceedbasiscreatetensorh1, CEEDBASISCREATETENSORH1) 418*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisCreateTensorH1(int *ceed, int *dim, int *num_comp, int *P_1d, int *Q_1d, const CeedScalar *interp_1d, 419*2b730f8bSJeremy L Thompson const CeedScalar *grad_1d, const CeedScalar *q_ref_1d, const CeedScalar *q_weight_1d, int *basis, 4208795c945Sjeremylt int *err) { 421d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 422d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max / 2 + 1; 423d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 424d7b241e6Sjeremylt } 425d7b241e6Sjeremylt 426*2b730f8bSJeremy L Thompson *err = CeedBasisCreateTensorH1(Ceed_dict[*ceed], *dim, *num_comp, *P_1d, *Q_1d, interp_1d, grad_1d, q_ref_1d, q_weight_1d, 4278795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 428d7b241e6Sjeremylt 429d7b241e6Sjeremylt if (*err == 0) { 430d7b241e6Sjeremylt *basis = CeedBasis_count++; 431d7b241e6Sjeremylt CeedBasis_n++; 432d7b241e6Sjeremylt } 433d7b241e6Sjeremylt } 434d7b241e6Sjeremylt 435*2b730f8bSJeremy L Thompson #define fCeedBasisCreateH1 FORTRAN_NAME(ceedbasiscreateh1, CEEDBASISCREATEH1) 436*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisCreateH1(int *ceed, int *topo, int *num_comp, int *nnodes, int *nqpts, const CeedScalar *interp, const CeedScalar *grad, 437*2b730f8bSJeremy L Thompson const CeedScalar *qref, const CeedScalar *qweight, int *basis, int *err) { 438a8de75f0Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 439a8de75f0Sjeremylt CeedBasis_count_max += CeedBasis_count_max / 2 + 1; 440a8de75f0Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 441a8de75f0Sjeremylt } 442a8de75f0Sjeremylt 443*2b730f8bSJeremy L Thompson *err = CeedBasisCreateH1(Ceed_dict[*ceed], (CeedElemTopology)*topo, *num_comp, *nnodes, *nqpts, interp, grad, qref, qweight, 4448795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 445a8de75f0Sjeremylt 446a8de75f0Sjeremylt if (*err == 0) { 447a8de75f0Sjeremylt *basis = CeedBasis_count++; 448a8de75f0Sjeremylt CeedBasis_n++; 449a8de75f0Sjeremylt } 450a8de75f0Sjeremylt } 451a8de75f0Sjeremylt 452d7b241e6Sjeremylt #define fCeedBasisView FORTRAN_NAME(ceedbasisview, CEEDBASISVIEW) 453*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisView(int *basis, int *err) { *err = CeedBasisView(CeedBasis_dict[*basis], stdout); } 454d7b241e6Sjeremylt 455*2b730f8bSJeremy L Thompson #define fCeedQRFactorization FORTRAN_NAME(ceedqrfactorization, CEEDQRFACTORIZATION) 456*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQRFactorization(int *ceed, CeedScalar *mat, CeedScalar *tau, int *m, int *n, int *err) { 457a7bd39daSjeremylt *err = CeedQRFactorization(Ceed_dict[*ceed], mat, tau, *m, *n); 458d7b241e6Sjeremylt } 459d7b241e6Sjeremylt 460*2b730f8bSJeremy L Thompson #define fCeedHouseholderApplyQ FORTRAN_NAME(ceedhouseholderapplyq, CEEDHOUSEHOLDERAPPLYQ) 461*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedHouseholderApplyQ(CeedScalar *A, CeedScalar *Q, CeedScalar *tau, int *t_mode, int *m, int *n, int *k, int *row, int *col, 462*2b730f8bSJeremy L Thompson int *err) { 463*2b730f8bSJeremy L Thompson *err = CeedHouseholderApplyQ(A, Q, tau, (CeedTransposeMode)*t_mode, *m, *n, *k, *row, *col); 4649ac7b42eSJeremy L Thompson } 4659ac7b42eSJeremy L Thompson 466*2b730f8bSJeremy L Thompson #define fCeedSymmetricSchurDecomposition FORTRAN_NAME(ceedsymmetricschurdecomposition, CEEDSYMMETRICSCHURDECOMPOSITION) 467*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedSymmetricSchurDecomposition(int *ceed, CeedScalar *mat, CeedScalar *lambda, int *n, int *err) { 46852bfb9bbSJeremy L Thompson *err = CeedSymmetricSchurDecomposition(Ceed_dict[*ceed], mat, lambda, *n); 46952bfb9bbSJeremy L Thompson } 47052bfb9bbSJeremy L Thompson 471*2b730f8bSJeremy L Thompson #define fCeedSimultaneousDiagonalization FORTRAN_NAME(ceedsimultaneousdiagonalization, CEEDSIMULTANEOUSDIAGONALIZATION) 472*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedSimultaneousDiagonalization(int *ceed, CeedScalar *matA, CeedScalar *matB, CeedScalar *x, CeedScalar *lambda, int *n, 473*2b730f8bSJeremy L Thompson int *err) { 474*2b730f8bSJeremy L Thompson *err = CeedSimultaneousDiagonalization(Ceed_dict[*ceed], matA, matB, x, lambda, *n); 47552bfb9bbSJeremy L Thompson } 47652bfb9bbSJeremy L Thompson 477*2b730f8bSJeremy L Thompson #define fCeedBasisGetCollocatedGrad FORTRAN_NAME(ceedbasisgetcollocatedgrad, CEEDBASISGETCOLLOCATEDGRAD) 478*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetCollocatedGrad(int *basis, CeedScalar *colo_grad_1d, int *err) { 479d1d35e2fSjeremylt *err = CeedBasisGetCollocatedGrad(CeedBasis_dict[*basis], colo_grad_1d); 480d7b241e6Sjeremylt } 481d7b241e6Sjeremylt 482d7b241e6Sjeremylt #define fCeedBasisApply FORTRAN_NAME(ceedbasisapply, CEEDBASISAPPLY) 483*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisApply(int *basis, int *num_elem, int *tmode, int *eval_mode, int *u, int *v, int *err) { 484*2b730f8bSJeremy L Thompson *err = CeedBasisApply(CeedBasis_dict[*basis], *num_elem, (CeedTransposeMode)*tmode, (CeedEvalMode)*eval_mode, 485*2b730f8bSJeremy L Thompson *u == FORTRAN_VECTOR_NONE ? CEED_VECTOR_NONE : CeedVector_dict[*u], CeedVector_dict[*v]); 486d7b241e6Sjeremylt } 487d7b241e6Sjeremylt 488*2b730f8bSJeremy L Thompson #define fCeedBasisGetNumNodes FORTRAN_NAME(ceedbasisgetnumnodes, CEEDBASISGETNUMNODES) 489*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetNumNodes(int *basis, int *P, int *err) { *err = CeedBasisGetNumNodes(CeedBasis_dict[*basis], P); } 490d7b241e6Sjeremylt 491*2b730f8bSJeremy L Thompson #define fCeedBasisGetNumQuadraturePoints FORTRAN_NAME(ceedbasisgetnumquadraturepoints, CEEDBASISGETNUMQUADRATUREPOINTS) 492*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetNumQuadraturePoints(int *basis, int *Q, int *err) { *err = CeedBasisGetNumQuadraturePoints(CeedBasis_dict[*basis], Q); } 493d7b241e6Sjeremylt 494*2b730f8bSJeremy L Thompson #define fCeedBasisGetInterp1D FORTRAN_NAME(ceedbasisgetinterp1d, CEEDBASISGETINTERP1D) 495*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetInterp1D(int *basis, CeedScalar *interp_1d, int64_t *offset, int *err) { 496d99fa3c5SJeremy L Thompson const CeedScalar *interp1d_; 497d99fa3c5SJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 498d99fa3c5SJeremy L Thompson *err = CeedBasisGetInterp1D(basis_, &interp1d_); 499d1d35e2fSjeremylt *offset = interp1d_ - interp_1d; 500d99fa3c5SJeremy L Thompson } 501d99fa3c5SJeremy L Thompson 502*2b730f8bSJeremy L Thompson #define fCeedBasisGetGrad1D FORTRAN_NAME(ceedbasisgetgrad1d, CEEDBASISGETGRAD1D) 503*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetGrad1D(int *basis, CeedScalar *grad_1d, int64_t *offset, int *err) { 5049ac7b42eSJeremy L Thompson const CeedScalar *grad1d_; 5059ac7b42eSJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 5069ac7b42eSJeremy L Thompson *err = CeedBasisGetGrad1D(basis_, &grad1d_); 5079ac7b42eSJeremy L Thompson *offset = grad1d_ - grad_1d; 5089ac7b42eSJeremy L Thompson } 5099ac7b42eSJeremy L Thompson 510*2b730f8bSJeremy L Thompson #define fCeedBasisGetQRef FORTRAN_NAME(ceedbasisgetqref, CEEDBASISGETQREF) 511*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedBasisGetQRef(int *basis, CeedScalar *q_ref, int64_t *offset, int *err) { 5129ac7b42eSJeremy L Thompson const CeedScalar *qref_; 5139ac7b42eSJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 5149ac7b42eSJeremy L Thompson *err = CeedBasisGetQRef(basis_, &qref_); 5159ac7b42eSJeremy L Thompson *offset = qref_ - q_ref; 5169ac7b42eSJeremy L Thompson } 5179ac7b42eSJeremy L Thompson 518d7b241e6Sjeremylt #define fCeedBasisDestroy FORTRAN_NAME(ceedbasisdestroy, CEEDBASISDESTROY) 5193b271f31SJed Brown CEED_EXTERN void fCeedBasisDestroy(int *basis, int *err) { 520b22696a8SJeremy L Thompson if (*basis == FORTRAN_NULL) return; 521d7b241e6Sjeremylt *err = CeedBasisDestroy(&CeedBasis_dict[*basis]); 522d7b241e6Sjeremylt 523d7b241e6Sjeremylt if (*err == 0) { 52416595307SJeremy L Thompson *basis = FORTRAN_NULL; 525d7b241e6Sjeremylt CeedBasis_n--; 526d7b241e6Sjeremylt if (CeedBasis_n == 0) { 527d7b241e6Sjeremylt CeedFree(&CeedBasis_dict); 528d7b241e6Sjeremylt CeedBasis_count = 0; 529d7b241e6Sjeremylt CeedBasis_count_max = 0; 530d7b241e6Sjeremylt } 531d7b241e6Sjeremylt } 532d7b241e6Sjeremylt } 533d7b241e6Sjeremylt 534d7b241e6Sjeremylt #define fCeedGaussQuadrature FORTRAN_NAME(ceedgaussquadrature, CEEDGAUSSQUADRATURE) 535*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedGaussQuadrature(int *Q, CeedScalar *q_ref_1d, CeedScalar *q_weight_1d, int *err) { 536d1d35e2fSjeremylt *err = CeedGaussQuadrature(*Q, q_ref_1d, q_weight_1d); 537d7b241e6Sjeremylt } 538d7b241e6Sjeremylt 539*2b730f8bSJeremy L Thompson #define fCeedLobattoQuadrature FORTRAN_NAME(ceedlobattoquadrature, CEEDLOBATTOQUADRATURE) 540*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedLobattoQuadrature(int *Q, CeedScalar *q_ref_1d, CeedScalar *q_weight_1d, int *err) { 541d1d35e2fSjeremylt *err = CeedLobattoQuadrature(*Q, q_ref_1d, q_weight_1d); 542d7b241e6Sjeremylt } 543d7b241e6Sjeremylt 544b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 545777ff853SJeremy L Thompson // CeedQFunctionContext 546777ff853SJeremy L Thompson // ----------------------------------------------------------------------------- 547777ff853SJeremy L Thompson static CeedQFunctionContext *CeedQFunctionContext_dict = NULL; 548777ff853SJeremy L Thompson static int CeedQFunctionContext_count = 0; 549777ff853SJeremy L Thompson static int CeedQFunctionContext_n = 0; 550777ff853SJeremy L Thompson static int CeedQFunctionContext_count_max = 0; 551777ff853SJeremy L Thompson 552*2b730f8bSJeremy L Thompson #define fCeedQFunctionContextCreate FORTRAN_NAME(ceedqfunctioncontextcreate, CEEDQFUNCTIONCONTEXTCREATE) 5533b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextCreate(int *ceed, int *ctx, int *err) { 554777ff853SJeremy L Thompson if (CeedQFunctionContext_count == CeedQFunctionContext_count_max) { 555777ff853SJeremy L Thompson CeedQFunctionContext_count_max += CeedQFunctionContext_count_max / 2 + 1; 556777ff853SJeremy L Thompson CeedRealloc(CeedQFunctionContext_count_max, &CeedQFunctionContext_dict); 557777ff853SJeremy L Thompson } 558777ff853SJeremy L Thompson 559*2b730f8bSJeremy L Thompson CeedQFunctionContext *ctx_ = &CeedQFunctionContext_dict[CeedQFunctionContext_count]; 560777ff853SJeremy L Thompson 561777ff853SJeremy L Thompson *err = CeedQFunctionContextCreate(Ceed_dict[*ceed], ctx_); 562777ff853SJeremy L Thompson if (*err) return; 563777ff853SJeremy L Thompson *ctx = CeedQFunctionContext_count++; 564777ff853SJeremy L Thompson CeedQFunctionContext_n++; 565777ff853SJeremy L Thompson } 566777ff853SJeremy L Thompson 567*2b730f8bSJeremy L Thompson #define fCeedQFunctionContextSetData FORTRAN_NAME(ceedqfunctioncontextsetdata, CEEDQFUNCTIONCONTEXTSETDATA) 568*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextSetData(int *ctx, int *memtype, int *copymode, CeedInt *n, CeedScalar *data, int64_t *offset, int *err) { 569d1d35e2fSjeremylt size_t ctx_size = ((size_t)*n) * sizeof(CeedScalar); 570*2b730f8bSJeremy L Thompson *err = CeedQFunctionContextSetData(CeedQFunctionContext_dict[*ctx], (CeedMemType)*memtype, (CeedCopyMode)*copymode, ctx_size, data + *offset); 571777ff853SJeremy L Thompson } 572777ff853SJeremy L Thompson 573*2b730f8bSJeremy L Thompson #define fCeedQFunctionContextGetData FORTRAN_NAME(ceedqfunctioncontextgetdata, CEEDQFUNCTIONCONTEXTGETDATA) 574*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextGetData(int *ctx, int *memtype, CeedScalar *data, int64_t *offset, int *err) { 575777ff853SJeremy L Thompson CeedScalar *b; 576777ff853SJeremy L Thompson CeedQFunctionContext ctx_ = CeedQFunctionContext_dict[*ctx]; 577777ff853SJeremy L Thompson *err = CeedQFunctionContextGetData(ctx_, (CeedMemType)*memtype, &b); 578777ff853SJeremy L Thompson *offset = b - data; 579777ff853SJeremy L Thompson } 580777ff853SJeremy L Thompson 581*2b730f8bSJeremy L Thompson #define fCeedQFunctionContextRestoreData FORTRAN_NAME(ceedqfunctioncontextrestoredata, CEEDQFUNCTIONCONTEXTRESTOREDATA) 582*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextRestoreData(int *ctx, CeedScalar *data, int64_t *offset, int *err) { 583*2b730f8bSJeremy L Thompson *err = CeedQFunctionContextRestoreData(CeedQFunctionContext_dict[*ctx], (void **)&data); 584777ff853SJeremy L Thompson *offset = 0; 585777ff853SJeremy L Thompson } 586777ff853SJeremy L Thompson 587*2b730f8bSJeremy L Thompson #define fCeedQFunctionContextView FORTRAN_NAME(ceedqfunctioncontextview, CEEDQFUNCTIONCONTEXTVIEW) 588*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionContextView(int *ctx, int *err) { *err = CeedQFunctionContextView(CeedQFunctionContext_dict[*ctx], stdout); } 589777ff853SJeremy L Thompson 590*2b730f8bSJeremy L Thompson #define fCeedQFunctionContextDestroy FORTRAN_NAME(ceedqfunctioncontextdestroy, CEEDQFUNCTIONCONTEXTDESTROY) 5913b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextDestroy(int *ctx, int *err) { 592777ff853SJeremy L Thompson if (*ctx == FORTRAN_NULL) return; 593777ff853SJeremy L Thompson *err = CeedQFunctionContextDestroy(&CeedQFunctionContext_dict[*ctx]); 594777ff853SJeremy L Thompson 595777ff853SJeremy L Thompson if (*err == 0) { 596777ff853SJeremy L Thompson *ctx = FORTRAN_NULL; 597777ff853SJeremy L Thompson CeedQFunctionContext_n--; 598777ff853SJeremy L Thompson if (CeedQFunctionContext_n == 0) { 599777ff853SJeremy L Thompson CeedFree(&CeedQFunctionContext_dict); 600777ff853SJeremy L Thompson CeedQFunctionContext_count = 0; 601777ff853SJeremy L Thompson CeedQFunctionContext_count_max = 0; 602777ff853SJeremy L Thompson } 603777ff853SJeremy L Thompson } 604777ff853SJeremy L Thompson } 605777ff853SJeremy L Thompson 606777ff853SJeremy L Thompson // ----------------------------------------------------------------------------- 607b00b43a6SJeremy L Thompson // CeedQFunction 608b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 609d7b241e6Sjeremylt static CeedQFunction *CeedQFunction_dict = NULL; 610d7b241e6Sjeremylt static int CeedQFunction_count = 0; 611d7b241e6Sjeremylt static int CeedQFunction_n = 0; 612d7b241e6Sjeremylt static int CeedQFunction_count_max = 0; 613d7b241e6Sjeremylt 614*2b730f8bSJeremy L Thompson static int CeedQFunctionFortranStub(void *ctx, int nq, const CeedScalar *const *u, CeedScalar *const *v) { 615777ff853SJeremy L Thompson CeedFortranContext fctx = ctx; 6168cb0412aSJeremy L Thompson CeedQFunctionContext inner_ctx = fctx->inner_ctx; 617d7b241e6Sjeremylt int ierr; 618d7b241e6Sjeremylt 619777ff853SJeremy L Thompson CeedScalar *ctx_ = NULL; 620777ff853SJeremy L Thompson // Note: Device backends are generating their own kernels from 621777ff853SJeremy L Thompson // single source files, so only Host backends need to 622777ff853SJeremy L Thompson // use this Fortran stub. 6238cb0412aSJeremy L Thompson if (inner_ctx) { 6248cb0412aSJeremy L Thompson ierr = CeedQFunctionContextGetData(inner_ctx, CEED_MEM_HOST, &ctx_); 625777ff853SJeremy L Thompson CeedChk(ierr); 626777ff853SJeremy L Thompson } 627777ff853SJeremy L Thompson 628*2b730f8bSJeremy 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], 629*2b730f8bSJeremy 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); 630777ff853SJeremy L Thompson 6318cb0412aSJeremy L Thompson if (inner_ctx) { 6328cb0412aSJeremy L Thompson ierr = CeedQFunctionContextRestoreData(inner_ctx, (void *)&ctx_); 633777ff853SJeremy L Thompson CeedChk(ierr); 634777ff853SJeremy L Thompson } 635777ff853SJeremy L Thompson 636d7b241e6Sjeremylt return ierr; 637d7b241e6Sjeremylt } 638d7b241e6Sjeremylt 639*2b730f8bSJeremy L Thompson #define fCeedQFunctionCreateInterior FORTRAN_NAME(ceedqfunctioncreateinterior, CEEDQFUNCTIONCREATEINTERIOR) 640*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionCreateInterior( 641*2b730f8bSJeremy L Thompson int *ceed, int *vec_length, 642*2b730f8bSJeremy L Thompson void (*f)(void *ctx, int *nq, const CeedScalar *u, const CeedScalar *u1, const CeedScalar *u2, const CeedScalar *u3, const CeedScalar *u4, 643*2b730f8bSJeremy L Thompson const CeedScalar *u5, const CeedScalar *u6, const CeedScalar *u7, const CeedScalar *u8, const CeedScalar *u9, const CeedScalar *u10, 644*2b730f8bSJeremy L Thompson const CeedScalar *u11, const CeedScalar *u12, const CeedScalar *u13, const CeedScalar *u14, const CeedScalar *u15, CeedScalar *v, 645*2b730f8bSJeremy L Thompson CeedScalar *v1, CeedScalar *v2, CeedScalar *v3, CeedScalar *v4, CeedScalar *v5, CeedScalar *v6, CeedScalar *v7, CeedScalar *v8, 646*2b730f8bSJeremy L Thompson CeedScalar *v9, CeedScalar *v10, CeedScalar *v11, CeedScalar *v12, CeedScalar *v13, CeedScalar *v14, CeedScalar *v15, int *err), 647*2b730f8bSJeremy L Thompson const char *source, int *qf, int *err, fortran_charlen_t source_len) { 648288c0443SJeremy L Thompson FIX_STRING(source); 649d7b241e6Sjeremylt if (CeedQFunction_count == CeedQFunction_count_max) { 650d7b241e6Sjeremylt CeedQFunction_count_max += CeedQFunction_count_max / 2 + 1; 651d7b241e6Sjeremylt CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 652d7b241e6Sjeremylt } 653d7b241e6Sjeremylt 654d7b241e6Sjeremylt CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 655*2b730f8bSJeremy L Thompson *err = CeedQFunctionCreateInterior(Ceed_dict[*ceed], *vec_length, CeedQFunctionFortranStub, source_c, qf_); 656d7b241e6Sjeremylt 657d7b241e6Sjeremylt if (*err == 0) { 658d7b241e6Sjeremylt *qf = CeedQFunction_count++; 659d7b241e6Sjeremylt CeedQFunction_n++; 660d7b241e6Sjeremylt } 661d7b241e6Sjeremylt 662777ff853SJeremy L Thompson CeedFortranContext fctxdata; 663777ff853SJeremy L Thompson *err = CeedCalloc(1, &fctxdata); 664d7b241e6Sjeremylt if (*err) return; 665*2b730f8bSJeremy L Thompson fctxdata->f = f; 666*2b730f8bSJeremy L Thompson fctxdata->inner_ctx = NULL; 667777ff853SJeremy L Thompson CeedQFunctionContext fctx; 668777ff853SJeremy L Thompson *err = CeedQFunctionContextCreate(Ceed_dict[*ceed], &fctx); 669777ff853SJeremy L Thompson if (*err) return; 670*2b730f8bSJeremy L Thompson *err = CeedQFunctionContextSetData(fctx, CEED_MEM_HOST, CEED_OWN_POINTER, sizeof(*fctxdata), fctxdata); 671777ff853SJeremy L Thompson if (*err) return; 672777ff853SJeremy L Thompson *err = CeedQFunctionSetContext(*qf_, fctx); 673777ff853SJeremy L Thompson if (*err) return; 674777ff853SJeremy L Thompson CeedQFunctionContextDestroy(&fctx); 675777ff853SJeremy L Thompson if (*err) return; 676d7b241e6Sjeremylt 677777ff853SJeremy L Thompson *err = CeedQFunctionSetFortranStatus(*qf_, true); 678d7b241e6Sjeremylt } 679d7b241e6Sjeremylt 680*2b730f8bSJeremy L Thompson #define fCeedQFunctionCreateInteriorByName FORTRAN_NAME(ceedqfunctioncreateinteriorbyname, CEEDQFUNCTIONCREATEINTERIORBYNAME) 681*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionCreateInteriorByName(int *ceed, const char *name, int *qf, int *err, fortran_charlen_t name_len) { 682288c0443SJeremy L Thompson FIX_STRING(name); 683288c0443SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 684288c0443SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max / 2 + 1; 685288c0443SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 686288c0443SJeremy L Thompson } 687288c0443SJeremy L Thompson 688288c0443SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 689288c0443SJeremy L Thompson *err = CeedQFunctionCreateInteriorByName(Ceed_dict[*ceed], name_c, qf_); 690288c0443SJeremy L Thompson 691288c0443SJeremy L Thompson if (*err == 0) { 692288c0443SJeremy L Thompson *qf = CeedQFunction_count++; 693288c0443SJeremy L Thompson CeedQFunction_n++; 694288c0443SJeremy L Thompson } 695288c0443SJeremy L Thompson } 696288c0443SJeremy L Thompson 697*2b730f8bSJeremy L Thompson #define fCeedQFunctionCreateIdentity FORTRAN_NAME(ceedqfunctioncreateidentity, CEEDQFUNCTIONCREATEIDENTITY) 698*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionCreateIdentity(int *ceed, int *size, int *inmode, int *outmode, int *qf, int *err) { 6990219ea01SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 7000219ea01SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max / 2 + 1; 7010219ea01SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 7020219ea01SJeremy L Thompson } 7030219ea01SJeremy L Thompson 7040219ea01SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 705*2b730f8bSJeremy L Thompson *err = CeedQFunctionCreateIdentity(Ceed_dict[*ceed], *size, (CeedEvalMode)*inmode, (CeedEvalMode)*outmode, qf_); 7060219ea01SJeremy L Thompson 7070219ea01SJeremy L Thompson if (*err == 0) { 7080219ea01SJeremy L Thompson *qf = CeedQFunction_count++; 7090219ea01SJeremy L Thompson CeedQFunction_n++; 7100219ea01SJeremy L Thompson } 7110219ea01SJeremy L Thompson } 7120219ea01SJeremy L Thompson 713*2b730f8bSJeremy L Thompson #define fCeedQFunctionAddInput FORTRAN_NAME(ceedqfunctionaddinput, CEEDQFUNCTIONADDINPUT) 714*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionAddInput(int *qf, const char *field_name, CeedInt *num_comp, CeedEvalMode *eval_mode, int *err, 715d1d35e2fSjeremylt fortran_charlen_t field_name_len) { 716d1d35e2fSjeremylt FIX_STRING(field_name); 717d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 718d7b241e6Sjeremylt 719d1d35e2fSjeremylt *err = CeedQFunctionAddInput(qf_, field_name_c, *num_comp, *eval_mode); 720d7b241e6Sjeremylt } 721d7b241e6Sjeremylt 722*2b730f8bSJeremy L Thompson #define fCeedQFunctionAddOutput FORTRAN_NAME(ceedqfunctionaddoutput, CEEDQFUNCTIONADDOUTPUT) 723*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedQFunctionAddOutput(int *qf, const char *field_name, CeedInt *num_comp, CeedEvalMode *eval_mode, int *err, 724d1d35e2fSjeremylt fortran_charlen_t field_name_len) { 725d1d35e2fSjeremylt FIX_STRING(field_name); 726d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 727d7b241e6Sjeremylt 728d1d35e2fSjeremylt *err = CeedQFunctionAddOutput(qf_, field_name_c, *num_comp, *eval_mode); 729d7b241e6Sjeremylt } 730d7b241e6Sjeremylt 731*2b730f8bSJeremy L Thompson #define fCeedQFunctionSetContext FORTRAN_NAME(ceedqfunctionsetcontext, CEEDQFUNCTIONSETCONTEXT) 7323b271f31SJed Brown CEED_EXTERN void fCeedQFunctionSetContext(int *qf, int *ctx, int *err) { 733418fb8c2Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 734777ff853SJeremy L Thompson CeedQFunctionContext ctx_ = CeedQFunctionContext_dict[*ctx]; 735418fb8c2Sjeremylt 736777ff853SJeremy L Thompson CeedQFunctionContext fctx; 737777ff853SJeremy L Thompson *err = CeedQFunctionGetContext(qf_, &fctx); 738777ff853SJeremy L Thompson if (*err) return; 739777ff853SJeremy L Thompson CeedFortranContext fctxdata; 740777ff853SJeremy L Thompson *err = CeedQFunctionContextGetData(fctx, CEED_MEM_HOST, &fctxdata); 741777ff853SJeremy L Thompson if (*err) return; 7428cb0412aSJeremy L Thompson fctxdata->inner_ctx = ctx_; 743777ff853SJeremy L Thompson *err = CeedQFunctionContextRestoreData(fctx, (void **)&fctxdata); 744418fb8c2Sjeremylt } 745418fb8c2Sjeremylt 746*2b730f8bSJeremy L Thompson #define fCeedQFunctionView FORTRAN_NAME(ceedqfunctionview, CEEDQFUNCTIONVIEW) 7473b271f31SJed Brown CEED_EXTERN void fCeedQFunctionView(int *qf, int *err) { 74875affc3bSjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 74975affc3bSjeremylt 75075affc3bSjeremylt *err = CeedQFunctionView(qf_, stdout); 75175affc3bSjeremylt } 75275affc3bSjeremylt 753*2b730f8bSJeremy L Thompson #define fCeedQFunctionApply FORTRAN_NAME(ceedqfunctionapply, CEEDQFUNCTIONAPPLY) 754d7b241e6Sjeremylt // TODO Need Fixing, double pointer 755*2b730f8bSJeremy 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, 756*2b730f8bSJeremy 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, 757*2b730f8bSJeremy L Thompson int *v6, int *v7, int *v8, int *v9, int *v10, int *v11, int *v12, int *v13, int *v14, int *v15, int *err) { 758d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 759aedaa0e5Sjeremylt CeedVector *in; 760bf4cb664SJeremy L Thompson *err = CeedCalloc(CEED_FIELD_MAX, &in); 761d7b241e6Sjeremylt if (*err) return; 762aedaa0e5Sjeremylt in[0] = *u == FORTRAN_NULL ? NULL : CeedVector_dict[*u]; 763aedaa0e5Sjeremylt in[1] = *u1 == FORTRAN_NULL ? NULL : CeedVector_dict[*u1]; 764aedaa0e5Sjeremylt in[2] = *u2 == FORTRAN_NULL ? NULL : CeedVector_dict[*u2]; 765aedaa0e5Sjeremylt in[3] = *u3 == FORTRAN_NULL ? NULL : CeedVector_dict[*u3]; 766aedaa0e5Sjeremylt in[4] = *u4 == FORTRAN_NULL ? NULL : CeedVector_dict[*u4]; 767aedaa0e5Sjeremylt in[5] = *u5 == FORTRAN_NULL ? NULL : CeedVector_dict[*u5]; 768aedaa0e5Sjeremylt in[6] = *u6 == FORTRAN_NULL ? NULL : CeedVector_dict[*u6]; 769aedaa0e5Sjeremylt in[7] = *u7 == FORTRAN_NULL ? NULL : CeedVector_dict[*u7]; 770aedaa0e5Sjeremylt in[8] = *u8 == FORTRAN_NULL ? NULL : CeedVector_dict[*u8]; 771aedaa0e5Sjeremylt in[9] = *u9 == FORTRAN_NULL ? NULL : CeedVector_dict[*u9]; 772aedaa0e5Sjeremylt in[10] = *u10 == FORTRAN_NULL ? NULL : CeedVector_dict[*u10]; 773aedaa0e5Sjeremylt in[11] = *u11 == FORTRAN_NULL ? NULL : CeedVector_dict[*u11]; 774aedaa0e5Sjeremylt in[12] = *u12 == FORTRAN_NULL ? NULL : CeedVector_dict[*u12]; 775aedaa0e5Sjeremylt in[13] = *u13 == FORTRAN_NULL ? NULL : CeedVector_dict[*u13]; 776aedaa0e5Sjeremylt in[14] = *u14 == FORTRAN_NULL ? NULL : CeedVector_dict[*u14]; 777aedaa0e5Sjeremylt in[15] = *u15 == FORTRAN_NULL ? NULL : CeedVector_dict[*u15]; 778aedaa0e5Sjeremylt CeedVector *out; 779bf4cb664SJeremy L Thompson *err = CeedCalloc(CEED_FIELD_MAX, &out); 780d7b241e6Sjeremylt if (*err) return; 781aedaa0e5Sjeremylt out[0] = *v == FORTRAN_NULL ? NULL : CeedVector_dict[*v]; 782aedaa0e5Sjeremylt out[1] = *v1 == FORTRAN_NULL ? NULL : CeedVector_dict[*v1]; 783aedaa0e5Sjeremylt out[2] = *v2 == FORTRAN_NULL ? NULL : CeedVector_dict[*v2]; 784aedaa0e5Sjeremylt out[3] = *v3 == FORTRAN_NULL ? NULL : CeedVector_dict[*v3]; 785aedaa0e5Sjeremylt out[4] = *v4 == FORTRAN_NULL ? NULL : CeedVector_dict[*v4]; 786aedaa0e5Sjeremylt out[5] = *v5 == FORTRAN_NULL ? NULL : CeedVector_dict[*v5]; 787aedaa0e5Sjeremylt out[6] = *v6 == FORTRAN_NULL ? NULL : CeedVector_dict[*v6]; 788aedaa0e5Sjeremylt out[7] = *v7 == FORTRAN_NULL ? NULL : CeedVector_dict[*v7]; 789aedaa0e5Sjeremylt out[8] = *v8 == FORTRAN_NULL ? NULL : CeedVector_dict[*v8]; 790aedaa0e5Sjeremylt out[9] = *v9 == FORTRAN_NULL ? NULL : CeedVector_dict[*v9]; 791aedaa0e5Sjeremylt out[10] = *v10 == FORTRAN_NULL ? NULL : CeedVector_dict[*v10]; 792aedaa0e5Sjeremylt out[11] = *v11 == FORTRAN_NULL ? NULL : CeedVector_dict[*v11]; 793aedaa0e5Sjeremylt out[12] = *v12 == FORTRAN_NULL ? NULL : CeedVector_dict[*v12]; 794aedaa0e5Sjeremylt out[13] = *v13 == FORTRAN_NULL ? NULL : CeedVector_dict[*v13]; 795aedaa0e5Sjeremylt out[14] = *v14 == FORTRAN_NULL ? NULL : CeedVector_dict[*v14]; 796aedaa0e5Sjeremylt out[15] = *v15 == FORTRAN_NULL ? NULL : CeedVector_dict[*v15]; 797aedaa0e5Sjeremylt *err = CeedQFunctionApply(qf_, *Q, in, out); 798d7b241e6Sjeremylt if (*err) return; 799d7b241e6Sjeremylt 800d7b241e6Sjeremylt *err = CeedFree(&in); 801d7b241e6Sjeremylt if (*err) return; 802d7b241e6Sjeremylt *err = CeedFree(&out); 803d7b241e6Sjeremylt } 804d7b241e6Sjeremylt 805*2b730f8bSJeremy L Thompson #define fCeedQFunctionDestroy FORTRAN_NAME(ceedqfunctiondestroy, CEEDQFUNCTIONDESTROY) 8063b271f31SJed Brown CEED_EXTERN void fCeedQFunctionDestroy(int *qf, int *err) { 807b22696a8SJeremy L Thompson if (*qf == FORTRAN_NULL) return; 808c4b814baSjeremylt 809c4b814baSjeremylt *err = CeedQFunctionDestroy(&CeedQFunction_dict[*qf]); 81016595307SJeremy L Thompson if (*err == 0) { 81116595307SJeremy L Thompson *qf = FORTRAN_NULL; 812d7b241e6Sjeremylt CeedQFunction_n--; 813d7b241e6Sjeremylt if (CeedQFunction_n == 0) { 814d7b241e6Sjeremylt *err = CeedFree(&CeedQFunction_dict); 815d7b241e6Sjeremylt CeedQFunction_count = 0; 816d7b241e6Sjeremylt CeedQFunction_count_max = 0; 817d7b241e6Sjeremylt } 818d7b241e6Sjeremylt } 81916595307SJeremy L Thompson } 820d7b241e6Sjeremylt 821b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 822b00b43a6SJeremy L Thompson // CeedOperator 823b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 824d7b241e6Sjeremylt static CeedOperator *CeedOperator_dict = NULL; 825d7b241e6Sjeremylt static int CeedOperator_count = 0; 826d7b241e6Sjeremylt static int CeedOperator_n = 0; 827d7b241e6Sjeremylt static int CeedOperator_count_max = 0; 828d7b241e6Sjeremylt 829*2b730f8bSJeremy L Thompson #define fCeedOperatorCreate FORTRAN_NAME(ceedoperatorcreate, CEEDOPERATORCREATE) 830*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorCreate(int *ceed, int *qf, int *dqf, int *dqfT, int *op, int *err) { 831777ff853SJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) { 832777ff853SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 833777ff853SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 834777ff853SJeremy L Thompson } 835d7b241e6Sjeremylt 836d7b241e6Sjeremylt CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 837d7b241e6Sjeremylt 838442e7f0bSjeremylt CeedQFunction dqf_ = CEED_QFUNCTION_NONE, dqfT_ = CEED_QFUNCTION_NONE; 839442e7f0bSjeremylt if (*dqf != FORTRAN_QFUNCTION_NONE) dqf_ = CeedQFunction_dict[*dqf]; 840442e7f0bSjeremylt if (*dqfT != FORTRAN_QFUNCTION_NONE) dqfT_ = CeedQFunction_dict[*dqfT]; 841d7b241e6Sjeremylt 842*2b730f8bSJeremy L Thompson *err = CeedOperatorCreate(Ceed_dict[*ceed], CeedQFunction_dict[*qf], dqf_, dqfT_, op_); 843d7b241e6Sjeremylt if (*err) return; 844d7b241e6Sjeremylt *op = CeedOperator_count++; 845d7b241e6Sjeremylt CeedOperator_n++; 846d7b241e6Sjeremylt } 847d7b241e6Sjeremylt 848*2b730f8bSJeremy L Thompson #define fCeedCompositeOperatorCreate FORTRAN_NAME(ceedcompositeoperatorcreate, CEEDCOMPOSITEOPERATORCREATE) 8493b271f31SJed Brown CEED_EXTERN void fCeedCompositeOperatorCreate(int *ceed, int *op, int *err) { 850777ff853SJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) { 851777ff853SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 852777ff853SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 853777ff853SJeremy L Thompson } 85452d6035fSJeremy L Thompson 85552d6035fSJeremy L Thompson CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 85652d6035fSJeremy L Thompson 85752d6035fSJeremy L Thompson *err = CeedCompositeOperatorCreate(Ceed_dict[*ceed], op_); 85852d6035fSJeremy L Thompson if (*err) return; 85952d6035fSJeremy L Thompson *op = CeedOperator_count++; 86052d6035fSJeremy L Thompson CeedOperator_n++; 86152d6035fSJeremy L Thompson } 86252d6035fSJeremy L Thompson 863*2b730f8bSJeremy L Thompson #define fCeedOperatorSetField FORTRAN_NAME(ceedoperatorsetfield, CEEDOPERATORSETFIELD) 864*2b730f8bSJeremy 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) { 865d1d35e2fSjeremylt FIX_STRING(field_name); 866d7b241e6Sjeremylt CeedElemRestriction r_; 867d7b241e6Sjeremylt CeedBasis b_; 868d7b241e6Sjeremylt CeedVector v_; 869d7b241e6Sjeremylt 870d7b241e6Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 871d7b241e6Sjeremylt 872d7b241e6Sjeremylt if (*r == FORTRAN_NULL) { 873d7b241e6Sjeremylt r_ = NULL; 87415910d16Sjeremylt } else if (*r == FORTRAN_ELEMRESTRICTION_NONE) { 87515910d16Sjeremylt r_ = CEED_ELEMRESTRICTION_NONE; 876d7b241e6Sjeremylt } else { 877d7b241e6Sjeremylt r_ = CeedElemRestriction_dict[*r]; 878d7b241e6Sjeremylt } 8794dccadb6Sjeremylt 880d7b241e6Sjeremylt if (*b == FORTRAN_NULL) { 881d7b241e6Sjeremylt b_ = NULL; 882783c99b3SValeria Barra } else if (*b == FORTRAN_BASIS_COLLOCATED) { 883783c99b3SValeria Barra b_ = CEED_BASIS_COLLOCATED; 884d7b241e6Sjeremylt } else { 885d7b241e6Sjeremylt b_ = CeedBasis_dict[*b]; 886d7b241e6Sjeremylt } 887d7b241e6Sjeremylt if (*v == FORTRAN_NULL) { 888d7b241e6Sjeremylt v_ = NULL; 889d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_ACTIVE) { 890d7b241e6Sjeremylt v_ = CEED_VECTOR_ACTIVE; 891d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_NONE) { 892d7b241e6Sjeremylt v_ = CEED_VECTOR_NONE; 893d7b241e6Sjeremylt } else { 894d7b241e6Sjeremylt v_ = CeedVector_dict[*v]; 895d7b241e6Sjeremylt } 896d7b241e6Sjeremylt 897d1d35e2fSjeremylt *err = CeedOperatorSetField(op_, field_name_c, r_, b_, v_); 898d7b241e6Sjeremylt } 899d7b241e6Sjeremylt 900*2b730f8bSJeremy L Thompson #define fCeedCompositeOperatorAddSub FORTRAN_NAME(ceedcompositeoperatoraddsub, CEEDCOMPOSITEOPERATORADDSUB) 901*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedCompositeOperatorAddSub(int *compositeop, int *subop, int *err) { 90252d6035fSJeremy L Thompson CeedOperator compositeop_ = CeedOperator_dict[*compositeop]; 90352d6035fSJeremy L Thompson CeedOperator subop_ = CeedOperator_dict[*subop]; 90452d6035fSJeremy L Thompson 90552d6035fSJeremy L Thompson *err = CeedCompositeOperatorAddSub(compositeop_, subop_); 906ea6b5821SJeremy L Thompson } 907ea6b5821SJeremy L Thompson 908*2b730f8bSJeremy L Thompson #define fCeedOperatorSetName FORTRAN_NAME(ceedoperatorsetname, CEEDOPERATORSETNAME) 909*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorSetName(int *op, const char *name, int *err, fortran_charlen_t name_len) { 910ea6b5821SJeremy L Thompson FIX_STRING(name); 911ea6b5821SJeremy L Thompson CeedOperator op_ = CeedOperator_dict[*op]; 912ea6b5821SJeremy L Thompson 913ea6b5821SJeremy L Thompson *err = CeedOperatorSetName(op_, name_c); 91452d6035fSJeremy L Thompson } 91552d6035fSJeremy L Thompson 916*2b730f8bSJeremy L Thompson #define fCeedOperatorLinearAssembleQFunction FORTRAN_NAME(ceedoperatorlinearassembleqfunction, CEEDOPERATORLINEARASSEMBLEQFUNCTION) 917*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorLinearAssembleQFunction(int *op, int *assembledvec, int *assembledrstr, int *rqst, int *err) { 9181d102b48SJeremy L Thompson // Vector 9191d102b48SJeremy L Thompson if (CeedVector_count == CeedVector_count_max) { 9201d102b48SJeremy L Thompson CeedVector_count_max += CeedVector_count_max / 2 + 1; 9211d102b48SJeremy L Thompson CeedRealloc(CeedVector_count_max, &CeedVector_dict); 9221d102b48SJeremy L Thompson } 9231d102b48SJeremy L Thompson CeedVector *assembledvec_ = &CeedVector_dict[CeedVector_count]; 9241d102b48SJeremy L Thompson 9251d102b48SJeremy L Thompson // Restriction 9261d102b48SJeremy L Thompson if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 9271d102b48SJeremy L Thompson CeedElemRestriction_count_max += CeedElemRestriction_count_max / 2 + 1; 9281d102b48SJeremy L Thompson CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 9291d102b48SJeremy L Thompson } 930*2b730f8bSJeremy L Thompson CeedElemRestriction *rstr_ = &CeedElemRestriction_dict[CeedElemRestriction_count]; 9311d102b48SJeremy L Thompson 9321d102b48SJeremy L Thompson int createRequest = 1; 9331d102b48SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 9341d102b48SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 9351d102b48SJeremy L Thompson createRequest = 0; 9361d102b48SJeremy L Thompson } 9371d102b48SJeremy L Thompson 9381d102b48SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 9391d102b48SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 9401d102b48SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 9411d102b48SJeremy L Thompson } 9421d102b48SJeremy L Thompson 9431d102b48SJeremy L Thompson CeedRequest *rqst_; 9441d102b48SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 9451d102b48SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 9461d102b48SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 9471d102b48SJeremy L Thompson 948*2b730f8bSJeremy L Thompson *err = CeedOperatorLinearAssembleQFunction(CeedOperator_dict[*op], assembledvec_, rstr_, rqst_); 9491d102b48SJeremy L Thompson if (*err) return; 9501d102b48SJeremy L Thompson if (createRequest) { 9511d102b48SJeremy L Thompson *rqst = CeedRequest_count++; 9521d102b48SJeremy L Thompson CeedRequest_n++; 9531d102b48SJeremy L Thompson } 9541d102b48SJeremy L Thompson 9551d102b48SJeremy L Thompson if (*err == 0) { 9561d102b48SJeremy L Thompson *assembledrstr = CeedElemRestriction_count++; 9571d102b48SJeremy L Thompson CeedElemRestriction_n++; 9581d102b48SJeremy L Thompson *assembledvec = CeedVector_count++; 9591d102b48SJeremy L Thompson CeedVector_n++; 9601d102b48SJeremy L Thompson } 9611d102b48SJeremy L Thompson } 9621d102b48SJeremy L Thompson 963*2b730f8bSJeremy L Thompson #define fCeedOperatorLinearAssembleDiagonal FORTRAN_NAME(ceedoperatorlinearassemblediagonal, CEEDOPERATORLINEARASSEMBLEDIAGONAL) 964*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorLinearAssembleDiagonal(int *op, int *assembledvec, int *rqst, int *err) { 965b7ec98d8SJeremy L Thompson int createRequest = 1; 966b7ec98d8SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 967b7ec98d8SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 968b7ec98d8SJeremy L Thompson createRequest = 0; 969b7ec98d8SJeremy L Thompson } 970b7ec98d8SJeremy L Thompson 971b7ec98d8SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 972b7ec98d8SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 973b7ec98d8SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 974b7ec98d8SJeremy L Thompson } 975b7ec98d8SJeremy L Thompson 976b7ec98d8SJeremy L Thompson CeedRequest *rqst_; 977b7ec98d8SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 978b7ec98d8SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 979b7ec98d8SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 980b7ec98d8SJeremy L Thompson 981*2b730f8bSJeremy L Thompson *err = CeedOperatorLinearAssembleDiagonal(CeedOperator_dict[*op], CeedVector_dict[*assembledvec], rqst_); 982b7ec98d8SJeremy L Thompson if (*err) return; 983b7ec98d8SJeremy L Thompson if (createRequest) { 984b7ec98d8SJeremy L Thompson *rqst = CeedRequest_count++; 985b7ec98d8SJeremy L Thompson CeedRequest_n++; 986b7ec98d8SJeremy L Thompson } 987b7ec98d8SJeremy L Thompson } 988b7ec98d8SJeremy L Thompson 989*2b730f8bSJeremy L Thompson #define fCeedOperatorMultigridLevelCreate FORTRAN_NAME(ceedoperatormultigridlevelcreate, CEEDOPERATORMULTIGRIDLEVELCREATE) 990*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorMultigridLevelCreate(int *opFine, int *pMultFine, int *rstrCoarse, int *basisCoarse, int *opCoarse, int *opProlong, 991*2b730f8bSJeremy L Thompson int *opRestrict, int *err) { 992d99fa3c5SJeremy L Thompson // Operators 993d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 994d99fa3c5SJeremy L Thompson 995d99fa3c5SJeremy L Thompson // C interface call 996*2b730f8bSJeremy L Thompson *err = CeedOperatorMultigridLevelCreate(CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], CeedElemRestriction_dict[*rstrCoarse], 997*2b730f8bSJeremy L Thompson CeedBasis_dict[*basisCoarse], &opCoarse_, &opProlong_, &opRestrict_); 998d99fa3c5SJeremy L Thompson 999d99fa3c5SJeremy L Thompson if (*err) return; 1000d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1001d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 1002d99fa3c5SJeremy L Thompson } 1003d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1004d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1005d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1006d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1007d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1008d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1009d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1010d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1011d99fa3c5SJeremy L Thompson } 1012d99fa3c5SJeremy L Thompson 1013*2b730f8bSJeremy L Thompson #define fCeedOperatorMultigridLevelCreateTensorH1 FORTRAN_NAME(ceedoperatormultigridlevelcreatetensorh1, CEEDOPERATORMULTIGRIDLEVELCREATETENSORH1) 1014*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorMultigridLevelCreateTensorH1(int *opFine, int *pMultFine, int *rstrCoarse, int *basisCoarse, 1015*2b730f8bSJeremy L Thompson const CeedScalar *interpCtoF, int *opCoarse, int *opProlong, int *opRestrict, int *err) { 1016*2b730f8bSJeremy L Thompson // Operators 1017*2b730f8bSJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1018*2b730f8bSJeremy L Thompson 1019*2b730f8bSJeremy L Thompson // C interface call 1020*2b730f8bSJeremy L Thompson *err = CeedOperatorMultigridLevelCreateTensorH1(CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], CeedElemRestriction_dict[*rstrCoarse], 1021*2b730f8bSJeremy L Thompson CeedBasis_dict[*basisCoarse], interpCtoF, &opCoarse_, &opProlong_, &opRestrict_); 1022*2b730f8bSJeremy L Thompson 1023*2b730f8bSJeremy L Thompson if (*err) return; 1024*2b730f8bSJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1025*2b730f8bSJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 1026*2b730f8bSJeremy L Thompson } 1027*2b730f8bSJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1028*2b730f8bSJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1029*2b730f8bSJeremy L Thompson *opCoarse = CeedOperator_count++; 1030*2b730f8bSJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1031*2b730f8bSJeremy L Thompson *opProlong = CeedOperator_count++; 1032*2b730f8bSJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1033*2b730f8bSJeremy L Thompson *opRestrict = CeedOperator_count++; 1034*2b730f8bSJeremy L Thompson CeedOperator_n += 3; 1035*2b730f8bSJeremy L Thompson } 1036*2b730f8bSJeremy L Thompson 1037*2b730f8bSJeremy L Thompson #define fCeedOperatorMultigridLevelCreateH1 FORTRAN_NAME(ceedoperatormultigridlevelcreateh1, CEEDOPERATORMULTIGRIDLEVELCREATEH1) 1038*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorMultigridLevelCreateH1(int *opFine, int *pMultFine, int *rstrCoarse, int *basisCoarse, const CeedScalar *interpCtoF, 1039d99fa3c5SJeremy L Thompson int *opCoarse, int *opProlong, int *opRestrict, int *err) { 1040d99fa3c5SJeremy L Thompson // Operators 1041d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1042d99fa3c5SJeremy L Thompson 1043d99fa3c5SJeremy L Thompson // C interface call 1044*2b730f8bSJeremy L Thompson *err = CeedOperatorMultigridLevelCreateH1(CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], CeedElemRestriction_dict[*rstrCoarse], 1045*2b730f8bSJeremy L Thompson CeedBasis_dict[*basisCoarse], interpCtoF, &opCoarse_, &opProlong_, &opRestrict_); 1046d99fa3c5SJeremy L Thompson 1047d99fa3c5SJeremy L Thompson if (*err) return; 1048d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1049d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 1050d99fa3c5SJeremy L Thompson } 1051d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1052d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1053d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1054d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1055d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1056d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1057d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1058d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1059d99fa3c5SJeremy L Thompson } 1060d99fa3c5SJeremy L Thompson 1061*2b730f8bSJeremy L Thompson #define fCeedOperatorView FORTRAN_NAME(ceedoperatorview, CEEDOPERATORVIEW) 10623b271f31SJed Brown CEED_EXTERN void fCeedOperatorView(int *op, int *err) { 10632ebaca42Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 10642ebaca42Sjeremylt 10652ebaca42Sjeremylt *err = CeedOperatorView(op_, stdout); 10662ebaca42Sjeremylt } 10672ebaca42Sjeremylt 1068*2b730f8bSJeremy L Thompson #define fCeedOperatorCreateFDMElementInverse FORTRAN_NAME(ceedoperatorcreatefdmelementinverse, CEEDOPERATORCREATEFDMELEMENTINVERSE) 1069*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorCreateFDMElementInverse(int *op, int *fdminv, int *rqst, int *err) { 10703bd813ffSjeremylt // Operator 10713bd813ffSjeremylt if (CeedOperator_count == CeedOperator_count_max) { 10723bd813ffSjeremylt CeedOperator_count_max += CeedOperator_count_max / 2 + 1; 10733bd813ffSjeremylt CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 10743bd813ffSjeremylt } 1075*2b730f8bSJeremy L Thompson CeedOperator *fdminv_ = &CeedOperator_dict[CeedOperator_count]; 10763bd813ffSjeremylt 10773bd813ffSjeremylt int createRequest = 1; 10783bd813ffSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 10793bd813ffSjeremylt if (*rqst == -1 || *rqst == -2) { 10803bd813ffSjeremylt createRequest = 0; 10813bd813ffSjeremylt } 10823bd813ffSjeremylt 10833bd813ffSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 10843bd813ffSjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 10853bd813ffSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 10863bd813ffSjeremylt } 10873bd813ffSjeremylt 10883bd813ffSjeremylt CeedRequest *rqst_; 10893bd813ffSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 10903bd813ffSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 10913bd813ffSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 10923bd813ffSjeremylt 1093*2b730f8bSJeremy L Thompson *err = CeedOperatorCreateFDMElementInverse(CeedOperator_dict[*op], fdminv_, rqst_); 10943bd813ffSjeremylt if (*err) return; 10953bd813ffSjeremylt if (createRequest) { 10963bd813ffSjeremylt *rqst = CeedRequest_count++; 10973bd813ffSjeremylt CeedRequest_n++; 10983bd813ffSjeremylt } 10993bd813ffSjeremylt 11003bd813ffSjeremylt if (*err == 0) { 11013bd813ffSjeremylt *fdminv = CeedOperator_count++; 11023bd813ffSjeremylt CeedOperator_n++; 11033bd813ffSjeremylt } 11043bd813ffSjeremylt } 11053bd813ffSjeremylt 1106d7b241e6Sjeremylt #define fCeedOperatorApply FORTRAN_NAME(ceedoperatorapply, CEEDOPERATORAPPLY) 1107*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorApply(int *op, int *ustatevec, int *resvec, int *rqst, int *err) { 1108*2b730f8bSJeremy L Thompson CeedVector ustatevec_ = (*ustatevec == FORTRAN_NULL) ? NULL : (*ustatevec == FORTRAN_VECTOR_NONE ? CEED_VECTOR_NONE : CeedVector_dict[*ustatevec]); 1109*2b730f8bSJeremy L Thompson CeedVector resvec_ = (*resvec == FORTRAN_NULL) ? NULL : (*resvec == FORTRAN_VECTOR_NONE ? CEED_VECTOR_NONE : CeedVector_dict[*resvec]); 1110d7b241e6Sjeremylt 1111d7b241e6Sjeremylt int createRequest = 1; 1112d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1113d7b241e6Sjeremylt if (*rqst == -1 || *rqst == -2) { 1114d7b241e6Sjeremylt createRequest = 0; 1115d7b241e6Sjeremylt } 1116d7b241e6Sjeremylt 1117d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1118d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 1119d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1120d7b241e6Sjeremylt } 1121d7b241e6Sjeremylt 1122d7b241e6Sjeremylt CeedRequest *rqst_; 1123d7b241e6Sjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1124d7b241e6Sjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1125d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1126d7b241e6Sjeremylt 1127*2b730f8bSJeremy L Thompson *err = CeedOperatorApply(CeedOperator_dict[*op], ustatevec_, resvec_, rqst_); 1128d7b241e6Sjeremylt if (*err) return; 1129d7b241e6Sjeremylt if (createRequest) { 1130d7b241e6Sjeremylt *rqst = CeedRequest_count++; 1131d7b241e6Sjeremylt CeedRequest_n++; 1132d7b241e6Sjeremylt } 1133d7b241e6Sjeremylt } 1134d7b241e6Sjeremylt 1135cae8b89aSjeremylt #define fCeedOperatorApplyAdd FORTRAN_NAME(ceedoperatorapplyadd, CEEDOPERATORAPPLYADD) 1136*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorApplyAdd(int *op, int *ustatevec, int *resvec, int *rqst, int *err) { 1137*2b730f8bSJeremy L Thompson CeedVector ustatevec_ = *ustatevec == FORTRAN_NULL ? NULL : CeedVector_dict[*ustatevec]; 1138*2b730f8bSJeremy L Thompson CeedVector resvec_ = *resvec == FORTRAN_NULL ? NULL : CeedVector_dict[*resvec]; 1139cae8b89aSjeremylt 1140cae8b89aSjeremylt int createRequest = 1; 1141cae8b89aSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1142cae8b89aSjeremylt if (*rqst == -1 || *rqst == -2) { 1143cae8b89aSjeremylt createRequest = 0; 1144cae8b89aSjeremylt } 1145cae8b89aSjeremylt 1146cae8b89aSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1147cae8b89aSjeremylt CeedRequest_count_max += CeedRequest_count_max / 2 + 1; 1148cae8b89aSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1149cae8b89aSjeremylt } 1150cae8b89aSjeremylt 1151cae8b89aSjeremylt CeedRequest *rqst_; 1152cae8b89aSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1153cae8b89aSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1154cae8b89aSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1155cae8b89aSjeremylt 1156*2b730f8bSJeremy L Thompson *err = CeedOperatorApplyAdd(CeedOperator_dict[*op], ustatevec_, resvec_, rqst_); 1157cae8b89aSjeremylt if (*err) return; 1158cae8b89aSjeremylt if (createRequest) { 1159cae8b89aSjeremylt *rqst = CeedRequest_count++; 1160cae8b89aSjeremylt CeedRequest_n++; 1161cae8b89aSjeremylt } 1162cae8b89aSjeremylt } 1163cae8b89aSjeremylt 1164*2b730f8bSJeremy L Thompson #define fCeedOperatorApplyJacobian FORTRAN_NAME(ceedoperatorapplyjacobian, CEEDOPERATORAPPLYJACOBIAN) 1165*2b730f8bSJeremy L Thompson CEED_EXTERN void fCeedOperatorApplyJacobian(int *op, int *qdatavec, int *ustatevec, int *dustatevec, int *dresvec, int *rqst, int *err) { 1166d7b241e6Sjeremylt // TODO Uncomment this when CeedOperatorApplyJacobian is implemented 1167d7b241e6Sjeremylt // *err = CeedOperatorApplyJacobian(CeedOperator_dict[*op], CeedVector_dict[*qdatavec], 1168d7b241e6Sjeremylt // CeedVector_dict[*ustatevec], CeedVector_dict[*dustatevec], 1169d7b241e6Sjeremylt // CeedVector_dict[*dresvec], &CeedRequest_dict[*rqst]); 1170d7b241e6Sjeremylt } 1171d7b241e6Sjeremylt 1172*2b730f8bSJeremy L Thompson #define fCeedOperatorDestroy FORTRAN_NAME(ceedoperatordestroy, CEEDOPERATORDESTROY) 11733b271f31SJed Brown CEED_EXTERN void fCeedOperatorDestroy(int *op, int *err) { 1174b22696a8SJeremy L Thompson if (*op == FORTRAN_NULL) return; 1175d7b241e6Sjeremylt *err = CeedOperatorDestroy(&CeedOperator_dict[*op]); 117616595307SJeremy L Thompson if (*err == 0) { 117716595307SJeremy L Thompson *op = FORTRAN_NULL; 1178d7b241e6Sjeremylt CeedOperator_n--; 1179d7b241e6Sjeremylt if (CeedOperator_n == 0) { 1180d7b241e6Sjeremylt *err = CeedFree(&CeedOperator_dict); 1181d7b241e6Sjeremylt CeedOperator_count = 0; 1182d7b241e6Sjeremylt CeedOperator_count_max = 0; 1183d7b241e6Sjeremylt } 1184d7b241e6Sjeremylt } 118516595307SJeremy L Thompson } 1186b00b43a6SJeremy L Thompson 1187b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 1188