1d7b241e6Sjeremylt // Fortran interface 2d7b241e6Sjeremylt #include <ceed.h> 3d7b241e6Sjeremylt #include <ceed-impl.h> 4d7b241e6Sjeremylt #include <ceed-fortran-name.h> 5d7b241e6Sjeremylt 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 67d7b241e6Sjeremylt #define fCeedDestroy FORTRAN_NAME(ceeddestroy,CEEDDESTROY) 68d7b241e6Sjeremylt void fCeedDestroy(int *ceed, int *err) { 69d7b241e6Sjeremylt *err = CeedDestroy(&Ceed_dict[*ceed]); 70d7b241e6Sjeremylt 71d7b241e6Sjeremylt if (*err == 0) { 72d7b241e6Sjeremylt Ceed_n--; 73d7b241e6Sjeremylt if (Ceed_n == 0) { 74d7b241e6Sjeremylt CeedFree(&Ceed_dict); 75d7b241e6Sjeremylt Ceed_count = 0; 76d7b241e6Sjeremylt Ceed_count_max = 0; 77d7b241e6Sjeremylt } 78d7b241e6Sjeremylt } 79d7b241e6Sjeremylt } 80d7b241e6Sjeremylt 81d7b241e6Sjeremylt static CeedVector *CeedVector_dict = NULL; 82d7b241e6Sjeremylt static int CeedVector_count = 0; 83d7b241e6Sjeremylt static int CeedVector_n = 0; 84d7b241e6Sjeremylt static int CeedVector_count_max = 0; 85d7b241e6Sjeremylt 86d7b241e6Sjeremylt #define fCeedVectorCreate FORTRAN_NAME(ceedvectorcreate,CEEDVECTORCREATE) 87d7b241e6Sjeremylt void fCeedVectorCreate(int *ceed, int *length, int *vec, int *err) { 88d7b241e6Sjeremylt if (CeedVector_count == CeedVector_count_max) { 89d7b241e6Sjeremylt CeedVector_count_max += CeedVector_count_max/2 + 1; 90d7b241e6Sjeremylt CeedRealloc(CeedVector_count_max, &CeedVector_dict); 91d7b241e6Sjeremylt } 92d7b241e6Sjeremylt 93d7b241e6Sjeremylt CeedVector* vec_ = &CeedVector_dict[CeedVector_count]; 94d7b241e6Sjeremylt *err = CeedVectorCreate(Ceed_dict[*ceed], *length, vec_); 95d7b241e6Sjeremylt 96d7b241e6Sjeremylt if (*err == 0) { 97d7b241e6Sjeremylt *vec = CeedVector_count++; 98d7b241e6Sjeremylt CeedVector_n++; 99d7b241e6Sjeremylt } 100d7b241e6Sjeremylt } 101d7b241e6Sjeremylt 102d7b241e6Sjeremylt #define fCeedVectorSetArray FORTRAN_NAME(ceedvectorsetarray,CEEDVECTORSETARRAY) 103d7b241e6Sjeremylt void fCeedVectorSetArray(int *vec, int *memtype, int *copymode, 104d7b241e6Sjeremylt CeedScalar *array, int *err) { 105d7b241e6Sjeremylt *err = CeedVectorSetArray(CeedVector_dict[*vec], *memtype, *copymode, array); 106d7b241e6Sjeremylt } 107d7b241e6Sjeremylt 108d7b241e6Sjeremylt #define fCeedVectorSetValue FORTRAN_NAME(ceedvectorsetvalue,CEEDVECTORSETVALUE) 109d7b241e6Sjeremylt void fCeedVectorSetValue(int *vec, CeedScalar *value, int *err) { 110d7b241e6Sjeremylt *err = CeedVectorSetValue(CeedVector_dict[*vec], *value); 111d7b241e6Sjeremylt } 112d7b241e6Sjeremylt 113d7b241e6Sjeremylt #define fCeedVectorGetArray FORTRAN_NAME(ceedvectorgetarray,CEEDVECTORGETARRAY) 114*a8de75f0Sjeremylt void fCeedVectorGetArray(int *vec, int *memtype, CeedScalar *array, 115*a8de75f0Sjeremylt int64_t *offset, 116cbfd6a90SThilina Rathnayake int *err) { 117d7b241e6Sjeremylt CeedScalar *b; 118d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 119d7b241e6Sjeremylt *err = CeedVectorGetArray(vec_, *memtype, &b); 120cbfd6a90SThilina Rathnayake *offset = b - array; 121d7b241e6Sjeremylt } 122d7b241e6Sjeremylt 123d7b241e6Sjeremylt #define fCeedVectorGetArrayRead \ 124d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorgetarrayread,CEEDVECTORGETARRAYREAD) 125d7b241e6Sjeremylt void fCeedVectorGetArrayRead(int *vec, int *memtype, CeedScalar *array, 1263334379fSJed Brown int64_t *offset, int *err) { 127d7b241e6Sjeremylt const CeedScalar *b; 128d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 129cbfd6a90SThilina Rathnayake *err = CeedVectorGetArrayRead(vec_, *memtype, &b); 130cbfd6a90SThilina Rathnayake *offset = b - array; 131d7b241e6Sjeremylt } 132d7b241e6Sjeremylt 133d7b241e6Sjeremylt #define fCeedVectorRestoreArray \ 134d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearray,CEEDVECTORRESTOREARRAY) 1356456524eSJed Brown void fCeedVectorRestoreArray(int *vec, CeedScalar *array, 1366456524eSJed Brown int64_t *offset, int *err) { 137d7b241e6Sjeremylt *err = CeedVectorRestoreArray(CeedVector_dict[*vec], &array); 1386456524eSJed Brown *offset = 0; 139d7b241e6Sjeremylt } 140d7b241e6Sjeremylt 141d7b241e6Sjeremylt #define fCeedVectorRestoreArrayRead \ 142d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearrayread,CEEDVECTORRESTOREARRAYREAD) 1436456524eSJed Brown void fCeedVectorRestoreArrayRead(int *vec, const CeedScalar *array, 1446456524eSJed Brown int64_t *offset, int *err) { 145d7b241e6Sjeremylt *err = CeedVectorRestoreArrayRead(CeedVector_dict[*vec], &array); 1466456524eSJed Brown *offset = 0; 147d7b241e6Sjeremylt } 148d7b241e6Sjeremylt 149d7b241e6Sjeremylt #define fCeedVectorView FORTRAN_NAME(ceedvectorview,CEEDVECTORVIEW) 150d7b241e6Sjeremylt void fCeedVectorView(int *vec, int *err) { 151d7b241e6Sjeremylt *err = CeedVectorView(CeedVector_dict[*vec], "%12.8f", stdout); 152d7b241e6Sjeremylt } 153d7b241e6Sjeremylt 154d7b241e6Sjeremylt #define fCeedVectorDestroy FORTRAN_NAME(ceedvectordestroy,CEEDVECTORDESTROY) 155d7b241e6Sjeremylt void fCeedVectorDestroy(int *vec, int *err) { 156d7b241e6Sjeremylt *err = CeedVectorDestroy(&CeedVector_dict[*vec]); 157d7b241e6Sjeremylt 158d7b241e6Sjeremylt if (*err == 0) { 159d7b241e6Sjeremylt CeedVector_n--; 160d7b241e6Sjeremylt if (CeedVector_n == 0) { 161d7b241e6Sjeremylt CeedFree(&CeedVector_dict); 162d7b241e6Sjeremylt CeedVector_count = 0; 163d7b241e6Sjeremylt CeedVector_count_max = 0; 164d7b241e6Sjeremylt } 165d7b241e6Sjeremylt } 166d7b241e6Sjeremylt } 167d7b241e6Sjeremylt 168d7b241e6Sjeremylt static CeedElemRestriction *CeedElemRestriction_dict = NULL; 169d7b241e6Sjeremylt static int CeedElemRestriction_count = 0; 170d7b241e6Sjeremylt static int CeedElemRestriction_n = 0; 171d7b241e6Sjeremylt static int CeedElemRestriction_count_max = 0; 172d7b241e6Sjeremylt 173d7b241e6Sjeremylt #define fCeedElemRestrictionCreate \ 174d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreate, CEEDELEMRESTRICTIONCREATE) 175d7b241e6Sjeremylt void fCeedElemRestrictionCreate(int *ceed, int *nelements, 176d7b241e6Sjeremylt int *esize, int *ndof, int *ncomp, int *memtype, int *copymode, 177d7b241e6Sjeremylt const int *indices, int *elemrestriction, int *err) { 178d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 179d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 180d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 181d7b241e6Sjeremylt } 182d7b241e6Sjeremylt 183d7b241e6Sjeremylt const int *indices_ = indices; 184d7b241e6Sjeremylt 185d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 186d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 187d7b241e6Sjeremylt *err = CeedElemRestrictionCreate(Ceed_dict[*ceed], *nelements, *esize, *ndof, 188d7b241e6Sjeremylt *ncomp, 189d7b241e6Sjeremylt *memtype, *copymode, indices_, elemrestriction_); 190d7b241e6Sjeremylt 191d7b241e6Sjeremylt if (*err == 0) { 192d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 193d7b241e6Sjeremylt CeedElemRestriction_n++; 194d7b241e6Sjeremylt } 195d7b241e6Sjeremylt } 196d7b241e6Sjeremylt 197d7b241e6Sjeremylt #define fCeedElemRestrictionCreateIdentity \ 198d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateidentity, CEEDELEMRESTRICTIONCREATEIDENTITY) 199d7b241e6Sjeremylt void fCeedElemRestrictionCreateIdentity(int *ceed, int *nelements, 200d7b241e6Sjeremylt int *esize, int *ndof, int *ncomp, 201d7b241e6Sjeremylt int *elemrestriction, int *err) { 202d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 203d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 204d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 205d7b241e6Sjeremylt } 206d7b241e6Sjeremylt 207d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 208d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 2094b8bea3bSJed Brown *err = CeedElemRestrictionCreateIdentity(Ceed_dict[*ceed], *nelements, *esize, 2104b8bea3bSJed Brown *ndof, 211d7b241e6Sjeremylt *ncomp, elemrestriction_); 212d7b241e6Sjeremylt 213d7b241e6Sjeremylt if (*err == 0) { 214d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 215d7b241e6Sjeremylt CeedElemRestriction_n++; 216d7b241e6Sjeremylt } 217d7b241e6Sjeremylt } 218d7b241e6Sjeremylt 219d7b241e6Sjeremylt #define fCeedElemRestrictionCreateBlocked \ 220d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateblocked,CEEDELEMRESTRICTIONCREATEBLOCKED) 221d7b241e6Sjeremylt void fCeedElemRestrictionCreateBlocked(int *ceed, int *nelements, 222d7b241e6Sjeremylt int *esize, int *blocksize, int *ndof, int *ncomp, 223d7b241e6Sjeremylt int *mtype, int *cmode, 224d7b241e6Sjeremylt int *blkindices, int *elemrestriction, int *err) { 225d7b241e6Sjeremylt 226d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 227d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 228d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 229d7b241e6Sjeremylt } 230d7b241e6Sjeremylt 231d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 232d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 233d7b241e6Sjeremylt *err = CeedElemRestrictionCreateBlocked(Ceed_dict[*ceed], *nelements, *esize, 234d7b241e6Sjeremylt *blocksize, *ndof, *ncomp, *mtype, *cmode, blkindices, 235d7b241e6Sjeremylt elemrestriction_); 236d7b241e6Sjeremylt 237d7b241e6Sjeremylt if (*err == 0) { 238d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 239d7b241e6Sjeremylt CeedElemRestriction_n++; 240d7b241e6Sjeremylt } 241d7b241e6Sjeremylt } 242d7b241e6Sjeremylt 243d7b241e6Sjeremylt static CeedRequest *CeedRequest_dict = NULL; 244d7b241e6Sjeremylt static int CeedRequest_count = 0; 245d7b241e6Sjeremylt static int CeedRequest_n = 0; 246d7b241e6Sjeremylt static int CeedRequest_count_max = 0; 247d7b241e6Sjeremylt 248d7b241e6Sjeremylt #define fCeedElemRestrictionApply \ 249d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictionapply,CEEDELEMRESTRICTIONAPPLY) 250d7b241e6Sjeremylt void fCeedElemRestrictionApply(int *elemr, int *tmode, int *lmode, 251d7b241e6Sjeremylt int *uvec, int *ruvec, int *rqst, int *err) { 252d7b241e6Sjeremylt int createRequest = 1; 253d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 254d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) 255d7b241e6Sjeremylt createRequest = 0; 256d7b241e6Sjeremylt 257d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 258d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 259d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 260d7b241e6Sjeremylt } 261d7b241e6Sjeremylt 262d7b241e6Sjeremylt CeedRequest *rqst_; 263d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 264d7b241e6Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED ) rqst_ = CEED_REQUEST_ORDERED; 265d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 266d7b241e6Sjeremylt 267d7b241e6Sjeremylt *err = CeedElemRestrictionApply(CeedElemRestriction_dict[*elemr], *tmode, 268d7b241e6Sjeremylt *lmode, CeedVector_dict[*uvec], CeedVector_dict[*ruvec], rqst_); 269d7b241e6Sjeremylt 270d7b241e6Sjeremylt if (*err == 0 && createRequest) { 271d7b241e6Sjeremylt *rqst = CeedRequest_count++; 272d7b241e6Sjeremylt CeedRequest_n++; 273d7b241e6Sjeremylt } 274d7b241e6Sjeremylt } 275d7b241e6Sjeremylt 276d7b241e6Sjeremylt #define fCeedRequestWait FORTRAN_NAME(ceedrequestwait, CEEDREQUESTWAIT) 277d7b241e6Sjeremylt void fCeedRequestWait(int *rqst, int *err) { 278d7b241e6Sjeremylt // TODO Uncomment this once CeedRequestWait is implemented 279d7b241e6Sjeremylt //*err = CeedRequestWait(&CeedRequest_dict[*rqst]); 280d7b241e6Sjeremylt 281d7b241e6Sjeremylt if (*err == 0) { 282d7b241e6Sjeremylt CeedRequest_n--; 283d7b241e6Sjeremylt if (CeedRequest_n == 0) { 284d7b241e6Sjeremylt CeedFree(&CeedRequest_dict); 285d7b241e6Sjeremylt CeedRequest_count = 0; 286d7b241e6Sjeremylt CeedRequest_count_max = 0; 287d7b241e6Sjeremylt } 288d7b241e6Sjeremylt } 289d7b241e6Sjeremylt } 290d7b241e6Sjeremylt 291d7b241e6Sjeremylt #define fCeedElemRestrictionDestroy \ 292d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictiondestroy,CEEDELEMRESTRICTIONDESTROY) 293d7b241e6Sjeremylt void fCeedElemRestrictionDestroy(int *elem, int *err) { 294d7b241e6Sjeremylt *err = CeedElemRestrictionDestroy(&CeedElemRestriction_dict[*elem]); 295d7b241e6Sjeremylt 296d7b241e6Sjeremylt if (*err == 0) { 297d7b241e6Sjeremylt CeedElemRestriction_n--; 298d7b241e6Sjeremylt if (CeedElemRestriction_n == 0) { 299d7b241e6Sjeremylt CeedFree(&CeedElemRestriction_dict); 300d7b241e6Sjeremylt CeedElemRestriction_count = 0; 301d7b241e6Sjeremylt CeedElemRestriction_count_max = 0; 302d7b241e6Sjeremylt } 303d7b241e6Sjeremylt } 304d7b241e6Sjeremylt } 305d7b241e6Sjeremylt 306d7b241e6Sjeremylt static CeedBasis *CeedBasis_dict = NULL; 307d7b241e6Sjeremylt static int CeedBasis_count = 0; 308d7b241e6Sjeremylt static int CeedBasis_n = 0; 309d7b241e6Sjeremylt static int CeedBasis_count_max = 0; 310d7b241e6Sjeremylt 311d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1Lagrange \ 312d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1lagrange, CEEDBASISCREATETENSORH1LAGRANGE) 313d7b241e6Sjeremylt void fCeedBasisCreateTensorH1Lagrange(int *ceed, int *dim, 314*a8de75f0Sjeremylt int *ncomp, int *P, int *Q, int *quadmode, int *basis, 315d7b241e6Sjeremylt int *err) { 316d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 317d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 318d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 319d7b241e6Sjeremylt } 320d7b241e6Sjeremylt 321*a8de75f0Sjeremylt *err = CeedBasisCreateTensorH1Lagrange(Ceed_dict[*ceed], *dim, *ncomp, *P, *Q, 322d7b241e6Sjeremylt *quadmode, &CeedBasis_dict[CeedBasis_count]); 323d7b241e6Sjeremylt 324d7b241e6Sjeremylt if (*err == 0) { 325d7b241e6Sjeremylt *basis = CeedBasis_count++; 326d7b241e6Sjeremylt CeedBasis_n++; 327d7b241e6Sjeremylt } 328d7b241e6Sjeremylt } 329d7b241e6Sjeremylt 330d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1 \ 331d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1, CEEDBASISCREATETENSORH1) 332*a8de75f0Sjeremylt void fCeedBasisCreateTensorH1(int *ceed, int *dim, int *ncomp, int *P1d, 333d7b241e6Sjeremylt int *Q1d, const CeedScalar *interp1d, const CeedScalar *grad1d, 334d7b241e6Sjeremylt const CeedScalar *qref1d, const CeedScalar *qweight1d, int *basis, int *err) { 335d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 336d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 337d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 338d7b241e6Sjeremylt } 339d7b241e6Sjeremylt 340*a8de75f0Sjeremylt *err = CeedBasisCreateTensorH1(Ceed_dict[*ceed], *dim, *ncomp, *P1d, *Q1d, 341d7b241e6Sjeremylt interp1d, grad1d, 342d7b241e6Sjeremylt qref1d, qweight1d, &CeedBasis_dict[CeedBasis_count]); 343d7b241e6Sjeremylt 344d7b241e6Sjeremylt if (*err == 0) { 345d7b241e6Sjeremylt *basis = CeedBasis_count++; 346d7b241e6Sjeremylt CeedBasis_n++; 347d7b241e6Sjeremylt } 348d7b241e6Sjeremylt } 349d7b241e6Sjeremylt 350*a8de75f0Sjeremylt #define fCeedBasisCreateH1 \ 351*a8de75f0Sjeremylt FORTRAN_NAME(ceedbasiscreateh1, CEEDBASISCREATEH1) 352*a8de75f0Sjeremylt void fCeedBasisCreateH1(int *ceed, int *topo, int *ncomp, int *ndof, 353*a8de75f0Sjeremylt int *nqpts, const CeedScalar *interp, const CeedScalar *grad, 354*a8de75f0Sjeremylt const CeedScalar *qref, const CeedScalar *qweight, int *basis, int *err) { 355*a8de75f0Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 356*a8de75f0Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 357*a8de75f0Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 358*a8de75f0Sjeremylt } 359*a8de75f0Sjeremylt 360*a8de75f0Sjeremylt *err = CeedBasisCreateH1(Ceed_dict[*ceed], *topo, *ncomp, 361*a8de75f0Sjeremylt *ndof, *nqpts, interp, grad, 362*a8de75f0Sjeremylt qref, qweight, &CeedBasis_dict[CeedBasis_count]); 363*a8de75f0Sjeremylt 364*a8de75f0Sjeremylt if (*err == 0) { 365*a8de75f0Sjeremylt *basis = CeedBasis_count++; 366*a8de75f0Sjeremylt CeedBasis_n++; 367*a8de75f0Sjeremylt } 368*a8de75f0Sjeremylt } 369*a8de75f0Sjeremylt 370d7b241e6Sjeremylt #define fCeedBasisView FORTRAN_NAME(ceedbasisview, CEEDBASISVIEW) 371d7b241e6Sjeremylt void fCeedBasisView(int *basis, int *err) { 372d7b241e6Sjeremylt *err = CeedBasisView(CeedBasis_dict[*basis], stdout); 373d7b241e6Sjeremylt } 374d7b241e6Sjeremylt 375d7b241e6Sjeremylt #define fCeedQRFactorization \ 376d7b241e6Sjeremylt FORTRAN_NAME(ceedqrfactorization, CEEDQRFACTORIZATION) 377d7b241e6Sjeremylt void fCeedQRFactorization(CeedScalar *mat, CeedScalar *tau, int *m, int *n, 378d7b241e6Sjeremylt int *err) { 379d7b241e6Sjeremylt *err = CeedQRFactorization(mat, tau, *m, *n); 380d7b241e6Sjeremylt } 381d7b241e6Sjeremylt 382783c99b3SValeria Barra #define fCeedBasisGetCollocatedGrad \ 383783c99b3SValeria Barra FORTRAN_NAME(ceedbasisgetcollocatedgrad, CEEDBASISGETCOLLOCATEDGRAD) 384783c99b3SValeria Barra void fCeedBasisGetCollocatedGrad(int *basis, CeedScalar *colograd1d, 385d7b241e6Sjeremylt int *err) { 386783c99b3SValeria Barra *err = CeedBasisGetCollocatedGrad(CeedBasis_dict[*basis], colograd1d); 387d7b241e6Sjeremylt } 388d7b241e6Sjeremylt 389d7b241e6Sjeremylt #define fCeedBasisApply FORTRAN_NAME(ceedbasisapply, CEEDBASISAPPLY) 390d7b241e6Sjeremylt void fCeedBasisApply(int *basis, int *nelem, int *tmode, int *emode, 391d7b241e6Sjeremylt const CeedScalar *u, CeedScalar *v, int *err) { 392d7b241e6Sjeremylt *err = CeedBasisApply(CeedBasis_dict[*basis], *nelem, *tmode, *emode, u, v); 393d7b241e6Sjeremylt } 394d7b241e6Sjeremylt 395d7b241e6Sjeremylt #define fCeedBasisGetNumNodes \ 396d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumnodes, CEEDBASISGETNUMNODES) 397d7b241e6Sjeremylt void fCeedBasisGetNumNodes(int *basis, int *P, int *err) { 398d7b241e6Sjeremylt *err = CeedBasisGetNumNodes(CeedBasis_dict[*basis], P); 399d7b241e6Sjeremylt } 400d7b241e6Sjeremylt 401d7b241e6Sjeremylt #define fCeedBasisGetNumQuadraturePoints \ 402d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumquadraturepoints, CEEDBASISGETNUMQUADRATUREPOINTS) 403d7b241e6Sjeremylt void fCeedBasisGetNumQuadraturePoints(int *basis, int *Q, int *err) { 404d7b241e6Sjeremylt *err = CeedBasisGetNumQuadraturePoints(CeedBasis_dict[*basis], Q); 405d7b241e6Sjeremylt } 406d7b241e6Sjeremylt 407d7b241e6Sjeremylt #define fCeedBasisDestroy FORTRAN_NAME(ceedbasisdestroy,CEEDBASISDESTROY) 408d7b241e6Sjeremylt void fCeedBasisDestroy(int *basis, int *err) { 409d7b241e6Sjeremylt *err = CeedBasisDestroy(&CeedBasis_dict[*basis]); 410d7b241e6Sjeremylt 411d7b241e6Sjeremylt if (*err == 0) { 412d7b241e6Sjeremylt CeedBasis_n--; 413d7b241e6Sjeremylt if (CeedBasis_n == 0) { 414d7b241e6Sjeremylt CeedFree(&CeedBasis_dict); 415d7b241e6Sjeremylt CeedBasis_count = 0; 416d7b241e6Sjeremylt CeedBasis_count_max = 0; 417d7b241e6Sjeremylt } 418d7b241e6Sjeremylt } 419d7b241e6Sjeremylt } 420d7b241e6Sjeremylt 421d7b241e6Sjeremylt #define fCeedGaussQuadrature FORTRAN_NAME(ceedgaussquadrature, CEEDGAUSSQUADRATURE) 422d7b241e6Sjeremylt void fCeedGaussQuadrature(int *Q, CeedScalar *qref1d, CeedScalar *qweight1d, 423d7b241e6Sjeremylt int *err) { 424d7b241e6Sjeremylt *err = CeedGaussQuadrature(*Q, qref1d, qweight1d); 425d7b241e6Sjeremylt } 426d7b241e6Sjeremylt 427d7b241e6Sjeremylt #define fCeedLobattoQuadrature \ 428d7b241e6Sjeremylt FORTRAN_NAME(ceedlobattoquadrature, CEEDLOBATTOQUADRATURE) 429d7b241e6Sjeremylt void fCeedLobattoQuadrature(int *Q, CeedScalar *qref1d, CeedScalar *qweight1d, 430d7b241e6Sjeremylt int *err) { 431d7b241e6Sjeremylt *err = CeedLobattoQuadrature(*Q, qref1d, qweight1d); 432d7b241e6Sjeremylt } 433d7b241e6Sjeremylt 434d7b241e6Sjeremylt static CeedQFunction *CeedQFunction_dict = NULL; 435d7b241e6Sjeremylt static int CeedQFunction_count = 0; 436d7b241e6Sjeremylt static int CeedQFunction_n = 0; 437d7b241e6Sjeremylt static int CeedQFunction_count_max = 0; 438d7b241e6Sjeremylt 439d7b241e6Sjeremylt struct fContext { 440d7b241e6Sjeremylt void (*f)(void *ctx, int *nq, 441d7b241e6Sjeremylt const CeedScalar *u,const CeedScalar *u1,const CeedScalar *u2, 442d7b241e6Sjeremylt const CeedScalar *u3, 443d7b241e6Sjeremylt const CeedScalar *u4,const CeedScalar *u5,const CeedScalar *u6, 444d7b241e6Sjeremylt const CeedScalar *u7, 445d7b241e6Sjeremylt const CeedScalar *u8,const CeedScalar *u9,const CeedScalar *u10, 446d7b241e6Sjeremylt const CeedScalar *u11, 447d7b241e6Sjeremylt const CeedScalar *u12,const CeedScalar *u13,const CeedScalar *u14, 448d7b241e6Sjeremylt const CeedScalar *u15, 449d7b241e6Sjeremylt CeedScalar *v,CeedScalar *v1, CeedScalar *v2,CeedScalar *v3, 450d7b241e6Sjeremylt CeedScalar *v4,CeedScalar *v5, CeedScalar *v6,CeedScalar *v7, 451d7b241e6Sjeremylt CeedScalar *v8,CeedScalar *v9, CeedScalar *v10,CeedScalar *v11, 452d7b241e6Sjeremylt CeedScalar *v12,CeedScalar *v13, CeedScalar *v14,CeedScalar *v15, int *err); 453d7b241e6Sjeremylt void *innerctx; 454d7b241e6Sjeremylt }; 455d7b241e6Sjeremylt 456d7b241e6Sjeremylt static int CeedQFunctionFortranStub(void *ctx, int nq, 457d7b241e6Sjeremylt const CeedScalar *const *u, CeedScalar *const *v) { 458d7b241e6Sjeremylt struct fContext *fctx = ctx; 459d7b241e6Sjeremylt int ierr; 460d7b241e6Sjeremylt 461d7b241e6Sjeremylt CeedScalar ctx_=1.0; 462d7b241e6Sjeremylt fctx->f((void*)&ctx_,&nq,u[0],u[1],u[2],u[3],u[4],u[5],u[6],u[7], 463d7b241e6Sjeremylt u[8],u[9],u[10],u[11],u[12],u[13],u[14],u[15], 464d7b241e6Sjeremylt v[0],v[1],v[2],v[3],v[4],v[5],v[6],v[7], 465d7b241e6Sjeremylt v[8],v[9],v[10],v[11],v[12],v[13],v[14],v[15],&ierr); 466d7b241e6Sjeremylt return ierr; 467d7b241e6Sjeremylt } 468d7b241e6Sjeremylt 469d7b241e6Sjeremylt #define fCeedQFunctionCreateInterior \ 470d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctioncreateinterior, CEEDQFUNCTIONCREATEINTERIOR) 471d7b241e6Sjeremylt void fCeedQFunctionCreateInterior(int* ceed, int* vlength, 472d7b241e6Sjeremylt void (*f)(void *ctx, int *nq, 473d7b241e6Sjeremylt const CeedScalar *u,const CeedScalar *u1,const CeedScalar *u2, 474d7b241e6Sjeremylt const CeedScalar *u3, 475d7b241e6Sjeremylt const CeedScalar *u4,const CeedScalar *u5,const CeedScalar *u6, 476d7b241e6Sjeremylt const CeedScalar *u7, 477d7b241e6Sjeremylt const CeedScalar *u8,const CeedScalar *u9,const CeedScalar *u10, 478d7b241e6Sjeremylt const CeedScalar *u11, 479d7b241e6Sjeremylt const CeedScalar *u12,const CeedScalar *u13,const CeedScalar *u14, 480d7b241e6Sjeremylt const CeedScalar *u15, 481d7b241e6Sjeremylt CeedScalar *v,CeedScalar *v1, CeedScalar *v2,CeedScalar *v3, 482d7b241e6Sjeremylt CeedScalar *v4,CeedScalar *v5, CeedScalar *v6,CeedScalar *v7, 483d7b241e6Sjeremylt CeedScalar *v8,CeedScalar *v9, CeedScalar *v10,CeedScalar *v11, 484d7b241e6Sjeremylt CeedScalar *v12,CeedScalar *v13, CeedScalar *v14,CeedScalar *v15, 485d7b241e6Sjeremylt int *err), 486c2ae719dSJed Brown const char *focca, int *qf, int *err, 487c2ae719dSJed Brown fortran_charlen_t focca_len) { 488c2ae719dSJed Brown FIX_STRING(focca); 489d7b241e6Sjeremylt if (CeedQFunction_count == CeedQFunction_count_max) { 490d7b241e6Sjeremylt CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 491d7b241e6Sjeremylt CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 492d7b241e6Sjeremylt } 493d7b241e6Sjeremylt 494d7b241e6Sjeremylt CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 495d7b241e6Sjeremylt *err = CeedQFunctionCreateInterior(Ceed_dict[*ceed], *vlength, 496c2ae719dSJed Brown CeedQFunctionFortranStub,focca_c, qf_); 497d7b241e6Sjeremylt 498d7b241e6Sjeremylt if (*err == 0) { 499d7b241e6Sjeremylt *qf = CeedQFunction_count++; 500d7b241e6Sjeremylt CeedQFunction_n++; 501d7b241e6Sjeremylt } 502d7b241e6Sjeremylt 503d7b241e6Sjeremylt struct fContext *fctx; 504d7b241e6Sjeremylt *err = CeedMalloc(1, &fctx); 505d7b241e6Sjeremylt if (*err) return; 506d7b241e6Sjeremylt fctx->f = f; fctx->innerctx = NULL; 507d7b241e6Sjeremylt 508d7b241e6Sjeremylt *err = CeedQFunctionSetContext(*qf_, fctx, sizeof(struct fContext)); 509d7b241e6Sjeremylt 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 534d7b241e6Sjeremylt #define fCeedQFunctionApply \ 535d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionapply,CEEDQFUNCTIONAPPLY) 536d7b241e6Sjeremylt //TODO Need Fixing, double pointer 537d7b241e6Sjeremylt void fCeedQFunctionApply(int *qf, int *Q, 538d7b241e6Sjeremylt const CeedScalar *u,const CeedScalar *u1,const CeedScalar *u2, 539d7b241e6Sjeremylt const CeedScalar *u3, 540d7b241e6Sjeremylt const CeedScalar *u4,const CeedScalar *u5,const CeedScalar *u6, 541d7b241e6Sjeremylt const CeedScalar *u7, 542d7b241e6Sjeremylt const CeedScalar *u8,const CeedScalar *u9,const CeedScalar *u10, 543d7b241e6Sjeremylt const CeedScalar *u11, 544d7b241e6Sjeremylt const CeedScalar *u12,const CeedScalar *u13,const CeedScalar *u14, 545d7b241e6Sjeremylt const CeedScalar *u15, 546d7b241e6Sjeremylt CeedScalar *v,CeedScalar *v1, CeedScalar *v2,CeedScalar *v3, 547d7b241e6Sjeremylt CeedScalar *v4,CeedScalar *v5, CeedScalar *v6,CeedScalar *v7, 548d7b241e6Sjeremylt CeedScalar *v8,CeedScalar *v9, CeedScalar *v10,CeedScalar *v11, 549d7b241e6Sjeremylt CeedScalar *v12,CeedScalar *v13, CeedScalar *v14,CeedScalar *v15, int *err) { 550d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 551d7b241e6Sjeremylt const CeedScalar **in; 552d7b241e6Sjeremylt *err = CeedCalloc(16, &in); 553d7b241e6Sjeremylt if (*err) return; 554d7b241e6Sjeremylt in[0] = u; 555d7b241e6Sjeremylt in[1] = u1; 556d7b241e6Sjeremylt in[2] = u2; 557d7b241e6Sjeremylt in[3] = u3; 558d7b241e6Sjeremylt in[4] = u4; 559d7b241e6Sjeremylt in[5] = u5; 560d7b241e6Sjeremylt in[6] = u6; 561d7b241e6Sjeremylt in[7] = u7; 562d7b241e6Sjeremylt in[8] = u8; 563d7b241e6Sjeremylt in[9] = u9; 564d7b241e6Sjeremylt in[10] = u10; 565d7b241e6Sjeremylt in[11] = u11; 566d7b241e6Sjeremylt in[12] = u12; 567d7b241e6Sjeremylt in[13] = u13; 568d7b241e6Sjeremylt in[14] = u14; 569d7b241e6Sjeremylt in[15] = u15; 570d7b241e6Sjeremylt CeedScalar **out; 571d7b241e6Sjeremylt *err = CeedCalloc(16, &out); 572d7b241e6Sjeremylt if (*err) return; 573d7b241e6Sjeremylt out[0] = v; 574d7b241e6Sjeremylt out[1] = v1; 575d7b241e6Sjeremylt out[2] = v2; 576d7b241e6Sjeremylt out[3] = v3; 577d7b241e6Sjeremylt out[4] = v4; 578d7b241e6Sjeremylt out[5] = v5; 579d7b241e6Sjeremylt out[6] = v6; 580d7b241e6Sjeremylt out[7] = v7; 581d7b241e6Sjeremylt out[8] = v8; 582d7b241e6Sjeremylt out[9] = v9; 583d7b241e6Sjeremylt out[10] = v10; 584d7b241e6Sjeremylt out[11] = v11; 585d7b241e6Sjeremylt out[12] = v12; 586d7b241e6Sjeremylt out[13] = v13; 587d7b241e6Sjeremylt out[14] = v14; 588d7b241e6Sjeremylt out[15] = v15; 589d7b241e6Sjeremylt *err = CeedQFunctionApply(qf_, *Q, (const CeedScalar * const*)in, out); 590d7b241e6Sjeremylt if (*err) return; 591d7b241e6Sjeremylt 592d7b241e6Sjeremylt *err = CeedFree(&in); 593d7b241e6Sjeremylt if (*err) return; 594d7b241e6Sjeremylt *err = CeedFree(&out); 595d7b241e6Sjeremylt } 596d7b241e6Sjeremylt 597d7b241e6Sjeremylt #define fCeedQFunctionDestroy \ 598d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctiondestroy,CEEDQFUNCTIONDESTROY) 599d7b241e6Sjeremylt void fCeedQFunctionDestroy(int *qf, int *err) { 600d7b241e6Sjeremylt CeedFree(&CeedQFunction_dict[*qf]->ctx); 601d7b241e6Sjeremylt *err = CeedQFunctionDestroy(&CeedQFunction_dict[*qf]); 602d7b241e6Sjeremylt 603d7b241e6Sjeremylt if (*err) return; 604d7b241e6Sjeremylt CeedQFunction_n--; 605d7b241e6Sjeremylt if (CeedQFunction_n == 0) { 606d7b241e6Sjeremylt *err = CeedFree(&CeedQFunction_dict); 607d7b241e6Sjeremylt CeedQFunction_count = 0; 608d7b241e6Sjeremylt CeedQFunction_count_max = 0; 609d7b241e6Sjeremylt } 610d7b241e6Sjeremylt } 611d7b241e6Sjeremylt 612d7b241e6Sjeremylt static CeedOperator *CeedOperator_dict = NULL; 613d7b241e6Sjeremylt static int CeedOperator_count = 0; 614d7b241e6Sjeremylt static int CeedOperator_n = 0; 615d7b241e6Sjeremylt static int CeedOperator_count_max = 0; 616d7b241e6Sjeremylt 617d7b241e6Sjeremylt #define fCeedOperatorCreate \ 618d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorcreate, CEEDOPERATORCREATE) 619d7b241e6Sjeremylt void fCeedOperatorCreate(int* ceed, 620d7b241e6Sjeremylt int* qf, int* dqf, int* dqfT, int *op, int *err) { 621d7b241e6Sjeremylt if (CeedOperator_count == CeedOperator_count_max) 622d7b241e6Sjeremylt CeedOperator_count_max += CeedOperator_count_max/2 + 1, 623d7b241e6Sjeremylt CeedOperator_dict = 624d7b241e6Sjeremylt realloc(CeedOperator_dict, sizeof(CeedOperator)*CeedOperator_count_max); 625d7b241e6Sjeremylt 626d7b241e6Sjeremylt CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 627d7b241e6Sjeremylt 628d7b241e6Sjeremylt CeedQFunction dqf_ = NULL, dqfT_ = NULL; 629d7b241e6Sjeremylt if (*dqf != FORTRAN_NULL) dqf_ = CeedQFunction_dict[*dqf ]; 630d7b241e6Sjeremylt if (*dqfT != FORTRAN_NULL) dqfT_ = CeedQFunction_dict[*dqfT]; 631d7b241e6Sjeremylt 632d7b241e6Sjeremylt *err = CeedOperatorCreate(Ceed_dict[*ceed], CeedQFunction_dict[*qf], dqf_, 633d7b241e6Sjeremylt dqfT_, op_); 634d7b241e6Sjeremylt if (*err) return; 635d7b241e6Sjeremylt *op = CeedOperator_count++; 636d7b241e6Sjeremylt CeedOperator_n++; 637d7b241e6Sjeremylt } 638d7b241e6Sjeremylt 639d7b241e6Sjeremylt #define fCeedOperatorSetField \ 640d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorsetfield,CEEDOPERATORSETFIELD) 641d7b241e6Sjeremylt void fCeedOperatorSetField(int *op, const char *fieldname, 642c2ae719dSJed Brown int *r, int *b, int *v, int *err, 643c2ae719dSJed Brown fortran_charlen_t fieldname_len) { 644c2ae719dSJed Brown FIX_STRING(fieldname); 645d7b241e6Sjeremylt CeedElemRestriction r_; 646d7b241e6Sjeremylt CeedBasis b_; 647d7b241e6Sjeremylt CeedVector v_; 648d7b241e6Sjeremylt 649d7b241e6Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 650d7b241e6Sjeremylt 651d7b241e6Sjeremylt if (*r == FORTRAN_NULL) { 652d7b241e6Sjeremylt r_ = NULL; 653d7b241e6Sjeremylt } else { 654d7b241e6Sjeremylt r_ = CeedElemRestriction_dict[*r]; 655d7b241e6Sjeremylt } 656d7b241e6Sjeremylt if (*b == FORTRAN_NULL) { 657d7b241e6Sjeremylt b_ = NULL; 658783c99b3SValeria Barra } else if (*b == FORTRAN_BASIS_COLLOCATED) { 659783c99b3SValeria Barra b_ = CEED_BASIS_COLLOCATED; 660d7b241e6Sjeremylt } else { 661d7b241e6Sjeremylt b_ = CeedBasis_dict[*b]; 662d7b241e6Sjeremylt } 663d7b241e6Sjeremylt if (*v == FORTRAN_NULL) { 664d7b241e6Sjeremylt v_ = NULL; 665d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_ACTIVE) { 666d7b241e6Sjeremylt v_ = CEED_VECTOR_ACTIVE; 667d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_NONE) { 668d7b241e6Sjeremylt v_ = CEED_VECTOR_NONE; 669d7b241e6Sjeremylt } else { 670d7b241e6Sjeremylt v_ = CeedVector_dict[*v]; 671d7b241e6Sjeremylt } 672d7b241e6Sjeremylt 673c2ae719dSJed Brown *err = CeedOperatorSetField(op_, fieldname_c, r_, b_, v_); 674d7b241e6Sjeremylt } 675d7b241e6Sjeremylt 676d7b241e6Sjeremylt #define fCeedOperatorApply FORTRAN_NAME(ceedoperatorapply, CEEDOPERATORAPPLY) 677d7b241e6Sjeremylt void fCeedOperatorApply(int *op, int *ustatevec, 678d7b241e6Sjeremylt int *resvec, int *rqst, int *err) { 67912bf2c11SJed Brown CeedVector ustatevec_ = *ustatevec == FORTRAN_NULL 68012bf2c11SJed Brown ? NULL : CeedVector_dict[*ustatevec]; 68112bf2c11SJed Brown CeedVector resvec_ = *resvec == FORTRAN_NULL 68212bf2c11SJed Brown ? NULL : CeedVector_dict[*resvec]; 683d7b241e6Sjeremylt 684d7b241e6Sjeremylt int createRequest = 1; 685d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 686d7b241e6Sjeremylt if (*rqst == -1 || *rqst == -2) { 687d7b241e6Sjeremylt createRequest = 0; 688d7b241e6Sjeremylt } 689d7b241e6Sjeremylt 690d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 691d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 692d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 693d7b241e6Sjeremylt } 694d7b241e6Sjeremylt 695d7b241e6Sjeremylt CeedRequest *rqst_; 696d7b241e6Sjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 697d7b241e6Sjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 698d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 699d7b241e6Sjeremylt 700d7b241e6Sjeremylt *err = CeedOperatorApply(CeedOperator_dict[*op], 70112bf2c11SJed Brown ustatevec_, resvec_, rqst_); 702d7b241e6Sjeremylt if (*err) return; 703d7b241e6Sjeremylt if (createRequest) { 704d7b241e6Sjeremylt *rqst = CeedRequest_count++; 705d7b241e6Sjeremylt CeedRequest_n++; 706d7b241e6Sjeremylt } 707d7b241e6Sjeremylt } 708d7b241e6Sjeremylt 709d7b241e6Sjeremylt #define fCeedOperatorApplyJacobian \ 710d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorapplyjacobian, CEEDOPERATORAPPLYJACOBIAN) 711d7b241e6Sjeremylt void fCeedOperatorApplyJacobian(int *op, int *qdatavec, int *ustatevec, 712d7b241e6Sjeremylt int *dustatevec, int *dresvec, int *rqst, int *err) { 713d7b241e6Sjeremylt // TODO Uncomment this when CeedOperatorApplyJacobian is implemented 714d7b241e6Sjeremylt // *err = CeedOperatorApplyJacobian(CeedOperator_dict[*op], CeedVector_dict[*qdatavec], 715d7b241e6Sjeremylt // CeedVector_dict[*ustatevec], CeedVector_dict[*dustatevec], 716d7b241e6Sjeremylt // CeedVector_dict[*dresvec], &CeedRequest_dict[*rqst]); 717d7b241e6Sjeremylt } 718d7b241e6Sjeremylt 719d7b241e6Sjeremylt #define fCeedOperatorDestroy \ 720d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatordestroy, CEEDOPERATORDESTROY) 721d7b241e6Sjeremylt void fCeedOperatorDestroy(int *op, int *err) { 722d7b241e6Sjeremylt *err = CeedOperatorDestroy(&CeedOperator_dict[*op]); 723d7b241e6Sjeremylt if (*err) return; 724d7b241e6Sjeremylt CeedOperator_n--; 725d7b241e6Sjeremylt if (CeedOperator_n == 0) { 726d7b241e6Sjeremylt *err = CeedFree(&CeedOperator_dict); 727d7b241e6Sjeremylt CeedOperator_count = 0; 728d7b241e6Sjeremylt CeedOperator_count_max = 0; 729d7b241e6Sjeremylt } 730d7b241e6Sjeremylt } 731