xref: /libCEED/rust/libceed-sys/c-src/include/ceed/ceed.h (revision 3047f7896db8a74672e0eaf4fe2741a47841cabb)
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
8ec3da8bcSJed Brown /// 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 
49ec3da8bcSJed Brown /**
50ec3da8bcSJed Brown   CEED_EXTERN is used in this header to denote all publicly visible symbols.
51ec3da8bcSJed Brown 
52ec3da8bcSJed Brown   No other file should declare publicly visible symbols, thus it should never be
53ec3da8bcSJed Brown   used outside ceed.h.
54ec3da8bcSJed Brown  */
55ec3da8bcSJed Brown #ifdef __cplusplus
56ec3da8bcSJed Brown #  define CEED_EXTERN extern "C"
57ec3da8bcSJed Brown #else
58ec3da8bcSJed Brown #  define CEED_EXTERN extern
59ec3da8bcSJed Brown #endif
60ec3da8bcSJed Brown 
61ec3da8bcSJed Brown /**
62ec3da8bcSJed Brown   @ingroup CeedQFunction
63ec3da8bcSJed Brown   This macro populates the correct function annotations for User QFunction
64ec3da8bcSJed Brown     source for code generation backends or populates default values for CPU
654b97e6d3SJames Wright     backends. It also creates a variable `name_loc` populated with the correct
664b97e6d3SJames Wright     source path for creating the respective User QFunction.
67ec3da8bcSJed Brown **/
68ec3da8bcSJed Brown #ifndef CEED_QFUNCTION
69ec3da8bcSJed Brown #define CEED_QFUNCTION(name) \
70ec3da8bcSJed Brown   static const char name ## _loc[] = __FILE__ ":" #name;        \
71ec3da8bcSJed Brown   static int name
72ec3da8bcSJed Brown #endif
73ec3da8bcSJed Brown 
74ec3da8bcSJed Brown /**
75ec3da8bcSJed Brown   @ingroup CeedQFunction
7677841947SLeila Ghaffari   This macro populates the correct function annotations for User QFunction
7777841947SLeila Ghaffari     helper function source for code generation backends or populates default
7877841947SLeila Ghaffari     values for CPU backends.
7977841947SLeila Ghaffari **/
8077841947SLeila Ghaffari #ifndef CEED_QFUNCTION_HELPER
8177841947SLeila Ghaffari #define CEED_QFUNCTION_HELPER static inline
8277841947SLeila Ghaffari #endif
8377841947SLeila Ghaffari 
8477841947SLeila Ghaffari /**
8577841947SLeila Ghaffari   @ingroup CeedQFunction
86ec3da8bcSJed Brown   Using VLA syntax to reshape User QFunction inputs and outputs can make
87ec3da8bcSJed Brown     user code more readable. VLA is a C99 feature that is not supported by
88ec3da8bcSJed Brown     the C++ dialect used by CUDA. This macro allows users to use the VLA
89ec3da8bcSJed Brown     syntax with the CUDA backends.
90ec3da8bcSJed Brown **/
91ec3da8bcSJed Brown #ifndef CEED_Q_VLA
92ec3da8bcSJed Brown #  define CEED_Q_VLA Q
93ec3da8bcSJed Brown #endif
94ec3da8bcSJed Brown 
95ec3da8bcSJed Brown /**
96ec3da8bcSJed Brown   @ingroup Ceed
97ec3da8bcSJed Brown   This macro provides the appropriate SIMD Pragma for the compilation
98ec3da8bcSJed Brown     environment. Code generation backends may redefine this macro, as needed.
99ec3da8bcSJed Brown **/
100ec3da8bcSJed Brown #ifndef CeedPragmaSIMD
101ec3da8bcSJed Brown #  if defined(__INTEL_COMPILER)
102ec3da8bcSJed Brown #    define CeedPragmaSIMD _Pragma("vector")
103ec3da8bcSJed Brown // Cannot use Intel pragma ivdep because it miscompiles unpacking symmetric tensors, as in
104ec3da8bcSJed Brown // Poisson2DApply, where the SIMD loop body contains temporaries such as the following.
105ec3da8bcSJed Brown //
106ec3da8bcSJed Brown //     const CeedScalar dXdxdXdxT[2][2] = {{qd[i+0*Q], qd[i+2*Q]},
107ec3da8bcSJed Brown //                                         {qd[i+2*Q], qd[i+1*Q]}};
108ec3da8bcSJed Brown //     for (int j=0; j<2; j++)
109ec3da8bcSJed Brown //        vg[i+j*Q] = (du[0] * dXdxdXdxT[0][j] + du[1] * dXdxdXdxT[1][j]);
110ec3da8bcSJed Brown //
111ec3da8bcSJed Brown // Miscompilation with pragma ivdep observed with icc (ICC) 19.0.5.281 20190815
112ec3da8bcSJed Brown // at -O2 and above.
113ec3da8bcSJed Brown #  elif defined(__GNUC__) && __GNUC__ >= 5
114ec3da8bcSJed Brown #    define CeedPragmaSIMD _Pragma("GCC ivdep")
115ec3da8bcSJed Brown #  elif defined(_OPENMP) && _OPENMP >= 201307 // OpenMP-4.0 (July, 2013)
116ec3da8bcSJed Brown #    define CeedPragmaSIMD _Pragma("omp simd")
117ec3da8bcSJed Brown #  else
118ec3da8bcSJed Brown #    define CeedPragmaSIMD
119ec3da8bcSJed Brown #  endif
120ec3da8bcSJed Brown #endif
121ec3da8bcSJed Brown 
122ec3da8bcSJed Brown #include <stdarg.h>
123ec3da8bcSJed Brown #include <stdbool.h>
124*3047f789SJeremy L Thompson #include <stddef.h>
125*3047f789SJeremy L Thompson #include <stdint.h>
126*3047f789SJeremy L Thompson #include <stdio.h>
127ec3da8bcSJed Brown 
128ec3da8bcSJed Brown /// Integer type, used for indexing
129ec3da8bcSJed Brown /// @ingroup Ceed
130ec3da8bcSJed Brown typedef int32_t CeedInt;
13180a9ef05SNatalie Beams 
13280a9ef05SNatalie Beams /// Scalar (floating point) types
13380a9ef05SNatalie Beams ///
134ec3da8bcSJed Brown /// @ingroup Ceed
13580a9ef05SNatalie Beams typedef enum {
13680a9ef05SNatalie Beams   /// Single precision
13780a9ef05SNatalie Beams   CEED_SCALAR_FP32,
13880a9ef05SNatalie Beams   /// Double precision
13980a9ef05SNatalie Beams   CEED_SCALAR_FP64
14080a9ef05SNatalie Beams } CeedScalarType;
14180a9ef05SNatalie Beams /// Base scalar type for the library to use: change which header is
14280a9ef05SNatalie Beams /// included to change the precision.
14380a9ef05SNatalie Beams #include "ceed-f64.h"
144ec3da8bcSJed Brown 
145ec3da8bcSJed Brown /// Library context created by CeedInit()
146ec3da8bcSJed Brown /// @ingroup CeedUser
147ec3da8bcSJed Brown typedef struct Ceed_private *Ceed;
148ec3da8bcSJed Brown /// Non-blocking Ceed interfaces return a CeedRequest.
149ec3da8bcSJed Brown /// To perform an operation immediately, pass \ref CEED_REQUEST_IMMEDIATE instead.
150ec3da8bcSJed Brown /// @ingroup CeedUser
151ec3da8bcSJed Brown typedef struct CeedRequest_private *CeedRequest;
152ec3da8bcSJed Brown /// Handle for vectors over the field \ref CeedScalar
153ec3da8bcSJed Brown /// @ingroup CeedVectorUser
154ec3da8bcSJed Brown typedef struct CeedVector_private *CeedVector;
155ec3da8bcSJed Brown /// Handle for object describing restriction to elements
156ec3da8bcSJed Brown /// @ingroup CeedElemRestrictionUser
157ec3da8bcSJed Brown typedef struct CeedElemRestriction_private *CeedElemRestriction;
158ec3da8bcSJed Brown /// Handle for object describing discrete finite element evaluations
159ec3da8bcSJed Brown /// @ingroup CeedBasisUser
160ec3da8bcSJed Brown typedef struct CeedBasis_private *CeedBasis;
16143bbe138SJeremy L Thompson /// Handle for object describing CeedQFunction fields
16243bbe138SJeremy L Thompson /// @ingroup CeedQFunctionBackend
16343bbe138SJeremy L Thompson typedef struct CeedQFunctionField_private *CeedQFunctionField;
164ec3da8bcSJed Brown /// Handle for object describing functions evaluated independently at quadrature points
165ec3da8bcSJed Brown /// @ingroup CeedQFunctionUser
166ec3da8bcSJed Brown typedef struct CeedQFunction_private *CeedQFunction;
16743bbe138SJeremy L Thompson /// Handle for object describing CeedOperator fields
16843bbe138SJeremy L Thompson /// @ingroup CeedOperatorBackend
16943bbe138SJeremy L Thompson typedef struct CeedOperatorField_private *CeedOperatorField;
170ec3da8bcSJed Brown /// Handle for object describing context data for CeedQFunctions
171ec3da8bcSJed Brown /// @ingroup CeedQFunctionUser
172ec3da8bcSJed Brown typedef struct CeedQFunctionContext_private *CeedQFunctionContext;
1733668ca4bSJeremy L Thompson /// Handle for object describing registered fields for CeedQFunctionContext
1743668ca4bSJeremy L Thompson /// @ingroup CeedQFunctionUser
1753668ca4bSJeremy L Thompson typedef struct CeedContextFieldLabel_private *CeedContextFieldLabel;
176ec3da8bcSJed Brown /// Handle for object describing FE-type operators acting on vectors
177ec3da8bcSJed Brown ///
178ec3da8bcSJed Brown /// Given an element restriction \f$E\f$, basis evaluator \f$B\f$, and
179ec3da8bcSJed Brown ///   quadrature function\f$f\f$, a CeedOperator expresses operations of the form
180ec3da8bcSJed Brown ///   $$ E^T B^T f(B E u) $$
181ec3da8bcSJed Brown ///   acting on the vector \f$u\f$.
182ec3da8bcSJed Brown /// @ingroup CeedOperatorUser
183ec3da8bcSJed Brown typedef struct CeedOperator_private *CeedOperator;
184ec3da8bcSJed Brown 
18522e44211Sjeremylt CEED_EXTERN int CeedRegistryGetList(size_t *n, char ***const resources, CeedInt **array);
186ec3da8bcSJed Brown CEED_EXTERN int CeedInit(const char *resource, Ceed *ceed);
1879560d06aSjeremylt CEED_EXTERN int CeedReferenceCopy(Ceed ceed, Ceed *ceed_copy);
188ec3da8bcSJed Brown CEED_EXTERN int CeedGetResource(Ceed ceed, const char **resource);
189d1d35e2fSjeremylt CEED_EXTERN int CeedIsDeterministic(Ceed ceed, bool *is_deterministic);
190ec3da8bcSJed Brown CEED_EXTERN int CeedView(Ceed ceed, FILE *stream);
191ec3da8bcSJed Brown CEED_EXTERN int CeedDestroy(Ceed *ceed);
192ec3da8bcSJed Brown 
193ec3da8bcSJed Brown CEED_EXTERN int CeedErrorImpl(Ceed, const char *, int, const char *, int,
194ec3da8bcSJed Brown                               const char *, ...);
195ec3da8bcSJed Brown /// Raise an error on ceed object
196ec3da8bcSJed Brown ///
197ec3da8bcSJed Brown /// @param ceed Ceed library context or NULL
198ec3da8bcSJed Brown /// @param ecode Error code (int)
199ec3da8bcSJed Brown /// @param ... printf-style format string followed by arguments as needed
200ec3da8bcSJed Brown ///
201ec3da8bcSJed Brown /// @ingroup Ceed
202ec3da8bcSJed Brown /// @sa CeedSetErrorHandler()
203ec3da8bcSJed Brown #if defined(__clang__)
204ec3da8bcSJed Brown /// Use nonstandard ternary to convince the compiler/clang-tidy that this
205ec3da8bcSJed Brown /// function never returns zero.
206ec3da8bcSJed Brown #  define CeedError(ceed, ecode, ...)                                     \
20745f1e315Sjeremylt   (CeedErrorImpl((ceed), __FILE__, __LINE__, __func__, (ecode), __VA_ARGS__), (ecode))
208ec3da8bcSJed Brown #else
209ec3da8bcSJed Brown #  define CeedError(ceed, ecode, ...)                                     \
210ec3da8bcSJed Brown   CeedErrorImpl((ceed), __FILE__, __LINE__, __func__, (ecode), __VA_ARGS__) ?: (ecode)
211ec3da8bcSJed Brown #endif
212ec3da8bcSJed Brown 
213ec3da8bcSJed Brown /// Ceed error handlers
214ec3da8bcSJed Brown CEED_EXTERN int CeedErrorReturn(Ceed, const char *, int, const char *, int,
215ec3da8bcSJed Brown                                 const char *, va_list *);
216ec3da8bcSJed Brown CEED_EXTERN int CeedErrorStore(Ceed, const char *, int, const char *, int,
217ec3da8bcSJed Brown                                const char *, va_list *);
218ec3da8bcSJed Brown CEED_EXTERN int CeedErrorAbort(Ceed, const char *, int, const char *, int,
219ec3da8bcSJed Brown                                const char *, va_list *);
220ec3da8bcSJed Brown CEED_EXTERN int CeedErrorExit(Ceed, const char *, int, const char *, int,
221ec3da8bcSJed Brown                               const char *, va_list *);
222ec3da8bcSJed Brown typedef int (*CeedErrorHandler)(Ceed, const char *, int,
223ec3da8bcSJed Brown                                 const char *, int, const char *,
224ec3da8bcSJed Brown                                 va_list *);
225ec3da8bcSJed Brown CEED_EXTERN int CeedSetErrorHandler(Ceed ceed, CeedErrorHandler eh);
226d1d35e2fSjeremylt CEED_EXTERN int CeedGetErrorMessage(Ceed, const char **err_msg);
227d1d35e2fSjeremylt CEED_EXTERN int CeedResetErrorMessage(Ceed, const char **err_msg);
228ec3da8bcSJed Brown 
229ec3da8bcSJed Brown /// libCEED library version numbering
230ec3da8bcSJed Brown /// @ingroup Ceed
231ec3da8bcSJed Brown #define CEED_VERSION_MAJOR 0
232d66340f5SJed Brown #define CEED_VERSION_MINOR 9
233ec3da8bcSJed Brown #define CEED_VERSION_PATCH 0
234a6fd1c49SJed Brown #define CEED_VERSION_RELEASE false
235ec3da8bcSJed Brown 
236ec3da8bcSJed Brown /// Compile-time check that the the current library version is at least as
237ec3da8bcSJed Brown /// recent as the specified version. This macro is typically used in
238ec3da8bcSJed Brown /// @code
239ec3da8bcSJed Brown /// #if CEED_VERSION_GE(0, 8, 0)
240ec3da8bcSJed Brown ///   code path that needs at least 0.8.0
241ec3da8bcSJed Brown /// #else
242ec3da8bcSJed Brown ///   fallback code for older versions
243ec3da8bcSJed Brown /// #endif
244ec3da8bcSJed Brown /// @endcode
245ec3da8bcSJed Brown ///
246ec3da8bcSJed Brown /// A non-release version always compares as positive infinity.
247ec3da8bcSJed Brown ///
248ec3da8bcSJed Brown /// @param major   Major version
249ec3da8bcSJed Brown /// @param minor   Minor version
250ec3da8bcSJed Brown /// @param patch   Patch (subminor) version
251ec3da8bcSJed Brown ///
252ec3da8bcSJed Brown /// @ingroup Ceed
253ec3da8bcSJed Brown /// @sa CeedGetVersion()
254ec3da8bcSJed Brown #define CEED_VERSION_GE(major, minor, patch)                                   \
255ec3da8bcSJed Brown   (!CEED_VERSION_RELEASE ||                                                    \
256ec3da8bcSJed Brown    (CEED_VERSION_MAJOR > major ||                                              \
257ec3da8bcSJed Brown     (CEED_VERSION_MAJOR == major &&                                            \
258ec3da8bcSJed Brown      (CEED_VERSION_MINOR > minor ||                                            \
259ec3da8bcSJed Brown       (CEED_VERSION_MINOR == minor && CEED_VERSION_PATCH >= patch)))))
260ec3da8bcSJed Brown 
261ec3da8bcSJed Brown CEED_EXTERN int CeedGetVersion(int *major, int *minor, int *patch,
262ec3da8bcSJed Brown                                bool *release);
263ec3da8bcSJed Brown 
26480a9ef05SNatalie Beams CEED_EXTERN int CeedGetScalarType(CeedScalarType *scalar_type);
26580a9ef05SNatalie Beams 
266ec3da8bcSJed Brown /// Ceed Errors
267ec3da8bcSJed Brown ///
268ec3da8bcSJed Brown /// This enum is used to specify the type of error returned by a function.
269ec3da8bcSJed Brown /// A zero error code is success, negative error codes indicate terminal errors
270ec3da8bcSJed Brown /// and positive error codes indicate nonterminal errors. With nonterminal errors
271ec3da8bcSJed Brown /// the object state has not been modifiend, but with terminal errors the object
272ec3da8bcSJed Brown /// data is likely modified or corrupted.
273ec3da8bcSJed Brown /// @ingroup Ceed
274ec3da8bcSJed Brown typedef enum {
2750f7fd0f8Sjeremylt   /// Success error code
276ec3da8bcSJed Brown   CEED_ERROR_SUCCESS      = 0,
277ec3da8bcSJed Brown   /// Minor error, generic
278ec3da8bcSJed Brown   CEED_ERROR_MINOR        = 1,
279ec3da8bcSJed Brown   /// Minor error, dimension mismatch in inputs
280ec3da8bcSJed Brown   CEED_ERROR_DIMENSION    = 2,
281ec3da8bcSJed Brown   /// Minor error, incomplete object setup
282ec3da8bcSJed Brown   CEED_ERROR_INCOMPLETE   = 3,
283ec3da8bcSJed Brown   /// Minor error, incompatible arguments/configuration
284ec3da8bcSJed Brown   CEED_ERROR_INCOMPATIBLE = 4,
285ec3da8bcSJed Brown   /// Minor error, access lock problem
286ec3da8bcSJed Brown   CEED_ERROR_ACCESS       = 5,
287ec3da8bcSJed Brown   /// Major error, generic
288ec3da8bcSJed Brown   CEED_ERROR_MAJOR        = -1,
289ec3da8bcSJed Brown   /// Major error, internal backend error
290ec3da8bcSJed Brown   CEED_ERROR_BACKEND      = -2,
291ec3da8bcSJed Brown   /// Major error, operation unsupported by current backend
292ec3da8bcSJed Brown   CEED_ERROR_UNSUPPORTED  = -3,
293ec3da8bcSJed Brown } CeedErrorType;
294ec3da8bcSJed Brown CEED_EXTERN const char *const *CeedErrorTypes;
295ec3da8bcSJed Brown 
296ec3da8bcSJed Brown /// Specify memory type
297ec3da8bcSJed Brown ///
298ec3da8bcSJed Brown /// Many Ceed interfaces take or return pointers to memory.  This enum is used to
299ec3da8bcSJed Brown /// specify where the memory being provided or requested must reside.
300ec3da8bcSJed Brown /// @ingroup Ceed
301ec3da8bcSJed Brown typedef enum {
302ec3da8bcSJed Brown   /// Memory resides on the host
303ec3da8bcSJed Brown   CEED_MEM_HOST,
304ec3da8bcSJed Brown   /// Memory resides on a device (corresponding to \ref Ceed resource)
305ec3da8bcSJed Brown   CEED_MEM_DEVICE,
306ec3da8bcSJed Brown } CeedMemType;
307ec3da8bcSJed Brown CEED_EXTERN const char *const CeedMemTypes[];
308ec3da8bcSJed Brown 
309ec3da8bcSJed Brown CEED_EXTERN int CeedGetPreferredMemType(Ceed ceed, CeedMemType *type);
310ec3da8bcSJed Brown 
311ec3da8bcSJed Brown /// Conveys ownership status of arrays passed to Ceed interfaces.
312ec3da8bcSJed Brown /// @ingroup Ceed
313ec3da8bcSJed Brown typedef enum {
314ec3da8bcSJed Brown   /// Implementation will copy the values and not store the passed pointer.
315ec3da8bcSJed Brown   CEED_COPY_VALUES,
316ec3da8bcSJed Brown   /// Implementation can use and modify the data provided by the user, but does
317ec3da8bcSJed Brown   /// not take ownership.
318ec3da8bcSJed Brown   CEED_USE_POINTER,
319ec3da8bcSJed Brown   /// Implementation takes ownership of the pointer and will free using
320ec3da8bcSJed Brown   /// CeedFree() when done using it.  The user should not assume that the
321ec3da8bcSJed Brown   /// pointer remains valid after ownership has been transferred.  Note that
322ec3da8bcSJed Brown   /// arrays allocated using C++ operator new or other allocators cannot
323ec3da8bcSJed Brown   /// generally be freed using CeedFree().  CeedFree() is capable of freeing any
324ec3da8bcSJed Brown   /// memory that can be freed using free(3).
325ec3da8bcSJed Brown   CEED_OWN_POINTER,
326ec3da8bcSJed Brown } CeedCopyMode;
327cdf32b93SJeremy L Thompson CEED_EXTERN const char *const CeedCopyModes[];
328ec3da8bcSJed Brown 
329ec3da8bcSJed Brown /// Denotes type of vector norm to be computed
330ec3da8bcSJed Brown /// @ingroup CeedVector
331ec3da8bcSJed Brown typedef enum {
332ec3da8bcSJed Brown   /// L_1 norm: sum_i |x_i|
333ec3da8bcSJed Brown   CEED_NORM_1,
334ec3da8bcSJed Brown   /// L_2 norm: sqrt(sum_i |x_i|^2)
335ec3da8bcSJed Brown   CEED_NORM_2,
336ec3da8bcSJed Brown   /// L_Infinity norm: max_i |x_i|
337ec3da8bcSJed Brown   CEED_NORM_MAX,
338ec3da8bcSJed Brown } CeedNormType;
339ec3da8bcSJed Brown 
340ec3da8bcSJed Brown CEED_EXTERN int CeedVectorCreate(Ceed ceed, CeedInt len, CeedVector *vec);
3419560d06aSjeremylt CEED_EXTERN int CeedVectorReferenceCopy(CeedVector vec, CeedVector *vec_copy);
342d1d35e2fSjeremylt CEED_EXTERN int CeedVectorSetArray(CeedVector vec, CeedMemType mem_type,
343d1d35e2fSjeremylt                                    CeedCopyMode copy_mode, CeedScalar *array);
344ec3da8bcSJed Brown CEED_EXTERN int CeedVectorSetValue(CeedVector vec, CeedScalar value);
345d1d35e2fSjeremylt CEED_EXTERN int CeedVectorSyncArray(CeedVector vec, CeedMemType mem_type);
346d1d35e2fSjeremylt CEED_EXTERN int CeedVectorTakeArray(CeedVector vec, CeedMemType mem_type,
347ec3da8bcSJed Brown                                     CeedScalar **array);
348d1d35e2fSjeremylt CEED_EXTERN int CeedVectorGetArray(CeedVector vec, CeedMemType mem_type,
349ec3da8bcSJed Brown                                    CeedScalar **array);
350d1d35e2fSjeremylt CEED_EXTERN int CeedVectorGetArrayRead(CeedVector vec, CeedMemType mem_type,
351ec3da8bcSJed Brown                                        const CeedScalar **array);
3529c774eddSJeremy L Thompson CEED_EXTERN int CeedVectorGetArrayWrite(CeedVector vec, CeedMemType mem_type,
3539c774eddSJeremy L Thompson                                         CeedScalar **array);
354ec3da8bcSJed Brown CEED_EXTERN int CeedVectorRestoreArray(CeedVector vec, CeedScalar **array);
355ec3da8bcSJed Brown CEED_EXTERN int CeedVectorRestoreArrayRead(CeedVector vec,
356ec3da8bcSJed Brown     const CeedScalar **array);
357ec3da8bcSJed Brown CEED_EXTERN int CeedVectorNorm(CeedVector vec, CeedNormType type,
358ec3da8bcSJed Brown                                CeedScalar *norm);
359e0dd3b27Sjeremylt CEED_EXTERN int CeedVectorScale(CeedVector x, CeedScalar alpha);
3600f7fd0f8Sjeremylt CEED_EXTERN int CeedVectorAXPY(CeedVector y, CeedScalar alpha, CeedVector x);
3610f7fd0f8Sjeremylt CEED_EXTERN int CeedVectorPointwiseMult(CeedVector w, CeedVector x, CeedVector y);
362ec3da8bcSJed Brown CEED_EXTERN int CeedVectorReciprocal(CeedVector vec);
363d1d35e2fSjeremylt CEED_EXTERN int CeedVectorView(CeedVector vec, const char *fp_fmt, FILE *stream);
364b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedVectorGetCeed(CeedVector vec, Ceed *ceed);
365ec3da8bcSJed Brown CEED_EXTERN int CeedVectorGetLength(CeedVector vec, CeedInt *length);
366ec3da8bcSJed Brown CEED_EXTERN int CeedVectorDestroy(CeedVector *vec);
367ec3da8bcSJed Brown 
368ec3da8bcSJed Brown CEED_EXTERN CeedRequest *const CEED_REQUEST_IMMEDIATE;
369ec3da8bcSJed Brown CEED_EXTERN CeedRequest *const CEED_REQUEST_ORDERED;
370ec3da8bcSJed Brown CEED_EXTERN int CeedRequestWait(CeedRequest *req);
371ec3da8bcSJed Brown 
372ec3da8bcSJed Brown /// Argument for CeedOperatorSetField that vector is collocated with
373ec3da8bcSJed Brown /// quadrature points, used with QFunction eval mode CEED_EVAL_NONE
374ec3da8bcSJed Brown /// or CEED_EVAL_INTERP only, not with CEED_EVAL_GRAD, CEED_EVAL_DIV,
375ec3da8bcSJed Brown /// or CEED_EVAL_CURL
376ec3da8bcSJed Brown /// @ingroup CeedBasis
377ec3da8bcSJed Brown CEED_EXTERN const CeedBasis CEED_BASIS_COLLOCATED;
378ec3da8bcSJed Brown 
379ec3da8bcSJed Brown /// Argument for CeedOperatorSetField to use active input or output
380ec3da8bcSJed Brown /// @ingroup CeedVector
381ec3da8bcSJed Brown CEED_EXTERN const CeedVector CEED_VECTOR_ACTIVE;
382ec3da8bcSJed Brown 
383ec3da8bcSJed Brown /// Argument for CeedOperatorSetField to use no vector, used with
384ec3da8bcSJed Brown /// qfunction input with eval mode CEED_EVAL_WEIGHT
385ec3da8bcSJed Brown /// @ingroup CeedVector
386ec3da8bcSJed Brown CEED_EXTERN const CeedVector CEED_VECTOR_NONE;
387ec3da8bcSJed Brown 
388ec3da8bcSJed Brown /// Argument for CeedOperatorSetField to use no ElemRestriction, only used with
389ec3da8bcSJed Brown /// eval mode CEED_EVAL_WEIGHT.
390ec3da8bcSJed Brown /// @ingroup CeedElemRestriction
391ec3da8bcSJed Brown CEED_EXTERN const CeedElemRestriction CEED_ELEMRESTRICTION_NONE;
392ec3da8bcSJed Brown 
393ec3da8bcSJed Brown /// Argument for CeedOperatorCreate that QFunction is not created by user.
394ec3da8bcSJed Brown /// Only used for QFunctions dqf and dqfT. If implemented, a backend may
395ec3da8bcSJed Brown /// attempt to provide the action of these QFunctions.
396ec3da8bcSJed Brown /// @ingroup CeedQFunction
397ec3da8bcSJed Brown CEED_EXTERN const CeedQFunction CEED_QFUNCTION_NONE;
398ec3da8bcSJed Brown 
399ec3da8bcSJed Brown /// Denotes whether a linear transformation or its transpose should be applied
400ec3da8bcSJed Brown /// @ingroup CeedBasis
401ec3da8bcSJed Brown typedef enum {
402ec3da8bcSJed Brown   /// Apply the linear transformation
403ec3da8bcSJed Brown   CEED_NOTRANSPOSE,
404ec3da8bcSJed Brown   /// Apply the transpose
405ec3da8bcSJed Brown   CEED_TRANSPOSE
406ec3da8bcSJed Brown } CeedTransposeMode;
407ec3da8bcSJed Brown CEED_EXTERN const char *const CeedTransposeModes[];
408ec3da8bcSJed Brown 
409ec3da8bcSJed Brown /// Argument for CeedElemRestrictionCreateStrided that L-vector is in
410ec3da8bcSJed Brown /// the Ceed backend's preferred layout. This argument should only be used
411ec3da8bcSJed Brown /// with vectors created by a Ceed backend.
412ec3da8bcSJed Brown /// @ingroup CeedElemRestriction
413ec3da8bcSJed Brown CEED_EXTERN const CeedInt CEED_STRIDES_BACKEND[3];
414ec3da8bcSJed Brown 
415d1d35e2fSjeremylt CEED_EXTERN int CeedElemRestrictionCreate(Ceed ceed, CeedInt num_elem,
416d1d35e2fSjeremylt     CeedInt elem_size, CeedInt num_comp, CeedInt comp_stride, CeedInt l_size,
417d1d35e2fSjeremylt     CeedMemType mem_type, CeedCopyMode copy_mode, const CeedInt *offsets,
418ec3da8bcSJed Brown     CeedElemRestriction *rstr);
419fc0567d9Srezgarshakeri CEED_EXTERN int CeedElemRestrictionCreateOriented(Ceed ceed, CeedInt num_elem,
420fc0567d9Srezgarshakeri     CeedInt elem_size, CeedInt num_comp, CeedInt comp_stride, CeedInt l_size,
421fc0567d9Srezgarshakeri     CeedMemType mem_type, CeedCopyMode copy_mode, const CeedInt *offsets,
422fc0567d9Srezgarshakeri     const bool *orient, CeedElemRestriction *rstr);
423ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionCreateStrided(Ceed ceed,
424d1d35e2fSjeremylt     CeedInt num_elem, CeedInt elem_size, CeedInt num_comp, CeedInt l_size,
425ec3da8bcSJed Brown     const CeedInt strides[3], CeedElemRestriction *rstr);
426d1d35e2fSjeremylt CEED_EXTERN int CeedElemRestrictionCreateBlocked(Ceed ceed, CeedInt num_elem,
427d1d35e2fSjeremylt     CeedInt elem_size, CeedInt blk_size, CeedInt num_comp, CeedInt comp_stride,
428d1d35e2fSjeremylt     CeedInt l_size, CeedMemType mem_type, CeedCopyMode copy_mode,
429ec3da8bcSJed Brown     const CeedInt *offsets, CeedElemRestriction *rstr);
430ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionCreateBlockedStrided(Ceed ceed,
431d1d35e2fSjeremylt     CeedInt num_elem, CeedInt elem_size, CeedInt blk_size, CeedInt num_comp,
432d1d35e2fSjeremylt     CeedInt l_size, const CeedInt strides[3], CeedElemRestriction *rstr);
4339560d06aSjeremylt CEED_EXTERN int CeedElemRestrictionReferenceCopy(CeedElemRestriction rstr,
4349560d06aSjeremylt     CeedElemRestriction *rstr_copy);
435ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionCreateVector(CeedElemRestriction rstr,
436ec3da8bcSJed Brown     CeedVector *lvec, CeedVector *evec);
437ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionApply(CeedElemRestriction rstr,
438d1d35e2fSjeremylt     CeedTransposeMode t_mode, CeedVector u, CeedVector ru, CeedRequest *request);
439ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionApplyBlock(CeedElemRestriction rstr,
440d1d35e2fSjeremylt     CeedInt block, CeedTransposeMode t_mode, CeedVector u, CeedVector ru,
441ec3da8bcSJed Brown     CeedRequest *request);
442b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedElemRestrictionGetCeed(CeedElemRestriction rstr,
443b7c9bbdaSJeremy L Thompson     Ceed *ceed);
444ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetCompStride(CeedElemRestriction rstr,
445d1d35e2fSjeremylt     CeedInt *comp_stride);
446ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetNumElements(CeedElemRestriction rstr,
447d1d35e2fSjeremylt     CeedInt *num_elem);
448ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetElementSize(CeedElemRestriction rstr,
449d1d35e2fSjeremylt     CeedInt *elem_size);
450ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetLVectorSize(CeedElemRestriction rstr,
451d1d35e2fSjeremylt     CeedInt *l_size);
452ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetNumComponents(CeedElemRestriction rstr,
453d1d35e2fSjeremylt     CeedInt *num_comp);
454ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetNumBlocks(CeedElemRestriction rstr,
455d1d35e2fSjeremylt     CeedInt *num_blk);
456ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetBlockSize(CeedElemRestriction rstr,
457d1d35e2fSjeremylt     CeedInt *blk_size);
458ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionGetMultiplicity(CeedElemRestriction rstr,
459ec3da8bcSJed Brown     CeedVector mult);
460ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionView(CeedElemRestriction rstr, FILE *stream);
461ec3da8bcSJed Brown CEED_EXTERN int CeedElemRestrictionDestroy(CeedElemRestriction *rstr);
462ec3da8bcSJed Brown 
463ec3da8bcSJed Brown // The formalism here is that we have the structure
464ec3da8bcSJed Brown //  \int_\Omega v^T f_0(u, \nabla u, qdata) + (\nabla v)^T f_1(u, \nabla u, qdata)
465ec3da8bcSJed Brown // where gradients are with respect to the reference element.
466ec3da8bcSJed Brown 
467ec3da8bcSJed Brown /// Basis evaluation mode
468ec3da8bcSJed Brown ///
469ec3da8bcSJed Brown /// Modes can be bitwise ORed when passing to most functions.
470ec3da8bcSJed Brown /// @ingroup CeedBasis
471ec3da8bcSJed Brown typedef enum {
472ec3da8bcSJed Brown   /// Perform no evaluation (either because there is no data or it is already at
473ec3da8bcSJed Brown   /// quadrature points)
474ec3da8bcSJed Brown   CEED_EVAL_NONE   = 0,
475ec3da8bcSJed Brown   /// Interpolate from nodes to quadrature points
476ec3da8bcSJed Brown   CEED_EVAL_INTERP = 1,
477ec3da8bcSJed Brown   /// Evaluate gradients at quadrature points from input in a nodal basis
478ec3da8bcSJed Brown   CEED_EVAL_GRAD   = 2,
479ec3da8bcSJed Brown   /// Evaluate divergence at quadrature points from input in a nodal basis
480ec3da8bcSJed Brown   CEED_EVAL_DIV    = 4,
481ec3da8bcSJed Brown   /// Evaluate curl at quadrature points from input in a nodal basis
482ec3da8bcSJed Brown   CEED_EVAL_CURL   = 8,
483ec3da8bcSJed Brown   /// Using no input, evaluate quadrature weights on the reference element
484ec3da8bcSJed Brown   CEED_EVAL_WEIGHT = 16,
485ec3da8bcSJed Brown } CeedEvalMode;
486ec3da8bcSJed Brown CEED_EXTERN const char *const CeedEvalModes[];
487ec3da8bcSJed Brown 
488ec3da8bcSJed Brown /// Type of quadrature; also used for location of nodes
489ec3da8bcSJed Brown /// @ingroup CeedBasis
490ec3da8bcSJed Brown typedef enum {
491ec3da8bcSJed Brown   /// Gauss-Legendre quadrature
492ec3da8bcSJed Brown   CEED_GAUSS         = 0,
493ec3da8bcSJed Brown   /// Gauss-Legendre-Lobatto quadrature
494ec3da8bcSJed Brown   CEED_GAUSS_LOBATTO = 1,
495ec3da8bcSJed Brown } CeedQuadMode;
496ec3da8bcSJed Brown CEED_EXTERN const char *const CeedQuadModes[];
497ec3da8bcSJed Brown 
498ec3da8bcSJed Brown /// Type of basis shape to create non-tensor H1 element basis
499ec3da8bcSJed Brown ///
500ec3da8bcSJed Brown /// Dimension can be extracted with bitwise AND
501ec3da8bcSJed Brown /// (CeedElemTopology & 2**(dim + 2)) == TRUE
502ec3da8bcSJed Brown /// @ingroup CeedBasis
503ec3da8bcSJed Brown typedef enum {
504ec3da8bcSJed Brown   /// Line
50550c301a5SRezgar Shakeri   CEED_TOPOLOGY_LINE     = 1 << 16 | 0,
506ec3da8bcSJed Brown   /// Triangle - 2D shape
50750c301a5SRezgar Shakeri   CEED_TOPOLOGY_TRIANGLE = 2 << 16 | 1,
508ec3da8bcSJed Brown   /// Quadralateral - 2D shape
50950c301a5SRezgar Shakeri   CEED_TOPOLOGY_QUAD     = 2 << 16 | 2,
510ec3da8bcSJed Brown   /// Tetrahedron - 3D shape
51150c301a5SRezgar Shakeri   CEED_TOPOLOGY_TET      = 3 << 16 | 3,
512ec3da8bcSJed Brown   /// Pyramid - 3D shape
51350c301a5SRezgar Shakeri   CEED_TOPOLOGY_PYRAMID  = 3 << 16 | 4,
514ec3da8bcSJed Brown   /// Prism - 3D shape
51550c301a5SRezgar Shakeri   CEED_TOPOLOGY_PRISM    = 3 << 16 | 5,
516ec3da8bcSJed Brown   /// Hexehedron - 3D shape
51750c301a5SRezgar Shakeri   CEED_TOPOLOGY_HEX      = 3 << 16 | 6,
518ec3da8bcSJed Brown } CeedElemTopology;
519ec3da8bcSJed Brown CEED_EXTERN const char *const CeedElemTopologies[];
520ec3da8bcSJed Brown 
521ec3da8bcSJed Brown CEED_EXTERN int CeedBasisCreateTensorH1Lagrange(Ceed ceed, CeedInt dim,
522d1d35e2fSjeremylt     CeedInt num_comp, CeedInt P, CeedInt Q, CeedQuadMode quad_mode, CeedBasis *basis);
523d1d35e2fSjeremylt CEED_EXTERN int CeedBasisCreateTensorH1(Ceed ceed, CeedInt dim, CeedInt num_comp,
524d1d35e2fSjeremylt                                         CeedInt P_1d, CeedInt Q_1d,
525d1d35e2fSjeremylt                                         const CeedScalar *interp_1d,
526d1d35e2fSjeremylt                                         const CeedScalar *grad_1d,
527d1d35e2fSjeremylt                                         const CeedScalar *q_ref_1d,
528d1d35e2fSjeremylt                                         const CeedScalar *q_weight_1d,
529ec3da8bcSJed Brown                                         CeedBasis *basis);
530ec3da8bcSJed Brown CEED_EXTERN int CeedBasisCreateH1(Ceed ceed, CeedElemTopology topo,
531d1d35e2fSjeremylt                                   CeedInt num_comp,
532d1d35e2fSjeremylt                                   CeedInt num_nodes, CeedInt nqpts,
533ec3da8bcSJed Brown                                   const CeedScalar *interp,
534ec3da8bcSJed Brown                                   const CeedScalar *grad,
535d1d35e2fSjeremylt                                   const CeedScalar *q_ref,
536d1d35e2fSjeremylt                                   const CeedScalar *q_weights, CeedBasis *basis);
53750c301a5SRezgar Shakeri CEED_EXTERN int CeedBasisCreateHdiv(Ceed ceed, CeedElemTopology topo,
53850c301a5SRezgar Shakeri                                     CeedInt num_comp,
53950c301a5SRezgar Shakeri                                     CeedInt num_nodes, CeedInt nqpts,
54050c301a5SRezgar Shakeri                                     const CeedScalar *interp,
54150c301a5SRezgar Shakeri                                     const CeedScalar *div,
54250c301a5SRezgar Shakeri                                     const CeedScalar *q_ref,
54350c301a5SRezgar Shakeri                                     const CeedScalar *q_weights, CeedBasis *basis);
5449560d06aSjeremylt CEED_EXTERN int CeedBasisReferenceCopy(CeedBasis basis, CeedBasis *basis_copy);
545ec3da8bcSJed Brown CEED_EXTERN int CeedBasisView(CeedBasis basis, FILE *stream);
546d1d35e2fSjeremylt CEED_EXTERN int CeedBasisApply(CeedBasis basis, CeedInt num_elem,
547d1d35e2fSjeremylt                                CeedTransposeMode t_mode,
548d1d35e2fSjeremylt                                CeedEvalMode eval_mode, CeedVector u, CeedVector v);
549b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedBasisGetCeed(CeedBasis basis, Ceed *ceed);
550ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetDimension(CeedBasis basis, CeedInt *dim);
551ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetTopology(CeedBasis basis, CeedElemTopology *topo);
55250c301a5SRezgar Shakeri CEED_EXTERN int CeedBasisGetNumQuadratureComponents(CeedBasis basis, CeedInt *Q_comp);
553d1d35e2fSjeremylt CEED_EXTERN int CeedBasisGetNumComponents(CeedBasis basis, CeedInt *num_comp);
554ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetNumNodes(CeedBasis basis, CeedInt *P);
555d1d35e2fSjeremylt CEED_EXTERN int CeedBasisGetNumNodes1D(CeedBasis basis, CeedInt *P_1d);
556ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetNumQuadraturePoints(CeedBasis basis, CeedInt *Q);
557ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetNumQuadraturePoints1D(CeedBasis basis,
558d1d35e2fSjeremylt     CeedInt *Q_1d);
559d1d35e2fSjeremylt CEED_EXTERN int CeedBasisGetQRef(CeedBasis basis, const CeedScalar **q_ref);
560ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetQWeights(CeedBasis basis,
561d1d35e2fSjeremylt                                      const CeedScalar **q_weights);
562ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetInterp(CeedBasis basis, const CeedScalar **interp);
563ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetInterp1D(CeedBasis basis,
564d1d35e2fSjeremylt                                      const CeedScalar **interp_1d);
565ec3da8bcSJed Brown CEED_EXTERN int CeedBasisGetGrad(CeedBasis basis, const CeedScalar **grad);
566d1d35e2fSjeremylt CEED_EXTERN int CeedBasisGetGrad1D(CeedBasis basis, const CeedScalar **grad_1d);
56750c301a5SRezgar Shakeri CEED_EXTERN int CeedBasisGetDiv(CeedBasis basis, const CeedScalar **div);
568ec3da8bcSJed Brown CEED_EXTERN int CeedBasisDestroy(CeedBasis *basis);
569ec3da8bcSJed Brown 
570d1d35e2fSjeremylt CEED_EXTERN int CeedGaussQuadrature(CeedInt Q, CeedScalar *q_ref_1d,
571d1d35e2fSjeremylt                                     CeedScalar *q_weight_1d);
572d1d35e2fSjeremylt CEED_EXTERN int CeedLobattoQuadrature(CeedInt Q, CeedScalar *q_ref_1d,
573d1d35e2fSjeremylt                                       CeedScalar *q_weight_1d);
574ec3da8bcSJed Brown CEED_EXTERN int CeedQRFactorization(Ceed ceed, CeedScalar *mat, CeedScalar *tau,
575ec3da8bcSJed Brown                                     CeedInt m, CeedInt n);
576ec3da8bcSJed Brown CEED_EXTERN int CeedSymmetricSchurDecomposition(Ceed ceed, CeedScalar *mat,
577ec3da8bcSJed Brown     CeedScalar *lambda, CeedInt n);
578d1d35e2fSjeremylt CEED_EXTERN int CeedSimultaneousDiagonalization(Ceed ceed, CeedScalar *mat_A,
579d1d35e2fSjeremylt     CeedScalar *mat_B, CeedScalar *x, CeedScalar *lambda, CeedInt n);
580ec3da8bcSJed Brown 
581cdf32b93SJeremy L Thompson /** Handle for the user provided CeedQFunction callback function
582ec3da8bcSJed Brown 
583cdf32b93SJeremy L Thompson  @param[in,out] ctx  User-defined context set using CeedQFunctionSetContext() or NULL
584cdf32b93SJeremy L Thompson  @param[in] Q        Number of quadrature points at which to evaluate
585cdf32b93SJeremy L Thompson  @param[in] in       Array of pointers to each input argument in the order provided
586ec3da8bcSJed Brown                        by the user in CeedQFunctionAddInput().  Each array has shape
587d1d35e2fSjeremylt                        `[dim, num_comp, Q]` where `dim` is the geometric dimension for
588ec3da8bcSJed Brown                        \ref CEED_EVAL_GRAD (`dim=1` for \ref CEED_EVAL_INTERP) and
589d1d35e2fSjeremylt                        `num_comp` is the number of field components (`num_comp=1` for
590ec3da8bcSJed Brown                        scalar fields).  This results in indexing the `i`th input at
591d1d35e2fSjeremylt                        quadrature point `j` as `in[i][(d*num_comp + c)*Q + j]`.
592cdf32b93SJeremy L Thompson  @param[out]   out   Array of pointers to each output array in the order provided
593ec3da8bcSJed Brown                        using CeedQFunctionAddOutput().  The shapes are as above for
594ec3da8bcSJed Brown                        \a in.
595ec3da8bcSJed Brown 
596ec3da8bcSJed Brown  @return An error code: 0 - success, otherwise - failure
597ec3da8bcSJed Brown 
598ec3da8bcSJed Brown  @ingroup CeedQFunction
599ec3da8bcSJed Brown **/
600ec3da8bcSJed Brown typedef int (*CeedQFunctionUser)(void *ctx, const CeedInt Q,
601ec3da8bcSJed Brown                                  const CeedScalar *const *in,
602ec3da8bcSJed Brown                                  CeedScalar *const *out);
603ec3da8bcSJed Brown 
604d1d35e2fSjeremylt CEED_EXTERN int CeedQFunctionCreateInterior(Ceed ceed, CeedInt vec_length,
605ec3da8bcSJed Brown     CeedQFunctionUser f, const char *source, CeedQFunction *qf);
606ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionCreateInteriorByName(Ceed ceed, const char *name,
607ec3da8bcSJed Brown     CeedQFunction *qf);
608ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionCreateIdentity(Ceed ceed, CeedInt size,
609d1d35e2fSjeremylt     CeedEvalMode in_mode, CeedEvalMode out_mode, CeedQFunction *qf);
6109560d06aSjeremylt CEED_EXTERN int CeedQFunctionReferenceCopy(CeedQFunction qf, CeedQFunction *qf_copy);
611d1d35e2fSjeremylt CEED_EXTERN int CeedQFunctionAddInput(CeedQFunction qf, const char *field_name,
612d1d35e2fSjeremylt                                       CeedInt size, CeedEvalMode eval_mode);
613d1d35e2fSjeremylt CEED_EXTERN int CeedQFunctionAddOutput(CeedQFunction qf, const char *field_name,
614d1d35e2fSjeremylt                                        CeedInt size, CeedEvalMode eval_mode);
61543bbe138SJeremy L Thompson CEED_EXTERN int CeedQFunctionGetFields(CeedQFunction qf,
61643bbe138SJeremy L Thompson                                        CeedInt *num_input_fields,
61743bbe138SJeremy L Thompson                                        CeedQFunctionField **input_fields,
61843bbe138SJeremy L Thompson                                        CeedInt *num_output_fields,
61943bbe138SJeremy L Thompson                                        CeedQFunctionField **output_fields);
620ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionSetContext(CeedQFunction qf,
621ec3da8bcSJed Brown                                         CeedQFunctionContext ctx);
622441428dfSJeremy L Thompson CEED_EXTERN int CeedQFunctionSetContextWritable(CeedQFunction qf, bool is_writable);
623ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionView(CeedQFunction qf, FILE *stream);
624b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedQFunctionGetCeed(CeedQFunction qf, Ceed *ceed);
625ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionApply(CeedQFunction qf, CeedInt Q,
626ec3da8bcSJed Brown                                    CeedVector *u, CeedVector *v);
627ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionDestroy(CeedQFunction *qf);
628ec3da8bcSJed Brown 
62943bbe138SJeremy L Thompson CEED_EXTERN int CeedQFunctionFieldGetName(CeedQFunctionField qf_field,
63043bbe138SJeremy L Thompson     char **field_name);
63143bbe138SJeremy L Thompson CEED_EXTERN int CeedQFunctionFieldGetSize(CeedQFunctionField qf_field,
63243bbe138SJeremy L Thompson     CeedInt *size);
63343bbe138SJeremy L Thompson CEED_EXTERN int CeedQFunctionFieldGetEvalMode(CeedQFunctionField qf_field,
63443bbe138SJeremy L Thompson     CeedEvalMode *eval_mode);
63543bbe138SJeremy L Thompson 
636cdf32b93SJeremy L Thompson /// Denotes type of data stored in a CeedQFunctionContext field
637cdf32b93SJeremy L Thompson /// @ingroup CeedQFunction
638cdf32b93SJeremy L Thompson typedef enum {
639cdf32b93SJeremy L Thompson   /// Double precision value
6407bfe0f0eSJeremy L Thompson   CEED_CONTEXT_FIELD_DOUBLE = 1,
641cdf32b93SJeremy L Thompson   /// 32 bit integer value
6427bfe0f0eSJeremy L Thompson   CEED_CONTEXT_FIELD_INT32  = 2,
643cdf32b93SJeremy L Thompson } CeedContextFieldType;
644cdf32b93SJeremy L Thompson CEED_EXTERN const char *const CeedContextFieldTypes[];
645cdf32b93SJeremy L Thompson 
646ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextCreate(Ceed ceed,
647ec3da8bcSJed Brown     CeedQFunctionContext *ctx);
6489560d06aSjeremylt CEED_EXTERN int CeedQFunctionContextReferenceCopy(CeedQFunctionContext ctx,
6499560d06aSjeremylt     CeedQFunctionContext *ctx_copy);
650ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextSetData(CeedQFunctionContext ctx,
651d1d35e2fSjeremylt     CeedMemType mem_type, CeedCopyMode copy_mode, size_t size, void *data);
652891038deSjeremylt CEED_EXTERN int CeedQFunctionContextTakeData(CeedQFunctionContext ctx,
653891038deSjeremylt     CeedMemType mem_type, void *data);
654ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextGetData(CeedQFunctionContext ctx,
655891038deSjeremylt     CeedMemType mem_type, void *data);
65628bfd0b7SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextGetDataRead(CeedQFunctionContext ctx,
65728bfd0b7SJeremy L Thompson     CeedMemType mem_type, void *data);
658ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextRestoreData(CeedQFunctionContext ctx,
659ec3da8bcSJed Brown     void *data);
66028bfd0b7SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextRestoreDataRead(CeedQFunctionContext ctx,
66128bfd0b7SJeremy L Thompson     void *data);
662cdf32b93SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextRegisterDouble(CeedQFunctionContext ctx,
6637bfe0f0eSJeremy L Thompson     const char *field_name, size_t field_offset, size_t num_values,
6647bfe0f0eSJeremy L Thompson     const char *field_description);
665cdf32b93SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextRegisterInt32(CeedQFunctionContext ctx,
6667bfe0f0eSJeremy L Thompson     const char *field_name, size_t field_offset, size_t num_values,
6677bfe0f0eSJeremy L Thompson     const char *field_description);
6683668ca4bSJeremy L Thompson CEED_EXTERN int CeedQFunctionContextGetFieldLabel(CeedQFunctionContext ctx,
6693668ca4bSJeremy L Thompson     const char *field_name, CeedContextFieldLabel *field_label);
6703668ca4bSJeremy L Thompson CEED_EXTERN int CeedQFunctionContextGetAllFieldLabels(CeedQFunctionContext ctx,
6713668ca4bSJeremy L Thompson     const CeedContextFieldLabel **field_labels, CeedInt *num_fields);
6720f86cbe7SJeremy L Thompson CEED_EXTERN int CeedContextFieldLabelGetDescription(CeedContextFieldLabel label,
6737bfe0f0eSJeremy L Thompson     const char **field_name, const char **field_description, size_t *num_values,
6740f86cbe7SJeremy L Thompson     CeedContextFieldType *field_type);
675cdf32b93SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextSetDouble(CeedQFunctionContext ctx,
6767bfe0f0eSJeremy L Thompson     CeedContextFieldLabel field_label, double *values);
677cdf32b93SJeremy L Thompson CEED_EXTERN int CeedQFunctionContextSetInt32(CeedQFunctionContext ctx,
6787bfe0f0eSJeremy L Thompson     CeedContextFieldLabel field_label, int *values);
67980a9ef05SNatalie Beams CEED_EXTERN int CeedQFunctionContextGetContextSize(CeedQFunctionContext ctx,
68080a9ef05SNatalie Beams     size_t *ctx_size);
681ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextView(CeedQFunctionContext ctx,
682ec3da8bcSJed Brown     FILE *stream);
683ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionContextDestroy(CeedQFunctionContext *ctx);
684ec3da8bcSJed Brown 
685ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorCreate(Ceed ceed, CeedQFunction qf,
686ec3da8bcSJed Brown                                    CeedQFunction dqf, CeedQFunction dqfT,
687ec3da8bcSJed Brown                                    CeedOperator *op);
688ec3da8bcSJed Brown CEED_EXTERN int CeedCompositeOperatorCreate(Ceed ceed, CeedOperator *op);
6899560d06aSjeremylt CEED_EXTERN int CeedOperatorReferenceCopy(CeedOperator op, CeedOperator *op_copy);
690d1d35e2fSjeremylt CEED_EXTERN int CeedOperatorSetField(CeedOperator op, const char *field_name,
691ec3da8bcSJed Brown                                      CeedElemRestriction r, CeedBasis b,
692ec3da8bcSJed Brown                                      CeedVector v);
69343bbe138SJeremy L Thompson CEED_EXTERN int CeedOperatorGetFields(CeedOperator op,
69443bbe138SJeremy L Thompson                                       CeedInt *num_input_fields,
69543bbe138SJeremy L Thompson                                       CeedOperatorField **input_fields,
69643bbe138SJeremy L Thompson                                       CeedInt *num_output_fields,
69743bbe138SJeremy L Thompson                                       CeedOperatorField **output_fields);
698d1d35e2fSjeremylt CEED_EXTERN int CeedCompositeOperatorAddSub(CeedOperator composite_op,
699d1d35e2fSjeremylt     CeedOperator sub_op);
7004db537f9SJeremy L Thompson CEED_EXTERN int CeedOperatorCheckReady(CeedOperator op);
701ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleQFunction(CeedOperator op,
702ec3da8bcSJed Brown     CeedVector *assembled, CeedElemRestriction *rstr, CeedRequest *request);
70370a7ffb3SJeremy L Thompson CEED_EXTERN int CeedOperatorLinearAssembleQFunctionBuildOrUpdate(CeedOperator op,
70470a7ffb3SJeremy L Thompson     CeedVector *assembled, CeedElemRestriction *rstr, CeedRequest *request);
705ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleDiagonal(CeedOperator op,
706ec3da8bcSJed Brown     CeedVector assembled, CeedRequest *request);
707ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleAddDiagonal(CeedOperator op,
708ec3da8bcSJed Brown     CeedVector assembled, CeedRequest *request);
709ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssemblePointBlockDiagonal(CeedOperator op,
710ec3da8bcSJed Brown     CeedVector assembled, CeedRequest *request);
711ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleAddPointBlockDiagonal(CeedOperator op,
712ec3da8bcSJed Brown     CeedVector assembled, CeedRequest *request);
713ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssembleSymbolic(CeedOperator op,
714*3047f789SJeremy L Thompson      ptrdiff_t *num_entries, CeedInt **rows, CeedInt **cols);
715ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorLinearAssemble(CeedOperator op, CeedVector values);
716d1d35e2fSjeremylt CEED_EXTERN int CeedOperatorMultigridLevelCreate(CeedOperator op_fine,
717d1d35e2fSjeremylt     CeedVector p_mult_fine, CeedElemRestriction rstr_coarse, CeedBasis basis_coarse,
718d1d35e2fSjeremylt     CeedOperator *op_coarse, CeedOperator *op_prolong, CeedOperator *op_restrict);
719ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorMultigridLevelCreateTensorH1(
720d1d35e2fSjeremylt   CeedOperator op_fine, CeedVector p_mult_fine, CeedElemRestriction rstr_coarse,
721d1d35e2fSjeremylt   CeedBasis basis_coarse, const CeedScalar *interp_c_to_f, CeedOperator *op_coarse,
722d1d35e2fSjeremylt   CeedOperator *op_prolong, CeedOperator *op_restrict);
723d1d35e2fSjeremylt CEED_EXTERN int CeedOperatorMultigridLevelCreateH1(CeedOperator op_fine,
724d1d35e2fSjeremylt     CeedVector p_mult_fine, CeedElemRestriction rstr_coarse, CeedBasis basis_coarse,
725d1d35e2fSjeremylt     const CeedScalar *interp_c_to_f, CeedOperator *op_coarse,
726d1d35e2fSjeremylt     CeedOperator *op_prolong, CeedOperator *op_restrict);
727ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorCreateFDMElementInverse(CeedOperator op,
728d1d35e2fSjeremylt     CeedOperator *fdm_inv, CeedRequest *request);
729cd4dfc48Sjeremylt CEED_EXTERN int CeedOperatorSetNumQuadraturePoints(CeedOperator op, CeedInt num_qpts);
730ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorView(CeedOperator op, FILE *stream);
731b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedOperatorGetCeed(CeedOperator op, Ceed *ceed);
732b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedOperatorGetNumElements(CeedOperator op, CeedInt *num_elem);
733b7c9bbdaSJeremy L Thompson CEED_EXTERN int CeedOperatorGetNumQuadraturePoints(CeedOperator op,
734b7c9bbdaSJeremy L Thompson     CeedInt *num_qpts);
7353668ca4bSJeremy L Thompson CEED_EXTERN int CeedOperatorContextGetFieldLabel(CeedOperator op,
7363668ca4bSJeremy L Thompson     const char *field_name, CeedContextFieldLabel *field_label);
7373668ca4bSJeremy L Thompson CEED_EXTERN int CeedOperatorContextSetDouble(CeedOperator op,
7387bfe0f0eSJeremy L Thompson     CeedContextFieldLabel field_label, double *values);
7393668ca4bSJeremy L Thompson CEED_EXTERN int CeedOperatorContextSetInt32(CeedOperator op,
7407bfe0f0eSJeremy L Thompson     CeedContextFieldLabel field_label, int *values);
741ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorApply(CeedOperator op, CeedVector in,
742ec3da8bcSJed Brown                                   CeedVector out, CeedRequest *request);
743ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorApplyAdd(CeedOperator op, CeedVector in,
744ec3da8bcSJed Brown                                      CeedVector out, CeedRequest *request);
745ec3da8bcSJed Brown CEED_EXTERN int CeedOperatorDestroy(CeedOperator *op);
746ec3da8bcSJed Brown 
74728567f8fSJeremy L Thompson CEED_EXTERN int CeedOperatorFieldGetName(CeedOperatorField op_field,
74828567f8fSJeremy L Thompson     char **field_name);
74943bbe138SJeremy L Thompson CEED_EXTERN int CeedOperatorFieldGetElemRestriction(CeedOperatorField op_field,
75043bbe138SJeremy L Thompson     CeedElemRestriction *rstr);
75143bbe138SJeremy L Thompson CEED_EXTERN int CeedOperatorFieldGetBasis(CeedOperatorField op_field,
75243bbe138SJeremy L Thompson     CeedBasis *basis);
75343bbe138SJeremy L Thompson CEED_EXTERN int CeedOperatorFieldGetVector(CeedOperatorField op_field,
75443bbe138SJeremy L Thompson     CeedVector *vec);
75543bbe138SJeremy L Thompson 
756ec3da8bcSJed Brown /**
757ec3da8bcSJed Brown   @brief Return integer power
758ec3da8bcSJed Brown 
759ec3da8bcSJed Brown   @param[in] base   The base to exponentiate
760ec3da8bcSJed Brown   @param[in] power  The power to raise the base to
761ec3da8bcSJed Brown 
762ec3da8bcSJed Brown   @return base^power
763ec3da8bcSJed Brown 
764ec3da8bcSJed Brown   @ref Utility
765ec3da8bcSJed Brown **/
766ec3da8bcSJed Brown static inline CeedInt CeedIntPow(CeedInt base, CeedInt power) {
767ec3da8bcSJed Brown   CeedInt result = 1;
768ec3da8bcSJed Brown   while (power) {
769ec3da8bcSJed Brown     if (power & 1) result *= base;
770ec3da8bcSJed Brown     power >>= 1;
771ec3da8bcSJed Brown     base *= base;
772ec3da8bcSJed Brown   }
773ec3da8bcSJed Brown   return result;
774ec3da8bcSJed Brown }
775ec3da8bcSJed Brown 
776ec3da8bcSJed Brown /**
777ec3da8bcSJed Brown   @brief Return minimum of two integers
778ec3da8bcSJed Brown 
779ec3da8bcSJed Brown   @param[in] a  The first integer to compare
780ec3da8bcSJed Brown   @param[in] b  The second integer to compare
781ec3da8bcSJed Brown 
782ec3da8bcSJed Brown   @return The minimum of the two integers
783ec3da8bcSJed Brown 
784ec3da8bcSJed Brown   @ref Utility
785ec3da8bcSJed Brown **/
786ec3da8bcSJed Brown static inline CeedInt CeedIntMin(CeedInt a, CeedInt b) { return a < b ? a : b; }
787ec3da8bcSJed Brown 
788ec3da8bcSJed Brown /**
789ec3da8bcSJed Brown   @brief Return maximum of two integers
790ec3da8bcSJed Brown 
791ec3da8bcSJed Brown   @param[in] a  The first integer to compare
792ec3da8bcSJed Brown   @param[in] b  The second integer to compare
793ec3da8bcSJed Brown 
794ec3da8bcSJed Brown   @return The maximum of the two integers
795ec3da8bcSJed Brown 
796ec3da8bcSJed Brown   @ref Utility
797ec3da8bcSJed Brown **/
798ec3da8bcSJed Brown static inline CeedInt CeedIntMax(CeedInt a, CeedInt b) { return a > b ? a : b; }
799ec3da8bcSJed Brown 
800ec3da8bcSJed Brown // Used to ensure initialization before CeedInit()
801ec3da8bcSJed Brown CEED_EXTERN int CeedRegisterAll(void);
802ec3da8bcSJed Brown // Used to ensure initialization before CeedQFunctionCreate*()
803ec3da8bcSJed Brown CEED_EXTERN int CeedQFunctionRegisterAll(void);
804ec3da8bcSJed Brown 
805ec3da8bcSJed Brown #endif
806