1d7b241e6Sjeremylt // Fortran interface 2d7b241e6Sjeremylt #include <ceed.h> 3d7b241e6Sjeremylt #include <ceed-impl.h> 4d863ab9bSjeremylt #include <ceed-backend.h> 5d7b241e6Sjeremylt #include <ceed-fortran-name.h> 6d7b241e6Sjeremylt #include <stdlib.h> 7d7b241e6Sjeremylt #include <string.h> 8d7b241e6Sjeremylt 9d7b241e6Sjeremylt #define FORTRAN_REQUEST_IMMEDIATE -1 10d7b241e6Sjeremylt #define FORTRAN_REQUEST_ORDERED -2 11d7b241e6Sjeremylt #define FORTRAN_NULL -3 12783c99b3SValeria Barra #define FORTRAN_BASIS_COLLOCATED -1 13d7b241e6Sjeremylt #define FORTRAN_VECTOR_ACTIVE -1 14d7b241e6Sjeremylt #define FORTRAN_VECTOR_NONE -2 15d7b241e6Sjeremylt 16d7b241e6Sjeremylt static Ceed *Ceed_dict = NULL; 17d7b241e6Sjeremylt static int Ceed_count = 0; 18d7b241e6Sjeremylt static int Ceed_n = 0; 19d7b241e6Sjeremylt static int Ceed_count_max = 0; 20d7b241e6Sjeremylt 21c2ae719dSJed Brown // This test should actually be for the gfortran version, but we don't currently 22c2ae719dSJed Brown // have a configure system to determine that (TODO). At present, this will use 23c2ae719dSJed Brown // the smaller integer when run with clang+gfortran=8, for example. (That is 24c2ae719dSJed Brown // sketchy, but will likely work for users that don't have huge character 25c2ae719dSJed Brown // strings.) 26c2ae719dSJed Brown #if __GNUC__ >= 8 27c2ae719dSJed Brown typedef size_t fortran_charlen_t; 28c2ae719dSJed Brown #else 29c2ae719dSJed Brown typedef int fortran_charlen_t; 30c2ae719dSJed Brown #endif 31c2ae719dSJed Brown 32c2ae719dSJed Brown #define Splice(a, b) a ## b 33c2ae719dSJed Brown 34c2ae719dSJed Brown // Fortran strings are generally unterminated and the length is passed as an 35c2ae719dSJed Brown // extra argument after all the normal arguments. Some compilers (I only know 36c2ae719dSJed Brown // of Windows) place the length argument immediately after the string parameter 37c2ae719dSJed Brown // (TODO). 38c2ae719dSJed Brown // 39c2ae719dSJed Brown // We can't just NULL-terminate the string in-place because that could overwrite 40c2ae719dSJed Brown // other strings or attempt to write to read-only memory. This macro allocates 41c2ae719dSJed Brown // a string to hold the null-terminated version of the string that C expects. 42c2ae719dSJed Brown #define FIX_STRING(stringname) \ 43c2ae719dSJed Brown char Splice(stringname, _c)[1024]; \ 44c2ae719dSJed Brown if (Splice(stringname, _len) > 1023) \ 45c2ae719dSJed Brown CeedError(NULL, 1, "Fortran string length too long %zd", (size_t)Splice(stringname, _len)); \ 46c2ae719dSJed Brown strncpy(Splice(stringname, _c), stringname, Splice(stringname, _len)); \ 47c2ae719dSJed Brown Splice(stringname, _c)[Splice(stringname, _len)] = 0; \ 48c2ae719dSJed Brown 49d7b241e6Sjeremylt #define fCeedInit FORTRAN_NAME(ceedinit,CEEDINIT) 50c2ae719dSJed Brown void fCeedInit(const char *resource, int *ceed, int *err, 51c2ae719dSJed Brown fortran_charlen_t resource_len) { 52c2ae719dSJed Brown FIX_STRING(resource); 53d7b241e6Sjeremylt if (Ceed_count == Ceed_count_max) { 54d7b241e6Sjeremylt Ceed_count_max += Ceed_count_max/2 + 1; 55d7b241e6Sjeremylt CeedRealloc(Ceed_count_max, &Ceed_dict); 56d7b241e6Sjeremylt } 57d7b241e6Sjeremylt 58d7b241e6Sjeremylt Ceed *ceed_ = &Ceed_dict[Ceed_count]; 59c2ae719dSJed Brown *err = CeedInit(resource_c, ceed_); 60d7b241e6Sjeremylt 61d7b241e6Sjeremylt if (*err == 0) { 62d7b241e6Sjeremylt *ceed = Ceed_count++; 63d7b241e6Sjeremylt Ceed_n++; 64d7b241e6Sjeremylt } 65d7b241e6Sjeremylt } 66d7b241e6Sjeremylt 67c907536fSjeremylt #define fCeedGetPreferredMemType \ 68c907536fSjeremylt FORTRAN_NAME(ceedgetpreferredmemtype,CEEDGETPREFERREDMEMTYPE) 69c907536fSjeremylt void fCeedGetPreferredMemType(int *ceed, int *type, int *err) { 70c907536fSjeremylt *err = CeedGetPreferredMemType(Ceed_dict[*ceed], (CeedMemType *)type); 71c907536fSjeremylt } 72c907536fSjeremylt 73d7b241e6Sjeremylt #define fCeedDestroy FORTRAN_NAME(ceeddestroy,CEEDDESTROY) 74d7b241e6Sjeremylt void fCeedDestroy(int *ceed, int *err) { 75d7b241e6Sjeremylt *err = CeedDestroy(&Ceed_dict[*ceed]); 76d7b241e6Sjeremylt 77d7b241e6Sjeremylt if (*err == 0) { 78d7b241e6Sjeremylt Ceed_n--; 79d7b241e6Sjeremylt if (Ceed_n == 0) { 80d7b241e6Sjeremylt CeedFree(&Ceed_dict); 81d7b241e6Sjeremylt Ceed_count = 0; 82d7b241e6Sjeremylt Ceed_count_max = 0; 83d7b241e6Sjeremylt } 84d7b241e6Sjeremylt } 85d7b241e6Sjeremylt } 86d7b241e6Sjeremylt 87d7b241e6Sjeremylt static CeedVector *CeedVector_dict = NULL; 88d7b241e6Sjeremylt static int CeedVector_count = 0; 89d7b241e6Sjeremylt static int CeedVector_n = 0; 90d7b241e6Sjeremylt static int CeedVector_count_max = 0; 91d7b241e6Sjeremylt 92d7b241e6Sjeremylt #define fCeedVectorCreate FORTRAN_NAME(ceedvectorcreate,CEEDVECTORCREATE) 93d7b241e6Sjeremylt void fCeedVectorCreate(int *ceed, int *length, int *vec, int *err) { 94d7b241e6Sjeremylt if (CeedVector_count == CeedVector_count_max) { 95d7b241e6Sjeremylt CeedVector_count_max += CeedVector_count_max/2 + 1; 96d7b241e6Sjeremylt CeedRealloc(CeedVector_count_max, &CeedVector_dict); 97d7b241e6Sjeremylt } 98d7b241e6Sjeremylt 99d7b241e6Sjeremylt CeedVector *vec_ = &CeedVector_dict[CeedVector_count]; 100d7b241e6Sjeremylt *err = CeedVectorCreate(Ceed_dict[*ceed], *length, vec_); 101d7b241e6Sjeremylt 102d7b241e6Sjeremylt if (*err == 0) { 103d7b241e6Sjeremylt *vec = CeedVector_count++; 104d7b241e6Sjeremylt CeedVector_n++; 105d7b241e6Sjeremylt } 106d7b241e6Sjeremylt } 107d7b241e6Sjeremylt 108d7b241e6Sjeremylt #define fCeedVectorSetArray FORTRAN_NAME(ceedvectorsetarray,CEEDVECTORSETARRAY) 109d7b241e6Sjeremylt void fCeedVectorSetArray(int *vec, int *memtype, int *copymode, 110*c8b9fe72Sjeremylt CeedScalar *array, int64_t *offset, int *err) { 111*c8b9fe72Sjeremylt *err = CeedVectorSetArray(CeedVector_dict[*vec], *memtype, *copymode, 112*c8b9fe72Sjeremylt (CeedScalar *)(array + *offset)); 113d7b241e6Sjeremylt } 114d7b241e6Sjeremylt 11554540941SJeremy L Thompson #define fCeedVectorSyncArray FORTRAN_NAME(ceedvectorsyncarray,CEEDVECTORSYNCARRAY) 11654540941SJeremy L Thompson void fCeedVectorSyncArray(int *vec, int *memtype, int *err) { 11754540941SJeremy L Thompson *err = CeedVectorSyncArray(CeedVector_dict[*vec], *memtype); 11854540941SJeremy L Thompson } 11954540941SJeremy L Thompson 120d7b241e6Sjeremylt #define fCeedVectorSetValue FORTRAN_NAME(ceedvectorsetvalue,CEEDVECTORSETVALUE) 121d7b241e6Sjeremylt void fCeedVectorSetValue(int *vec, CeedScalar *value, int *err) { 122d7b241e6Sjeremylt *err = CeedVectorSetValue(CeedVector_dict[*vec], *value); 123d7b241e6Sjeremylt } 124d7b241e6Sjeremylt 125d7b241e6Sjeremylt #define fCeedVectorGetArray FORTRAN_NAME(ceedvectorgetarray,CEEDVECTORGETARRAY) 126a8de75f0Sjeremylt void fCeedVectorGetArray(int *vec, int *memtype, CeedScalar *array, 127a8de75f0Sjeremylt int64_t *offset, 128cbfd6a90SThilina Rathnayake int *err) { 129d7b241e6Sjeremylt CeedScalar *b; 130d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 131d7b241e6Sjeremylt *err = CeedVectorGetArray(vec_, *memtype, &b); 132cbfd6a90SThilina Rathnayake *offset = b - array; 133d7b241e6Sjeremylt } 134d7b241e6Sjeremylt 135d7b241e6Sjeremylt #define fCeedVectorGetArrayRead \ 136d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorgetarrayread,CEEDVECTORGETARRAYREAD) 137d7b241e6Sjeremylt void fCeedVectorGetArrayRead(int *vec, int *memtype, CeedScalar *array, 1383334379fSJed Brown int64_t *offset, int *err) { 139d7b241e6Sjeremylt const CeedScalar *b; 140d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 141cbfd6a90SThilina Rathnayake *err = CeedVectorGetArrayRead(vec_, *memtype, &b); 142cbfd6a90SThilina Rathnayake *offset = b - array; 143d7b241e6Sjeremylt } 144d7b241e6Sjeremylt 145d7b241e6Sjeremylt #define fCeedVectorRestoreArray \ 146d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearray,CEEDVECTORRESTOREARRAY) 1476456524eSJed Brown void fCeedVectorRestoreArray(int *vec, CeedScalar *array, 1486456524eSJed Brown int64_t *offset, int *err) { 149d7b241e6Sjeremylt *err = CeedVectorRestoreArray(CeedVector_dict[*vec], &array); 1506456524eSJed Brown *offset = 0; 151d7b241e6Sjeremylt } 152d7b241e6Sjeremylt 153d7b241e6Sjeremylt #define fCeedVectorRestoreArrayRead \ 154d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearrayread,CEEDVECTORRESTOREARRAYREAD) 1556456524eSJed Brown void fCeedVectorRestoreArrayRead(int *vec, const CeedScalar *array, 1566456524eSJed Brown int64_t *offset, int *err) { 157d7b241e6Sjeremylt *err = CeedVectorRestoreArrayRead(CeedVector_dict[*vec], &array); 1586456524eSJed Brown *offset = 0; 159d7b241e6Sjeremylt } 160d7b241e6Sjeremylt 161d7b241e6Sjeremylt #define fCeedVectorView FORTRAN_NAME(ceedvectorview,CEEDVECTORVIEW) 162d7b241e6Sjeremylt void fCeedVectorView(int *vec, int *err) { 163d7b241e6Sjeremylt *err = CeedVectorView(CeedVector_dict[*vec], "%12.8f", stdout); 164d7b241e6Sjeremylt } 165d7b241e6Sjeremylt 166d7b241e6Sjeremylt #define fCeedVectorDestroy FORTRAN_NAME(ceedvectordestroy,CEEDVECTORDESTROY) 167d7b241e6Sjeremylt void fCeedVectorDestroy(int *vec, int *err) { 168d7b241e6Sjeremylt *err = CeedVectorDestroy(&CeedVector_dict[*vec]); 169d7b241e6Sjeremylt 170d7b241e6Sjeremylt if (*err == 0) { 171d7b241e6Sjeremylt CeedVector_n--; 172d7b241e6Sjeremylt if (CeedVector_n == 0) { 173d7b241e6Sjeremylt CeedFree(&CeedVector_dict); 174d7b241e6Sjeremylt CeedVector_count = 0; 175d7b241e6Sjeremylt CeedVector_count_max = 0; 176d7b241e6Sjeremylt } 177d7b241e6Sjeremylt } 178d7b241e6Sjeremylt } 179d7b241e6Sjeremylt 180d7b241e6Sjeremylt static CeedElemRestriction *CeedElemRestriction_dict = NULL; 181d7b241e6Sjeremylt static int CeedElemRestriction_count = 0; 182d7b241e6Sjeremylt static int CeedElemRestriction_n = 0; 183d7b241e6Sjeremylt static int CeedElemRestriction_count_max = 0; 184d7b241e6Sjeremylt 185d7b241e6Sjeremylt #define fCeedElemRestrictionCreate \ 186d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreate, CEEDELEMRESTRICTIONCREATE) 187d7b241e6Sjeremylt void fCeedElemRestrictionCreate(int *ceed, int *nelements, 188d7b241e6Sjeremylt int *esize, int *ndof, int *ncomp, int *memtype, int *copymode, 189d7b241e6Sjeremylt const int *indices, int *elemrestriction, int *err) { 190d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 191d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 192d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 193d7b241e6Sjeremylt } 194d7b241e6Sjeremylt 195d7b241e6Sjeremylt const int *indices_ = indices; 196d7b241e6Sjeremylt 197d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 198d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 199d7b241e6Sjeremylt *err = CeedElemRestrictionCreate(Ceed_dict[*ceed], *nelements, *esize, *ndof, 200d7b241e6Sjeremylt *ncomp, 201d7b241e6Sjeremylt *memtype, *copymode, indices_, elemrestriction_); 202d7b241e6Sjeremylt 203d7b241e6Sjeremylt if (*err == 0) { 204d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 205d7b241e6Sjeremylt CeedElemRestriction_n++; 206d7b241e6Sjeremylt } 207d7b241e6Sjeremylt } 208d7b241e6Sjeremylt 209d7b241e6Sjeremylt #define fCeedElemRestrictionCreateIdentity \ 210d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateidentity, CEEDELEMRESTRICTIONCREATEIDENTITY) 211d7b241e6Sjeremylt void fCeedElemRestrictionCreateIdentity(int *ceed, int *nelements, 212d7b241e6Sjeremylt int *esize, int *ndof, int *ncomp, 213d7b241e6Sjeremylt int *elemrestriction, int *err) { 214d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 215d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 216d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 217d7b241e6Sjeremylt } 218d7b241e6Sjeremylt 219d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 220d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 2214b8bea3bSJed Brown *err = CeedElemRestrictionCreateIdentity(Ceed_dict[*ceed], *nelements, *esize, 2224b8bea3bSJed Brown *ndof, 223d7b241e6Sjeremylt *ncomp, elemrestriction_); 224d7b241e6Sjeremylt 225d7b241e6Sjeremylt if (*err == 0) { 226d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 227d7b241e6Sjeremylt CeedElemRestriction_n++; 228d7b241e6Sjeremylt } 229d7b241e6Sjeremylt } 230d7b241e6Sjeremylt 231d7b241e6Sjeremylt #define fCeedElemRestrictionCreateBlocked \ 232d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateblocked,CEEDELEMRESTRICTIONCREATEBLOCKED) 233d7b241e6Sjeremylt void fCeedElemRestrictionCreateBlocked(int *ceed, int *nelements, 234d7b241e6Sjeremylt int *esize, int *blocksize, int *ndof, int *ncomp, 235d7b241e6Sjeremylt int *mtype, int *cmode, 236d7b241e6Sjeremylt int *blkindices, int *elemrestriction, int *err) { 237d7b241e6Sjeremylt 238d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 239d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 240d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 241d7b241e6Sjeremylt } 242d7b241e6Sjeremylt 243d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 244d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 245d7b241e6Sjeremylt *err = CeedElemRestrictionCreateBlocked(Ceed_dict[*ceed], *nelements, *esize, 246d7b241e6Sjeremylt *blocksize, *ndof, *ncomp, *mtype, *cmode, blkindices, 247d7b241e6Sjeremylt elemrestriction_); 248d7b241e6Sjeremylt 249d7b241e6Sjeremylt if (*err == 0) { 250d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 251d7b241e6Sjeremylt CeedElemRestriction_n++; 252d7b241e6Sjeremylt } 253d7b241e6Sjeremylt } 254d7b241e6Sjeremylt 255d7b241e6Sjeremylt static CeedRequest *CeedRequest_dict = NULL; 256d7b241e6Sjeremylt static int CeedRequest_count = 0; 257d7b241e6Sjeremylt static int CeedRequest_n = 0; 258d7b241e6Sjeremylt static int CeedRequest_count_max = 0; 259d7b241e6Sjeremylt 260d7b241e6Sjeremylt #define fCeedElemRestrictionApply \ 261d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictionapply,CEEDELEMRESTRICTIONAPPLY) 262d7b241e6Sjeremylt void fCeedElemRestrictionApply(int *elemr, int *tmode, int *lmode, 263d7b241e6Sjeremylt int *uvec, int *ruvec, int *rqst, int *err) { 264d7b241e6Sjeremylt int createRequest = 1; 265d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 266d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) 267d7b241e6Sjeremylt createRequest = 0; 268d7b241e6Sjeremylt 269d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 270d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 271d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 272d7b241e6Sjeremylt } 273d7b241e6Sjeremylt 274d7b241e6Sjeremylt CeedRequest *rqst_; 275d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 276d7b241e6Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED ) rqst_ = CEED_REQUEST_ORDERED; 277d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 278d7b241e6Sjeremylt 279d7b241e6Sjeremylt *err = CeedElemRestrictionApply(CeedElemRestriction_dict[*elemr], *tmode, 280d7b241e6Sjeremylt *lmode, CeedVector_dict[*uvec], CeedVector_dict[*ruvec], rqst_); 281d7b241e6Sjeremylt 282d7b241e6Sjeremylt if (*err == 0 && createRequest) { 283d7b241e6Sjeremylt *rqst = CeedRequest_count++; 284d7b241e6Sjeremylt CeedRequest_n++; 285d7b241e6Sjeremylt } 286d7b241e6Sjeremylt } 287d7b241e6Sjeremylt 288d7b241e6Sjeremylt #define fCeedRequestWait FORTRAN_NAME(ceedrequestwait, CEEDREQUESTWAIT) 289d7b241e6Sjeremylt void fCeedRequestWait(int *rqst, int *err) { 290d7b241e6Sjeremylt // TODO Uncomment this once CeedRequestWait is implemented 291d7b241e6Sjeremylt //*err = CeedRequestWait(&CeedRequest_dict[*rqst]); 292d7b241e6Sjeremylt 293d7b241e6Sjeremylt if (*err == 0) { 294d7b241e6Sjeremylt CeedRequest_n--; 295d7b241e6Sjeremylt if (CeedRequest_n == 0) { 296d7b241e6Sjeremylt CeedFree(&CeedRequest_dict); 297d7b241e6Sjeremylt CeedRequest_count = 0; 298d7b241e6Sjeremylt CeedRequest_count_max = 0; 299d7b241e6Sjeremylt } 300d7b241e6Sjeremylt } 301d7b241e6Sjeremylt } 302d7b241e6Sjeremylt 303d7b241e6Sjeremylt #define fCeedElemRestrictionDestroy \ 304d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictiondestroy,CEEDELEMRESTRICTIONDESTROY) 305d7b241e6Sjeremylt void fCeedElemRestrictionDestroy(int *elem, int *err) { 306d7b241e6Sjeremylt *err = CeedElemRestrictionDestroy(&CeedElemRestriction_dict[*elem]); 307d7b241e6Sjeremylt 308d7b241e6Sjeremylt if (*err == 0) { 309d7b241e6Sjeremylt CeedElemRestriction_n--; 310d7b241e6Sjeremylt if (CeedElemRestriction_n == 0) { 311d7b241e6Sjeremylt CeedFree(&CeedElemRestriction_dict); 312d7b241e6Sjeremylt CeedElemRestriction_count = 0; 313d7b241e6Sjeremylt CeedElemRestriction_count_max = 0; 314d7b241e6Sjeremylt } 315d7b241e6Sjeremylt } 316d7b241e6Sjeremylt } 317d7b241e6Sjeremylt 318d7b241e6Sjeremylt static CeedBasis *CeedBasis_dict = NULL; 319d7b241e6Sjeremylt static int CeedBasis_count = 0; 320d7b241e6Sjeremylt static int CeedBasis_n = 0; 321d7b241e6Sjeremylt static int CeedBasis_count_max = 0; 322d7b241e6Sjeremylt 323d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1Lagrange \ 324d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1lagrange, CEEDBASISCREATETENSORH1LAGRANGE) 325d7b241e6Sjeremylt void fCeedBasisCreateTensorH1Lagrange(int *ceed, int *dim, 326a8de75f0Sjeremylt int *ncomp, int *P, int *Q, int *quadmode, int *basis, 327d7b241e6Sjeremylt int *err) { 328d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 329d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 330d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 331d7b241e6Sjeremylt } 332d7b241e6Sjeremylt 333a8de75f0Sjeremylt *err = CeedBasisCreateTensorH1Lagrange(Ceed_dict[*ceed], *dim, *ncomp, *P, *Q, 334d7b241e6Sjeremylt *quadmode, &CeedBasis_dict[CeedBasis_count]); 335d7b241e6Sjeremylt 336d7b241e6Sjeremylt if (*err == 0) { 337d7b241e6Sjeremylt *basis = CeedBasis_count++; 338d7b241e6Sjeremylt CeedBasis_n++; 339d7b241e6Sjeremylt } 340d7b241e6Sjeremylt } 341d7b241e6Sjeremylt 342d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1 \ 343d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1, CEEDBASISCREATETENSORH1) 344a8de75f0Sjeremylt void fCeedBasisCreateTensorH1(int *ceed, int *dim, int *ncomp, int *P1d, 345d7b241e6Sjeremylt int *Q1d, const CeedScalar *interp1d, const CeedScalar *grad1d, 346d7b241e6Sjeremylt const CeedScalar *qref1d, const CeedScalar *qweight1d, int *basis, int *err) { 347d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 348d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 349d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 350d7b241e6Sjeremylt } 351d7b241e6Sjeremylt 352a8de75f0Sjeremylt *err = CeedBasisCreateTensorH1(Ceed_dict[*ceed], *dim, *ncomp, *P1d, *Q1d, 353d7b241e6Sjeremylt interp1d, grad1d, 354d7b241e6Sjeremylt qref1d, qweight1d, &CeedBasis_dict[CeedBasis_count]); 355d7b241e6Sjeremylt 356d7b241e6Sjeremylt if (*err == 0) { 357d7b241e6Sjeremylt *basis = CeedBasis_count++; 358d7b241e6Sjeremylt CeedBasis_n++; 359d7b241e6Sjeremylt } 360d7b241e6Sjeremylt } 361d7b241e6Sjeremylt 362a8de75f0Sjeremylt #define fCeedBasisCreateH1 \ 363a8de75f0Sjeremylt FORTRAN_NAME(ceedbasiscreateh1, CEEDBASISCREATEH1) 364a8de75f0Sjeremylt void fCeedBasisCreateH1(int *ceed, int *topo, int *ncomp, int *ndof, 365a8de75f0Sjeremylt int *nqpts, const CeedScalar *interp, const CeedScalar *grad, 366a8de75f0Sjeremylt const CeedScalar *qref, const CeedScalar *qweight, int *basis, int *err) { 367a8de75f0Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 368a8de75f0Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 369a8de75f0Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 370a8de75f0Sjeremylt } 371a8de75f0Sjeremylt 372a8de75f0Sjeremylt *err = CeedBasisCreateH1(Ceed_dict[*ceed], *topo, *ncomp, 373a8de75f0Sjeremylt *ndof, *nqpts, interp, grad, 374a8de75f0Sjeremylt qref, qweight, &CeedBasis_dict[CeedBasis_count]); 375a8de75f0Sjeremylt 376a8de75f0Sjeremylt if (*err == 0) { 377a8de75f0Sjeremylt *basis = CeedBasis_count++; 378a8de75f0Sjeremylt CeedBasis_n++; 379a8de75f0Sjeremylt } 380a8de75f0Sjeremylt } 381a8de75f0Sjeremylt 382d7b241e6Sjeremylt #define fCeedBasisView FORTRAN_NAME(ceedbasisview, CEEDBASISVIEW) 383d7b241e6Sjeremylt void fCeedBasisView(int *basis, int *err) { 384d7b241e6Sjeremylt *err = CeedBasisView(CeedBasis_dict[*basis], stdout); 385d7b241e6Sjeremylt } 386d7b241e6Sjeremylt 387d7b241e6Sjeremylt #define fCeedQRFactorization \ 388d7b241e6Sjeremylt FORTRAN_NAME(ceedqrfactorization, CEEDQRFACTORIZATION) 389d7b241e6Sjeremylt void fCeedQRFactorization(CeedScalar *mat, CeedScalar *tau, int *m, int *n, 390d7b241e6Sjeremylt int *err) { 391d7b241e6Sjeremylt *err = CeedQRFactorization(mat, tau, *m, *n); 392d7b241e6Sjeremylt } 393d7b241e6Sjeremylt 394783c99b3SValeria Barra #define fCeedBasisGetCollocatedGrad \ 395783c99b3SValeria Barra FORTRAN_NAME(ceedbasisgetcollocatedgrad, CEEDBASISGETCOLLOCATEDGRAD) 396783c99b3SValeria Barra void fCeedBasisGetCollocatedGrad(int *basis, CeedScalar *colograd1d, 397d7b241e6Sjeremylt int *err) { 398783c99b3SValeria Barra *err = CeedBasisGetCollocatedGrad(CeedBasis_dict[*basis], colograd1d); 399d7b241e6Sjeremylt } 400d7b241e6Sjeremylt 401d7b241e6Sjeremylt #define fCeedBasisApply FORTRAN_NAME(ceedbasisapply, CEEDBASISAPPLY) 402d7b241e6Sjeremylt void fCeedBasisApply(int *basis, int *nelem, int *tmode, int *emode, 403aedaa0e5Sjeremylt int *u, int *v, int *err) { 404aedaa0e5Sjeremylt *err = CeedBasisApply(CeedBasis_dict[*basis], *nelem, *tmode, *emode, 405aedaa0e5Sjeremylt *u==FORTRAN_NULL?NULL:CeedVector_dict[*u], 406aedaa0e5Sjeremylt CeedVector_dict[*v]); 407d7b241e6Sjeremylt } 408d7b241e6Sjeremylt 409d7b241e6Sjeremylt #define fCeedBasisGetNumNodes \ 410d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumnodes, CEEDBASISGETNUMNODES) 411d7b241e6Sjeremylt void fCeedBasisGetNumNodes(int *basis, int *P, int *err) { 412d7b241e6Sjeremylt *err = CeedBasisGetNumNodes(CeedBasis_dict[*basis], P); 413d7b241e6Sjeremylt } 414d7b241e6Sjeremylt 415d7b241e6Sjeremylt #define fCeedBasisGetNumQuadraturePoints \ 416d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumquadraturepoints, CEEDBASISGETNUMQUADRATUREPOINTS) 417d7b241e6Sjeremylt void fCeedBasisGetNumQuadraturePoints(int *basis, int *Q, int *err) { 418d7b241e6Sjeremylt *err = CeedBasisGetNumQuadraturePoints(CeedBasis_dict[*basis], Q); 419d7b241e6Sjeremylt } 420d7b241e6Sjeremylt 421d7b241e6Sjeremylt #define fCeedBasisDestroy FORTRAN_NAME(ceedbasisdestroy,CEEDBASISDESTROY) 422d7b241e6Sjeremylt void fCeedBasisDestroy(int *basis, int *err) { 423d7b241e6Sjeremylt *err = CeedBasisDestroy(&CeedBasis_dict[*basis]); 424d7b241e6Sjeremylt 425d7b241e6Sjeremylt if (*err == 0) { 426d7b241e6Sjeremylt CeedBasis_n--; 427d7b241e6Sjeremylt if (CeedBasis_n == 0) { 428d7b241e6Sjeremylt CeedFree(&CeedBasis_dict); 429d7b241e6Sjeremylt CeedBasis_count = 0; 430d7b241e6Sjeremylt CeedBasis_count_max = 0; 431d7b241e6Sjeremylt } 432d7b241e6Sjeremylt } 433d7b241e6Sjeremylt } 434d7b241e6Sjeremylt 435d7b241e6Sjeremylt #define fCeedGaussQuadrature FORTRAN_NAME(ceedgaussquadrature, CEEDGAUSSQUADRATURE) 436d7b241e6Sjeremylt void fCeedGaussQuadrature(int *Q, CeedScalar *qref1d, CeedScalar *qweight1d, 437d7b241e6Sjeremylt int *err) { 438d7b241e6Sjeremylt *err = CeedGaussQuadrature(*Q, qref1d, qweight1d); 439d7b241e6Sjeremylt } 440d7b241e6Sjeremylt 441d7b241e6Sjeremylt #define fCeedLobattoQuadrature \ 442d7b241e6Sjeremylt FORTRAN_NAME(ceedlobattoquadrature, CEEDLOBATTOQUADRATURE) 443d7b241e6Sjeremylt void fCeedLobattoQuadrature(int *Q, CeedScalar *qref1d, CeedScalar *qweight1d, 444d7b241e6Sjeremylt int *err) { 445d7b241e6Sjeremylt *err = CeedLobattoQuadrature(*Q, qref1d, qweight1d); 446d7b241e6Sjeremylt } 447d7b241e6Sjeremylt 448d7b241e6Sjeremylt static CeedQFunction *CeedQFunction_dict = NULL; 449d7b241e6Sjeremylt static int CeedQFunction_count = 0; 450d7b241e6Sjeremylt static int CeedQFunction_n = 0; 451d7b241e6Sjeremylt static int CeedQFunction_count_max = 0; 452d7b241e6Sjeremylt 453d7b241e6Sjeremylt static int CeedQFunctionFortranStub(void *ctx, int nq, 454d7b241e6Sjeremylt const CeedScalar *const *u, CeedScalar *const *v) { 455418fb8c2Sjeremylt fContext *fctx = ctx; 456d7b241e6Sjeremylt int ierr; 457d7b241e6Sjeremylt 4581e35832bSjeremylt CeedScalar *ctx_ = (CeedScalar *) fctx->innerctx; 4591e35832bSjeremylt fctx->f((void *)ctx_,&nq,u[0],u[1],u[2],u[3],u[4],u[5],u[6], 460418fb8c2Sjeremylt u[7],u[8],u[9],u[10],u[11],u[12],u[13],u[14],u[15], 461418fb8c2Sjeremylt v[0],v[1],v[2],v[3],v[4],v[5],v[6],v[7],v[8],v[9], 462418fb8c2Sjeremylt v[10],v[11],v[12],v[13],v[14],v[15],&ierr); 463d7b241e6Sjeremylt return ierr; 464d7b241e6Sjeremylt } 465d7b241e6Sjeremylt 466d7b241e6Sjeremylt #define fCeedQFunctionCreateInterior \ 467d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctioncreateinterior, CEEDQFUNCTIONCREATEINTERIOR) 468d7b241e6Sjeremylt void fCeedQFunctionCreateInterior(int *ceed, int *vlength, 469d7b241e6Sjeremylt void (*f)(void *ctx, int *nq, 470418fb8c2Sjeremylt const CeedScalar *u,const CeedScalar *u1, 471418fb8c2Sjeremylt const CeedScalar *u2,const CeedScalar *u3, 472418fb8c2Sjeremylt const CeedScalar *u4,const CeedScalar *u5, 473418fb8c2Sjeremylt const CeedScalar *u6,const CeedScalar *u7, 474418fb8c2Sjeremylt const CeedScalar *u8,const CeedScalar *u9, 475418fb8c2Sjeremylt const CeedScalar *u10,const CeedScalar *u11, 476418fb8c2Sjeremylt const CeedScalar *u12,const CeedScalar *u13, 477418fb8c2Sjeremylt const CeedScalar *u14,const CeedScalar *u15, 478418fb8c2Sjeremylt CeedScalar *v,CeedScalar *v1,CeedScalar *v2, 479418fb8c2Sjeremylt CeedScalar *v3,CeedScalar *v4,CeedScalar *v5, 480418fb8c2Sjeremylt CeedScalar *v6,CeedScalar *v7,CeedScalar *v8, 481418fb8c2Sjeremylt CeedScalar *v9,CeedScalar *v10,CeedScalar *v11, 482418fb8c2Sjeremylt CeedScalar *v12,CeedScalar *v13,CeedScalar *v14, 483418fb8c2Sjeremylt CeedScalar *v15,int *err), 484c2ae719dSJed Brown const char *focca, int *qf, int *err, 485c2ae719dSJed Brown fortran_charlen_t focca_len) { 486c2ae719dSJed Brown FIX_STRING(focca); 487d7b241e6Sjeremylt if (CeedQFunction_count == CeedQFunction_count_max) { 488d7b241e6Sjeremylt CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 489d7b241e6Sjeremylt CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 490d7b241e6Sjeremylt } 491d7b241e6Sjeremylt 492d7b241e6Sjeremylt CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 493d7b241e6Sjeremylt *err = CeedQFunctionCreateInterior(Ceed_dict[*ceed], *vlength, 494c2ae719dSJed Brown CeedQFunctionFortranStub, focca_c, qf_); 495d7b241e6Sjeremylt 496d7b241e6Sjeremylt if (*err == 0) { 497d7b241e6Sjeremylt *qf = CeedQFunction_count++; 498d7b241e6Sjeremylt CeedQFunction_n++; 499418fb8c2Sjeremylt 500d7b241e6Sjeremylt } 501d7b241e6Sjeremylt 502418fb8c2Sjeremylt fContext *fctx; 503d7b241e6Sjeremylt *err = CeedMalloc(1, &fctx); 504d7b241e6Sjeremylt if (*err) return; 505418fb8c2Sjeremylt fctx->f = f; fctx->innerctx = NULL; fctx->innerctxsize = 0; 506d7b241e6Sjeremylt 507418fb8c2Sjeremylt *err = CeedQFunctionSetContext(*qf_, fctx, sizeof(fContext)); 508d7b241e6Sjeremylt 509418fb8c2Sjeremylt (*qf_)->fortranstatus = true; 510d7b241e6Sjeremylt } 511d7b241e6Sjeremylt 512d7b241e6Sjeremylt #define fCeedQFunctionAddInput \ 513d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionaddinput,CEEDQFUNCTIONADDINPUT) 514d7b241e6Sjeremylt void fCeedQFunctionAddInput(int *qf, const char *fieldname, 515c2ae719dSJed Brown CeedInt *ncomp, CeedEvalMode *emode, int *err, 516c2ae719dSJed Brown fortran_charlen_t fieldname_len) { 517c2ae719dSJed Brown FIX_STRING(fieldname); 518d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 519d7b241e6Sjeremylt 520c2ae719dSJed Brown *err = CeedQFunctionAddInput(qf_, fieldname_c, *ncomp, *emode); 521d7b241e6Sjeremylt } 522d7b241e6Sjeremylt 523d7b241e6Sjeremylt #define fCeedQFunctionAddOutput \ 524d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionaddoutput,CEEDQFUNCTIONADDOUTPUT) 525d7b241e6Sjeremylt void fCeedQFunctionAddOutput(int *qf, const char *fieldname, 526c2ae719dSJed Brown CeedInt *ncomp, CeedEvalMode *emode, int *err, 527c2ae719dSJed Brown fortran_charlen_t fieldname_len) { 528c2ae719dSJed Brown FIX_STRING(fieldname); 529d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 530d7b241e6Sjeremylt 531c2ae719dSJed Brown *err = CeedQFunctionAddOutput(qf_, fieldname_c, *ncomp, *emode); 532d7b241e6Sjeremylt } 533d7b241e6Sjeremylt 534418fb8c2Sjeremylt #define fCeedQFunctionSetContext \ 535418fb8c2Sjeremylt FORTRAN_NAME(ceedqfunctionsetcontext,CEEDQFUNCTIONSETCONTEXT) 536418fb8c2Sjeremylt void fCeedQFunctionSetContext(int *qf, CeedScalar *ctx, CeedInt *n, int *err) { 537418fb8c2Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 538418fb8c2Sjeremylt 539418fb8c2Sjeremylt fContext *fctx = qf_->ctx; 540418fb8c2Sjeremylt fctx->innerctx = ctx; 5411e35832bSjeremylt fctx->innerctxsize = ((size_t) *n)*sizeof(CeedScalar); 542418fb8c2Sjeremylt } 543418fb8c2Sjeremylt 544d7b241e6Sjeremylt #define fCeedQFunctionApply \ 545d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionapply,CEEDQFUNCTIONAPPLY) 546d7b241e6Sjeremylt //TODO Need Fixing, double pointer 547d7b241e6Sjeremylt void fCeedQFunctionApply(int *qf, int *Q, 548aedaa0e5Sjeremylt int *u, int *u1, int *u2, int *u3, 549aedaa0e5Sjeremylt int *u4, int *u5, int *u6, int *u7, 550aedaa0e5Sjeremylt int *u8, int *u9, int *u10, int *u11, 551aedaa0e5Sjeremylt int *u12, int *u13, int *u14, int *u15, 552aedaa0e5Sjeremylt int *v, int *v1, int *v2, int *v3, 553aedaa0e5Sjeremylt int *v4, int *v5, int *v6, int *v7, 554aedaa0e5Sjeremylt int *v8, int *v9, int *v10, int *v11, 555aedaa0e5Sjeremylt int *v12, int *v13, int *v14, int *v15, int *err) { 556d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 557aedaa0e5Sjeremylt CeedVector *in; 558d7b241e6Sjeremylt *err = CeedCalloc(16, &in); 559d7b241e6Sjeremylt if (*err) return; 560aedaa0e5Sjeremylt in[0] = *u==FORTRAN_NULL?NULL:CeedVector_dict[*u]; 561aedaa0e5Sjeremylt in[1] = *u1==FORTRAN_NULL?NULL:CeedVector_dict[*u1]; 562aedaa0e5Sjeremylt in[2] = *u2==FORTRAN_NULL?NULL:CeedVector_dict[*u2]; 563aedaa0e5Sjeremylt in[3] = *u3==FORTRAN_NULL?NULL:CeedVector_dict[*u3]; 564aedaa0e5Sjeremylt in[4] = *u4==FORTRAN_NULL?NULL:CeedVector_dict[*u4]; 565aedaa0e5Sjeremylt in[5] = *u5==FORTRAN_NULL?NULL:CeedVector_dict[*u5]; 566aedaa0e5Sjeremylt in[6] = *u6==FORTRAN_NULL?NULL:CeedVector_dict[*u6]; 567aedaa0e5Sjeremylt in[7] = *u7==FORTRAN_NULL?NULL:CeedVector_dict[*u7]; 568aedaa0e5Sjeremylt in[8] = *u8==FORTRAN_NULL?NULL:CeedVector_dict[*u8]; 569aedaa0e5Sjeremylt in[9] = *u9==FORTRAN_NULL?NULL:CeedVector_dict[*u9]; 570aedaa0e5Sjeremylt in[10] = *u10==FORTRAN_NULL?NULL:CeedVector_dict[*u10]; 571aedaa0e5Sjeremylt in[11] = *u11==FORTRAN_NULL?NULL:CeedVector_dict[*u11]; 572aedaa0e5Sjeremylt in[12] = *u12==FORTRAN_NULL?NULL:CeedVector_dict[*u12]; 573aedaa0e5Sjeremylt in[13] = *u13==FORTRAN_NULL?NULL:CeedVector_dict[*u13]; 574aedaa0e5Sjeremylt in[14] = *u14==FORTRAN_NULL?NULL:CeedVector_dict[*u14]; 575aedaa0e5Sjeremylt in[15] = *u15==FORTRAN_NULL?NULL:CeedVector_dict[*u15]; 576aedaa0e5Sjeremylt CeedVector *out; 577d7b241e6Sjeremylt *err = CeedCalloc(16, &out); 578d7b241e6Sjeremylt if (*err) return; 579aedaa0e5Sjeremylt out[0] = *v==FORTRAN_NULL?NULL:CeedVector_dict[*v]; 580aedaa0e5Sjeremylt out[1] = *v1==FORTRAN_NULL?NULL:CeedVector_dict[*v1]; 581aedaa0e5Sjeremylt out[2] = *v2==FORTRAN_NULL?NULL:CeedVector_dict[*v2]; 582aedaa0e5Sjeremylt out[3] = *v3==FORTRAN_NULL?NULL:CeedVector_dict[*v3]; 583aedaa0e5Sjeremylt out[4] = *v4==FORTRAN_NULL?NULL:CeedVector_dict[*v4]; 584aedaa0e5Sjeremylt out[5] = *v5==FORTRAN_NULL?NULL:CeedVector_dict[*v5]; 585aedaa0e5Sjeremylt out[6] = *v6==FORTRAN_NULL?NULL:CeedVector_dict[*v6]; 586aedaa0e5Sjeremylt out[7] = *v7==FORTRAN_NULL?NULL:CeedVector_dict[*v7]; 587aedaa0e5Sjeremylt out[8] = *v8==FORTRAN_NULL?NULL:CeedVector_dict[*v8]; 588aedaa0e5Sjeremylt out[9] = *v9==FORTRAN_NULL?NULL:CeedVector_dict[*v9]; 589aedaa0e5Sjeremylt out[10] = *v10==FORTRAN_NULL?NULL:CeedVector_dict[*v10]; 590aedaa0e5Sjeremylt out[11] = *v11==FORTRAN_NULL?NULL:CeedVector_dict[*v11]; 591aedaa0e5Sjeremylt out[12] = *v12==FORTRAN_NULL?NULL:CeedVector_dict[*v12]; 592aedaa0e5Sjeremylt out[13] = *v13==FORTRAN_NULL?NULL:CeedVector_dict[*v13]; 593aedaa0e5Sjeremylt out[14] = *v14==FORTRAN_NULL?NULL:CeedVector_dict[*v14]; 594aedaa0e5Sjeremylt out[15] = *v15==FORTRAN_NULL?NULL:CeedVector_dict[*v15]; 595aedaa0e5Sjeremylt *err = CeedQFunctionApply(qf_, *Q, in, out); 596d7b241e6Sjeremylt if (*err) return; 597d7b241e6Sjeremylt 598d7b241e6Sjeremylt *err = CeedFree(&in); 599d7b241e6Sjeremylt if (*err) return; 600d7b241e6Sjeremylt *err = CeedFree(&out); 601d7b241e6Sjeremylt } 602d7b241e6Sjeremylt 603d7b241e6Sjeremylt #define fCeedQFunctionDestroy \ 604d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctiondestroy,CEEDQFUNCTIONDESTROY) 605d7b241e6Sjeremylt void fCeedQFunctionDestroy(int *qf, int *err) { 606c4b814baSjeremylt fContext *fctx = CeedQFunction_dict[*qf]->ctx; 607c4b814baSjeremylt *err = CeedFree(&fctx); 608d7b241e6Sjeremylt if (*err) return; 609c4b814baSjeremylt 610c4b814baSjeremylt *err = CeedQFunctionDestroy(&CeedQFunction_dict[*qf]); 611c4b814baSjeremylt if (*err) return; 612c4b814baSjeremylt 613d7b241e6Sjeremylt CeedQFunction_n--; 614d7b241e6Sjeremylt if (CeedQFunction_n == 0) { 615d7b241e6Sjeremylt *err = CeedFree(&CeedQFunction_dict); 616d7b241e6Sjeremylt CeedQFunction_count = 0; 617d7b241e6Sjeremylt CeedQFunction_count_max = 0; 618d7b241e6Sjeremylt } 619d7b241e6Sjeremylt } 620d7b241e6Sjeremylt 621d7b241e6Sjeremylt static CeedOperator *CeedOperator_dict = NULL; 622d7b241e6Sjeremylt static int CeedOperator_count = 0; 623d7b241e6Sjeremylt static int CeedOperator_n = 0; 624d7b241e6Sjeremylt static int CeedOperator_count_max = 0; 625d7b241e6Sjeremylt 626d7b241e6Sjeremylt #define fCeedOperatorCreate \ 627d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorcreate, CEEDOPERATORCREATE) 628d7b241e6Sjeremylt void fCeedOperatorCreate(int *ceed, 629d7b241e6Sjeremylt int *qf, int *dqf, int *dqfT, int *op, int *err) { 630d7b241e6Sjeremylt if (CeedOperator_count == CeedOperator_count_max) 631d7b241e6Sjeremylt CeedOperator_count_max += CeedOperator_count_max/2 + 1, 632d7b241e6Sjeremylt CeedOperator_dict = 633d7b241e6Sjeremylt realloc(CeedOperator_dict, sizeof(CeedOperator)*CeedOperator_count_max); 634d7b241e6Sjeremylt 635d7b241e6Sjeremylt CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 636d7b241e6Sjeremylt 637d7b241e6Sjeremylt CeedQFunction dqf_ = NULL, dqfT_ = NULL; 638d7b241e6Sjeremylt if (*dqf != FORTRAN_NULL) dqf_ = CeedQFunction_dict[*dqf ]; 639d7b241e6Sjeremylt if (*dqfT != FORTRAN_NULL) dqfT_ = CeedQFunction_dict[*dqfT]; 640d7b241e6Sjeremylt 641d7b241e6Sjeremylt *err = CeedOperatorCreate(Ceed_dict[*ceed], CeedQFunction_dict[*qf], dqf_, 642d7b241e6Sjeremylt dqfT_, op_); 643d7b241e6Sjeremylt if (*err) return; 644d7b241e6Sjeremylt *op = CeedOperator_count++; 645d7b241e6Sjeremylt CeedOperator_n++; 646d7b241e6Sjeremylt } 647d7b241e6Sjeremylt 64852d6035fSJeremy L Thompson #define fCeedCompositeOperatorCreate \ 64952d6035fSJeremy L Thompson FORTRAN_NAME(ceedcompositeoperatorcreate, CEEDCOMPOSITEOPERATORCREATE) 65052d6035fSJeremy L Thompson void fCeedCompositeOperatorCreate(int *ceed, int *op, int *err) { 65152d6035fSJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) 65252d6035fSJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1, 65352d6035fSJeremy L Thompson CeedOperator_dict = 65452d6035fSJeremy L Thompson realloc(CeedOperator_dict, sizeof(CeedOperator)*CeedOperator_count_max); 65552d6035fSJeremy L Thompson 65652d6035fSJeremy L Thompson CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 65752d6035fSJeremy L Thompson 65852d6035fSJeremy L Thompson *err = CeedCompositeOperatorCreate(Ceed_dict[*ceed], op_); 65952d6035fSJeremy L Thompson if (*err) return; 66052d6035fSJeremy L Thompson *op = CeedOperator_count++; 66152d6035fSJeremy L Thompson CeedOperator_n++; 66252d6035fSJeremy L Thompson } 66352d6035fSJeremy L Thompson 664d7b241e6Sjeremylt #define fCeedOperatorSetField \ 665d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorsetfield,CEEDOPERATORSETFIELD) 666d7b241e6Sjeremylt void fCeedOperatorSetField(int *op, const char *fieldname, 6674dccadb6Sjeremylt int *r, int *lmode, int *b, int *v, int *err, 668c2ae719dSJed Brown fortran_charlen_t fieldname_len) { 669c2ae719dSJed Brown FIX_STRING(fieldname); 670d7b241e6Sjeremylt CeedElemRestriction r_; 671d7b241e6Sjeremylt CeedBasis b_; 672d7b241e6Sjeremylt CeedVector v_; 673d7b241e6Sjeremylt 674d7b241e6Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 675d7b241e6Sjeremylt 676d7b241e6Sjeremylt if (*r == FORTRAN_NULL) { 677d7b241e6Sjeremylt r_ = NULL; 678d7b241e6Sjeremylt } else { 679d7b241e6Sjeremylt r_ = CeedElemRestriction_dict[*r]; 680d7b241e6Sjeremylt } 6814dccadb6Sjeremylt 682d7b241e6Sjeremylt if (*b == FORTRAN_NULL) { 683d7b241e6Sjeremylt b_ = NULL; 684783c99b3SValeria Barra } else if (*b == FORTRAN_BASIS_COLLOCATED) { 685783c99b3SValeria Barra b_ = CEED_BASIS_COLLOCATED; 686d7b241e6Sjeremylt } else { 687d7b241e6Sjeremylt b_ = CeedBasis_dict[*b]; 688d7b241e6Sjeremylt } 689d7b241e6Sjeremylt if (*v == FORTRAN_NULL) { 690d7b241e6Sjeremylt v_ = NULL; 691d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_ACTIVE) { 692d7b241e6Sjeremylt v_ = CEED_VECTOR_ACTIVE; 693d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_NONE) { 694d7b241e6Sjeremylt v_ = CEED_VECTOR_NONE; 695d7b241e6Sjeremylt } else { 696d7b241e6Sjeremylt v_ = CeedVector_dict[*v]; 697d7b241e6Sjeremylt } 698d7b241e6Sjeremylt 6994dccadb6Sjeremylt *err = CeedOperatorSetField(op_, fieldname_c, r_, *lmode, b_, v_); 700d7b241e6Sjeremylt } 701d7b241e6Sjeremylt 70252d6035fSJeremy L Thompson #define fCeedCompositeOperatorAddSub \ 70352d6035fSJeremy L Thompson FORTRAN_NAME(ceedcompositeoperatoraddsub, CEEDCOMPOSITEOPERATORADDSUB) 70452d6035fSJeremy L Thompson void fCeedCompositeOperatorAddSub(int *compositeop, int *subop, int *err) { 70552d6035fSJeremy L Thompson CeedOperator compositeop_ = CeedOperator_dict[*compositeop]; 70652d6035fSJeremy L Thompson CeedOperator subop_ = CeedOperator_dict[*subop]; 70752d6035fSJeremy L Thompson 70852d6035fSJeremy L Thompson *err = CeedCompositeOperatorAddSub(compositeop_, subop_); 70952d6035fSJeremy L Thompson if (*err) return; 71052d6035fSJeremy L Thompson } 71152d6035fSJeremy L Thompson 712d7b241e6Sjeremylt #define fCeedOperatorApply FORTRAN_NAME(ceedoperatorapply, CEEDOPERATORAPPLY) 713d7b241e6Sjeremylt void fCeedOperatorApply(int *op, int *ustatevec, 714d7b241e6Sjeremylt int *resvec, int *rqst, int *err) { 71512bf2c11SJed Brown CeedVector ustatevec_ = *ustatevec == FORTRAN_NULL 71612bf2c11SJed Brown ? NULL : CeedVector_dict[*ustatevec]; 71712bf2c11SJed Brown CeedVector resvec_ = *resvec == FORTRAN_NULL 71812bf2c11SJed Brown ? NULL : CeedVector_dict[*resvec]; 719d7b241e6Sjeremylt 720d7b241e6Sjeremylt int createRequest = 1; 721d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 722d7b241e6Sjeremylt if (*rqst == -1 || *rqst == -2) { 723d7b241e6Sjeremylt createRequest = 0; 724d7b241e6Sjeremylt } 725d7b241e6Sjeremylt 726d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 727d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 728d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 729d7b241e6Sjeremylt } 730d7b241e6Sjeremylt 731d7b241e6Sjeremylt CeedRequest *rqst_; 732d7b241e6Sjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 733d7b241e6Sjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 734d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 735d7b241e6Sjeremylt 736d7b241e6Sjeremylt *err = CeedOperatorApply(CeedOperator_dict[*op], 73712bf2c11SJed Brown ustatevec_, resvec_, rqst_); 738d7b241e6Sjeremylt if (*err) return; 739d7b241e6Sjeremylt if (createRequest) { 740d7b241e6Sjeremylt *rqst = CeedRequest_count++; 741d7b241e6Sjeremylt CeedRequest_n++; 742d7b241e6Sjeremylt } 743d7b241e6Sjeremylt } 744d7b241e6Sjeremylt 745d7b241e6Sjeremylt #define fCeedOperatorApplyJacobian \ 746d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorapplyjacobian, CEEDOPERATORAPPLYJACOBIAN) 747d7b241e6Sjeremylt void fCeedOperatorApplyJacobian(int *op, int *qdatavec, int *ustatevec, 748d7b241e6Sjeremylt int *dustatevec, int *dresvec, int *rqst, int *err) { 749d7b241e6Sjeremylt // TODO Uncomment this when CeedOperatorApplyJacobian is implemented 750d7b241e6Sjeremylt // *err = CeedOperatorApplyJacobian(CeedOperator_dict[*op], CeedVector_dict[*qdatavec], 751d7b241e6Sjeremylt // CeedVector_dict[*ustatevec], CeedVector_dict[*dustatevec], 752d7b241e6Sjeremylt // CeedVector_dict[*dresvec], &CeedRequest_dict[*rqst]); 753d7b241e6Sjeremylt } 754d7b241e6Sjeremylt 755d7b241e6Sjeremylt #define fCeedOperatorDestroy \ 756d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatordestroy, CEEDOPERATORDESTROY) 757d7b241e6Sjeremylt void fCeedOperatorDestroy(int *op, int *err) { 758d7b241e6Sjeremylt *err = CeedOperatorDestroy(&CeedOperator_dict[*op]); 759d7b241e6Sjeremylt if (*err) return; 760d7b241e6Sjeremylt CeedOperator_n--; 761d7b241e6Sjeremylt if (CeedOperator_n == 0) { 762d7b241e6Sjeremylt *err = CeedFree(&CeedOperator_dict); 763d7b241e6Sjeremylt CeedOperator_count = 0; 764d7b241e6Sjeremylt CeedOperator_count_max = 0; 765d7b241e6Sjeremylt } 766d7b241e6Sjeremylt } 767