1c8817ca9Svaleriabarra // Copyright (c) 2017-2018, Lawrence Livermore National Security, LLC. 2c8817ca9Svaleriabarra // Produced at the Lawrence Livermore National Laboratory. LLNL-CODE-734707. 3c8817ca9Svaleriabarra // All Rights reserved. See files LICENSE and NOTICE for details. 4c8817ca9Svaleriabarra // 5c8817ca9Svaleriabarra // This file is part of CEED, a collection of benchmarks, miniapps, software 6c8817ca9Svaleriabarra // libraries and APIs for efficient high-order finite element and spectral 7c8817ca9Svaleriabarra // element discretizations for exascale applications. For more information and 8c8817ca9Svaleriabarra // source code availability see http://github.com/ceed. 9c8817ca9Svaleriabarra // 10c8817ca9Svaleriabarra // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC, 11c8817ca9Svaleriabarra // a collaborative effort of two U.S. Department of Energy organizations (Office 12c8817ca9Svaleriabarra // of Science and the National Nuclear Security Administration) responsible for 13c8817ca9Svaleriabarra // the planning and preparation of a capable exascale ecosystem, including 14c8817ca9Svaleriabarra // software, applications, hardware, advanced system engineering and early 15c8817ca9Svaleriabarra // testbed platforms, in support of the nation's exascale computing imperative. 16c8817ca9Svaleriabarra 17d7b241e6Sjeremylt // Fortran interface 18d7b241e6Sjeremylt #include <ceed.h> 19d7b241e6Sjeremylt #include <ceed-impl.h> 20d863ab9bSjeremylt #include <ceed-backend.h> 21d7b241e6Sjeremylt #include <ceed-fortran-name.h> 22d7b241e6Sjeremylt #include <stdlib.h> 23d7b241e6Sjeremylt #include <string.h> 24d7b241e6Sjeremylt 25d7b241e6Sjeremylt #define FORTRAN_REQUEST_IMMEDIATE -1 26d7b241e6Sjeremylt #define FORTRAN_REQUEST_ORDERED -2 27d7b241e6Sjeremylt #define FORTRAN_NULL -3 2815910d16Sjeremylt #define FORTRAN_STRIDES_BACKEND -4 2915910d16Sjeremylt #define FORTRAN_VECTOR_ACTIVE -5 3015910d16Sjeremylt #define FORTRAN_VECTOR_NONE -6 3115910d16Sjeremylt #define FORTRAN_ELEMRESTRICTION_NONE -7 3215910d16Sjeremylt #define FORTRAN_BASIS_COLLOCATED -8 3315910d16Sjeremylt #define FORTRAN_QFUNCTION_NONE -9 34d7b241e6Sjeremylt 35d7b241e6Sjeremylt static Ceed *Ceed_dict = NULL; 36d7b241e6Sjeremylt static int Ceed_count = 0; 37d7b241e6Sjeremylt static int Ceed_n = 0; 38d7b241e6Sjeremylt static int Ceed_count_max = 0; 39d7b241e6Sjeremylt 40c2ae719dSJed Brown // This test should actually be for the gfortran version, but we don't currently 41c2ae719dSJed Brown // have a configure system to determine that (TODO). At present, this will use 42c2ae719dSJed Brown // the smaller integer when run with clang+gfortran=8, for example. (That is 43c2ae719dSJed Brown // sketchy, but will likely work for users that don't have huge character 44c2ae719dSJed Brown // strings.) 45c2ae719dSJed Brown #if __GNUC__ >= 8 46c2ae719dSJed Brown typedef size_t fortran_charlen_t; 47c2ae719dSJed Brown #else 48c2ae719dSJed Brown typedef int fortran_charlen_t; 49c2ae719dSJed Brown #endif 50c2ae719dSJed Brown 51c2ae719dSJed Brown #define Splice(a, b) a ## b 52c2ae719dSJed Brown 53c2ae719dSJed Brown // Fortran strings are generally unterminated and the length is passed as an 54c2ae719dSJed Brown // extra argument after all the normal arguments. Some compilers (I only know 55c2ae719dSJed Brown // of Windows) place the length argument immediately after the string parameter 56c2ae719dSJed Brown // (TODO). 57c2ae719dSJed Brown // 58c2ae719dSJed Brown // We can't just NULL-terminate the string in-place because that could overwrite 59c2ae719dSJed Brown // other strings or attempt to write to read-only memory. This macro allocates 60c2ae719dSJed Brown // a string to hold the null-terminated version of the string that C expects. 61c2ae719dSJed Brown #define FIX_STRING(stringname) \ 62c2ae719dSJed Brown char Splice(stringname, _c)[1024]; \ 63c2ae719dSJed Brown if (Splice(stringname, _len) > 1023) \ 64c2ae719dSJed Brown CeedError(NULL, 1, "Fortran string length too long %zd", (size_t)Splice(stringname, _len)); \ 65c2ae719dSJed Brown strncpy(Splice(stringname, _c), stringname, Splice(stringname, _len)); \ 66c2ae719dSJed Brown Splice(stringname, _c)[Splice(stringname, _len)] = 0; \ 67c2ae719dSJed Brown 68b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 69b00b43a6SJeremy L Thompson // Ceed 70b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 71d7b241e6Sjeremylt #define fCeedInit FORTRAN_NAME(ceedinit,CEEDINIT) 72c2ae719dSJed Brown void fCeedInit(const char *resource, int *ceed, int *err, 73c2ae719dSJed Brown fortran_charlen_t resource_len) { 74c2ae719dSJed Brown FIX_STRING(resource); 75d7b241e6Sjeremylt if (Ceed_count == Ceed_count_max) { 76d7b241e6Sjeremylt Ceed_count_max += Ceed_count_max/2 + 1; 77d7b241e6Sjeremylt CeedRealloc(Ceed_count_max, &Ceed_dict); 78d7b241e6Sjeremylt } 79d7b241e6Sjeremylt 80d7b241e6Sjeremylt Ceed *ceed_ = &Ceed_dict[Ceed_count]; 81c2ae719dSJed Brown *err = CeedInit(resource_c, ceed_); 82d7b241e6Sjeremylt 83d7b241e6Sjeremylt if (*err == 0) { 84d7b241e6Sjeremylt *ceed = Ceed_count++; 85d7b241e6Sjeremylt Ceed_n++; 86d7b241e6Sjeremylt } 87d7b241e6Sjeremylt } 88d7b241e6Sjeremylt 899525855cSJeremy L Thompson #define fCeedIsDeterministic \ 909525855cSJeremy L Thompson FORTRAN_NAME(ceedisdeterministic,CEEDISDETERMINISTIC) 919525855cSJeremy L Thompson void fCeedIsDeterministic(int *ceed, int *isDeterministic, int *err) { 929525855cSJeremy L Thompson *err = CeedIsDeterministic(Ceed_dict[*ceed], (bool *)isDeterministic); 939525855cSJeremy L Thompson } 949525855cSJeremy L Thompson 95c907536fSjeremylt #define fCeedGetPreferredMemType \ 96c907536fSjeremylt FORTRAN_NAME(ceedgetpreferredmemtype,CEEDGETPREFERREDMEMTYPE) 97c907536fSjeremylt void fCeedGetPreferredMemType(int *ceed, int *type, int *err) { 98c907536fSjeremylt *err = CeedGetPreferredMemType(Ceed_dict[*ceed], (CeedMemType *)type); 99c907536fSjeremylt } 100c907536fSjeremylt 1010a0da059Sjeremylt #define fCeedView FORTRAN_NAME(ceedview,CEEDVIEW) 1020a0da059Sjeremylt void fCeedView(int *ceed, int *err) { 1030a0da059Sjeremylt *err = CeedView(Ceed_dict[*ceed], stdout); 1040a0da059Sjeremylt } 1050a0da059Sjeremylt 106d7b241e6Sjeremylt #define fCeedDestroy FORTRAN_NAME(ceeddestroy,CEEDDESTROY) 107d7b241e6Sjeremylt void fCeedDestroy(int *ceed, int *err) { 108b22696a8SJeremy L Thompson if (*ceed == FORTRAN_NULL) return; 109d7b241e6Sjeremylt *err = CeedDestroy(&Ceed_dict[*ceed]); 110d7b241e6Sjeremylt 111d7b241e6Sjeremylt if (*err == 0) { 11216595307SJeremy L Thompson *ceed = FORTRAN_NULL; 113d7b241e6Sjeremylt Ceed_n--; 114d7b241e6Sjeremylt if (Ceed_n == 0) { 115d7b241e6Sjeremylt CeedFree(&Ceed_dict); 116d7b241e6Sjeremylt Ceed_count = 0; 117d7b241e6Sjeremylt Ceed_count_max = 0; 118d7b241e6Sjeremylt } 119d7b241e6Sjeremylt } 120d7b241e6Sjeremylt } 121d7b241e6Sjeremylt 122b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 123b00b43a6SJeremy L Thompson // CeedVector 124b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 125d7b241e6Sjeremylt static CeedVector *CeedVector_dict = NULL; 126d7b241e6Sjeremylt static int CeedVector_count = 0; 127d7b241e6Sjeremylt static int CeedVector_n = 0; 128d7b241e6Sjeremylt static int CeedVector_count_max = 0; 129d7b241e6Sjeremylt 130d7b241e6Sjeremylt #define fCeedVectorCreate FORTRAN_NAME(ceedvectorcreate,CEEDVECTORCREATE) 131d7b241e6Sjeremylt void fCeedVectorCreate(int *ceed, int *length, int *vec, int *err) { 132d7b241e6Sjeremylt if (CeedVector_count == CeedVector_count_max) { 133d7b241e6Sjeremylt CeedVector_count_max += CeedVector_count_max/2 + 1; 134d7b241e6Sjeremylt CeedRealloc(CeedVector_count_max, &CeedVector_dict); 135d7b241e6Sjeremylt } 136d7b241e6Sjeremylt 137d7b241e6Sjeremylt CeedVector *vec_ = &CeedVector_dict[CeedVector_count]; 138d7b241e6Sjeremylt *err = CeedVectorCreate(Ceed_dict[*ceed], *length, vec_); 139d7b241e6Sjeremylt 140d7b241e6Sjeremylt if (*err == 0) { 141d7b241e6Sjeremylt *vec = CeedVector_count++; 142d7b241e6Sjeremylt CeedVector_n++; 143d7b241e6Sjeremylt } 144d7b241e6Sjeremylt } 145d7b241e6Sjeremylt 146d7b241e6Sjeremylt #define fCeedVectorSetArray FORTRAN_NAME(ceedvectorsetarray,CEEDVECTORSETARRAY) 147d7b241e6Sjeremylt void fCeedVectorSetArray(int *vec, int *memtype, int *copymode, 148c8b9fe72Sjeremylt CeedScalar *array, int64_t *offset, int *err) { 149399bb197SJed Brown *err = CeedVectorSetArray(CeedVector_dict[*vec], (CeedMemType)*memtype, 150399bb197SJed Brown (CeedCopyMode)*copymode, 151c8b9fe72Sjeremylt (CeedScalar *)(array + *offset)); 152d7b241e6Sjeremylt } 153d7b241e6Sjeremylt 1546a6c615bSJeremy L Thompson #define fCeedVectorTakeArray FORTRAN_NAME(ceedvectortakearray,CEEDVECTORTAKEARRAY) 1556a6c615bSJeremy L Thompson void fCeedVectorTakeArray(int *vec, int *memtype, CeedScalar *array, 1566a6c615bSJeremy L Thompson int64_t *offset, int *err) { 1576a6c615bSJeremy L Thompson CeedScalar *b; 1586a6c615bSJeremy L Thompson CeedVector vec_ = CeedVector_dict[*vec]; 1596a6c615bSJeremy L Thompson *err = CeedVectorTakeArray(vec_, (CeedMemType)*memtype, &b); 1606a6c615bSJeremy L Thompson *offset = b - array; 1616a6c615bSJeremy L Thompson } 1626a6c615bSJeremy L Thompson 16354540941SJeremy L Thompson #define fCeedVectorSyncArray FORTRAN_NAME(ceedvectorsyncarray,CEEDVECTORSYNCARRAY) 16454540941SJeremy L Thompson void fCeedVectorSyncArray(int *vec, int *memtype, int *err) { 165399bb197SJed Brown *err = CeedVectorSyncArray(CeedVector_dict[*vec], (CeedMemType)*memtype); 16654540941SJeremy L Thompson } 16754540941SJeremy L Thompson 168d7b241e6Sjeremylt #define fCeedVectorSetValue FORTRAN_NAME(ceedvectorsetvalue,CEEDVECTORSETVALUE) 169d7b241e6Sjeremylt void fCeedVectorSetValue(int *vec, CeedScalar *value, int *err) { 170d7b241e6Sjeremylt *err = CeedVectorSetValue(CeedVector_dict[*vec], *value); 171d7b241e6Sjeremylt } 172d7b241e6Sjeremylt 173d7b241e6Sjeremylt #define fCeedVectorGetArray FORTRAN_NAME(ceedvectorgetarray,CEEDVECTORGETARRAY) 174a8de75f0Sjeremylt void fCeedVectorGetArray(int *vec, int *memtype, CeedScalar *array, 1756a6c615bSJeremy L Thompson int64_t *offset, int *err) { 176d7b241e6Sjeremylt CeedScalar *b; 177d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 178399bb197SJed Brown *err = CeedVectorGetArray(vec_, (CeedMemType)*memtype, &b); 179cbfd6a90SThilina Rathnayake *offset = b - array; 180d7b241e6Sjeremylt } 181d7b241e6Sjeremylt 182d7b241e6Sjeremylt #define fCeedVectorGetArrayRead \ 183d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorgetarrayread,CEEDVECTORGETARRAYREAD) 184d7b241e6Sjeremylt void fCeedVectorGetArrayRead(int *vec, int *memtype, CeedScalar *array, 1853334379fSJed Brown int64_t *offset, int *err) { 186d7b241e6Sjeremylt const CeedScalar *b; 187d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 188399bb197SJed Brown *err = CeedVectorGetArrayRead(vec_, (CeedMemType)*memtype, &b); 189cbfd6a90SThilina Rathnayake *offset = b - array; 190d7b241e6Sjeremylt } 191d7b241e6Sjeremylt 192d7b241e6Sjeremylt #define fCeedVectorRestoreArray \ 193d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearray,CEEDVECTORRESTOREARRAY) 1946456524eSJed Brown void fCeedVectorRestoreArray(int *vec, CeedScalar *array, 1956456524eSJed Brown int64_t *offset, int *err) { 196*9647a07eSDavid Medina CeedScalar *offsetArray = array + *offset; 197*9647a07eSDavid Medina *err = CeedVectorRestoreArray(CeedVector_dict[*vec], &offsetArray); 1986456524eSJed Brown *offset = 0; 199d7b241e6Sjeremylt } 200d7b241e6Sjeremylt 201d7b241e6Sjeremylt #define fCeedVectorRestoreArrayRead \ 202d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearrayread,CEEDVECTORRESTOREARRAYREAD) 2036456524eSJed Brown void fCeedVectorRestoreArrayRead(int *vec, const CeedScalar *array, 2046456524eSJed Brown int64_t *offset, int *err) { 205d7b241e6Sjeremylt *err = CeedVectorRestoreArrayRead(CeedVector_dict[*vec], &array); 2066456524eSJed Brown *offset = 0; 207d7b241e6Sjeremylt } 208d7b241e6Sjeremylt 209547d9b97Sjeremylt #define fCeedVectorNorm \ 210547d9b97Sjeremylt FORTRAN_NAME(ceedvectornorm,CEEDVECTORNORM) 211547d9b97Sjeremylt void fCeedVectorNorm(int *vec, int *type, CeedScalar *norm, int *err) { 212399bb197SJed Brown *err = CeedVectorNorm(CeedVector_dict[*vec], (CeedNormType)*type, norm); 213547d9b97Sjeremylt } 214547d9b97Sjeremylt 215d99fa3c5SJeremy L Thompson #define fCeedVectorReciprocal \ 216d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedvectorreciprocal,CEEDVECTORRECIPROCAL) 217d99fa3c5SJeremy L Thompson void fCeedVectorReciprocal(int *vec, int *err) { 218d99fa3c5SJeremy L Thompson *err = CeedVectorReciprocal(CeedVector_dict[*vec]); 219d99fa3c5SJeremy L Thompson } 220d99fa3c5SJeremy L Thompson 221d7b241e6Sjeremylt #define fCeedVectorView FORTRAN_NAME(ceedvectorview,CEEDVECTORVIEW) 222d7b241e6Sjeremylt void fCeedVectorView(int *vec, int *err) { 223d7b241e6Sjeremylt *err = CeedVectorView(CeedVector_dict[*vec], "%12.8f", stdout); 224d7b241e6Sjeremylt } 225d7b241e6Sjeremylt 226d7b241e6Sjeremylt #define fCeedVectorDestroy FORTRAN_NAME(ceedvectordestroy,CEEDVECTORDESTROY) 227d7b241e6Sjeremylt void fCeedVectorDestroy(int *vec, int *err) { 228b22696a8SJeremy L Thompson if (*vec == FORTRAN_NULL) return; 229d7b241e6Sjeremylt *err = CeedVectorDestroy(&CeedVector_dict[*vec]); 230d7b241e6Sjeremylt 231d7b241e6Sjeremylt if (*err == 0) { 23216595307SJeremy L Thompson *vec = FORTRAN_NULL; 233d7b241e6Sjeremylt CeedVector_n--; 234d7b241e6Sjeremylt if (CeedVector_n == 0) { 235d7b241e6Sjeremylt CeedFree(&CeedVector_dict); 236d7b241e6Sjeremylt CeedVector_count = 0; 237d7b241e6Sjeremylt CeedVector_count_max = 0; 238d7b241e6Sjeremylt } 239d7b241e6Sjeremylt } 240d7b241e6Sjeremylt } 241d7b241e6Sjeremylt 242b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 243b00b43a6SJeremy L Thompson // CeedElemRestriction 244b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 245d7b241e6Sjeremylt static CeedElemRestriction *CeedElemRestriction_dict = NULL; 246d7b241e6Sjeremylt static int CeedElemRestriction_count = 0; 247d7b241e6Sjeremylt static int CeedElemRestriction_n = 0; 248d7b241e6Sjeremylt static int CeedElemRestriction_count_max = 0; 249d7b241e6Sjeremylt 250d7b241e6Sjeremylt #define fCeedElemRestrictionCreate \ 251d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreate, CEEDELEMRESTRICTIONCREATE) 252d979a051Sjeremylt void fCeedElemRestrictionCreate(int *ceed, int *nelements, int *esize, 253d979a051Sjeremylt int *ncomp, int *compstride, int *lsize, 254d979a051Sjeremylt int *memtype, int *copymode, const int *offsets, 2558795c945Sjeremylt int *elemrestriction, int *err) { 256d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 257d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 258d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 259d7b241e6Sjeremylt } 260d7b241e6Sjeremylt 261d979a051Sjeremylt const int *offsets_ = offsets; 262d7b241e6Sjeremylt 263d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 264d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 265d979a051Sjeremylt *err = CeedElemRestrictionCreate(Ceed_dict[*ceed], *nelements, *esize, 266d979a051Sjeremylt *ncomp, *compstride, *lsize, 267d979a051Sjeremylt (CeedMemType)*memtype, 268d979a051Sjeremylt (CeedCopyMode)*copymode, offsets_, 269d979a051Sjeremylt elemrestriction_); 270d7b241e6Sjeremylt 271d7b241e6Sjeremylt if (*err == 0) { 272d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 273d7b241e6Sjeremylt CeedElemRestriction_n++; 274d7b241e6Sjeremylt } 275d7b241e6Sjeremylt } 276d7b241e6Sjeremylt 2777509a596Sjeremylt #define fCeedElemRestrictionCreateStrided \ 2787509a596Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreatestrided, CEEDELEMRESTRICTIONCREATESTRIDED) 2797509a596Sjeremylt void fCeedElemRestrictionCreateStrided(int *ceed, int *nelements, int *esize, 280d979a051Sjeremylt int *ncomp, int *lsize, int *strides, 281d7b241e6Sjeremylt int *elemrestriction, int *err) { 282d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 283d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 284d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 285d7b241e6Sjeremylt } 286d7b241e6Sjeremylt 287d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 288d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 2897509a596Sjeremylt *err = CeedElemRestrictionCreateStrided(Ceed_dict[*ceed], *nelements, *esize, 290d979a051Sjeremylt *ncomp, *lsize, 291d979a051Sjeremylt *strides == FORTRAN_STRIDES_BACKEND ? 292d979a051Sjeremylt CEED_STRIDES_BACKEND : strides, 293523b8ea0Sjeremylt elemrestriction_); 294d7b241e6Sjeremylt if (*err == 0) { 295d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 296d7b241e6Sjeremylt CeedElemRestriction_n++; 297d7b241e6Sjeremylt } 298d7b241e6Sjeremylt } 299d7b241e6Sjeremylt 300d7b241e6Sjeremylt #define fCeedElemRestrictionCreateBlocked \ 301d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateblocked,CEEDELEMRESTRICTIONCREATEBLOCKED) 302d979a051Sjeremylt void fCeedElemRestrictionCreateBlocked(int *ceed, int *nelements, int *esize, 303d979a051Sjeremylt int *blocksize, int *ncomp, 304d979a051Sjeremylt int *compstride, int *lsize, 305d979a051Sjeremylt int *mtype, int *cmode, 3068795c945Sjeremylt int *blkindices, int *elemrestriction, 3078795c945Sjeremylt int *err) { 308d7b241e6Sjeremylt 309d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 310d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 311d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 312d7b241e6Sjeremylt } 313d7b241e6Sjeremylt 314d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 315d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 316399bb197SJed Brown *err = CeedElemRestrictionCreateBlocked(Ceed_dict[*ceed], 317d979a051Sjeremylt *nelements, *esize, *blocksize, 318d979a051Sjeremylt *ncomp, *compstride, *lsize, 319d979a051Sjeremylt (CeedMemType)*mtype, 320d979a051Sjeremylt (CeedCopyMode)*cmode, blkindices, 321a8d32208Sjeremylt elemrestriction_); 322d7b241e6Sjeremylt 323d7b241e6Sjeremylt if (*err == 0) { 324d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 325d7b241e6Sjeremylt CeedElemRestriction_n++; 326d7b241e6Sjeremylt } 327d7b241e6Sjeremylt } 328d7b241e6Sjeremylt 3297509a596Sjeremylt #define fCeedElemRestrictionCreateBlockedStrided \ 3307509a596Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateblockedstrided, CEEDELEMRESTRICTIONCREATEBLOCKEDSTRIDED) 3317509a596Sjeremylt void fCeedElemRestrictionCreateBlockedStrided(int *ceed, int *nelements, 332d979a051Sjeremylt int *esize, int *blksize, int *ncomp, int *lsize, int *strides, 3337509a596Sjeremylt int *elemrestriction, int *err) { 3347509a596Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 3357509a596Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 3367509a596Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 3377509a596Sjeremylt } 3387509a596Sjeremylt 3397509a596Sjeremylt CeedElemRestriction *elemrestriction_ = 3407509a596Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 3417509a596Sjeremylt *err = CeedElemRestrictionCreateBlockedStrided(Ceed_dict[*ceed], *nelements, 342d979a051Sjeremylt *esize, *blksize, *ncomp, *lsize, strides, elemrestriction_); 3437509a596Sjeremylt if (*err == 0) { 3447509a596Sjeremylt *elemrestriction = CeedElemRestriction_count++; 3457509a596Sjeremylt CeedElemRestriction_n++; 3467509a596Sjeremylt } 3477509a596Sjeremylt } 3487509a596Sjeremylt 349d7b241e6Sjeremylt static CeedRequest *CeedRequest_dict = NULL; 350d7b241e6Sjeremylt static int CeedRequest_count = 0; 351d7b241e6Sjeremylt static int CeedRequest_n = 0; 352d7b241e6Sjeremylt static int CeedRequest_count_max = 0; 353d7b241e6Sjeremylt 354d7b241e6Sjeremylt #define fCeedElemRestrictionApply \ 355d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictionapply,CEEDELEMRESTRICTIONAPPLY) 356a8d32208Sjeremylt void fCeedElemRestrictionApply(int *elemr, int *tmode, int *uvec, int *ruvec, 357a8d32208Sjeremylt int *rqst, int *err) { 358d7b241e6Sjeremylt int createRequest = 1; 359d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 360d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) 361d7b241e6Sjeremylt createRequest = 0; 362d7b241e6Sjeremylt 363d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 364d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 365d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 366d7b241e6Sjeremylt } 367d7b241e6Sjeremylt 368d7b241e6Sjeremylt CeedRequest *rqst_; 369d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 370d7b241e6Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED ) rqst_ = CEED_REQUEST_ORDERED; 371d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 372d7b241e6Sjeremylt 373399bb197SJed Brown *err = CeedElemRestrictionApply(CeedElemRestriction_dict[*elemr], 374399bb197SJed Brown (CeedTransposeMode)*tmode, 375a8d32208Sjeremylt CeedVector_dict[*uvec], 3768795c945Sjeremylt CeedVector_dict[*ruvec], rqst_); 377d7b241e6Sjeremylt 378d7b241e6Sjeremylt if (*err == 0 && createRequest) { 379d7b241e6Sjeremylt *rqst = CeedRequest_count++; 380d7b241e6Sjeremylt CeedRequest_n++; 381d7b241e6Sjeremylt } 382d7b241e6Sjeremylt } 383d7b241e6Sjeremylt 384be9261b7Sjeremylt #define fCeedElemRestrictionApplyBlock \ 385be9261b7Sjeremylt FORTRAN_NAME(ceedelemrestrictionapplyblock,CEEDELEMRESTRICTIONAPPLYBLOCK) 386f90c8643Sjeremylt void fCeedElemRestrictionApplyBlock(int *elemr, int *block, int *tmode, 387be9261b7Sjeremylt int *uvec, int *ruvec, int *rqst, int *err) { 388be9261b7Sjeremylt int createRequest = 1; 389be9261b7Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 390be9261b7Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) 391be9261b7Sjeremylt createRequest = 0; 392be9261b7Sjeremylt 393be9261b7Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 394be9261b7Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 395be9261b7Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 396be9261b7Sjeremylt } 397be9261b7Sjeremylt 398be9261b7Sjeremylt CeedRequest *rqst_; 399be9261b7Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 400be9261b7Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED ) rqst_ = CEED_REQUEST_ORDERED; 401be9261b7Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 402be9261b7Sjeremylt 403be9261b7Sjeremylt *err = CeedElemRestrictionApplyBlock(CeedElemRestriction_dict[*elemr], *block, 404399bb197SJed Brown (CeedTransposeMode)*tmode, CeedVector_dict[*uvec], 405be9261b7Sjeremylt CeedVector_dict[*ruvec], rqst_); 406be9261b7Sjeremylt 407be9261b7Sjeremylt if (*err == 0 && createRequest) { 408be9261b7Sjeremylt *rqst = CeedRequest_count++; 409be9261b7Sjeremylt CeedRequest_n++; 410be9261b7Sjeremylt } 411be9261b7Sjeremylt } 412be9261b7Sjeremylt 4131469ee4dSjeremylt #define fCeedElemRestrictionGetMultiplicity \ 4141469ee4dSjeremylt FORTRAN_NAME(ceedelemrestrictiongetmultiplicity,CEEDELEMRESTRICTIONGETMULTIPLICITY) 415a8d32208Sjeremylt void fCeedElemRestrictionGetMultiplicity(int *elemr, int *mult, int *err) { 4161469ee4dSjeremylt *err = CeedElemRestrictionGetMultiplicity(CeedElemRestriction_dict[*elemr], 417a8d32208Sjeremylt CeedVector_dict[*mult]); 4181469ee4dSjeremylt } 4191469ee4dSjeremylt 42049fd234cSJeremy L Thompson #define fCeedElemRestrictionGetELayout \ 42149fd234cSJeremy L Thompson FORTRAN_NAME(ceedelemrestrictiongetelayout,CEEDELEMRESTRICTIONGETELAYOUT) 42249fd234cSJeremy L Thompson void fCeedElemRestrictionGetELayout(int *elemr, int *layout, int *err) { 42349fd234cSJeremy L Thompson CeedInt layout_c[3]; 42449fd234cSJeremy L Thompson *err = CeedElemRestrictionGetELayout(CeedElemRestriction_dict[*elemr], 42549fd234cSJeremy L Thompson &layout_c); 42649fd234cSJeremy L Thompson for (int i=0; i<3; i++) 42749fd234cSJeremy L Thompson layout[i] = layout_c[i]; 42849fd234cSJeremy L Thompson } 42949fd234cSJeremy L Thompson 4305a86f033Sjeremylt #define fCeedElemRestrictionView \ 4315a86f033Sjeremylt FORTRAN_NAME(ceedelemrestrictionview,CEEDELEMRESTRICTIONVIEW) 4325a86f033Sjeremylt void fCeedElemRestrictionView(int *elemr, int *err) { 4335a86f033Sjeremylt *err = CeedElemRestrictionView(CeedElemRestriction_dict[*elemr], stdout); 4345a86f033Sjeremylt } 4351469ee4dSjeremylt 436d7b241e6Sjeremylt #define fCeedRequestWait FORTRAN_NAME(ceedrequestwait, CEEDREQUESTWAIT) 437d7b241e6Sjeremylt void fCeedRequestWait(int *rqst, int *err) { 438d7b241e6Sjeremylt // TODO Uncomment this once CeedRequestWait is implemented 439d7b241e6Sjeremylt //*err = CeedRequestWait(&CeedRequest_dict[*rqst]); 440d7b241e6Sjeremylt 441d7b241e6Sjeremylt if (*err == 0) { 442d7b241e6Sjeremylt CeedRequest_n--; 443d7b241e6Sjeremylt if (CeedRequest_n == 0) { 444d7b241e6Sjeremylt CeedFree(&CeedRequest_dict); 445d7b241e6Sjeremylt CeedRequest_count = 0; 446d7b241e6Sjeremylt CeedRequest_count_max = 0; 447d7b241e6Sjeremylt } 448d7b241e6Sjeremylt } 449d7b241e6Sjeremylt } 450d7b241e6Sjeremylt 451d7b241e6Sjeremylt #define fCeedElemRestrictionDestroy \ 452d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictiondestroy,CEEDELEMRESTRICTIONDESTROY) 453d7b241e6Sjeremylt void fCeedElemRestrictionDestroy(int *elem, int *err) { 454b22696a8SJeremy L Thompson if (*elem == FORTRAN_NULL) return; 455d7b241e6Sjeremylt *err = CeedElemRestrictionDestroy(&CeedElemRestriction_dict[*elem]); 456d7b241e6Sjeremylt 457d7b241e6Sjeremylt if (*err == 0) { 45816595307SJeremy L Thompson *elem = FORTRAN_NULL; 459d7b241e6Sjeremylt CeedElemRestriction_n--; 460d7b241e6Sjeremylt if (CeedElemRestriction_n == 0) { 461d7b241e6Sjeremylt CeedFree(&CeedElemRestriction_dict); 462d7b241e6Sjeremylt CeedElemRestriction_count = 0; 463d7b241e6Sjeremylt CeedElemRestriction_count_max = 0; 464d7b241e6Sjeremylt } 465d7b241e6Sjeremylt } 466d7b241e6Sjeremylt } 467d7b241e6Sjeremylt 468b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 469b00b43a6SJeremy L Thompson // CeedBasis 470b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 471d7b241e6Sjeremylt static CeedBasis *CeedBasis_dict = NULL; 472d7b241e6Sjeremylt static int CeedBasis_count = 0; 473d7b241e6Sjeremylt static int CeedBasis_n = 0; 474d7b241e6Sjeremylt static int CeedBasis_count_max = 0; 475d7b241e6Sjeremylt 476d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1Lagrange \ 477d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1lagrange, CEEDBASISCREATETENSORH1LAGRANGE) 478d7b241e6Sjeremylt void fCeedBasisCreateTensorH1Lagrange(int *ceed, int *dim, 4798795c945Sjeremylt int *ncomp, int *P, int *Q, int *quadmode, 4808795c945Sjeremylt int *basis, int *err) { 481d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 482d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 483d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 484d7b241e6Sjeremylt } 485d7b241e6Sjeremylt 486a8de75f0Sjeremylt *err = CeedBasisCreateTensorH1Lagrange(Ceed_dict[*ceed], *dim, *ncomp, *P, *Q, 487399bb197SJed Brown (CeedQuadMode)*quadmode, 4888795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 489d7b241e6Sjeremylt 490d7b241e6Sjeremylt if (*err == 0) { 491d7b241e6Sjeremylt *basis = CeedBasis_count++; 492d7b241e6Sjeremylt CeedBasis_n++; 493d7b241e6Sjeremylt } 494d7b241e6Sjeremylt } 495d7b241e6Sjeremylt 496d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1 \ 497d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1, CEEDBASISCREATETENSORH1) 498a8de75f0Sjeremylt void fCeedBasisCreateTensorH1(int *ceed, int *dim, int *ncomp, int *P1d, 4998795c945Sjeremylt int *Q1d, const CeedScalar *interp1d, 5008795c945Sjeremylt const CeedScalar *grad1d, 5018795c945Sjeremylt const CeedScalar *qref1d, 5028795c945Sjeremylt const CeedScalar *qweight1d, int *basis, 5038795c945Sjeremylt int *err) { 504d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 505d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 506d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 507d7b241e6Sjeremylt } 508d7b241e6Sjeremylt 509a8de75f0Sjeremylt *err = CeedBasisCreateTensorH1(Ceed_dict[*ceed], *dim, *ncomp, *P1d, *Q1d, 5108795c945Sjeremylt interp1d, grad1d, qref1d, qweight1d, 5118795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 512d7b241e6Sjeremylt 513d7b241e6Sjeremylt if (*err == 0) { 514d7b241e6Sjeremylt *basis = CeedBasis_count++; 515d7b241e6Sjeremylt CeedBasis_n++; 516d7b241e6Sjeremylt } 517d7b241e6Sjeremylt } 518d7b241e6Sjeremylt 519a8de75f0Sjeremylt #define fCeedBasisCreateH1 \ 520a8de75f0Sjeremylt FORTRAN_NAME(ceedbasiscreateh1, CEEDBASISCREATEH1) 5218795c945Sjeremylt void fCeedBasisCreateH1(int *ceed, int *topo, int *ncomp, int *nnodes, 5228795c945Sjeremylt int *nqpts, const CeedScalar *interp, 5238795c945Sjeremylt const CeedScalar *grad, const CeedScalar *qref, 5248795c945Sjeremylt const CeedScalar *qweight, int *basis, int *err) { 525a8de75f0Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 526a8de75f0Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 527a8de75f0Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 528a8de75f0Sjeremylt } 529a8de75f0Sjeremylt 530399bb197SJed Brown *err = CeedBasisCreateH1(Ceed_dict[*ceed], (CeedElemTopology)*topo, *ncomp, 531399bb197SJed Brown *nnodes, *nqpts, interp, grad, qref, qweight, 5328795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 533a8de75f0Sjeremylt 534a8de75f0Sjeremylt if (*err == 0) { 535a8de75f0Sjeremylt *basis = CeedBasis_count++; 536a8de75f0Sjeremylt CeedBasis_n++; 537a8de75f0Sjeremylt } 538a8de75f0Sjeremylt } 539a8de75f0Sjeremylt 540d7b241e6Sjeremylt #define fCeedBasisView FORTRAN_NAME(ceedbasisview, CEEDBASISVIEW) 541d7b241e6Sjeremylt void fCeedBasisView(int *basis, int *err) { 542d7b241e6Sjeremylt *err = CeedBasisView(CeedBasis_dict[*basis], stdout); 543d7b241e6Sjeremylt } 544d7b241e6Sjeremylt 545d7b241e6Sjeremylt #define fCeedQRFactorization \ 546d7b241e6Sjeremylt FORTRAN_NAME(ceedqrfactorization, CEEDQRFACTORIZATION) 547a7bd39daSjeremylt void fCeedQRFactorization(int *ceed, CeedScalar *mat, CeedScalar *tau, int *m, 548a7bd39daSjeremylt int *n, int *err) { 549a7bd39daSjeremylt *err = CeedQRFactorization(Ceed_dict[*ceed], mat, tau, *m, *n); 550d7b241e6Sjeremylt } 551d7b241e6Sjeremylt 55252bfb9bbSJeremy L Thompson #define fCeedSymmetricSchurDecomposition \ 55352bfb9bbSJeremy L Thompson FORTRAN_NAME(ceedsymmetricschurdecomposition, CEEDSYMMETRICSCHURDECOMPOSITION) 55452bfb9bbSJeremy L Thompson void fCeedSymmetricSchurDecomposition(int *ceed, CeedScalar *mat, 55552bfb9bbSJeremy L Thompson CeedScalar *lambda, int *n, int *err) { 55652bfb9bbSJeremy L Thompson *err = CeedSymmetricSchurDecomposition(Ceed_dict[*ceed], mat, lambda, *n); 55752bfb9bbSJeremy L Thompson } 55852bfb9bbSJeremy L Thompson 55952bfb9bbSJeremy L Thompson #define fCeedSimultaneousDiagonalization \ 56052bfb9bbSJeremy L Thompson FORTRAN_NAME(ceedsimultaneousdiagonalization, CEEDSIMULTANEOUSDIAGONALIZATION) 56152bfb9bbSJeremy L Thompson void fCeedSimultaneousDiagonalization(int *ceed, CeedScalar *matA, 56252bfb9bbSJeremy L Thompson CeedScalar *matB, CeedScalar *x, 56352bfb9bbSJeremy L Thompson CeedScalar *lambda, int *n, int *err) { 56452bfb9bbSJeremy L Thompson *err = CeedSimultaneousDiagonalization(Ceed_dict[*ceed], matA, matB, x, 56552bfb9bbSJeremy L Thompson lambda, *n); 56652bfb9bbSJeremy L Thompson } 56752bfb9bbSJeremy L Thompson 568783c99b3SValeria Barra #define fCeedBasisGetCollocatedGrad \ 569783c99b3SValeria Barra FORTRAN_NAME(ceedbasisgetcollocatedgrad, CEEDBASISGETCOLLOCATEDGRAD) 570783c99b3SValeria Barra void fCeedBasisGetCollocatedGrad(int *basis, CeedScalar *colograd1d, 571d7b241e6Sjeremylt int *err) { 572783c99b3SValeria Barra *err = CeedBasisGetCollocatedGrad(CeedBasis_dict[*basis], colograd1d); 573d7b241e6Sjeremylt } 574d7b241e6Sjeremylt 575d7b241e6Sjeremylt #define fCeedBasisApply FORTRAN_NAME(ceedbasisapply, CEEDBASISAPPLY) 576d7b241e6Sjeremylt void fCeedBasisApply(int *basis, int *nelem, int *tmode, int *emode, 577aedaa0e5Sjeremylt int *u, int *v, int *err) { 578399bb197SJed Brown *err = CeedBasisApply(CeedBasis_dict[*basis], *nelem, (CeedTransposeMode)*tmode, 579399bb197SJed Brown (CeedEvalMode)*emode, 580399bb197SJed Brown *u == FORTRAN_VECTOR_NONE ? CEED_VECTOR_NONE : CeedVector_dict[*u], 581aedaa0e5Sjeremylt CeedVector_dict[*v]); 582d7b241e6Sjeremylt } 583d7b241e6Sjeremylt 584d7b241e6Sjeremylt #define fCeedBasisGetNumNodes \ 585d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumnodes, CEEDBASISGETNUMNODES) 586d7b241e6Sjeremylt void fCeedBasisGetNumNodes(int *basis, int *P, int *err) { 587d7b241e6Sjeremylt *err = CeedBasisGetNumNodes(CeedBasis_dict[*basis], P); 588d7b241e6Sjeremylt } 589d7b241e6Sjeremylt 590d7b241e6Sjeremylt #define fCeedBasisGetNumQuadraturePoints \ 591d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumquadraturepoints, CEEDBASISGETNUMQUADRATUREPOINTS) 592d7b241e6Sjeremylt void fCeedBasisGetNumQuadraturePoints(int *basis, int *Q, int *err) { 593d7b241e6Sjeremylt *err = CeedBasisGetNumQuadraturePoints(CeedBasis_dict[*basis], Q); 594d7b241e6Sjeremylt } 595d7b241e6Sjeremylt 596d99fa3c5SJeremy L Thompson #define fCeedBasisGetInterp1D \ 597d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedbasisgetinterp1d, CEEDBASISGETINTERP1D) 598d99fa3c5SJeremy L Thompson void fCeedBasisGetInterp1D(int *basis, CeedScalar *interp1d, int64_t *offset, 599d99fa3c5SJeremy L Thompson int *err) { 600d99fa3c5SJeremy L Thompson const CeedScalar *interp1d_; 601d99fa3c5SJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 602d99fa3c5SJeremy L Thompson *err = CeedBasisGetInterp1D(basis_, &interp1d_); 603d99fa3c5SJeremy L Thompson *offset = interp1d_ - interp1d; 604d99fa3c5SJeremy L Thompson } 605d99fa3c5SJeremy L Thompson 606d7b241e6Sjeremylt #define fCeedBasisDestroy FORTRAN_NAME(ceedbasisdestroy,CEEDBASISDESTROY) 607d7b241e6Sjeremylt void fCeedBasisDestroy(int *basis, int *err) { 608b22696a8SJeremy L Thompson if (*basis == FORTRAN_NULL) return; 609d7b241e6Sjeremylt *err = CeedBasisDestroy(&CeedBasis_dict[*basis]); 610d7b241e6Sjeremylt 611d7b241e6Sjeremylt if (*err == 0) { 61216595307SJeremy L Thompson *basis = FORTRAN_NULL; 613d7b241e6Sjeremylt CeedBasis_n--; 614d7b241e6Sjeremylt if (CeedBasis_n == 0) { 615d7b241e6Sjeremylt CeedFree(&CeedBasis_dict); 616d7b241e6Sjeremylt CeedBasis_count = 0; 617d7b241e6Sjeremylt CeedBasis_count_max = 0; 618d7b241e6Sjeremylt } 619d7b241e6Sjeremylt } 620d7b241e6Sjeremylt } 621d7b241e6Sjeremylt 622d7b241e6Sjeremylt #define fCeedGaussQuadrature FORTRAN_NAME(ceedgaussquadrature, CEEDGAUSSQUADRATURE) 623d7b241e6Sjeremylt void fCeedGaussQuadrature(int *Q, CeedScalar *qref1d, CeedScalar *qweight1d, 624d7b241e6Sjeremylt int *err) { 625d7b241e6Sjeremylt *err = CeedGaussQuadrature(*Q, qref1d, qweight1d); 626d7b241e6Sjeremylt } 627d7b241e6Sjeremylt 628d7b241e6Sjeremylt #define fCeedLobattoQuadrature \ 629d7b241e6Sjeremylt FORTRAN_NAME(ceedlobattoquadrature, CEEDLOBATTOQUADRATURE) 630d7b241e6Sjeremylt void fCeedLobattoQuadrature(int *Q, CeedScalar *qref1d, CeedScalar *qweight1d, 631d7b241e6Sjeremylt int *err) { 632d7b241e6Sjeremylt *err = CeedLobattoQuadrature(*Q, qref1d, qweight1d); 633d7b241e6Sjeremylt } 634d7b241e6Sjeremylt 635b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 636777ff853SJeremy L Thompson // CeedQFunctionContext 637777ff853SJeremy L Thompson // ----------------------------------------------------------------------------- 638777ff853SJeremy L Thompson static CeedQFunctionContext *CeedQFunctionContext_dict = NULL; 639777ff853SJeremy L Thompson static int CeedQFunctionContext_count = 0; 640777ff853SJeremy L Thompson static int CeedQFunctionContext_n = 0; 641777ff853SJeremy L Thompson static int CeedQFunctionContext_count_max = 0; 642777ff853SJeremy L Thompson 643777ff853SJeremy L Thompson #define fCeedQFunctionContextCreate \ 644777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextcreate,CEEDQFUNCTIONCONTEXTCREATE) 645777ff853SJeremy L Thompson void fCeedQFunctionContextCreate(int *ceed, int *ctx, int *err) { 646777ff853SJeremy L Thompson if (CeedQFunctionContext_count == CeedQFunctionContext_count_max) { 647777ff853SJeremy L Thompson CeedQFunctionContext_count_max += CeedQFunctionContext_count_max/2 + 1; 648777ff853SJeremy L Thompson CeedRealloc(CeedQFunctionContext_count_max, &CeedQFunctionContext_dict); 649777ff853SJeremy L Thompson } 650777ff853SJeremy L Thompson 651777ff853SJeremy L Thompson CeedQFunctionContext *ctx_ = 652777ff853SJeremy L Thompson &CeedQFunctionContext_dict[CeedQFunctionContext_count]; 653777ff853SJeremy L Thompson 654777ff853SJeremy L Thompson *err = CeedQFunctionContextCreate(Ceed_dict[*ceed], ctx_); 655777ff853SJeremy L Thompson if (*err) return; 656777ff853SJeremy L Thompson *ctx = CeedQFunctionContext_count++; 657777ff853SJeremy L Thompson CeedQFunctionContext_n++; 658777ff853SJeremy L Thompson } 659777ff853SJeremy L Thompson 660777ff853SJeremy L Thompson #define fCeedQFunctionContextSetData \ 661777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextsetdata,CEEDQFUNCTIONCONTEXTSETDATA) 662777ff853SJeremy L Thompson void fCeedQFunctionContextSetData(int *ctx, int *memtype, int *copymode, 663777ff853SJeremy L Thompson CeedInt *n, 664777ff853SJeremy L Thompson CeedScalar *data, int64_t *offset, int *err) { 665777ff853SJeremy L Thompson size_t ctxsize = ((size_t) *n)*sizeof(CeedScalar); 666777ff853SJeremy L Thompson *err = CeedQFunctionContextSetData(CeedQFunctionContext_dict[*ctx], 667777ff853SJeremy L Thompson (CeedMemType)*memtype, 668777ff853SJeremy L Thompson (CeedCopyMode)*copymode, ctxsize, 669777ff853SJeremy L Thompson data + *offset); 670777ff853SJeremy L Thompson } 671777ff853SJeremy L Thompson 672777ff853SJeremy L Thompson #define fCeedQFunctionContextGetData \ 673777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextgetdata,CEEDQFUNCTIONCONTEXTGETDATA) 674777ff853SJeremy L Thompson void fCeedQFunctionContextGetData(int *ctx, int *memtype, CeedScalar *data, 675777ff853SJeremy L Thompson int64_t *offset, int *err) { 676777ff853SJeremy L Thompson CeedScalar *b; 677777ff853SJeremy L Thompson CeedQFunctionContext ctx_ = CeedQFunctionContext_dict[*ctx]; 678777ff853SJeremy L Thompson *err = CeedQFunctionContextGetData(ctx_, (CeedMemType)*memtype, &b); 679777ff853SJeremy L Thompson *offset = b - data; 680777ff853SJeremy L Thompson } 681777ff853SJeremy L Thompson 682777ff853SJeremy L Thompson #define fCeedQFunctionContextRestoreData \ 683777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextrestoredata,CEEDQFUNCTIONCONTEXTRESTOREDATA) 684777ff853SJeremy L Thompson void fCeedQFunctionContextRestoreData(int *ctx, CeedScalar *data, 685777ff853SJeremy L Thompson int64_t *offset, int *err) { 686777ff853SJeremy L Thompson *err = CeedQFunctionContextRestoreData(CeedQFunctionContext_dict[*ctx], 687777ff853SJeremy L Thompson (void **)&data); 688777ff853SJeremy L Thompson *offset = 0; 689777ff853SJeremy L Thompson } 690777ff853SJeremy L Thompson 691777ff853SJeremy L Thompson #define fCeedQFunctionContextView \ 692777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextview,CEEDQFUNCTIONCONTEXTVIEW) 693777ff853SJeremy L Thompson void fCeedQFunctionContextView(int *ctx, int *err) { 694777ff853SJeremy L Thompson *err = CeedQFunctionContextView(CeedQFunctionContext_dict[*ctx], stdout); 695777ff853SJeremy L Thompson } 696777ff853SJeremy L Thompson 697777ff853SJeremy L Thompson #define fCeedQFunctionContextDestroy \ 698777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextdestroy,CEEDQFUNCTIONCONTEXTDESTROY) 699777ff853SJeremy L Thompson void fCeedQFunctionContextDestroy(int *ctx, int *err) { 700777ff853SJeremy L Thompson if (*ctx == FORTRAN_NULL) return; 701777ff853SJeremy L Thompson *err = CeedQFunctionContextDestroy(&CeedQFunctionContext_dict[*ctx]); 702777ff853SJeremy L Thompson 703777ff853SJeremy L Thompson if (*err == 0) { 704777ff853SJeremy L Thompson *ctx = FORTRAN_NULL; 705777ff853SJeremy L Thompson CeedQFunctionContext_n--; 706777ff853SJeremy L Thompson if (CeedQFunctionContext_n == 0) { 707777ff853SJeremy L Thompson CeedFree(&CeedQFunctionContext_dict); 708777ff853SJeremy L Thompson CeedQFunctionContext_count = 0; 709777ff853SJeremy L Thompson CeedQFunctionContext_count_max = 0; 710777ff853SJeremy L Thompson } 711777ff853SJeremy L Thompson } 712777ff853SJeremy L Thompson } 713777ff853SJeremy L Thompson 714777ff853SJeremy L Thompson // ----------------------------------------------------------------------------- 715b00b43a6SJeremy L Thompson // CeedQFunction 716b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 717d7b241e6Sjeremylt static CeedQFunction *CeedQFunction_dict = NULL; 718d7b241e6Sjeremylt static int CeedQFunction_count = 0; 719d7b241e6Sjeremylt static int CeedQFunction_n = 0; 720d7b241e6Sjeremylt static int CeedQFunction_count_max = 0; 721d7b241e6Sjeremylt 722d7b241e6Sjeremylt static int CeedQFunctionFortranStub(void *ctx, int nq, 7238795c945Sjeremylt const CeedScalar *const *u, 7248795c945Sjeremylt CeedScalar *const *v) { 725777ff853SJeremy L Thompson CeedFortranContext fctx = ctx; 726777ff853SJeremy L Thompson CeedQFunctionContext innerctx = fctx->innerctx; 727d7b241e6Sjeremylt int ierr; 728d7b241e6Sjeremylt 729777ff853SJeremy L Thompson CeedScalar *ctx_ = NULL; 730777ff853SJeremy L Thompson // Note: Device backends are generating their own kernels from 731777ff853SJeremy L Thompson // single source files, so only Host backends need to 732777ff853SJeremy L Thompson // use this Fortran stub. 733777ff853SJeremy L Thompson if (innerctx) { 734777ff853SJeremy L Thompson ierr = CeedQFunctionContextGetData(innerctx, CEED_MEM_HOST, &ctx_); 735777ff853SJeremy L Thompson CeedChk(ierr); 736777ff853SJeremy L Thompson } 737777ff853SJeremy L Thompson 7381e35832bSjeremylt fctx->f((void *)ctx_,&nq,u[0],u[1],u[2],u[3],u[4],u[5],u[6], 739418fb8c2Sjeremylt u[7],u[8],u[9],u[10],u[11],u[12],u[13],u[14],u[15], 740418fb8c2Sjeremylt v[0],v[1],v[2],v[3],v[4],v[5],v[6],v[7],v[8],v[9], 741418fb8c2Sjeremylt v[10],v[11],v[12],v[13],v[14],v[15],&ierr); 742777ff853SJeremy L Thompson 743777ff853SJeremy L Thompson if (innerctx) { 744777ff853SJeremy L Thompson ierr = CeedQFunctionContextRestoreData(innerctx, (void *)&ctx_); 745777ff853SJeremy L Thompson CeedChk(ierr); 746777ff853SJeremy L Thompson } 747777ff853SJeremy L Thompson 748d7b241e6Sjeremylt return ierr; 749d7b241e6Sjeremylt } 750d7b241e6Sjeremylt 751d7b241e6Sjeremylt #define fCeedQFunctionCreateInterior \ 752d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctioncreateinterior, CEEDQFUNCTIONCREATEINTERIOR) 753d7b241e6Sjeremylt void fCeedQFunctionCreateInterior(int *ceed, int *vlength, 754d7b241e6Sjeremylt void (*f)(void *ctx, int *nq, 755418fb8c2Sjeremylt const CeedScalar *u,const CeedScalar *u1, 756418fb8c2Sjeremylt const CeedScalar *u2,const CeedScalar *u3, 757418fb8c2Sjeremylt const CeedScalar *u4,const CeedScalar *u5, 758418fb8c2Sjeremylt const CeedScalar *u6,const CeedScalar *u7, 759418fb8c2Sjeremylt const CeedScalar *u8,const CeedScalar *u9, 760418fb8c2Sjeremylt const CeedScalar *u10,const CeedScalar *u11, 761418fb8c2Sjeremylt const CeedScalar *u12,const CeedScalar *u13, 762418fb8c2Sjeremylt const CeedScalar *u14,const CeedScalar *u15, 763418fb8c2Sjeremylt CeedScalar *v,CeedScalar *v1,CeedScalar *v2, 7648795c945Sjeremylt CeedScalar *v3,CeedScalar *v4, 7658795c945Sjeremylt CeedScalar *v5,CeedScalar *v6, 7668795c945Sjeremylt CeedScalar *v7,CeedScalar *v8, 7678795c945Sjeremylt CeedScalar *v9,CeedScalar *v10, 7688795c945Sjeremylt CeedScalar *v11,CeedScalar *v12, 7698795c945Sjeremylt CeedScalar *v13,CeedScalar *v14, 770418fb8c2Sjeremylt CeedScalar *v15,int *err), 771288c0443SJeremy L Thompson const char *source, int *qf, int *err, 772288c0443SJeremy L Thompson fortran_charlen_t source_len) { 773288c0443SJeremy L Thompson FIX_STRING(source); 774d7b241e6Sjeremylt if (CeedQFunction_count == CeedQFunction_count_max) { 775d7b241e6Sjeremylt CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 776d7b241e6Sjeremylt CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 777d7b241e6Sjeremylt } 778d7b241e6Sjeremylt 779d7b241e6Sjeremylt CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 780d7b241e6Sjeremylt *err = CeedQFunctionCreateInterior(Ceed_dict[*ceed], *vlength, 781288c0443SJeremy L Thompson CeedQFunctionFortranStub, source_c, qf_); 782d7b241e6Sjeremylt 783d7b241e6Sjeremylt if (*err == 0) { 784d7b241e6Sjeremylt *qf = CeedQFunction_count++; 785d7b241e6Sjeremylt CeedQFunction_n++; 786d7b241e6Sjeremylt } 787d7b241e6Sjeremylt 788777ff853SJeremy L Thompson CeedFortranContext fctxdata; 789777ff853SJeremy L Thompson *err = CeedCalloc(1, &fctxdata); 790d7b241e6Sjeremylt if (*err) return; 791777ff853SJeremy L Thompson fctxdata->f = f; fctxdata->innerctx = NULL; 792777ff853SJeremy L Thompson CeedQFunctionContext fctx; 793777ff853SJeremy L Thompson *err = CeedQFunctionContextCreate(Ceed_dict[*ceed], &fctx); 794777ff853SJeremy L Thompson if (*err) return; 795777ff853SJeremy L Thompson *err = CeedQFunctionContextSetData(fctx, CEED_MEM_HOST, CEED_OWN_POINTER, 796777ff853SJeremy L Thompson sizeof(*fctxdata), fctxdata); 797777ff853SJeremy L Thompson if (*err) return; 798777ff853SJeremy L Thompson *err = CeedQFunctionSetContext(*qf_, fctx); 799777ff853SJeremy L Thompson if (*err) return; 800777ff853SJeremy L Thompson CeedQFunctionContextDestroy(&fctx); 801777ff853SJeremy L Thompson if (*err) return; 802d7b241e6Sjeremylt 803777ff853SJeremy L Thompson *err = CeedQFunctionSetFortranStatus(*qf_, true); 804d7b241e6Sjeremylt } 805d7b241e6Sjeremylt 806288c0443SJeremy L Thompson #define fCeedQFunctionCreateInteriorByName \ 807288c0443SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncreateinteriorbyname, CEEDQFUNCTIONCREATEINTERIORBYNAME) 808288c0443SJeremy L Thompson void fCeedQFunctionCreateInteriorByName(int *ceed, const char *name, int *qf, 809288c0443SJeremy L Thompson int *err, fortran_charlen_t name_len) { 810288c0443SJeremy L Thompson FIX_STRING(name); 811288c0443SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 812288c0443SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 813288c0443SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 814288c0443SJeremy L Thompson } 815288c0443SJeremy L Thompson 816288c0443SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 817288c0443SJeremy L Thompson *err = CeedQFunctionCreateInteriorByName(Ceed_dict[*ceed], name_c, qf_); 818288c0443SJeremy L Thompson 819288c0443SJeremy L Thompson if (*err == 0) { 820288c0443SJeremy L Thompson *qf = CeedQFunction_count++; 821288c0443SJeremy L Thompson CeedQFunction_n++; 822288c0443SJeremy L Thompson } 823288c0443SJeremy L Thompson } 824288c0443SJeremy L Thompson 8250219ea01SJeremy L Thompson #define fCeedQFunctionCreateIdentity \ 8260219ea01SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncreateidentity, CEEDQFUNCTIONCREATEIDENTITY) 82760f77c51Sjeremylt void fCeedQFunctionCreateIdentity(int *ceed, int *size, int *inmode, 82860f77c51Sjeremylt int *outmode, int *qf, int *err) { 8290219ea01SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 8300219ea01SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 8310219ea01SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 8320219ea01SJeremy L Thompson } 8330219ea01SJeremy L Thompson 8340219ea01SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 835399bb197SJed Brown *err = CeedQFunctionCreateIdentity(Ceed_dict[*ceed], *size, 836399bb197SJed Brown (CeedEvalMode)*inmode, 837399bb197SJed Brown (CeedEvalMode)*outmode, qf_); 8380219ea01SJeremy L Thompson 8390219ea01SJeremy L Thompson if (*err == 0) { 8400219ea01SJeremy L Thompson *qf = CeedQFunction_count++; 8410219ea01SJeremy L Thompson CeedQFunction_n++; 8420219ea01SJeremy L Thompson } 8430219ea01SJeremy L Thompson } 8440219ea01SJeremy L Thompson 845d7b241e6Sjeremylt #define fCeedQFunctionAddInput \ 846d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionaddinput,CEEDQFUNCTIONADDINPUT) 847d7b241e6Sjeremylt void fCeedQFunctionAddInput(int *qf, const char *fieldname, 848c2ae719dSJed Brown CeedInt *ncomp, CeedEvalMode *emode, int *err, 849c2ae719dSJed Brown fortran_charlen_t fieldname_len) { 850c2ae719dSJed Brown FIX_STRING(fieldname); 851d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 852d7b241e6Sjeremylt 853c2ae719dSJed Brown *err = CeedQFunctionAddInput(qf_, fieldname_c, *ncomp, *emode); 854d7b241e6Sjeremylt } 855d7b241e6Sjeremylt 856d7b241e6Sjeremylt #define fCeedQFunctionAddOutput \ 857d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionaddoutput,CEEDQFUNCTIONADDOUTPUT) 858d7b241e6Sjeremylt void fCeedQFunctionAddOutput(int *qf, const char *fieldname, 859c2ae719dSJed Brown CeedInt *ncomp, CeedEvalMode *emode, int *err, 860c2ae719dSJed Brown fortran_charlen_t fieldname_len) { 861c2ae719dSJed Brown FIX_STRING(fieldname); 862d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 863d7b241e6Sjeremylt 864c2ae719dSJed Brown *err = CeedQFunctionAddOutput(qf_, fieldname_c, *ncomp, *emode); 865d7b241e6Sjeremylt } 866d7b241e6Sjeremylt 867418fb8c2Sjeremylt #define fCeedQFunctionSetContext \ 868418fb8c2Sjeremylt FORTRAN_NAME(ceedqfunctionsetcontext,CEEDQFUNCTIONSETCONTEXT) 869777ff853SJeremy L Thompson void fCeedQFunctionSetContext(int *qf, int *ctx, int *err) { 870418fb8c2Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 871777ff853SJeremy L Thompson CeedQFunctionContext ctx_ = CeedQFunctionContext_dict[*ctx]; 872418fb8c2Sjeremylt 873777ff853SJeremy L Thompson CeedQFunctionContext fctx; 874777ff853SJeremy L Thompson *err = CeedQFunctionGetContext(qf_, &fctx); 875777ff853SJeremy L Thompson if (*err) return; 876777ff853SJeremy L Thompson CeedFortranContext fctxdata; 877777ff853SJeremy L Thompson *err = CeedQFunctionContextGetData(fctx, CEED_MEM_HOST, &fctxdata); 878777ff853SJeremy L Thompson if (*err) return; 879777ff853SJeremy L Thompson fctxdata->innerctx = ctx_; 880777ff853SJeremy L Thompson *err = CeedQFunctionContextRestoreData(fctx, (void **)&fctxdata); 881418fb8c2Sjeremylt } 882418fb8c2Sjeremylt 88375affc3bSjeremylt #define fCeedQFunctionView \ 88475affc3bSjeremylt FORTRAN_NAME(ceedqfunctionview,CEEDQFUNCTIONVIEW) 88575affc3bSjeremylt void fCeedQFunctionView(int *qf, int *err) { 88675affc3bSjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 88775affc3bSjeremylt 88875affc3bSjeremylt *err = CeedQFunctionView(qf_, stdout); 88975affc3bSjeremylt } 89075affc3bSjeremylt 891d7b241e6Sjeremylt #define fCeedQFunctionApply \ 892d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionapply,CEEDQFUNCTIONAPPLY) 893d7b241e6Sjeremylt //TODO Need Fixing, double pointer 894d7b241e6Sjeremylt void fCeedQFunctionApply(int *qf, int *Q, 895aedaa0e5Sjeremylt int *u, int *u1, int *u2, int *u3, 896aedaa0e5Sjeremylt int *u4, int *u5, int *u6, int *u7, 897aedaa0e5Sjeremylt int *u8, int *u9, int *u10, int *u11, 898aedaa0e5Sjeremylt int *u12, int *u13, int *u14, int *u15, 899aedaa0e5Sjeremylt int *v, int *v1, int *v2, int *v3, 900aedaa0e5Sjeremylt int *v4, int *v5, int *v6, int *v7, 901aedaa0e5Sjeremylt int *v8, int *v9, int *v10, int *v11, 902aedaa0e5Sjeremylt int *v12, int *v13, int *v14, int *v15, int *err) { 903d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 904aedaa0e5Sjeremylt CeedVector *in; 905d7b241e6Sjeremylt *err = CeedCalloc(16, &in); 906d7b241e6Sjeremylt if (*err) return; 907aedaa0e5Sjeremylt in[0] = *u==FORTRAN_NULL?NULL:CeedVector_dict[*u]; 908aedaa0e5Sjeremylt in[1] = *u1==FORTRAN_NULL?NULL:CeedVector_dict[*u1]; 909aedaa0e5Sjeremylt in[2] = *u2==FORTRAN_NULL?NULL:CeedVector_dict[*u2]; 910aedaa0e5Sjeremylt in[3] = *u3==FORTRAN_NULL?NULL:CeedVector_dict[*u3]; 911aedaa0e5Sjeremylt in[4] = *u4==FORTRAN_NULL?NULL:CeedVector_dict[*u4]; 912aedaa0e5Sjeremylt in[5] = *u5==FORTRAN_NULL?NULL:CeedVector_dict[*u5]; 913aedaa0e5Sjeremylt in[6] = *u6==FORTRAN_NULL?NULL:CeedVector_dict[*u6]; 914aedaa0e5Sjeremylt in[7] = *u7==FORTRAN_NULL?NULL:CeedVector_dict[*u7]; 915aedaa0e5Sjeremylt in[8] = *u8==FORTRAN_NULL?NULL:CeedVector_dict[*u8]; 916aedaa0e5Sjeremylt in[9] = *u9==FORTRAN_NULL?NULL:CeedVector_dict[*u9]; 917aedaa0e5Sjeremylt in[10] = *u10==FORTRAN_NULL?NULL:CeedVector_dict[*u10]; 918aedaa0e5Sjeremylt in[11] = *u11==FORTRAN_NULL?NULL:CeedVector_dict[*u11]; 919aedaa0e5Sjeremylt in[12] = *u12==FORTRAN_NULL?NULL:CeedVector_dict[*u12]; 920aedaa0e5Sjeremylt in[13] = *u13==FORTRAN_NULL?NULL:CeedVector_dict[*u13]; 921aedaa0e5Sjeremylt in[14] = *u14==FORTRAN_NULL?NULL:CeedVector_dict[*u14]; 922aedaa0e5Sjeremylt in[15] = *u15==FORTRAN_NULL?NULL:CeedVector_dict[*u15]; 923aedaa0e5Sjeremylt CeedVector *out; 924d7b241e6Sjeremylt *err = CeedCalloc(16, &out); 925d7b241e6Sjeremylt if (*err) return; 926aedaa0e5Sjeremylt out[0] = *v==FORTRAN_NULL?NULL:CeedVector_dict[*v]; 927aedaa0e5Sjeremylt out[1] = *v1==FORTRAN_NULL?NULL:CeedVector_dict[*v1]; 928aedaa0e5Sjeremylt out[2] = *v2==FORTRAN_NULL?NULL:CeedVector_dict[*v2]; 929aedaa0e5Sjeremylt out[3] = *v3==FORTRAN_NULL?NULL:CeedVector_dict[*v3]; 930aedaa0e5Sjeremylt out[4] = *v4==FORTRAN_NULL?NULL:CeedVector_dict[*v4]; 931aedaa0e5Sjeremylt out[5] = *v5==FORTRAN_NULL?NULL:CeedVector_dict[*v5]; 932aedaa0e5Sjeremylt out[6] = *v6==FORTRAN_NULL?NULL:CeedVector_dict[*v6]; 933aedaa0e5Sjeremylt out[7] = *v7==FORTRAN_NULL?NULL:CeedVector_dict[*v7]; 934aedaa0e5Sjeremylt out[8] = *v8==FORTRAN_NULL?NULL:CeedVector_dict[*v8]; 935aedaa0e5Sjeremylt out[9] = *v9==FORTRAN_NULL?NULL:CeedVector_dict[*v9]; 936aedaa0e5Sjeremylt out[10] = *v10==FORTRAN_NULL?NULL:CeedVector_dict[*v10]; 937aedaa0e5Sjeremylt out[11] = *v11==FORTRAN_NULL?NULL:CeedVector_dict[*v11]; 938aedaa0e5Sjeremylt out[12] = *v12==FORTRAN_NULL?NULL:CeedVector_dict[*v12]; 939aedaa0e5Sjeremylt out[13] = *v13==FORTRAN_NULL?NULL:CeedVector_dict[*v13]; 940aedaa0e5Sjeremylt out[14] = *v14==FORTRAN_NULL?NULL:CeedVector_dict[*v14]; 941aedaa0e5Sjeremylt out[15] = *v15==FORTRAN_NULL?NULL:CeedVector_dict[*v15]; 942aedaa0e5Sjeremylt *err = CeedQFunctionApply(qf_, *Q, in, out); 943d7b241e6Sjeremylt if (*err) return; 944d7b241e6Sjeremylt 945d7b241e6Sjeremylt *err = CeedFree(&in); 946d7b241e6Sjeremylt if (*err) return; 947d7b241e6Sjeremylt *err = CeedFree(&out); 948d7b241e6Sjeremylt } 949d7b241e6Sjeremylt 950d7b241e6Sjeremylt #define fCeedQFunctionDestroy \ 951d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctiondestroy,CEEDQFUNCTIONDESTROY) 952d7b241e6Sjeremylt void fCeedQFunctionDestroy(int *qf, int *err) { 953b22696a8SJeremy L Thompson if (*qf == FORTRAN_NULL) return; 954c4b814baSjeremylt 955c4b814baSjeremylt *err = CeedQFunctionDestroy(&CeedQFunction_dict[*qf]); 95616595307SJeremy L Thompson if (*err == 0) { 95716595307SJeremy L Thompson *qf = FORTRAN_NULL; 958d7b241e6Sjeremylt CeedQFunction_n--; 959d7b241e6Sjeremylt if (CeedQFunction_n == 0) { 960d7b241e6Sjeremylt *err = CeedFree(&CeedQFunction_dict); 961d7b241e6Sjeremylt CeedQFunction_count = 0; 962d7b241e6Sjeremylt CeedQFunction_count_max = 0; 963d7b241e6Sjeremylt } 964d7b241e6Sjeremylt } 96516595307SJeremy L Thompson } 966d7b241e6Sjeremylt 967b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 968b00b43a6SJeremy L Thompson // CeedOperator 969b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 970d7b241e6Sjeremylt static CeedOperator *CeedOperator_dict = NULL; 971d7b241e6Sjeremylt static int CeedOperator_count = 0; 972d7b241e6Sjeremylt static int CeedOperator_n = 0; 973d7b241e6Sjeremylt static int CeedOperator_count_max = 0; 974d7b241e6Sjeremylt 975d7b241e6Sjeremylt #define fCeedOperatorCreate \ 976d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorcreate, CEEDOPERATORCREATE) 977d7b241e6Sjeremylt void fCeedOperatorCreate(int *ceed, 978d7b241e6Sjeremylt int *qf, int *dqf, int *dqfT, int *op, int *err) { 979777ff853SJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) { 980777ff853SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 981777ff853SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 982777ff853SJeremy L Thompson } 983d7b241e6Sjeremylt 984d7b241e6Sjeremylt CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 985d7b241e6Sjeremylt 986442e7f0bSjeremylt CeedQFunction dqf_ = CEED_QFUNCTION_NONE, dqfT_ = CEED_QFUNCTION_NONE; 987442e7f0bSjeremylt if (*dqf != FORTRAN_QFUNCTION_NONE) dqf_ = CeedQFunction_dict[*dqf ]; 988442e7f0bSjeremylt if (*dqfT != FORTRAN_QFUNCTION_NONE) dqfT_ = CeedQFunction_dict[*dqfT]; 989d7b241e6Sjeremylt 990d7b241e6Sjeremylt *err = CeedOperatorCreate(Ceed_dict[*ceed], CeedQFunction_dict[*qf], dqf_, 991d7b241e6Sjeremylt dqfT_, op_); 992d7b241e6Sjeremylt if (*err) return; 993d7b241e6Sjeremylt *op = CeedOperator_count++; 994d7b241e6Sjeremylt CeedOperator_n++; 995d7b241e6Sjeremylt } 996d7b241e6Sjeremylt 99752d6035fSJeremy L Thompson #define fCeedCompositeOperatorCreate \ 99852d6035fSJeremy L Thompson FORTRAN_NAME(ceedcompositeoperatorcreate, CEEDCOMPOSITEOPERATORCREATE) 99952d6035fSJeremy L Thompson void fCeedCompositeOperatorCreate(int *ceed, int *op, int *err) { 1000777ff853SJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) { 1001777ff853SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1002777ff853SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1003777ff853SJeremy L Thompson } 100452d6035fSJeremy L Thompson 100552d6035fSJeremy L Thompson CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 100652d6035fSJeremy L Thompson 100752d6035fSJeremy L Thompson *err = CeedCompositeOperatorCreate(Ceed_dict[*ceed], op_); 100852d6035fSJeremy L Thompson if (*err) return; 100952d6035fSJeremy L Thompson *op = CeedOperator_count++; 101052d6035fSJeremy L Thompson CeedOperator_n++; 101152d6035fSJeremy L Thompson } 101252d6035fSJeremy L Thompson 1013d7b241e6Sjeremylt #define fCeedOperatorSetField \ 1014d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorsetfield,CEEDOPERATORSETFIELD) 1015a8d32208Sjeremylt void fCeedOperatorSetField(int *op, const char *fieldname, int *r, int *b, 1016a8d32208Sjeremylt int *v, int *err, fortran_charlen_t fieldname_len) { 1017c2ae719dSJed Brown FIX_STRING(fieldname); 1018d7b241e6Sjeremylt CeedElemRestriction r_; 1019d7b241e6Sjeremylt CeedBasis b_; 1020d7b241e6Sjeremylt CeedVector v_; 1021d7b241e6Sjeremylt 1022d7b241e6Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 1023d7b241e6Sjeremylt 1024d7b241e6Sjeremylt if (*r == FORTRAN_NULL) { 1025d7b241e6Sjeremylt r_ = NULL; 102615910d16Sjeremylt } else if (*r == FORTRAN_ELEMRESTRICTION_NONE) { 102715910d16Sjeremylt r_ = CEED_ELEMRESTRICTION_NONE; 1028d7b241e6Sjeremylt } else { 1029d7b241e6Sjeremylt r_ = CeedElemRestriction_dict[*r]; 1030d7b241e6Sjeremylt } 10314dccadb6Sjeremylt 1032d7b241e6Sjeremylt if (*b == FORTRAN_NULL) { 1033d7b241e6Sjeremylt b_ = NULL; 1034783c99b3SValeria Barra } else if (*b == FORTRAN_BASIS_COLLOCATED) { 1035783c99b3SValeria Barra b_ = CEED_BASIS_COLLOCATED; 1036d7b241e6Sjeremylt } else { 1037d7b241e6Sjeremylt b_ = CeedBasis_dict[*b]; 1038d7b241e6Sjeremylt } 1039d7b241e6Sjeremylt if (*v == FORTRAN_NULL) { 1040d7b241e6Sjeremylt v_ = NULL; 1041d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_ACTIVE) { 1042d7b241e6Sjeremylt v_ = CEED_VECTOR_ACTIVE; 1043d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_NONE) { 1044d7b241e6Sjeremylt v_ = CEED_VECTOR_NONE; 1045d7b241e6Sjeremylt } else { 1046d7b241e6Sjeremylt v_ = CeedVector_dict[*v]; 1047d7b241e6Sjeremylt } 1048d7b241e6Sjeremylt 1049a8d32208Sjeremylt *err = CeedOperatorSetField(op_, fieldname_c, r_, b_, v_); 1050d7b241e6Sjeremylt } 1051d7b241e6Sjeremylt 105252d6035fSJeremy L Thompson #define fCeedCompositeOperatorAddSub \ 105352d6035fSJeremy L Thompson FORTRAN_NAME(ceedcompositeoperatoraddsub, CEEDCOMPOSITEOPERATORADDSUB) 105452d6035fSJeremy L Thompson void fCeedCompositeOperatorAddSub(int *compositeop, int *subop, int *err) { 105552d6035fSJeremy L Thompson CeedOperator compositeop_ = CeedOperator_dict[*compositeop]; 105652d6035fSJeremy L Thompson CeedOperator subop_ = CeedOperator_dict[*subop]; 105752d6035fSJeremy L Thompson 105852d6035fSJeremy L Thompson *err = CeedCompositeOperatorAddSub(compositeop_, subop_); 105952d6035fSJeremy L Thompson if (*err) return; 106052d6035fSJeremy L Thompson } 106152d6035fSJeremy L Thompson 1062d99fa3c5SJeremy L Thompson #define fCeedOperatorLinearAssembleQFunction \ 1063d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatorlinearassembleqfunction, CEEDOPERATORLINEARASSEMBLEQFUNCTION) 106480ac2e43SJeremy L Thompson void fCeedOperatorLinearAssembleQFunction(int *op, int *assembledvec, 10651d102b48SJeremy L Thompson int *assembledrstr, int *rqst, int *err) { 10661d102b48SJeremy L Thompson // Vector 10671d102b48SJeremy L Thompson if (CeedVector_count == CeedVector_count_max) { 10681d102b48SJeremy L Thompson CeedVector_count_max += CeedVector_count_max/2 + 1; 10691d102b48SJeremy L Thompson CeedRealloc(CeedVector_count_max, &CeedVector_dict); 10701d102b48SJeremy L Thompson } 10711d102b48SJeremy L Thompson CeedVector *assembledvec_ = &CeedVector_dict[CeedVector_count]; 10721d102b48SJeremy L Thompson 10731d102b48SJeremy L Thompson // Restriction 10741d102b48SJeremy L Thompson if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 10751d102b48SJeremy L Thompson CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 10761d102b48SJeremy L Thompson CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 10771d102b48SJeremy L Thompson } 10781d102b48SJeremy L Thompson CeedElemRestriction *rstr_ = 10791d102b48SJeremy L Thompson &CeedElemRestriction_dict[CeedElemRestriction_count]; 10801d102b48SJeremy L Thompson 10811d102b48SJeremy L Thompson int createRequest = 1; 10821d102b48SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 10831d102b48SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 10841d102b48SJeremy L Thompson createRequest = 0; 10851d102b48SJeremy L Thompson } 10861d102b48SJeremy L Thompson 10871d102b48SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 10881d102b48SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max/2 + 1; 10891d102b48SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 10901d102b48SJeremy L Thompson } 10911d102b48SJeremy L Thompson 10921d102b48SJeremy L Thompson CeedRequest *rqst_; 10931d102b48SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 10941d102b48SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 10951d102b48SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 10961d102b48SJeremy L Thompson 109780ac2e43SJeremy L Thompson *err = CeedOperatorLinearAssembleQFunction(CeedOperator_dict[*op], 10981d102b48SJeremy L Thompson assembledvec_, rstr_, rqst_); 10991d102b48SJeremy L Thompson if (*err) return; 11001d102b48SJeremy L Thompson if (createRequest) { 11011d102b48SJeremy L Thompson *rqst = CeedRequest_count++; 11021d102b48SJeremy L Thompson CeedRequest_n++; 11031d102b48SJeremy L Thompson } 11041d102b48SJeremy L Thompson 11051d102b48SJeremy L Thompson if (*err == 0) { 11061d102b48SJeremy L Thompson *assembledrstr = CeedElemRestriction_count++; 11071d102b48SJeremy L Thompson CeedElemRestriction_n++; 11081d102b48SJeremy L Thompson *assembledvec = CeedVector_count++; 11091d102b48SJeremy L Thompson CeedVector_n++; 11101d102b48SJeremy L Thompson } 11111d102b48SJeremy L Thompson } 11121d102b48SJeremy L Thompson 1113d99fa3c5SJeremy L Thompson #define fCeedOperatorLinearAssembleDiagonal \ 1114d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatorlinearassemblediagonal, CEEDOPERATORLINEARASSEMBLEDIAGONAL) 111580ac2e43SJeremy L Thompson void fCeedOperatorLinearAssembleDiagonal(int *op, int *assembledvec, 1116b7ec98d8SJeremy L Thompson int *rqst, int *err) { 1117b7ec98d8SJeremy L Thompson int createRequest = 1; 1118b7ec98d8SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1119b7ec98d8SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 1120b7ec98d8SJeremy L Thompson createRequest = 0; 1121b7ec98d8SJeremy L Thompson } 1122b7ec98d8SJeremy L Thompson 1123b7ec98d8SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1124b7ec98d8SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max/2 + 1; 1125b7ec98d8SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1126b7ec98d8SJeremy L Thompson } 1127b7ec98d8SJeremy L Thompson 1128b7ec98d8SJeremy L Thompson CeedRequest *rqst_; 1129b7ec98d8SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1130b7ec98d8SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1131b7ec98d8SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1132b7ec98d8SJeremy L Thompson 113380ac2e43SJeremy L Thompson *err = CeedOperatorLinearAssembleDiagonal(CeedOperator_dict[*op], 11342bba3ffaSJeremy L Thompson CeedVector_dict[*assembledvec], rqst_); 1135b7ec98d8SJeremy L Thompson if (*err) return; 1136b7ec98d8SJeremy L Thompson if (createRequest) { 1137b7ec98d8SJeremy L Thompson *rqst = CeedRequest_count++; 1138b7ec98d8SJeremy L Thompson CeedRequest_n++; 1139b7ec98d8SJeremy L Thompson } 1140b7ec98d8SJeremy L Thompson } 1141b7ec98d8SJeremy L Thompson 1142d99fa3c5SJeremy L Thompson #define fCeedOperatorMultigridLevelCreate \ 1143d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatormultigridlevelcreate, CEEDOPERATORMULTIGRIDLEVELCREATE) 1144d99fa3c5SJeremy L Thompson void fCeedOperatorMultigridLevelCreate(int *opFine, int *pMultFine, 1145d99fa3c5SJeremy L Thompson int *rstrCoarse, int *basisCoarse, int *opCoarse, 1146d99fa3c5SJeremy L Thompson int *opProlong, int *opRestrict, int *err) { 1147d99fa3c5SJeremy L Thompson // Operators 1148d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1149d99fa3c5SJeremy L Thompson 1150d99fa3c5SJeremy L Thompson // C interface call 1151d99fa3c5SJeremy L Thompson *err = CeedOperatorMultigridLevelCreate( 1152d99fa3c5SJeremy L Thompson CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], 1153d99fa3c5SJeremy L Thompson CeedElemRestriction_dict[*rstrCoarse], 1154d99fa3c5SJeremy L Thompson CeedBasis_dict[*basisCoarse], 1155d99fa3c5SJeremy L Thompson &opCoarse_, &opProlong_, &opRestrict_); 1156d99fa3c5SJeremy L Thompson 1157d99fa3c5SJeremy L Thompson if (*err) return; 1158d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1159d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1160d99fa3c5SJeremy L Thompson } 1161d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1162d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1163d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1164d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1165d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1166d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1167d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1168d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1169d99fa3c5SJeremy L Thompson } 1170d99fa3c5SJeremy L Thompson 1171d99fa3c5SJeremy L Thompson #define fCeedOperatorMultigridLevelCreateTensorH1 \ 1172d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatormultigridlevelcreatetensorh1, CEEDOPERATORMULTIGRIDLEVELCREATETENSORH1) 1173d99fa3c5SJeremy L Thompson void fCeedOperatorMultigridLevelCreateTensorH1(int *opFine, int *pMultFine, 1174d99fa3c5SJeremy L Thompson int *rstrCoarse, int *basisCoarse, const CeedScalar *interpCtoF, 1175d99fa3c5SJeremy L Thompson int *opCoarse, int *opProlong, int *opRestrict, int *err) { 1176d99fa3c5SJeremy L Thompson // Operators 1177d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1178d99fa3c5SJeremy L Thompson 1179d99fa3c5SJeremy L Thompson // C interface call 1180d99fa3c5SJeremy L Thompson *err = CeedOperatorMultigridLevelCreateTensorH1( 1181d99fa3c5SJeremy L Thompson CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], 1182d99fa3c5SJeremy L Thompson CeedElemRestriction_dict[*rstrCoarse], CeedBasis_dict[*basisCoarse], 1183d99fa3c5SJeremy L Thompson interpCtoF, &opCoarse_, &opProlong_, &opRestrict_); 1184d99fa3c5SJeremy L Thompson 1185d99fa3c5SJeremy L Thompson if (*err) return; 1186d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1187d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1188d99fa3c5SJeremy L Thompson } 1189d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1190d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1191d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1192d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1193d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1194d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1195d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1196d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1197d99fa3c5SJeremy L Thompson } 1198d99fa3c5SJeremy L Thompson 1199d99fa3c5SJeremy L Thompson #define fCeedOperatorMultigridLevelCreateH1 \ 1200d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatormultigridlevelcreateh1, CEEDOPERATORMULTIGRIDLEVELCREATEH1) 1201d99fa3c5SJeremy L Thompson void fCeedOperatorMultigridLevelCreateH1(int *opFine, int *pMultFine, 1202d99fa3c5SJeremy L Thompson int *rstrCoarse, int *basisCoarse, const CeedScalar *interpCtoF, 1203d99fa3c5SJeremy L Thompson int *opCoarse, int *opProlong, int *opRestrict, int *err) { 1204d99fa3c5SJeremy L Thompson // Operators 1205d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1206d99fa3c5SJeremy L Thompson 1207d99fa3c5SJeremy L Thompson // C interface call 1208d99fa3c5SJeremy L Thompson *err = CeedOperatorMultigridLevelCreateH1( 1209d99fa3c5SJeremy L Thompson CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], 1210d99fa3c5SJeremy L Thompson CeedElemRestriction_dict[*rstrCoarse], CeedBasis_dict[*basisCoarse], 1211d99fa3c5SJeremy L Thompson interpCtoF, &opCoarse_, &opProlong_, &opRestrict_); 1212d99fa3c5SJeremy L Thompson 1213d99fa3c5SJeremy L Thompson if (*err) return; 1214d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1215d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1216d99fa3c5SJeremy L Thompson } 1217d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1218d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1219d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1220d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1221d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1222d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1223d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1224d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1225d99fa3c5SJeremy L Thompson } 1226d99fa3c5SJeremy L Thompson 12272ebaca42Sjeremylt #define fCeedOperatorView \ 12282ebaca42Sjeremylt FORTRAN_NAME(ceedoperatorview,CEEDOPERATORVIEW) 12292ebaca42Sjeremylt void fCeedOperatorView(int *op, int *err) { 12302ebaca42Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 12312ebaca42Sjeremylt 12322ebaca42Sjeremylt *err = CeedOperatorView(op_, stdout); 12332ebaca42Sjeremylt } 12342ebaca42Sjeremylt 1235d99fa3c5SJeremy L Thompson #define fCeedOperatorCreateFDMElementInverse \ 1236d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatorcreatefdmelementinverse, CEEDOPERATORCREATEFDMELEMENTINVERSE) 12373bd813ffSjeremylt void fCeedOperatorCreateFDMElementInverse(int *op, int *fdminv, 12383bd813ffSjeremylt int *rqst, int *err) { 12393bd813ffSjeremylt // Operator 12403bd813ffSjeremylt if (CeedOperator_count == CeedOperator_count_max) { 12413bd813ffSjeremylt CeedOperator_count_max += CeedOperator_count_max/2 + 1; 12423bd813ffSjeremylt CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 12433bd813ffSjeremylt } 12443bd813ffSjeremylt CeedOperator *fdminv_ = 12453bd813ffSjeremylt &CeedOperator_dict[CeedOperator_count]; 12463bd813ffSjeremylt 12473bd813ffSjeremylt int createRequest = 1; 12483bd813ffSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 12493bd813ffSjeremylt if (*rqst == -1 || *rqst == -2) { 12503bd813ffSjeremylt createRequest = 0; 12513bd813ffSjeremylt } 12523bd813ffSjeremylt 12533bd813ffSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 12543bd813ffSjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 12553bd813ffSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 12563bd813ffSjeremylt } 12573bd813ffSjeremylt 12583bd813ffSjeremylt CeedRequest *rqst_; 12593bd813ffSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 12603bd813ffSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 12613bd813ffSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 12623bd813ffSjeremylt 12633bd813ffSjeremylt *err = CeedOperatorCreateFDMElementInverse(CeedOperator_dict[*op], 12643bd813ffSjeremylt fdminv_, rqst_); 12653bd813ffSjeremylt if (*err) return; 12663bd813ffSjeremylt if (createRequest) { 12673bd813ffSjeremylt *rqst = CeedRequest_count++; 12683bd813ffSjeremylt CeedRequest_n++; 12693bd813ffSjeremylt } 12703bd813ffSjeremylt 12713bd813ffSjeremylt if (*err == 0) { 12723bd813ffSjeremylt *fdminv = CeedOperator_count++; 12733bd813ffSjeremylt CeedOperator_n++; 12743bd813ffSjeremylt } 12753bd813ffSjeremylt } 12763bd813ffSjeremylt 1277d7b241e6Sjeremylt #define fCeedOperatorApply FORTRAN_NAME(ceedoperatorapply, CEEDOPERATORAPPLY) 1278d7b241e6Sjeremylt void fCeedOperatorApply(int *op, int *ustatevec, 1279d7b241e6Sjeremylt int *resvec, int *rqst, int *err) { 1280e97ff134Sjeremylt CeedVector ustatevec_ = (*ustatevec == FORTRAN_NULL) ? 1281e97ff134Sjeremylt NULL : (*ustatevec == FORTRAN_VECTOR_NONE ? 1282e97ff134Sjeremylt CEED_VECTOR_NONE : CeedVector_dict[*ustatevec]); 1283e97ff134Sjeremylt CeedVector resvec_ = (*resvec == FORTRAN_NULL) ? 1284e97ff134Sjeremylt NULL : (*resvec == FORTRAN_VECTOR_NONE ? 1285e97ff134Sjeremylt CEED_VECTOR_NONE : CeedVector_dict[*resvec]); 1286d7b241e6Sjeremylt 1287d7b241e6Sjeremylt int createRequest = 1; 1288d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1289d7b241e6Sjeremylt if (*rqst == -1 || *rqst == -2) { 1290d7b241e6Sjeremylt createRequest = 0; 1291d7b241e6Sjeremylt } 1292d7b241e6Sjeremylt 1293d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1294d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 1295d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1296d7b241e6Sjeremylt } 1297d7b241e6Sjeremylt 1298d7b241e6Sjeremylt CeedRequest *rqst_; 1299d7b241e6Sjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1300d7b241e6Sjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1301d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1302d7b241e6Sjeremylt 1303d7b241e6Sjeremylt *err = CeedOperatorApply(CeedOperator_dict[*op], 130412bf2c11SJed Brown ustatevec_, resvec_, rqst_); 1305d7b241e6Sjeremylt if (*err) return; 1306d7b241e6Sjeremylt if (createRequest) { 1307d7b241e6Sjeremylt *rqst = CeedRequest_count++; 1308d7b241e6Sjeremylt CeedRequest_n++; 1309d7b241e6Sjeremylt } 1310d7b241e6Sjeremylt } 1311d7b241e6Sjeremylt 1312cae8b89aSjeremylt #define fCeedOperatorApplyAdd FORTRAN_NAME(ceedoperatorapplyadd, CEEDOPERATORAPPLYADD) 1313cae8b89aSjeremylt void fCeedOperatorApplyAdd(int *op, int *ustatevec, 1314cae8b89aSjeremylt int *resvec, int *rqst, int *err) { 1315cae8b89aSjeremylt CeedVector ustatevec_ = *ustatevec == FORTRAN_NULL 1316cae8b89aSjeremylt ? NULL : CeedVector_dict[*ustatevec]; 1317cae8b89aSjeremylt CeedVector resvec_ = *resvec == FORTRAN_NULL 1318cae8b89aSjeremylt ? NULL : CeedVector_dict[*resvec]; 1319cae8b89aSjeremylt 1320cae8b89aSjeremylt int createRequest = 1; 1321cae8b89aSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1322cae8b89aSjeremylt if (*rqst == -1 || *rqst == -2) { 1323cae8b89aSjeremylt createRequest = 0; 1324cae8b89aSjeremylt } 1325cae8b89aSjeremylt 1326cae8b89aSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1327cae8b89aSjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 1328cae8b89aSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1329cae8b89aSjeremylt } 1330cae8b89aSjeremylt 1331cae8b89aSjeremylt CeedRequest *rqst_; 1332cae8b89aSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1333cae8b89aSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1334cae8b89aSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1335cae8b89aSjeremylt 1336cae8b89aSjeremylt *err = CeedOperatorApplyAdd(CeedOperator_dict[*op], 1337cae8b89aSjeremylt ustatevec_, resvec_, rqst_); 1338cae8b89aSjeremylt if (*err) return; 1339cae8b89aSjeremylt if (createRequest) { 1340cae8b89aSjeremylt *rqst = CeedRequest_count++; 1341cae8b89aSjeremylt CeedRequest_n++; 1342cae8b89aSjeremylt } 1343cae8b89aSjeremylt } 1344cae8b89aSjeremylt 1345d7b241e6Sjeremylt #define fCeedOperatorApplyJacobian \ 1346d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorapplyjacobian, CEEDOPERATORAPPLYJACOBIAN) 1347d7b241e6Sjeremylt void fCeedOperatorApplyJacobian(int *op, int *qdatavec, int *ustatevec, 13488795c945Sjeremylt int *dustatevec, int *dresvec, int *rqst, 13498795c945Sjeremylt int *err) { 1350d7b241e6Sjeremylt // TODO Uncomment this when CeedOperatorApplyJacobian is implemented 1351d7b241e6Sjeremylt // *err = CeedOperatorApplyJacobian(CeedOperator_dict[*op], CeedVector_dict[*qdatavec], 1352d7b241e6Sjeremylt // CeedVector_dict[*ustatevec], CeedVector_dict[*dustatevec], 1353d7b241e6Sjeremylt // CeedVector_dict[*dresvec], &CeedRequest_dict[*rqst]); 1354d7b241e6Sjeremylt } 1355d7b241e6Sjeremylt 1356d7b241e6Sjeremylt #define fCeedOperatorDestroy \ 1357d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatordestroy, CEEDOPERATORDESTROY) 1358d7b241e6Sjeremylt void fCeedOperatorDestroy(int *op, int *err) { 1359b22696a8SJeremy L Thompson if (*op == FORTRAN_NULL) return; 1360d7b241e6Sjeremylt *err = CeedOperatorDestroy(&CeedOperator_dict[*op]); 136116595307SJeremy L Thompson if (*err == 0) { 136216595307SJeremy L Thompson *op = FORTRAN_NULL; 1363d7b241e6Sjeremylt CeedOperator_n--; 1364d7b241e6Sjeremylt if (CeedOperator_n == 0) { 1365d7b241e6Sjeremylt *err = CeedFree(&CeedOperator_dict); 1366d7b241e6Sjeremylt CeedOperator_count = 0; 1367d7b241e6Sjeremylt CeedOperator_count_max = 0; 1368d7b241e6Sjeremylt } 1369d7b241e6Sjeremylt } 137016595307SJeremy L Thompson } 1371b00b43a6SJeremy L Thompson 1372b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 1373