1ec3da8bcSJed Brown /// Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at 2ec3da8bcSJed Brown /// the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights 3ec3da8bcSJed Brown /// reserved. See files LICENSE and NOTICE for details. 4ec3da8bcSJed Brown /// 5ec3da8bcSJed Brown /// This file is part of CEED, a collection of benchmarks, miniapps, software 6ec3da8bcSJed Brown /// libraries and APIs for efficient high-order finite element and spectral 7ec3da8bcSJed Brown /// element discretizations for exascale applications. For more information and 83d8e8822SJeremy L Thompson /// source code availability see http://github.com/ceed 9ec3da8bcSJed Brown /// 10ec3da8bcSJed Brown /// The CEED research is supported by the Exascale Computing Project 17-SC-20-SC, 11ec3da8bcSJed Brown /// a collaborative effort of two U.S. Department of Energy organizations (Office 12ec3da8bcSJed Brown /// of Science and the National Nuclear Security Administration) responsible for 13ec3da8bcSJed Brown /// the planning and preparation of a capable exascale ecosystem, including 14ec3da8bcSJed Brown /// software, applications, hardware, advanced system engineering and early 15ec3da8bcSJed Brown /// testbed platforms, in support of the nation's exascale computing imperative. 16ec3da8bcSJed Brown 17ec3da8bcSJed Brown /// @file 18ec3da8bcSJed Brown /// Public header for user and utility components of libCEED 19ec3da8bcSJed Brown #ifndef _ceed_h 20ec3da8bcSJed Brown #define _ceed_h 21ec3da8bcSJed Brown 22ec3da8bcSJed Brown /// @defgroup Ceed Ceed: core components 23ec3da8bcSJed Brown /// @defgroup CeedVector CeedVector: storing and manipulating vectors 24ec3da8bcSJed Brown /// @defgroup CeedElemRestriction CeedElemRestriction: restriction from local vectors to elements 25ec3da8bcSJed Brown /// @defgroup CeedBasis CeedBasis: fully discrete finite element-like objects 26ec3da8bcSJed Brown /// @defgroup CeedQFunction CeedQFunction: independent operations at quadrature points 27ec3da8bcSJed Brown /// @defgroup CeedOperator CeedOperator: composed FE-type operations on vectors 28ec3da8bcSJed Brown /// 29ec3da8bcSJed Brown /// @page FunctionCategories libCEED: Types of Functions 30ec3da8bcSJed Brown /// libCEED provides three different header files depending upon the type of 31ec3da8bcSJed Brown /// functions a user requires. 32ec3da8bcSJed Brown /// @section Utility Utility Functions 33ec3da8bcSJed Brown /// These functions are intended general utilities that may be useful to 34ec3da8bcSJed Brown /// libCEED developers and users. These functions can generally be found in 35ec3da8bcSJed Brown /// "ceed.h". 36ec3da8bcSJed Brown /// @section User User Functions 37ec3da8bcSJed Brown /// These functions are intended to be used by general users of libCEED 38ec3da8bcSJed Brown /// and can generally be found in "ceed.h". 39e9b533fbSJeremy L Thompson /// @section Advanced Advanced Functions 40e9b533fbSJeremy L Thompson /// These functions are intended to be used by advanced users of libCEED 41e9b533fbSJeremy L Thompson /// and can generally be found in "ceed.h". 42ec3da8bcSJed Brown /// @section Backend Backend Developer Functions 43ec3da8bcSJed Brown /// These functions are intended to be used by backend developers of 44ec3da8bcSJed Brown /// libCEED and can generally be found in "ceed-backend.h". 45ec3da8bcSJed Brown /// @section Developer Library Developer Functions 46ec3da8bcSJed Brown /// These functions are intended to be used by library developers of 47ec3da8bcSJed Brown /// libCEED and can generally be found in "ceed-impl.h". 48ec3da8bcSJed Brown 493b271f31SJed Brown #if !defined(CEED_SKIP_VISIBILITY) 503b271f31SJed Brown # define CEED_VISIBILITY(mode) __attribute__((visibility(#mode))) 513b271f31SJed Brown #else 523b271f31SJed Brown # define CEED_VISIBILITY(mode) 533b271f31SJed Brown #endif 543b271f31SJed Brown 55ec3da8bcSJed Brown /** 56ec3da8bcSJed Brown CEED_EXTERN is used in this header to denote all publicly visible symbols. 57ec3da8bcSJed Brown 58ec3da8bcSJed Brown No other file should declare publicly visible symbols, thus it should never be 59ec3da8bcSJed Brown used outside ceed.h. 60ec3da8bcSJed Brown */ 61ec3da8bcSJed Brown #ifdef __cplusplus 623b271f31SJed Brown # define CEED_EXTERN extern "C" CEED_VISIBILITY(default) 63ec3da8bcSJed Brown #else 643b271f31SJed Brown # define CEED_EXTERN extern CEED_VISIBILITY(default) 65ec3da8bcSJed Brown #endif 66ec3da8bcSJed Brown 67ec3da8bcSJed Brown /** 68ec3da8bcSJed Brown @ingroup CeedQFunction 69ec3da8bcSJed Brown This macro populates the correct function annotations for User QFunction 70ec3da8bcSJed Brown source for code generation backends or populates default values for CPU 714b97e6d3SJames Wright backends. It also creates a variable `name_loc` populated with the correct 724b97e6d3SJames Wright source path for creating the respective User QFunction. 73ec3da8bcSJed Brown **/ 74ec3da8bcSJed Brown #ifndef CEED_QFUNCTION 75ec3da8bcSJed Brown #define CEED_QFUNCTION(name) \ 76ec3da8bcSJed Brown static const char name ## _loc[] = __FILE__ ":" #name; \ 77ec3da8bcSJed Brown static int name 78ec3da8bcSJed Brown #endif 79ec3da8bcSJed Brown 80ec3da8bcSJed Brown /** 81ec3da8bcSJed Brown @ingroup CeedQFunction 8277841947SLeila Ghaffari This macro populates the correct function annotations for User QFunction 8377841947SLeila Ghaffari helper function source for code generation backends or populates default 8477841947SLeila Ghaffari values for CPU backends. 8577841947SLeila Ghaffari **/ 8677841947SLeila Ghaffari #ifndef CEED_QFUNCTION_HELPER 8777841947SLeila Ghaffari #define CEED_QFUNCTION_HELPER static inline 8877841947SLeila Ghaffari #endif 8977841947SLeila Ghaffari 9077841947SLeila Ghaffari /** 9177841947SLeila Ghaffari @ingroup CeedQFunction 92ec3da8bcSJed Brown Using VLA syntax to reshape User QFunction inputs and outputs can make 93ec3da8bcSJed Brown user code more readable. VLA is a C99 feature that is not supported by 94ec3da8bcSJed Brown the C++ dialect used by CUDA. This macro allows users to use the VLA 95ec3da8bcSJed Brown syntax with the CUDA backends. 96ec3da8bcSJed Brown **/ 97ec3da8bcSJed Brown #ifndef CEED_Q_VLA 98ec3da8bcSJed Brown # define CEED_Q_VLA Q 99ec3da8bcSJed Brown #endif 100ec3da8bcSJed Brown 101ec3da8bcSJed Brown /** 102ec3da8bcSJed Brown @ingroup Ceed 103ec3da8bcSJed Brown This macro provides the appropriate SIMD Pragma for the compilation 104ec3da8bcSJed Brown environment. Code generation backends may redefine this macro, as needed. 105ec3da8bcSJed Brown **/ 106ec3da8bcSJed Brown #ifndef CeedPragmaSIMD 107ec3da8bcSJed Brown # if defined(__INTEL_COMPILER) 108ec3da8bcSJed Brown # define CeedPragmaSIMD _Pragma("vector") 109ec3da8bcSJed Brown // Cannot use Intel pragma ivdep because it miscompiles unpacking symmetric tensors, as in 110ec3da8bcSJed Brown // Poisson2DApply, where the SIMD loop body contains temporaries such as the following. 111ec3da8bcSJed Brown // 112ec3da8bcSJed Brown // const CeedScalar dXdxdXdxT[2][2] = {{qd[i+0*Q], qd[i+2*Q]}, 113ec3da8bcSJed Brown // {qd[i+2*Q], qd[i+1*Q]}}; 114ec3da8bcSJed Brown // for (int j=0; j<2; j++) 115ec3da8bcSJed Brown // vg[i+j*Q] = (du[0] * dXdxdXdxT[0][j] + du[1] * dXdxdXdxT[1][j]); 116ec3da8bcSJed Brown // 117ec3da8bcSJed Brown // Miscompilation with pragma ivdep observed with icc (ICC) 19.0.5.281 20190815 118ec3da8bcSJed Brown // at -O2 and above. 119ec3da8bcSJed Brown # elif defined(__GNUC__) && __GNUC__ >= 5 120ec3da8bcSJed Brown # define CeedPragmaSIMD _Pragma("GCC ivdep") 121ec3da8bcSJed Brown # elif defined(_OPENMP) && _OPENMP >= 201307 // OpenMP-4.0 (July, 2013) 122ec3da8bcSJed Brown # define CeedPragmaSIMD _Pragma("omp simd") 123ec3da8bcSJed Brown # else 124ec3da8bcSJed Brown # define CeedPragmaSIMD 125ec3da8bcSJed Brown # endif 126ec3da8bcSJed Brown #endif 127ec3da8bcSJed Brown 128ec3da8bcSJed Brown #include <stdarg.h> 129ec3da8bcSJed Brown #include <stdbool.h> 1303047f789SJeremy L Thompson #include <stddef.h> 1313047f789SJeremy L Thompson #include <stdint.h> 1323047f789SJeremy L Thompson #include <stdio.h> 133ec3da8bcSJed Brown 134ec3da8bcSJed Brown /// Integer type, used for indexing 135ec3da8bcSJed Brown /// @ingroup Ceed 136ec3da8bcSJed Brown typedef int32_t CeedInt; 137990fdeb6SJeremy L Thompson #define CeedInt_FMT "d" 13880a9ef05SNatalie Beams 139c30b7fbdSJeremy L Thompson /// Integer type, used array sizes 140c30b7fbdSJeremy L Thompson /// @ingroup Ceed 141c30b7fbdSJeremy L Thompson typedef ptrdiff_t CeedSize; 142c30b7fbdSJeremy L Thompson 14380a9ef05SNatalie Beams /// Scalar (floating point) types 14480a9ef05SNatalie Beams /// 145ec3da8bcSJed Brown /// @ingroup Ceed 14680a9ef05SNatalie Beams typedef enum { 14780a9ef05SNatalie Beams /// Single precision 14880a9ef05SNatalie Beams CEED_SCALAR_FP32, 14980a9ef05SNatalie Beams /// Double precision 15080a9ef05SNatalie Beams CEED_SCALAR_FP64 15180a9ef05SNatalie Beams } CeedScalarType; 15280a9ef05SNatalie Beams /// Base scalar type for the library to use: change which header is 15380a9ef05SNatalie Beams /// included to change the precision. 15480a9ef05SNatalie Beams #include "ceed-f64.h" 155ec3da8bcSJed Brown 156ec3da8bcSJed Brown /// Library context created by CeedInit() 157ec3da8bcSJed Brown /// @ingroup CeedUser 158ec3da8bcSJed Brown typedef struct Ceed_private *Ceed; 159ec3da8bcSJed Brown /// Non-blocking Ceed interfaces return a CeedRequest. 160ec3da8bcSJed Brown /// To perform an operation immediately, pass \ref CEED_REQUEST_IMMEDIATE instead. 161ec3da8bcSJed Brown /// @ingroup CeedUser 162ec3da8bcSJed Brown typedef struct CeedRequest_private *CeedRequest; 163ec3da8bcSJed Brown /// Handle for vectors over the field \ref CeedScalar 164ec3da8bcSJed Brown /// @ingroup CeedVectorUser 165ec3da8bcSJed Brown typedef struct CeedVector_private *CeedVector; 166ec3da8bcSJed Brown /// Handle for object describing restriction to elements 167ec3da8bcSJed Brown /// @ingroup CeedElemRestrictionUser 168ec3da8bcSJed Brown typedef struct CeedElemRestriction_private *CeedElemRestriction; 169ec3da8bcSJed Brown /// Handle for object describing discrete finite element evaluations 170ec3da8bcSJed Brown /// @ingroup CeedBasisUser 171ec3da8bcSJed Brown typedef struct CeedBasis_private *CeedBasis; 17243bbe138SJeremy L Thompson /// Handle for object describing CeedQFunction fields 17343bbe138SJeremy L Thompson /// @ingroup CeedQFunctionBackend 17443bbe138SJeremy L Thompson typedef struct CeedQFunctionField_private *CeedQFunctionField; 175ec3da8bcSJed Brown /// Handle for object describing functions evaluated independently at quadrature points 176ec3da8bcSJed Brown /// @ingroup CeedQFunctionUser 177ec3da8bcSJed Brown typedef struct CeedQFunction_private *CeedQFunction; 17843bbe138SJeremy L Thompson /// Handle for object describing CeedOperator fields 17943bbe138SJeremy L Thompson /// @ingroup CeedOperatorBackend 18043bbe138SJeremy L Thompson typedef struct CeedOperatorField_private *CeedOperatorField; 181ec3da8bcSJed Brown /// Handle for object describing context data for CeedQFunctions 182ec3da8bcSJed Brown /// @ingroup CeedQFunctionUser 183ec3da8bcSJed Brown typedef struct CeedQFunctionContext_private *CeedQFunctionContext; 1843668ca4bSJeremy L Thompson /// Handle for object describing registered fields for CeedQFunctionContext 1853668ca4bSJeremy L Thompson /// @ingroup CeedQFunctionUser 1863668ca4bSJeremy L Thompson typedef struct CeedContextFieldLabel_private *CeedContextFieldLabel; 187ec3da8bcSJed Brown /// Handle for object describing FE-type operators acting on vectors 188ec3da8bcSJed Brown /// 189ec3da8bcSJed Brown /// Given an element restriction \f$E\f$, basis evaluator \f$B\f$, and 190ec3da8bcSJed Brown /// quadrature function\f$f\f$, a CeedOperator expresses operations of the form 191ec3da8bcSJed Brown /// $$ E^T B^T f(B E u) $$ 192ec3da8bcSJed Brown /// acting on the vector \f$u\f$. 193ec3da8bcSJed Brown /// @ingroup CeedOperatorUser 194ec3da8bcSJed Brown typedef struct CeedOperator_private *CeedOperator; 195ec3da8bcSJed Brown 19622e44211Sjeremylt CEED_EXTERN int CeedRegistryGetList(size_t *n, char ***const resources, CeedInt **array); 197ec3da8bcSJed Brown CEED_EXTERN int CeedInit(const char *resource, Ceed *ceed); 1989560d06aSjeremylt CEED_EXTERN int CeedReferenceCopy(Ceed ceed, Ceed *ceed_copy); 199ec3da8bcSJed Brown CEED_EXTERN int CeedGetResource(Ceed ceed, const char **resource); 200d1d35e2fSjeremylt CEED_EXTERN int CeedIsDeterministic(Ceed ceed, bool *is_deterministic); 201ee5a26f2SJeremy L Thompson CEED_EXTERN int CeedAddJitSourceRoot(Ceed ceed, const char *jit_source_root); 202ec3da8bcSJed Brown CEED_EXTERN int CeedView(Ceed ceed, FILE *stream); 203ec3da8bcSJed Brown CEED_EXTERN int CeedDestroy(Ceed *ceed); 204ec3da8bcSJed Brown 205ec3da8bcSJed Brown CEED_EXTERN int CeedErrorImpl(Ceed, const char *, int, const char *, int, 206ec3da8bcSJed Brown const char *, ...); 207ec3da8bcSJed Brown /// Raise an error on ceed object 208ec3da8bcSJed Brown /// 209ec3da8bcSJed Brown /// @param ceed Ceed library context or NULL 210ec3da8bcSJed Brown /// @param ecode Error code (int) 211ec3da8bcSJed Brown /// @param ... printf-style format string followed by arguments as needed 212ec3da8bcSJed Brown /// 213ec3da8bcSJed Brown /// @ingroup Ceed 214ec3da8bcSJed Brown /// @sa CeedSetErrorHandler() 215ec3da8bcSJed Brown #if defined(__clang__) 216ec3da8bcSJed Brown /// Use nonstandard ternary to convince the compiler/clang-tidy that this 217ec3da8bcSJed Brown /// function never returns zero. 218ec3da8bcSJed Brown # define CeedError(ceed, ecode, ...) \ 21945f1e315Sjeremylt (CeedErrorImpl((ceed), __FILE__, __LINE__, __func__, (ecode), __VA_ARGS__), (ecode)) 220ec3da8bcSJed Brown #else 221ec3da8bcSJed Brown # define CeedError(ceed, ecode, ...) \ 222ec3da8bcSJed Brown CeedErrorImpl((ceed), __FILE__, __LINE__, __func__, (ecode), __VA_ARGS__) ?: (ecode) 223ec3da8bcSJed Brown #endif 224ec3da8bcSJed Brown 225ec3da8bcSJed Brown /// Ceed error handlers 226ec3da8bcSJed Brown CEED_EXTERN int CeedErrorReturn(Ceed, const char *, int, const char *, int, 227ec3da8bcSJed Brown const char *, va_list *); 228ec3da8bcSJed Brown CEED_EXTERN int CeedErrorStore(Ceed, const char *, int, const char *, int, 229ec3da8bcSJed Brown const char *, va_list *); 230ec3da8bcSJed Brown CEED_EXTERN int CeedErrorAbort(Ceed, const char *, int, const char *, int, 231ec3da8bcSJed Brown const char *, va_list *); 232ec3da8bcSJed Brown CEED_EXTERN int CeedErrorExit(Ceed, const char *, int, const char *, int, 233ec3da8bcSJed Brown const char *, va_list *); 234ec3da8bcSJed Brown typedef int (*CeedErrorHandler)(Ceed, const char *, int, 235ec3da8bcSJed Brown const char *, int, const char *, 236ec3da8bcSJed Brown va_list *); 237ec3da8bcSJed Brown CEED_EXTERN int CeedSetErrorHandler(Ceed ceed, CeedErrorHandler eh); 238d1d35e2fSjeremylt CEED_EXTERN int CeedGetErrorMessage(Ceed, const char **err_msg); 239d1d35e2fSjeremylt CEED_EXTERN int CeedResetErrorMessage(Ceed, const char **err_msg); 240ec3da8bcSJed Brown 241ec3da8bcSJed Brown /// libCEED library version numbering 242ec3da8bcSJed Brown /// @ingroup Ceed 243ec3da8bcSJed Brown #define CEED_VERSION_MAJOR 0 244667e613fSJeremy L Thompson #define CEED_VERSION_MINOR 10 245f374d6a3SJeremy L Thompson #define CEED_VERSION_PATCH 1 24638faa2d6SJeremy L Thompson #define CEED_VERSION_RELEASE false 247ec3da8bcSJed Brown 248ec3da8bcSJed Brown /// Compile-time check that the the current library version is at least as 249ec3da8bcSJed Brown /// recent as the specified version. This macro is typically used in 250ec3da8bcSJed Brown /// @code 251ec3da8bcSJed Brown /// #if CEED_VERSION_GE(0, 8, 0) 252ec3da8bcSJed Brown /// code path that needs at least 0.8.0 253ec3da8bcSJed Brown /// #else 254ec3da8bcSJed Brown /// fallback code for older versions 255ec3da8bcSJed Brown /// #endif 256ec3da8bcSJed Brown /// @endcode 257ec3da8bcSJed Brown /// 258ec3da8bcSJed Brown /// A non-release version always compares as positive infinity. 259ec3da8bcSJed Brown /// 260ec3da8bcSJed Brown /// @param major Major version 261ec3da8bcSJed Brown /// @param minor Minor version 262ec3da8bcSJed Brown /// @param patch Patch (subminor) version 263ec3da8bcSJed Brown /// 264ec3da8bcSJed Brown /// @ingroup Ceed 265ec3da8bcSJed Brown /// @sa CeedGetVersion() 266ec3da8bcSJed Brown #define CEED_VERSION_GE(major, minor, patch) \ 267ec3da8bcSJed Brown (!CEED_VERSION_RELEASE || \ 268ec3da8bcSJed Brown (CEED_VERSION_MAJOR > major || \ 269ec3da8bcSJed Brown (CEED_VERSION_MAJOR == major && \ 270ec3da8bcSJed Brown (CEED_VERSION_MINOR > minor || \ 271ec3da8bcSJed Brown (CEED_VERSION_MINOR == minor && CEED_VERSION_PATCH >= patch))))) 272ec3da8bcSJed Brown 273ec3da8bcSJed Brown CEED_EXTERN int CeedGetVersion(int *major, int *minor, int *patch, 274ec3da8bcSJed Brown bool *release); 275ec3da8bcSJed Brown 27680a9ef05SNatalie Beams CEED_EXTERN int CeedGetScalarType(CeedScalarType *scalar_type); 27780a9ef05SNatalie Beams 278ec3da8bcSJed Brown /// Ceed Errors 279ec3da8bcSJed Brown /// 280ec3da8bcSJed Brown /// This enum is used to specify the type of error returned by a function. 281ec3da8bcSJed Brown /// A zero error code is success, negative error codes indicate terminal errors 282ec3da8bcSJed Brown /// and positive error codes indicate nonterminal errors. With nonterminal errors 283ec3da8bcSJed Brown /// the object state has not been modifiend, but with terminal errors the object 284ec3da8bcSJed Brown /// data is likely modified or corrupted. 285ec3da8bcSJed Brown /// @ingroup Ceed 286ec3da8bcSJed Brown typedef enum { 2870f7fd0f8Sjeremylt /// Success error code 288ec3da8bcSJed Brown CEED_ERROR_SUCCESS = 0, 289ec3da8bcSJed Brown /// Minor error, generic 290ec3da8bcSJed Brown CEED_ERROR_MINOR = 1, 291ec3da8bcSJed Brown /// Minor error, dimension mismatch in inputs 292ec3da8bcSJed Brown CEED_ERROR_DIMENSION = 2, 293ec3da8bcSJed Brown /// Minor error, incomplete object setup 294ec3da8bcSJed Brown CEED_ERROR_INCOMPLETE = 3, 295ec3da8bcSJed Brown /// Minor error, incompatible arguments/configuration 296ec3da8bcSJed Brown CEED_ERROR_INCOMPATIBLE = 4, 297ec3da8bcSJed Brown /// Minor error, access lock problem 298ec3da8bcSJed Brown CEED_ERROR_ACCESS = 5, 299ec3da8bcSJed Brown /// Major error, generic 300ec3da8bcSJed Brown CEED_ERROR_MAJOR = -1, 301ec3da8bcSJed Brown /// Major error, internal backend error 302ec3da8bcSJed Brown CEED_ERROR_BACKEND = -2, 303ec3da8bcSJed Brown /// Major error, operation unsupported by current backend 304ec3da8bcSJed Brown CEED_ERROR_UNSUPPORTED = -3, 305ec3da8bcSJed Brown } CeedErrorType; 306ec3da8bcSJed Brown CEED_EXTERN const char *const *CeedErrorTypes; 307ec3da8bcSJed Brown 308ec3da8bcSJed Brown /// Specify memory type 309ec3da8bcSJed Brown /// 310ec3da8bcSJed Brown /// Many Ceed interfaces take or return pointers to memory. This enum is used to 311ec3da8bcSJed Brown /// specify where the memory being provided or requested must reside. 312ec3da8bcSJed Brown /// @ingroup Ceed 313ec3da8bcSJed Brown typedef enum { 314ec3da8bcSJed Brown /// Memory resides on the host 315ec3da8bcSJed Brown CEED_MEM_HOST, 316ec3da8bcSJed Brown /// Memory resides on a device (corresponding to \ref Ceed resource) 317ec3da8bcSJed Brown CEED_MEM_DEVICE, 318ec3da8bcSJed Brown } CeedMemType; 319ec3da8bcSJed Brown CEED_EXTERN const char *const CeedMemTypes[]; 320ec3da8bcSJed Brown 321ec3da8bcSJed Brown CEED_EXTERN int CeedGetPreferredMemType(Ceed ceed, CeedMemType *type); 322ec3da8bcSJed Brown 323ec3da8bcSJed Brown /// Conveys ownership status of arrays passed to Ceed interfaces. 324ec3da8bcSJed Brown /// @ingroup Ceed 325ec3da8bcSJed Brown typedef enum { 326ec3da8bcSJed Brown /// Implementation will copy the values and not store the passed pointer. 327ec3da8bcSJed Brown CEED_COPY_VALUES, 328ec3da8bcSJed Brown /// Implementation can use and modify the data provided by the user, but does 329ec3da8bcSJed Brown /// not take ownership. 330ec3da8bcSJed Brown CEED_USE_POINTER, 331ec3da8bcSJed Brown /// Implementation takes ownership of the pointer and will free using 332ec3da8bcSJed Brown /// CeedFree() when done using it. The user should not assume that the 333ec3da8bcSJed Brown /// pointer remains valid after ownership has been transferred. Note that 334ec3da8bcSJed Brown /// arrays allocated using C++ operator new or other allocators cannot 335ec3da8bcSJed Brown /// generally be freed using CeedFree(). CeedFree() is capable of freeing any 336ec3da8bcSJed Brown /// memory that can be freed using free(3). 337ec3da8bcSJed Brown CEED_OWN_POINTER, 338ec3da8bcSJed Brown } CeedCopyMode; 339cdf32b93SJeremy L Thompson CEED_EXTERN const char *const CeedCopyModes[]; 340ec3da8bcSJed Brown 341ec3da8bcSJed Brown /// Denotes type of vector norm to be computed 342ec3da8bcSJed Brown /// @ingroup CeedVector 343ec3da8bcSJed Brown typedef enum { 344ec3da8bcSJed Brown /// L_1 norm: sum_i |x_i| 345ec3da8bcSJed Brown CEED_NORM_1, 346ec3da8bcSJed Brown /// L_2 norm: sqrt(sum_i |x_i|^2) 347ec3da8bcSJed Brown CEED_NORM_2, 348ec3da8bcSJed Brown /// L_Infinity norm: max_i |x_i| 349ec3da8bcSJed Brown CEED_NORM_MAX, 350ec3da8bcSJed Brown } CeedNormType; 351ec3da8bcSJed Brown 3521f9221feSJeremy L Thompson CEED_EXTERN int CeedVectorCreate(Ceed ceed, CeedSize len, CeedVector *vec); 3539560d06aSjeremylt CEED_EXTERN int CeedVectorReferenceCopy(CeedVector vec, CeedVector *vec_copy); 354d1d35e2fSjeremylt CEED_EXTERN int CeedVectorSetArray(CeedVector vec, CeedMemType mem_type, 355d1d35e2fSjeremylt CeedCopyMode copy_mode, CeedScalar *array); 356ec3da8bcSJed Brown CEED_EXTERN int CeedVectorSetValue(CeedVector vec, CeedScalar value); 357d1d35e2fSjeremylt CEED_EXTERN int CeedVectorSyncArray(CeedVector vec, CeedMemType mem_type); 358d1d35e2fSjeremylt CEED_EXTERN int CeedVectorTakeArray(CeedVector vec, CeedMemType mem_type, 359ec3da8bcSJed Brown CeedScalar **array); 360d1d35e2fSjeremylt CEED_EXTERN int CeedVectorGetArray(CeedVector vec, CeedMemType mem_type, 361ec3da8bcSJed Brown CeedScalar **array); 362d1d35e2fSjeremylt CEED_EXTERN int CeedVectorGetArrayRead(CeedVector vec, CeedMemType mem_type, 363ec3da8bcSJed Brown const CeedScalar **array); 3649c774eddSJeremy L Thompson CEED_EXTERN int CeedVectorGetArrayWrite(CeedVector vec, CeedMemType mem_type, 3659c774eddSJeremy L Thompson CeedScalar **array); 366ec3da8bcSJed Brown CEED_EXTERN int CeedVectorRestoreArray(CeedVector vec, CeedScalar **array); 367ec3da8bcSJed Brown CEED_EXTERN int CeedVectorRestoreArrayRead(CeedVector vec, 368ec3da8bcSJed Brown const CeedScalar **array); 369ec3da8bcSJed Brown CEED_EXTERN int CeedVectorNorm(CeedVector vec, CeedNormType type, 370ec3da8bcSJed Brown CeedScalar *norm); 371e0dd3b27Sjeremylt CEED_EXTERN int CeedVectorScale(CeedVector x, CeedScalar alpha); 3720f7fd0f8Sjeremylt CEED_EXTERN int CeedVectorAXPY(CeedVector y, CeedScalar alpha, CeedVector x); 3730f7fd0f8Sjeremylt CEED_EXTERN int CeedVectorPointwiseMult(CeedVector w, CeedVector x, CeedVector y); 374ec3da8bcSJed Brown CEED_EXTERN int CeedVectorReciprocal(CeedVector vec); 375d1d35e2fSjeremylt CEED_EXTERN int CeedVectorView(CeedVector vec, const char *fp_fmt, FILE *stream); 376b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedVectorGetCeed(CeedVector vec, Ceed *ceed); 3771f9221feSJeremy L Thompson CEED_EXTERN int CeedVectorGetLength(CeedVector vec, CeedSize *length); 378ec3da8bcSJed Brown CEED_EXTERN int CeedVectorDestroy(CeedVector *vec); 379ec3da8bcSJed Brown 380ec3da8bcSJed Brown CEED_EXTERN CeedRequest *const CEED_REQUEST_IMMEDIATE; 381ec3da8bcSJed Brown CEED_EXTERN CeedRequest *const CEED_REQUEST_ORDERED; 382ec3da8bcSJed Brown CEED_EXTERN int CeedRequestWait(CeedRequest *req); 383ec3da8bcSJed Brown 384ec3da8bcSJed Brown /// Argument for CeedOperatorSetField that vector is collocated with 385*151157abSJeremy L Thompson /// quadrature points, only used with CeedEvalMode CEED_EVAL_NONE 386ec3da8bcSJed Brown /// @ingroup CeedBasis 387ec3da8bcSJed Brown CEED_EXTERN const CeedBasis CEED_BASIS_COLLOCATED; 388ec3da8bcSJed Brown 389ec3da8bcSJed Brown /// Argument for CeedOperatorSetField to use active input or output 390ec3da8bcSJed Brown /// @ingroup CeedVector 391ec3da8bcSJed Brown CEED_EXTERN const CeedVector CEED_VECTOR_ACTIVE; 392ec3da8bcSJed Brown 393ec3da8bcSJed Brown /// Argument for CeedOperatorSetField to use no vector, used with 394ec3da8bcSJed Brown /// qfunction input with eval mode CEED_EVAL_WEIGHT 395ec3da8bcSJed Brown /// @ingroup CeedVector 396ec3da8bcSJed Brown CEED_EXTERN const CeedVector CEED_VECTOR_NONE; 397ec3da8bcSJed Brown 398ec3da8bcSJed Brown /// Argument for CeedOperatorSetField to use no ElemRestriction, only used with 399ec3da8bcSJed Brown /// eval mode CEED_EVAL_WEIGHT. 400ec3da8bcSJed Brown /// @ingroup CeedElemRestriction 401ec3da8bcSJed Brown CEED_EXTERN const CeedElemRestriction CEED_ELEMRESTRICTION_NONE; 402ec3da8bcSJed Brown 403ec3da8bcSJed Brown /// Argument for CeedOperatorCreate that QFunction is not created by user. 404ec3da8bcSJed Brown /// Only used for QFunctions dqf and dqfT. If implemented, a backend may 405ec3da8bcSJed Brown /// attempt to provide the action of these QFunctions. 406ec3da8bcSJed Brown /// @ingroup CeedQFunction 407ec3da8bcSJed Brown CEED_EXTERN const CeedQFunction CEED_QFUNCTION_NONE; 408ec3da8bcSJed Brown 409ec3da8bcSJed Brown /// Denotes whether a linear transformation or its transpose should be applied 410ec3da8bcSJed Brown /// @ingroup CeedBasis 411ec3da8bcSJed Brown typedef enum { 412ec3da8bcSJed Brown /// Apply the linear transformation 413ec3da8bcSJed Brown CEED_NOTRANSPOSE, 414ec3da8bcSJed Brown /// Apply the transpose 415ec3da8bcSJed Brown CEED_TRANSPOSE 416ec3da8bcSJed Brown } CeedTransposeMode; 417ec3da8bcSJed Brown CEED_EXTERN const char *const CeedTransposeModes[]; 418ec3da8bcSJed Brown 419ec3da8bcSJed Brown /// Argument for CeedElemRestrictionCreateStrided that L-vector is in 420ec3da8bcSJed Brown /// the Ceed backend's preferred layout. This argument should only be used 421ec3da8bcSJed Brown /// with vectors created by a Ceed backend. 422ec3da8bcSJed Brown /// @ingroup CeedElemRestriction 423ec3da8bcSJed Brown CEED_EXTERN const CeedInt CEED_STRIDES_BACKEND[3]; 424ec3da8bcSJed Brown 425d1d35e2fSjeremylt CEED_EXTERN int CeedElemRestrictionCreate(Ceed ceed, CeedInt num_elem, 426e79b91d9SJeremy L Thompson CeedInt elem_size, CeedInt num_comp, CeedInt comp_stride, CeedSize l_size, 427d1d35e2fSjeremylt CeedMemType mem_type, CeedCopyMode copy_mode, const CeedInt *offsets, 428ec3da8bcSJed Brown CeedElemRestriction *rstr); 429fc0567d9Srezgarshakeri CEED_EXTERN int CeedElemRestrictionCreateOriented(Ceed ceed, CeedInt num_elem, 430e79b91d9SJeremy L Thompson CeedInt elem_size, CeedInt num_comp, CeedInt comp_stride, CeedSize l_size, 431fc0567d9Srezgarshakeri CeedMemType mem_type, CeedCopyMode copy_mode, const CeedInt *offsets, 432fc0567d9Srezgarshakeri const bool *orient, CeedElemRestriction *rstr); 433ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionCreateStrided(Ceed ceed, 434e79b91d9SJeremy L Thompson CeedInt num_elem, CeedInt elem_size, CeedInt num_comp, CeedSize l_size, 435ec3da8bcSJed Brown const CeedInt strides[3], CeedElemRestriction *rstr); 436d1d35e2fSjeremylt CEED_EXTERN int CeedElemRestrictionCreateBlocked(Ceed ceed, CeedInt num_elem, 437d1d35e2fSjeremylt CeedInt elem_size, CeedInt blk_size, CeedInt num_comp, CeedInt comp_stride, 438e79b91d9SJeremy L Thompson CeedSize l_size, CeedMemType mem_type, CeedCopyMode copy_mode, 439ec3da8bcSJed Brown const CeedInt *offsets, CeedElemRestriction *rstr); 440ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionCreateBlockedStrided(Ceed ceed, 441d1d35e2fSjeremylt CeedInt num_elem, CeedInt elem_size, CeedInt blk_size, CeedInt num_comp, 442e79b91d9SJeremy L Thompson CeedSize l_size, const CeedInt strides[3], CeedElemRestriction *rstr); 4439560d06aSjeremylt CEED_EXTERN int CeedElemRestrictionReferenceCopy(CeedElemRestriction rstr, 4449560d06aSjeremylt CeedElemRestriction *rstr_copy); 445ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionCreateVector(CeedElemRestriction rstr, 446ec3da8bcSJed Brown CeedVector *lvec, CeedVector *evec); 447ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionApply(CeedElemRestriction rstr, 448d1d35e2fSjeremylt CeedTransposeMode t_mode, CeedVector u, CeedVector ru, CeedRequest *request); 449ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionApplyBlock(CeedElemRestriction rstr, 450d1d35e2fSjeremylt CeedInt block, CeedTransposeMode t_mode, CeedVector u, CeedVector ru, 451ec3da8bcSJed Brown CeedRequest *request); 452b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedElemRestrictionGetCeed(CeedElemRestriction rstr, 453b7c9bbdaSJeremy L Thompson Ceed *ceed); 454ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetCompStride(CeedElemRestriction rstr, 455d1d35e2fSjeremylt CeedInt *comp_stride); 456ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetNumElements(CeedElemRestriction rstr, 457d1d35e2fSjeremylt CeedInt *num_elem); 458ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetElementSize(CeedElemRestriction rstr, 459d1d35e2fSjeremylt CeedInt *elem_size); 460ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetLVectorSize(CeedElemRestriction rstr, 461e79b91d9SJeremy L Thompson CeedSize *l_size); 462ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetNumComponents(CeedElemRestriction rstr, 463d1d35e2fSjeremylt CeedInt *num_comp); 464ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetNumBlocks(CeedElemRestriction rstr, 465d1d35e2fSjeremylt CeedInt *num_blk); 466ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetBlockSize(CeedElemRestriction rstr, 467d1d35e2fSjeremylt CeedInt *blk_size); 468ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetMultiplicity(CeedElemRestriction rstr, 469ec3da8bcSJed Brown CeedVector mult); 470ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionView(CeedElemRestriction rstr, FILE *stream); 471ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionDestroy(CeedElemRestriction *rstr); 472ec3da8bcSJed Brown 473ec3da8bcSJed Brown // The formalism here is that we have the structure 474ec3da8bcSJed Brown // \int_\Omega v^T f_0(u, \nabla u, qdata) + (\nabla v)^T f_1(u, \nabla u, qdata) 475ec3da8bcSJed Brown // where gradients are with respect to the reference element. 476ec3da8bcSJed Brown 477ec3da8bcSJed Brown /// Basis evaluation mode 478ec3da8bcSJed Brown /// 479ec3da8bcSJed Brown /// Modes can be bitwise ORed when passing to most functions. 480ec3da8bcSJed Brown /// @ingroup CeedBasis 481ec3da8bcSJed Brown typedef enum { 482ec3da8bcSJed Brown /// Perform no evaluation (either because there is no data or it is already at 483ec3da8bcSJed Brown /// quadrature points) 484ec3da8bcSJed Brown CEED_EVAL_NONE = 0, 485ec3da8bcSJed Brown /// Interpolate from nodes to quadrature points 486ec3da8bcSJed Brown CEED_EVAL_INTERP = 1, 487ec3da8bcSJed Brown /// Evaluate gradients at quadrature points from input in a nodal basis 488ec3da8bcSJed Brown CEED_EVAL_GRAD = 2, 489ec3da8bcSJed Brown /// Evaluate divergence at quadrature points from input in a nodal basis 490ec3da8bcSJed Brown CEED_EVAL_DIV = 4, 491ec3da8bcSJed Brown /// Evaluate curl at quadrature points from input in a nodal basis 492ec3da8bcSJed Brown CEED_EVAL_CURL = 8, 493ec3da8bcSJed Brown /// Using no input, evaluate quadrature weights on the reference element 494ec3da8bcSJed Brown CEED_EVAL_WEIGHT = 16, 495ec3da8bcSJed Brown } CeedEvalMode; 496ec3da8bcSJed Brown CEED_EXTERN const char *const CeedEvalModes[]; 497ec3da8bcSJed Brown 498ec3da8bcSJed Brown /// Type of quadrature; also used for location of nodes 499ec3da8bcSJed Brown /// @ingroup CeedBasis 500ec3da8bcSJed Brown typedef enum { 501ec3da8bcSJed Brown /// Gauss-Legendre quadrature 502ec3da8bcSJed Brown CEED_GAUSS = 0, 503ec3da8bcSJed Brown /// Gauss-Legendre-Lobatto quadrature 504ec3da8bcSJed Brown CEED_GAUSS_LOBATTO = 1, 505ec3da8bcSJed Brown } CeedQuadMode; 506ec3da8bcSJed Brown CEED_EXTERN const char *const CeedQuadModes[]; 507ec3da8bcSJed Brown 508ec3da8bcSJed Brown /// Type of basis shape to create non-tensor H1 element basis 509ec3da8bcSJed Brown /// 510ec3da8bcSJed Brown /// Dimension can be extracted with bitwise AND 511ec3da8bcSJed Brown /// (CeedElemTopology & 2**(dim + 2)) == TRUE 512ec3da8bcSJed Brown /// @ingroup CeedBasis 513ec3da8bcSJed Brown typedef enum { 514ec3da8bcSJed Brown /// Line 51550c301a5SRezgar Shakeri CEED_TOPOLOGY_LINE = 1 << 16 | 0, 516ec3da8bcSJed Brown /// Triangle - 2D shape 51750c301a5SRezgar Shakeri CEED_TOPOLOGY_TRIANGLE = 2 << 16 | 1, 518ec3da8bcSJed Brown /// Quadralateral - 2D shape 51950c301a5SRezgar Shakeri CEED_TOPOLOGY_QUAD = 2 << 16 | 2, 520ec3da8bcSJed Brown /// Tetrahedron - 3D shape 52150c301a5SRezgar Shakeri CEED_TOPOLOGY_TET = 3 << 16 | 3, 522ec3da8bcSJed Brown /// Pyramid - 3D shape 52350c301a5SRezgar Shakeri CEED_TOPOLOGY_PYRAMID = 3 << 16 | 4, 524ec3da8bcSJed Brown /// Prism - 3D shape 52550c301a5SRezgar Shakeri CEED_TOPOLOGY_PRISM = 3 << 16 | 5, 526ec3da8bcSJed Brown /// Hexehedron - 3D shape 52750c301a5SRezgar Shakeri CEED_TOPOLOGY_HEX = 3 << 16 | 6, 528ec3da8bcSJed Brown } CeedElemTopology; 529ec3da8bcSJed Brown CEED_EXTERN const char *const CeedElemTopologies[]; 530ec3da8bcSJed Brown 531ec3da8bcSJed Brown CEED_EXTERN int CeedBasisCreateTensorH1Lagrange(Ceed ceed, CeedInt dim, 532d1d35e2fSjeremylt CeedInt num_comp, CeedInt P, CeedInt Q, CeedQuadMode quad_mode, CeedBasis *basis); 533d1d35e2fSjeremylt CEED_EXTERN int CeedBasisCreateTensorH1(Ceed ceed, CeedInt dim, CeedInt num_comp, 534d1d35e2fSjeremylt CeedInt P_1d, CeedInt Q_1d, 535d1d35e2fSjeremylt const CeedScalar *interp_1d, 536d1d35e2fSjeremylt const CeedScalar *grad_1d, 537d1d35e2fSjeremylt const CeedScalar *q_ref_1d, 538d1d35e2fSjeremylt const CeedScalar *q_weight_1d, 539ec3da8bcSJed Brown CeedBasis *basis); 540ec3da8bcSJed Brown CEED_EXTERN int CeedBasisCreateH1(Ceed ceed, CeedElemTopology topo, 541d1d35e2fSjeremylt CeedInt num_comp, 542d1d35e2fSjeremylt CeedInt num_nodes, CeedInt nqpts, 543ec3da8bcSJed Brown const CeedScalar *interp, 544ec3da8bcSJed Brown const CeedScalar *grad, 545d1d35e2fSjeremylt const CeedScalar *q_ref, 546d1d35e2fSjeremylt const CeedScalar *q_weights, CeedBasis *basis); 54750c301a5SRezgar Shakeri CEED_EXTERN int CeedBasisCreateHdiv(Ceed ceed, CeedElemTopology topo, 54850c301a5SRezgar Shakeri CeedInt num_comp, 54950c301a5SRezgar Shakeri CeedInt num_nodes, CeedInt nqpts, 55050c301a5SRezgar Shakeri const CeedScalar *interp, 55150c301a5SRezgar Shakeri const CeedScalar *div, 55250c301a5SRezgar Shakeri const CeedScalar *q_ref, 55350c301a5SRezgar Shakeri const CeedScalar *q_weights, CeedBasis *basis); 554446e7af4SJeremy L Thompson CEED_EXTERN int CeedBasisCreateProjection(CeedBasis basis_from, CeedBasis basis_to, CeedBasis *basis_project); 555446e7af4SJeremy L Thompson CEED_EXTERN int CeedBasisCreateProjectionMatrix(CeedBasis basis_from, CeedBasis basis_to, CeedScalar **interp_project); 5569560d06aSjeremylt CEED_EXTERN int CeedBasisReferenceCopy(CeedBasis basis, CeedBasis *basis_copy); 557ec3da8bcSJed Brown CEED_EXTERN int CeedBasisView(CeedBasis basis, FILE *stream); 558d1d35e2fSjeremylt CEED_EXTERN int CeedBasisApply(CeedBasis basis, CeedInt num_elem, 559d1d35e2fSjeremylt CeedTransposeMode t_mode, 560d1d35e2fSjeremylt CeedEvalMode eval_mode, CeedVector u, CeedVector v); 561b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedBasisGetCeed(CeedBasis basis, Ceed *ceed); 562ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetDimension(CeedBasis basis, CeedInt *dim); 563ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetTopology(CeedBasis basis, CeedElemTopology *topo); 56450c301a5SRezgar Shakeri CEED_EXTERN int CeedBasisGetNumQuadratureComponents(CeedBasis basis, CeedInt *Q_comp); 565d1d35e2fSjeremylt CEED_EXTERN int CeedBasisGetNumComponents(CeedBasis basis, CeedInt *num_comp); 566ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetNumNodes(CeedBasis basis, CeedInt *P); 567d1d35e2fSjeremylt CEED_EXTERN int CeedBasisGetNumNodes1D(CeedBasis basis, CeedInt *P_1d); 568ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetNumQuadraturePoints(CeedBasis basis, CeedInt *Q); 569ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetNumQuadraturePoints1D(CeedBasis basis, 570d1d35e2fSjeremylt CeedInt *Q_1d); 571d1d35e2fSjeremylt CEED_EXTERN int CeedBasisGetQRef(CeedBasis basis, const CeedScalar **q_ref); 572ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetQWeights(CeedBasis basis, 573d1d35e2fSjeremylt const CeedScalar **q_weights); 574ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetInterp(CeedBasis basis, const CeedScalar **interp); 575ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetInterp1D(CeedBasis basis, 576d1d35e2fSjeremylt const CeedScalar **interp_1d); 577ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetGrad(CeedBasis basis, const CeedScalar **grad); 578d1d35e2fSjeremylt CEED_EXTERN int CeedBasisGetGrad1D(CeedBasis basis, const CeedScalar **grad_1d); 57950c301a5SRezgar Shakeri CEED_EXTERN int CeedBasisGetDiv(CeedBasis basis, const CeedScalar **div); 580ec3da8bcSJed Brown CEED_EXTERN int CeedBasisDestroy(CeedBasis *basis); 581ec3da8bcSJed Brown 582d1d35e2fSjeremylt CEED_EXTERN int CeedGaussQuadrature(CeedInt Q, CeedScalar *q_ref_1d, 583d1d35e2fSjeremylt CeedScalar *q_weight_1d); 584d1d35e2fSjeremylt CEED_EXTERN int CeedLobattoQuadrature(CeedInt Q, CeedScalar *q_ref_1d, 585d1d35e2fSjeremylt CeedScalar *q_weight_1d); 586ec3da8bcSJed Brown CEED_EXTERN int CeedQRFactorization(Ceed ceed, CeedScalar *mat, CeedScalar *tau, 587ec3da8bcSJed Brown CeedInt m, CeedInt n); 588ec3da8bcSJed Brown CEED_EXTERN int CeedSymmetricSchurDecomposition(Ceed ceed, CeedScalar *mat, 589ec3da8bcSJed Brown CeedScalar *lambda, CeedInt n); 590d1d35e2fSjeremylt CEED_EXTERN int CeedSimultaneousDiagonalization(Ceed ceed, CeedScalar *mat_A, 591d1d35e2fSjeremylt CeedScalar *mat_B, CeedScalar *x, CeedScalar *lambda, CeedInt n); 592ec3da8bcSJed Brown 593cdf32b93SJeremy L Thompson /** Handle for the user provided CeedQFunction callback function 594ec3da8bcSJed Brown 595cdf32b93SJeremy L Thompson @param[in,out] ctx User-defined context set using CeedQFunctionSetContext() or NULL 596cdf32b93SJeremy L Thompson @param[in] Q Number of quadrature points at which to evaluate 597cdf32b93SJeremy L Thompson @param[in] in Array of pointers to each input argument in the order provided 598ec3da8bcSJed Brown by the user in CeedQFunctionAddInput(). Each array has shape 599d1d35e2fSjeremylt `[dim, num_comp, Q]` where `dim` is the geometric dimension for 600ec3da8bcSJed Brown \ref CEED_EVAL_GRAD (`dim=1` for \ref CEED_EVAL_INTERP) and 601d1d35e2fSjeremylt `num_comp` is the number of field components (`num_comp=1` for 602ec3da8bcSJed Brown scalar fields). This results in indexing the `i`th input at 603d1d35e2fSjeremylt quadrature point `j` as `in[i][(d*num_comp + c)*Q + j]`. 604cdf32b93SJeremy L Thompson @param[out] out Array of pointers to each output array in the order provided 605ec3da8bcSJed Brown using CeedQFunctionAddOutput(). The shapes are as above for 606ec3da8bcSJed Brown \a in. 607ec3da8bcSJed Brown 608ec3da8bcSJed Brown @return An error code: 0 - success, otherwise - failure 609ec3da8bcSJed Brown 610ec3da8bcSJed Brown @ingroup CeedQFunction 611ec3da8bcSJed Brown **/ 612ec3da8bcSJed Brown typedef int (*CeedQFunctionUser)(void *ctx, const CeedInt Q, 613ec3da8bcSJed Brown const CeedScalar *const *in, 614ec3da8bcSJed Brown CeedScalar *const *out); 615ec3da8bcSJed Brown 616d1d35e2fSjeremylt CEED_EXTERN int CeedQFunctionCreateInterior(Ceed ceed, CeedInt vec_length, 617ec3da8bcSJed Brown CeedQFunctionUser f, const char *source, CeedQFunction *qf); 618ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionCreateInteriorByName(Ceed ceed, const char *name, 619ec3da8bcSJed Brown CeedQFunction *qf); 620ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionCreateIdentity(Ceed ceed, CeedInt size, 621d1d35e2fSjeremylt CeedEvalMode in_mode, CeedEvalMode out_mode, CeedQFunction *qf); 6229560d06aSjeremylt CEED_EXTERN int CeedQFunctionReferenceCopy(CeedQFunction qf, CeedQFunction *qf_copy); 623d1d35e2fSjeremylt CEED_EXTERN int CeedQFunctionAddInput(CeedQFunction qf, const char *field_name, 624d1d35e2fSjeremylt CeedInt size, CeedEvalMode eval_mode); 625d1d35e2fSjeremylt CEED_EXTERN int CeedQFunctionAddOutput(CeedQFunction qf, const char *field_name, 626d1d35e2fSjeremylt CeedInt size, CeedEvalMode eval_mode); 62743bbe138SJeremy L Thompson CEED_EXTERN int CeedQFunctionGetFields(CeedQFunction qf, 62843bbe138SJeremy L Thompson CeedInt *num_input_fields, 62943bbe138SJeremy L Thompson CeedQFunctionField **input_fields, 63043bbe138SJeremy L Thompson CeedInt *num_output_fields, 63143bbe138SJeremy L Thompson CeedQFunctionField **output_fields); 632ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionSetContext(CeedQFunction qf, 633ec3da8bcSJed Brown CeedQFunctionContext ctx); 634441428dfSJeremy L Thompson CEED_EXTERN int CeedQFunctionSetContextWritable(CeedQFunction qf, bool is_writable); 6359d36ca50SJeremy L Thompson CEED_EXTERN int CeedQFunctionSetUserFlopsEstimate(CeedQFunction qf, CeedSize flops); 636ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionView(CeedQFunction qf, FILE *stream); 637b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedQFunctionGetCeed(CeedQFunction qf, Ceed *ceed); 638ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionApply(CeedQFunction qf, CeedInt Q, 639ec3da8bcSJed Brown CeedVector *u, CeedVector *v); 640ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionDestroy(CeedQFunction *qf); 641ec3da8bcSJed Brown 64243bbe138SJeremy L Thompson CEED_EXTERN int CeedQFunctionFieldGetName(CeedQFunctionField qf_field, 64343bbe138SJeremy L Thompson char **field_name); 64443bbe138SJeremy L Thompson CEED_EXTERN int CeedQFunctionFieldGetSize(CeedQFunctionField qf_field, 64543bbe138SJeremy L Thompson CeedInt *size); 64643bbe138SJeremy L Thompson CEED_EXTERN int CeedQFunctionFieldGetEvalMode(CeedQFunctionField qf_field, 64743bbe138SJeremy L Thompson CeedEvalMode *eval_mode); 64843bbe138SJeremy L Thompson 649cdf32b93SJeremy L Thompson /// Denotes type of data stored in a CeedQFunctionContext field 650cdf32b93SJeremy L Thompson /// @ingroup CeedQFunction 651cdf32b93SJeremy L Thompson typedef enum { 652cdf32b93SJeremy L Thompson /// Double precision value 6537bfe0f0eSJeremy L Thompson CEED_CONTEXT_FIELD_DOUBLE = 1, 654cdf32b93SJeremy L Thompson /// 32 bit integer value 6557bfe0f0eSJeremy L Thompson CEED_CONTEXT_FIELD_INT32 = 2, 656cdf32b93SJeremy L Thompson } CeedContextFieldType; 657cdf32b93SJeremy L Thompson CEED_EXTERN const char *const CeedContextFieldTypes[]; 658cdf32b93SJeremy L Thompson 6592790b72bSJeremy L Thompson /** Handle for the user provided CeedQFunctionContextDataDestroy callback function 6602790b72bSJeremy L Thompson 6612790b72bSJeremy L Thompson @param[in,out] data User-CeedQFunctionContext data 6622790b72bSJeremy L Thompson 6632790b72bSJeremy L Thompson @return An error code: 0 - success, otherwise - failure 6642790b72bSJeremy L Thompson 6652790b72bSJeremy L Thompson @ingroup CeedQFunction 6662790b72bSJeremy L Thompson **/ 6672790b72bSJeremy L Thompson typedef int (*CeedQFunctionContextDataDestroyUser)(void *data); 6682790b72bSJeremy L Thompson 669ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextCreate(Ceed ceed, 670ec3da8bcSJed Brown CeedQFunctionContext *ctx); 6719560d06aSjeremylt CEED_EXTERN int CeedQFunctionContextReferenceCopy(CeedQFunctionContext ctx, 6729560d06aSjeremylt CeedQFunctionContext *ctx_copy); 673ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextSetData(CeedQFunctionContext ctx, 674d1d35e2fSjeremylt CeedMemType mem_type, CeedCopyMode copy_mode, size_t size, void *data); 675891038deSjeremylt CEED_EXTERN int CeedQFunctionContextTakeData(CeedQFunctionContext ctx, 676891038deSjeremylt CeedMemType mem_type, void *data); 677ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextGetData(CeedQFunctionContext ctx, 678891038deSjeremylt CeedMemType mem_type, void *data); 67928bfd0b7SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextGetDataRead(CeedQFunctionContext ctx, 68028bfd0b7SJeremy L Thompson CeedMemType mem_type, void *data); 681ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextRestoreData(CeedQFunctionContext ctx, 682ec3da8bcSJed Brown void *data); 68328bfd0b7SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextRestoreDataRead(CeedQFunctionContext ctx, 68428bfd0b7SJeremy L Thompson void *data); 685cdf32b93SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextRegisterDouble(CeedQFunctionContext ctx, 6867bfe0f0eSJeremy L Thompson const char *field_name, size_t field_offset, size_t num_values, 6877bfe0f0eSJeremy L Thompson const char *field_description); 688cdf32b93SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextRegisterInt32(CeedQFunctionContext ctx, 6897bfe0f0eSJeremy L Thompson const char *field_name, size_t field_offset, size_t num_values, 6907bfe0f0eSJeremy L Thompson const char *field_description); 6913668ca4bSJeremy L Thompson CEED_EXTERN int CeedQFunctionContextGetAllFieldLabels(CeedQFunctionContext ctx, 6923668ca4bSJeremy L Thompson const CeedContextFieldLabel **field_labels, CeedInt *num_fields); 6930f86cbe7SJeremy L Thompson CEED_EXTERN int CeedContextFieldLabelGetDescription(CeedContextFieldLabel label, 6947bfe0f0eSJeremy L Thompson const char **field_name, const char **field_description, size_t *num_values, 6950f86cbe7SJeremy L Thompson CeedContextFieldType *field_type); 69680a9ef05SNatalie Beams CEED_EXTERN int CeedQFunctionContextGetContextSize(CeedQFunctionContext ctx, 69780a9ef05SNatalie Beams size_t *ctx_size); 698ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextView(CeedQFunctionContext ctx, 699ec3da8bcSJed Brown FILE *stream); 7002e64a2b9SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextSetDataDestroy(CeedQFunctionContext ctx, 7012e64a2b9SJeremy L Thompson CeedMemType f_mem_type, CeedQFunctionContextDataDestroyUser f); 702ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextDestroy(CeedQFunctionContext *ctx); 703ec3da8bcSJed Brown 704ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorCreate(Ceed ceed, CeedQFunction qf, 705ec3da8bcSJed Brown CeedQFunction dqf, CeedQFunction dqfT, 706ec3da8bcSJed Brown CeedOperator *op); 707ec3da8bcSJed Brown CEED_EXTERN int CeedCompositeOperatorCreate(Ceed ceed, CeedOperator *op); 7089560d06aSjeremylt CEED_EXTERN int CeedOperatorReferenceCopy(CeedOperator op, CeedOperator *op_copy); 709d1d35e2fSjeremylt CEED_EXTERN int CeedOperatorSetField(CeedOperator op, const char *field_name, 710ec3da8bcSJed Brown CeedElemRestriction r, CeedBasis b, 711ec3da8bcSJed Brown CeedVector v); 71243bbe138SJeremy L Thompson CEED_EXTERN int CeedOperatorGetFields(CeedOperator op, 71343bbe138SJeremy L Thompson CeedInt *num_input_fields, 71443bbe138SJeremy L Thompson CeedOperatorField **input_fields, 71543bbe138SJeremy L Thompson CeedInt *num_output_fields, 71643bbe138SJeremy L Thompson CeedOperatorField **output_fields); 717d1d35e2fSjeremylt CEED_EXTERN int CeedCompositeOperatorAddSub(CeedOperator composite_op, 718d1d35e2fSjeremylt CeedOperator sub_op); 7194db537f9SJeremy L Thompson CEED_EXTERN int CeedOperatorCheckReady(CeedOperator op); 720c9366a6bSJeremy L Thompson CEED_EXTERN int CeedOperatorGetActiveVectorLengths(CeedOperator op, CeedSize *input_size, CeedSize *output_size); 721beecbf24SJeremy L Thompson CEED_EXTERN int CeedOperatorSetQFunctionAssemblyReuse(CeedOperator op, bool reuse_assembly_data); 722beecbf24SJeremy L Thompson CEED_EXTERN int CeedOperatorSetQFunctionAssemblyDataUpdateNeeded(CeedOperator op, bool needs_data_update); 723ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleQFunction(CeedOperator op, 724ec3da8bcSJed Brown CeedVector *assembled, CeedElemRestriction *rstr, CeedRequest *request); 72570a7ffb3SJeremy L Thompson CEED_EXTERN int CeedOperatorLinearAssembleQFunctionBuildOrUpdate(CeedOperator op, 72670a7ffb3SJeremy L Thompson CeedVector *assembled, CeedElemRestriction *rstr, CeedRequest *request); 727ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleDiagonal(CeedOperator op, 728ec3da8bcSJed Brown CeedVector assembled, CeedRequest *request); 729ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleAddDiagonal(CeedOperator op, 730ec3da8bcSJed Brown CeedVector assembled, CeedRequest *request); 731ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssemblePointBlockDiagonal(CeedOperator op, 732ec3da8bcSJed Brown CeedVector assembled, CeedRequest *request); 733ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleAddPointBlockDiagonal(CeedOperator op, 734ec3da8bcSJed Brown CeedVector assembled, CeedRequest *request); 735ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleSymbolic(CeedOperator op, 736c30b7fbdSJeremy L Thompson CeedSize *num_entries, CeedInt **rows, CeedInt **cols); 737ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssemble(CeedOperator op, CeedVector values); 738d1d35e2fSjeremylt CEED_EXTERN int CeedOperatorMultigridLevelCreate(CeedOperator op_fine, 739d1d35e2fSjeremylt CeedVector p_mult_fine, CeedElemRestriction rstr_coarse, CeedBasis basis_coarse, 740d1d35e2fSjeremylt CeedOperator *op_coarse, CeedOperator *op_prolong, CeedOperator *op_restrict); 741ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorMultigridLevelCreateTensorH1( 742d1d35e2fSjeremylt CeedOperator op_fine, CeedVector p_mult_fine, CeedElemRestriction rstr_coarse, 743d1d35e2fSjeremylt CeedBasis basis_coarse, const CeedScalar *interp_c_to_f, CeedOperator *op_coarse, 744d1d35e2fSjeremylt CeedOperator *op_prolong, CeedOperator *op_restrict); 745d1d35e2fSjeremylt CEED_EXTERN int CeedOperatorMultigridLevelCreateH1(CeedOperator op_fine, 746d1d35e2fSjeremylt CeedVector p_mult_fine, CeedElemRestriction rstr_coarse, CeedBasis basis_coarse, 747d1d35e2fSjeremylt const CeedScalar *interp_c_to_f, CeedOperator *op_coarse, 748d1d35e2fSjeremylt CeedOperator *op_prolong, CeedOperator *op_restrict); 749ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorCreateFDMElementInverse(CeedOperator op, 750d1d35e2fSjeremylt CeedOperator *fdm_inv, CeedRequest *request); 751cd4dfc48Sjeremylt CEED_EXTERN int CeedOperatorSetNumQuadraturePoints(CeedOperator op, CeedInt num_qpts); 752ea6b5821SJeremy L Thompson CEED_EXTERN int CeedOperatorSetName(CeedOperator op, const char *name); 753ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorView(CeedOperator op, FILE *stream); 754b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedOperatorGetCeed(CeedOperator op, Ceed *ceed); 755b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedOperatorGetNumElements(CeedOperator op, CeedInt *num_elem); 756b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedOperatorGetNumQuadraturePoints(CeedOperator op, 757b7c9bbdaSJeremy L Thompson CeedInt *num_qpts); 7589d36ca50SJeremy L Thompson CEED_EXTERN int CeedOperatorGetFlopsEstimate(CeedOperator op, CeedSize *flops); 7593668ca4bSJeremy L Thompson CEED_EXTERN int CeedOperatorContextGetFieldLabel(CeedOperator op, 7603668ca4bSJeremy L Thompson const char *field_name, CeedContextFieldLabel *field_label); 7613668ca4bSJeremy L Thompson CEED_EXTERN int CeedOperatorContextSetDouble(CeedOperator op, 7627bfe0f0eSJeremy L Thompson CeedContextFieldLabel field_label, double *values); 7633668ca4bSJeremy L Thompson CEED_EXTERN int CeedOperatorContextSetInt32(CeedOperator op, 7647bfe0f0eSJeremy L Thompson CeedContextFieldLabel field_label, int *values); 765ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorApply(CeedOperator op, CeedVector in, 766ec3da8bcSJed Brown CeedVector out, CeedRequest *request); 767ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorApplyAdd(CeedOperator op, CeedVector in, 768ec3da8bcSJed Brown CeedVector out, CeedRequest *request); 769ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorDestroy(CeedOperator *op); 770ec3da8bcSJed Brown 77128567f8fSJeremy L Thompson CEED_EXTERN int CeedOperatorFieldGetName(CeedOperatorField op_field, 77228567f8fSJeremy L Thompson char **field_name); 77343bbe138SJeremy L Thompson CEED_EXTERN int CeedOperatorFieldGetElemRestriction(CeedOperatorField op_field, 77443bbe138SJeremy L Thompson CeedElemRestriction *rstr); 77543bbe138SJeremy L Thompson CEED_EXTERN int CeedOperatorFieldGetBasis(CeedOperatorField op_field, 77643bbe138SJeremy L Thompson CeedBasis *basis); 77743bbe138SJeremy L Thompson CEED_EXTERN int CeedOperatorFieldGetVector(CeedOperatorField op_field, 77843bbe138SJeremy L Thompson CeedVector *vec); 77943bbe138SJeremy L Thompson 780ec3da8bcSJed Brown /** 781ec3da8bcSJed Brown @brief Return integer power 782ec3da8bcSJed Brown 783ec3da8bcSJed Brown @param[in] base The base to exponentiate 784ec3da8bcSJed Brown @param[in] power The power to raise the base to 785ec3da8bcSJed Brown 786ec3da8bcSJed Brown @return base^power 787ec3da8bcSJed Brown 788ec3da8bcSJed Brown @ref Utility 789ec3da8bcSJed Brown **/ 790ec3da8bcSJed Brown static inline CeedInt CeedIntPow(CeedInt base, CeedInt power) { 791ec3da8bcSJed Brown CeedInt result = 1; 792ec3da8bcSJed Brown while (power) { 793ec3da8bcSJed Brown if (power & 1) result *= base; 794ec3da8bcSJed Brown power >>= 1; 795ec3da8bcSJed Brown base *= base; 796ec3da8bcSJed Brown } 797ec3da8bcSJed Brown return result; 798ec3da8bcSJed Brown } 799ec3da8bcSJed Brown 800ec3da8bcSJed Brown /** 801ec3da8bcSJed Brown @brief Return minimum of two integers 802ec3da8bcSJed Brown 803ec3da8bcSJed Brown @param[in] a The first integer to compare 804ec3da8bcSJed Brown @param[in] b The second integer to compare 805ec3da8bcSJed Brown 806ec3da8bcSJed Brown @return The minimum of the two integers 807ec3da8bcSJed Brown 808ec3da8bcSJed Brown @ref Utility 809ec3da8bcSJed Brown **/ 810ec3da8bcSJed Brown static inline CeedInt CeedIntMin(CeedInt a, CeedInt b) { return a < b ? a : b; } 811ec3da8bcSJed Brown 812ec3da8bcSJed Brown /** 813ec3da8bcSJed Brown @brief Return maximum of two integers 814ec3da8bcSJed Brown 815ec3da8bcSJed Brown @param[in] a The first integer to compare 816ec3da8bcSJed Brown @param[in] b The second integer to compare 817ec3da8bcSJed Brown 818ec3da8bcSJed Brown @return The maximum of the two integers 819ec3da8bcSJed Brown 820ec3da8bcSJed Brown @ref Utility 821ec3da8bcSJed Brown **/ 822ec3da8bcSJed Brown static inline CeedInt CeedIntMax(CeedInt a, CeedInt b) { return a > b ? a : b; } 823ec3da8bcSJed Brown 824ec3da8bcSJed Brown // Used to ensure initialization before CeedInit() 825ec3da8bcSJed Brown CEED_EXTERN int CeedRegisterAll(void); 826ec3da8bcSJed Brown // Used to ensure initialization before CeedQFunctionCreate*() 827ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionRegisterAll(void); 828ec3da8bcSJed Brown 829ec3da8bcSJed Brown #endif 830