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 18ec3da8bcSJed Brown #include <ceed/ceed.h> 19ec3da8bcSJed Brown #include <ceed/backend.h> 203d576824SJeremy L Thompson #include <ceed-impl.h> 21d7b241e6Sjeremylt #include <ceed-fortran-name.h> 223d576824SJeremy L Thompson #include <stdbool.h> 233d576824SJeremy L Thompson #include <stdint.h> 243d576824SJeremy L Thompson #include <stdio.h> 25d7b241e6Sjeremylt #include <string.h> 26d7b241e6Sjeremylt 27d7b241e6Sjeremylt #define FORTRAN_REQUEST_IMMEDIATE -1 28d7b241e6Sjeremylt #define FORTRAN_REQUEST_ORDERED -2 29d7b241e6Sjeremylt #define FORTRAN_NULL -3 3015910d16Sjeremylt #define FORTRAN_STRIDES_BACKEND -4 3115910d16Sjeremylt #define FORTRAN_VECTOR_ACTIVE -5 3215910d16Sjeremylt #define FORTRAN_VECTOR_NONE -6 3315910d16Sjeremylt #define FORTRAN_ELEMRESTRICTION_NONE -7 3415910d16Sjeremylt #define FORTRAN_BASIS_COLLOCATED -8 3515910d16Sjeremylt #define FORTRAN_QFUNCTION_NONE -9 36d7b241e6Sjeremylt 37d7b241e6Sjeremylt static Ceed *Ceed_dict = NULL; 38d7b241e6Sjeremylt static int Ceed_count = 0; 39d7b241e6Sjeremylt static int Ceed_n = 0; 40d7b241e6Sjeremylt static int Ceed_count_max = 0; 41d7b241e6Sjeremylt 42c2ae719dSJed Brown // This test should actually be for the gfortran version, but we don't currently 43c2ae719dSJed Brown // have a configure system to determine that (TODO). At present, this will use 44c2ae719dSJed Brown // the smaller integer when run with clang+gfortran=8, for example. (That is 45c2ae719dSJed Brown // sketchy, but will likely work for users that don't have huge character 46c2ae719dSJed Brown // strings.) 47c2ae719dSJed Brown #if __GNUC__ >= 8 48c2ae719dSJed Brown typedef size_t fortran_charlen_t; 49c2ae719dSJed Brown #else 50c2ae719dSJed Brown typedef int fortran_charlen_t; 51c2ae719dSJed Brown #endif 52c2ae719dSJed Brown 53c2ae719dSJed Brown #define Splice(a, b) a ## b 54c2ae719dSJed Brown 55c2ae719dSJed Brown // Fortran strings are generally unterminated and the length is passed as an 56c2ae719dSJed Brown // extra argument after all the normal arguments. Some compilers (I only know 57c2ae719dSJed Brown // of Windows) place the length argument immediately after the string parameter 58c2ae719dSJed Brown // (TODO). 59c2ae719dSJed Brown // 60c2ae719dSJed Brown // We can't just NULL-terminate the string in-place because that could overwrite 61c2ae719dSJed Brown // other strings or attempt to write to read-only memory. This macro allocates 62c2ae719dSJed Brown // a string to hold the null-terminated version of the string that C expects. 63c2ae719dSJed Brown #define FIX_STRING(stringname) \ 64c2ae719dSJed Brown char Splice(stringname, _c)[1024]; \ 65c2ae719dSJed Brown if (Splice(stringname, _len) > 1023) \ 6645f1e315Sjeremylt *err = CeedError(NULL, 1, "Fortran string length too long %zd", (size_t)Splice(stringname, _len)); \ 67c2ae719dSJed Brown strncpy(Splice(stringname, _c), stringname, Splice(stringname, _len)); \ 68c2ae719dSJed Brown Splice(stringname, _c)[Splice(stringname, _len)] = 0; \ 69c2ae719dSJed Brown 70b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 71b00b43a6SJeremy L Thompson // Ceed 72b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 73d7b241e6Sjeremylt #define fCeedInit FORTRAN_NAME(ceedinit,CEEDINIT) 74*3b271f31SJed Brown CEED_EXTERN void fCeedInit(const char *resource, int *ceed, int *err, 75c2ae719dSJed Brown fortran_charlen_t resource_len) { 76c2ae719dSJed Brown FIX_STRING(resource); 77d7b241e6Sjeremylt if (Ceed_count == Ceed_count_max) { 78d7b241e6Sjeremylt Ceed_count_max += Ceed_count_max/2 + 1; 79d7b241e6Sjeremylt CeedRealloc(Ceed_count_max, &Ceed_dict); 80d7b241e6Sjeremylt } 81d7b241e6Sjeremylt 82d7b241e6Sjeremylt Ceed *ceed_ = &Ceed_dict[Ceed_count]; 83c2ae719dSJed Brown *err = CeedInit(resource_c, ceed_); 84d7b241e6Sjeremylt 85d7b241e6Sjeremylt if (*err == 0) { 86d7b241e6Sjeremylt *ceed = Ceed_count++; 87d7b241e6Sjeremylt Ceed_n++; 88d7b241e6Sjeremylt } 89d7b241e6Sjeremylt } 90d7b241e6Sjeremylt 919525855cSJeremy L Thompson #define fCeedIsDeterministic \ 929525855cSJeremy L Thompson FORTRAN_NAME(ceedisdeterministic,CEEDISDETERMINISTIC) 93*3b271f31SJed Brown CEED_EXTERN void fCeedIsDeterministic(int *ceed, int *is_deterministic, 94*3b271f31SJed Brown int *err) { 95d1d35e2fSjeremylt *err = CeedIsDeterministic(Ceed_dict[*ceed], (bool *)is_deterministic); 969525855cSJeremy L Thompson } 979525855cSJeremy L Thompson 98c907536fSjeremylt #define fCeedGetPreferredMemType \ 99c907536fSjeremylt FORTRAN_NAME(ceedgetpreferredmemtype,CEEDGETPREFERREDMEMTYPE) 100*3b271f31SJed Brown CEED_EXTERN void fCeedGetPreferredMemType(int *ceed, int *type, int *err) { 101c907536fSjeremylt *err = CeedGetPreferredMemType(Ceed_dict[*ceed], (CeedMemType *)type); 102c907536fSjeremylt } 103c907536fSjeremylt 1040a0da059Sjeremylt #define fCeedView FORTRAN_NAME(ceedview,CEEDVIEW) 105*3b271f31SJed Brown CEED_EXTERN void fCeedView(int *ceed, int *err) { 1060a0da059Sjeremylt *err = CeedView(Ceed_dict[*ceed], stdout); 1070a0da059Sjeremylt } 1080a0da059Sjeremylt 109d7b241e6Sjeremylt #define fCeedDestroy FORTRAN_NAME(ceeddestroy,CEEDDESTROY) 110*3b271f31SJed Brown CEED_EXTERN void fCeedDestroy(int *ceed, int *err) { 111b22696a8SJeremy L Thompson if (*ceed == FORTRAN_NULL) return; 112d7b241e6Sjeremylt *err = CeedDestroy(&Ceed_dict[*ceed]); 113d7b241e6Sjeremylt 114d7b241e6Sjeremylt if (*err == 0) { 11516595307SJeremy L Thompson *ceed = FORTRAN_NULL; 116d7b241e6Sjeremylt Ceed_n--; 117d7b241e6Sjeremylt if (Ceed_n == 0) { 118d7b241e6Sjeremylt CeedFree(&Ceed_dict); 119d7b241e6Sjeremylt Ceed_count = 0; 120d7b241e6Sjeremylt Ceed_count_max = 0; 121d7b241e6Sjeremylt } 122d7b241e6Sjeremylt } 123d7b241e6Sjeremylt } 124d7b241e6Sjeremylt 125b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 126b00b43a6SJeremy L Thompson // CeedVector 127b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 128d7b241e6Sjeremylt static CeedVector *CeedVector_dict = NULL; 129d7b241e6Sjeremylt static int CeedVector_count = 0; 130d7b241e6Sjeremylt static int CeedVector_n = 0; 131d7b241e6Sjeremylt static int CeedVector_count_max = 0; 132d7b241e6Sjeremylt 133d7b241e6Sjeremylt #define fCeedVectorCreate FORTRAN_NAME(ceedvectorcreate,CEEDVECTORCREATE) 134*3b271f31SJed Brown CEED_EXTERN void fCeedVectorCreate(int *ceed, int *length, int *vec, int *err) { 135d7b241e6Sjeremylt if (CeedVector_count == CeedVector_count_max) { 136d7b241e6Sjeremylt CeedVector_count_max += CeedVector_count_max/2 + 1; 137d7b241e6Sjeremylt CeedRealloc(CeedVector_count_max, &CeedVector_dict); 138d7b241e6Sjeremylt } 139d7b241e6Sjeremylt 140d7b241e6Sjeremylt CeedVector *vec_ = &CeedVector_dict[CeedVector_count]; 141d7b241e6Sjeremylt *err = CeedVectorCreate(Ceed_dict[*ceed], *length, vec_); 142d7b241e6Sjeremylt 143d7b241e6Sjeremylt if (*err == 0) { 144d7b241e6Sjeremylt *vec = CeedVector_count++; 145d7b241e6Sjeremylt CeedVector_n++; 146d7b241e6Sjeremylt } 147d7b241e6Sjeremylt } 148d7b241e6Sjeremylt 149d7b241e6Sjeremylt #define fCeedVectorSetArray FORTRAN_NAME(ceedvectorsetarray,CEEDVECTORSETARRAY) 150*3b271f31SJed Brown CEED_EXTERN void fCeedVectorSetArray(int *vec, int *memtype, int *copymode, 151c8b9fe72Sjeremylt CeedScalar *array, int64_t *offset, int *err) { 152399bb197SJed Brown *err = CeedVectorSetArray(CeedVector_dict[*vec], (CeedMemType)*memtype, 153399bb197SJed Brown (CeedCopyMode)*copymode, 154c8b9fe72Sjeremylt (CeedScalar *)(array + *offset)); 155d7b241e6Sjeremylt } 156d7b241e6Sjeremylt 1576a6c615bSJeremy L Thompson #define fCeedVectorTakeArray FORTRAN_NAME(ceedvectortakearray,CEEDVECTORTAKEARRAY) 158*3b271f31SJed Brown CEED_EXTERN void fCeedVectorTakeArray(int *vec, int *memtype, CeedScalar *array, 1596a6c615bSJeremy L Thompson int64_t *offset, int *err) { 1606a6c615bSJeremy L Thompson CeedScalar *b; 1616a6c615bSJeremy L Thompson CeedVector vec_ = CeedVector_dict[*vec]; 1626a6c615bSJeremy L Thompson *err = CeedVectorTakeArray(vec_, (CeedMemType)*memtype, &b); 1636a6c615bSJeremy L Thompson *offset = b - array; 1646a6c615bSJeremy L Thompson } 1656a6c615bSJeremy L Thompson 16654540941SJeremy L Thompson #define fCeedVectorSyncArray FORTRAN_NAME(ceedvectorsyncarray,CEEDVECTORSYNCARRAY) 167*3b271f31SJed Brown CEED_EXTERN void fCeedVectorSyncArray(int *vec, int *memtype, int *err) { 168399bb197SJed Brown *err = CeedVectorSyncArray(CeedVector_dict[*vec], (CeedMemType)*memtype); 16954540941SJeremy L Thompson } 17054540941SJeremy L Thompson 171d7b241e6Sjeremylt #define fCeedVectorSetValue FORTRAN_NAME(ceedvectorsetvalue,CEEDVECTORSETVALUE) 172*3b271f31SJed Brown CEED_EXTERN void fCeedVectorSetValue(int *vec, CeedScalar *value, int *err) { 173d7b241e6Sjeremylt *err = CeedVectorSetValue(CeedVector_dict[*vec], *value); 174d7b241e6Sjeremylt } 175d7b241e6Sjeremylt 176d7b241e6Sjeremylt #define fCeedVectorGetArray FORTRAN_NAME(ceedvectorgetarray,CEEDVECTORGETARRAY) 177*3b271f31SJed Brown CEED_EXTERN void fCeedVectorGetArray(int *vec, int *memtype, CeedScalar *array, 1786a6c615bSJeremy L Thompson int64_t *offset, int *err) { 179d7b241e6Sjeremylt CeedScalar *b; 180d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 181399bb197SJed Brown *err = CeedVectorGetArray(vec_, (CeedMemType)*memtype, &b); 182cbfd6a90SThilina Rathnayake *offset = b - array; 183d7b241e6Sjeremylt } 184d7b241e6Sjeremylt 185d7b241e6Sjeremylt #define fCeedVectorGetArrayRead \ 186d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorgetarrayread,CEEDVECTORGETARRAYREAD) 187*3b271f31SJed Brown CEED_EXTERN void fCeedVectorGetArrayRead(int *vec, int *memtype, 188*3b271f31SJed Brown CeedScalar *array, 1893334379fSJed Brown int64_t *offset, int *err) { 190d7b241e6Sjeremylt const CeedScalar *b; 191d7b241e6Sjeremylt CeedVector vec_ = CeedVector_dict[*vec]; 192399bb197SJed Brown *err = CeedVectorGetArrayRead(vec_, (CeedMemType)*memtype, &b); 193cbfd6a90SThilina Rathnayake *offset = b - array; 194d7b241e6Sjeremylt } 195d7b241e6Sjeremylt 1969c774eddSJeremy L Thompson #define fCeedVectorGetArrayWrite \ 1979c774eddSJeremy L Thompson FORTRAN_NAME(ceedvectorgetarraywrite,CEEDVECTORGETARRAYWRITE) 198*3b271f31SJed Brown CEED_EXTERN void fCeedVectorGetArrayWrite(int *vec, int *memtype, 199*3b271f31SJed Brown CeedScalar *array, 2009c774eddSJeremy L Thompson int64_t *offset, int *err) { 2019c774eddSJeremy L Thompson CeedScalar *b; 2029c774eddSJeremy L Thompson CeedVector vec_ = CeedVector_dict[*vec]; 2039c774eddSJeremy L Thompson *err = CeedVectorGetArrayWrite(vec_, (CeedMemType)*memtype, &b); 2049c774eddSJeremy L Thompson *offset = b - array; 2059c774eddSJeremy L Thompson } 2069c774eddSJeremy L Thompson 207d7b241e6Sjeremylt #define fCeedVectorRestoreArray \ 208d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearray,CEEDVECTORRESTOREARRAY) 209*3b271f31SJed Brown CEED_EXTERN void fCeedVectorRestoreArray(int *vec, CeedScalar *array, 2106456524eSJed Brown int64_t *offset, int *err) { 2119647a07eSDavid Medina CeedScalar *offsetArray = array + *offset; 2129647a07eSDavid Medina *err = CeedVectorRestoreArray(CeedVector_dict[*vec], &offsetArray); 2136456524eSJed Brown *offset = 0; 214d7b241e6Sjeremylt } 215d7b241e6Sjeremylt 216d7b241e6Sjeremylt #define fCeedVectorRestoreArrayRead \ 217d7b241e6Sjeremylt FORTRAN_NAME(ceedvectorrestorearrayread,CEEDVECTORRESTOREARRAYREAD) 218*3b271f31SJed Brown CEED_EXTERN void fCeedVectorRestoreArrayRead(int *vec, const CeedScalar *array, 2196456524eSJed Brown int64_t *offset, int *err) { 220d7b241e6Sjeremylt *err = CeedVectorRestoreArrayRead(CeedVector_dict[*vec], &array); 2216456524eSJed Brown *offset = 0; 222d7b241e6Sjeremylt } 223d7b241e6Sjeremylt 224547d9b97Sjeremylt #define fCeedVectorNorm \ 225547d9b97Sjeremylt FORTRAN_NAME(ceedvectornorm,CEEDVECTORNORM) 226*3b271f31SJed Brown CEED_EXTERN void fCeedVectorNorm(int *vec, int *type, CeedScalar *norm, 227*3b271f31SJed Brown int *err) { 228399bb197SJed Brown *err = CeedVectorNorm(CeedVector_dict[*vec], (CeedNormType)*type, norm); 229547d9b97Sjeremylt } 230547d9b97Sjeremylt 231d99fa3c5SJeremy L Thompson #define fCeedVectorReciprocal \ 232d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedvectorreciprocal,CEEDVECTORRECIPROCAL) 233*3b271f31SJed Brown CEED_EXTERN void fCeedVectorReciprocal(int *vec, int *err) { 234d99fa3c5SJeremy L Thompson *err = CeedVectorReciprocal(CeedVector_dict[*vec]); 235d99fa3c5SJeremy L Thompson } 236d99fa3c5SJeremy L Thompson 237d7b241e6Sjeremylt #define fCeedVectorView FORTRAN_NAME(ceedvectorview,CEEDVECTORVIEW) 238*3b271f31SJed Brown CEED_EXTERN void fCeedVectorView(int *vec, int *err) { 239d7b241e6Sjeremylt *err = CeedVectorView(CeedVector_dict[*vec], "%12.8f", stdout); 240d7b241e6Sjeremylt } 241d7b241e6Sjeremylt 242d7b241e6Sjeremylt #define fCeedVectorDestroy FORTRAN_NAME(ceedvectordestroy,CEEDVECTORDESTROY) 243*3b271f31SJed Brown CEED_EXTERN void fCeedVectorDestroy(int *vec, int *err) { 244b22696a8SJeremy L Thompson if (*vec == FORTRAN_NULL) return; 245d7b241e6Sjeremylt *err = CeedVectorDestroy(&CeedVector_dict[*vec]); 246d7b241e6Sjeremylt 247d7b241e6Sjeremylt if (*err == 0) { 24816595307SJeremy L Thompson *vec = FORTRAN_NULL; 249d7b241e6Sjeremylt CeedVector_n--; 250d7b241e6Sjeremylt if (CeedVector_n == 0) { 251d7b241e6Sjeremylt CeedFree(&CeedVector_dict); 252d7b241e6Sjeremylt CeedVector_count = 0; 253d7b241e6Sjeremylt CeedVector_count_max = 0; 254d7b241e6Sjeremylt } 255d7b241e6Sjeremylt } 256d7b241e6Sjeremylt } 257d7b241e6Sjeremylt 258b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 259b00b43a6SJeremy L Thompson // CeedElemRestriction 260b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 261d7b241e6Sjeremylt static CeedElemRestriction *CeedElemRestriction_dict = NULL; 262d7b241e6Sjeremylt static int CeedElemRestriction_count = 0; 263d7b241e6Sjeremylt static int CeedElemRestriction_n = 0; 264d7b241e6Sjeremylt static int CeedElemRestriction_count_max = 0; 265d7b241e6Sjeremylt 266d7b241e6Sjeremylt #define fCeedElemRestrictionCreate \ 267d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreate, CEEDELEMRESTRICTIONCREATE) 268*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionCreate(int *ceed, int *nelements, 269*3b271f31SJed Brown int *esize, 270d1d35e2fSjeremylt int *num_comp, int *comp_stride, int *lsize, 271d979a051Sjeremylt int *memtype, int *copymode, const int *offsets, 2728795c945Sjeremylt int *elemrestriction, int *err) { 273d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 274d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 275d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 276d7b241e6Sjeremylt } 277d7b241e6Sjeremylt 278d979a051Sjeremylt const int *offsets_ = offsets; 279d7b241e6Sjeremylt 280d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 281d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 282d979a051Sjeremylt *err = CeedElemRestrictionCreate(Ceed_dict[*ceed], *nelements, *esize, 283d1d35e2fSjeremylt *num_comp, *comp_stride, *lsize, 284d979a051Sjeremylt (CeedMemType)*memtype, 285d979a051Sjeremylt (CeedCopyMode)*copymode, offsets_, 286d979a051Sjeremylt elemrestriction_); 287d7b241e6Sjeremylt 288d7b241e6Sjeremylt if (*err == 0) { 289d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 290d7b241e6Sjeremylt CeedElemRestriction_n++; 291d7b241e6Sjeremylt } 292d7b241e6Sjeremylt } 293d7b241e6Sjeremylt 2947509a596Sjeremylt #define fCeedElemRestrictionCreateStrided \ 2957509a596Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreatestrided, CEEDELEMRESTRICTIONCREATESTRIDED) 296*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionCreateStrided(int *ceed, int *nelements, 297*3b271f31SJed Brown int *esize, 298d1d35e2fSjeremylt int *num_comp, int *lsize, int *strides, 299d7b241e6Sjeremylt int *elemrestriction, int *err) { 300d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 301d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 302d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 303d7b241e6Sjeremylt } 304d7b241e6Sjeremylt 305d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 306d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 3077509a596Sjeremylt *err = CeedElemRestrictionCreateStrided(Ceed_dict[*ceed], *nelements, *esize, 308d1d35e2fSjeremylt *num_comp, *lsize, 309d979a051Sjeremylt *strides == FORTRAN_STRIDES_BACKEND ? 310d979a051Sjeremylt CEED_STRIDES_BACKEND : strides, 311523b8ea0Sjeremylt elemrestriction_); 312d7b241e6Sjeremylt if (*err == 0) { 313d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 314d7b241e6Sjeremylt CeedElemRestriction_n++; 315d7b241e6Sjeremylt } 316d7b241e6Sjeremylt } 317d7b241e6Sjeremylt 318d7b241e6Sjeremylt #define fCeedElemRestrictionCreateBlocked \ 319d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateblocked,CEEDELEMRESTRICTIONCREATEBLOCKED) 320*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionCreateBlocked(int *ceed, int *nelements, 321*3b271f31SJed Brown int *esize, 322d1d35e2fSjeremylt int *blocksize, int *num_comp, 323d1d35e2fSjeremylt int *comp_stride, int *lsize, 324d979a051Sjeremylt int *mtype, int *cmode, 3258795c945Sjeremylt int *blkindices, int *elemrestriction, 3268795c945Sjeremylt int *err) { 327d7b241e6Sjeremylt 328d7b241e6Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 329d7b241e6Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 330d7b241e6Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 331d7b241e6Sjeremylt } 332d7b241e6Sjeremylt 333d7b241e6Sjeremylt CeedElemRestriction *elemrestriction_ = 334d7b241e6Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 335399bb197SJed Brown *err = CeedElemRestrictionCreateBlocked(Ceed_dict[*ceed], 336d979a051Sjeremylt *nelements, *esize, *blocksize, 337d1d35e2fSjeremylt *num_comp, *comp_stride, *lsize, 338d979a051Sjeremylt (CeedMemType)*mtype, 339d979a051Sjeremylt (CeedCopyMode)*cmode, blkindices, 340a8d32208Sjeremylt elemrestriction_); 341d7b241e6Sjeremylt 342d7b241e6Sjeremylt if (*err == 0) { 343d7b241e6Sjeremylt *elemrestriction = CeedElemRestriction_count++; 344d7b241e6Sjeremylt CeedElemRestriction_n++; 345d7b241e6Sjeremylt } 346d7b241e6Sjeremylt } 347d7b241e6Sjeremylt 3487509a596Sjeremylt #define fCeedElemRestrictionCreateBlockedStrided \ 3497509a596Sjeremylt FORTRAN_NAME(ceedelemrestrictioncreateblockedstrided, CEEDELEMRESTRICTIONCREATEBLOCKEDSTRIDED) 350*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionCreateBlockedStrided(int *ceed, 351*3b271f31SJed Brown int *nelements, 352d1d35e2fSjeremylt int *esize, int *blk_size, int *num_comp, int *lsize, int *strides, 3537509a596Sjeremylt int *elemrestriction, int *err) { 3547509a596Sjeremylt if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 3557509a596Sjeremylt CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 3567509a596Sjeremylt CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 3577509a596Sjeremylt } 3587509a596Sjeremylt 3597509a596Sjeremylt CeedElemRestriction *elemrestriction_ = 3607509a596Sjeremylt &CeedElemRestriction_dict[CeedElemRestriction_count]; 3617509a596Sjeremylt *err = CeedElemRestrictionCreateBlockedStrided(Ceed_dict[*ceed], *nelements, 362d1d35e2fSjeremylt *esize, *blk_size, *num_comp, *lsize, strides, elemrestriction_); 3637509a596Sjeremylt if (*err == 0) { 3647509a596Sjeremylt *elemrestriction = CeedElemRestriction_count++; 3657509a596Sjeremylt CeedElemRestriction_n++; 3667509a596Sjeremylt } 3677509a596Sjeremylt } 3687509a596Sjeremylt 369d7b241e6Sjeremylt static CeedRequest *CeedRequest_dict = NULL; 370d7b241e6Sjeremylt static int CeedRequest_count = 0; 371d7b241e6Sjeremylt static int CeedRequest_n = 0; 372d7b241e6Sjeremylt static int CeedRequest_count_max = 0; 373d7b241e6Sjeremylt 374d7b241e6Sjeremylt #define fCeedElemRestrictionApply \ 375d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictionapply,CEEDELEMRESTRICTIONAPPLY) 376*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionApply(int *elemr, int *tmode, int *uvec, 377*3b271f31SJed Brown int *ruvec, 378a8d32208Sjeremylt int *rqst, int *err) { 379d7b241e6Sjeremylt int createRequest = 1; 380d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 381d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) 382d7b241e6Sjeremylt createRequest = 0; 383d7b241e6Sjeremylt 384d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 385d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 386d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 387d7b241e6Sjeremylt } 388d7b241e6Sjeremylt 389d7b241e6Sjeremylt CeedRequest *rqst_; 390d7b241e6Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 391d7b241e6Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED ) rqst_ = CEED_REQUEST_ORDERED; 392d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 393d7b241e6Sjeremylt 394399bb197SJed Brown *err = CeedElemRestrictionApply(CeedElemRestriction_dict[*elemr], 395399bb197SJed Brown (CeedTransposeMode)*tmode, 396a8d32208Sjeremylt CeedVector_dict[*uvec], 3978795c945Sjeremylt CeedVector_dict[*ruvec], rqst_); 398d7b241e6Sjeremylt 399d7b241e6Sjeremylt if (*err == 0 && createRequest) { 400d7b241e6Sjeremylt *rqst = CeedRequest_count++; 401d7b241e6Sjeremylt CeedRequest_n++; 402d7b241e6Sjeremylt } 403d7b241e6Sjeremylt } 404d7b241e6Sjeremylt 405be9261b7Sjeremylt #define fCeedElemRestrictionApplyBlock \ 406be9261b7Sjeremylt FORTRAN_NAME(ceedelemrestrictionapplyblock,CEEDELEMRESTRICTIONAPPLYBLOCK) 407*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionApplyBlock(int *elemr, int *block, 408*3b271f31SJed Brown int *tmode, 409be9261b7Sjeremylt int *uvec, int *ruvec, int *rqst, int *err) { 410be9261b7Sjeremylt int createRequest = 1; 411be9261b7Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 412be9261b7Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE || *rqst == FORTRAN_REQUEST_ORDERED) 413be9261b7Sjeremylt createRequest = 0; 414be9261b7Sjeremylt 415be9261b7Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 416be9261b7Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 417be9261b7Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 418be9261b7Sjeremylt } 419be9261b7Sjeremylt 420be9261b7Sjeremylt CeedRequest *rqst_; 421be9261b7Sjeremylt if (*rqst == FORTRAN_REQUEST_IMMEDIATE) rqst_ = CEED_REQUEST_IMMEDIATE; 422be9261b7Sjeremylt else if (*rqst == FORTRAN_REQUEST_ORDERED ) rqst_ = CEED_REQUEST_ORDERED; 423be9261b7Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 424be9261b7Sjeremylt 425be9261b7Sjeremylt *err = CeedElemRestrictionApplyBlock(CeedElemRestriction_dict[*elemr], *block, 426399bb197SJed Brown (CeedTransposeMode)*tmode, CeedVector_dict[*uvec], 427be9261b7Sjeremylt CeedVector_dict[*ruvec], rqst_); 428be9261b7Sjeremylt 429be9261b7Sjeremylt if (*err == 0 && createRequest) { 430be9261b7Sjeremylt *rqst = CeedRequest_count++; 431be9261b7Sjeremylt CeedRequest_n++; 432be9261b7Sjeremylt } 433be9261b7Sjeremylt } 434be9261b7Sjeremylt 4351469ee4dSjeremylt #define fCeedElemRestrictionGetMultiplicity \ 4361469ee4dSjeremylt FORTRAN_NAME(ceedelemrestrictiongetmultiplicity,CEEDELEMRESTRICTIONGETMULTIPLICITY) 437*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionGetMultiplicity(int *elemr, int *mult, 438*3b271f31SJed Brown int *err) { 4391469ee4dSjeremylt *err = CeedElemRestrictionGetMultiplicity(CeedElemRestriction_dict[*elemr], 440a8d32208Sjeremylt CeedVector_dict[*mult]); 4411469ee4dSjeremylt } 4421469ee4dSjeremylt 44349fd234cSJeremy L Thompson #define fCeedElemRestrictionGetELayout \ 44449fd234cSJeremy L Thompson FORTRAN_NAME(ceedelemrestrictiongetelayout,CEEDELEMRESTRICTIONGETELAYOUT) 445*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionGetELayout(int *elemr, int *layout, 446*3b271f31SJed Brown int *err) { 44749fd234cSJeremy L Thompson CeedInt layout_c[3]; 44849fd234cSJeremy L Thompson *err = CeedElemRestrictionGetELayout(CeedElemRestriction_dict[*elemr], 44949fd234cSJeremy L Thompson &layout_c); 45049fd234cSJeremy L Thompson for (int i=0; i<3; i++) 45149fd234cSJeremy L Thompson layout[i] = layout_c[i]; 45249fd234cSJeremy L Thompson } 45349fd234cSJeremy L Thompson 4545a86f033Sjeremylt #define fCeedElemRestrictionView \ 4555a86f033Sjeremylt FORTRAN_NAME(ceedelemrestrictionview,CEEDELEMRESTRICTIONVIEW) 456*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionView(int *elemr, int *err) { 4575a86f033Sjeremylt *err = CeedElemRestrictionView(CeedElemRestriction_dict[*elemr], stdout); 4585a86f033Sjeremylt } 4591469ee4dSjeremylt 460d7b241e6Sjeremylt #define fCeedRequestWait FORTRAN_NAME(ceedrequestwait, CEEDREQUESTWAIT) 461*3b271f31SJed Brown CEED_EXTERN void fCeedRequestWait(int *rqst, int *err) { 462d7b241e6Sjeremylt // TODO Uncomment this once CeedRequestWait is implemented 463d7b241e6Sjeremylt //*err = CeedRequestWait(&CeedRequest_dict[*rqst]); 464d7b241e6Sjeremylt 465d7b241e6Sjeremylt if (*err == 0) { 466d7b241e6Sjeremylt CeedRequest_n--; 467d7b241e6Sjeremylt if (CeedRequest_n == 0) { 468d7b241e6Sjeremylt CeedFree(&CeedRequest_dict); 469d7b241e6Sjeremylt CeedRequest_count = 0; 470d7b241e6Sjeremylt CeedRequest_count_max = 0; 471d7b241e6Sjeremylt } 472d7b241e6Sjeremylt } 473d7b241e6Sjeremylt } 474d7b241e6Sjeremylt 475d7b241e6Sjeremylt #define fCeedElemRestrictionDestroy \ 476d7b241e6Sjeremylt FORTRAN_NAME(ceedelemrestrictiondestroy,CEEDELEMRESTRICTIONDESTROY) 477*3b271f31SJed Brown CEED_EXTERN void fCeedElemRestrictionDestroy(int *elem, int *err) { 478b22696a8SJeremy L Thompson if (*elem == FORTRAN_NULL) return; 479d7b241e6Sjeremylt *err = CeedElemRestrictionDestroy(&CeedElemRestriction_dict[*elem]); 480d7b241e6Sjeremylt 481d7b241e6Sjeremylt if (*err == 0) { 48216595307SJeremy L Thompson *elem = FORTRAN_NULL; 483d7b241e6Sjeremylt CeedElemRestriction_n--; 484d7b241e6Sjeremylt if (CeedElemRestriction_n == 0) { 485d7b241e6Sjeremylt CeedFree(&CeedElemRestriction_dict); 486d7b241e6Sjeremylt CeedElemRestriction_count = 0; 487d7b241e6Sjeremylt CeedElemRestriction_count_max = 0; 488d7b241e6Sjeremylt } 489d7b241e6Sjeremylt } 490d7b241e6Sjeremylt } 491d7b241e6Sjeremylt 492b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 493b00b43a6SJeremy L Thompson // CeedBasis 494b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 495d7b241e6Sjeremylt static CeedBasis *CeedBasis_dict = NULL; 496d7b241e6Sjeremylt static int CeedBasis_count = 0; 497d7b241e6Sjeremylt static int CeedBasis_n = 0; 498d7b241e6Sjeremylt static int CeedBasis_count_max = 0; 499d7b241e6Sjeremylt 500d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1Lagrange \ 501d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1lagrange, CEEDBASISCREATETENSORH1LAGRANGE) 502*3b271f31SJed Brown CEED_EXTERN void fCeedBasisCreateTensorH1Lagrange(int *ceed, int *dim, 503d1d35e2fSjeremylt int *num_comp, int *P, int *Q, int *quadmode, 5048795c945Sjeremylt int *basis, int *err) { 505d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 506d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 507d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 508d7b241e6Sjeremylt } 509d7b241e6Sjeremylt 510d1d35e2fSjeremylt *err = CeedBasisCreateTensorH1Lagrange(Ceed_dict[*ceed], *dim, *num_comp, *P, 511d1d35e2fSjeremylt *Q, 512399bb197SJed Brown (CeedQuadMode)*quadmode, 5138795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 514d7b241e6Sjeremylt 515d7b241e6Sjeremylt if (*err == 0) { 516d7b241e6Sjeremylt *basis = CeedBasis_count++; 517d7b241e6Sjeremylt CeedBasis_n++; 518d7b241e6Sjeremylt } 519d7b241e6Sjeremylt } 520d7b241e6Sjeremylt 521d7b241e6Sjeremylt #define fCeedBasisCreateTensorH1 \ 522d7b241e6Sjeremylt FORTRAN_NAME(ceedbasiscreatetensorh1, CEEDBASISCREATETENSORH1) 523*3b271f31SJed Brown CEED_EXTERN void fCeedBasisCreateTensorH1(int *ceed, int *dim, int *num_comp, 524*3b271f31SJed Brown int *P_1d, 525d1d35e2fSjeremylt int *Q_1d, const CeedScalar *interp_1d, 526d1d35e2fSjeremylt const CeedScalar *grad_1d, 527d1d35e2fSjeremylt const CeedScalar *q_ref_1d, 528d1d35e2fSjeremylt const CeedScalar *q_weight_1d, int *basis, 5298795c945Sjeremylt int *err) { 530d7b241e6Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 531d7b241e6Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 532d7b241e6Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 533d7b241e6Sjeremylt } 534d7b241e6Sjeremylt 535d1d35e2fSjeremylt *err = CeedBasisCreateTensorH1(Ceed_dict[*ceed], *dim, *num_comp, *P_1d, *Q_1d, 536d1d35e2fSjeremylt interp_1d, grad_1d, q_ref_1d, q_weight_1d, 5378795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 538d7b241e6Sjeremylt 539d7b241e6Sjeremylt if (*err == 0) { 540d7b241e6Sjeremylt *basis = CeedBasis_count++; 541d7b241e6Sjeremylt CeedBasis_n++; 542d7b241e6Sjeremylt } 543d7b241e6Sjeremylt } 544d7b241e6Sjeremylt 545a8de75f0Sjeremylt #define fCeedBasisCreateH1 \ 546a8de75f0Sjeremylt FORTRAN_NAME(ceedbasiscreateh1, CEEDBASISCREATEH1) 547*3b271f31SJed Brown CEED_EXTERN void fCeedBasisCreateH1(int *ceed, int *topo, int *num_comp, 548*3b271f31SJed Brown int *nnodes, 5498795c945Sjeremylt int *nqpts, const CeedScalar *interp, 5508795c945Sjeremylt const CeedScalar *grad, const CeedScalar *qref, 5518795c945Sjeremylt const CeedScalar *qweight, int *basis, int *err) { 552a8de75f0Sjeremylt if (CeedBasis_count == CeedBasis_count_max) { 553a8de75f0Sjeremylt CeedBasis_count_max += CeedBasis_count_max/2 + 1; 554a8de75f0Sjeremylt CeedRealloc(CeedBasis_count_max, &CeedBasis_dict); 555a8de75f0Sjeremylt } 556a8de75f0Sjeremylt 557d1d35e2fSjeremylt *err = CeedBasisCreateH1(Ceed_dict[*ceed], (CeedElemTopology)*topo, *num_comp, 558399bb197SJed Brown *nnodes, *nqpts, interp, grad, qref, qweight, 5598795c945Sjeremylt &CeedBasis_dict[CeedBasis_count]); 560a8de75f0Sjeremylt 561a8de75f0Sjeremylt if (*err == 0) { 562a8de75f0Sjeremylt *basis = CeedBasis_count++; 563a8de75f0Sjeremylt CeedBasis_n++; 564a8de75f0Sjeremylt } 565a8de75f0Sjeremylt } 566a8de75f0Sjeremylt 567d7b241e6Sjeremylt #define fCeedBasisView FORTRAN_NAME(ceedbasisview, CEEDBASISVIEW) 568*3b271f31SJed Brown CEED_EXTERN void fCeedBasisView(int *basis, int *err) { 569d7b241e6Sjeremylt *err = CeedBasisView(CeedBasis_dict[*basis], stdout); 570d7b241e6Sjeremylt } 571d7b241e6Sjeremylt 572d7b241e6Sjeremylt #define fCeedQRFactorization \ 573d7b241e6Sjeremylt FORTRAN_NAME(ceedqrfactorization, CEEDQRFACTORIZATION) 574*3b271f31SJed Brown CEED_EXTERN void fCeedQRFactorization(int *ceed, CeedScalar *mat, 575*3b271f31SJed Brown CeedScalar *tau, int *m, 576a7bd39daSjeremylt int *n, int *err) { 577a7bd39daSjeremylt *err = CeedQRFactorization(Ceed_dict[*ceed], mat, tau, *m, *n); 578d7b241e6Sjeremylt } 579d7b241e6Sjeremylt 5809ac7b42eSJeremy L Thompson #define fCeedHouseholderApplyQ \ 5819ac7b42eSJeremy L Thompson FORTRAN_NAME(ceedhouseholderapplyq, CEEDHOUSEHOLDERAPPLYQ) 582*3b271f31SJed Brown CEED_EXTERN void fCeedHouseholderApplyQ(CeedScalar *A, CeedScalar *Q, 583*3b271f31SJed Brown CeedScalar *tau, 5849ac7b42eSJeremy L Thompson int *t_mode, 5859ac7b42eSJeremy L Thompson int *m, int *n, int *k, int *row, int *col, int *err) { 5869ac7b42eSJeremy L Thompson *err = CeedHouseholderApplyQ(A, Q, tau, (CeedTransposeMode)*t_mode, *m, *n, *k, 5879ac7b42eSJeremy L Thompson *row, *col); 5889ac7b42eSJeremy L Thompson } 5899ac7b42eSJeremy L Thompson 59052bfb9bbSJeremy L Thompson #define fCeedSymmetricSchurDecomposition \ 59152bfb9bbSJeremy L Thompson FORTRAN_NAME(ceedsymmetricschurdecomposition, CEEDSYMMETRICSCHURDECOMPOSITION) 592*3b271f31SJed Brown CEED_EXTERN void fCeedSymmetricSchurDecomposition(int *ceed, CeedScalar *mat, 59352bfb9bbSJeremy L Thompson CeedScalar *lambda, int *n, int *err) { 59452bfb9bbSJeremy L Thompson *err = CeedSymmetricSchurDecomposition(Ceed_dict[*ceed], mat, lambda, *n); 59552bfb9bbSJeremy L Thompson } 59652bfb9bbSJeremy L Thompson 59752bfb9bbSJeremy L Thompson #define fCeedSimultaneousDiagonalization \ 59852bfb9bbSJeremy L Thompson FORTRAN_NAME(ceedsimultaneousdiagonalization, CEEDSIMULTANEOUSDIAGONALIZATION) 599*3b271f31SJed Brown CEED_EXTERN void fCeedSimultaneousDiagonalization(int *ceed, CeedScalar *matA, 60052bfb9bbSJeremy L Thompson CeedScalar *matB, CeedScalar *x, 60152bfb9bbSJeremy L Thompson CeedScalar *lambda, int *n, int *err) { 60252bfb9bbSJeremy L Thompson *err = CeedSimultaneousDiagonalization(Ceed_dict[*ceed], matA, matB, x, 60352bfb9bbSJeremy L Thompson lambda, *n); 60452bfb9bbSJeremy L Thompson } 60552bfb9bbSJeremy L Thompson 606783c99b3SValeria Barra #define fCeedBasisGetCollocatedGrad \ 607783c99b3SValeria Barra FORTRAN_NAME(ceedbasisgetcollocatedgrad, CEEDBASISGETCOLLOCATEDGRAD) 608*3b271f31SJed Brown CEED_EXTERN void fCeedBasisGetCollocatedGrad(int *basis, 609*3b271f31SJed Brown CeedScalar *colo_grad_1d, 610d7b241e6Sjeremylt int *err) { 611d1d35e2fSjeremylt *err = CeedBasisGetCollocatedGrad(CeedBasis_dict[*basis], colo_grad_1d); 612d7b241e6Sjeremylt } 613d7b241e6Sjeremylt 614d7b241e6Sjeremylt #define fCeedBasisApply FORTRAN_NAME(ceedbasisapply, CEEDBASISAPPLY) 615*3b271f31SJed Brown CEED_EXTERN void fCeedBasisApply(int *basis, int *num_elem, int *tmode, 616*3b271f31SJed Brown int *eval_mode, 617aedaa0e5Sjeremylt int *u, int *v, int *err) { 618d1d35e2fSjeremylt *err = CeedBasisApply(CeedBasis_dict[*basis], *num_elem, 619d1d35e2fSjeremylt (CeedTransposeMode)*tmode, 620d1d35e2fSjeremylt (CeedEvalMode)*eval_mode, 621399bb197SJed Brown *u == FORTRAN_VECTOR_NONE ? CEED_VECTOR_NONE : CeedVector_dict[*u], 622aedaa0e5Sjeremylt CeedVector_dict[*v]); 623d7b241e6Sjeremylt } 624d7b241e6Sjeremylt 625d7b241e6Sjeremylt #define fCeedBasisGetNumNodes \ 626d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumnodes, CEEDBASISGETNUMNODES) 627*3b271f31SJed Brown CEED_EXTERN void fCeedBasisGetNumNodes(int *basis, int *P, int *err) { 628d7b241e6Sjeremylt *err = CeedBasisGetNumNodes(CeedBasis_dict[*basis], P); 629d7b241e6Sjeremylt } 630d7b241e6Sjeremylt 631d7b241e6Sjeremylt #define fCeedBasisGetNumQuadraturePoints \ 632d7b241e6Sjeremylt FORTRAN_NAME(ceedbasisgetnumquadraturepoints, CEEDBASISGETNUMQUADRATUREPOINTS) 633*3b271f31SJed Brown CEED_EXTERN void fCeedBasisGetNumQuadraturePoints(int *basis, int *Q, 634*3b271f31SJed Brown int *err) { 635d7b241e6Sjeremylt *err = CeedBasisGetNumQuadraturePoints(CeedBasis_dict[*basis], Q); 636d7b241e6Sjeremylt } 637d7b241e6Sjeremylt 638d99fa3c5SJeremy L Thompson #define fCeedBasisGetInterp1D \ 639d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedbasisgetinterp1d, CEEDBASISGETINTERP1D) 640*3b271f31SJed Brown CEED_EXTERN void fCeedBasisGetInterp1D(int *basis, CeedScalar *interp_1d, 641*3b271f31SJed Brown int64_t *offset, 642d99fa3c5SJeremy L Thompson int *err) { 643d99fa3c5SJeremy L Thompson const CeedScalar *interp1d_; 644d99fa3c5SJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 645d99fa3c5SJeremy L Thompson *err = CeedBasisGetInterp1D(basis_, &interp1d_); 646d1d35e2fSjeremylt *offset = interp1d_ - interp_1d; 647d99fa3c5SJeremy L Thompson } 648d99fa3c5SJeremy L Thompson 6499ac7b42eSJeremy L Thompson #define fCeedBasisGetGrad1D \ 6509ac7b42eSJeremy L Thompson FORTRAN_NAME(ceedbasisgetgrad1d, CEEDBASISGETGRAD1D) 651*3b271f31SJed Brown CEED_EXTERN void fCeedBasisGetGrad1D(int *basis, CeedScalar *grad_1d, 652*3b271f31SJed Brown int64_t *offset, 6539ac7b42eSJeremy L Thompson int *err) { 6549ac7b42eSJeremy L Thompson const CeedScalar *grad1d_; 6559ac7b42eSJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 6569ac7b42eSJeremy L Thompson *err = CeedBasisGetGrad1D(basis_, &grad1d_); 6579ac7b42eSJeremy L Thompson *offset = grad1d_ - grad_1d; 6589ac7b42eSJeremy L Thompson } 6599ac7b42eSJeremy L Thompson 6609ac7b42eSJeremy L Thompson #define fCeedBasisGetQRef \ 6619ac7b42eSJeremy L Thompson FORTRAN_NAME(ceedbasisgetqref, CEEDBASISGETQREF) 662*3b271f31SJed Brown CEED_EXTERN void fCeedBasisGetQRef(int *basis, CeedScalar *q_ref, 663*3b271f31SJed Brown int64_t *offset, 6649ac7b42eSJeremy L Thompson int *err) { 6659ac7b42eSJeremy L Thompson const CeedScalar *qref_; 6669ac7b42eSJeremy L Thompson CeedBasis basis_ = CeedBasis_dict[*basis]; 6679ac7b42eSJeremy L Thompson *err = CeedBasisGetQRef(basis_, &qref_); 6689ac7b42eSJeremy L Thompson *offset = qref_ - q_ref; 6699ac7b42eSJeremy L Thompson } 6709ac7b42eSJeremy L Thompson 671d7b241e6Sjeremylt #define fCeedBasisDestroy FORTRAN_NAME(ceedbasisdestroy,CEEDBASISDESTROY) 672*3b271f31SJed Brown CEED_EXTERN void fCeedBasisDestroy(int *basis, int *err) { 673b22696a8SJeremy L Thompson if (*basis == FORTRAN_NULL) return; 674d7b241e6Sjeremylt *err = CeedBasisDestroy(&CeedBasis_dict[*basis]); 675d7b241e6Sjeremylt 676d7b241e6Sjeremylt if (*err == 0) { 67716595307SJeremy L Thompson *basis = FORTRAN_NULL; 678d7b241e6Sjeremylt CeedBasis_n--; 679d7b241e6Sjeremylt if (CeedBasis_n == 0) { 680d7b241e6Sjeremylt CeedFree(&CeedBasis_dict); 681d7b241e6Sjeremylt CeedBasis_count = 0; 682d7b241e6Sjeremylt CeedBasis_count_max = 0; 683d7b241e6Sjeremylt } 684d7b241e6Sjeremylt } 685d7b241e6Sjeremylt } 686d7b241e6Sjeremylt 687d7b241e6Sjeremylt #define fCeedGaussQuadrature FORTRAN_NAME(ceedgaussquadrature, CEEDGAUSSQUADRATURE) 688*3b271f31SJed Brown CEED_EXTERN void fCeedGaussQuadrature(int *Q, CeedScalar *q_ref_1d, 689*3b271f31SJed Brown CeedScalar *q_weight_1d, 690d7b241e6Sjeremylt int *err) { 691d1d35e2fSjeremylt *err = CeedGaussQuadrature(*Q, q_ref_1d, q_weight_1d); 692d7b241e6Sjeremylt } 693d7b241e6Sjeremylt 694d7b241e6Sjeremylt #define fCeedLobattoQuadrature \ 695d7b241e6Sjeremylt FORTRAN_NAME(ceedlobattoquadrature, CEEDLOBATTOQUADRATURE) 696*3b271f31SJed Brown CEED_EXTERN void fCeedLobattoQuadrature(int *Q, CeedScalar *q_ref_1d, 697d1d35e2fSjeremylt CeedScalar *q_weight_1d, 698d7b241e6Sjeremylt int *err) { 699d1d35e2fSjeremylt *err = CeedLobattoQuadrature(*Q, q_ref_1d, q_weight_1d); 700d7b241e6Sjeremylt } 701d7b241e6Sjeremylt 702b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 703777ff853SJeremy L Thompson // CeedQFunctionContext 704777ff853SJeremy L Thompson // ----------------------------------------------------------------------------- 705777ff853SJeremy L Thompson static CeedQFunctionContext *CeedQFunctionContext_dict = NULL; 706777ff853SJeremy L Thompson static int CeedQFunctionContext_count = 0; 707777ff853SJeremy L Thompson static int CeedQFunctionContext_n = 0; 708777ff853SJeremy L Thompson static int CeedQFunctionContext_count_max = 0; 709777ff853SJeremy L Thompson 710777ff853SJeremy L Thompson #define fCeedQFunctionContextCreate \ 711777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextcreate,CEEDQFUNCTIONCONTEXTCREATE) 712*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextCreate(int *ceed, int *ctx, int *err) { 713777ff853SJeremy L Thompson if (CeedQFunctionContext_count == CeedQFunctionContext_count_max) { 714777ff853SJeremy L Thompson CeedQFunctionContext_count_max += CeedQFunctionContext_count_max/2 + 1; 715777ff853SJeremy L Thompson CeedRealloc(CeedQFunctionContext_count_max, &CeedQFunctionContext_dict); 716777ff853SJeremy L Thompson } 717777ff853SJeremy L Thompson 718777ff853SJeremy L Thompson CeedQFunctionContext *ctx_ = 719777ff853SJeremy L Thompson &CeedQFunctionContext_dict[CeedQFunctionContext_count]; 720777ff853SJeremy L Thompson 721777ff853SJeremy L Thompson *err = CeedQFunctionContextCreate(Ceed_dict[*ceed], ctx_); 722777ff853SJeremy L Thompson if (*err) return; 723777ff853SJeremy L Thompson *ctx = CeedQFunctionContext_count++; 724777ff853SJeremy L Thompson CeedQFunctionContext_n++; 725777ff853SJeremy L Thompson } 726777ff853SJeremy L Thompson 727777ff853SJeremy L Thompson #define fCeedQFunctionContextSetData \ 728777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextsetdata,CEEDQFUNCTIONCONTEXTSETDATA) 729*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextSetData(int *ctx, int *memtype, 730*3b271f31SJed Brown int *copymode, 731777ff853SJeremy L Thompson CeedInt *n, 732777ff853SJeremy L Thompson CeedScalar *data, int64_t *offset, int *err) { 733d1d35e2fSjeremylt size_t ctx_size = ((size_t) *n)*sizeof(CeedScalar); 734777ff853SJeremy L Thompson *err = CeedQFunctionContextSetData(CeedQFunctionContext_dict[*ctx], 735777ff853SJeremy L Thompson (CeedMemType)*memtype, 736d1d35e2fSjeremylt (CeedCopyMode)*copymode, ctx_size, 737777ff853SJeremy L Thompson data + *offset); 738777ff853SJeremy L Thompson } 739777ff853SJeremy L Thompson 740777ff853SJeremy L Thompson #define fCeedQFunctionContextGetData \ 741777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextgetdata,CEEDQFUNCTIONCONTEXTGETDATA) 742*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextGetData(int *ctx, int *memtype, 743*3b271f31SJed Brown CeedScalar *data, 744777ff853SJeremy L Thompson int64_t *offset, int *err) { 745777ff853SJeremy L Thompson CeedScalar *b; 746777ff853SJeremy L Thompson CeedQFunctionContext ctx_ = CeedQFunctionContext_dict[*ctx]; 747777ff853SJeremy L Thompson *err = CeedQFunctionContextGetData(ctx_, (CeedMemType)*memtype, &b); 748777ff853SJeremy L Thompson *offset = b - data; 749777ff853SJeremy L Thompson } 750777ff853SJeremy L Thompson 751777ff853SJeremy L Thompson #define fCeedQFunctionContextRestoreData \ 752777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextrestoredata,CEEDQFUNCTIONCONTEXTRESTOREDATA) 753*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextRestoreData(int *ctx, CeedScalar *data, 754777ff853SJeremy L Thompson int64_t *offset, int *err) { 755777ff853SJeremy L Thompson *err = CeedQFunctionContextRestoreData(CeedQFunctionContext_dict[*ctx], 756777ff853SJeremy L Thompson (void **)&data); 757777ff853SJeremy L Thompson *offset = 0; 758777ff853SJeremy L Thompson } 759777ff853SJeremy L Thompson 760777ff853SJeremy L Thompson #define fCeedQFunctionContextView \ 761777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextview,CEEDQFUNCTIONCONTEXTVIEW) 762*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextView(int *ctx, int *err) { 763777ff853SJeremy L Thompson *err = CeedQFunctionContextView(CeedQFunctionContext_dict[*ctx], stdout); 764777ff853SJeremy L Thompson } 765777ff853SJeremy L Thompson 766777ff853SJeremy L Thompson #define fCeedQFunctionContextDestroy \ 767777ff853SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncontextdestroy,CEEDQFUNCTIONCONTEXTDESTROY) 768*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionContextDestroy(int *ctx, int *err) { 769777ff853SJeremy L Thompson if (*ctx == FORTRAN_NULL) return; 770777ff853SJeremy L Thompson *err = CeedQFunctionContextDestroy(&CeedQFunctionContext_dict[*ctx]); 771777ff853SJeremy L Thompson 772777ff853SJeremy L Thompson if (*err == 0) { 773777ff853SJeremy L Thompson *ctx = FORTRAN_NULL; 774777ff853SJeremy L Thompson CeedQFunctionContext_n--; 775777ff853SJeremy L Thompson if (CeedQFunctionContext_n == 0) { 776777ff853SJeremy L Thompson CeedFree(&CeedQFunctionContext_dict); 777777ff853SJeremy L Thompson CeedQFunctionContext_count = 0; 778777ff853SJeremy L Thompson CeedQFunctionContext_count_max = 0; 779777ff853SJeremy L Thompson } 780777ff853SJeremy L Thompson } 781777ff853SJeremy L Thompson } 782777ff853SJeremy L Thompson 783777ff853SJeremy L Thompson // ----------------------------------------------------------------------------- 784b00b43a6SJeremy L Thompson // CeedQFunction 785b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 786d7b241e6Sjeremylt static CeedQFunction *CeedQFunction_dict = NULL; 787d7b241e6Sjeremylt static int CeedQFunction_count = 0; 788d7b241e6Sjeremylt static int CeedQFunction_n = 0; 789d7b241e6Sjeremylt static int CeedQFunction_count_max = 0; 790d7b241e6Sjeremylt 791d7b241e6Sjeremylt static int CeedQFunctionFortranStub(void *ctx, int nq, 7928795c945Sjeremylt const CeedScalar *const *u, 7938795c945Sjeremylt CeedScalar *const *v) { 794777ff853SJeremy L Thompson CeedFortranContext fctx = ctx; 7958cb0412aSJeremy L Thompson CeedQFunctionContext inner_ctx = fctx->inner_ctx; 796d7b241e6Sjeremylt int ierr; 797d7b241e6Sjeremylt 798777ff853SJeremy L Thompson CeedScalar *ctx_ = NULL; 799777ff853SJeremy L Thompson // Note: Device backends are generating their own kernels from 800777ff853SJeremy L Thompson // single source files, so only Host backends need to 801777ff853SJeremy L Thompson // use this Fortran stub. 8028cb0412aSJeremy L Thompson if (inner_ctx) { 8038cb0412aSJeremy L Thompson ierr = CeedQFunctionContextGetData(inner_ctx, CEED_MEM_HOST, &ctx_); 804777ff853SJeremy L Thompson CeedChk(ierr); 805777ff853SJeremy L Thompson } 806777ff853SJeremy L Thompson 8071e35832bSjeremylt fctx->f((void *)ctx_,&nq,u[0],u[1],u[2],u[3],u[4],u[5],u[6], 808418fb8c2Sjeremylt u[7],u[8],u[9],u[10],u[11],u[12],u[13],u[14],u[15], 809418fb8c2Sjeremylt v[0],v[1],v[2],v[3],v[4],v[5],v[6],v[7],v[8],v[9], 810418fb8c2Sjeremylt v[10],v[11],v[12],v[13],v[14],v[15],&ierr); 811777ff853SJeremy L Thompson 8128cb0412aSJeremy L Thompson if (inner_ctx) { 8138cb0412aSJeremy L Thompson ierr = CeedQFunctionContextRestoreData(inner_ctx, (void *)&ctx_); 814777ff853SJeremy L Thompson CeedChk(ierr); 815777ff853SJeremy L Thompson } 816777ff853SJeremy L Thompson 817d7b241e6Sjeremylt return ierr; 818d7b241e6Sjeremylt } 819d7b241e6Sjeremylt 820d7b241e6Sjeremylt #define fCeedQFunctionCreateInterior \ 821d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctioncreateinterior, CEEDQFUNCTIONCREATEINTERIOR) 822*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionCreateInterior(int *ceed, int *vec_length, 823d7b241e6Sjeremylt void (*f)(void *ctx, int *nq, 824418fb8c2Sjeremylt const CeedScalar *u,const CeedScalar *u1, 825418fb8c2Sjeremylt const CeedScalar *u2,const CeedScalar *u3, 826418fb8c2Sjeremylt const CeedScalar *u4,const CeedScalar *u5, 827418fb8c2Sjeremylt const CeedScalar *u6,const CeedScalar *u7, 828418fb8c2Sjeremylt const CeedScalar *u8,const CeedScalar *u9, 829418fb8c2Sjeremylt const CeedScalar *u10,const CeedScalar *u11, 830418fb8c2Sjeremylt const CeedScalar *u12,const CeedScalar *u13, 831418fb8c2Sjeremylt const CeedScalar *u14,const CeedScalar *u15, 832418fb8c2Sjeremylt CeedScalar *v,CeedScalar *v1,CeedScalar *v2, 8338795c945Sjeremylt CeedScalar *v3,CeedScalar *v4, 8348795c945Sjeremylt CeedScalar *v5,CeedScalar *v6, 8358795c945Sjeremylt CeedScalar *v7,CeedScalar *v8, 8368795c945Sjeremylt CeedScalar *v9,CeedScalar *v10, 8378795c945Sjeremylt CeedScalar *v11,CeedScalar *v12, 8388795c945Sjeremylt CeedScalar *v13,CeedScalar *v14, 839418fb8c2Sjeremylt CeedScalar *v15,int *err), 840288c0443SJeremy L Thompson const char *source, int *qf, int *err, 841288c0443SJeremy L Thompson fortran_charlen_t source_len) { 842288c0443SJeremy L Thompson FIX_STRING(source); 843d7b241e6Sjeremylt if (CeedQFunction_count == CeedQFunction_count_max) { 844d7b241e6Sjeremylt CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 845d7b241e6Sjeremylt CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 846d7b241e6Sjeremylt } 847d7b241e6Sjeremylt 848d7b241e6Sjeremylt CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 849d1d35e2fSjeremylt *err = CeedQFunctionCreateInterior(Ceed_dict[*ceed], *vec_length, 850288c0443SJeremy L Thompson CeedQFunctionFortranStub, source_c, qf_); 851d7b241e6Sjeremylt 852d7b241e6Sjeremylt if (*err == 0) { 853d7b241e6Sjeremylt *qf = CeedQFunction_count++; 854d7b241e6Sjeremylt CeedQFunction_n++; 855d7b241e6Sjeremylt } 856d7b241e6Sjeremylt 857777ff853SJeremy L Thompson CeedFortranContext fctxdata; 858777ff853SJeremy L Thompson *err = CeedCalloc(1, &fctxdata); 859d7b241e6Sjeremylt if (*err) return; 8608cb0412aSJeremy L Thompson fctxdata->f = f; fctxdata->inner_ctx = NULL; 861777ff853SJeremy L Thompson CeedQFunctionContext fctx; 862777ff853SJeremy L Thompson *err = CeedQFunctionContextCreate(Ceed_dict[*ceed], &fctx); 863777ff853SJeremy L Thompson if (*err) return; 864777ff853SJeremy L Thompson *err = CeedQFunctionContextSetData(fctx, CEED_MEM_HOST, CEED_OWN_POINTER, 865777ff853SJeremy L Thompson sizeof(*fctxdata), fctxdata); 866777ff853SJeremy L Thompson if (*err) return; 867777ff853SJeremy L Thompson *err = CeedQFunctionSetContext(*qf_, fctx); 868777ff853SJeremy L Thompson if (*err) return; 869777ff853SJeremy L Thompson CeedQFunctionContextDestroy(&fctx); 870777ff853SJeremy L Thompson if (*err) return; 871d7b241e6Sjeremylt 872777ff853SJeremy L Thompson *err = CeedQFunctionSetFortranStatus(*qf_, true); 873d7b241e6Sjeremylt } 874d7b241e6Sjeremylt 875288c0443SJeremy L Thompson #define fCeedQFunctionCreateInteriorByName \ 876288c0443SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncreateinteriorbyname, CEEDQFUNCTIONCREATEINTERIORBYNAME) 877*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionCreateInteriorByName(int *ceed, const char *name, 878*3b271f31SJed Brown int *qf, 879288c0443SJeremy L Thompson int *err, fortran_charlen_t name_len) { 880288c0443SJeremy L Thompson FIX_STRING(name); 881288c0443SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 882288c0443SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 883288c0443SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 884288c0443SJeremy L Thompson } 885288c0443SJeremy L Thompson 886288c0443SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 887288c0443SJeremy L Thompson *err = CeedQFunctionCreateInteriorByName(Ceed_dict[*ceed], name_c, qf_); 888288c0443SJeremy L Thompson 889288c0443SJeremy L Thompson if (*err == 0) { 890288c0443SJeremy L Thompson *qf = CeedQFunction_count++; 891288c0443SJeremy L Thompson CeedQFunction_n++; 892288c0443SJeremy L Thompson } 893288c0443SJeremy L Thompson } 894288c0443SJeremy L Thompson 8950219ea01SJeremy L Thompson #define fCeedQFunctionCreateIdentity \ 8960219ea01SJeremy L Thompson FORTRAN_NAME(ceedqfunctioncreateidentity, CEEDQFUNCTIONCREATEIDENTITY) 897*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionCreateIdentity(int *ceed, int *size, int *inmode, 89860f77c51Sjeremylt int *outmode, int *qf, int *err) { 8990219ea01SJeremy L Thompson if (CeedQFunction_count == CeedQFunction_count_max) { 9000219ea01SJeremy L Thompson CeedQFunction_count_max += CeedQFunction_count_max/2 + 1; 9010219ea01SJeremy L Thompson CeedRealloc(CeedQFunction_count_max, &CeedQFunction_dict); 9020219ea01SJeremy L Thompson } 9030219ea01SJeremy L Thompson 9040219ea01SJeremy L Thompson CeedQFunction *qf_ = &CeedQFunction_dict[CeedQFunction_count]; 905399bb197SJed Brown *err = CeedQFunctionCreateIdentity(Ceed_dict[*ceed], *size, 906399bb197SJed Brown (CeedEvalMode)*inmode, 907399bb197SJed Brown (CeedEvalMode)*outmode, qf_); 9080219ea01SJeremy L Thompson 9090219ea01SJeremy L Thompson if (*err == 0) { 9100219ea01SJeremy L Thompson *qf = CeedQFunction_count++; 9110219ea01SJeremy L Thompson CeedQFunction_n++; 9120219ea01SJeremy L Thompson } 9130219ea01SJeremy L Thompson } 9140219ea01SJeremy L Thompson 915d7b241e6Sjeremylt #define fCeedQFunctionAddInput \ 916d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionaddinput,CEEDQFUNCTIONADDINPUT) 917*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionAddInput(int *qf, const char *field_name, 918d1d35e2fSjeremylt CeedInt *num_comp, CeedEvalMode *eval_mode, int *err, 919d1d35e2fSjeremylt fortran_charlen_t field_name_len) { 920d1d35e2fSjeremylt FIX_STRING(field_name); 921d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 922d7b241e6Sjeremylt 923d1d35e2fSjeremylt *err = CeedQFunctionAddInput(qf_, field_name_c, *num_comp, *eval_mode); 924d7b241e6Sjeremylt } 925d7b241e6Sjeremylt 926d7b241e6Sjeremylt #define fCeedQFunctionAddOutput \ 927d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionaddoutput,CEEDQFUNCTIONADDOUTPUT) 928*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionAddOutput(int *qf, const char *field_name, 929d1d35e2fSjeremylt CeedInt *num_comp, CeedEvalMode *eval_mode, int *err, 930d1d35e2fSjeremylt fortran_charlen_t field_name_len) { 931d1d35e2fSjeremylt FIX_STRING(field_name); 932d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 933d7b241e6Sjeremylt 934d1d35e2fSjeremylt *err = CeedQFunctionAddOutput(qf_, field_name_c, *num_comp, *eval_mode); 935d7b241e6Sjeremylt } 936d7b241e6Sjeremylt 937418fb8c2Sjeremylt #define fCeedQFunctionSetContext \ 938418fb8c2Sjeremylt FORTRAN_NAME(ceedqfunctionsetcontext,CEEDQFUNCTIONSETCONTEXT) 939*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionSetContext(int *qf, int *ctx, int *err) { 940418fb8c2Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 941777ff853SJeremy L Thompson CeedQFunctionContext ctx_ = CeedQFunctionContext_dict[*ctx]; 942418fb8c2Sjeremylt 943777ff853SJeremy L Thompson CeedQFunctionContext fctx; 944777ff853SJeremy L Thompson *err = CeedQFunctionGetContext(qf_, &fctx); 945777ff853SJeremy L Thompson if (*err) return; 946777ff853SJeremy L Thompson CeedFortranContext fctxdata; 947777ff853SJeremy L Thompson *err = CeedQFunctionContextGetData(fctx, CEED_MEM_HOST, &fctxdata); 948777ff853SJeremy L Thompson if (*err) return; 9498cb0412aSJeremy L Thompson fctxdata->inner_ctx = ctx_; 950777ff853SJeremy L Thompson *err = CeedQFunctionContextRestoreData(fctx, (void **)&fctxdata); 951418fb8c2Sjeremylt } 952418fb8c2Sjeremylt 95375affc3bSjeremylt #define fCeedQFunctionView \ 95475affc3bSjeremylt FORTRAN_NAME(ceedqfunctionview,CEEDQFUNCTIONVIEW) 955*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionView(int *qf, int *err) { 95675affc3bSjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 95775affc3bSjeremylt 95875affc3bSjeremylt *err = CeedQFunctionView(qf_, stdout); 95975affc3bSjeremylt } 96075affc3bSjeremylt 961d7b241e6Sjeremylt #define fCeedQFunctionApply \ 962d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctionapply,CEEDQFUNCTIONAPPLY) 963d7b241e6Sjeremylt //TODO Need Fixing, double pointer 964*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionApply(int *qf, int *Q, 965aedaa0e5Sjeremylt int *u, int *u1, int *u2, int *u3, 966aedaa0e5Sjeremylt int *u4, int *u5, int *u6, int *u7, 967aedaa0e5Sjeremylt int *u8, int *u9, int *u10, int *u11, 968aedaa0e5Sjeremylt int *u12, int *u13, int *u14, int *u15, 969aedaa0e5Sjeremylt int *v, int *v1, int *v2, int *v3, 970aedaa0e5Sjeremylt int *v4, int *v5, int *v6, int *v7, 971aedaa0e5Sjeremylt int *v8, int *v9, int *v10, int *v11, 972aedaa0e5Sjeremylt int *v12, int *v13, int *v14, int *v15, int *err) { 973d7b241e6Sjeremylt CeedQFunction qf_ = CeedQFunction_dict[*qf]; 974aedaa0e5Sjeremylt CeedVector *in; 975bf4cb664SJeremy L Thompson *err = CeedCalloc(CEED_FIELD_MAX, &in); 976d7b241e6Sjeremylt if (*err) return; 977aedaa0e5Sjeremylt in[0] = *u==FORTRAN_NULL?NULL:CeedVector_dict[*u]; 978aedaa0e5Sjeremylt in[1] = *u1==FORTRAN_NULL?NULL:CeedVector_dict[*u1]; 979aedaa0e5Sjeremylt in[2] = *u2==FORTRAN_NULL?NULL:CeedVector_dict[*u2]; 980aedaa0e5Sjeremylt in[3] = *u3==FORTRAN_NULL?NULL:CeedVector_dict[*u3]; 981aedaa0e5Sjeremylt in[4] = *u4==FORTRAN_NULL?NULL:CeedVector_dict[*u4]; 982aedaa0e5Sjeremylt in[5] = *u5==FORTRAN_NULL?NULL:CeedVector_dict[*u5]; 983aedaa0e5Sjeremylt in[6] = *u6==FORTRAN_NULL?NULL:CeedVector_dict[*u6]; 984aedaa0e5Sjeremylt in[7] = *u7==FORTRAN_NULL?NULL:CeedVector_dict[*u7]; 985aedaa0e5Sjeremylt in[8] = *u8==FORTRAN_NULL?NULL:CeedVector_dict[*u8]; 986aedaa0e5Sjeremylt in[9] = *u9==FORTRAN_NULL?NULL:CeedVector_dict[*u9]; 987aedaa0e5Sjeremylt in[10] = *u10==FORTRAN_NULL?NULL:CeedVector_dict[*u10]; 988aedaa0e5Sjeremylt in[11] = *u11==FORTRAN_NULL?NULL:CeedVector_dict[*u11]; 989aedaa0e5Sjeremylt in[12] = *u12==FORTRAN_NULL?NULL:CeedVector_dict[*u12]; 990aedaa0e5Sjeremylt in[13] = *u13==FORTRAN_NULL?NULL:CeedVector_dict[*u13]; 991aedaa0e5Sjeremylt in[14] = *u14==FORTRAN_NULL?NULL:CeedVector_dict[*u14]; 992aedaa0e5Sjeremylt in[15] = *u15==FORTRAN_NULL?NULL:CeedVector_dict[*u15]; 993aedaa0e5Sjeremylt CeedVector *out; 994bf4cb664SJeremy L Thompson *err = CeedCalloc(CEED_FIELD_MAX, &out); 995d7b241e6Sjeremylt if (*err) return; 996aedaa0e5Sjeremylt out[0] = *v==FORTRAN_NULL?NULL:CeedVector_dict[*v]; 997aedaa0e5Sjeremylt out[1] = *v1==FORTRAN_NULL?NULL:CeedVector_dict[*v1]; 998aedaa0e5Sjeremylt out[2] = *v2==FORTRAN_NULL?NULL:CeedVector_dict[*v2]; 999aedaa0e5Sjeremylt out[3] = *v3==FORTRAN_NULL?NULL:CeedVector_dict[*v3]; 1000aedaa0e5Sjeremylt out[4] = *v4==FORTRAN_NULL?NULL:CeedVector_dict[*v4]; 1001aedaa0e5Sjeremylt out[5] = *v5==FORTRAN_NULL?NULL:CeedVector_dict[*v5]; 1002aedaa0e5Sjeremylt out[6] = *v6==FORTRAN_NULL?NULL:CeedVector_dict[*v6]; 1003aedaa0e5Sjeremylt out[7] = *v7==FORTRAN_NULL?NULL:CeedVector_dict[*v7]; 1004aedaa0e5Sjeremylt out[8] = *v8==FORTRAN_NULL?NULL:CeedVector_dict[*v8]; 1005aedaa0e5Sjeremylt out[9] = *v9==FORTRAN_NULL?NULL:CeedVector_dict[*v9]; 1006aedaa0e5Sjeremylt out[10] = *v10==FORTRAN_NULL?NULL:CeedVector_dict[*v10]; 1007aedaa0e5Sjeremylt out[11] = *v11==FORTRAN_NULL?NULL:CeedVector_dict[*v11]; 1008aedaa0e5Sjeremylt out[12] = *v12==FORTRAN_NULL?NULL:CeedVector_dict[*v12]; 1009aedaa0e5Sjeremylt out[13] = *v13==FORTRAN_NULL?NULL:CeedVector_dict[*v13]; 1010aedaa0e5Sjeremylt out[14] = *v14==FORTRAN_NULL?NULL:CeedVector_dict[*v14]; 1011aedaa0e5Sjeremylt out[15] = *v15==FORTRAN_NULL?NULL:CeedVector_dict[*v15]; 1012aedaa0e5Sjeremylt *err = CeedQFunctionApply(qf_, *Q, in, out); 1013d7b241e6Sjeremylt if (*err) return; 1014d7b241e6Sjeremylt 1015d7b241e6Sjeremylt *err = CeedFree(&in); 1016d7b241e6Sjeremylt if (*err) return; 1017d7b241e6Sjeremylt *err = CeedFree(&out); 1018d7b241e6Sjeremylt } 1019d7b241e6Sjeremylt 1020d7b241e6Sjeremylt #define fCeedQFunctionDestroy \ 1021d7b241e6Sjeremylt FORTRAN_NAME(ceedqfunctiondestroy,CEEDQFUNCTIONDESTROY) 1022*3b271f31SJed Brown CEED_EXTERN void fCeedQFunctionDestroy(int *qf, int *err) { 1023b22696a8SJeremy L Thompson if (*qf == FORTRAN_NULL) return; 1024c4b814baSjeremylt 1025c4b814baSjeremylt *err = CeedQFunctionDestroy(&CeedQFunction_dict[*qf]); 102616595307SJeremy L Thompson if (*err == 0) { 102716595307SJeremy L Thompson *qf = FORTRAN_NULL; 1028d7b241e6Sjeremylt CeedQFunction_n--; 1029d7b241e6Sjeremylt if (CeedQFunction_n == 0) { 1030d7b241e6Sjeremylt *err = CeedFree(&CeedQFunction_dict); 1031d7b241e6Sjeremylt CeedQFunction_count = 0; 1032d7b241e6Sjeremylt CeedQFunction_count_max = 0; 1033d7b241e6Sjeremylt } 1034d7b241e6Sjeremylt } 103516595307SJeremy L Thompson } 1036d7b241e6Sjeremylt 1037b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 1038b00b43a6SJeremy L Thompson // CeedOperator 1039b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 1040d7b241e6Sjeremylt static CeedOperator *CeedOperator_dict = NULL; 1041d7b241e6Sjeremylt static int CeedOperator_count = 0; 1042d7b241e6Sjeremylt static int CeedOperator_n = 0; 1043d7b241e6Sjeremylt static int CeedOperator_count_max = 0; 1044d7b241e6Sjeremylt 1045d7b241e6Sjeremylt #define fCeedOperatorCreate \ 1046d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorcreate, CEEDOPERATORCREATE) 1047*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorCreate(int *ceed, 1048d7b241e6Sjeremylt int *qf, int *dqf, int *dqfT, int *op, int *err) { 1049777ff853SJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) { 1050777ff853SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1051777ff853SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1052777ff853SJeremy L Thompson } 1053d7b241e6Sjeremylt 1054d7b241e6Sjeremylt CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 1055d7b241e6Sjeremylt 1056442e7f0bSjeremylt CeedQFunction dqf_ = CEED_QFUNCTION_NONE, dqfT_ = CEED_QFUNCTION_NONE; 1057442e7f0bSjeremylt if (*dqf != FORTRAN_QFUNCTION_NONE) dqf_ = CeedQFunction_dict[*dqf ]; 1058442e7f0bSjeremylt if (*dqfT != FORTRAN_QFUNCTION_NONE) dqfT_ = CeedQFunction_dict[*dqfT]; 1059d7b241e6Sjeremylt 1060d7b241e6Sjeremylt *err = CeedOperatorCreate(Ceed_dict[*ceed], CeedQFunction_dict[*qf], dqf_, 1061d7b241e6Sjeremylt dqfT_, op_); 1062d7b241e6Sjeremylt if (*err) return; 1063d7b241e6Sjeremylt *op = CeedOperator_count++; 1064d7b241e6Sjeremylt CeedOperator_n++; 1065d7b241e6Sjeremylt } 1066d7b241e6Sjeremylt 106752d6035fSJeremy L Thompson #define fCeedCompositeOperatorCreate \ 106852d6035fSJeremy L Thompson FORTRAN_NAME(ceedcompositeoperatorcreate, CEEDCOMPOSITEOPERATORCREATE) 1069*3b271f31SJed Brown CEED_EXTERN void fCeedCompositeOperatorCreate(int *ceed, int *op, int *err) { 1070777ff853SJeremy L Thompson if (CeedOperator_count == CeedOperator_count_max) { 1071777ff853SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1072777ff853SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1073777ff853SJeremy L Thompson } 107452d6035fSJeremy L Thompson 107552d6035fSJeremy L Thompson CeedOperator *op_ = &CeedOperator_dict[CeedOperator_count]; 107652d6035fSJeremy L Thompson 107752d6035fSJeremy L Thompson *err = CeedCompositeOperatorCreate(Ceed_dict[*ceed], op_); 107852d6035fSJeremy L Thompson if (*err) return; 107952d6035fSJeremy L Thompson *op = CeedOperator_count++; 108052d6035fSJeremy L Thompson CeedOperator_n++; 108152d6035fSJeremy L Thompson } 108252d6035fSJeremy L Thompson 1083d7b241e6Sjeremylt #define fCeedOperatorSetField \ 1084d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorsetfield,CEEDOPERATORSETFIELD) 1085*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorSetField(int *op, const char *field_name, int *r, 1086*3b271f31SJed Brown int *b, 1087d1d35e2fSjeremylt int *v, int *err, fortran_charlen_t field_name_len) { 1088d1d35e2fSjeremylt FIX_STRING(field_name); 1089d7b241e6Sjeremylt CeedElemRestriction r_; 1090d7b241e6Sjeremylt CeedBasis b_; 1091d7b241e6Sjeremylt CeedVector v_; 1092d7b241e6Sjeremylt 1093d7b241e6Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 1094d7b241e6Sjeremylt 1095d7b241e6Sjeremylt if (*r == FORTRAN_NULL) { 1096d7b241e6Sjeremylt r_ = NULL; 109715910d16Sjeremylt } else if (*r == FORTRAN_ELEMRESTRICTION_NONE) { 109815910d16Sjeremylt r_ = CEED_ELEMRESTRICTION_NONE; 1099d7b241e6Sjeremylt } else { 1100d7b241e6Sjeremylt r_ = CeedElemRestriction_dict[*r]; 1101d7b241e6Sjeremylt } 11024dccadb6Sjeremylt 1103d7b241e6Sjeremylt if (*b == FORTRAN_NULL) { 1104d7b241e6Sjeremylt b_ = NULL; 1105783c99b3SValeria Barra } else if (*b == FORTRAN_BASIS_COLLOCATED) { 1106783c99b3SValeria Barra b_ = CEED_BASIS_COLLOCATED; 1107d7b241e6Sjeremylt } else { 1108d7b241e6Sjeremylt b_ = CeedBasis_dict[*b]; 1109d7b241e6Sjeremylt } 1110d7b241e6Sjeremylt if (*v == FORTRAN_NULL) { 1111d7b241e6Sjeremylt v_ = NULL; 1112d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_ACTIVE) { 1113d7b241e6Sjeremylt v_ = CEED_VECTOR_ACTIVE; 1114d7b241e6Sjeremylt } else if (*v == FORTRAN_VECTOR_NONE) { 1115d7b241e6Sjeremylt v_ = CEED_VECTOR_NONE; 1116d7b241e6Sjeremylt } else { 1117d7b241e6Sjeremylt v_ = CeedVector_dict[*v]; 1118d7b241e6Sjeremylt } 1119d7b241e6Sjeremylt 1120d1d35e2fSjeremylt *err = CeedOperatorSetField(op_, field_name_c, r_, b_, v_); 1121d7b241e6Sjeremylt } 1122d7b241e6Sjeremylt 112352d6035fSJeremy L Thompson #define fCeedCompositeOperatorAddSub \ 112452d6035fSJeremy L Thompson FORTRAN_NAME(ceedcompositeoperatoraddsub, CEEDCOMPOSITEOPERATORADDSUB) 1125*3b271f31SJed Brown CEED_EXTERN void fCeedCompositeOperatorAddSub(int *compositeop, int *subop, 1126*3b271f31SJed Brown int *err) { 112752d6035fSJeremy L Thompson CeedOperator compositeop_ = CeedOperator_dict[*compositeop]; 112852d6035fSJeremy L Thompson CeedOperator subop_ = CeedOperator_dict[*subop]; 112952d6035fSJeremy L Thompson 113052d6035fSJeremy L Thompson *err = CeedCompositeOperatorAddSub(compositeop_, subop_); 113152d6035fSJeremy L Thompson if (*err) return; 113252d6035fSJeremy L Thompson } 113352d6035fSJeremy L Thompson 1134d99fa3c5SJeremy L Thompson #define fCeedOperatorLinearAssembleQFunction \ 1135d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatorlinearassembleqfunction, CEEDOPERATORLINEARASSEMBLEQFUNCTION) 1136*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorLinearAssembleQFunction(int *op, 1137*3b271f31SJed Brown int *assembledvec, 11381d102b48SJeremy L Thompson int *assembledrstr, int *rqst, int *err) { 11391d102b48SJeremy L Thompson // Vector 11401d102b48SJeremy L Thompson if (CeedVector_count == CeedVector_count_max) { 11411d102b48SJeremy L Thompson CeedVector_count_max += CeedVector_count_max/2 + 1; 11421d102b48SJeremy L Thompson CeedRealloc(CeedVector_count_max, &CeedVector_dict); 11431d102b48SJeremy L Thompson } 11441d102b48SJeremy L Thompson CeedVector *assembledvec_ = &CeedVector_dict[CeedVector_count]; 11451d102b48SJeremy L Thompson 11461d102b48SJeremy L Thompson // Restriction 11471d102b48SJeremy L Thompson if (CeedElemRestriction_count == CeedElemRestriction_count_max) { 11481d102b48SJeremy L Thompson CeedElemRestriction_count_max += CeedElemRestriction_count_max/2 + 1; 11491d102b48SJeremy L Thompson CeedRealloc(CeedElemRestriction_count_max, &CeedElemRestriction_dict); 11501d102b48SJeremy L Thompson } 11511d102b48SJeremy L Thompson CeedElemRestriction *rstr_ = 11521d102b48SJeremy L Thompson &CeedElemRestriction_dict[CeedElemRestriction_count]; 11531d102b48SJeremy L Thompson 11541d102b48SJeremy L Thompson int createRequest = 1; 11551d102b48SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 11561d102b48SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 11571d102b48SJeremy L Thompson createRequest = 0; 11581d102b48SJeremy L Thompson } 11591d102b48SJeremy L Thompson 11601d102b48SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 11611d102b48SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max/2 + 1; 11621d102b48SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 11631d102b48SJeremy L Thompson } 11641d102b48SJeremy L Thompson 11651d102b48SJeremy L Thompson CeedRequest *rqst_; 11661d102b48SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 11671d102b48SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 11681d102b48SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 11691d102b48SJeremy L Thompson 117080ac2e43SJeremy L Thompson *err = CeedOperatorLinearAssembleQFunction(CeedOperator_dict[*op], 11711d102b48SJeremy L Thompson assembledvec_, rstr_, rqst_); 11721d102b48SJeremy L Thompson if (*err) return; 11731d102b48SJeremy L Thompson if (createRequest) { 11741d102b48SJeremy L Thompson *rqst = CeedRequest_count++; 11751d102b48SJeremy L Thompson CeedRequest_n++; 11761d102b48SJeremy L Thompson } 11771d102b48SJeremy L Thompson 11781d102b48SJeremy L Thompson if (*err == 0) { 11791d102b48SJeremy L Thompson *assembledrstr = CeedElemRestriction_count++; 11801d102b48SJeremy L Thompson CeedElemRestriction_n++; 11811d102b48SJeremy L Thompson *assembledvec = CeedVector_count++; 11821d102b48SJeremy L Thompson CeedVector_n++; 11831d102b48SJeremy L Thompson } 11841d102b48SJeremy L Thompson } 11851d102b48SJeremy L Thompson 1186d99fa3c5SJeremy L Thompson #define fCeedOperatorLinearAssembleDiagonal \ 1187d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatorlinearassemblediagonal, CEEDOPERATORLINEARASSEMBLEDIAGONAL) 1188*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorLinearAssembleDiagonal(int *op, int *assembledvec, 1189b7ec98d8SJeremy L Thompson int *rqst, int *err) { 1190b7ec98d8SJeremy L Thompson int createRequest = 1; 1191b7ec98d8SJeremy L Thompson // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1192b7ec98d8SJeremy L Thompson if (*rqst == -1 || *rqst == -2) { 1193b7ec98d8SJeremy L Thompson createRequest = 0; 1194b7ec98d8SJeremy L Thompson } 1195b7ec98d8SJeremy L Thompson 1196b7ec98d8SJeremy L Thompson if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1197b7ec98d8SJeremy L Thompson CeedRequest_count_max += CeedRequest_count_max/2 + 1; 1198b7ec98d8SJeremy L Thompson CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1199b7ec98d8SJeremy L Thompson } 1200b7ec98d8SJeremy L Thompson 1201b7ec98d8SJeremy L Thompson CeedRequest *rqst_; 1202b7ec98d8SJeremy L Thompson if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1203b7ec98d8SJeremy L Thompson else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1204b7ec98d8SJeremy L Thompson else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1205b7ec98d8SJeremy L Thompson 120680ac2e43SJeremy L Thompson *err = CeedOperatorLinearAssembleDiagonal(CeedOperator_dict[*op], 12072bba3ffaSJeremy L Thompson CeedVector_dict[*assembledvec], rqst_); 1208b7ec98d8SJeremy L Thompson if (*err) return; 1209b7ec98d8SJeremy L Thompson if (createRequest) { 1210b7ec98d8SJeremy L Thompson *rqst = CeedRequest_count++; 1211b7ec98d8SJeremy L Thompson CeedRequest_n++; 1212b7ec98d8SJeremy L Thompson } 1213b7ec98d8SJeremy L Thompson } 1214b7ec98d8SJeremy L Thompson 1215d99fa3c5SJeremy L Thompson #define fCeedOperatorMultigridLevelCreate \ 1216d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatormultigridlevelcreate, CEEDOPERATORMULTIGRIDLEVELCREATE) 1217*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorMultigridLevelCreate(int *opFine, int *pMultFine, 1218d99fa3c5SJeremy L Thompson int *rstrCoarse, int *basisCoarse, int *opCoarse, 1219d99fa3c5SJeremy L Thompson int *opProlong, int *opRestrict, int *err) { 1220d99fa3c5SJeremy L Thompson // Operators 1221d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1222d99fa3c5SJeremy L Thompson 1223d99fa3c5SJeremy L Thompson // C interface call 1224d99fa3c5SJeremy L Thompson *err = CeedOperatorMultigridLevelCreate( 1225d99fa3c5SJeremy L Thompson CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], 1226d99fa3c5SJeremy L Thompson CeedElemRestriction_dict[*rstrCoarse], 1227d99fa3c5SJeremy L Thompson CeedBasis_dict[*basisCoarse], 1228d99fa3c5SJeremy L Thompson &opCoarse_, &opProlong_, &opRestrict_); 1229d99fa3c5SJeremy L Thompson 1230d99fa3c5SJeremy L Thompson if (*err) return; 1231d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1232d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1233d99fa3c5SJeremy L Thompson } 1234d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1235d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1236d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1237d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1238d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1239d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1240d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1241d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1242d99fa3c5SJeremy L Thompson } 1243d99fa3c5SJeremy L Thompson 1244d99fa3c5SJeremy L Thompson #define fCeedOperatorMultigridLevelCreateTensorH1 \ 1245d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatormultigridlevelcreatetensorh1, CEEDOPERATORMULTIGRIDLEVELCREATETENSORH1) 1246*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorMultigridLevelCreateTensorH1(int *opFine, 1247*3b271f31SJed Brown int *pMultFine, 1248d99fa3c5SJeremy L Thompson int *rstrCoarse, int *basisCoarse, const CeedScalar *interpCtoF, 1249d99fa3c5SJeremy L Thompson int *opCoarse, int *opProlong, int *opRestrict, int *err) { 1250d99fa3c5SJeremy L Thompson // Operators 1251d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1252d99fa3c5SJeremy L Thompson 1253d99fa3c5SJeremy L Thompson // C interface call 1254d99fa3c5SJeremy L Thompson *err = CeedOperatorMultigridLevelCreateTensorH1( 1255d99fa3c5SJeremy L Thompson CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], 1256d99fa3c5SJeremy L Thompson CeedElemRestriction_dict[*rstrCoarse], CeedBasis_dict[*basisCoarse], 1257d99fa3c5SJeremy L Thompson interpCtoF, &opCoarse_, &opProlong_, &opRestrict_); 1258d99fa3c5SJeremy L Thompson 1259d99fa3c5SJeremy L Thompson if (*err) return; 1260d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1261d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1262d99fa3c5SJeremy L Thompson } 1263d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1264d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1265d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1266d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1267d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1268d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1269d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1270d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1271d99fa3c5SJeremy L Thompson } 1272d99fa3c5SJeremy L Thompson 1273d99fa3c5SJeremy L Thompson #define fCeedOperatorMultigridLevelCreateH1 \ 1274d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatormultigridlevelcreateh1, CEEDOPERATORMULTIGRIDLEVELCREATEH1) 1275*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorMultigridLevelCreateH1(int *opFine, 1276*3b271f31SJed Brown int *pMultFine, 1277d99fa3c5SJeremy L Thompson int *rstrCoarse, int *basisCoarse, const CeedScalar *interpCtoF, 1278d99fa3c5SJeremy L Thompson int *opCoarse, int *opProlong, int *opRestrict, int *err) { 1279d99fa3c5SJeremy L Thompson // Operators 1280d99fa3c5SJeremy L Thompson CeedOperator opCoarse_, opProlong_, opRestrict_; 1281d99fa3c5SJeremy L Thompson 1282d99fa3c5SJeremy L Thompson // C interface call 1283d99fa3c5SJeremy L Thompson *err = CeedOperatorMultigridLevelCreateH1( 1284d99fa3c5SJeremy L Thompson CeedOperator_dict[*opFine], CeedVector_dict[*pMultFine], 1285d99fa3c5SJeremy L Thompson CeedElemRestriction_dict[*rstrCoarse], CeedBasis_dict[*basisCoarse], 1286d99fa3c5SJeremy L Thompson interpCtoF, &opCoarse_, &opProlong_, &opRestrict_); 1287d99fa3c5SJeremy L Thompson 1288d99fa3c5SJeremy L Thompson if (*err) return; 1289d99fa3c5SJeremy L Thompson while (CeedOperator_count + 2 >= CeedOperator_count_max) { 1290d99fa3c5SJeremy L Thompson CeedOperator_count_max += CeedOperator_count_max/2 + 1; 1291d99fa3c5SJeremy L Thompson } 1292d99fa3c5SJeremy L Thompson CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 1293d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opCoarse_; 1294d99fa3c5SJeremy L Thompson *opCoarse = CeedOperator_count++; 1295d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opProlong_; 1296d99fa3c5SJeremy L Thompson *opProlong = CeedOperator_count++; 1297d99fa3c5SJeremy L Thompson CeedOperator_dict[CeedOperator_count] = opRestrict_; 1298d99fa3c5SJeremy L Thompson *opRestrict = CeedOperator_count++; 1299d99fa3c5SJeremy L Thompson CeedOperator_n += 3; 1300d99fa3c5SJeremy L Thompson } 1301d99fa3c5SJeremy L Thompson 13022ebaca42Sjeremylt #define fCeedOperatorView \ 13032ebaca42Sjeremylt FORTRAN_NAME(ceedoperatorview,CEEDOPERATORVIEW) 1304*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorView(int *op, int *err) { 13052ebaca42Sjeremylt CeedOperator op_ = CeedOperator_dict[*op]; 13062ebaca42Sjeremylt 13072ebaca42Sjeremylt *err = CeedOperatorView(op_, stdout); 13082ebaca42Sjeremylt } 13092ebaca42Sjeremylt 1310d99fa3c5SJeremy L Thompson #define fCeedOperatorCreateFDMElementInverse \ 1311d99fa3c5SJeremy L Thompson FORTRAN_NAME(ceedoperatorcreatefdmelementinverse, CEEDOPERATORCREATEFDMELEMENTINVERSE) 1312*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorCreateFDMElementInverse(int *op, int *fdminv, 13133bd813ffSjeremylt int *rqst, int *err) { 13143bd813ffSjeremylt // Operator 13153bd813ffSjeremylt if (CeedOperator_count == CeedOperator_count_max) { 13163bd813ffSjeremylt CeedOperator_count_max += CeedOperator_count_max/2 + 1; 13173bd813ffSjeremylt CeedRealloc(CeedOperator_count_max, &CeedOperator_dict); 13183bd813ffSjeremylt } 13193bd813ffSjeremylt CeedOperator *fdminv_ = 13203bd813ffSjeremylt &CeedOperator_dict[CeedOperator_count]; 13213bd813ffSjeremylt 13223bd813ffSjeremylt int createRequest = 1; 13233bd813ffSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 13243bd813ffSjeremylt if (*rqst == -1 || *rqst == -2) { 13253bd813ffSjeremylt createRequest = 0; 13263bd813ffSjeremylt } 13273bd813ffSjeremylt 13283bd813ffSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 13293bd813ffSjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 13303bd813ffSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 13313bd813ffSjeremylt } 13323bd813ffSjeremylt 13333bd813ffSjeremylt CeedRequest *rqst_; 13343bd813ffSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 13353bd813ffSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 13363bd813ffSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 13373bd813ffSjeremylt 13383bd813ffSjeremylt *err = CeedOperatorCreateFDMElementInverse(CeedOperator_dict[*op], 13393bd813ffSjeremylt fdminv_, rqst_); 13403bd813ffSjeremylt if (*err) return; 13413bd813ffSjeremylt if (createRequest) { 13423bd813ffSjeremylt *rqst = CeedRequest_count++; 13433bd813ffSjeremylt CeedRequest_n++; 13443bd813ffSjeremylt } 13453bd813ffSjeremylt 13463bd813ffSjeremylt if (*err == 0) { 13473bd813ffSjeremylt *fdminv = CeedOperator_count++; 13483bd813ffSjeremylt CeedOperator_n++; 13493bd813ffSjeremylt } 13503bd813ffSjeremylt } 13513bd813ffSjeremylt 1352d7b241e6Sjeremylt #define fCeedOperatorApply FORTRAN_NAME(ceedoperatorapply, CEEDOPERATORAPPLY) 1353*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorApply(int *op, int *ustatevec, 1354d7b241e6Sjeremylt int *resvec, int *rqst, int *err) { 1355e97ff134Sjeremylt CeedVector ustatevec_ = (*ustatevec == FORTRAN_NULL) ? 1356e97ff134Sjeremylt NULL : (*ustatevec == FORTRAN_VECTOR_NONE ? 1357e97ff134Sjeremylt CEED_VECTOR_NONE : CeedVector_dict[*ustatevec]); 1358e97ff134Sjeremylt CeedVector resvec_ = (*resvec == FORTRAN_NULL) ? 1359e97ff134Sjeremylt NULL : (*resvec == FORTRAN_VECTOR_NONE ? 1360e97ff134Sjeremylt CEED_VECTOR_NONE : CeedVector_dict[*resvec]); 1361d7b241e6Sjeremylt 1362d7b241e6Sjeremylt int createRequest = 1; 1363d7b241e6Sjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1364d7b241e6Sjeremylt if (*rqst == -1 || *rqst == -2) { 1365d7b241e6Sjeremylt createRequest = 0; 1366d7b241e6Sjeremylt } 1367d7b241e6Sjeremylt 1368d7b241e6Sjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1369d7b241e6Sjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 1370d7b241e6Sjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1371d7b241e6Sjeremylt } 1372d7b241e6Sjeremylt 1373d7b241e6Sjeremylt CeedRequest *rqst_; 1374d7b241e6Sjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1375d7b241e6Sjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1376d7b241e6Sjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1377d7b241e6Sjeremylt 1378d7b241e6Sjeremylt *err = CeedOperatorApply(CeedOperator_dict[*op], 137912bf2c11SJed Brown ustatevec_, resvec_, rqst_); 1380d7b241e6Sjeremylt if (*err) return; 1381d7b241e6Sjeremylt if (createRequest) { 1382d7b241e6Sjeremylt *rqst = CeedRequest_count++; 1383d7b241e6Sjeremylt CeedRequest_n++; 1384d7b241e6Sjeremylt } 1385d7b241e6Sjeremylt } 1386d7b241e6Sjeremylt 1387cae8b89aSjeremylt #define fCeedOperatorApplyAdd FORTRAN_NAME(ceedoperatorapplyadd, CEEDOPERATORAPPLYADD) 1388*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorApplyAdd(int *op, int *ustatevec, 1389cae8b89aSjeremylt int *resvec, int *rqst, int *err) { 1390cae8b89aSjeremylt CeedVector ustatevec_ = *ustatevec == FORTRAN_NULL 1391cae8b89aSjeremylt ? NULL : CeedVector_dict[*ustatevec]; 1392cae8b89aSjeremylt CeedVector resvec_ = *resvec == FORTRAN_NULL 1393cae8b89aSjeremylt ? NULL : CeedVector_dict[*resvec]; 1394cae8b89aSjeremylt 1395cae8b89aSjeremylt int createRequest = 1; 1396cae8b89aSjeremylt // Check if input is CEED_REQUEST_ORDERED(-2) or CEED_REQUEST_IMMEDIATE(-1) 1397cae8b89aSjeremylt if (*rqst == -1 || *rqst == -2) { 1398cae8b89aSjeremylt createRequest = 0; 1399cae8b89aSjeremylt } 1400cae8b89aSjeremylt 1401cae8b89aSjeremylt if (createRequest && CeedRequest_count == CeedRequest_count_max) { 1402cae8b89aSjeremylt CeedRequest_count_max += CeedRequest_count_max/2 + 1; 1403cae8b89aSjeremylt CeedRealloc(CeedRequest_count_max, &CeedRequest_dict); 1404cae8b89aSjeremylt } 1405cae8b89aSjeremylt 1406cae8b89aSjeremylt CeedRequest *rqst_; 1407cae8b89aSjeremylt if (*rqst == -1) rqst_ = CEED_REQUEST_IMMEDIATE; 1408cae8b89aSjeremylt else if (*rqst == -2) rqst_ = CEED_REQUEST_ORDERED; 1409cae8b89aSjeremylt else rqst_ = &CeedRequest_dict[CeedRequest_count]; 1410cae8b89aSjeremylt 1411cae8b89aSjeremylt *err = CeedOperatorApplyAdd(CeedOperator_dict[*op], 1412cae8b89aSjeremylt ustatevec_, resvec_, rqst_); 1413cae8b89aSjeremylt if (*err) return; 1414cae8b89aSjeremylt if (createRequest) { 1415cae8b89aSjeremylt *rqst = CeedRequest_count++; 1416cae8b89aSjeremylt CeedRequest_n++; 1417cae8b89aSjeremylt } 1418cae8b89aSjeremylt } 1419cae8b89aSjeremylt 1420d7b241e6Sjeremylt #define fCeedOperatorApplyJacobian \ 1421d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatorapplyjacobian, CEEDOPERATORAPPLYJACOBIAN) 1422*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorApplyJacobian(int *op, int *qdatavec, 1423*3b271f31SJed Brown int *ustatevec, 14248795c945Sjeremylt int *dustatevec, int *dresvec, int *rqst, 14258795c945Sjeremylt int *err) { 1426d7b241e6Sjeremylt // TODO Uncomment this when CeedOperatorApplyJacobian is implemented 1427d7b241e6Sjeremylt // *err = CeedOperatorApplyJacobian(CeedOperator_dict[*op], CeedVector_dict[*qdatavec], 1428d7b241e6Sjeremylt // CeedVector_dict[*ustatevec], CeedVector_dict[*dustatevec], 1429d7b241e6Sjeremylt // CeedVector_dict[*dresvec], &CeedRequest_dict[*rqst]); 1430d7b241e6Sjeremylt } 1431d7b241e6Sjeremylt 1432d7b241e6Sjeremylt #define fCeedOperatorDestroy \ 1433d7b241e6Sjeremylt FORTRAN_NAME(ceedoperatordestroy, CEEDOPERATORDESTROY) 1434*3b271f31SJed Brown CEED_EXTERN void fCeedOperatorDestroy(int *op, int *err) { 1435b22696a8SJeremy L Thompson if (*op == FORTRAN_NULL) return; 1436d7b241e6Sjeremylt *err = CeedOperatorDestroy(&CeedOperator_dict[*op]); 143716595307SJeremy L Thompson if (*err == 0) { 143816595307SJeremy L Thompson *op = FORTRAN_NULL; 1439d7b241e6Sjeremylt CeedOperator_n--; 1440d7b241e6Sjeremylt if (CeedOperator_n == 0) { 1441d7b241e6Sjeremylt *err = CeedFree(&CeedOperator_dict); 1442d7b241e6Sjeremylt CeedOperator_count = 0; 1443d7b241e6Sjeremylt CeedOperator_count_max = 0; 1444d7b241e6Sjeremylt } 1445d7b241e6Sjeremylt } 144616595307SJeremy L Thompson } 1447b00b43a6SJeremy L Thompson 1448b00b43a6SJeremy L Thompson // ----------------------------------------------------------------------------- 1449