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 28*523b8ea0Sjeremylt #define FORTRAN_STRIDES_BACKEND -1 29783c99b3SValeria Barra #define FORTRAN_BASIS_COLLOCATED -1 30d7b241e6Sjeremylt #define FORTRAN_VECTOR_ACTIVE -1 31d7b241e6Sjeremylt #define FORTRAN_VECTOR_NONE -2 32442e7f0bSjeremylt #define FORTRAN_QFUNCTION_NONE -1 33d7b241e6Sjeremylt 34d7b241e6Sjeremylt static Ceed *Ceed_dict = NULL; 35d7b241e6Sjeremylt static int Ceed_count = 0; 36d7b241e6Sjeremylt static int Ceed_n = 0; 37d7b241e6Sjeremylt static int Ceed_count_max = 0; 38d7b241e6Sjeremylt 39c2ae719dSJed Brown // This test should actually be for the gfortran version, but we don't currently 40c2ae719dSJed Brown // have a configure system to determine that (TODO). At present, this will use 41c2ae719dSJed Brown // the smaller integer when run with clang+gfortran=8, for example. (That is 42c2ae719dSJed Brown // sketchy, but will likely work for users that don't have huge character 43c2ae719dSJed Brown // strings.) 44c2ae719dSJed Brown #if __GNUC__ >= 8 45c2ae719dSJed Brown typedef size_t fortran_charlen_t; 46c2ae719dSJed Brown #else 47c2ae719dSJed Brown typedef int fortran_charlen_t; 48c2ae719dSJed Brown #endif 49c2ae719dSJed Brown 50c2ae719dSJed Brown #define Splice(a, b) a ## b 51c2ae719dSJed Brown 52c2ae719dSJed Brown // Fortran strings are generally unterminated and the length is passed as an 53c2ae719dSJed Brown // extra argument after all the normal arguments. Some compilers (I only know 54c2ae719dSJed Brown // of Windows) place the length argument immediately after the string parameter 55c2ae719dSJed Brown // (TODO). 56c2ae719dSJed Brown // 57c2ae719dSJed Brown // We can't just NULL-terminate the string in-place because that could overwrite 58c2ae719dSJed Brown // other strings or attempt to write to read-only memory. This macro allocates 59c2ae719dSJed Brown // a string to hold the null-terminated version of the string that C expects. 60c2ae719dSJed Brown #define FIX_STRING(stringname) \ 61c2ae719dSJed Brown char Splice(stringname, _c)[1024]; \ 62c2ae719dSJed Brown if (Splice(stringname, _len) > 1023) \ 63c2ae719dSJed Brown CeedError(NULL, 1, "Fortran string length too long %zd", (size_t)Splice(stringname, _len)); \ 64c2ae719dSJed Brown strncpy(Splice(stringname, _c), stringname, Splice(stringname, _len)); \ 65c2ae719dSJed Brown Splice(stringname, _c)[Splice(stringname, _len)] = 0; \ 66c2ae719dSJed Brown 67d7b241e6Sjeremylt #define fCeedInit FORTRAN_NAME(ceedinit,CEEDINIT) 68c2ae719dSJed Brown void fCeedInit(const char *resource, int *ceed, int *err, 69c2ae719dSJed Brown fortran_charlen_t resource_len) { 70c2ae719dSJed Brown FIX_STRING(resource); 71d7b241e6Sjeremylt if (Ceed_count == Ceed_count_max) { 72d7b241e6Sjeremylt Ceed_count_max += Ceed_count_max/2 + 1; 73d7b241e6Sjeremylt CeedRealloc(Ceed_count_max, &Ceed_dict); 74d7b241e6Sjeremylt } 75d7b241e6Sjeremylt 76d7b241e6Sjeremylt Ceed *ceed_ = &Ceed_dict[Ceed_count]; 77c2ae719dSJed Brown *err = CeedInit(resource_c, ceed_); 78d7b241e6Sjeremylt 79d7b241e6Sjeremylt if (*err == 0) { 80d7b241e6Sjeremylt *ceed = Ceed_count++; 81d7b241e6Sjeremylt Ceed_n++; 82d7b241e6Sjeremylt } 83d7b241e6Sjeremylt } 84d7b241e6Sjeremylt 85c907536fSjeremylt #define fCeedGetPreferredMemType \ 86c907536fSjeremylt FORTRAN_NAME(ceedgetpreferredmemtype,CEEDGETPREFERREDMEMTYPE) 87c907536fSjeremylt void fCeedGetPreferredMemType(int *ceed, int *type, int *err) { 88c907536fSjeremylt *err = CeedGetPreferredMemType(Ceed_dict[*ceed], (CeedMemType *)type); 89c907536fSjeremylt } 90c907536fSjeremylt 91d7b241e6Sjeremylt #define fCeedDestroy FORTRAN_NAME(ceeddestroy,CEEDDESTROY) 92d7b241e6Sjeremylt void fCeedDestroy(int *ceed, int *err) { 93d7b241e6Sjeremylt *err = CeedDestroy(&Ceed_dict[*ceed]); 94d7b241e6Sjeremylt 95d7b241e6Sjeremylt if (*err == 0) { 96d7b241e6Sjeremylt Ceed_n--; 97d7b241e6Sjeremylt if (Ceed_n == 0) { 98d7b241e6Sjeremylt CeedFree(&Ceed_dict); 99d7b241e6Sjeremylt Ceed_count = 0; 100d7b241e6Sjeremylt Ceed_count_max = 0; 101d7b241e6Sjeremylt } 102d7b241e6Sjeremylt } 103d7b241e6Sjeremylt } 104d7b241e6Sjeremylt 105d7b241e6Sjeremylt static CeedVector *CeedVector_dict = NULL; 106d7b241e6Sjeremylt static int CeedVector_count = 0; 107d7b241e6Sjeremylt static int CeedVector_n = 0; 108d7b241e6Sjeremylt static int CeedVector_count_max = 0; 109d7b241e6Sjeremylt 110d7b241e6Sjeremylt #define fCeedVectorCreate FORTRAN_NAME(ceedvectorcreate,CEEDVECTORCREATE) 111d7b241e6Sjeremylt void fCeedVectorCreate(int *ceed, int *length, int *vec, int *err) { 112d7b241e6Sjeremylt if (CeedVector_count == CeedVector_count_max) { 113d7b241e6Sjeremylt CeedVector_count_max += CeedVector_count_max/2 + 1; 114d7b241e6Sjeremylt CeedRealloc(CeedVector_count_max, &CeedVector_dict); 115d7b241e6Sjeremylt } 116d7b241e6Sjeremylt 117d7b241e6Sjeremylt CeedVector *vec_ = &CeedVector_dict[CeedVector_count]; 118d7b241e6Sjeremylt *err = CeedVectorCreate(Ceed_dict[*ceed], *length, vec_); 119d7b241e6Sjeremylt 120d7b241e6Sjeremylt if (*err == 0) { 121d7b241e6Sjeremylt *vec = CeedVector_count++; 122d7b241e6Sjeremylt CeedVector_n++; 123d7b241e6Sjeremylt } 124d7b241e6Sjeremylt } 125d7b241e6Sjeremylt 126d7b241e6Sjeremylt #define fCeedVectorSetArray FORTRAN_NAME(ceedvectorsetarray,CEEDVECTORSETARRAY) 127d7b241e6Sjeremylt void fCeedVectorSetArray(int *vec, int *memtype, int *copymode, 128c8b9fe72Sjeremylt CeedScalar *array, int64_t *offset, int *err) { 129c8b9fe72Sjeremylt *err = CeedVectorSetArray(CeedVector_dict[*vec], *memtype, *copymode, 130c8b9fe72Sjeremylt (CeedScalar *)(array + *offset)); 131d7b241e6Sjeremylt } 132d7b241e6Sjeremylt 13354540941SJeremy L Thompson #define fCeedVectorSyncArray FORTRAN_NAME(ceedvectorsyncarray,CEEDVECTORSYNCARRAY) 13454540941SJeremy L Thompson void fCeedVectorSyncArray(int *vec, int *memtype, int *err) { 13554540941SJeremy L Thompson *err = CeedVectorSyncArray(CeedVector_dict[*vec], *memtype); 13654540941SJeremy L Thompson } 13754540941SJeremy L Thompson 138d7b241e6Sjeremylt #define fCeedVectorSetValue FORTRAN_NAME(ceedvectorsetvalue,CEEDVECTORSETVALUE) 139d7b241e6Sjeremylt void fCeedVectorSetValue(int *vec, CeedScalar *value, int *err) { 140d7b241e6Sjeremylt *err = CeedVectorSetValue(CeedVector_dict[*vec], *value); 141d7b241e6Sjeremylt } 142d7b241e6Sjeremylt 143d7b241e6Sjeremylt #define fCeedVectorGetArray FORTRAN_NAME(ceedvectorgetarray,CEEDVECTORGETARRAY) 144a8de75f0Sjeremylt void fCeedVectorGetArray(int *vec, int *memtype, CeedScalar *array, 145a8de75f0Sjeremylt int64_t *offset, 146cbfd6a90SThilina Rathnayake int *err) { 147d7b241e6Sjeremylt CeedScalar *b; 148d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 149d7b241e6Sjeremylt *err = CeedVectorGetArray(vec_, *memtype, &b); 150cbfd6a90SThilina Rathnayake *offset = b - array; 151d7b241e6Sjeremylt } 152d7b241e6Sjeremylt 153d7b241e6Sjeremylt #define fCeedVectorGetArrayRead \ 154d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorgetarrayread,CEEDVECTORGETARRAYREAD) 155d7b241e6Sjeremylt void fCeedVectorGetArrayRead(int *vec, int *memtype, CeedScalar *array, 1563334379fSJed Brown int64_t *offset, int *err) { 157d7b241e6Sjeremylt const CeedScalar *b; 158d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 159cbfd6a90SThilina Rathnayake *err = CeedVectorGetArrayRead(vec_, *memtype, &b); 160cbfd6a90SThilina Rathnayake *offset = b - array; 161d7b241e6Sjeremylt } 162d7b241e6Sjeremylt 163d7b241e6Sjeremylt #define fCeedVectorRestoreArray \ 164d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearray,CEEDVECTORRESTOREARRAY) 1656456524eSJed Brown void fCeedVectorRestoreArray(int *vec, CeedScalar *array, 1666456524eSJed Brown int64_t *offset, int *err) { 167d7b241e6Sjeremylt *err = CeedVectorRestoreArray(CeedVector_dict[*vec], &array); 1686456524eSJed Brown *offset = 0; 169d7b241e6Sjeremylt } 170d7b241e6Sjeremylt 171d7b241e6Sjeremylt #define fCeedVectorRestoreArrayRead \ 172d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearrayread,CEEDVECTORRESTOREARRAYREAD) 1736456524eSJed Brown void fCeedVectorRestoreArrayRead(int *vec, const CeedScalar *array, 1746456524eSJed Brown int64_t *offset, int *err) { 175d7b241e6Sjeremylt *err = CeedVectorRestoreArrayRead(CeedVector_dict[*vec], &array); 1766456524eSJed Brown *offset = 0; 177d7b241e6Sjeremylt } 178d7b241e6Sjeremylt 179547d9b97Sjeremylt #define fCeedVectorNorm \ 180547d9b97Sjeremylt FORTRAN_NAME(ceedvectornorm,CEEDVECTORNORM) 181547d9b97Sjeremylt void fCeedVectorNorm(int *vec, int *type, CeedScalar *norm, int *err) { 182547d9b97Sjeremylt *err = CeedVectorNorm(CeedVector_dict[*vec], *type, norm); 183547d9b97Sjeremylt } 184547d9b97Sjeremylt 185d7b241e6Sjeremylt #define fCeedVectorView FORTRAN_NAME(ceedvectorview,CEEDVECTORVIEW) 186d7b241e6Sjeremylt void fCeedVectorView(int *vec, int *err) { 187d7b241e6Sjeremylt *err = CeedVectorView(CeedVector_dict[*vec], "%12.8f", stdout); 188d7b241e6Sjeremylt } 189d7b241e6Sjeremylt 190d7b241e6Sjeremylt #define fCeedVectorDestroy FORTRAN_NAME(ceedvectordestroy,CEEDVECTORDESTROY) 191d7b241e6Sjeremylt void fCeedVectorDestroy(int *vec, int *err) { 192d7b241e6Sjeremylt *err = CeedVectorDestroy(&CeedVector_dict[*vec]); 193d7b241e6Sjeremylt 194d7b241e6Sjeremylt if (*err == 0) { 195d7b241e6Sjeremylt CeedVector_n--; 196d7b241e6Sjeremylt if (CeedVector_n == 0) { 197d7b241e6Sjeremylt CeedFree(&CeedVector_dict); 198d7b241e6Sjeremylt CeedVector_count = 0; 199d7b241e6Sjeremylt CeedVector_count_max = 0; 200d7b241e6Sjeremylt } 201d7b241e6Sjeremylt } 202d7b241e6Sjeremylt } 203d7b241e6Sjeremylt 204d7b241e6Sjeremylt static CeedElemRestriction *CeedElemRestriction_dict = NULL; 205d7b241e6Sjeremylt static int CeedElemRestriction_count = 0; 206d7b241e6Sjeremylt static int CeedElemRestriction_n = 0; 207d7b241e6Sjeremylt static int CeedElemRestriction_count_max = 0; 208d7b241e6Sjeremylt 209d7b241e6Sjeremylt #define fCeedElemRestrictionCreate \ 210d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreate, CEEDELEMRESTRICTIONCREATE) 21161dbc9d2Sjeremylt void fCeedElemRestrictionCreate(int *ceed, int *imode, int *nelements, 2128795c945Sjeremylt int *esize, int *nnodes, int *ncomp, 2138795c945Sjeremylt int *memtype, int *copymode, const int *indices, 2148795c945Sjeremylt int *elemrestriction, int *err) { 215d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 216d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 217d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 218d7b241e6Sjeremylt } 219d7b241e6Sjeremylt 220d7b241e6Sjeremylt const int *indices_ = indices; 221d7b241e6Sjeremylt 222d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 223d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 22461dbc9d2Sjeremylt *err = CeedElemRestrictionCreate(Ceed_dict[*ceed], *imode, *nelements, *esize, 2258795c945Sjeremylt *nnodes, *ncomp, *memtype, *copymode, 2268795c945Sjeremylt indices_, elemrestriction_); 227d7b241e6Sjeremylt 228d7b241e6Sjeremylt if (*err == 0) { 229d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 230d7b241e6Sjeremylt CeedElemRestriction_n++; 231d7b241e6Sjeremylt } 232d7b241e6Sjeremylt } 233d7b241e6Sjeremylt 2347509a596Sjeremylt #define fCeedElemRestrictionCreateStrided \ 2357509a596Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreatestrided, CEEDELEMRESTRICTIONCREATESTRIDED) 2367509a596Sjeremylt void fCeedElemRestrictionCreateStrided(int *ceed, int *nelements, int *esize, 2377509a596Sjeremylt int *nnodes, int *ncomp, int *strides, 238d7b241e6Sjeremylt int *elemrestriction, int *err) { 239d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 240d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 241d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 242d7b241e6Sjeremylt } 243d7b241e6Sjeremylt 244d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 245d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 2467509a596Sjeremylt *err = CeedElemRestrictionCreateStrided(Ceed_dict[*ceed], *nelements, *esize, 247*523b8ea0Sjeremylt *nnodes, *ncomp, 248*523b8ea0Sjeremylt *strides == FORTRAN_STRIDES_BACKEND ? 249*523b8ea0Sjeremylt CEED_STRIDES_BACKEND : strides, 250*523b8ea0Sjeremylt elemrestriction_); 251d7b241e6Sjeremylt if (*err == 0) { 252d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 253d7b241e6Sjeremylt CeedElemRestriction_n++; 254d7b241e6Sjeremylt } 255d7b241e6Sjeremylt } 256d7b241e6Sjeremylt 257d7b241e6Sjeremylt #define fCeedElemRestrictionCreateBlocked \ 258d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateblocked,CEEDELEMRESTRICTIONCREATEBLOCKED) 25961dbc9d2Sjeremylt void fCeedElemRestrictionCreateBlocked(int *ceed, int *imode, int *nelements, 2608795c945Sjeremylt int *esize, int *blocksize, int *nnodes, 2618795c945Sjeremylt int *ncomp, int *mtype, int *cmode, 2628795c945Sjeremylt int *blkindices, int *elemrestriction, 2638795c945Sjeremylt int *err) { 264d7b241e6Sjeremylt 265d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 266d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 267d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 268d7b241e6Sjeremylt } 269d7b241e6Sjeremylt 270d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 271d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 27261dbc9d2Sjeremylt *err = CeedElemRestrictionCreateBlocked(Ceed_dict[*ceed], *imode, *nelements, 273a8d32208Sjeremylt *esize, *blocksize, *nnodes, *ncomp, 274a8d32208Sjeremylt *mtype, *cmode, blkindices, 275a8d32208Sjeremylt elemrestriction_); 276d7b241e6Sjeremylt 277d7b241e6Sjeremylt if (*err == 0) { 278d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 279d7b241e6Sjeremylt CeedElemRestriction_n++; 280d7b241e6Sjeremylt } 281d7b241e6Sjeremylt } 282d7b241e6Sjeremylt 2837509a596Sjeremylt #define fCeedElemRestrictionCreateBlockedStrided \ 2847509a596Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateblockedstrided, CEEDELEMRESTRICTIONCREATEBLOCKEDSTRIDED) 2857509a596Sjeremylt void fCeedElemRestrictionCreateBlockedStrided(int *ceed, int *nelements, 2867509a596Sjeremylt int *esize, int *blksize, 2877509a596Sjeremylt int *nnodes, int *ncomp, 2887509a596Sjeremylt int *strides, 2897509a596Sjeremylt int *elemrestriction, int *err) { 2907509a596Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 2917509a596Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 2927509a596Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 2937509a596Sjeremylt } 2947509a596Sjeremylt 2957509a596Sjeremylt CeedElemRestriction *elemrestriction_ = 2967509a596Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 2977509a596Sjeremylt *err = CeedElemRestrictionCreateBlockedStrided(Ceed_dict[*ceed], *nelements, 2987509a596Sjeremylt *esize, *blksize, *nnodes, *ncomp, strides, elemrestriction_); 2997509a596Sjeremylt if (*err == 0) { 3007509a596Sjeremylt *elemrestriction = CeedElemRestriction_count++; 3017509a596Sjeremylt CeedElemRestriction_n++; 3027509a596Sjeremylt } 3037509a596Sjeremylt } 3047509a596Sjeremylt 305d7b241e6Sjeremylt static CeedRequest *CeedRequest_dict = NULL; 306d7b241e6Sjeremylt static int CeedRequest_count = 0; 307d7b241e6Sjeremylt static int CeedRequest_n = 0; 308d7b241e6Sjeremylt static int CeedRequest_count_max = 0; 309d7b241e6Sjeremylt 310d7b241e6Sjeremylt #define fCeedElemRestrictionApply \ 311d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictionapply,CEEDELEMRESTRICTIONAPPLY) 312a8d32208Sjeremylt void fCeedElemRestrictionApply(int *elemr, int *tmode, int *uvec, int *ruvec, 313a8d32208Sjeremylt int *rqst, int *err) { 314d7b241e6Sjeremylt int createRequest = 1; 315d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 316d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) 317d7b241e6Sjeremylt createRequest = 0; 318d7b241e6Sjeremylt 319d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 320d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 321d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 322d7b241e6Sjeremylt } 323d7b241e6Sjeremylt 324d7b241e6Sjeremylt CeedRequest *rqst_; 325d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 326d7b241e6Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED ) rqst_ = CEED_REQUEST_ORDERED; 327d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 328d7b241e6Sjeremylt 329d7b241e6Sjeremylt *err = CeedElemRestrictionApply(CeedElemRestriction_dict[*elemr], *tmode, 330a8d32208Sjeremylt CeedVector_dict[*uvec], 3318795c945Sjeremylt CeedVector_dict[*ruvec], rqst_); 332d7b241e6Sjeremylt 333d7b241e6Sjeremylt if (*err == 0 && createRequest) { 334d7b241e6Sjeremylt *rqst = CeedRequest_count++; 335d7b241e6Sjeremylt CeedRequest_n++; 336d7b241e6Sjeremylt } 337d7b241e6Sjeremylt } 338d7b241e6Sjeremylt 339be9261b7Sjeremylt #define fCeedElemRestrictionApplyBlock \ 340be9261b7Sjeremylt FORTRAN_NAME(ceedelemrestrictionapplyblock,CEEDELEMRESTRICTIONAPPLYBLOCK) 341f90c8643Sjeremylt void fCeedElemRestrictionApplyBlock(int *elemr, int *block, int *tmode, 342be9261b7Sjeremylt int *uvec, int *ruvec, int *rqst, int *err) { 343be9261b7Sjeremylt int createRequest = 1; 344be9261b7Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 345be9261b7Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) 346be9261b7Sjeremylt createRequest = 0; 347be9261b7Sjeremylt 348be9261b7Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 349be9261b7Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 350be9261b7Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 351be9261b7Sjeremylt } 352be9261b7Sjeremylt 353be9261b7Sjeremylt CeedRequest *rqst_; 354be9261b7Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 355be9261b7Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED ) rqst_ = CEED_REQUEST_ORDERED; 356be9261b7Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 357be9261b7Sjeremylt 358be9261b7Sjeremylt *err = CeedElemRestrictionApplyBlock(CeedElemRestriction_dict[*elemr], *block, 359a8d32208Sjeremylt *tmode, CeedVector_dict[*uvec], 360be9261b7Sjeremylt CeedVector_dict[*ruvec], rqst_); 361be9261b7Sjeremylt 362be9261b7Sjeremylt if (*err == 0 && createRequest) { 363be9261b7Sjeremylt *rqst = CeedRequest_count++; 364be9261b7Sjeremylt CeedRequest_n++; 365be9261b7Sjeremylt } 366be9261b7Sjeremylt } 367be9261b7Sjeremylt 3681469ee4dSjeremylt #define fCeedElemRestrictionGetMultiplicity \ 3691469ee4dSjeremylt FORTRAN_NAME(ceedelemrestrictiongetmultiplicity,CEEDELEMRESTRICTIONGETMULTIPLICITY) 370a8d32208Sjeremylt void fCeedElemRestrictionGetMultiplicity(int *elemr, int *mult, int *err) { 3711469ee4dSjeremylt *err = CeedElemRestrictionGetMultiplicity(CeedElemRestriction_dict[*elemr], 372a8d32208Sjeremylt CeedVector_dict[*mult]); 3731469ee4dSjeremylt } 3741469ee4dSjeremylt 3755a86f033Sjeremylt #define fCeedElemRestrictionView \ 3765a86f033Sjeremylt FORTRAN_NAME(ceedelemrestrictionview,CEEDELEMRESTRICTIONVIEW) 3775a86f033Sjeremylt void fCeedElemRestrictionView(int *elemr, int *err) { 3785a86f033Sjeremylt *err = CeedElemRestrictionView(CeedElemRestriction_dict[*elemr], stdout); 3795a86f033Sjeremylt } 3801469ee4dSjeremylt 381d7b241e6Sjeremylt #define fCeedRequestWait FORTRAN_NAME(ceedrequestwait, CEEDREQUESTWAIT) 382d7b241e6Sjeremylt void fCeedRequestWait(int *rqst, int *err) { 383d7b241e6Sjeremylt // TODO Uncomment this once CeedRequestWait is implemented 384d7b241e6Sjeremylt //*err = CeedRequestWait(&CeedRequest_dict[*rqst]); 385d7b241e6Sjeremylt 386d7b241e6Sjeremylt if (*err == 0) { 387d7b241e6Sjeremylt CeedRequest_n--; 388d7b241e6Sjeremylt if (CeedRequest_n == 0) { 389d7b241e6Sjeremylt CeedFree(&CeedRequest_dict); 390d7b241e6Sjeremylt CeedRequest_count = 0; 391d7b241e6Sjeremylt CeedRequest_count_max = 0; 392d7b241e6Sjeremylt } 393d7b241e6Sjeremylt } 394d7b241e6Sjeremylt } 395d7b241e6Sjeremylt 396d7b241e6Sjeremylt #define fCeedElemRestrictionDestroy \ 397d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictiondestroy,CEEDELEMRESTRICTIONDESTROY) 398d7b241e6Sjeremylt void fCeedElemRestrictionDestroy(int *elem, int *err) { 399d7b241e6Sjeremylt *err = CeedElemRestrictionDestroy(&CeedElemRestriction_dict[*elem]); 400d7b241e6Sjeremylt 401d7b241e6Sjeremylt if (*err == 0) { 402d7b241e6Sjeremylt CeedElemRestriction_n--; 403d7b241e6Sjeremylt if (CeedElemRestriction_n == 0) { 404d7b241e6Sjeremylt CeedFree(&CeedElemRestriction_dict); 405d7b241e6Sjeremylt CeedElemRestriction_count = 0; 406d7b241e6Sjeremylt CeedElemRestriction_count_max = 0; 407d7b241e6Sjeremylt } 408d7b241e6Sjeremylt } 409d7b241e6Sjeremylt } 410d7b241e6Sjeremylt 411d7b241e6Sjeremylt static CeedBasis *CeedBasis_dict = NULL; 412d7b241e6Sjeremylt static int CeedBasis_count = 0; 413d7b241e6Sjeremylt static int CeedBasis_n = 0; 414d7b241e6Sjeremylt static int CeedBasis_count_max = 0; 415d7b241e6Sjeremylt 416d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1Lagrange \ 417d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1lagrange, CEEDBASISCREATETENSORH1LAGRANGE) 418d7b241e6Sjeremylt void fCeedBasisCreateTensorH1Lagrange(int *ceed, int *dim, 4198795c945Sjeremylt int *ncomp, int *P, int *Q, int *quadmode, 4208795c945Sjeremylt int *basis, int *err) { 421d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 422d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 423d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 424d7b241e6Sjeremylt } 425d7b241e6Sjeremylt 426a8de75f0Sjeremylt *err = CeedBasisCreateTensorH1Lagrange(Ceed_dict[*ceed], *dim, *ncomp, *P, *Q, 4278795c945Sjeremylt *quadmode, 4288795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 429d7b241e6Sjeremylt 430d7b241e6Sjeremylt if (*err == 0) { 431d7b241e6Sjeremylt *basis = CeedBasis_count++; 432d7b241e6Sjeremylt CeedBasis_n++; 433d7b241e6Sjeremylt } 434d7b241e6Sjeremylt } 435d7b241e6Sjeremylt 436d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1 \ 437d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1, CEEDBASISCREATETENSORH1) 438a8de75f0Sjeremylt void fCeedBasisCreateTensorH1(int *ceed, int *dim, int *ncomp, int *P1d, 4398795c945Sjeremylt int *Q1d, const CeedScalar *interp1d, 4408795c945Sjeremylt const CeedScalar *grad1d, 4418795c945Sjeremylt const CeedScalar *qref1d, 4428795c945Sjeremylt const CeedScalar *qweight1d, int *basis, 4438795c945Sjeremylt int *err) { 444d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 445d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 446d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 447d7b241e6Sjeremylt } 448d7b241e6Sjeremylt 449a8de75f0Sjeremylt *err = CeedBasisCreateTensorH1(Ceed_dict[*ceed], *dim, *ncomp, *P1d, *Q1d, 4508795c945Sjeremylt interp1d, grad1d, qref1d, qweight1d, 4518795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 452d7b241e6Sjeremylt 453d7b241e6Sjeremylt if (*err == 0) { 454d7b241e6Sjeremylt *basis = CeedBasis_count++; 455d7b241e6Sjeremylt CeedBasis_n++; 456d7b241e6Sjeremylt } 457d7b241e6Sjeremylt } 458d7b241e6Sjeremylt 459a8de75f0Sjeremylt #define fCeedBasisCreateH1 \ 460a8de75f0Sjeremylt FORTRAN_NAME(ceedbasiscreateh1, CEEDBASISCREATEH1) 4618795c945Sjeremylt void fCeedBasisCreateH1(int *ceed, int *topo, int *ncomp, int *nnodes, 4628795c945Sjeremylt int *nqpts, const CeedScalar *interp, 4638795c945Sjeremylt const CeedScalar *grad, const CeedScalar *qref, 4648795c945Sjeremylt const CeedScalar *qweight, int *basis, int *err) { 465a8de75f0Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 466a8de75f0Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 467a8de75f0Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 468a8de75f0Sjeremylt } 469a8de75f0Sjeremylt 4708795c945Sjeremylt *err = CeedBasisCreateH1(Ceed_dict[*ceed], *topo, *ncomp, *nnodes, *nqpts, 4718795c945Sjeremylt interp, grad, qref, qweight, 4728795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 473a8de75f0Sjeremylt 474a8de75f0Sjeremylt if (*err == 0) { 475a8de75f0Sjeremylt *basis = CeedBasis_count++; 476a8de75f0Sjeremylt CeedBasis_n++; 477a8de75f0Sjeremylt } 478a8de75f0Sjeremylt } 479a8de75f0Sjeremylt 480d7b241e6Sjeremylt #define fCeedBasisView FORTRAN_NAME(ceedbasisview, CEEDBASISVIEW) 481d7b241e6Sjeremylt void fCeedBasisView(int *basis, int *err) { 482d7b241e6Sjeremylt *err = CeedBasisView(CeedBasis_dict[*basis], stdout); 483d7b241e6Sjeremylt } 484d7b241e6Sjeremylt 485d7b241e6Sjeremylt #define fCeedQRFactorization \ 486d7b241e6Sjeremylt FORTRAN_NAME(ceedqrfactorization, CEEDQRFACTORIZATION) 487a7bd39daSjeremylt void fCeedQRFactorization(int *ceed, CeedScalar *mat, CeedScalar *tau, int *m, 488a7bd39daSjeremylt int *n, int *err) { 489a7bd39daSjeremylt *err = CeedQRFactorization(Ceed_dict[*ceed], mat, tau, *m, *n); 490d7b241e6Sjeremylt } 491d7b241e6Sjeremylt 49252bfb9bbSJeremy L Thompson #define fCeedSymmetricSchurDecomposition \ 49352bfb9bbSJeremy L Thompson FORTRAN_NAME(ceedsymmetricschurdecomposition, CEEDSYMMETRICSCHURDECOMPOSITION) 49452bfb9bbSJeremy L Thompson void fCeedSymmetricSchurDecomposition(int *ceed, CeedScalar *mat, 49552bfb9bbSJeremy L Thompson CeedScalar *lambda, int *n, int *err) { 49652bfb9bbSJeremy L Thompson *err = CeedSymmetricSchurDecomposition(Ceed_dict[*ceed], mat, lambda, *n); 49752bfb9bbSJeremy L Thompson } 49852bfb9bbSJeremy L Thompson 49952bfb9bbSJeremy L Thompson #define fCeedSimultaneousDiagonalization \ 50052bfb9bbSJeremy L Thompson FORTRAN_NAME(ceedsimultaneousdiagonalization, CEEDSIMULTANEOUSDIAGONALIZATION) 50152bfb9bbSJeremy L Thompson void fCeedSimultaneousDiagonalization(int *ceed, CeedScalar *matA, 50252bfb9bbSJeremy L Thompson CeedScalar *matB, CeedScalar *x, 50352bfb9bbSJeremy L Thompson CeedScalar *lambda, int *n, int *err) { 50452bfb9bbSJeremy L Thompson *err = CeedSimultaneousDiagonalization(Ceed_dict[*ceed], matA, matB, x, 50552bfb9bbSJeremy L Thompson lambda, *n); 50652bfb9bbSJeremy L Thompson } 50752bfb9bbSJeremy L Thompson 508783c99b3SValeria Barra #define fCeedBasisGetCollocatedGrad \ 509783c99b3SValeria Barra FORTRAN_NAME(ceedbasisgetcollocatedgrad, CEEDBASISGETCOLLOCATEDGRAD) 510783c99b3SValeria Barra void fCeedBasisGetCollocatedGrad(int *basis, CeedScalar *colograd1d, 511d7b241e6Sjeremylt int *err) { 512783c99b3SValeria Barra *err = CeedBasisGetCollocatedGrad(CeedBasis_dict[*basis], colograd1d); 513d7b241e6Sjeremylt } 514d7b241e6Sjeremylt 515d7b241e6Sjeremylt #define fCeedBasisApply FORTRAN_NAME(ceedbasisapply, CEEDBASISAPPLY) 516d7b241e6Sjeremylt void fCeedBasisApply(int *basis, int *nelem, int *tmode, int *emode, 517aedaa0e5Sjeremylt int *u, int *v, int *err) { 518aedaa0e5Sjeremylt *err = CeedBasisApply(CeedBasis_dict[*basis], *nelem, *tmode, *emode, 519a7b7f929Sjeremylt *u==FORTRAN_VECTOR_NONE? 520a7b7f929Sjeremylt CEED_VECTOR_NONE:CeedVector_dict[*u], 521aedaa0e5Sjeremylt CeedVector_dict[*v]); 522d7b241e6Sjeremylt } 523d7b241e6Sjeremylt 524d7b241e6Sjeremylt #define fCeedBasisGetNumNodes \ 525d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumnodes, CEEDBASISGETNUMNODES) 526d7b241e6Sjeremylt void fCeedBasisGetNumNodes(int *basis, int *P, int *err) { 527d7b241e6Sjeremylt *err = CeedBasisGetNumNodes(CeedBasis_dict[*basis], P); 528d7b241e6Sjeremylt } 529d7b241e6Sjeremylt 530d7b241e6Sjeremylt #define fCeedBasisGetNumQuadraturePoints \ 531d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumquadraturepoints, CEEDBASISGETNUMQUADRATUREPOINTS) 532d7b241e6Sjeremylt void fCeedBasisGetNumQuadraturePoints(int *basis, int *Q, int *err) { 533d7b241e6Sjeremylt *err = CeedBasisGetNumQuadraturePoints(CeedBasis_dict[*basis], Q); 534d7b241e6Sjeremylt } 535d7b241e6Sjeremylt 536d7b241e6Sjeremylt #define fCeedBasisDestroy FORTRAN_NAME(ceedbasisdestroy,CEEDBASISDESTROY) 537d7b241e6Sjeremylt void fCeedBasisDestroy(int *basis, int *err) { 538d7b241e6Sjeremylt *err = CeedBasisDestroy(&CeedBasis_dict[*basis]); 539d7b241e6Sjeremylt 540d7b241e6Sjeremylt if (*err == 0) { 541d7b241e6Sjeremylt CeedBasis_n--; 542d7b241e6Sjeremylt if (CeedBasis_n == 0) { 543d7b241e6Sjeremylt CeedFree(&CeedBasis_dict); 544d7b241e6Sjeremylt CeedBasis_count = 0; 545d7b241e6Sjeremylt CeedBasis_count_max = 0; 546d7b241e6Sjeremylt } 547d7b241e6Sjeremylt } 548d7b241e6Sjeremylt } 549d7b241e6Sjeremylt 550d7b241e6Sjeremylt #define fCeedGaussQuadrature FORTRAN_NAME(ceedgaussquadrature, CEEDGAUSSQUADRATURE) 551d7b241e6Sjeremylt void fCeedGaussQuadrature(int *Q, CeedScalar *qref1d, CeedScalar *qweight1d, 552d7b241e6Sjeremylt int *err) { 553d7b241e6Sjeremylt *err = CeedGaussQuadrature(*Q, qref1d, qweight1d); 554d7b241e6Sjeremylt } 555d7b241e6Sjeremylt 556d7b241e6Sjeremylt #define fCeedLobattoQuadrature \ 557d7b241e6Sjeremylt FORTRAN_NAME(ceedlobattoquadrature, CEEDLOBATTOQUADRATURE) 558d7b241e6Sjeremylt void fCeedLobattoQuadrature(int *Q, CeedScalar *qref1d, CeedScalar *qweight1d, 559d7b241e6Sjeremylt int *err) { 560d7b241e6Sjeremylt *err = CeedLobattoQuadrature(*Q, qref1d, qweight1d); 561d7b241e6Sjeremylt } 562d7b241e6Sjeremylt 563d7b241e6Sjeremylt static CeedQFunction *CeedQFunction_dict = NULL; 564d7b241e6Sjeremylt static int CeedQFunction_count = 0; 565d7b241e6Sjeremylt static int CeedQFunction_n = 0; 566d7b241e6Sjeremylt static int CeedQFunction_count_max = 0; 567d7b241e6Sjeremylt 568d7b241e6Sjeremylt static int CeedQFunctionFortranStub(void *ctx, int nq, 5698795c945Sjeremylt const CeedScalar *const *u, 5708795c945Sjeremylt CeedScalar *const *v) { 571418fb8c2Sjeremylt fContext *fctx = ctx; 572d7b241e6Sjeremylt int ierr; 573d7b241e6Sjeremylt 5741e35832bSjeremylt CeedScalar *ctx_ = (CeedScalar *) fctx->innerctx; 5751e35832bSjeremylt fctx->f((void *)ctx_,&nq,u[0],u[1],u[2],u[3],u[4],u[5],u[6], 576418fb8c2Sjeremylt u[7],u[8],u[9],u[10],u[11],u[12],u[13],u[14],u[15], 577418fb8c2Sjeremylt v[0],v[1],v[2],v[3],v[4],v[5],v[6],v[7],v[8],v[9], 578418fb8c2Sjeremylt v[10],v[11],v[12],v[13],v[14],v[15],&ierr); 579d7b241e6Sjeremylt return ierr; 580d7b241e6Sjeremylt } 581d7b241e6Sjeremylt 582d7b241e6Sjeremylt #define fCeedQFunctionCreateInterior \ 583d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctioncreateinterior, CEEDQFUNCTIONCREATEINTERIOR) 584d7b241e6Sjeremylt void fCeedQFunctionCreateInterior(int *ceed, int *vlength, 585d7b241e6Sjeremylt void (*f)(void *ctx, int *nq, 586418fb8c2Sjeremylt const CeedScalar *u,const CeedScalar *u1, 587418fb8c2Sjeremylt const CeedScalar *u2,const CeedScalar *u3, 588418fb8c2Sjeremylt const CeedScalar *u4,const CeedScalar *u5, 589418fb8c2Sjeremylt const CeedScalar *u6,const CeedScalar *u7, 590418fb8c2Sjeremylt const CeedScalar *u8,const CeedScalar *u9, 591418fb8c2Sjeremylt const CeedScalar *u10,const CeedScalar *u11, 592418fb8c2Sjeremylt const CeedScalar *u12,const CeedScalar *u13, 593418fb8c2Sjeremylt const CeedScalar *u14,const CeedScalar *u15, 594418fb8c2Sjeremylt CeedScalar *v,CeedScalar *v1,CeedScalar *v2, 5958795c945Sjeremylt CeedScalar *v3,CeedScalar *v4, 5968795c945Sjeremylt CeedScalar *v5,CeedScalar *v6, 5978795c945Sjeremylt CeedScalar *v7,CeedScalar *v8, 5988795c945Sjeremylt CeedScalar *v9,CeedScalar *v10, 5998795c945Sjeremylt CeedScalar *v11,CeedScalar *v12, 6008795c945Sjeremylt CeedScalar *v13,CeedScalar *v14, 601418fb8c2Sjeremylt CeedScalar *v15,int *err), 602288c0443SJeremy L Thompson const char *source, int *qf, int *err, 603288c0443SJeremy L Thompson fortran_charlen_t source_len) { 604288c0443SJeremy L Thompson FIX_STRING(source); 605d7b241e6Sjeremylt if (CeedQFunction_count == CeedQFunction_count_max) { 606d7b241e6Sjeremylt CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 607d7b241e6Sjeremylt CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 608d7b241e6Sjeremylt } 609d7b241e6Sjeremylt 610d7b241e6Sjeremylt CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 611d7b241e6Sjeremylt *err = CeedQFunctionCreateInterior(Ceed_dict[*ceed], *vlength, 612288c0443SJeremy L Thompson CeedQFunctionFortranStub, source_c, qf_); 613d7b241e6Sjeremylt 614d7b241e6Sjeremylt if (*err == 0) { 615d7b241e6Sjeremylt *qf = CeedQFunction_count++; 616d7b241e6Sjeremylt CeedQFunction_n++; 617d7b241e6Sjeremylt } 618d7b241e6Sjeremylt 619418fb8c2Sjeremylt fContext *fctx; 620d7b241e6Sjeremylt *err = CeedMalloc(1, &fctx); 621d7b241e6Sjeremylt if (*err) return; 622418fb8c2Sjeremylt fctx->f = f; fctx->innerctx = NULL; fctx->innerctxsize = 0; 623d7b241e6Sjeremylt 624418fb8c2Sjeremylt *err = CeedQFunctionSetContext(*qf_, fctx, sizeof(fContext)); 625d7b241e6Sjeremylt 626418fb8c2Sjeremylt (*qf_)->fortranstatus = true; 627d7b241e6Sjeremylt } 628d7b241e6Sjeremylt 629288c0443SJeremy L Thompson #define fCeedQFunctionCreateInteriorByName \ 630288c0443SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncreateinteriorbyname, CEEDQFUNCTIONCREATEINTERIORBYNAME) 631288c0443SJeremy L Thompson void fCeedQFunctionCreateInteriorByName(int *ceed, const char *name, int *qf, 632288c0443SJeremy L Thompson int *err, fortran_charlen_t name_len) { 633288c0443SJeremy L Thompson FIX_STRING(name); 634288c0443SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 635288c0443SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 636288c0443SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 637288c0443SJeremy L Thompson } 638288c0443SJeremy L Thompson 639288c0443SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 640288c0443SJeremy L Thompson *err = CeedQFunctionCreateInteriorByName(Ceed_dict[*ceed], name_c, qf_); 641288c0443SJeremy L Thompson 642288c0443SJeremy L Thompson if (*err == 0) { 643288c0443SJeremy L Thompson *qf = CeedQFunction_count++; 644288c0443SJeremy L Thompson CeedQFunction_n++; 645288c0443SJeremy L Thompson } 646288c0443SJeremy L Thompson } 647288c0443SJeremy L Thompson 6480219ea01SJeremy L Thompson #define fCeedQFunctionCreateIdentity \ 6490219ea01SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncreateidentity, CEEDQFUNCTIONCREATEIDENTITY) 65060f77c51Sjeremylt void fCeedQFunctionCreateIdentity(int *ceed, int *size, int *inmode, 65160f77c51Sjeremylt int *outmode, int *qf, int *err) { 6520219ea01SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 6530219ea01SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 6540219ea01SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 6550219ea01SJeremy L Thompson } 6560219ea01SJeremy L Thompson 6570219ea01SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 65860f77c51Sjeremylt *err = CeedQFunctionCreateIdentity(Ceed_dict[*ceed], *size, *inmode, 65960f77c51Sjeremylt *outmode, qf_); 6600219ea01SJeremy L Thompson 6610219ea01SJeremy L Thompson if (*err == 0) { 6620219ea01SJeremy L Thompson *qf = CeedQFunction_count++; 6630219ea01SJeremy L Thompson CeedQFunction_n++; 6640219ea01SJeremy L Thompson } 6650219ea01SJeremy L Thompson } 6660219ea01SJeremy L Thompson 667d7b241e6Sjeremylt #define fCeedQFunctionAddInput \ 668d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionaddinput,CEEDQFUNCTIONADDINPUT) 669d7b241e6Sjeremylt void fCeedQFunctionAddInput(int *qf, const char *fieldname, 670c2ae719dSJed Brown CeedInt *ncomp, CeedEvalMode *emode, int *err, 671c2ae719dSJed Brown fortran_charlen_t fieldname_len) { 672c2ae719dSJed Brown FIX_STRING(fieldname); 673d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 674d7b241e6Sjeremylt 675c2ae719dSJed Brown *err = CeedQFunctionAddInput(qf_, fieldname_c, *ncomp, *emode); 676d7b241e6Sjeremylt } 677d7b241e6Sjeremylt 678d7b241e6Sjeremylt #define fCeedQFunctionAddOutput \ 679d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionaddoutput,CEEDQFUNCTIONADDOUTPUT) 680d7b241e6Sjeremylt void fCeedQFunctionAddOutput(int *qf, const char *fieldname, 681c2ae719dSJed Brown CeedInt *ncomp, CeedEvalMode *emode, int *err, 682c2ae719dSJed Brown fortran_charlen_t fieldname_len) { 683c2ae719dSJed Brown FIX_STRING(fieldname); 684d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 685d7b241e6Sjeremylt 686c2ae719dSJed Brown *err = CeedQFunctionAddOutput(qf_, fieldname_c, *ncomp, *emode); 687d7b241e6Sjeremylt } 688d7b241e6Sjeremylt 689418fb8c2Sjeremylt #define fCeedQFunctionSetContext \ 690418fb8c2Sjeremylt FORTRAN_NAME(ceedqfunctionsetcontext,CEEDQFUNCTIONSETCONTEXT) 691418fb8c2Sjeremylt void fCeedQFunctionSetContext(int *qf, CeedScalar *ctx, CeedInt *n, int *err) { 692418fb8c2Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 693418fb8c2Sjeremylt 694418fb8c2Sjeremylt fContext *fctx = qf_->ctx; 695418fb8c2Sjeremylt fctx->innerctx = ctx; 6961e35832bSjeremylt fctx->innerctxsize = ((size_t) *n)*sizeof(CeedScalar); 697418fb8c2Sjeremylt } 698418fb8c2Sjeremylt 69975affc3bSjeremylt #define fCeedQFunctionView \ 70075affc3bSjeremylt FORTRAN_NAME(ceedqfunctionview,CEEDQFUNCTIONVIEW) 70175affc3bSjeremylt void fCeedQFunctionView(int *qf, int *err) { 70275affc3bSjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 70375affc3bSjeremylt 70475affc3bSjeremylt *err = CeedQFunctionView(qf_, stdout); 70575affc3bSjeremylt } 70675affc3bSjeremylt 707d7b241e6Sjeremylt #define fCeedQFunctionApply \ 708d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionapply,CEEDQFUNCTIONAPPLY) 709d7b241e6Sjeremylt //TODO Need Fixing, double pointer 710d7b241e6Sjeremylt void fCeedQFunctionApply(int *qf, int *Q, 711aedaa0e5Sjeremylt int *u, int *u1, int *u2, int *u3, 712aedaa0e5Sjeremylt int *u4, int *u5, int *u6, int *u7, 713aedaa0e5Sjeremylt int *u8, int *u9, int *u10, int *u11, 714aedaa0e5Sjeremylt int *u12, int *u13, int *u14, int *u15, 715aedaa0e5Sjeremylt int *v, int *v1, int *v2, int *v3, 716aedaa0e5Sjeremylt int *v4, int *v5, int *v6, int *v7, 717aedaa0e5Sjeremylt int *v8, int *v9, int *v10, int *v11, 718aedaa0e5Sjeremylt int *v12, int *v13, int *v14, int *v15, int *err) { 719d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 720aedaa0e5Sjeremylt CeedVector *in; 721d7b241e6Sjeremylt *err = CeedCalloc(16, &in); 722d7b241e6Sjeremylt if (*err) return; 723aedaa0e5Sjeremylt in[0] = *u==FORTRAN_NULL?NULL:CeedVector_dict[*u]; 724aedaa0e5Sjeremylt in[1] = *u1==FORTRAN_NULL?NULL:CeedVector_dict[*u1]; 725aedaa0e5Sjeremylt in[2] = *u2==FORTRAN_NULL?NULL:CeedVector_dict[*u2]; 726aedaa0e5Sjeremylt in[3] = *u3==FORTRAN_NULL?NULL:CeedVector_dict[*u3]; 727aedaa0e5Sjeremylt in[4] = *u4==FORTRAN_NULL?NULL:CeedVector_dict[*u4]; 728aedaa0e5Sjeremylt in[5] = *u5==FORTRAN_NULL?NULL:CeedVector_dict[*u5]; 729aedaa0e5Sjeremylt in[6] = *u6==FORTRAN_NULL?NULL:CeedVector_dict[*u6]; 730aedaa0e5Sjeremylt in[7] = *u7==FORTRAN_NULL?NULL:CeedVector_dict[*u7]; 731aedaa0e5Sjeremylt in[8] = *u8==FORTRAN_NULL?NULL:CeedVector_dict[*u8]; 732aedaa0e5Sjeremylt in[9] = *u9==FORTRAN_NULL?NULL:CeedVector_dict[*u9]; 733aedaa0e5Sjeremylt in[10] = *u10==FORTRAN_NULL?NULL:CeedVector_dict[*u10]; 734aedaa0e5Sjeremylt in[11] = *u11==FORTRAN_NULL?NULL:CeedVector_dict[*u11]; 735aedaa0e5Sjeremylt in[12] = *u12==FORTRAN_NULL?NULL:CeedVector_dict[*u12]; 736aedaa0e5Sjeremylt in[13] = *u13==FORTRAN_NULL?NULL:CeedVector_dict[*u13]; 737aedaa0e5Sjeremylt in[14] = *u14==FORTRAN_NULL?NULL:CeedVector_dict[*u14]; 738aedaa0e5Sjeremylt in[15] = *u15==FORTRAN_NULL?NULL:CeedVector_dict[*u15]; 739aedaa0e5Sjeremylt CeedVector *out; 740d7b241e6Sjeremylt *err = CeedCalloc(16, &out); 741d7b241e6Sjeremylt if (*err) return; 742aedaa0e5Sjeremylt out[0] = *v==FORTRAN_NULL?NULL:CeedVector_dict[*v]; 743aedaa0e5Sjeremylt out[1] = *v1==FORTRAN_NULL?NULL:CeedVector_dict[*v1]; 744aedaa0e5Sjeremylt out[2] = *v2==FORTRAN_NULL?NULL:CeedVector_dict[*v2]; 745aedaa0e5Sjeremylt out[3] = *v3==FORTRAN_NULL?NULL:CeedVector_dict[*v3]; 746aedaa0e5Sjeremylt out[4] = *v4==FORTRAN_NULL?NULL:CeedVector_dict[*v4]; 747aedaa0e5Sjeremylt out[5] = *v5==FORTRAN_NULL?NULL:CeedVector_dict[*v5]; 748aedaa0e5Sjeremylt out[6] = *v6==FORTRAN_NULL?NULL:CeedVector_dict[*v6]; 749aedaa0e5Sjeremylt out[7] = *v7==FORTRAN_NULL?NULL:CeedVector_dict[*v7]; 750aedaa0e5Sjeremylt out[8] = *v8==FORTRAN_NULL?NULL:CeedVector_dict[*v8]; 751aedaa0e5Sjeremylt out[9] = *v9==FORTRAN_NULL?NULL:CeedVector_dict[*v9]; 752aedaa0e5Sjeremylt out[10] = *v10==FORTRAN_NULL?NULL:CeedVector_dict[*v10]; 753aedaa0e5Sjeremylt out[11] = *v11==FORTRAN_NULL?NULL:CeedVector_dict[*v11]; 754aedaa0e5Sjeremylt out[12] = *v12==FORTRAN_NULL?NULL:CeedVector_dict[*v12]; 755aedaa0e5Sjeremylt out[13] = *v13==FORTRAN_NULL?NULL:CeedVector_dict[*v13]; 756aedaa0e5Sjeremylt out[14] = *v14==FORTRAN_NULL?NULL:CeedVector_dict[*v14]; 757aedaa0e5Sjeremylt out[15] = *v15==FORTRAN_NULL?NULL:CeedVector_dict[*v15]; 758aedaa0e5Sjeremylt *err = CeedQFunctionApply(qf_, *Q, in, out); 759d7b241e6Sjeremylt if (*err) return; 760d7b241e6Sjeremylt 761d7b241e6Sjeremylt *err = CeedFree(&in); 762d7b241e6Sjeremylt if (*err) return; 763d7b241e6Sjeremylt *err = CeedFree(&out); 764d7b241e6Sjeremylt } 765d7b241e6Sjeremylt 766d7b241e6Sjeremylt #define fCeedQFunctionDestroy \ 767d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctiondestroy,CEEDQFUNCTIONDESTROY) 768d7b241e6Sjeremylt void fCeedQFunctionDestroy(int *qf, int *err) { 7691134a487Sjeremylt bool fstatus; 7701134a487Sjeremylt *err = CeedQFunctionGetFortranStatus(CeedQFunction_dict[*qf], &fstatus); 7711134a487Sjeremylt if (*err) return; 7721134a487Sjeremylt if (fstatus) { 773c4b814baSjeremylt fContext *fctx = CeedQFunction_dict[*qf]->ctx; 774c4b814baSjeremylt *err = CeedFree(&fctx); 775d7b241e6Sjeremylt if (*err) return; 7761134a487Sjeremylt } 777c4b814baSjeremylt 778c4b814baSjeremylt *err = CeedQFunctionDestroy(&CeedQFunction_dict[*qf]); 779c4b814baSjeremylt if (*err) return; 780c4b814baSjeremylt 781d7b241e6Sjeremylt CeedQFunction_n--; 782d7b241e6Sjeremylt if (CeedQFunction_n == 0) { 783d7b241e6Sjeremylt *err = CeedFree(&CeedQFunction_dict); 784d7b241e6Sjeremylt CeedQFunction_count = 0; 785d7b241e6Sjeremylt CeedQFunction_count_max = 0; 786d7b241e6Sjeremylt } 787d7b241e6Sjeremylt } 788d7b241e6Sjeremylt 789d7b241e6Sjeremylt static CeedOperator *CeedOperator_dict = NULL; 790d7b241e6Sjeremylt static int CeedOperator_count = 0; 791d7b241e6Sjeremylt static int CeedOperator_n = 0; 792d7b241e6Sjeremylt static int CeedOperator_count_max = 0; 793d7b241e6Sjeremylt 794d7b241e6Sjeremylt #define fCeedOperatorCreate \ 795d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorcreate, CEEDOPERATORCREATE) 796d7b241e6Sjeremylt void fCeedOperatorCreate(int *ceed, 797d7b241e6Sjeremylt int *qf, int *dqf, int *dqfT, int *op, int *err) { 798d7b241e6Sjeremylt if (CeedOperator_count == CeedOperator_count_max) 799d7b241e6Sjeremylt CeedOperator_count_max += CeedOperator_count_max/2 + 1, 8008795c945Sjeremylt CeedOperator_dict = realloc(CeedOperator_dict, 8018795c945Sjeremylt sizeof(CeedOperator)*CeedOperator_count_max); 802d7b241e6Sjeremylt 803d7b241e6Sjeremylt CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 804d7b241e6Sjeremylt 805442e7f0bSjeremylt CeedQFunction dqf_ = CEED_QFUNCTION_NONE, dqfT_ = CEED_QFUNCTION_NONE; 806442e7f0bSjeremylt if (*dqf != FORTRAN_QFUNCTION_NONE) dqf_ = CeedQFunction_dict[*dqf ]; 807442e7f0bSjeremylt if (*dqfT != FORTRAN_QFUNCTION_NONE) dqfT_ = CeedQFunction_dict[*dqfT]; 808d7b241e6Sjeremylt 809d7b241e6Sjeremylt *err = CeedOperatorCreate(Ceed_dict[*ceed], CeedQFunction_dict[*qf], dqf_, 810d7b241e6Sjeremylt dqfT_, op_); 811d7b241e6Sjeremylt if (*err) return; 812d7b241e6Sjeremylt *op = CeedOperator_count++; 813d7b241e6Sjeremylt CeedOperator_n++; 814d7b241e6Sjeremylt } 815d7b241e6Sjeremylt 81652d6035fSJeremy L Thompson #define fCeedCompositeOperatorCreate \ 81752d6035fSJeremy L Thompson FORTRAN_NAME(ceedcompositeoperatorcreate, CEEDCOMPOSITEOPERATORCREATE) 81852d6035fSJeremy L Thompson void fCeedCompositeOperatorCreate(int *ceed, int *op, int *err) { 81952d6035fSJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) 82052d6035fSJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1, 8218795c945Sjeremylt CeedOperator_dict = realloc(CeedOperator_dict, 8228795c945Sjeremylt sizeof(CeedOperator)*CeedOperator_count_max); 82352d6035fSJeremy L Thompson 82452d6035fSJeremy L Thompson CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 82552d6035fSJeremy L Thompson 82652d6035fSJeremy L Thompson *err = CeedCompositeOperatorCreate(Ceed_dict[*ceed], op_); 82752d6035fSJeremy L Thompson if (*err) return; 82852d6035fSJeremy L Thompson *op = CeedOperator_count++; 82952d6035fSJeremy L Thompson CeedOperator_n++; 83052d6035fSJeremy L Thompson } 83152d6035fSJeremy L Thompson 832d7b241e6Sjeremylt #define fCeedOperatorSetField \ 833d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorsetfield,CEEDOPERATORSETFIELD) 834a8d32208Sjeremylt void fCeedOperatorSetField(int *op, const char *fieldname, int *r, int *b, 835a8d32208Sjeremylt int *v, int *err, fortran_charlen_t fieldname_len) { 836c2ae719dSJed Brown FIX_STRING(fieldname); 837d7b241e6Sjeremylt CeedElemRestriction r_; 838d7b241e6Sjeremylt CeedBasis b_; 839d7b241e6Sjeremylt CeedVector v_; 840d7b241e6Sjeremylt 841d7b241e6Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 842d7b241e6Sjeremylt 843d7b241e6Sjeremylt if (*r == FORTRAN_NULL) { 844d7b241e6Sjeremylt r_ = NULL; 845d7b241e6Sjeremylt } else { 846d7b241e6Sjeremylt r_ = CeedElemRestriction_dict[*r]; 847d7b241e6Sjeremylt } 8484dccadb6Sjeremylt 849d7b241e6Sjeremylt if (*b == FORTRAN_NULL) { 850d7b241e6Sjeremylt b_ = NULL; 851783c99b3SValeria Barra } else if (*b == FORTRAN_BASIS_COLLOCATED) { 852783c99b3SValeria Barra b_ = CEED_BASIS_COLLOCATED; 853d7b241e6Sjeremylt } else { 854d7b241e6Sjeremylt b_ = CeedBasis_dict[*b]; 855d7b241e6Sjeremylt } 856d7b241e6Sjeremylt if (*v == FORTRAN_NULL) { 857d7b241e6Sjeremylt v_ = NULL; 858d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_ACTIVE) { 859d7b241e6Sjeremylt v_ = CEED_VECTOR_ACTIVE; 860d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_NONE) { 861d7b241e6Sjeremylt v_ = CEED_VECTOR_NONE; 862d7b241e6Sjeremylt } else { 863d7b241e6Sjeremylt v_ = CeedVector_dict[*v]; 864d7b241e6Sjeremylt } 865d7b241e6Sjeremylt 866a8d32208Sjeremylt *err = CeedOperatorSetField(op_, fieldname_c, r_, b_, v_); 867d7b241e6Sjeremylt } 868d7b241e6Sjeremylt 86952d6035fSJeremy L Thompson #define fCeedCompositeOperatorAddSub \ 87052d6035fSJeremy L Thompson FORTRAN_NAME(ceedcompositeoperatoraddsub, CEEDCOMPOSITEOPERATORADDSUB) 87152d6035fSJeremy L Thompson void fCeedCompositeOperatorAddSub(int *compositeop, int *subop, int *err) { 87252d6035fSJeremy L Thompson CeedOperator compositeop_ = CeedOperator_dict[*compositeop]; 87352d6035fSJeremy L Thompson CeedOperator subop_ = CeedOperator_dict[*subop]; 87452d6035fSJeremy L Thompson 87552d6035fSJeremy L Thompson *err = CeedCompositeOperatorAddSub(compositeop_, subop_); 87652d6035fSJeremy L Thompson if (*err) return; 87752d6035fSJeremy L Thompson } 87852d6035fSJeremy L Thompson 8791d102b48SJeremy L Thompson #define fCeedOperatorAssembleLinearQFunction FORTRAN_NAME(ceedoperatorassemblelinearqfunction, CEEDOPERATORASSEMBLELINEARQFUNCTION) 8801d102b48SJeremy L Thompson void fCeedOperatorAssembleLinearQFunction(int *op, int *assembledvec, 8811d102b48SJeremy L Thompson int *assembledrstr, int *rqst, int *err) { 8821d102b48SJeremy L Thompson // Vector 8831d102b48SJeremy L Thompson if (CeedVector_count == CeedVector_count_max) { 8841d102b48SJeremy L Thompson CeedVector_count_max += CeedVector_count_max/2 + 1; 8851d102b48SJeremy L Thompson CeedRealloc(CeedVector_count_max, &CeedVector_dict); 8861d102b48SJeremy L Thompson } 8871d102b48SJeremy L Thompson CeedVector *assembledvec_ = &CeedVector_dict[CeedVector_count]; 8881d102b48SJeremy L Thompson 8891d102b48SJeremy L Thompson // Restriction 8901d102b48SJeremy L Thompson if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 8911d102b48SJeremy L Thompson CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 8921d102b48SJeremy L Thompson CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 8931d102b48SJeremy L Thompson } 8941d102b48SJeremy L Thompson CeedElemRestriction *rstr_ = 8951d102b48SJeremy L Thompson &CeedElemRestriction_dict[CeedElemRestriction_count]; 8961d102b48SJeremy L Thompson 8971d102b48SJeremy L Thompson int createRequest = 1; 8981d102b48SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 8991d102b48SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 9001d102b48SJeremy L Thompson createRequest = 0; 9011d102b48SJeremy L Thompson } 9021d102b48SJeremy L Thompson 9031d102b48SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 9041d102b48SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max/2 + 1; 9051d102b48SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 9061d102b48SJeremy L Thompson } 9071d102b48SJeremy L Thompson 9081d102b48SJeremy L Thompson CeedRequest *rqst_; 9091d102b48SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 9101d102b48SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 9111d102b48SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 9121d102b48SJeremy L Thompson 9131d102b48SJeremy L Thompson *err = CeedOperatorAssembleLinearQFunction(CeedOperator_dict[*op], 9141d102b48SJeremy L Thompson assembledvec_, rstr_, rqst_); 9151d102b48SJeremy L Thompson if (*err) return; 9161d102b48SJeremy L Thompson if (createRequest) { 9171d102b48SJeremy L Thompson *rqst = CeedRequest_count++; 9181d102b48SJeremy L Thompson CeedRequest_n++; 9191d102b48SJeremy L Thompson } 9201d102b48SJeremy L Thompson 9211d102b48SJeremy L Thompson if (*err == 0) { 9221d102b48SJeremy L Thompson *assembledrstr = CeedElemRestriction_count++; 9231d102b48SJeremy L Thompson CeedElemRestriction_n++; 9241d102b48SJeremy L Thompson *assembledvec = CeedVector_count++; 9251d102b48SJeremy L Thompson CeedVector_n++; 9261d102b48SJeremy L Thompson } 9271d102b48SJeremy L Thompson } 9281d102b48SJeremy L Thompson 929b7ec98d8SJeremy L Thompson #define fCeedOperatorAssembleLinearDiagonal FORTRAN_NAME(ceedoperatorassemblelineardiagonal, CEEDOPERATORASSEMBLELINEARDIAGONAL) 930b7ec98d8SJeremy L Thompson void fCeedOperatorAssembleLinearDiagonal(int *op, int *assembledvec, 931b7ec98d8SJeremy L Thompson int *rqst, int *err) { 932b7ec98d8SJeremy L Thompson // Vector 933b7ec98d8SJeremy L Thompson if (CeedVector_count == CeedVector_count_max) { 934b7ec98d8SJeremy L Thompson CeedVector_count_max += CeedVector_count_max/2 + 1; 935b7ec98d8SJeremy L Thompson CeedRealloc(CeedVector_count_max, &CeedVector_dict); 936b7ec98d8SJeremy L Thompson } 937b7ec98d8SJeremy L Thompson CeedVector *assembledvec_ = &CeedVector_dict[CeedVector_count]; 938b7ec98d8SJeremy L Thompson 939b7ec98d8SJeremy L Thompson int createRequest = 1; 940b7ec98d8SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 941b7ec98d8SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 942b7ec98d8SJeremy L Thompson createRequest = 0; 943b7ec98d8SJeremy L Thompson } 944b7ec98d8SJeremy L Thompson 945b7ec98d8SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 946b7ec98d8SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max/2 + 1; 947b7ec98d8SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 948b7ec98d8SJeremy L Thompson } 949b7ec98d8SJeremy L Thompson 950b7ec98d8SJeremy L Thompson CeedRequest *rqst_; 951b7ec98d8SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 952b7ec98d8SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 953b7ec98d8SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 954b7ec98d8SJeremy L Thompson 955b7ec98d8SJeremy L Thompson *err = CeedOperatorAssembleLinearDiagonal(CeedOperator_dict[*op], 956b7ec98d8SJeremy L Thompson assembledvec_, rqst_); 957b7ec98d8SJeremy L Thompson if (*err) return; 958b7ec98d8SJeremy L Thompson if (createRequest) { 959b7ec98d8SJeremy L Thompson *rqst = CeedRequest_count++; 960b7ec98d8SJeremy L Thompson CeedRequest_n++; 961b7ec98d8SJeremy L Thompson } 962b7ec98d8SJeremy L Thompson 963b7ec98d8SJeremy L Thompson if (*err == 0) { 964b7ec98d8SJeremy L Thompson *assembledvec = CeedVector_count++; 965b7ec98d8SJeremy L Thompson CeedVector_n++; 966b7ec98d8SJeremy L Thompson } 967b7ec98d8SJeremy L Thompson } 968b7ec98d8SJeremy L Thompson 9692ebaca42Sjeremylt #define fCeedOperatorView \ 9702ebaca42Sjeremylt FORTRAN_NAME(ceedoperatorview,CEEDOPERATORVIEW) 9712ebaca42Sjeremylt void fCeedOperatorView(int *op, int *err) { 9722ebaca42Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 9732ebaca42Sjeremylt 9742ebaca42Sjeremylt *err = CeedOperatorView(op_, stdout); 9752ebaca42Sjeremylt } 9762ebaca42Sjeremylt 9773bd813ffSjeremylt #define fCeedOperatorCreateFDMElementInverse FORTRAN_NAME(ceedoperatorcreatefdmelementinverse, CEEDOPERATORCREATEFDMELEMENTINVERSE) 9783bd813ffSjeremylt void fCeedOperatorCreateFDMElementInverse(int *op, int *fdminv, 9793bd813ffSjeremylt int *rqst, int *err) { 9803bd813ffSjeremylt // Operator 9813bd813ffSjeremylt if (CeedOperator_count == CeedOperator_count_max) { 9823bd813ffSjeremylt CeedOperator_count_max += CeedOperator_count_max/2 + 1; 9833bd813ffSjeremylt CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 9843bd813ffSjeremylt } 9853bd813ffSjeremylt CeedOperator *fdminv_ = 9863bd813ffSjeremylt &CeedOperator_dict[CeedOperator_count]; 9873bd813ffSjeremylt 9883bd813ffSjeremylt int createRequest = 1; 9893bd813ffSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 9903bd813ffSjeremylt if (*rqst == -1 || *rqst == -2) { 9913bd813ffSjeremylt createRequest = 0; 9923bd813ffSjeremylt } 9933bd813ffSjeremylt 9943bd813ffSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 9953bd813ffSjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 9963bd813ffSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 9973bd813ffSjeremylt } 9983bd813ffSjeremylt 9993bd813ffSjeremylt CeedRequest *rqst_; 10003bd813ffSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 10013bd813ffSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 10023bd813ffSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 10033bd813ffSjeremylt 10043bd813ffSjeremylt *err = CeedOperatorCreateFDMElementInverse(CeedOperator_dict[*op], 10053bd813ffSjeremylt fdminv_, rqst_); 10063bd813ffSjeremylt if (*err) return; 10073bd813ffSjeremylt if (createRequest) { 10083bd813ffSjeremylt *rqst = CeedRequest_count++; 10093bd813ffSjeremylt CeedRequest_n++; 10103bd813ffSjeremylt } 10113bd813ffSjeremylt 10123bd813ffSjeremylt if (*err == 0) { 10133bd813ffSjeremylt *fdminv = CeedOperator_count++; 10143bd813ffSjeremylt CeedOperator_n++; 10153bd813ffSjeremylt } 10163bd813ffSjeremylt } 10173bd813ffSjeremylt 1018d7b241e6Sjeremylt #define fCeedOperatorApply FORTRAN_NAME(ceedoperatorapply, CEEDOPERATORAPPLY) 1019d7b241e6Sjeremylt void fCeedOperatorApply(int *op, int *ustatevec, 1020d7b241e6Sjeremylt int *resvec, int *rqst, int *err) { 1021e97ff134Sjeremylt CeedVector ustatevec_ = (*ustatevec == FORTRAN_NULL) ? 1022e97ff134Sjeremylt NULL : (*ustatevec == FORTRAN_VECTOR_NONE ? 1023e97ff134Sjeremylt CEED_VECTOR_NONE : CeedVector_dict[*ustatevec]); 1024e97ff134Sjeremylt CeedVector resvec_ = (*resvec == FORTRAN_NULL) ? 1025e97ff134Sjeremylt NULL : (*resvec == FORTRAN_VECTOR_NONE ? 1026e97ff134Sjeremylt CEED_VECTOR_NONE : CeedVector_dict[*resvec]); 1027d7b241e6Sjeremylt 1028d7b241e6Sjeremylt int createRequest = 1; 1029d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1030d7b241e6Sjeremylt if (*rqst == -1 || *rqst == -2) { 1031d7b241e6Sjeremylt createRequest = 0; 1032d7b241e6Sjeremylt } 1033d7b241e6Sjeremylt 1034d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1035d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 1036d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1037d7b241e6Sjeremylt } 1038d7b241e6Sjeremylt 1039d7b241e6Sjeremylt CeedRequest *rqst_; 1040d7b241e6Sjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1041d7b241e6Sjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1042d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1043d7b241e6Sjeremylt 1044d7b241e6Sjeremylt *err = CeedOperatorApply(CeedOperator_dict[*op], 104512bf2c11SJed Brown ustatevec_, resvec_, rqst_); 1046d7b241e6Sjeremylt if (*err) return; 1047d7b241e6Sjeremylt if (createRequest) { 1048d7b241e6Sjeremylt *rqst = CeedRequest_count++; 1049d7b241e6Sjeremylt CeedRequest_n++; 1050d7b241e6Sjeremylt } 1051d7b241e6Sjeremylt } 1052d7b241e6Sjeremylt 1053cae8b89aSjeremylt #define fCeedOperatorApplyAdd FORTRAN_NAME(ceedoperatorapplyadd, CEEDOPERATORAPPLYADD) 1054cae8b89aSjeremylt void fCeedOperatorApplyAdd(int *op, int *ustatevec, 1055cae8b89aSjeremylt int *resvec, int *rqst, int *err) { 1056cae8b89aSjeremylt CeedVector ustatevec_ = *ustatevec == FORTRAN_NULL 1057cae8b89aSjeremylt ? NULL : CeedVector_dict[*ustatevec]; 1058cae8b89aSjeremylt CeedVector resvec_ = *resvec == FORTRAN_NULL 1059cae8b89aSjeremylt ? NULL : CeedVector_dict[*resvec]; 1060cae8b89aSjeremylt 1061cae8b89aSjeremylt int createRequest = 1; 1062cae8b89aSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1063cae8b89aSjeremylt if (*rqst == -1 || *rqst == -2) { 1064cae8b89aSjeremylt createRequest = 0; 1065cae8b89aSjeremylt } 1066cae8b89aSjeremylt 1067cae8b89aSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1068cae8b89aSjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 1069cae8b89aSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1070cae8b89aSjeremylt } 1071cae8b89aSjeremylt 1072cae8b89aSjeremylt CeedRequest *rqst_; 1073cae8b89aSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1074cae8b89aSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1075cae8b89aSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1076cae8b89aSjeremylt 1077cae8b89aSjeremylt *err = CeedOperatorApplyAdd(CeedOperator_dict[*op], 1078cae8b89aSjeremylt ustatevec_, resvec_, rqst_); 1079cae8b89aSjeremylt if (*err) return; 1080cae8b89aSjeremylt if (createRequest) { 1081cae8b89aSjeremylt *rqst = CeedRequest_count++; 1082cae8b89aSjeremylt CeedRequest_n++; 1083cae8b89aSjeremylt } 1084cae8b89aSjeremylt } 1085cae8b89aSjeremylt 1086d7b241e6Sjeremylt #define fCeedOperatorApplyJacobian \ 1087d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorapplyjacobian, CEEDOPERATORAPPLYJACOBIAN) 1088d7b241e6Sjeremylt void fCeedOperatorApplyJacobian(int *op, int *qdatavec, int *ustatevec, 10898795c945Sjeremylt int *dustatevec, int *dresvec, int *rqst, 10908795c945Sjeremylt int *err) { 1091d7b241e6Sjeremylt // TODO Uncomment this when CeedOperatorApplyJacobian is implemented 1092d7b241e6Sjeremylt // *err = CeedOperatorApplyJacobian(CeedOperator_dict[*op], CeedVector_dict[*qdatavec], 1093d7b241e6Sjeremylt // CeedVector_dict[*ustatevec], CeedVector_dict[*dustatevec], 1094d7b241e6Sjeremylt // CeedVector_dict[*dresvec], &CeedRequest_dict[*rqst]); 1095d7b241e6Sjeremylt } 1096d7b241e6Sjeremylt 1097d7b241e6Sjeremylt #define fCeedOperatorDestroy \ 1098d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatordestroy, CEEDOPERATORDESTROY) 1099d7b241e6Sjeremylt void fCeedOperatorDestroy(int *op, int *err) { 1100d7b241e6Sjeremylt *err = CeedOperatorDestroy(&CeedOperator_dict[*op]); 1101d7b241e6Sjeremylt if (*err) return; 1102d7b241e6Sjeremylt CeedOperator_n--; 1103d7b241e6Sjeremylt if (CeedOperator_n == 0) { 1104d7b241e6Sjeremylt *err = CeedFree(&CeedOperator_dict); 1105d7b241e6Sjeremylt CeedOperator_count = 0; 1106d7b241e6Sjeremylt CeedOperator_count_max = 0; 1107d7b241e6Sjeremylt } 1108d7b241e6Sjeremylt } 1109