xref: /libCEED/rust/libceed-sys/c-src/interface/ceed-basis.c (revision 446e7af4d6e7c58f469d4618e9b5e398a4084523)
13d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3d7b241e6Sjeremylt //
43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
5d7b241e6Sjeremylt //
63d8e8822SJeremy L Thompson // This file is part of CEED:  http://github.com/ceed
7d7b241e6Sjeremylt 
8ec3da8bcSJed Brown #include <ceed/ceed.h>
9ec3da8bcSJed Brown #include <ceed/backend.h>
103d576824SJeremy L Thompson #include <ceed-impl.h>
11d7b241e6Sjeremylt #include <math.h>
123d576824SJeremy L Thompson #include <stdbool.h>
13d7b241e6Sjeremylt #include <stdio.h>
14d7b241e6Sjeremylt #include <string.h>
15d7b241e6Sjeremylt 
167a982d89SJeremy L. Thompson /// @file
177a982d89SJeremy L. Thompson /// Implementation of CeedBasis interfaces
187a982d89SJeremy L. Thompson 
19d7b241e6Sjeremylt /// @cond DOXYGEN_SKIP
20783c99b3SValeria Barra static struct CeedBasis_private ceed_basis_collocated;
21d7b241e6Sjeremylt /// @endcond
22d7b241e6Sjeremylt 
237a982d89SJeremy L. Thompson /// @addtogroup CeedBasisUser
247a982d89SJeremy L. Thompson /// @{
257a982d89SJeremy L. Thompson 
267a982d89SJeremy L. Thompson /// Indicate that the quadrature points are collocated with the nodes
277a982d89SJeremy L. Thompson const CeedBasis CEED_BASIS_COLLOCATED = &ceed_basis_collocated;
287a982d89SJeremy L. Thompson 
297a982d89SJeremy L. Thompson /// @}
307a982d89SJeremy L. Thompson 
317a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
327a982d89SJeremy L. Thompson /// CeedBasis Library Internal Functions
337a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
347a982d89SJeremy L. Thompson /// @addtogroup CeedBasisDeveloper
357a982d89SJeremy L. Thompson /// @{
367a982d89SJeremy L. Thompson 
377a982d89SJeremy L. Thompson /**
387a982d89SJeremy L. Thompson   @brief Compute Householder reflection
397a982d89SJeremy L. Thompson 
407a982d89SJeremy L. Thompson     Computes A = (I - b v v^T) A
417a982d89SJeremy L. Thompson     where A is an mxn matrix indexed as A[i*row + j*col]
427a982d89SJeremy L. Thompson 
437a982d89SJeremy L. Thompson   @param[in,out] A  Matrix to apply Householder reflection to, in place
447a982d89SJeremy L. Thompson   @param v          Householder vector
457a982d89SJeremy L. Thompson   @param b          Scaling factor
467a982d89SJeremy L. Thompson   @param m          Number of rows in A
477a982d89SJeremy L. Thompson   @param n          Number of columns in A
487a982d89SJeremy L. Thompson   @param row        Row stride
497a982d89SJeremy L. Thompson   @param col        Col stride
507a982d89SJeremy L. Thompson 
517a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
527a982d89SJeremy L. Thompson 
537a982d89SJeremy L. Thompson   @ref Developer
547a982d89SJeremy L. Thompson **/
557a982d89SJeremy L. Thompson static int CeedHouseholderReflect(CeedScalar *A, const CeedScalar *v,
567a982d89SJeremy L. Thompson                                   CeedScalar b, CeedInt m, CeedInt n,
577a982d89SJeremy L. Thompson                                   CeedInt row, CeedInt col) {
587a982d89SJeremy L. Thompson   for (CeedInt j=0; j<n; j++) {
597a982d89SJeremy L. Thompson     CeedScalar w = A[0*row + j*col];
607a982d89SJeremy L. Thompson     for (CeedInt i=1; i<m; i++)
617a982d89SJeremy L. Thompson       w += v[i] * A[i*row + j*col];
627a982d89SJeremy L. Thompson     A[0*row + j*col] -= b * w;
637a982d89SJeremy L. Thompson     for (CeedInt i=1; i<m; i++)
647a982d89SJeremy L. Thompson       A[i*row + j*col] -= b * w * v[i];
657a982d89SJeremy L. Thompson   }
66e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
677a982d89SJeremy L. Thompson }
687a982d89SJeremy L. Thompson 
697a982d89SJeremy L. Thompson /**
707a982d89SJeremy L. Thompson   @brief Apply Householder Q matrix
717a982d89SJeremy L. Thompson 
727a982d89SJeremy L. Thompson     Compute A = Q A where Q is mxm and A is mxn.
737a982d89SJeremy L. Thompson 
747a982d89SJeremy L. Thompson   @param[in,out] A  Matrix to apply Householder Q to, in place
757a982d89SJeremy L. Thompson   @param Q          Householder Q matrix
767a982d89SJeremy L. Thompson   @param tau        Householder scaling factors
77d1d35e2fSjeremylt   @param t_mode     Transpose mode for application
787a982d89SJeremy L. Thompson   @param m          Number of rows in A
797a982d89SJeremy L. Thompson   @param n          Number of columns in A
807a982d89SJeremy L. Thompson   @param k          Number of elementary reflectors in Q, k<m
817a982d89SJeremy L. Thompson   @param row        Row stride in A
827a982d89SJeremy L. Thompson   @param col        Col stride in A
837a982d89SJeremy L. Thompson 
847a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
857a982d89SJeremy L. Thompson 
867a982d89SJeremy L. Thompson   @ref Developer
877a982d89SJeremy L. Thompson **/
88d99fa3c5SJeremy L Thompson int CeedHouseholderApplyQ(CeedScalar *A, const CeedScalar *Q,
89d1d35e2fSjeremylt                           const CeedScalar *tau, CeedTransposeMode t_mode,
907a982d89SJeremy L. Thompson                           CeedInt m, CeedInt n, CeedInt k,
917a982d89SJeremy L. Thompson                           CeedInt row, CeedInt col) {
92e15f9bd0SJeremy L Thompson   int ierr;
9378464608Sjeremylt   CeedScalar *v;
9478464608Sjeremylt   ierr = CeedMalloc(m, &v); CeedChk(ierr);
957a982d89SJeremy L. Thompson   for (CeedInt ii=0; ii<k; ii++) {
96d1d35e2fSjeremylt     CeedInt i = t_mode == CEED_TRANSPOSE ? ii : k-1-ii;
977a982d89SJeremy L. Thompson     for (CeedInt j=i+1; j<m; j++)
987a982d89SJeremy L. Thompson       v[j] = Q[j*k+i];
99d1d35e2fSjeremylt     // Apply Householder reflector (I - tau v v^T) collo_grad_1d^T
100e15f9bd0SJeremy L Thompson     ierr = CeedHouseholderReflect(&A[i*row], &v[i], tau[i], m-i, n, row, col);
101e15f9bd0SJeremy L Thompson     CeedChk(ierr);
1027a982d89SJeremy L. Thompson   }
10378464608Sjeremylt   ierr = CeedFree(&v); CeedChk(ierr);
104e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1057a982d89SJeremy L. Thompson }
1067a982d89SJeremy L. Thompson 
1077a982d89SJeremy L. Thompson /**
1087a982d89SJeremy L. Thompson   @brief Compute Givens rotation
1097a982d89SJeremy L. Thompson 
1107a982d89SJeremy L. Thompson     Computes A = G A (or G^T A in transpose mode)
1117a982d89SJeremy L. Thompson     where A is an mxn matrix indexed as A[i*n + j*m]
1127a982d89SJeremy L. Thompson 
1137a982d89SJeremy L. Thompson   @param[in,out] A  Row major matrix to apply Givens rotation to, in place
1147a982d89SJeremy L. Thompson   @param c          Cosine factor
1157a982d89SJeremy L. Thompson   @param s          Sine factor
116d1d35e2fSjeremylt   @param t_mode     @ref CEED_NOTRANSPOSE to rotate the basis counter-clockwise,
1174c4400c7SValeria Barra                     which has the effect of rotating columns of A clockwise;
1184cc79fe7SJed Brown                     @ref CEED_TRANSPOSE for the opposite rotation
1197a982d89SJeremy L. Thompson   @param i          First row/column to apply rotation
1207a982d89SJeremy L. Thompson   @param k          Second row/column to apply rotation
1217a982d89SJeremy L. Thompson   @param m          Number of rows in A
1227a982d89SJeremy L. Thompson   @param n          Number of columns in A
1237a982d89SJeremy L. Thompson 
1247a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
1257a982d89SJeremy L. Thompson 
1267a982d89SJeremy L. Thompson   @ref Developer
1277a982d89SJeremy L. Thompson **/
1287a982d89SJeremy L. Thompson static int CeedGivensRotation(CeedScalar *A, CeedScalar c, CeedScalar s,
129d1d35e2fSjeremylt                               CeedTransposeMode t_mode, CeedInt i, CeedInt k,
1307a982d89SJeremy L. Thompson                               CeedInt m, CeedInt n) {
131d1d35e2fSjeremylt   CeedInt stride_j = 1, stride_ik = m, num_its = n;
132d1d35e2fSjeremylt   if (t_mode == CEED_NOTRANSPOSE) {
133d1d35e2fSjeremylt     stride_j = n; stride_ik = 1; num_its = m;
1347a982d89SJeremy L. Thompson   }
1357a982d89SJeremy L. Thompson 
1367a982d89SJeremy L. Thompson   // Apply rotation
137d1d35e2fSjeremylt   for (CeedInt j=0; j<num_its; j++) {
138d1d35e2fSjeremylt     CeedScalar tau1 = A[i*stride_ik+j*stride_j], tau2 = A[k*stride_ik+j*stride_j];
139d1d35e2fSjeremylt     A[i*stride_ik+j*stride_j] = c*tau1 - s*tau2;
140d1d35e2fSjeremylt     A[k*stride_ik+j*stride_j] = s*tau1 + c*tau2;
1417a982d89SJeremy L. Thompson   }
142e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1437a982d89SJeremy L. Thompson }
1447a982d89SJeremy L. Thompson 
1457a982d89SJeremy L. Thompson /**
1467a982d89SJeremy L. Thompson   @brief View an array stored in a CeedBasis
1477a982d89SJeremy L. Thompson 
1480a0da059Sjeremylt   @param[in] name      Name of array
149d1d35e2fSjeremylt   @param[in] fp_fmt    Printing format
1500a0da059Sjeremylt   @param[in] m         Number of rows in array
1510a0da059Sjeremylt   @param[in] n         Number of columns in array
1520a0da059Sjeremylt   @param[in] a         Array to be viewed
1530a0da059Sjeremylt   @param[in] stream    Stream to view to, e.g., stdout
1547a982d89SJeremy L. Thompson 
1557a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
1567a982d89SJeremy L. Thompson 
1577a982d89SJeremy L. Thompson   @ref Developer
1587a982d89SJeremy L. Thompson **/
159d1d35e2fSjeremylt static int CeedScalarView(const char *name, const char *fp_fmt, CeedInt m,
1607a982d89SJeremy L. Thompson                           CeedInt n, const CeedScalar *a, FILE *stream) {
16192ae7e47SJeremy L Thompson   for (CeedInt i=0; i<m; i++) {
1627a982d89SJeremy L. Thompson     if (m > 1)
1637a982d89SJeremy L. Thompson       fprintf(stream, "%12s[%d]:", name, i);
1647a982d89SJeremy L. Thompson     else
1657a982d89SJeremy L. Thompson       fprintf(stream, "%12s:", name);
16692ae7e47SJeremy L Thompson     for (CeedInt j=0; j<n; j++)
167d1d35e2fSjeremylt       fprintf(stream, fp_fmt, fabs(a[i*n+j]) > 1E-14 ? a[i*n+j] : 0);
1687a982d89SJeremy L. Thompson     fputs("\n", stream);
1697a982d89SJeremy L. Thompson   }
170e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1717a982d89SJeremy L. Thompson }
1727a982d89SJeremy L. Thompson 
1737a982d89SJeremy L. Thompson /// @}
1747a982d89SJeremy L. Thompson 
1757a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
1767a982d89SJeremy L. Thompson /// Ceed Backend API
1777a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
1787a982d89SJeremy L. Thompson /// @addtogroup CeedBasisBackend
1797a982d89SJeremy L. Thompson /// @{
1807a982d89SJeremy L. Thompson 
1817a982d89SJeremy L. Thompson /**
1827a982d89SJeremy L. Thompson   @brief Return collocated grad matrix
1837a982d89SJeremy L. Thompson 
1847a982d89SJeremy L. Thompson   @param basis               CeedBasis
185d1d35e2fSjeremylt   @param[out] collo_grad_1d  Row-major (Q_1d * Q_1d) matrix expressing derivatives of
1867a982d89SJeremy L. Thompson                                basis functions at quadrature points
1877a982d89SJeremy L. Thompson 
1887a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
1897a982d89SJeremy L. Thompson 
1907a982d89SJeremy L. Thompson   @ref Backend
1917a982d89SJeremy L. Thompson **/
192d1d35e2fSjeremylt int CeedBasisGetCollocatedGrad(CeedBasis basis, CeedScalar *collo_grad_1d) {
1937a982d89SJeremy L. Thompson   int i, j, k;
1947a982d89SJeremy L. Thompson   Ceed ceed;
195d1d35e2fSjeremylt   CeedInt ierr, P_1d=(basis)->P_1d, Q_1d=(basis)->Q_1d;
19678464608Sjeremylt   CeedScalar *interp_1d, *grad_1d, *tau;
1977a982d89SJeremy L. Thompson 
198d1d35e2fSjeremylt   ierr = CeedMalloc(Q_1d*P_1d, &interp_1d); CeedChk(ierr);
199d1d35e2fSjeremylt   ierr = CeedMalloc(Q_1d*P_1d, &grad_1d); CeedChk(ierr);
20078464608Sjeremylt   ierr = CeedMalloc(Q_1d, &tau); CeedChk(ierr);
201d1d35e2fSjeremylt   memcpy(interp_1d, (basis)->interp_1d, Q_1d*P_1d*sizeof(basis)->interp_1d[0]);
202d1d35e2fSjeremylt   memcpy(grad_1d, (basis)->grad_1d, Q_1d*P_1d*sizeof(basis)->interp_1d[0]);
2037a982d89SJeremy L. Thompson 
204d1d35e2fSjeremylt   // QR Factorization, interp_1d = Q R
2057a982d89SJeremy L. Thompson   ierr = CeedBasisGetCeed(basis, &ceed); CeedChk(ierr);
206d1d35e2fSjeremylt   ierr = CeedQRFactorization(ceed, interp_1d, tau, Q_1d, P_1d); CeedChk(ierr);
207e15f9bd0SJeremy L Thompson   // Note: This function is for backend use, so all errors are terminal
208e15f9bd0SJeremy L Thompson   //   and we do not need to clean up memory on failure.
2097a982d89SJeremy L. Thompson 
210d1d35e2fSjeremylt   // Apply Rinv, collo_grad_1d = grad_1d Rinv
211d1d35e2fSjeremylt   for (i=0; i<Q_1d; i++) { // Row i
212d1d35e2fSjeremylt     collo_grad_1d[Q_1d*i] = grad_1d[P_1d*i]/interp_1d[0];
213d1d35e2fSjeremylt     for (j=1; j<P_1d; j++) { // Column j
214d1d35e2fSjeremylt       collo_grad_1d[j+Q_1d*i] = grad_1d[j+P_1d*i];
2157a982d89SJeremy L. Thompson       for (k=0; k<j; k++)
216d1d35e2fSjeremylt         collo_grad_1d[j+Q_1d*i] -= interp_1d[j+P_1d*k]*collo_grad_1d[k+Q_1d*i];
217d1d35e2fSjeremylt       collo_grad_1d[j+Q_1d*i] /= interp_1d[j+P_1d*j];
2187a982d89SJeremy L. Thompson     }
219d1d35e2fSjeremylt     for (j=P_1d; j<Q_1d; j++)
220d1d35e2fSjeremylt       collo_grad_1d[j+Q_1d*i] = 0;
2217a982d89SJeremy L. Thompson   }
2227a982d89SJeremy L. Thompson 
223673160d7Sjeremylt   // Apply Qtranspose, collo_grad = collo_grad Q_transpose
224d1d35e2fSjeremylt   ierr = CeedHouseholderApplyQ(collo_grad_1d, interp_1d, tau, CEED_NOTRANSPOSE,
225d1d35e2fSjeremylt                                Q_1d, Q_1d, P_1d, 1, Q_1d); CeedChk(ierr);
2267a982d89SJeremy L. Thompson 
227d1d35e2fSjeremylt   ierr = CeedFree(&interp_1d); CeedChk(ierr);
228d1d35e2fSjeremylt   ierr = CeedFree(&grad_1d); CeedChk(ierr);
22978464608Sjeremylt   ierr = CeedFree(&tau); CeedChk(ierr);
230e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2317a982d89SJeremy L. Thompson }
2327a982d89SJeremy L. Thompson 
2337a982d89SJeremy L. Thompson /**
2347a982d89SJeremy L. Thompson   @brief Get tensor status for given CeedBasis
2357a982d89SJeremy L. Thompson 
2367a982d89SJeremy L. Thompson   @param basis           CeedBasis
237d1d35e2fSjeremylt   @param[out] is_tensor  Variable to store tensor status
2387a982d89SJeremy L. Thompson 
2397a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
2407a982d89SJeremy L. Thompson 
2417a982d89SJeremy L. Thompson   @ref Backend
2427a982d89SJeremy L. Thompson **/
243d1d35e2fSjeremylt int CeedBasisIsTensor(CeedBasis basis, bool *is_tensor) {
244d1d35e2fSjeremylt   *is_tensor = basis->tensor_basis;
245e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2467a982d89SJeremy L. Thompson }
2477a982d89SJeremy L. Thompson 
2487a982d89SJeremy L. Thompson /**
2497a982d89SJeremy L. Thompson   @brief Get backend data of a CeedBasis
2507a982d89SJeremy L. Thompson 
2517a982d89SJeremy L. Thompson   @param basis      CeedBasis
2527a982d89SJeremy L. Thompson   @param[out] data  Variable to store data
2537a982d89SJeremy L. Thompson 
2547a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
2557a982d89SJeremy L. Thompson 
2567a982d89SJeremy L. Thompson   @ref Backend
2577a982d89SJeremy L. Thompson **/
258777ff853SJeremy L Thompson int CeedBasisGetData(CeedBasis basis, void *data) {
259777ff853SJeremy L Thompson   *(void **)data = basis->data;
260e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2617a982d89SJeremy L. Thompson }
2627a982d89SJeremy L. Thompson 
2637a982d89SJeremy L. Thompson /**
2647a982d89SJeremy L. Thompson   @brief Set backend data of a CeedBasis
2657a982d89SJeremy L. Thompson 
2667a982d89SJeremy L. Thompson   @param[out] basis  CeedBasis
2677a982d89SJeremy L. Thompson   @param data        Data to set
2687a982d89SJeremy L. Thompson 
2697a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
2707a982d89SJeremy L. Thompson 
2717a982d89SJeremy L. Thompson   @ref Backend
2727a982d89SJeremy L. Thompson **/
273777ff853SJeremy L Thompson int CeedBasisSetData(CeedBasis basis, void *data) {
274777ff853SJeremy L Thompson   basis->data = data;
275e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2767a982d89SJeremy L. Thompson }
2777a982d89SJeremy L. Thompson 
2787a982d89SJeremy L. Thompson /**
27934359f16Sjeremylt   @brief Increment the reference counter for a CeedBasis
28034359f16Sjeremylt 
28134359f16Sjeremylt   @param basis  Basis to increment the reference counter
28234359f16Sjeremylt 
28334359f16Sjeremylt   @return An error code: 0 - success, otherwise - failure
28434359f16Sjeremylt 
28534359f16Sjeremylt   @ref Backend
28634359f16Sjeremylt **/
2879560d06aSjeremylt int CeedBasisReference(CeedBasis basis) {
28834359f16Sjeremylt   basis->ref_count++;
28934359f16Sjeremylt   return CEED_ERROR_SUCCESS;
29034359f16Sjeremylt }
29134359f16Sjeremylt 
29234359f16Sjeremylt /**
2936e15d496SJeremy L Thompson   @brief Estimate number of FLOPs required to apply CeedBasis in t_mode and eval_mode
2946e15d496SJeremy L Thompson 
2956e15d496SJeremy L Thompson   @param basis     Basis to estimate FLOPs for
2966e15d496SJeremy L Thompson   @param t_mode    Apply basis or transpose
2976e15d496SJeremy L Thompson   @param eval_mode Basis evaluation mode
2986e15d496SJeremy L Thompson   @param flops     Address of variable to hold FLOPs estimate
2996e15d496SJeremy L Thompson 
3006e15d496SJeremy L Thompson   @ref Backend
3016e15d496SJeremy L Thompson **/
3026e15d496SJeremy L Thompson int CeedBasisGetFlopsEstimate(CeedBasis basis, CeedTransposeMode t_mode,
3039d36ca50SJeremy L Thompson                               CeedEvalMode eval_mode, CeedSize *flops) {
3046e15d496SJeremy L Thompson   int ierr;
3056e15d496SJeremy L Thompson   bool is_tensor;
3066e15d496SJeremy L Thompson 
3076e15d496SJeremy L Thompson   ierr = CeedBasisIsTensor(basis, &is_tensor); CeedChk(ierr);
3086e15d496SJeremy L Thompson   if (is_tensor) {
3096e15d496SJeremy L Thompson     CeedInt dim, num_comp, P_1d, Q_1d;
3106e15d496SJeremy L Thompson     ierr = CeedBasisGetDimension(basis, &dim); CeedChk(ierr);
3116e15d496SJeremy L Thompson     ierr = CeedBasisGetNumComponents(basis, &num_comp); CeedChk(ierr);
3126e15d496SJeremy L Thompson     ierr = CeedBasisGetNumNodes1D(basis, &P_1d);  CeedChk(ierr);
3136e15d496SJeremy L Thompson     ierr = CeedBasisGetNumQuadraturePoints1D(basis, &Q_1d);  CeedChk(ierr);
3146e15d496SJeremy L Thompson     if (t_mode == CEED_TRANSPOSE) {
3156e15d496SJeremy L Thompson       P_1d = Q_1d; Q_1d = P_1d;
3166e15d496SJeremy L Thompson     }
3176e15d496SJeremy L Thompson     CeedInt tensor_flops = 0, pre = num_comp * CeedIntPow(P_1d, dim-1), post = 1;
3186e15d496SJeremy L Thompson     for (CeedInt d = 0; d < dim; d++) {
3196e15d496SJeremy L Thompson       tensor_flops += 2 * pre * P_1d * post * Q_1d;
3206e15d496SJeremy L Thompson       pre /= P_1d;
3216e15d496SJeremy L Thompson       post *= Q_1d;
3226e15d496SJeremy L Thompson     }
3236e15d496SJeremy L Thompson     switch (eval_mode) {
3246e15d496SJeremy L Thompson     case CEED_EVAL_NONE:   *flops = 0; break;
3256e15d496SJeremy L Thompson     case CEED_EVAL_INTERP: *flops = tensor_flops; break;
3266e15d496SJeremy L Thompson     case CEED_EVAL_GRAD:   *flops = tensor_flops * 2; break;
3276e15d496SJeremy L Thompson     case CEED_EVAL_DIV:
3286e15d496SJeremy L Thompson       // LCOV_EXCL_START
3296e15d496SJeremy L Thompson       return CeedError(basis->ceed, CEED_ERROR_INCOMPATIBLE,
3306e15d496SJeremy L Thompson                        "Tensor CEED_EVAL_DIV not supported"); break;
3316e15d496SJeremy L Thompson     case CEED_EVAL_CURL:
3326e15d496SJeremy L Thompson       return CeedError(basis->ceed, CEED_ERROR_INCOMPATIBLE,
3336e15d496SJeremy L Thompson                        "Tensor CEED_EVAL_CURL not supported"); break;
3346e15d496SJeremy L Thompson     // LCOV_EXCL_STOP
3356e15d496SJeremy L Thompson     case CEED_EVAL_WEIGHT: *flops = dim * CeedIntPow(Q_1d, dim); break;
3366e15d496SJeremy L Thompson     }
3376e15d496SJeremy L Thompson   } else {
3386e15d496SJeremy L Thompson     CeedInt dim, num_comp, num_nodes, num_qpts, Q_comp;
3396e15d496SJeremy L Thompson     ierr = CeedBasisGetDimension(basis, &dim); CeedChk(ierr);
3406e15d496SJeremy L Thompson     ierr = CeedBasisGetNumComponents(basis, &num_comp); CeedChk(ierr);
3416e15d496SJeremy L Thompson     ierr = CeedBasisGetNumNodes(basis, &num_nodes); CeedChk(ierr);
3426e15d496SJeremy L Thompson     ierr = CeedBasisGetNumQuadraturePoints(basis, &num_qpts); CeedChk(ierr);
3436e15d496SJeremy L Thompson     ierr = CeedBasisGetNumQuadratureComponents(basis, &Q_comp); CeedChk(ierr);
3446e15d496SJeremy L Thompson     switch (eval_mode) {
3456e15d496SJeremy L Thompson     case CEED_EVAL_NONE:   *flops = 0; break;
3466e15d496SJeremy L Thompson     case CEED_EVAL_INTERP: *flops = num_nodes * num_qpts * num_comp; break;
3476e15d496SJeremy L Thompson     case CEED_EVAL_GRAD:   *flops = num_nodes * num_qpts * num_comp * dim; break;
3486e15d496SJeremy L Thompson     case CEED_EVAL_DIV:    *flops = num_nodes * num_qpts; break;
3496e15d496SJeremy L Thompson     case CEED_EVAL_CURL:   *flops = num_nodes * num_qpts * dim; break;
3506e15d496SJeremy L Thompson     case CEED_EVAL_WEIGHT: *flops = 0; break;
3516e15d496SJeremy L Thompson     }
3526e15d496SJeremy L Thompson   }
3536e15d496SJeremy L Thompson 
3546e15d496SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3556e15d496SJeremy L Thompson }
3566e15d496SJeremy L Thompson 
3576e15d496SJeremy L Thompson /**
3587a982d89SJeremy L. Thompson   @brief Get dimension for given CeedElemTopology
3597a982d89SJeremy L. Thompson 
3607a982d89SJeremy L. Thompson   @param topo      CeedElemTopology
3617a982d89SJeremy L. Thompson   @param[out] dim  Variable to store dimension of topology
3627a982d89SJeremy L. Thompson 
3637a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
3647a982d89SJeremy L. Thompson 
3657a982d89SJeremy L. Thompson   @ref Backend
3667a982d89SJeremy L. Thompson **/
3677a982d89SJeremy L. Thompson int CeedBasisGetTopologyDimension(CeedElemTopology topo, CeedInt *dim) {
3687a982d89SJeremy L. Thompson   *dim = (CeedInt) topo >> 16;
369e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3707a982d89SJeremy L. Thompson }
3717a982d89SJeremy L. Thompson 
3727a982d89SJeremy L. Thompson /**
3737a982d89SJeremy L. Thompson   @brief Get CeedTensorContract of a CeedBasis
3747a982d89SJeremy L. Thompson 
3757a982d89SJeremy L. Thompson   @param basis          CeedBasis
3767a982d89SJeremy L. Thompson   @param[out] contract  Variable to store CeedTensorContract
3777a982d89SJeremy L. Thompson 
3787a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
3797a982d89SJeremy L. Thompson 
3807a982d89SJeremy L. Thompson   @ref Backend
3817a982d89SJeremy L. Thompson **/
3827a982d89SJeremy L. Thompson int CeedBasisGetTensorContract(CeedBasis basis, CeedTensorContract *contract) {
3837a982d89SJeremy L. Thompson   *contract = basis->contract;
384e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
3857a982d89SJeremy L. Thompson }
3867a982d89SJeremy L. Thompson 
3877a982d89SJeremy L. Thompson /**
3887a982d89SJeremy L. Thompson   @brief Set CeedTensorContract of a CeedBasis
3897a982d89SJeremy L. Thompson 
3907a982d89SJeremy L. Thompson   @param[out] basis  CeedBasis
3917a982d89SJeremy L. Thompson   @param contract    CeedTensorContract to set
3927a982d89SJeremy L. Thompson 
3937a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
3947a982d89SJeremy L. Thompson 
3957a982d89SJeremy L. Thompson   @ref Backend
3967a982d89SJeremy L. Thompson **/
39734359f16Sjeremylt int CeedBasisSetTensorContract(CeedBasis basis, CeedTensorContract contract) {
39834359f16Sjeremylt   int ierr;
39934359f16Sjeremylt   basis->contract = contract;
4009560d06aSjeremylt   ierr = CeedTensorContractReference(contract); CeedChk(ierr);
401e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4027a982d89SJeremy L. Thompson }
4037a982d89SJeremy L. Thompson 
4047a982d89SJeremy L. Thompson /**
4057a982d89SJeremy L. Thompson   @brief Return a reference implementation of matrix multiplication C = A B.
4067a982d89SJeremy L. Thompson            Note, this is a reference implementation for CPU CeedScalar pointers
4077a982d89SJeremy L. Thompson            that is not intended for high performance.
4087a982d89SJeremy L. Thompson 
4097a982d89SJeremy L. Thompson   @param ceed        A Ceed context for error handling
410d1d35e2fSjeremylt   @param[in] mat_A   Row-major matrix A
411d1d35e2fSjeremylt   @param[in] mat_B   Row-major matrix B
412d1d35e2fSjeremylt   @param[out] mat_C  Row-major output matrix C
4137a982d89SJeremy L. Thompson   @param m           Number of rows of C
4147a982d89SJeremy L. Thompson   @param n           Number of columns of C
4157a982d89SJeremy L. Thompson   @param kk          Number of columns of A/rows of B
4167a982d89SJeremy L. Thompson 
4177a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
4187a982d89SJeremy L. Thompson 
4197a982d89SJeremy L. Thompson   @ref Utility
4207a982d89SJeremy L. Thompson **/
421ed9e99e6SJeremy L Thompson int CeedMatrixMatrixMultiply(Ceed ceed, const CeedScalar *mat_A,
422ed9e99e6SJeremy L Thompson                              const CeedScalar *mat_B, CeedScalar *mat_C,
423ed9e99e6SJeremy L Thompson                              CeedInt m, CeedInt n, CeedInt kk) {
4247a982d89SJeremy L. Thompson   for (CeedInt i=0; i<m; i++)
4257a982d89SJeremy L. Thompson     for (CeedInt j=0; j<n; j++) {
4267a982d89SJeremy L. Thompson       CeedScalar sum = 0;
4277a982d89SJeremy L. Thompson       for (CeedInt k=0; k<kk; k++)
428d1d35e2fSjeremylt         sum += mat_A[k+i*kk]*mat_B[j+k*n];
429d1d35e2fSjeremylt       mat_C[j+i*n] = sum;
4307a982d89SJeremy L. Thompson     }
431e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
4327a982d89SJeremy L. Thompson }
4337a982d89SJeremy L. Thompson 
4347a982d89SJeremy L. Thompson /// @}
4357a982d89SJeremy L. Thompson 
4367a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
4377a982d89SJeremy L. Thompson /// CeedBasis Public API
4387a982d89SJeremy L. Thompson /// ----------------------------------------------------------------------------
4397a982d89SJeremy L. Thompson /// @addtogroup CeedBasisUser
440d7b241e6Sjeremylt /// @{
441d7b241e6Sjeremylt 
442b11c1e72Sjeremylt /**
44395bb1877Svaleriabarra   @brief Create a tensor-product basis for H^1 discretizations
444b11c1e72Sjeremylt 
445b11c1e72Sjeremylt   @param ceed        A Ceed object where the CeedBasis will be created
446b11c1e72Sjeremylt   @param dim         Topological dimension
447d1d35e2fSjeremylt   @param num_comp    Number of field components (1 for scalar fields)
448d1d35e2fSjeremylt   @param P_1d        Number of nodes in one dimension
449d1d35e2fSjeremylt   @param Q_1d        Number of quadrature points in one dimension
450d1d35e2fSjeremylt   @param interp_1d   Row-major (Q_1d * P_1d) matrix expressing the values of nodal
451b11c1e72Sjeremylt                        basis functions at quadrature points
452d1d35e2fSjeremylt   @param grad_1d     Row-major (Q_1d * P_1d) matrix expressing derivatives of nodal
453b11c1e72Sjeremylt                        basis functions at quadrature points
454d1d35e2fSjeremylt   @param q_ref_1d    Array of length Q_1d holding the locations of quadrature points
455b11c1e72Sjeremylt                        on the 1D reference element [-1, 1]
456d1d35e2fSjeremylt   @param q_weight_1d Array of length Q_1d holding the quadrature weights on the
457b11c1e72Sjeremylt                        reference element
458b11c1e72Sjeremylt   @param[out] basis  Address of the variable where the newly created
459b11c1e72Sjeremylt                        CeedBasis will be stored.
460b11c1e72Sjeremylt 
461b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
462dfdf5a53Sjeremylt 
4637a982d89SJeremy L. Thompson   @ref User
464b11c1e72Sjeremylt **/
465d1d35e2fSjeremylt int CeedBasisCreateTensorH1(Ceed ceed, CeedInt dim, CeedInt num_comp,
466d1d35e2fSjeremylt                             CeedInt P_1d, CeedInt Q_1d,
467d1d35e2fSjeremylt                             const CeedScalar *interp_1d,
468d1d35e2fSjeremylt                             const CeedScalar *grad_1d, const CeedScalar *q_ref_1d,
469d1d35e2fSjeremylt                             const CeedScalar *q_weight_1d, CeedBasis *basis) {
470d7b241e6Sjeremylt   int ierr;
471d7b241e6Sjeremylt 
4725fe0d4faSjeremylt   if (!ceed->BasisCreateTensorH1) {
4735fe0d4faSjeremylt     Ceed delegate;
474aefd8378Sjeremylt     ierr = CeedGetObjectDelegate(ceed, &delegate, "Basis"); CeedChk(ierr);
4755fe0d4faSjeremylt 
4765fe0d4faSjeremylt     if (!delegate)
477c042f62fSJeremy L Thompson       // LCOV_EXCL_START
478e15f9bd0SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
479e15f9bd0SJeremy L Thompson                        "Backend does not support BasisCreateTensorH1");
480c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
4815fe0d4faSjeremylt 
482d1d35e2fSjeremylt     ierr = CeedBasisCreateTensorH1(delegate, dim, num_comp, P_1d,
483d1d35e2fSjeremylt                                    Q_1d, interp_1d, grad_1d, q_ref_1d,
484d1d35e2fSjeremylt                                    q_weight_1d, basis); CeedChk(ierr);
485e15f9bd0SJeremy L Thompson     return CEED_ERROR_SUCCESS;
4865fe0d4faSjeremylt   }
487e15f9bd0SJeremy L Thompson 
488e15f9bd0SJeremy L Thompson   if (dim < 1)
489e15f9bd0SJeremy L Thompson     // LCOV_EXCL_START
490e15f9bd0SJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
491e15f9bd0SJeremy L Thompson                      "Basis dimension must be a positive value");
492e15f9bd0SJeremy L Thompson   // LCOV_EXCL_STOP
493227444bfSJeremy L Thompson 
494227444bfSJeremy L Thompson   if (num_comp < 1)
495227444bfSJeremy L Thompson     // LCOV_EXCL_START
496227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
497227444bfSJeremy L Thompson                      "Basis must have at least 1 component");
498227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
499227444bfSJeremy L Thompson 
500227444bfSJeremy L Thompson   if (P_1d < 1)
501227444bfSJeremy L Thompson     // LCOV_EXCL_START
502227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
503227444bfSJeremy L Thompson                      "Basis must have at least 1 node");
504227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
505227444bfSJeremy L Thompson 
506227444bfSJeremy L Thompson   if (Q_1d < 1)
507227444bfSJeremy L Thompson     // LCOV_EXCL_START
508227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
509227444bfSJeremy L Thompson                      "Basis must have at least 1 quadrature point");
510227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
511227444bfSJeremy L Thompson 
51250c301a5SRezgar Shakeri   CeedElemTopology topo = dim == 1 ? CEED_TOPOLOGY_LINE
51350c301a5SRezgar Shakeri                           : dim == 2 ? CEED_TOPOLOGY_QUAD
51450c301a5SRezgar Shakeri                           : CEED_TOPOLOGY_HEX;
515e15f9bd0SJeremy L Thompson 
516d7b241e6Sjeremylt   ierr = CeedCalloc(1, basis); CeedChk(ierr);
517d7b241e6Sjeremylt   (*basis)->ceed = ceed;
5189560d06aSjeremylt   ierr = CeedReference(ceed); CeedChk(ierr);
519d1d35e2fSjeremylt   (*basis)->ref_count = 1;
520d1d35e2fSjeremylt   (*basis)->tensor_basis = 1;
521d7b241e6Sjeremylt   (*basis)->dim = dim;
522d99fa3c5SJeremy L Thompson   (*basis)->topo = topo;
523d1d35e2fSjeremylt   (*basis)->num_comp = num_comp;
524d1d35e2fSjeremylt   (*basis)->P_1d = P_1d;
525d1d35e2fSjeremylt   (*basis)->Q_1d = Q_1d;
526d1d35e2fSjeremylt   (*basis)->P = CeedIntPow(P_1d, dim);
527d1d35e2fSjeremylt   (*basis)->Q = CeedIntPow(Q_1d, dim);
52850c301a5SRezgar Shakeri   (*basis)->Q_comp = 1;
52950c301a5SRezgar Shakeri   (*basis)->basis_space = 1; // 1 for H^1 space
530ff3a0f91SJeremy L Thompson   ierr = CeedCalloc(Q_1d, &(*basis)->q_ref_1d); CeedChk(ierr);
531ff3a0f91SJeremy L Thompson   ierr = CeedCalloc(Q_1d, &(*basis)->q_weight_1d); CeedChk(ierr);
532ff3a0f91SJeremy L Thompson   if (q_ref_1d) memcpy((*basis)->q_ref_1d, q_ref_1d, Q_1d*sizeof(q_ref_1d[0]));
533ff3a0f91SJeremy L Thompson   if (q_weight_1d) memcpy((*basis)->q_weight_1d, q_weight_1d,
534ff3a0f91SJeremy L Thompson                             Q_1d*sizeof(q_weight_1d[0]));
535ff3a0f91SJeremy L Thompson   ierr = CeedCalloc(Q_1d*P_1d, &(*basis)->interp_1d); CeedChk(ierr);
536ff3a0f91SJeremy L Thompson   ierr = CeedCalloc(Q_1d*P_1d, &(*basis)->grad_1d); CeedChk(ierr);
537ff3a0f91SJeremy L Thompson   if (interp_1d) memcpy((*basis)->interp_1d, interp_1d,
538ff3a0f91SJeremy L Thompson                           Q_1d*P_1d*sizeof(interp_1d[0]));
539ff3a0f91SJeremy L Thompson   if (grad_1d) memcpy((*basis)->grad_1d, grad_1d, Q_1d*P_1d*sizeof(grad_1d[0]));
540d1d35e2fSjeremylt   ierr = ceed->BasisCreateTensorH1(dim, P_1d, Q_1d, interp_1d, grad_1d, q_ref_1d,
541d1d35e2fSjeremylt                                    q_weight_1d, *basis); CeedChk(ierr);
542e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
543d7b241e6Sjeremylt }
544d7b241e6Sjeremylt 
545b11c1e72Sjeremylt /**
54695bb1877Svaleriabarra   @brief Create a tensor-product Lagrange basis
547b11c1e72Sjeremylt 
548b11c1e72Sjeremylt   @param ceed        A Ceed object where the CeedBasis will be created
549b11c1e72Sjeremylt   @param dim         Topological dimension of element
550d1d35e2fSjeremylt   @param num_comp      Number of field components (1 for scalar fields)
551b11c1e72Sjeremylt   @param P           Number of Gauss-Lobatto nodes in one dimension.  The
552b11c1e72Sjeremylt                        polynomial degree of the resulting Q_k element is k=P-1.
553b11c1e72Sjeremylt   @param Q           Number of quadrature points in one dimension.
554d1d35e2fSjeremylt   @param quad_mode   Distribution of the Q quadrature points (affects order of
555b11c1e72Sjeremylt                        accuracy for the quadrature)
556b11c1e72Sjeremylt   @param[out] basis  Address of the variable where the newly created
557b11c1e72Sjeremylt                        CeedBasis will be stored.
558b11c1e72Sjeremylt 
559b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
560dfdf5a53Sjeremylt 
5617a982d89SJeremy L. Thompson   @ref User
562b11c1e72Sjeremylt **/
563d1d35e2fSjeremylt int CeedBasisCreateTensorH1Lagrange(Ceed ceed, CeedInt dim, CeedInt num_comp,
564d1d35e2fSjeremylt                                     CeedInt P, CeedInt Q, CeedQuadMode quad_mode,
565692c2638Sjeremylt                                     CeedBasis *basis) {
566d7b241e6Sjeremylt   // Allocate
567e15f9bd0SJeremy L Thompson   int ierr, ierr2, i, j, k;
568d1d35e2fSjeremylt   CeedScalar c1, c2, c3, c4, dx, *nodes, *interp_1d, *grad_1d, *q_ref_1d,
569d1d35e2fSjeremylt              *q_weight_1d;
5704d537eeaSYohann 
5714d537eeaSYohann   if (dim < 1)
572c042f62fSJeremy L Thompson     // LCOV_EXCL_START
573e15f9bd0SJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
574e15f9bd0SJeremy L Thompson                      "Basis dimension must be a positive value");
575c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
5764d537eeaSYohann 
577227444bfSJeremy L Thompson   if (num_comp < 1)
578227444bfSJeremy L Thompson     // LCOV_EXCL_START
579227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
580227444bfSJeremy L Thompson                      "Basis must have at least 1 component");
581227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
582227444bfSJeremy L Thompson 
583227444bfSJeremy L Thompson   if (P < 1)
584227444bfSJeremy L Thompson     // LCOV_EXCL_START
585227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
586227444bfSJeremy L Thompson                      "Basis must have at least 1 node");
587227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
588227444bfSJeremy L Thompson 
589227444bfSJeremy L Thompson   if (Q < 1)
590227444bfSJeremy L Thompson     // LCOV_EXCL_START
591227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
592227444bfSJeremy L Thompson                      "Basis must have at least 1 quadrature point");
593227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
594227444bfSJeremy L Thompson 
595e15f9bd0SJeremy L Thompson   // Get Nodes and Weights
596d1d35e2fSjeremylt   ierr = CeedCalloc(P*Q, &interp_1d); CeedChk(ierr);
597d1d35e2fSjeremylt   ierr = CeedCalloc(P*Q, &grad_1d); CeedChk(ierr);
598d7b241e6Sjeremylt   ierr = CeedCalloc(P, &nodes); CeedChk(ierr);
599d1d35e2fSjeremylt   ierr = CeedCalloc(Q, &q_ref_1d); CeedChk(ierr);
600d1d35e2fSjeremylt   ierr = CeedCalloc(Q, &q_weight_1d); CeedChk(ierr);
601e15f9bd0SJeremy L Thompson   ierr = CeedLobattoQuadrature(P, nodes, NULL);
602e15f9bd0SJeremy L Thompson   if (ierr) { goto cleanup; } CeedChk(ierr);
603d1d35e2fSjeremylt   switch (quad_mode) {
604d7b241e6Sjeremylt   case CEED_GAUSS:
605d1d35e2fSjeremylt     ierr = CeedGaussQuadrature(Q, q_ref_1d, q_weight_1d);
606d7b241e6Sjeremylt     break;
607d7b241e6Sjeremylt   case CEED_GAUSS_LOBATTO:
608d1d35e2fSjeremylt     ierr = CeedLobattoQuadrature(Q, q_ref_1d, q_weight_1d);
609d7b241e6Sjeremylt     break;
610d7b241e6Sjeremylt   }
611e15f9bd0SJeremy L Thompson   if (ierr) { goto cleanup; } CeedChk(ierr);
612e15f9bd0SJeremy L Thompson 
613d7b241e6Sjeremylt   // Build B, D matrix
614d7b241e6Sjeremylt   // Fornberg, 1998
615d7b241e6Sjeremylt   for (i = 0; i  < Q; i++) {
616d7b241e6Sjeremylt     c1 = 1.0;
617d1d35e2fSjeremylt     c3 = nodes[0] - q_ref_1d[i];
618d1d35e2fSjeremylt     interp_1d[i*P+0] = 1.0;
619d7b241e6Sjeremylt     for (j = 1; j < P; j++) {
620d7b241e6Sjeremylt       c2 = 1.0;
621d7b241e6Sjeremylt       c4 = c3;
622d1d35e2fSjeremylt       c3 = nodes[j] - q_ref_1d[i];
623d7b241e6Sjeremylt       for (k = 0; k < j; k++) {
624d7b241e6Sjeremylt         dx = nodes[j] - nodes[k];
625d7b241e6Sjeremylt         c2 *= dx;
626d7b241e6Sjeremylt         if (k == j - 1) {
627d1d35e2fSjeremylt           grad_1d[i*P + j] = c1*(interp_1d[i*P + k] - c4*grad_1d[i*P + k]) / c2;
628d1d35e2fSjeremylt           interp_1d[i*P + j] = - c1*c4*interp_1d[i*P + k] / c2;
629d7b241e6Sjeremylt         }
630d1d35e2fSjeremylt         grad_1d[i*P + k] = (c3*grad_1d[i*P + k] - interp_1d[i*P + k]) / dx;
631d1d35e2fSjeremylt         interp_1d[i*P + k] = c3*interp_1d[i*P + k] / dx;
632d7b241e6Sjeremylt       }
633d7b241e6Sjeremylt       c1 = c2;
634d7b241e6Sjeremylt     }
635d7b241e6Sjeremylt   }
6369ac7b42eSJeremy L Thompson   // Pass to CeedBasisCreateTensorH1
637d1d35e2fSjeremylt   ierr = CeedBasisCreateTensorH1(ceed, dim, num_comp, P, Q, interp_1d, grad_1d,
6389ac7b42eSJeremy L Thompson                                  q_ref_1d, q_weight_1d, basis); CeedChk(ierr);
639e15f9bd0SJeremy L Thompson cleanup:
640d1d35e2fSjeremylt   ierr2 = CeedFree(&interp_1d); CeedChk(ierr2);
641d1d35e2fSjeremylt   ierr2 = CeedFree(&grad_1d); CeedChk(ierr2);
642e15f9bd0SJeremy L Thompson   ierr2 = CeedFree(&nodes); CeedChk(ierr2);
643d1d35e2fSjeremylt   ierr2 = CeedFree(&q_ref_1d); CeedChk(ierr2);
644d1d35e2fSjeremylt   ierr2 = CeedFree(&q_weight_1d); CeedChk(ierr2);
645e15f9bd0SJeremy L Thompson   CeedChk(ierr);
646e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
647d7b241e6Sjeremylt }
648d7b241e6Sjeremylt 
649b11c1e72Sjeremylt /**
65095bb1877Svaleriabarra   @brief Create a non tensor-product basis for H^1 discretizations
651a8de75f0Sjeremylt 
652a8de75f0Sjeremylt   @param ceed        A Ceed object where the CeedBasis will be created
653a8de75f0Sjeremylt   @param topo        Topology of element, e.g. hypercube, simplex, ect
654d1d35e2fSjeremylt   @param num_comp    Number of field components (1 for scalar fields)
655d1d35e2fSjeremylt   @param num_nodes   Total number of nodes
656d1d35e2fSjeremylt   @param num_qpts    Total number of quadrature points
657d1d35e2fSjeremylt   @param interp      Row-major (num_qpts * num_nodes) matrix expressing the values of
6588795c945Sjeremylt                        nodal basis functions at quadrature points
659d1d35e2fSjeremylt   @param grad        Row-major (num_qpts * dim * num_nodes) matrix expressing
6608795c945Sjeremylt                        derivatives of nodal basis functions at quadrature points
661d1d35e2fSjeremylt   @param q_ref       Array of length num_qpts holding the locations of quadrature
66250c301a5SRezgar Shakeri                        points on the reference element
663d1d35e2fSjeremylt   @param q_weight    Array of length num_qpts holding the quadrature weights on the
664a8de75f0Sjeremylt                        reference element
665a8de75f0Sjeremylt   @param[out] basis  Address of the variable where the newly created
666a8de75f0Sjeremylt                        CeedBasis will be stored.
667a8de75f0Sjeremylt 
668a8de75f0Sjeremylt   @return An error code: 0 - success, otherwise - failure
669a8de75f0Sjeremylt 
6707a982d89SJeremy L. Thompson   @ref User
671a8de75f0Sjeremylt **/
672d1d35e2fSjeremylt int CeedBasisCreateH1(Ceed ceed, CeedElemTopology topo, CeedInt num_comp,
673d1d35e2fSjeremylt                       CeedInt num_nodes, CeedInt num_qpts, const CeedScalar *interp,
674d1d35e2fSjeremylt                       const CeedScalar *grad, const CeedScalar *q_ref,
675d1d35e2fSjeremylt                       const CeedScalar *q_weight, CeedBasis *basis) {
676a8de75f0Sjeremylt   int ierr;
677d1d35e2fSjeremylt   CeedInt P = num_nodes, Q = num_qpts, dim = 0;
678a8de75f0Sjeremylt 
6795fe0d4faSjeremylt   if (!ceed->BasisCreateH1) {
6805fe0d4faSjeremylt     Ceed delegate;
681aefd8378Sjeremylt     ierr = CeedGetObjectDelegate(ceed, &delegate, "Basis"); CeedChk(ierr);
6825fe0d4faSjeremylt 
6835fe0d4faSjeremylt     if (!delegate)
684c042f62fSJeremy L Thompson       // LCOV_EXCL_START
685e15f9bd0SJeremy L Thompson       return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
686e15f9bd0SJeremy L Thompson                        "Backend does not support BasisCreateH1");
687c042f62fSJeremy L Thompson     // LCOV_EXCL_STOP
6885fe0d4faSjeremylt 
689d1d35e2fSjeremylt     ierr = CeedBasisCreateH1(delegate, topo, num_comp, num_nodes,
690d1d35e2fSjeremylt                              num_qpts, interp, grad, q_ref,
691d1d35e2fSjeremylt                              q_weight, basis); CeedChk(ierr);
692e15f9bd0SJeremy L Thompson     return CEED_ERROR_SUCCESS;
6935fe0d4faSjeremylt   }
6945fe0d4faSjeremylt 
695227444bfSJeremy L Thompson   if (num_comp < 1)
696227444bfSJeremy L Thompson     // LCOV_EXCL_START
697227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
698227444bfSJeremy L Thompson                      "Basis must have at least 1 component");
699227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
700227444bfSJeremy L Thompson 
701227444bfSJeremy L Thompson   if (num_nodes < 1)
702227444bfSJeremy L Thompson     // LCOV_EXCL_START
703227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
704227444bfSJeremy L Thompson                      "Basis must have at least 1 node");
705227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
706227444bfSJeremy L Thompson 
707227444bfSJeremy L Thompson   if (num_qpts < 1)
708227444bfSJeremy L Thompson     // LCOV_EXCL_START
709227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
710227444bfSJeremy L Thompson                      "Basis must have at least 1 quadrature point");
711227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
712227444bfSJeremy L Thompson 
713a8de75f0Sjeremylt   ierr = CeedCalloc(1, basis); CeedChk(ierr);
714a8de75f0Sjeremylt 
715a8de75f0Sjeremylt   ierr = CeedBasisGetTopologyDimension(topo, &dim); CeedChk(ierr);
716a8de75f0Sjeremylt 
717a8de75f0Sjeremylt   (*basis)->ceed = ceed;
7189560d06aSjeremylt   ierr = CeedReference(ceed); CeedChk(ierr);
719d1d35e2fSjeremylt   (*basis)->ref_count = 1;
720d1d35e2fSjeremylt   (*basis)->tensor_basis = 0;
721a8de75f0Sjeremylt   (*basis)->dim = dim;
722d99fa3c5SJeremy L Thompson   (*basis)->topo = topo;
723d1d35e2fSjeremylt   (*basis)->num_comp = num_comp;
724a8de75f0Sjeremylt   (*basis)->P = P;
725a8de75f0Sjeremylt   (*basis)->Q = Q;
72650c301a5SRezgar Shakeri   (*basis)->Q_comp = 1;
72750c301a5SRezgar Shakeri   (*basis)->basis_space = 1; // 1 for H^1 space
728ff3a0f91SJeremy L Thompson   ierr = CeedCalloc(Q*dim, &(*basis)->q_ref_1d); CeedChk(ierr);
729ff3a0f91SJeremy L Thompson   ierr = CeedCalloc(Q, &(*basis)->q_weight_1d); CeedChk(ierr);
730ff3a0f91SJeremy L Thompson   if (q_ref) memcpy((*basis)->q_ref_1d, q_ref, Q*dim*sizeof(q_ref[0]));
731ff3a0f91SJeremy L Thompson   if(q_weight) memcpy((*basis)->q_weight_1d, q_weight, Q*sizeof(q_weight[0]));
732ff3a0f91SJeremy L Thompson   ierr = CeedCalloc(Q*P, &(*basis)->interp); CeedChk(ierr);
733ff3a0f91SJeremy L Thompson   ierr = CeedCalloc(dim*Q*P, &(*basis)->grad); CeedChk(ierr);
734ff3a0f91SJeremy L Thompson   if(interp) memcpy((*basis)->interp, interp, Q*P*sizeof(interp[0]));
735ff3a0f91SJeremy L Thompson   if(grad) memcpy((*basis)->grad, grad, dim*Q*P*sizeof(grad[0]));
736d1d35e2fSjeremylt   ierr = ceed->BasisCreateH1(topo, dim, P, Q, interp, grad, q_ref,
737d1d35e2fSjeremylt                              q_weight, *basis); CeedChk(ierr);
738e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
739a8de75f0Sjeremylt }
740a8de75f0Sjeremylt 
741a8de75f0Sjeremylt /**
74250c301a5SRezgar Shakeri   @brief Create a non tensor-product basis for H(div) discretizations
74350c301a5SRezgar Shakeri 
74450c301a5SRezgar Shakeri   @param ceed        A Ceed object where the CeedBasis will be created
74550c301a5SRezgar Shakeri   @param topo        Topology of element (`CEED_TOPOLOGY_QUAD`, `CEED_TOPOLOGY_PRISM`, etc.),
74650c301a5SRezgar Shakeri                      dimension of which is used in some array sizes below
74750c301a5SRezgar Shakeri   @param num_comp    Number of components (usually 1 for vectors in H(div) bases)
74850c301a5SRezgar Shakeri   @param num_nodes   Total number of nodes (dofs per element)
74950c301a5SRezgar Shakeri   @param num_qpts    Total number of quadrature points
75050c301a5SRezgar Shakeri   @param interp      Row-major (dim*num_qpts * num_nodes) matrix expressing the values of
75150c301a5SRezgar Shakeri                        nodal basis functions at quadrature points
75250c301a5SRezgar Shakeri   @param div        Row-major (num_qpts * num_nodes) matrix expressing
75350c301a5SRezgar Shakeri                        divergence of nodal basis functions at quadrature points
75450c301a5SRezgar Shakeri   @param q_ref       Array of length num_qpts holding the locations of quadrature
75550c301a5SRezgar Shakeri                        points on the reference element
75650c301a5SRezgar Shakeri   @param q_weight    Array of length num_qpts holding the quadrature weights on the
75750c301a5SRezgar Shakeri                        reference element
75850c301a5SRezgar Shakeri   @param[out] basis  Address of the variable where the newly created
75950c301a5SRezgar Shakeri                        CeedBasis will be stored.
76050c301a5SRezgar Shakeri 
76150c301a5SRezgar Shakeri   @return An error code: 0 - success, otherwise - failure
76250c301a5SRezgar Shakeri 
76350c301a5SRezgar Shakeri   @ref User
76450c301a5SRezgar Shakeri **/
76550c301a5SRezgar Shakeri int CeedBasisCreateHdiv(Ceed ceed, CeedElemTopology topo, CeedInt num_comp,
76650c301a5SRezgar Shakeri                         CeedInt num_nodes, CeedInt num_qpts, const CeedScalar *interp,
76750c301a5SRezgar Shakeri                         const CeedScalar *div, const CeedScalar *q_ref,
76850c301a5SRezgar Shakeri                         const CeedScalar *q_weight, CeedBasis *basis) {
76950c301a5SRezgar Shakeri   int ierr;
77050c301a5SRezgar Shakeri   CeedInt Q = num_qpts, P = num_nodes, dim = 0;
77150c301a5SRezgar Shakeri   ierr = CeedBasisGetTopologyDimension(topo, &dim); CeedChk(ierr);
77250c301a5SRezgar Shakeri   if (!ceed->BasisCreateHdiv) {
77350c301a5SRezgar Shakeri     Ceed delegate;
77450c301a5SRezgar Shakeri     ierr = CeedGetObjectDelegate(ceed, &delegate, "Basis"); CeedChk(ierr);
77550c301a5SRezgar Shakeri 
77650c301a5SRezgar Shakeri     if (!delegate)
77750c301a5SRezgar Shakeri       // LCOV_EXCL_START
77850c301a5SRezgar Shakeri       return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
77950c301a5SRezgar Shakeri                        "Backend does not implement BasisCreateHdiv");
78050c301a5SRezgar Shakeri     // LCOV_EXCL_STOP
78150c301a5SRezgar Shakeri 
78250c301a5SRezgar Shakeri     ierr = CeedBasisCreateHdiv(delegate, topo, num_comp, num_nodes,
78350c301a5SRezgar Shakeri                                num_qpts, interp, div, q_ref,
78450c301a5SRezgar Shakeri                                q_weight, basis); CeedChk(ierr);
78550c301a5SRezgar Shakeri     return CEED_ERROR_SUCCESS;
78650c301a5SRezgar Shakeri   }
78750c301a5SRezgar Shakeri 
788227444bfSJeremy L Thompson   if (num_comp < 1)
789227444bfSJeremy L Thompson     // LCOV_EXCL_START
790227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
791227444bfSJeremy L Thompson                      "Basis must have at least 1 component");
792227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
793227444bfSJeremy L Thompson 
794227444bfSJeremy L Thompson   if (num_nodes < 1)
795227444bfSJeremy L Thompson     // LCOV_EXCL_START
796227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
797227444bfSJeremy L Thompson                      "Basis must have at least 1 node");
798227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
799227444bfSJeremy L Thompson 
800227444bfSJeremy L Thompson   if (num_qpts < 1)
801227444bfSJeremy L Thompson     // LCOV_EXCL_START
802227444bfSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
803227444bfSJeremy L Thompson                      "Basis must have at least 1 quadrature point");
804227444bfSJeremy L Thompson   // LCOV_EXCL_STOP
805227444bfSJeremy L Thompson 
80650c301a5SRezgar Shakeri   ierr = CeedCalloc(1, basis); CeedChk(ierr);
80750c301a5SRezgar Shakeri 
80850c301a5SRezgar Shakeri   (*basis)->ceed = ceed;
80950c301a5SRezgar Shakeri   ierr = CeedReference(ceed); CeedChk(ierr);
81050c301a5SRezgar Shakeri   (*basis)->ref_count = 1;
81150c301a5SRezgar Shakeri   (*basis)->tensor_basis = 0;
81250c301a5SRezgar Shakeri   (*basis)->dim = dim;
81350c301a5SRezgar Shakeri   (*basis)->topo = topo;
81450c301a5SRezgar Shakeri   (*basis)->num_comp = num_comp;
81550c301a5SRezgar Shakeri   (*basis)->P = P;
81650c301a5SRezgar Shakeri   (*basis)->Q = Q;
81750c301a5SRezgar Shakeri   (*basis)->Q_comp = dim;
81850c301a5SRezgar Shakeri   (*basis)->basis_space = 2; // 2 for H(div) space
81950c301a5SRezgar Shakeri   ierr = CeedMalloc(Q*dim, &(*basis)->q_ref_1d); CeedChk(ierr);
82050c301a5SRezgar Shakeri   ierr = CeedMalloc(Q, &(*basis)->q_weight_1d); CeedChk(ierr);
82150c301a5SRezgar Shakeri   if (q_ref) memcpy((*basis)->q_ref_1d, q_ref, Q*dim*sizeof(q_ref[0]));
82250c301a5SRezgar Shakeri   if (q_weight) memcpy((*basis)->q_weight_1d, q_weight, Q*sizeof(q_weight[0]));
82350c301a5SRezgar Shakeri   ierr = CeedMalloc(dim*Q*P, &(*basis)->interp); CeedChk(ierr);
82450c301a5SRezgar Shakeri   ierr = CeedMalloc(Q*P, &(*basis)->div); CeedChk(ierr);
82550c301a5SRezgar Shakeri   if (interp) memcpy((*basis)->interp, interp, dim*Q*P*sizeof(interp[0]));
82650c301a5SRezgar Shakeri   if (div) memcpy((*basis)->div, div, Q*P*sizeof(div[0]));
82750c301a5SRezgar Shakeri   ierr = ceed->BasisCreateHdiv(topo, dim, P, Q, interp, div, q_ref,
82850c301a5SRezgar Shakeri                                q_weight, *basis); CeedChk(ierr);
82950c301a5SRezgar Shakeri   return CEED_ERROR_SUCCESS;
83050c301a5SRezgar Shakeri }
83150c301a5SRezgar Shakeri 
83250c301a5SRezgar Shakeri /**
833f113e5dcSJeremy L Thompson   @brief Create a CeedBasis for projection from the nodes of `basis_from`
834f113e5dcSJeremy L Thompson            to the nodes of `basis_to`. Only `CEED_EVAL_INTERP` will be
835f113e5dcSJeremy L Thompson            valid for the new basis, `basis_project`. This projection is
836f113e5dcSJeremy L Thompson            given by `interp_project = interp_to^+ * interp_from`, where
837f113e5dcSJeremy L Thompson            the pesudoinverse `interp_to^+` is given by QR factorization.
838f113e5dcSJeremy L Thompson          Note: `basis_from` and `basis_to` must have compatible quadrature
839f113e5dcSJeremy L Thompson            spaces.
840*446e7af4SJeremy L Thompson          Note: `basis_project` will have the same number of components as
841*446e7af4SJeremy L Thompson            `basis_from`, regardless of the number of components that
842*446e7af4SJeremy L Thompson            `basis_to` has. If `basis_from` has 3 components and `basis_to`
843*446e7af4SJeremy L Thompson            has 5 components, then `basis_project` will have 3 components.
844f113e5dcSJeremy L Thompson 
845f113e5dcSJeremy L Thompson   @param[in] basis_from      CeedBasis to prolong from
846*446e7af4SJeremy L Thompson   @param[in] basis_to        CeedBasis to prolong to
847f113e5dcSJeremy L Thompson   @param[out] basis_project  Address of the variable where the newly created
848f113e5dcSJeremy L Thompson                                CeedBasis will be stored.
849f113e5dcSJeremy L Thompson 
850f113e5dcSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
851f113e5dcSJeremy L Thompson 
852f113e5dcSJeremy L Thompson   @ref User
853f113e5dcSJeremy L Thompson **/
854*446e7af4SJeremy L Thompson int CeedBasisCreateProjection(CeedBasis basis_from, CeedBasis basis_to,
855f113e5dcSJeremy L Thompson                               CeedBasis *basis_project) {
856f113e5dcSJeremy L Thompson   int ierr;
857f113e5dcSJeremy L Thompson   Ceed ceed;
858f113e5dcSJeremy L Thompson   ierr = CeedBasisGetCeed(basis_to, &ceed); CeedChk(ierr);
859f113e5dcSJeremy L Thompson 
860f113e5dcSJeremy L Thompson   // Create projectior matrix
861f113e5dcSJeremy L Thompson   CeedScalar *interp_project;
862*446e7af4SJeremy L Thompson   ierr = CeedBasisCreateProjectionMatrix(basis_from, basis_to,
863f113e5dcSJeremy L Thompson                                          &interp_project); CeedChk(ierr);
864f113e5dcSJeremy L Thompson 
865f113e5dcSJeremy L Thompson   // Build basis
866f113e5dcSJeremy L Thompson   bool is_tensor;
867f113e5dcSJeremy L Thompson   CeedInt dim, num_comp;
868f113e5dcSJeremy L Thompson   CeedScalar *q_ref, *q_weight, *grad;
869f113e5dcSJeremy L Thompson   ierr = CeedBasisIsTensor(basis_to, &is_tensor); CeedChk(ierr);
870f113e5dcSJeremy L Thompson   ierr = CeedBasisGetDimension(basis_to, &dim); CeedChk(ierr);
871*446e7af4SJeremy L Thompson   ierr = CeedBasisGetNumComponents(basis_from, &num_comp); CeedChk(ierr);
872f113e5dcSJeremy L Thompson   if (is_tensor) {
873f113e5dcSJeremy L Thompson     CeedInt P_1d_to, P_1d_from;
874f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumNodes1D(basis_from, &P_1d_from); CeedChk(ierr);
875f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumNodes1D(basis_to, &P_1d_to); CeedChk(ierr);
876f113e5dcSJeremy L Thompson     ierr = CeedCalloc(P_1d_to, &q_ref); CeedChk(ierr);
877f113e5dcSJeremy L Thompson     ierr = CeedCalloc(P_1d_to, &q_weight); CeedChk(ierr);
878f113e5dcSJeremy L Thompson     ierr = CeedCalloc(P_1d_to * P_1d_from * dim, &grad); CeedChk(ierr);
879f113e5dcSJeremy L Thompson     ierr = CeedBasisCreateTensorH1(ceed, dim, num_comp, P_1d_from, P_1d_to,
880f113e5dcSJeremy L Thompson                                    interp_project, grad, q_ref, q_weight, basis_project);
881f113e5dcSJeremy L Thompson     CeedChk(ierr);
882f113e5dcSJeremy L Thompson   } else {
883f113e5dcSJeremy L Thompson     CeedElemTopology topo;
884f113e5dcSJeremy L Thompson     ierr = CeedBasisGetTopology(basis_to, &topo); CeedChk(ierr);
885f113e5dcSJeremy L Thompson     CeedInt num_nodes_to, num_nodes_from;
886f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumNodes(basis_from, &num_nodes_from); CeedChk(ierr);
887f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumNodes(basis_to, &num_nodes_to); CeedChk(ierr);
888f113e5dcSJeremy L Thompson     ierr = CeedCalloc(num_nodes_to * dim, &q_ref); CeedChk(ierr);
889f113e5dcSJeremy L Thompson     ierr = CeedCalloc(num_nodes_to, &q_weight); CeedChk(ierr);
890f113e5dcSJeremy L Thompson     ierr = CeedCalloc(num_nodes_to * num_nodes_from * dim, &grad); CeedChk(ierr);
891f113e5dcSJeremy L Thompson     ierr = CeedBasisCreateH1(ceed, topo, num_comp, num_nodes_from, num_nodes_to,
892f113e5dcSJeremy L Thompson                              interp_project, grad, q_ref, q_weight, basis_project);
893f113e5dcSJeremy L Thompson     CeedChk(ierr);
894f113e5dcSJeremy L Thompson   }
895f113e5dcSJeremy L Thompson 
896f113e5dcSJeremy L Thompson   // Cleanup
897f113e5dcSJeremy L Thompson   ierr = CeedFree(&interp_project); CeedChk(ierr);
898f113e5dcSJeremy L Thompson   ierr = CeedFree(&q_ref); CeedChk(ierr);
899f113e5dcSJeremy L Thompson   ierr = CeedFree(&q_weight); CeedChk(ierr);
900f113e5dcSJeremy L Thompson   ierr = CeedFree(&grad); CeedChk(ierr);
901f113e5dcSJeremy L Thompson 
902f113e5dcSJeremy L Thompson   return CEED_ERROR_SUCCESS;
903f113e5dcSJeremy L Thompson }
904f113e5dcSJeremy L Thompson 
905f113e5dcSJeremy L Thompson /**
906f113e5dcSJeremy L Thompson   @brief Create the interpolation matrix for projection from the nodes of
907f113e5dcSJeremy L Thompson            `basis_from` to the nodes of `basis_to`. This projection is
908f113e5dcSJeremy L Thompson            given by `interp_project = interp_to^+ * interp_from`, where
909f113e5dcSJeremy L Thompson            the pesudoinverse `interp_to^+` is given by QR factorization.
910f113e5dcSJeremy L Thompson          Note: `basis_from` and `basis_to` must have compatible quadrature
911f113e5dcSJeremy L Thompson            spaces.
912f113e5dcSJeremy L Thompson 
913f113e5dcSJeremy L Thompson   @param[in] basis_from       CeedBasis to project from
914*446e7af4SJeremy L Thompson   @param[in] basis_to         CeedBasis to project to
915f113e5dcSJeremy L Thompson   @param[out] interp_project  Address of the variable where the newly created
916f113e5dcSJeremy L Thompson                                 projection matrix will be stored.
917f113e5dcSJeremy L Thompson 
918f113e5dcSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
919f113e5dcSJeremy L Thompson 
920f113e5dcSJeremy L Thompson   @ref User
921f113e5dcSJeremy L Thompson **/
922*446e7af4SJeremy L Thompson int CeedBasisCreateProjectionMatrix(CeedBasis basis_from,
923*446e7af4SJeremy L Thompson                                     CeedBasis basis_to,
924f113e5dcSJeremy L Thompson                                     CeedScalar **interp_project) {
925f113e5dcSJeremy L Thompson   int ierr;
926f113e5dcSJeremy L Thompson   Ceed ceed;
927f113e5dcSJeremy L Thompson   ierr = CeedBasisGetCeed(basis_to, &ceed); CeedChk(ierr);
928f113e5dcSJeremy L Thompson 
929f113e5dcSJeremy L Thompson   // Check for compatible quadrature spaces
930f113e5dcSJeremy L Thompson   CeedInt Q_to, Q_from;
931f113e5dcSJeremy L Thompson   ierr = CeedBasisGetNumQuadraturePoints(basis_to, &Q_to); CeedChk(ierr);
932f113e5dcSJeremy L Thompson   ierr = CeedBasisGetNumQuadraturePoints(basis_from, &Q_from); CeedChk(ierr);
933f113e5dcSJeremy L Thompson   if (Q_to != Q_from)
934f113e5dcSJeremy L Thompson     // LCOV_EXCL_START
935f113e5dcSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_DIMENSION,
936f113e5dcSJeremy L Thompson                      "Bases must have compatible quadrature spaces");
937f113e5dcSJeremy L Thompson   // LCOV_EXCL_STOP
938f113e5dcSJeremy L Thompson 
939f113e5dcSJeremy L Thompson   // Coarse to fine basis
940f113e5dcSJeremy L Thompson   CeedInt P_to, P_from, Q = Q_to;
941f113e5dcSJeremy L Thompson   bool is_tensor_to, is_tensor_from;
942f113e5dcSJeremy L Thompson   ierr = CeedBasisIsTensor(basis_to, &is_tensor_to); CeedChk(ierr);
943f113e5dcSJeremy L Thompson   ierr = CeedBasisIsTensor(basis_from, &is_tensor_from); CeedChk(ierr);
944f113e5dcSJeremy L Thompson   CeedScalar *interp_to, *interp_from, *tau;
945f113e5dcSJeremy L Thompson   if (is_tensor_to && is_tensor_from) {
946f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumNodes1D(basis_to, &P_to); CeedChk(ierr);
947f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumNodes1D(basis_from, &P_from); CeedChk(ierr);
948f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumQuadraturePoints1D(basis_from, &Q); CeedChk(ierr);
949f113e5dcSJeremy L Thompson   } else if (!is_tensor_to && !is_tensor_from) {
950f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumNodes(basis_to, &P_to); CeedChk(ierr);
951f113e5dcSJeremy L Thompson     ierr = CeedBasisGetNumNodes(basis_from, &P_from); CeedChk(ierr);
952f113e5dcSJeremy L Thompson   } else {
953f113e5dcSJeremy L Thompson     // LCOV_EXCL_START
954f113e5dcSJeremy L Thompson     return CeedError(ceed, CEED_ERROR_MINOR,
955f113e5dcSJeremy L Thompson                      "Bases must both be tensor or non-tensor");
956f113e5dcSJeremy L Thompson     // LCOV_EXCL_STOP
957f113e5dcSJeremy L Thompson   }
958f113e5dcSJeremy L Thompson 
959f113e5dcSJeremy L Thompson   ierr = CeedMalloc(Q * P_from, &interp_from); CeedChk(ierr);
960f113e5dcSJeremy L Thompson   ierr = CeedMalloc(Q * P_to, &interp_to); CeedChk(ierr);
961f113e5dcSJeremy L Thompson   ierr = CeedCalloc(P_to * P_from, interp_project); CeedChk(ierr);
962f113e5dcSJeremy L Thompson   ierr = CeedMalloc(Q, &tau); CeedChk(ierr);
963f113e5dcSJeremy L Thompson   const CeedScalar *interp_to_source = NULL, *interp_from_source = NULL;
964f113e5dcSJeremy L Thompson   if (is_tensor_to) {
965f113e5dcSJeremy L Thompson     ierr = CeedBasisGetInterp1D(basis_to, &interp_to_source); CeedChk(ierr);
966f113e5dcSJeremy L Thompson     ierr = CeedBasisGetInterp1D(basis_from, &interp_from_source); CeedChk(ierr);
967f113e5dcSJeremy L Thompson   } else {
968f113e5dcSJeremy L Thompson     ierr = CeedBasisGetInterp(basis_to, &interp_to_source); CeedChk(ierr);
969f113e5dcSJeremy L Thompson     ierr = CeedBasisGetInterp(basis_from, &interp_from_source); CeedChk(ierr);
970f113e5dcSJeremy L Thompson   }
971f113e5dcSJeremy L Thompson   memcpy(interp_to, interp_to_source, Q * P_to * sizeof(interp_to_source[0]));
972f113e5dcSJeremy L Thompson   memcpy(interp_from, interp_from_source,
973f113e5dcSJeremy L Thompson          Q * P_from * sizeof(interp_from_source[0]));
974f113e5dcSJeremy L Thompson 
975f113e5dcSJeremy L Thompson   // -- QR Factorization, interp_to = Q R
976f113e5dcSJeremy L Thompson   ierr = CeedQRFactorization(ceed, interp_to, tau, Q, P_to); CeedChk(ierr);
977f113e5dcSJeremy L Thompson 
978f113e5dcSJeremy L Thompson   // -- Apply Qtranspose, interp_to = Qtranspose interp_from
979f113e5dcSJeremy L Thompson   ierr = CeedHouseholderApplyQ(interp_from, interp_to, tau, CEED_TRANSPOSE,
980f113e5dcSJeremy L Thompson                                Q, P_from, P_to, P_from, 1); CeedChk(ierr);
981f113e5dcSJeremy L Thompson 
982f113e5dcSJeremy L Thompson   // -- Apply Rinv, interp_project = Rinv interp_c
983f113e5dcSJeremy L Thompson   for (CeedInt j = 0; j < P_from; j++) { // Column j
984f113e5dcSJeremy L Thompson     (*interp_project)[j + P_from * (P_to - 1)] = interp_from[j + P_from *
985f113e5dcSJeremy L Thompson         (P_to - 1)] / interp_to[P_to * P_to - 1];
986f113e5dcSJeremy L Thompson     for (CeedInt i = P_to - 2; i >= 0; i--) { // Row i
987f113e5dcSJeremy L Thompson       (*interp_project)[j + P_from * i] = interp_from[j + P_from * i];
988f113e5dcSJeremy L Thompson       for (CeedInt k = i+1; k < P_to; k++) {
989f113e5dcSJeremy L Thompson         (*interp_project)[j + P_from * i] -= interp_to[k + P_to * i]*
990f113e5dcSJeremy L Thompson                                              (*interp_project)[j + P_from * k];
991f113e5dcSJeremy L Thompson       }
992f113e5dcSJeremy L Thompson       (*interp_project)[j + P_from * i] /= interp_to[i + P_to * i];
993f113e5dcSJeremy L Thompson     }
994f113e5dcSJeremy L Thompson   }
995f113e5dcSJeremy L Thompson   ierr = CeedFree(&tau); CeedChk(ierr);
996f113e5dcSJeremy L Thompson   ierr = CeedFree(&interp_to); CeedChk(ierr);
997f113e5dcSJeremy L Thompson   ierr = CeedFree(&interp_from); CeedChk(ierr);
998f113e5dcSJeremy L Thompson 
999f113e5dcSJeremy L Thompson   return CEED_ERROR_SUCCESS;
1000f113e5dcSJeremy L Thompson }
1001f113e5dcSJeremy L Thompson 
1002f113e5dcSJeremy L Thompson /**
10039560d06aSjeremylt   @brief Copy the pointer to a CeedBasis. Both pointers should
10049560d06aSjeremylt            be destroyed with `CeedBasisDestroy()`;
10059560d06aSjeremylt            Note: If `*basis_copy` is non-NULL, then it is assumed that
10069560d06aSjeremylt            `*basis_copy` is a pointer to a CeedBasis. This CeedBasis
10079560d06aSjeremylt            will be destroyed if `*basis_copy` is the only
10089560d06aSjeremylt            reference to this CeedBasis.
10099560d06aSjeremylt 
10109560d06aSjeremylt   @param basis            CeedBasis to copy reference to
10119560d06aSjeremylt   @param[out] basis_copy  Variable to store copied reference
10129560d06aSjeremylt 
10139560d06aSjeremylt   @return An error code: 0 - success, otherwise - failure
10149560d06aSjeremylt 
10159560d06aSjeremylt   @ref User
10169560d06aSjeremylt **/
10179560d06aSjeremylt int CeedBasisReferenceCopy(CeedBasis basis, CeedBasis *basis_copy) {
10189560d06aSjeremylt   int ierr;
10199560d06aSjeremylt 
10209560d06aSjeremylt   ierr = CeedBasisReference(basis); CeedChk(ierr);
10219560d06aSjeremylt   ierr = CeedBasisDestroy(basis_copy); CeedChk(ierr);
10229560d06aSjeremylt   *basis_copy = basis;
10239560d06aSjeremylt   return CEED_ERROR_SUCCESS;
10249560d06aSjeremylt }
10259560d06aSjeremylt 
10269560d06aSjeremylt /**
10277a982d89SJeremy L. Thompson   @brief View a CeedBasis
10287a982d89SJeremy L. Thompson 
10297a982d89SJeremy L. Thompson   @param basis   CeedBasis to view
10307a982d89SJeremy L. Thompson   @param stream  Stream to view to, e.g., stdout
10317a982d89SJeremy L. Thompson 
10327a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
10337a982d89SJeremy L. Thompson 
10347a982d89SJeremy L. Thompson   @ref User
10357a982d89SJeremy L. Thompson **/
10367a982d89SJeremy L. Thompson int CeedBasisView(CeedBasis basis, FILE *stream) {
10377a982d89SJeremy L. Thompson   int ierr;
103850c301a5SRezgar Shakeri   CeedFESpace FE_space = basis->basis_space;
103950c301a5SRezgar Shakeri   CeedElemTopology topo = basis->topo;
104050c301a5SRezgar Shakeri   // Print FE space and element topology of the basis
1041d1d35e2fSjeremylt   if (basis->tensor_basis) {
104250c301a5SRezgar Shakeri     fprintf(stream, "CeedBasis (%s on a %s element): dim=%d P=%d Q=%d\n",
104350c301a5SRezgar Shakeri             CeedFESpaces[FE_space], CeedElemTopologies[topo],
104450c301a5SRezgar Shakeri             basis->dim, basis->P_1d, basis->Q_1d);
104550c301a5SRezgar Shakeri   } else {
104650c301a5SRezgar Shakeri     fprintf(stream, "CeedBasis (%s on a %s element): dim=%d P=%d Q=%d\n",
104750c301a5SRezgar Shakeri             CeedFESpaces[FE_space], CeedElemTopologies[topo],
104850c301a5SRezgar Shakeri             basis->dim, basis->P, basis->Q);
104950c301a5SRezgar Shakeri   }
105050c301a5SRezgar Shakeri   // Print quadrature data, interpolation/gradient/divergene/curl of the basis
105150c301a5SRezgar Shakeri   if (basis->tensor_basis) { // tensor basis
1052d1d35e2fSjeremylt     ierr = CeedScalarView("qref1d", "\t% 12.8f", 1, basis->Q_1d, basis->q_ref_1d,
10537a982d89SJeremy L. Thompson                           stream); CeedChk(ierr);
1054d1d35e2fSjeremylt     ierr = CeedScalarView("qweight1d", "\t% 12.8f", 1, basis->Q_1d,
1055d1d35e2fSjeremylt                           basis->q_weight_1d, stream); CeedChk(ierr);
1056d1d35e2fSjeremylt     ierr = CeedScalarView("interp1d", "\t% 12.8f", basis->Q_1d, basis->P_1d,
1057d1d35e2fSjeremylt                           basis->interp_1d, stream); CeedChk(ierr);
1058d1d35e2fSjeremylt     ierr = CeedScalarView("grad1d", "\t% 12.8f", basis->Q_1d, basis->P_1d,
1059d1d35e2fSjeremylt                           basis->grad_1d, stream); CeedChk(ierr);
106050c301a5SRezgar Shakeri   } else { // non-tensor basis
10617a982d89SJeremy L. Thompson     ierr = CeedScalarView("qref", "\t% 12.8f", 1, basis->Q*basis->dim,
1062d1d35e2fSjeremylt                           basis->q_ref_1d,
10637a982d89SJeremy L. Thompson                           stream); CeedChk(ierr);
1064d1d35e2fSjeremylt     ierr = CeedScalarView("qweight", "\t% 12.8f", 1, basis->Q, basis->q_weight_1d,
10657a982d89SJeremy L. Thompson                           stream); CeedChk(ierr);
106650c301a5SRezgar Shakeri     ierr = CeedScalarView("interp", "\t% 12.8f", basis->Q_comp*basis->Q, basis->P,
10677a982d89SJeremy L. Thompson                           basis->interp, stream); CeedChk(ierr);
106850c301a5SRezgar Shakeri     if (basis->grad) {
10697a982d89SJeremy L. Thompson       ierr = CeedScalarView("grad", "\t% 12.8f", basis->dim*basis->Q, basis->P,
10707a982d89SJeremy L. Thompson                             basis->grad, stream); CeedChk(ierr);
10717a982d89SJeremy L. Thompson     }
107250c301a5SRezgar Shakeri     if (basis->div) {
107350c301a5SRezgar Shakeri       ierr = CeedScalarView("div", "\t% 12.8f", basis->Q, basis->P,
107450c301a5SRezgar Shakeri                             basis->div, stream); CeedChk(ierr);
107550c301a5SRezgar Shakeri     }
107650c301a5SRezgar Shakeri   }
1077e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
10787a982d89SJeremy L. Thompson }
10797a982d89SJeremy L. Thompson 
10807a982d89SJeremy L. Thompson /**
10817a982d89SJeremy L. Thompson   @brief Apply basis evaluation from nodes to quadrature points or vice versa
10827a982d89SJeremy L. Thompson 
10837a982d89SJeremy L. Thompson   @param basis     CeedBasis to evaluate
1084d1d35e2fSjeremylt   @param num_elem  The number of elements to apply the basis evaluation to;
10857a982d89SJeremy L. Thompson                      the backend will specify the ordering in
10864cc79fe7SJed Brown                      CeedElemRestrictionCreateBlocked()
1087d1d35e2fSjeremylt   @param t_mode    \ref CEED_NOTRANSPOSE to evaluate from nodes to quadrature
10887a982d89SJeremy L. Thompson                      points, \ref CEED_TRANSPOSE to apply the transpose, mapping
10897a982d89SJeremy L. Thompson                      from quadrature points to nodes
1090d1d35e2fSjeremylt   @param eval_mode \ref CEED_EVAL_NONE to use values directly,
10917a982d89SJeremy L. Thompson                      \ref CEED_EVAL_INTERP to use interpolated values,
10927a982d89SJeremy L. Thompson                      \ref CEED_EVAL_GRAD to use gradients,
10937a982d89SJeremy L. Thompson                      \ref CEED_EVAL_WEIGHT to use quadrature weights.
10947a982d89SJeremy L. Thompson   @param[in] u     Input CeedVector
10957a982d89SJeremy L. Thompson   @param[out] v    Output CeedVector
10967a982d89SJeremy L. Thompson 
10977a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
10987a982d89SJeremy L. Thompson 
10997a982d89SJeremy L. Thompson   @ref User
11007a982d89SJeremy L. Thompson **/
1101d1d35e2fSjeremylt int CeedBasisApply(CeedBasis basis, CeedInt num_elem, CeedTransposeMode t_mode,
1102d1d35e2fSjeremylt                    CeedEvalMode eval_mode, CeedVector u, CeedVector v) {
11037a982d89SJeremy L. Thompson   int ierr;
11041f9221feSJeremy L Thompson   CeedSize u_length = 0, v_length;
11051f9221feSJeremy L Thompson   CeedInt dim, num_comp, num_nodes, num_qpts;
1106e15f9bd0SJeremy L Thompson   ierr = CeedBasisGetDimension(basis, &dim); CeedChk(ierr);
1107d1d35e2fSjeremylt   ierr = CeedBasisGetNumComponents(basis, &num_comp); CeedChk(ierr);
1108d1d35e2fSjeremylt   ierr = CeedBasisGetNumNodes(basis, &num_nodes); CeedChk(ierr);
1109d1d35e2fSjeremylt   ierr = CeedBasisGetNumQuadraturePoints(basis, &num_qpts); CeedChk(ierr);
1110d1d35e2fSjeremylt   ierr = CeedVectorGetLength(v, &v_length); CeedChk(ierr);
11117a982d89SJeremy L. Thompson   if (u) {
1112d1d35e2fSjeremylt     ierr = CeedVectorGetLength(u, &u_length); CeedChk(ierr);
11137a982d89SJeremy L. Thompson   }
11147a982d89SJeremy L. Thompson 
1115e15f9bd0SJeremy L Thompson   if (!basis->Apply)
1116e15f9bd0SJeremy L Thompson     // LCOV_EXCL_START
1117e15f9bd0SJeremy L Thompson     return CeedError(basis->ceed, CEED_ERROR_UNSUPPORTED,
1118e15f9bd0SJeremy L Thompson                      "Backend does not support BasisApply");
1119e15f9bd0SJeremy L Thompson   // LCOV_EXCL_STOP
1120e15f9bd0SJeremy L Thompson 
1121e15f9bd0SJeremy L Thompson   // Check compatibility of topological and geometrical dimensions
1122d1d35e2fSjeremylt   if ((t_mode == CEED_TRANSPOSE && (v_length%num_nodes != 0 ||
1123d1d35e2fSjeremylt                                     u_length%num_qpts != 0)) ||
1124d1d35e2fSjeremylt       (t_mode == CEED_NOTRANSPOSE && (u_length%num_nodes != 0 ||
1125d1d35e2fSjeremylt                                       v_length%num_qpts != 0)))
11268229195eSjeremylt     // LCOV_EXCL_START
1127e15f9bd0SJeremy L Thompson     return CeedError(basis->ceed, CEED_ERROR_DIMENSION,
1128e15f9bd0SJeremy L Thompson                      "Length of input/output vectors "
11297a982d89SJeremy L. Thompson                      "incompatible with basis dimensions");
11308229195eSjeremylt   // LCOV_EXCL_STOP
11317a982d89SJeremy L. Thompson 
1132e15f9bd0SJeremy L Thompson   // Check vector lengths to prevent out of bounds issues
1133d1d35e2fSjeremylt   bool bad_dims = false;
1134d1d35e2fSjeremylt   switch (eval_mode) {
1135e15f9bd0SJeremy L Thompson   case CEED_EVAL_NONE:
1136d1d35e2fSjeremylt   case CEED_EVAL_INTERP: bad_dims =
1137d1d35e2fSjeremylt       ((t_mode == CEED_TRANSPOSE && (u_length < num_elem*num_comp*num_qpts ||
1138d1d35e2fSjeremylt                                      v_length < num_elem*num_comp*num_nodes)) ||
1139d1d35e2fSjeremylt        (t_mode == CEED_NOTRANSPOSE && (v_length < num_elem*num_qpts*num_comp ||
1140d1d35e2fSjeremylt                                        u_length < num_elem*num_comp*num_nodes)));
1141e15f9bd0SJeremy L Thompson     break;
1142d1d35e2fSjeremylt   case CEED_EVAL_GRAD: bad_dims =
1143d1d35e2fSjeremylt       ((t_mode == CEED_TRANSPOSE && (u_length < num_elem*num_comp*num_qpts*dim ||
1144d1d35e2fSjeremylt                                      v_length < num_elem*num_comp*num_nodes)) ||
1145d1d35e2fSjeremylt        (t_mode == CEED_NOTRANSPOSE && (v_length < num_elem*num_qpts*num_comp*dim ||
1146d1d35e2fSjeremylt                                        u_length < num_elem*num_comp*num_nodes)));
1147e15f9bd0SJeremy L Thompson     break;
1148e15f9bd0SJeremy L Thompson   case CEED_EVAL_WEIGHT:
1149d1d35e2fSjeremylt     bad_dims = v_length < num_elem*num_qpts;
1150e15f9bd0SJeremy L Thompson     break;
1151e15f9bd0SJeremy L Thompson   // LCOV_EXCL_START
1152d1d35e2fSjeremylt   case CEED_EVAL_DIV: bad_dims =
1153d1d35e2fSjeremylt       ((t_mode == CEED_TRANSPOSE && (u_length < num_elem*num_comp*num_qpts ||
1154d1d35e2fSjeremylt                                      v_length < num_elem*num_comp*num_nodes)) ||
1155d1d35e2fSjeremylt        (t_mode == CEED_NOTRANSPOSE && (v_length < num_elem*num_qpts*num_comp ||
1156d1d35e2fSjeremylt                                        u_length < num_elem*num_comp*num_nodes)));
1157e15f9bd0SJeremy L Thompson     break;
1158d1d35e2fSjeremylt   case CEED_EVAL_CURL: bad_dims =
1159d1d35e2fSjeremylt       ((t_mode == CEED_TRANSPOSE && (u_length < num_elem*num_comp*num_qpts ||
1160d1d35e2fSjeremylt                                      v_length < num_elem*num_comp*num_nodes)) ||
1161d1d35e2fSjeremylt        (t_mode == CEED_NOTRANSPOSE && (v_length < num_elem*num_qpts*num_comp ||
1162d1d35e2fSjeremylt                                        u_length < num_elem*num_comp*num_nodes)));
1163e15f9bd0SJeremy L Thompson     break;
1164e15f9bd0SJeremy L Thompson     // LCOV_EXCL_STOP
1165e15f9bd0SJeremy L Thompson   }
1166d1d35e2fSjeremylt   if (bad_dims)
1167e15f9bd0SJeremy L Thompson     // LCOV_EXCL_START
1168e15f9bd0SJeremy L Thompson     return CeedError(basis->ceed, CEED_ERROR_DIMENSION,
1169d1d35e2fSjeremylt                      "Input/output vectors too short for basis and evaluation mode");
1170e15f9bd0SJeremy L Thompson   // LCOV_EXCL_STOP
1171e15f9bd0SJeremy L Thompson 
1172d1d35e2fSjeremylt   ierr = basis->Apply(basis, num_elem, t_mode, eval_mode, u, v); CeedChk(ierr);
1173e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
11747a982d89SJeremy L. Thompson }
11757a982d89SJeremy L. Thompson 
11767a982d89SJeremy L. Thompson /**
1177b7c9bbdaSJeremy L Thompson   @brief Get Ceed associated with a CeedBasis
1178b7c9bbdaSJeremy L Thompson 
1179b7c9bbdaSJeremy L Thompson   @param basis      CeedBasis
1180b7c9bbdaSJeremy L Thompson   @param[out] ceed  Variable to store Ceed
1181b7c9bbdaSJeremy L Thompson 
1182b7c9bbdaSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1183b7c9bbdaSJeremy L Thompson 
1184b7c9bbdaSJeremy L Thompson   @ref Advanced
1185b7c9bbdaSJeremy L Thompson **/
1186b7c9bbdaSJeremy L Thompson int CeedBasisGetCeed(CeedBasis basis, Ceed *ceed) {
1187b7c9bbdaSJeremy L Thompson   *ceed = basis->ceed;
1188b7c9bbdaSJeremy L Thompson   return CEED_ERROR_SUCCESS;
1189b7c9bbdaSJeremy L Thompson }
1190b7c9bbdaSJeremy L Thompson 
1191b7c9bbdaSJeremy L Thompson /**
11929d007619Sjeremylt   @brief Get dimension for given CeedBasis
11939d007619Sjeremylt 
11949d007619Sjeremylt   @param basis     CeedBasis
11959d007619Sjeremylt   @param[out] dim  Variable to store dimension of basis
11969d007619Sjeremylt 
11979d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
11989d007619Sjeremylt 
1199b7c9bbdaSJeremy L Thompson   @ref Advanced
12009d007619Sjeremylt **/
12019d007619Sjeremylt int CeedBasisGetDimension(CeedBasis basis, CeedInt *dim) {
12029d007619Sjeremylt   *dim = basis->dim;
1203e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
12049d007619Sjeremylt }
12059d007619Sjeremylt 
12069d007619Sjeremylt /**
1207d99fa3c5SJeremy L Thompson   @brief Get topology for given CeedBasis
1208d99fa3c5SJeremy L Thompson 
1209d99fa3c5SJeremy L Thompson   @param basis      CeedBasis
1210d99fa3c5SJeremy L Thompson   @param[out] topo  Variable to store topology of basis
1211d99fa3c5SJeremy L Thompson 
1212d99fa3c5SJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
1213d99fa3c5SJeremy L Thompson 
1214b7c9bbdaSJeremy L Thompson   @ref Advanced
1215d99fa3c5SJeremy L Thompson **/
1216d99fa3c5SJeremy L Thompson int CeedBasisGetTopology(CeedBasis basis, CeedElemTopology *topo) {
1217d99fa3c5SJeremy L Thompson   *topo = basis->topo;
1218e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1219d99fa3c5SJeremy L Thompson }
1220d99fa3c5SJeremy L Thompson 
1221d99fa3c5SJeremy L Thompson /**
122250c301a5SRezgar Shakeri   @brief Get number of Q-vector components for given CeedBasis
122350c301a5SRezgar Shakeri 
122450c301a5SRezgar Shakeri   @param basis          CeedBasis
122550c301a5SRezgar Shakeri   @param[out] Q_comp  Variable to store number of Q-vector components of basis
122650c301a5SRezgar Shakeri 
122750c301a5SRezgar Shakeri   @return An error code: 0 - success, otherwise - failure
122850c301a5SRezgar Shakeri 
122950c301a5SRezgar Shakeri   @ref Advanced
123050c301a5SRezgar Shakeri **/
123150c301a5SRezgar Shakeri int CeedBasisGetNumQuadratureComponents(CeedBasis basis, CeedInt *Q_comp) {
123250c301a5SRezgar Shakeri   *Q_comp = basis->Q_comp;
123350c301a5SRezgar Shakeri   return CEED_ERROR_SUCCESS;
123450c301a5SRezgar Shakeri }
123550c301a5SRezgar Shakeri 
123650c301a5SRezgar Shakeri /**
12379d007619Sjeremylt   @brief Get number of components for given CeedBasis
12389d007619Sjeremylt 
12399d007619Sjeremylt   @param basis          CeedBasis
1240d1d35e2fSjeremylt   @param[out] num_comp  Variable to store number of components of basis
12419d007619Sjeremylt 
12429d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
12439d007619Sjeremylt 
1244b7c9bbdaSJeremy L Thompson   @ref Advanced
12459d007619Sjeremylt **/
1246d1d35e2fSjeremylt int CeedBasisGetNumComponents(CeedBasis basis, CeedInt *num_comp) {
1247d1d35e2fSjeremylt   *num_comp = basis->num_comp;
1248e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
12499d007619Sjeremylt }
12509d007619Sjeremylt 
12519d007619Sjeremylt /**
12529d007619Sjeremylt   @brief Get total number of nodes (in dim dimensions) of a CeedBasis
12539d007619Sjeremylt 
12549d007619Sjeremylt   @param basis   CeedBasis
12559d007619Sjeremylt   @param[out] P  Variable to store number of nodes
12569d007619Sjeremylt 
12579d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
12589d007619Sjeremylt 
12599d007619Sjeremylt   @ref Utility
12609d007619Sjeremylt **/
12619d007619Sjeremylt int CeedBasisGetNumNodes(CeedBasis basis, CeedInt *P) {
12629d007619Sjeremylt   *P = basis->P;
1263e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
12649d007619Sjeremylt }
12659d007619Sjeremylt 
12669d007619Sjeremylt /**
12679d007619Sjeremylt   @brief Get total number of nodes (in 1 dimension) of a CeedBasis
12689d007619Sjeremylt 
12699d007619Sjeremylt   @param basis     CeedBasis
1270d1d35e2fSjeremylt   @param[out] P_1d  Variable to store number of nodes
12719d007619Sjeremylt 
12729d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
12739d007619Sjeremylt 
1274b7c9bbdaSJeremy L Thompson   @ref Advanced
12759d007619Sjeremylt **/
1276d1d35e2fSjeremylt int CeedBasisGetNumNodes1D(CeedBasis basis, CeedInt *P_1d) {
1277d1d35e2fSjeremylt   if (!basis->tensor_basis)
12789d007619Sjeremylt     // LCOV_EXCL_START
1279e15f9bd0SJeremy L Thompson     return CeedError(basis->ceed, CEED_ERROR_MINOR,
1280d1d35e2fSjeremylt                      "Cannot supply P_1d for non-tensor basis");
12819d007619Sjeremylt   // LCOV_EXCL_STOP
12829d007619Sjeremylt 
1283d1d35e2fSjeremylt   *P_1d = basis->P_1d;
1284e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
12859d007619Sjeremylt }
12869d007619Sjeremylt 
12879d007619Sjeremylt /**
12889d007619Sjeremylt   @brief Get total number of quadrature points (in dim dimensions) of a CeedBasis
12899d007619Sjeremylt 
12909d007619Sjeremylt   @param basis   CeedBasis
12919d007619Sjeremylt   @param[out] Q  Variable to store number of quadrature points
12929d007619Sjeremylt 
12939d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
12949d007619Sjeremylt 
12959d007619Sjeremylt   @ref Utility
12969d007619Sjeremylt **/
12979d007619Sjeremylt int CeedBasisGetNumQuadraturePoints(CeedBasis basis, CeedInt *Q) {
12989d007619Sjeremylt   *Q = basis->Q;
1299e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
13009d007619Sjeremylt }
13019d007619Sjeremylt 
13029d007619Sjeremylt /**
13039d007619Sjeremylt   @brief Get total number of quadrature points (in 1 dimension) of a CeedBasis
13049d007619Sjeremylt 
13059d007619Sjeremylt   @param basis      CeedBasis
1306d1d35e2fSjeremylt   @param[out] Q_1d  Variable to store number of quadrature points
13079d007619Sjeremylt 
13089d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
13099d007619Sjeremylt 
1310b7c9bbdaSJeremy L Thompson   @ref Advanced
13119d007619Sjeremylt **/
1312d1d35e2fSjeremylt int CeedBasisGetNumQuadraturePoints1D(CeedBasis basis, CeedInt *Q_1d) {
1313d1d35e2fSjeremylt   if (!basis->tensor_basis)
13149d007619Sjeremylt     // LCOV_EXCL_START
1315e15f9bd0SJeremy L Thompson     return CeedError(basis->ceed, CEED_ERROR_MINOR,
1316d1d35e2fSjeremylt                      "Cannot supply Q_1d for non-tensor basis");
13179d007619Sjeremylt   // LCOV_EXCL_STOP
13189d007619Sjeremylt 
1319d1d35e2fSjeremylt   *Q_1d = basis->Q_1d;
1320e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
13219d007619Sjeremylt }
13229d007619Sjeremylt 
13239d007619Sjeremylt /**
13249d007619Sjeremylt   @brief Get reference coordinates of quadrature points (in dim dimensions)
13259d007619Sjeremylt          of a CeedBasis
13269d007619Sjeremylt 
13279d007619Sjeremylt   @param basis       CeedBasis
1328d1d35e2fSjeremylt   @param[out] q_ref  Variable to store reference coordinates of quadrature points
13299d007619Sjeremylt 
13309d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
13319d007619Sjeremylt 
1332b7c9bbdaSJeremy L Thompson   @ref Advanced
13339d007619Sjeremylt **/
1334d1d35e2fSjeremylt int CeedBasisGetQRef(CeedBasis basis, const CeedScalar **q_ref) {
1335d1d35e2fSjeremylt   *q_ref = basis->q_ref_1d;
1336e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
13379d007619Sjeremylt }
13389d007619Sjeremylt 
13399d007619Sjeremylt /**
13409d007619Sjeremylt   @brief Get quadrature weights of quadrature points (in dim dimensions)
13419d007619Sjeremylt          of a CeedBasis
13429d007619Sjeremylt 
13439d007619Sjeremylt   @param basis          CeedBasis
1344d1d35e2fSjeremylt   @param[out] q_weight  Variable to store quadrature weights
13459d007619Sjeremylt 
13469d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
13479d007619Sjeremylt 
1348b7c9bbdaSJeremy L Thompson   @ref Advanced
13499d007619Sjeremylt **/
1350d1d35e2fSjeremylt int CeedBasisGetQWeights(CeedBasis basis, const CeedScalar **q_weight) {
1351d1d35e2fSjeremylt   *q_weight = basis->q_weight_1d;
1352e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
13539d007619Sjeremylt }
13549d007619Sjeremylt 
13559d007619Sjeremylt /**
13569d007619Sjeremylt   @brief Get interpolation matrix of a CeedBasis
13579d007619Sjeremylt 
13589d007619Sjeremylt   @param basis        CeedBasis
13599d007619Sjeremylt   @param[out] interp  Variable to store interpolation matrix
13609d007619Sjeremylt 
13619d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
13629d007619Sjeremylt 
1363b7c9bbdaSJeremy L Thompson   @ref Advanced
13649d007619Sjeremylt **/
13656c58de82SJeremy L Thompson int CeedBasisGetInterp(CeedBasis basis, const CeedScalar **interp) {
1366d1d35e2fSjeremylt   if (!basis->interp && basis->tensor_basis) {
13679d007619Sjeremylt     // Allocate
13689d007619Sjeremylt     int ierr;
13699d007619Sjeremylt     ierr = CeedMalloc(basis->Q*basis->P, &basis->interp); CeedChk(ierr);
13709d007619Sjeremylt 
13719d007619Sjeremylt     // Initialize
13729d007619Sjeremylt     for (CeedInt i=0; i<basis->Q*basis->P; i++)
13739d007619Sjeremylt       basis->interp[i] = 1.0;
13749d007619Sjeremylt 
13759d007619Sjeremylt     // Calculate
13769d007619Sjeremylt     for (CeedInt d=0; d<basis->dim; d++)
13779d007619Sjeremylt       for (CeedInt qpt=0; qpt<basis->Q; qpt++)
13789d007619Sjeremylt         for (CeedInt node=0; node<basis->P; node++) {
1379d1d35e2fSjeremylt           CeedInt p = (node / CeedIntPow(basis->P_1d, d)) % basis->P_1d;
1380d1d35e2fSjeremylt           CeedInt q = (qpt / CeedIntPow(basis->Q_1d, d)) % basis->Q_1d;
1381d1d35e2fSjeremylt           basis->interp[qpt*(basis->P)+node] *= basis->interp_1d[q*basis->P_1d+p];
13829d007619Sjeremylt         }
13839d007619Sjeremylt   }
13849d007619Sjeremylt   *interp = basis->interp;
1385e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
13869d007619Sjeremylt }
13879d007619Sjeremylt 
13889d007619Sjeremylt /**
13899d007619Sjeremylt   @brief Get 1D interpolation matrix of a tensor product CeedBasis
13909d007619Sjeremylt 
13919d007619Sjeremylt   @param basis           CeedBasis
1392d1d35e2fSjeremylt   @param[out] interp_1d  Variable to store interpolation matrix
13939d007619Sjeremylt 
13949d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
13959d007619Sjeremylt 
13969d007619Sjeremylt   @ref Backend
13979d007619Sjeremylt **/
1398d1d35e2fSjeremylt int CeedBasisGetInterp1D(CeedBasis basis, const CeedScalar **interp_1d) {
1399d1d35e2fSjeremylt   if (!basis->tensor_basis)
14009d007619Sjeremylt     // LCOV_EXCL_START
1401e15f9bd0SJeremy L Thompson     return CeedError(basis->ceed, CEED_ERROR_MINOR,
1402e15f9bd0SJeremy L Thompson                      "CeedBasis is not a tensor product basis.");
14039d007619Sjeremylt   // LCOV_EXCL_STOP
14049d007619Sjeremylt 
1405d1d35e2fSjeremylt   *interp_1d = basis->interp_1d;
1406e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
14079d007619Sjeremylt }
14089d007619Sjeremylt 
14099d007619Sjeremylt /**
14109d007619Sjeremylt   @brief Get gradient matrix of a CeedBasis
14119d007619Sjeremylt 
14129d007619Sjeremylt   @param basis      CeedBasis
14139d007619Sjeremylt   @param[out] grad  Variable to store gradient matrix
14149d007619Sjeremylt 
14159d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
14169d007619Sjeremylt 
1417b7c9bbdaSJeremy L Thompson   @ref Advanced
14189d007619Sjeremylt **/
14196c58de82SJeremy L Thompson int CeedBasisGetGrad(CeedBasis basis, const CeedScalar **grad) {
1420d1d35e2fSjeremylt   if (!basis->grad && basis->tensor_basis) {
14219d007619Sjeremylt     // Allocate
14229d007619Sjeremylt     int ierr;
14239d007619Sjeremylt     ierr = CeedMalloc(basis->dim*basis->Q*basis->P, &basis->grad);
14249d007619Sjeremylt     CeedChk(ierr);
14259d007619Sjeremylt 
14269d007619Sjeremylt     // Initialize
14279d007619Sjeremylt     for (CeedInt i=0; i<basis->dim*basis->Q*basis->P; i++)
14289d007619Sjeremylt       basis->grad[i] = 1.0;
14299d007619Sjeremylt 
14309d007619Sjeremylt     // Calculate
14319d007619Sjeremylt     for (CeedInt d=0; d<basis->dim; d++)
14329d007619Sjeremylt       for (CeedInt i=0; i<basis->dim; i++)
14339d007619Sjeremylt         for (CeedInt qpt=0; qpt<basis->Q; qpt++)
14349d007619Sjeremylt           for (CeedInt node=0; node<basis->P; node++) {
1435d1d35e2fSjeremylt             CeedInt p = (node / CeedIntPow(basis->P_1d, d)) % basis->P_1d;
1436d1d35e2fSjeremylt             CeedInt q = (qpt / CeedIntPow(basis->Q_1d, d)) % basis->Q_1d;
14379d007619Sjeremylt             if (i == d)
14389d007619Sjeremylt               basis->grad[(i*basis->Q+qpt)*(basis->P)+node] *=
1439d1d35e2fSjeremylt                 basis->grad_1d[q*basis->P_1d+p];
14409d007619Sjeremylt             else
14419d007619Sjeremylt               basis->grad[(i*basis->Q+qpt)*(basis->P)+node] *=
1442d1d35e2fSjeremylt                 basis->interp_1d[q*basis->P_1d+p];
14439d007619Sjeremylt           }
14449d007619Sjeremylt   }
14459d007619Sjeremylt   *grad = basis->grad;
1446e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
14479d007619Sjeremylt }
14489d007619Sjeremylt 
14499d007619Sjeremylt /**
14509d007619Sjeremylt   @brief Get 1D gradient matrix of a tensor product CeedBasis
14519d007619Sjeremylt 
14529d007619Sjeremylt   @param basis         CeedBasis
1453d1d35e2fSjeremylt   @param[out] grad_1d  Variable to store gradient matrix
14549d007619Sjeremylt 
14559d007619Sjeremylt   @return An error code: 0 - success, otherwise - failure
14569d007619Sjeremylt 
1457b7c9bbdaSJeremy L Thompson   @ref Advanced
14589d007619Sjeremylt **/
1459d1d35e2fSjeremylt int CeedBasisGetGrad1D(CeedBasis basis, const CeedScalar **grad_1d) {
1460d1d35e2fSjeremylt   if (!basis->tensor_basis)
14619d007619Sjeremylt     // LCOV_EXCL_START
1462e15f9bd0SJeremy L Thompson     return CeedError(basis->ceed, CEED_ERROR_MINOR,
1463e15f9bd0SJeremy L Thompson                      "CeedBasis is not a tensor product basis.");
14649d007619Sjeremylt   // LCOV_EXCL_STOP
14659d007619Sjeremylt 
1466d1d35e2fSjeremylt   *grad_1d = basis->grad_1d;
1467e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
14689d007619Sjeremylt }
14699d007619Sjeremylt 
14709d007619Sjeremylt /**
147150c301a5SRezgar Shakeri   @brief Get divergence matrix of a CeedBasis
147250c301a5SRezgar Shakeri 
147350c301a5SRezgar Shakeri   @param basis     CeedBasis
147450c301a5SRezgar Shakeri   @param[out] div  Variable to store divergence matrix
147550c301a5SRezgar Shakeri 
147650c301a5SRezgar Shakeri   @return An error code: 0 - success, otherwise - failure
147750c301a5SRezgar Shakeri 
147850c301a5SRezgar Shakeri   @ref Advanced
147950c301a5SRezgar Shakeri **/
148050c301a5SRezgar Shakeri int CeedBasisGetDiv(CeedBasis basis, const CeedScalar **div) {
148150c301a5SRezgar Shakeri   if (!basis->div)
148250c301a5SRezgar Shakeri     // LCOV_EXCL_START
148350c301a5SRezgar Shakeri     return CeedError(basis->ceed, CEED_ERROR_MINOR,
148450c301a5SRezgar Shakeri                      "CeedBasis does not have divergence matrix.");
148550c301a5SRezgar Shakeri   // LCOV_EXCL_STOP
148650c301a5SRezgar Shakeri 
148750c301a5SRezgar Shakeri   *div = basis->div;
148850c301a5SRezgar Shakeri   return CEED_ERROR_SUCCESS;
148950c301a5SRezgar Shakeri }
149050c301a5SRezgar Shakeri 
149150c301a5SRezgar Shakeri /**
14927a982d89SJeremy L. Thompson   @brief Destroy a CeedBasis
14937a982d89SJeremy L. Thompson 
14947a982d89SJeremy L. Thompson   @param basis CeedBasis to destroy
14957a982d89SJeremy L. Thompson 
14967a982d89SJeremy L. Thompson   @return An error code: 0 - success, otherwise - failure
14977a982d89SJeremy L. Thompson 
14987a982d89SJeremy L. Thompson   @ref User
14997a982d89SJeremy L. Thompson **/
15007a982d89SJeremy L. Thompson int CeedBasisDestroy(CeedBasis *basis) {
15017a982d89SJeremy L. Thompson   int ierr;
15027a982d89SJeremy L. Thompson 
1503d1d35e2fSjeremylt   if (!*basis || --(*basis)->ref_count > 0) return CEED_ERROR_SUCCESS;
15047a982d89SJeremy L. Thompson   if ((*basis)->Destroy) {
15057a982d89SJeremy L. Thompson     ierr = (*basis)->Destroy(*basis); CeedChk(ierr);
15067a982d89SJeremy L. Thompson   }
150734359f16Sjeremylt   if ((*basis)->contract) {
150834359f16Sjeremylt     ierr = CeedTensorContractDestroy(&(*basis)->contract); CeedChk(ierr);
150934359f16Sjeremylt   }
15107a982d89SJeremy L. Thompson   ierr = CeedFree(&(*basis)->interp); CeedChk(ierr);
1511d1d35e2fSjeremylt   ierr = CeedFree(&(*basis)->interp_1d); CeedChk(ierr);
15127a982d89SJeremy L. Thompson   ierr = CeedFree(&(*basis)->grad); CeedChk(ierr);
151350c301a5SRezgar Shakeri   ierr = CeedFree(&(*basis)->div); CeedChk(ierr);
1514d1d35e2fSjeremylt   ierr = CeedFree(&(*basis)->grad_1d); CeedChk(ierr);
1515d1d35e2fSjeremylt   ierr = CeedFree(&(*basis)->q_ref_1d); CeedChk(ierr);
1516d1d35e2fSjeremylt   ierr = CeedFree(&(*basis)->q_weight_1d); CeedChk(ierr);
15177a982d89SJeremy L. Thompson   ierr = CeedDestroy(&(*basis)->ceed); CeedChk(ierr);
15187a982d89SJeremy L. Thompson   ierr = CeedFree(basis); CeedChk(ierr);
1519e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
15207a982d89SJeremy L. Thompson }
15217a982d89SJeremy L. Thompson 
15227a982d89SJeremy L. Thompson /**
1523b11c1e72Sjeremylt   @brief Construct a Gauss-Legendre quadrature
1524b11c1e72Sjeremylt 
1525b11c1e72Sjeremylt   @param Q                 Number of quadrature points (integrates polynomials of
1526b11c1e72Sjeremylt                              degree 2*Q-1 exactly)
1527d1d35e2fSjeremylt   @param[out] q_ref_1d     Array of length Q to hold the abscissa on [-1, 1]
1528d1d35e2fSjeremylt   @param[out] q_weight_1d  Array of length Q to hold the weights
1529b11c1e72Sjeremylt 
1530b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
1531dfdf5a53Sjeremylt 
1532dfdf5a53Sjeremylt   @ref Utility
1533b11c1e72Sjeremylt **/
1534d1d35e2fSjeremylt int CeedGaussQuadrature(CeedInt Q, CeedScalar *q_ref_1d,
1535d1d35e2fSjeremylt                         CeedScalar *q_weight_1d) {
1536d7b241e6Sjeremylt   // Allocate
1537d7b241e6Sjeremylt   CeedScalar P0, P1, P2, dP2, xi, wi, PI = 4.0*atan(1.0);
1538d1d35e2fSjeremylt   // Build q_ref_1d, q_weight_1d
153992ae7e47SJeremy L Thompson   for (CeedInt i = 0; i <= Q/2; i++) {
1540d7b241e6Sjeremylt     // Guess
1541d7b241e6Sjeremylt     xi = cos(PI*(CeedScalar)(2*i+1)/((CeedScalar)(2*Q)));
1542d7b241e6Sjeremylt     // Pn(xi)
1543d7b241e6Sjeremylt     P0 = 1.0;
1544d7b241e6Sjeremylt     P1 = xi;
1545d7b241e6Sjeremylt     P2 = 0.0;
154692ae7e47SJeremy L Thompson     for (CeedInt j = 2; j <= Q; j++) {
1547d7b241e6Sjeremylt       P2 = (((CeedScalar)(2*j-1))*xi*P1-((CeedScalar)(j-1))*P0)/((CeedScalar)(j));
1548d7b241e6Sjeremylt       P0 = P1;
1549d7b241e6Sjeremylt       P1 = P2;
1550d7b241e6Sjeremylt     }
1551d7b241e6Sjeremylt     // First Newton Step
1552d7b241e6Sjeremylt     dP2 = (xi*P2 - P0)*(CeedScalar)Q/(xi*xi-1.0);
1553d7b241e6Sjeremylt     xi = xi-P2/dP2;
1554d7b241e6Sjeremylt     // Newton to convergence
155592ae7e47SJeremy L Thompson     for (CeedInt k=0; k<100 && fabs(P2)>10*CEED_EPSILON; k++) {
1556d7b241e6Sjeremylt       P0 = 1.0;
1557d7b241e6Sjeremylt       P1 = xi;
155892ae7e47SJeremy L Thompson       for (CeedInt j = 2; j <= Q; j++) {
1559d7b241e6Sjeremylt         P2 = (((CeedScalar)(2*j-1))*xi*P1-((CeedScalar)(j-1))*P0)/((CeedScalar)(j));
1560d7b241e6Sjeremylt         P0 = P1;
1561d7b241e6Sjeremylt         P1 = P2;
1562d7b241e6Sjeremylt       }
1563d7b241e6Sjeremylt       dP2 = (xi*P2 - P0)*(CeedScalar)Q/(xi*xi-1.0);
1564d7b241e6Sjeremylt       xi = xi-P2/dP2;
1565d7b241e6Sjeremylt     }
1566d7b241e6Sjeremylt     // Save xi, wi
1567d7b241e6Sjeremylt     wi = 2.0/((1.0-xi*xi)*dP2*dP2);
1568d1d35e2fSjeremylt     q_weight_1d[i] = wi;
1569d1d35e2fSjeremylt     q_weight_1d[Q-1-i] = wi;
1570d1d35e2fSjeremylt     q_ref_1d[i] = -xi;
1571d1d35e2fSjeremylt     q_ref_1d[Q-1-i]= xi;
1572d7b241e6Sjeremylt   }
1573e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1574d7b241e6Sjeremylt }
1575d7b241e6Sjeremylt 
1576b11c1e72Sjeremylt /**
1577b11c1e72Sjeremylt   @brief Construct a Gauss-Legendre-Lobatto quadrature
1578b11c1e72Sjeremylt 
1579b11c1e72Sjeremylt   @param Q                 Number of quadrature points (integrates polynomials of
1580b11c1e72Sjeremylt                              degree 2*Q-3 exactly)
1581d1d35e2fSjeremylt   @param[out] q_ref_1d     Array of length Q to hold the abscissa on [-1, 1]
1582d1d35e2fSjeremylt   @param[out] q_weight_1d  Array of length Q to hold the weights
1583b11c1e72Sjeremylt 
1584b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
1585dfdf5a53Sjeremylt 
1586dfdf5a53Sjeremylt   @ref Utility
1587b11c1e72Sjeremylt **/
1588d1d35e2fSjeremylt int CeedLobattoQuadrature(CeedInt Q, CeedScalar *q_ref_1d,
1589d1d35e2fSjeremylt                           CeedScalar *q_weight_1d) {
1590d7b241e6Sjeremylt   // Allocate
1591d7b241e6Sjeremylt   CeedScalar P0, P1, P2, dP2, d2P2, xi, wi, PI = 4.0*atan(1.0);
1592d1d35e2fSjeremylt   // Build q_ref_1d, q_weight_1d
1593d7b241e6Sjeremylt   // Set endpoints
159430a100c3SJed Brown   if (Q < 2)
1595b0d62198Sjeremylt     // LCOV_EXCL_START
1596e15f9bd0SJeremy L Thompson     return CeedError(NULL, CEED_ERROR_DIMENSION,
15977ed52d01Sjeremylt                      "Cannot create Lobatto quadrature with Q=%d < 2 points", Q);
1598b0d62198Sjeremylt   // LCOV_EXCL_STOP
1599d7b241e6Sjeremylt   wi = 2.0/((CeedScalar)(Q*(Q-1)));
1600d1d35e2fSjeremylt   if (q_weight_1d) {
1601d1d35e2fSjeremylt     q_weight_1d[0] = wi;
1602d1d35e2fSjeremylt     q_weight_1d[Q-1] = wi;
1603d7b241e6Sjeremylt   }
1604d1d35e2fSjeremylt   q_ref_1d[0] = -1.0;
1605d1d35e2fSjeremylt   q_ref_1d[Q-1] = 1.0;
1606d7b241e6Sjeremylt   // Interior
160792ae7e47SJeremy L Thompson   for (CeedInt i = 1; i <= (Q-1)/2; i++) {
1608d7b241e6Sjeremylt     // Guess
1609d7b241e6Sjeremylt     xi = cos(PI*(CeedScalar)(i)/(CeedScalar)(Q-1));
1610d7b241e6Sjeremylt     // Pn(xi)
1611d7b241e6Sjeremylt     P0 = 1.0;
1612d7b241e6Sjeremylt     P1 = xi;
1613d7b241e6Sjeremylt     P2 = 0.0;
161492ae7e47SJeremy L Thompson     for (CeedInt j = 2; j < Q; j++) {
1615d7b241e6Sjeremylt       P2 = (((CeedScalar)(2*j-1))*xi*P1-((CeedScalar)(j-1))*P0)/((CeedScalar)(j));
1616d7b241e6Sjeremylt       P0 = P1;
1617d7b241e6Sjeremylt       P1 = P2;
1618d7b241e6Sjeremylt     }
1619d7b241e6Sjeremylt     // First Newton step
1620d7b241e6Sjeremylt     dP2 = (xi*P2 - P0)*(CeedScalar)Q/(xi*xi-1.0);
1621d7b241e6Sjeremylt     d2P2 = (2*xi*dP2 - (CeedScalar)(Q*(Q-1))*P2)/(1.0-xi*xi);
1622d7b241e6Sjeremylt     xi = xi-dP2/d2P2;
1623d7b241e6Sjeremylt     // Newton to convergence
162492ae7e47SJeremy L Thompson     for (CeedInt k=0; k<100 && fabs(dP2)>10*CEED_EPSILON; k++) {
1625d7b241e6Sjeremylt       P0 = 1.0;
1626d7b241e6Sjeremylt       P1 = xi;
162792ae7e47SJeremy L Thompson       for (CeedInt j = 2; j < Q; j++) {
1628d7b241e6Sjeremylt         P2 = (((CeedScalar)(2*j-1))*xi*P1-((CeedScalar)(j-1))*P0)/((CeedScalar)(j));
1629d7b241e6Sjeremylt         P0 = P1;
1630d7b241e6Sjeremylt         P1 = P2;
1631d7b241e6Sjeremylt       }
1632d7b241e6Sjeremylt       dP2 = (xi*P2 - P0)*(CeedScalar)Q/(xi*xi-1.0);
1633d7b241e6Sjeremylt       d2P2 = (2*xi*dP2 - (CeedScalar)(Q*(Q-1))*P2)/(1.0-xi*xi);
1634d7b241e6Sjeremylt       xi = xi-dP2/d2P2;
1635d7b241e6Sjeremylt     }
1636d7b241e6Sjeremylt     // Save xi, wi
1637d7b241e6Sjeremylt     wi = 2.0/(((CeedScalar)(Q*(Q-1)))*P2*P2);
1638d1d35e2fSjeremylt     if (q_weight_1d) {
1639d1d35e2fSjeremylt       q_weight_1d[i] = wi;
1640d1d35e2fSjeremylt       q_weight_1d[Q-1-i] = wi;
1641d7b241e6Sjeremylt     }
1642d1d35e2fSjeremylt     q_ref_1d[i] = -xi;
1643d1d35e2fSjeremylt     q_ref_1d[Q-1-i]= xi;
1644d7b241e6Sjeremylt   }
1645e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1646d7b241e6Sjeremylt }
1647d7b241e6Sjeremylt 
1648dfdf5a53Sjeremylt /**
164995bb1877Svaleriabarra   @brief Return QR Factorization of a matrix
1650b11c1e72Sjeremylt 
165177645d7bSjeremylt   @param ceed         A Ceed context for error handling
165252bfb9bbSJeremy L Thompson   @param[in,out] mat  Row-major matrix to be factorized in place
165352bfb9bbSJeremy L Thompson   @param[in,out] tau  Vector of length m of scaling factors
1654b11c1e72Sjeremylt   @param m            Number of rows
1655b11c1e72Sjeremylt   @param n            Number of columns
1656b11c1e72Sjeremylt 
1657b11c1e72Sjeremylt   @return An error code: 0 - success, otherwise - failure
1658dfdf5a53Sjeremylt 
1659dfdf5a53Sjeremylt   @ref Utility
1660b11c1e72Sjeremylt **/
1661a7bd39daSjeremylt int CeedQRFactorization(Ceed ceed, CeedScalar *mat, CeedScalar *tau,
1662d7b241e6Sjeremylt                         CeedInt m, CeedInt n) {
1663d7b241e6Sjeremylt   CeedScalar v[m];
1664d7b241e6Sjeremylt 
1665a7bd39daSjeremylt   // Check m >= n
1666a7bd39daSjeremylt   if (n > m)
1667c042f62fSJeremy L Thompson     // LCOV_EXCL_START
1668e15f9bd0SJeremy L Thompson     return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
1669e15f9bd0SJeremy L Thompson                      "Cannot compute QR factorization with n > m");
1670c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
1671a7bd39daSjeremylt 
167252bfb9bbSJeremy L Thompson   for (CeedInt i=0; i<n; i++) {
1673bde37e8cSJed Brown     if (i >= m-1) { // last row of matrix, no reflection needed
1674bde37e8cSJed Brown       tau[i] = 0.;
1675bde37e8cSJed Brown       break;
1676bde37e8cSJed Brown     }
1677d7b241e6Sjeremylt     // Calculate Householder vector, magnitude
1678d7b241e6Sjeremylt     CeedScalar sigma = 0.0;
1679d7b241e6Sjeremylt     v[i] = mat[i+n*i];
168052bfb9bbSJeremy L Thompson     for (CeedInt j=i+1; j<m; j++) {
1681d7b241e6Sjeremylt       v[j] = mat[i+n*j];
1682d7b241e6Sjeremylt       sigma += v[j] * v[j];
1683d7b241e6Sjeremylt     }
1684d7b241e6Sjeremylt     CeedScalar norm = sqrt(v[i]*v[i] + sigma); // norm of v[i:m]
1685673160d7Sjeremylt     CeedScalar R_ii = -copysign(norm, v[i]);
1686673160d7Sjeremylt     v[i] -= R_ii;
1687d7b241e6Sjeremylt     // norm of v[i:m] after modification above and scaling below
1688d7b241e6Sjeremylt     //   norm = sqrt(v[i]*v[i] + sigma) / v[i];
1689d7b241e6Sjeremylt     //   tau = 2 / (norm*norm)
1690d7b241e6Sjeremylt     tau[i] = 2 * v[i]*v[i] / (v[i]*v[i] + sigma);
16911d102b48SJeremy L Thompson     for (CeedInt j=i+1; j<m; j++)
16921d102b48SJeremy L Thompson       v[j] /= v[i];
1693d7b241e6Sjeremylt 
1694d7b241e6Sjeremylt     // Apply Householder reflector to lower right panel
1695d7b241e6Sjeremylt     CeedHouseholderReflect(&mat[i*n+i+1], &v[i], tau[i], m-i, n-i-1, n, 1);
1696d7b241e6Sjeremylt     // Save v
1697673160d7Sjeremylt     mat[i+n*i] = R_ii;
16981d102b48SJeremy L Thompson     for (CeedInt j=i+1; j<m; j++)
1699d7b241e6Sjeremylt       mat[i+n*j] = v[j];
1700d7b241e6Sjeremylt   }
1701e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
1702d7b241e6Sjeremylt }
1703d7b241e6Sjeremylt 
1704b11c1e72Sjeremylt /**
170552bfb9bbSJeremy L Thompson   @brief Return symmetric Schur decomposition of the symmetric matrix mat via
170652bfb9bbSJeremy L Thompson            symmetric QR factorization
170752bfb9bbSJeremy L Thompson 
170877645d7bSjeremylt   @param ceed         A Ceed context for error handling
170952bfb9bbSJeremy L Thompson   @param[in,out] mat  Row-major matrix to be factorized in place
1710460bf743SValeria Barra   @param[out] lambda  Vector of length n of eigenvalues
171152bfb9bbSJeremy L Thompson   @param n            Number of rows/columns
171252bfb9bbSJeremy L Thompson 
171352bfb9bbSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
171452bfb9bbSJeremy L Thompson 
171552bfb9bbSJeremy L Thompson   @ref Utility
171652bfb9bbSJeremy L Thompson **/
171703d18186Sjeremylt CeedPragmaOptimizeOff
171852bfb9bbSJeremy L Thompson int CeedSymmetricSchurDecomposition(Ceed ceed, CeedScalar *mat,
171952bfb9bbSJeremy L Thompson                                     CeedScalar *lambda, CeedInt n) {
172052bfb9bbSJeremy L Thompson   // Check bounds for clang-tidy
172152bfb9bbSJeremy L Thompson   if (n<2)
1722c042f62fSJeremy L Thompson     // LCOV_EXCL_START
1723e15f9bd0SJeremy L Thompson     return CeedError(ceed, CEED_ERROR_UNSUPPORTED,
1724c042f62fSJeremy L Thompson                      "Cannot compute symmetric Schur decomposition of scalars");
1725c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
172652bfb9bbSJeremy L Thompson 
1727673160d7Sjeremylt   CeedScalar v[n-1], tau[n-1], mat_T[n*n];
172852bfb9bbSJeremy L Thompson 
1729673160d7Sjeremylt   // Copy mat to mat_T and set mat to I
1730673160d7Sjeremylt   memcpy(mat_T, mat, n*n*sizeof(mat[0]));
173152bfb9bbSJeremy L Thompson   for (CeedInt i=0; i<n; i++)
173252bfb9bbSJeremy L Thompson     for (CeedInt j=0; j<n; j++)
173352bfb9bbSJeremy L Thompson       mat[j+n*i] = (i==j) ? 1 : 0;
173452bfb9bbSJeremy L Thompson 
173552bfb9bbSJeremy L Thompson   // Reduce to tridiagonal
173652bfb9bbSJeremy L Thompson   for (CeedInt i=0; i<n-1; i++) {
173752bfb9bbSJeremy L Thompson     // Calculate Householder vector, magnitude
173852bfb9bbSJeremy L Thompson     CeedScalar sigma = 0.0;
1739673160d7Sjeremylt     v[i] = mat_T[i+n*(i+1)];
174052bfb9bbSJeremy L Thompson     for (CeedInt j=i+1; j<n-1; j++) {
1741673160d7Sjeremylt       v[j] = mat_T[i+n*(j+1)];
174252bfb9bbSJeremy L Thompson       sigma += v[j] * v[j];
174352bfb9bbSJeremy L Thompson     }
174452bfb9bbSJeremy L Thompson     CeedScalar norm = sqrt(v[i]*v[i] + sigma); // norm of v[i:n-1]
1745673160d7Sjeremylt     CeedScalar R_ii = -copysign(norm, v[i]);
1746673160d7Sjeremylt     v[i] -= R_ii;
174752bfb9bbSJeremy L Thompson     // norm of v[i:m] after modification above and scaling below
174852bfb9bbSJeremy L Thompson     //   norm = sqrt(v[i]*v[i] + sigma) / v[i];
174952bfb9bbSJeremy L Thompson     //   tau = 2 / (norm*norm)
175080a9ef05SNatalie Beams     tau[i] = i == n - 2 ? 2 : 2 * v[i]*v[i] / (v[i]*v[i] + sigma);
1751fb551037Sjeremylt     for (CeedInt j=i+1; j<n-1; j++)
1752fb551037Sjeremylt       v[j] /= v[i];
175352bfb9bbSJeremy L Thompson 
175452bfb9bbSJeremy L Thompson     // Update sub and super diagonal
175552bfb9bbSJeremy L Thompson     for (CeedInt j=i+2; j<n; j++) {
1756673160d7Sjeremylt       mat_T[i+n*j] = 0; mat_T[j+n*i] = 0;
175752bfb9bbSJeremy L Thompson     }
175852bfb9bbSJeremy L Thompson     // Apply symmetric Householder reflector to lower right panel
1759673160d7Sjeremylt     CeedHouseholderReflect(&mat_T[(i+1)+n*(i+1)], &v[i], tau[i],
176052bfb9bbSJeremy L Thompson                            n-(i+1), n-(i+1), n, 1);
1761673160d7Sjeremylt     CeedHouseholderReflect(&mat_T[(i+1)+n*(i+1)], &v[i], tau[i],
176252bfb9bbSJeremy L Thompson                            n-(i+1), n-(i+1), 1, n);
1763673160d7Sjeremylt 
176452bfb9bbSJeremy L Thompson     // Save v
1765673160d7Sjeremylt     mat_T[i+n*(i+1)] = R_ii;
1766673160d7Sjeremylt     mat_T[(i+1)+n*i] = R_ii;
176752bfb9bbSJeremy L Thompson     for (CeedInt j=i+1; j<n-1; j++) {
1768673160d7Sjeremylt       mat_T[i+n*(j+1)] = v[j];
176952bfb9bbSJeremy L Thompson     }
177052bfb9bbSJeremy L Thompson   }
177152bfb9bbSJeremy L Thompson   // Backwards accumulation of Q
177252bfb9bbSJeremy L Thompson   for (CeedInt i=n-2; i>=0; i--) {
177385cf89eaSjeremylt     if (tau[i] > 0.0) {
177452bfb9bbSJeremy L Thompson       v[i] = 1;
177552bfb9bbSJeremy L Thompson       for (CeedInt j=i+1; j<n-1; j++) {
1776673160d7Sjeremylt         v[j] = mat_T[i+n*(j+1)];
1777673160d7Sjeremylt         mat_T[i+n*(j+1)] = 0;
177852bfb9bbSJeremy L Thompson       }
177952bfb9bbSJeremy L Thompson       CeedHouseholderReflect(&mat[(i+1)+n*(i+1)], &v[i], tau[i],
178052bfb9bbSJeremy L Thompson                              n-(i+1), n-(i+1), n, 1);
178152bfb9bbSJeremy L Thompson     }
178285cf89eaSjeremylt   }
178352bfb9bbSJeremy L Thompson 
178452bfb9bbSJeremy L Thompson   // Reduce sub and super diagonal
1785673160d7Sjeremylt   CeedInt p = 0, q = 0, itr = 0, max_itr = n*n*n*n;
1786673160d7Sjeremylt   CeedScalar tol = CEED_EPSILON;
178752bfb9bbSJeremy L Thompson 
1788673160d7Sjeremylt   while (itr < max_itr) {
178952bfb9bbSJeremy L Thompson     // Update p, q, size of reduced portions of diagonal
179052bfb9bbSJeremy L Thompson     p = 0; q = 0;
179152bfb9bbSJeremy L Thompson     for (CeedInt i=n-2; i>=0; i--) {
1792673160d7Sjeremylt       if (fabs(mat_T[i+n*(i+1)]) < tol)
179352bfb9bbSJeremy L Thompson         q += 1;
179452bfb9bbSJeremy L Thompson       else
179552bfb9bbSJeremy L Thompson         break;
179652bfb9bbSJeremy L Thompson     }
1797673160d7Sjeremylt     for (CeedInt i=0; i<n-q-1; i++) {
1798673160d7Sjeremylt       if (fabs(mat_T[i+n*(i+1)]) < tol)
179952bfb9bbSJeremy L Thompson         p += 1;
180052bfb9bbSJeremy L Thompson       else
180152bfb9bbSJeremy L Thompson         break;
180252bfb9bbSJeremy L Thompson     }
180352bfb9bbSJeremy L Thompson     if (q == n-1) break; // Finished reducing
180452bfb9bbSJeremy L Thompson 
180552bfb9bbSJeremy L Thompson     // Reduce tridiagonal portion
1806673160d7Sjeremylt     CeedScalar t_nn = mat_T[(n-1-q)+n*(n-1-q)],
1807673160d7Sjeremylt                t_nnm1 = mat_T[(n-2-q)+n*(n-1-q)];
1808673160d7Sjeremylt     CeedScalar d = (mat_T[(n-2-q)+n*(n-2-q)] - t_nn)/2;
1809673160d7Sjeremylt     CeedScalar mu = t_nn - t_nnm1*t_nnm1 /
1810673160d7Sjeremylt                     (d + copysign(sqrt(d*d + t_nnm1*t_nnm1), d));
1811673160d7Sjeremylt     CeedScalar x = mat_T[p+n*p] - mu;
1812673160d7Sjeremylt     CeedScalar z = mat_T[p+n*(p+1)];
1813673160d7Sjeremylt     for (CeedInt k=p; k<n-q-1; k++) {
181452bfb9bbSJeremy L Thompson       // Compute Givens rotation
181552bfb9bbSJeremy L Thompson       CeedScalar c = 1, s = 0;
181652bfb9bbSJeremy L Thompson       if (fabs(z) > tol) {
181752bfb9bbSJeremy L Thompson         if (fabs(z) > fabs(x)) {
181852bfb9bbSJeremy L Thompson           CeedScalar tau = -x/z;
181952bfb9bbSJeremy L Thompson           s = 1/sqrt(1+tau*tau), c = s*tau;
182052bfb9bbSJeremy L Thompson         } else {
182152bfb9bbSJeremy L Thompson           CeedScalar tau = -z/x;
182252bfb9bbSJeremy L Thompson           c = 1/sqrt(1+tau*tau), s = c*tau;
182352bfb9bbSJeremy L Thompson         }
182452bfb9bbSJeremy L Thompson       }
182552bfb9bbSJeremy L Thompson 
182652bfb9bbSJeremy L Thompson       // Apply Givens rotation to T
1827673160d7Sjeremylt       CeedGivensRotation(mat_T, c, s, CEED_NOTRANSPOSE, k, k+1, n, n);
1828673160d7Sjeremylt       CeedGivensRotation(mat_T, c, s, CEED_TRANSPOSE, k, k+1, n, n);
182952bfb9bbSJeremy L Thompson 
183052bfb9bbSJeremy L Thompson       // Apply Givens rotation to Q
183152bfb9bbSJeremy L Thompson       CeedGivensRotation(mat, c, s, CEED_NOTRANSPOSE, k, k+1, n, n);
183252bfb9bbSJeremy L Thompson 
183352bfb9bbSJeremy L Thompson       // Update x, z
183452bfb9bbSJeremy L Thompson       if (k < n-q-2) {
1835673160d7Sjeremylt         x = mat_T[k+n*(k+1)];
1836673160d7Sjeremylt         z = mat_T[k+n*(k+2)];
183752bfb9bbSJeremy L Thompson       }
183852bfb9bbSJeremy L Thompson     }
183952bfb9bbSJeremy L Thompson     itr++;
184052bfb9bbSJeremy L Thompson   }
1841673160d7Sjeremylt 
184252bfb9bbSJeremy L Thompson   // Save eigenvalues
184352bfb9bbSJeremy L Thompson   for (CeedInt i=0; i<n; i++)
1844673160d7Sjeremylt     lambda[i] = mat_T[i+n*i];
184552bfb9bbSJeremy L Thompson 
184652bfb9bbSJeremy L Thompson   // Check convergence
1847673160d7Sjeremylt   if (itr == max_itr && q < n-1)
1848c042f62fSJeremy L Thompson     // LCOV_EXCL_START
1849e15f9bd0SJeremy L Thompson     return CeedError(ceed, CEED_ERROR_MINOR,
1850e15f9bd0SJeremy L Thompson                      "Symmetric QR failed to converge");
1851c042f62fSJeremy L Thompson   // LCOV_EXCL_STOP
1852e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
185352bfb9bbSJeremy L Thompson }
185403d18186Sjeremylt CeedPragmaOptimizeOn
185552bfb9bbSJeremy L Thompson 
185652bfb9bbSJeremy L Thompson /**
185752bfb9bbSJeremy L Thompson   @brief Return Simultaneous Diagonalization of two matrices. This solves the
185852bfb9bbSJeremy L Thompson            generalized eigenvalue problem A x = lambda B x, where A and B
185952bfb9bbSJeremy L Thompson            are symmetric and B is positive definite. We generate the matrix X
186052bfb9bbSJeremy L Thompson            and vector Lambda such that X^T A X = Lambda and X^T B X = I. This
186152bfb9bbSJeremy L Thompson            is equivalent to the LAPACK routine 'sygv' with TYPE = 1.
186252bfb9bbSJeremy L Thompson 
186377645d7bSjeremylt   @param ceed         A Ceed context for error handling
1864d1d35e2fSjeremylt   @param[in] mat_A    Row-major matrix to be factorized with eigenvalues
1865d1d35e2fSjeremylt   @param[in] mat_B    Row-major matrix to be factorized to identity
1866d3331725Sjeremylt   @param[out] mat_X   Row-major orthogonal matrix
1867460bf743SValeria Barra   @param[out] lambda  Vector of length n of generalized eigenvalues
186852bfb9bbSJeremy L Thompson   @param n            Number of rows/columns
186952bfb9bbSJeremy L Thompson 
187052bfb9bbSJeremy L Thompson   @return An error code: 0 - success, otherwise - failure
187152bfb9bbSJeremy L Thompson 
187252bfb9bbSJeremy L Thompson   @ref Utility
187352bfb9bbSJeremy L Thompson **/
187403d18186Sjeremylt CeedPragmaOptimizeOff
1875d1d35e2fSjeremylt int CeedSimultaneousDiagonalization(Ceed ceed, CeedScalar *mat_A,
1876d3331725Sjeremylt                                     CeedScalar *mat_B, CeedScalar *mat_X,
187752bfb9bbSJeremy L Thompson                                     CeedScalar *lambda, CeedInt n) {
187852bfb9bbSJeremy L Thompson   int ierr;
1879d3331725Sjeremylt   CeedScalar *mat_C, *mat_G, *vec_D;
188078464608Sjeremylt   ierr = CeedCalloc(n*n, &mat_C); CeedChk(ierr);
188178464608Sjeremylt   ierr = CeedCalloc(n*n, &mat_G); CeedChk(ierr);
1882d3331725Sjeremylt   ierr = CeedCalloc(n, &vec_D); CeedChk(ierr);
188352bfb9bbSJeremy L Thompson 
188452bfb9bbSJeremy L Thompson   // Compute B = G D G^T
188578464608Sjeremylt   memcpy(mat_G, mat_B, n*n*sizeof(mat_B[0]));
1886d3331725Sjeremylt   ierr = CeedSymmetricSchurDecomposition(ceed, mat_G, vec_D, n); CeedChk(ierr);
188752bfb9bbSJeremy L Thompson 
188885cf89eaSjeremylt   // Sort eigenvalues
188985cf89eaSjeremylt   for (CeedInt i=n-1; i>=0; i--)
189085cf89eaSjeremylt     for (CeedInt j=0; j<i; j++) {
189185cf89eaSjeremylt       if (fabs(vec_D[j]) > fabs(vec_D[j+1])) {
189285cf89eaSjeremylt         CeedScalar temp;
189385cf89eaSjeremylt         temp = vec_D[j]; vec_D[j] = vec_D[j+1]; vec_D[j+1] = temp;
189485cf89eaSjeremylt         for (CeedInt k=0; k<n; k++) {
189585cf89eaSjeremylt           temp = mat_G[k*n+j]; mat_G[k*n+j] = mat_G[k*n+j+1]; mat_G[k*n+j+1] = temp;
189685cf89eaSjeremylt         }
189785cf89eaSjeremylt       }
189885cf89eaSjeremylt     }
189985cf89eaSjeremylt 
1900fb551037Sjeremylt   // Compute C = (G D^1/2)^-1 A (G D^1/2)^-T
1901fb551037Sjeremylt   //           = D^-1/2 G^T A G D^-1/2
1902d3331725Sjeremylt   // -- D = D^-1/2
190352bfb9bbSJeremy L Thompson   for (CeedInt i=0; i<n; i++)
1904d3331725Sjeremylt     vec_D[i] = 1./sqrt(vec_D[i]);
1905d3331725Sjeremylt   // -- G = G D^-1/2
1906d3331725Sjeremylt   // -- C = D^-1/2 G^T
1907d3331725Sjeremylt   for (CeedInt i=0; i<n; i++)
1908d3331725Sjeremylt     for (CeedInt j=0; j<n; j++) {
1909673160d7Sjeremylt       mat_G[i*n+j] *= vec_D[j];
1910673160d7Sjeremylt       mat_C[j*n+i]  = mat_G[i*n+j];
1911d3331725Sjeremylt     }
1912673160d7Sjeremylt   // -- X = (D^-1/2 G^T) A
1913ed9e99e6SJeremy L Thompson   ierr = CeedMatrixMatrixMultiply(ceed, (const CeedScalar *)mat_C,
1914d3331725Sjeremylt                                   (const CeedScalar *)mat_A, mat_X, n, n, n);
19159289e5bfSjeremylt   CeedChk(ierr);
1916673160d7Sjeremylt   // -- C = (D^-1/2 G^T A) (G D^-1/2)
1917ed9e99e6SJeremy L Thompson   ierr = CeedMatrixMatrixMultiply(ceed, (const CeedScalar *)mat_X,
191878464608Sjeremylt                                   (const CeedScalar *)mat_G, mat_C, n, n, n);
19199289e5bfSjeremylt   CeedChk(ierr);
192052bfb9bbSJeremy L Thompson 
192152bfb9bbSJeremy L Thompson   // Compute Q^T C Q = lambda
1922d1d35e2fSjeremylt   ierr = CeedSymmetricSchurDecomposition(ceed, mat_C, lambda, n); CeedChk(ierr);
192352bfb9bbSJeremy L Thompson 
192485cf89eaSjeremylt   // Sort eigenvalues
192585cf89eaSjeremylt   for (CeedInt i=n-1; i>=0; i--)
192685cf89eaSjeremylt     for (CeedInt j=0; j<i; j++) {
192785cf89eaSjeremylt       if (fabs(lambda[j]) > fabs(lambda[j+1])) {
192885cf89eaSjeremylt         CeedScalar temp;
192985cf89eaSjeremylt         temp = lambda[j]; lambda[j] = lambda[j+1]; lambda[j+1] = temp;
193085cf89eaSjeremylt         for (CeedInt k=0; k<n; k++) {
193185cf89eaSjeremylt           temp = mat_C[k*n+j]; mat_C[k*n+j] = mat_C[k*n+j+1]; mat_C[k*n+j+1] = temp;
193285cf89eaSjeremylt         }
193385cf89eaSjeremylt       }
193485cf89eaSjeremylt     }
193585cf89eaSjeremylt 
1936d3331725Sjeremylt   // Set X = (G D^1/2)^-T Q
1937fb551037Sjeremylt   //       = G D^-1/2 Q
1938ed9e99e6SJeremy L Thompson   ierr = CeedMatrixMatrixMultiply(ceed, (const CeedScalar *)mat_G,
1939d3331725Sjeremylt                                   (const CeedScalar *)mat_C, mat_X, n, n, n);
19409289e5bfSjeremylt   CeedChk(ierr);
194178464608Sjeremylt 
194278464608Sjeremylt   // Cleanup
194378464608Sjeremylt   ierr = CeedFree(&mat_C); CeedChk(ierr);
194478464608Sjeremylt   ierr = CeedFree(&mat_G); CeedChk(ierr);
1945d3331725Sjeremylt   ierr = CeedFree(&vec_D); CeedChk(ierr);
1946e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
194752bfb9bbSJeremy L Thompson }
194803d18186Sjeremylt CeedPragmaOptimizeOn
194952bfb9bbSJeremy L Thompson 
1950d7b241e6Sjeremylt /// @}
1951