xref: /libCEED/backends/ref/ceed-ref-operator.c (revision f10650af6497af6d0949b45f2eee824400fc9b71)
121617c04Sjeremylt // Copyright (c) 2017-2018, Lawrence Livermore National Security, LLC.
221617c04Sjeremylt // Produced at the Lawrence Livermore National Laboratory. LLNL-CODE-734707.
321617c04Sjeremylt // All Rights reserved. See files LICENSE and NOTICE for details.
421617c04Sjeremylt //
521617c04Sjeremylt // This file is part of CEED, a collection of benchmarks, miniapps, software
621617c04Sjeremylt // libraries and APIs for efficient high-order finite element and spectral
721617c04Sjeremylt // element discretizations for exascale applications. For more information and
821617c04Sjeremylt // source code availability see http://github.com/ceed.
921617c04Sjeremylt //
1021617c04Sjeremylt // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
1121617c04Sjeremylt // a collaborative effort of two U.S. Department of Energy organizations (Office
1221617c04Sjeremylt // of Science and the National Nuclear Security Administration) responsible for
1321617c04Sjeremylt // the planning and preparation of a capable exascale ecosystem, including
1421617c04Sjeremylt // software, applications, hardware, advanced system engineering and early
1521617c04Sjeremylt // testbed platforms, in support of the nation's exascale computing imperative.
1621617c04Sjeremylt 
1721617c04Sjeremylt #include "ceed-ref.h"
1821617c04Sjeremylt 
19*f10650afSjeremylt //------------------------------------------------------------------------------
20*f10650afSjeremylt // Setup Input/Output Fields
21*f10650afSjeremylt //------------------------------------------------------------------------------
22fe2413ffSjeremylt static int CeedOperatorSetupFields_Ref(CeedQFunction qf, CeedOperator op,
2391703d3fSjeremylt                                        bool inOrOut,
2491703d3fSjeremylt                                        CeedVector *fullevecs, CeedVector *evecs,
25aedaa0e5Sjeremylt                                        CeedVector *qvecs, CeedInt starte,
26135a076eSjeremylt                                        CeedInt numfields, CeedInt Q) {
274d537eeaSYohann   CeedInt dim, ierr, size, P;
28aedaa0e5Sjeremylt   Ceed ceed;
29aedaa0e5Sjeremylt   ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
30d1bcdac9Sjeremylt   CeedBasis basis;
31d1bcdac9Sjeremylt   CeedElemRestriction Erestrict;
32aedaa0e5Sjeremylt   CeedOperatorField *opfields;
33aedaa0e5Sjeremylt   CeedQFunctionField *qffields;
34fe2413ffSjeremylt   if (inOrOut) {
3516911fdaSjeremylt     ierr = CeedOperatorGetFields(op, NULL, &opfields); CeedChk(ierr);
3616911fdaSjeremylt     ierr = CeedQFunctionGetFields(qf, NULL, &qffields); CeedChk(ierr);
37fe2413ffSjeremylt   } else {
3816911fdaSjeremylt     ierr = CeedOperatorGetFields(op, &opfields, NULL); CeedChk(ierr);
3916911fdaSjeremylt     ierr = CeedQFunctionGetFields(qf, &qffields, NULL); CeedChk(ierr);
40fe2413ffSjeremylt   }
4121617c04Sjeremylt 
42885ac19cSjeremylt   // Loop over fields
43885ac19cSjeremylt   for (CeedInt i=0; i<numfields; i++) {
44d1bcdac9Sjeremylt     CeedEvalMode emode;
45aedaa0e5Sjeremylt     ierr = CeedQFunctionFieldGetEvalMode(qffields[i], &emode); CeedChk(ierr);
46135a076eSjeremylt 
47135a076eSjeremylt     if (emode != CEED_EVAL_WEIGHT) {
48aedaa0e5Sjeremylt       ierr = CeedOperatorFieldGetElemRestriction(opfields[i], &Erestrict);
49d1bcdac9Sjeremylt       CeedChk(ierr);
50d1bcdac9Sjeremylt       ierr = CeedElemRestrictionCreateVector(Erestrict, NULL,
5191703d3fSjeremylt                                              &fullevecs[i+starte]);
52135a076eSjeremylt       CeedChk(ierr);
53135a076eSjeremylt     }
54135a076eSjeremylt 
55885ac19cSjeremylt     switch(emode) {
56885ac19cSjeremylt     case CEED_EVAL_NONE:
574d537eeaSYohann       ierr = CeedQFunctionFieldGetSize(qffields[i], &size); CeedChk(ierr);
584d537eeaSYohann       ierr = CeedVectorCreate(ceed, Q*size, &qvecs[i]); CeedChk(ierr);
59aedaa0e5Sjeremylt       break;
60aedaa0e5Sjeremylt     case CEED_EVAL_INTERP:
614d537eeaSYohann       ierr = CeedQFunctionFieldGetSize(qffields[i], &size); CeedChk(ierr);
624d1cd9fcSJeremy L Thompson       ierr = CeedElemRestrictionGetElementSize(Erestrict, &P);
634d1cd9fcSJeremy L Thompson       CeedChk(ierr);
644d537eeaSYohann       ierr = CeedVectorCreate(ceed, P*size, &evecs[i]); CeedChk(ierr);
654d537eeaSYohann       ierr = CeedVectorCreate(ceed, Q*size, &qvecs[i]); CeedChk(ierr);
66885ac19cSjeremylt       break;
67885ac19cSjeremylt     case CEED_EVAL_GRAD:
68aedaa0e5Sjeremylt       ierr = CeedOperatorFieldGetBasis(opfields[i], &basis); CeedChk(ierr);
694d537eeaSYohann       ierr = CeedQFunctionFieldGetSize(qffields[i], &size); CeedChk(ierr);
70d1bcdac9Sjeremylt       ierr = CeedBasisGetDimension(basis, &dim); CeedChk(ierr);
714d1cd9fcSJeremy L Thompson       ierr = CeedElemRestrictionGetElementSize(Erestrict, &P);
724d1cd9fcSJeremy L Thompson       CeedChk(ierr);
734d537eeaSYohann       ierr = CeedVectorCreate(ceed, P*size/dim, &evecs[i]); CeedChk(ierr);
744d537eeaSYohann       ierr = CeedVectorCreate(ceed, Q*size, &qvecs[i]); CeedChk(ierr);
75885ac19cSjeremylt       break;
76885ac19cSjeremylt     case CEED_EVAL_WEIGHT: // Only on input fields
77aedaa0e5Sjeremylt       ierr = CeedOperatorFieldGetBasis(opfields[i], &basis); CeedChk(ierr);
78aedaa0e5Sjeremylt       ierr = CeedVectorCreate(ceed, Q, &qvecs[i]); CeedChk(ierr);
79d1bcdac9Sjeremylt       ierr = CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, CEED_EVAL_WEIGHT,
80a7b7f929Sjeremylt                             CEED_VECTOR_NONE, qvecs[i]); CeedChk(ierr);
81885ac19cSjeremylt       break;
82885ac19cSjeremylt     case CEED_EVAL_DIV:
834d537eeaSYohann       break; // Not implemented
84885ac19cSjeremylt     case CEED_EVAL_CURL:
854d537eeaSYohann       break; // Not implemented
8621617c04Sjeremylt     }
87885ac19cSjeremylt   }
8821617c04Sjeremylt   return 0;
8921617c04Sjeremylt }
9021617c04Sjeremylt 
91*f10650afSjeremylt //------------------------------------------------------------------------------
92*f10650afSjeremylt // Setup Operator
93*f10650afSjeremylt //------------------------------------------------------------------------------/*
94885ac19cSjeremylt static int CeedOperatorSetup_Ref(CeedOperator op) {
9521617c04Sjeremylt   int ierr;
964ce2993fSjeremylt   bool setupdone;
974ce2993fSjeremylt   ierr = CeedOperatorGetSetupStatus(op, &setupdone); CeedChk(ierr);
984ce2993fSjeremylt   if (setupdone) return 0;
99aedaa0e5Sjeremylt   Ceed ceed;
100aedaa0e5Sjeremylt   ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
1014ce2993fSjeremylt   CeedOperator_Ref *impl;
1024ce2993fSjeremylt   ierr = CeedOperatorGetData(op, (void *)&impl); CeedChk(ierr);
1034ce2993fSjeremylt   CeedQFunction qf;
1044ce2993fSjeremylt   ierr = CeedOperatorGetQFunction(op, &qf); CeedChk(ierr);
1054ce2993fSjeremylt   CeedInt Q, numinputfields, numoutputfields;
1064ce2993fSjeremylt   ierr = CeedOperatorGetNumQuadraturePoints(op, &Q); CeedChk(ierr);
10716911fdaSjeremylt   ierr = CeedQFunctionGetIdentityStatus(qf, &impl->identityqf); CeedChk(ierr);
108a8de75f0Sjeremylt   ierr = CeedQFunctionGetNumArgs(qf, &numinputfields, &numoutputfields);
109a8de75f0Sjeremylt   CeedChk(ierr);
110d1bcdac9Sjeremylt   CeedOperatorField *opinputfields, *opoutputfields;
111d1bcdac9Sjeremylt   ierr = CeedOperatorGetFields(op, &opinputfields, &opoutputfields);
112d1bcdac9Sjeremylt   CeedChk(ierr);
113d1bcdac9Sjeremylt   CeedQFunctionField *qfinputfields, *qfoutputfields;
114d1bcdac9Sjeremylt   ierr = CeedQFunctionGetFields(qf, &qfinputfields, &qfoutputfields);
115d1bcdac9Sjeremylt   CeedChk(ierr);
116885ac19cSjeremylt 
117885ac19cSjeremylt   // Allocate
118aedaa0e5Sjeremylt   ierr = CeedCalloc(numinputfields + numoutputfields, &impl->evecs);
119aedaa0e5Sjeremylt   CeedChk(ierr);
120aedaa0e5Sjeremylt   ierr = CeedCalloc(numinputfields + numoutputfields, &impl->edata);
121885ac19cSjeremylt   CeedChk(ierr);
122885ac19cSjeremylt 
1238d713cf6Sjeremylt   ierr = CeedCalloc(16, &impl->inputstate); CeedChk(ierr);
12491703d3fSjeremylt   ierr = CeedCalloc(16, &impl->evecsin); CeedChk(ierr);
12591703d3fSjeremylt   ierr = CeedCalloc(16, &impl->evecsout); CeedChk(ierr);
126aedaa0e5Sjeremylt   ierr = CeedCalloc(16, &impl->qvecsin); CeedChk(ierr);
127aedaa0e5Sjeremylt   ierr = CeedCalloc(16, &impl->qvecsout); CeedChk(ierr);
128885ac19cSjeremylt 
129aedaa0e5Sjeremylt   impl->numein = numinputfields; impl->numeout = numoutputfields;
130885ac19cSjeremylt 
131aedaa0e5Sjeremylt   // Set up infield and outfield evecs and qvecs
132885ac19cSjeremylt   // Infields
13391703d3fSjeremylt   ierr = CeedOperatorSetupFields_Ref(qf, op, 0, impl->evecs,
13491703d3fSjeremylt                                      impl->evecsin, impl->qvecsin, 0,
135aedaa0e5Sjeremylt                                      numinputfields, Q);
136aedaa0e5Sjeremylt   CeedChk(ierr);
137885ac19cSjeremylt   // Outfields
13891703d3fSjeremylt   ierr = CeedOperatorSetupFields_Ref(qf, op, 1, impl->evecs,
13991703d3fSjeremylt                                      impl->evecsout, impl->qvecsout,
140aedaa0e5Sjeremylt                                      numinputfields, numoutputfields, Q);
141d1bcdac9Sjeremylt   CeedChk(ierr);
142885ac19cSjeremylt 
14316911fdaSjeremylt   // Identity QFunctions
14416911fdaSjeremylt   if (impl->identityqf) {
14516911fdaSjeremylt     CeedEvalMode inmode, outmode;
14616911fdaSjeremylt     CeedQFunctionField *infields, *outfields;
14716911fdaSjeremylt     ierr = CeedQFunctionGetFields(qf, &infields, &outfields); CeedChk(ierr);
14816911fdaSjeremylt 
14916911fdaSjeremylt     for (CeedInt i=0; i<numinputfields; i++) {
15016911fdaSjeremylt       ierr = CeedQFunctionFieldGetEvalMode(infields[i], &inmode);
15116911fdaSjeremylt       CeedChk(ierr);
15216911fdaSjeremylt       ierr = CeedQFunctionFieldGetEvalMode(outfields[i], &outmode);
15316911fdaSjeremylt       CeedChk(ierr);
15416911fdaSjeremylt 
15516911fdaSjeremylt       ierr = CeedVectorDestroy(&impl->qvecsout[i]); CeedChk(ierr);
15616911fdaSjeremylt       impl->qvecsout[i] = impl->qvecsin[i];
15755e4cc5bSjeremylt       ierr = CeedVectorAddReference(impl->qvecsin[i]); CeedChk(ierr);
15816911fdaSjeremylt     }
15916911fdaSjeremylt   }
16016911fdaSjeremylt 
1614ce2993fSjeremylt   ierr = CeedOperatorSetSetupDone(op); CeedChk(ierr);
162885ac19cSjeremylt 
163885ac19cSjeremylt   return 0;
164885ac19cSjeremylt }
165885ac19cSjeremylt 
166*f10650afSjeremylt //------------------------------------------------------------------------------
167*f10650afSjeremylt // Setup Operator Inputs
168*f10650afSjeremylt //------------------------------------------------------------------------------
1691d102b48SJeremy L Thompson static inline int CeedOperatorSetupInputs_Ref(CeedInt numinputfields,
1701d102b48SJeremy L Thompson     CeedQFunctionField *qfinputfields, CeedOperatorField *opinputfields,
1711d102b48SJeremy L Thompson     CeedVector invec, const bool skipactive, CeedOperator_Ref *impl,
1721d102b48SJeremy L Thompson     CeedRequest *request) {
1731d102b48SJeremy L Thompson   CeedInt ierr;
174d1bcdac9Sjeremylt   CeedEvalMode emode;
175d1bcdac9Sjeremylt   CeedVector vec;
176d1bcdac9Sjeremylt   CeedElemRestriction Erestrict;
1771d102b48SJeremy L Thompson   CeedTransposeMode lmode;
1788d713cf6Sjeremylt   uint64_t state;
179885ac19cSjeremylt 
1801a4ead9bSjeremylt   for (CeedInt i=0; i<numinputfields; i++) {
181d1bcdac9Sjeremylt     // Get input vector
182d1bcdac9Sjeremylt     ierr = CeedOperatorFieldGetVector(opinputfields[i], &vec); CeedChk(ierr);
1831d102b48SJeremy L Thompson     if (vec == CEED_VECTOR_ACTIVE) {
1841d102b48SJeremy L Thompson       if (skipactive)
1851d102b48SJeremy L Thompson         continue;
1861d102b48SJeremy L Thompson       else
187d1bcdac9Sjeremylt         vec = invec;
1881d102b48SJeremy L Thompson     }
1891d102b48SJeremy L Thompson 
1901d102b48SJeremy L Thompson     ierr = CeedQFunctionFieldGetEvalMode(qfinputfields[i], &emode);
1911d102b48SJeremy L Thompson     CeedChk(ierr);
1921d102b48SJeremy L Thompson     // Restrict and Evec
1931d102b48SJeremy L Thompson     if (emode == CEED_EVAL_WEIGHT) { // Skip
1941d102b48SJeremy L Thompson     } else {
195668048e2SJed Brown       // Restrict
1968d713cf6Sjeremylt       ierr = CeedVectorGetState(vec, &state); CeedChk(ierr);
1978d713cf6Sjeremylt       // Skip restriction if input is unchanged
1988d713cf6Sjeremylt       if (state != impl->inputstate[i] || vec == invec) {
199d1bcdac9Sjeremylt         ierr = CeedOperatorFieldGetElemRestriction(opinputfields[i], &Erestrict);
200d1bcdac9Sjeremylt         CeedChk(ierr);
2011d102b48SJeremy L Thompson         ierr = CeedOperatorFieldGetLMode(opinputfields[i], &lmode);
2021d102b48SJeremy L Thompson         CeedChk(ierr);
203d1bcdac9Sjeremylt         ierr = CeedElemRestrictionApply(Erestrict, CEED_NOTRANSPOSE,
204d1bcdac9Sjeremylt                                         lmode, vec, impl->evecs[i],
205668048e2SJed Brown                                         request); CeedChk(ierr);
2068d713cf6Sjeremylt         impl->inputstate[i] = state;
2078d713cf6Sjeremylt       }
208668048e2SJed Brown       // Get evec
209a2b73c81Sjeremylt       ierr = CeedVectorGetArrayRead(impl->evecs[i], CEED_MEM_HOST,
210d1bcdac9Sjeremylt                                     (const CeedScalar **) &impl->edata[i]);
211d1bcdac9Sjeremylt       CeedChk(ierr);
212885ac19cSjeremylt     }
213885ac19cSjeremylt   }
2141d102b48SJeremy L Thompson   return 0;
215885ac19cSjeremylt }
216885ac19cSjeremylt 
217*f10650afSjeremylt //------------------------------------------------------------------------------
218*f10650afSjeremylt // Input Basis Action
219*f10650afSjeremylt //------------------------------------------------------------------------------
2201d102b48SJeremy L Thompson static inline int CeedOperatorInputBasis_Ref(CeedInt e, CeedInt Q,
2211d102b48SJeremy L Thompson     CeedQFunctionField *qfinputfields, CeedOperatorField *opinputfields,
2221d102b48SJeremy L Thompson     CeedInt numinputfields, const bool skipactive, CeedOperator_Ref *impl) {
2231d102b48SJeremy L Thompson   CeedInt ierr;
2241d102b48SJeremy L Thompson   CeedInt dim, elemsize, size;
2251d102b48SJeremy L Thompson   CeedElemRestriction Erestrict;
2261d102b48SJeremy L Thompson   CeedEvalMode emode;
2271d102b48SJeremy L Thompson   CeedBasis basis;
2281d102b48SJeremy L Thompson 
2291a4ead9bSjeremylt   for (CeedInt i=0; i<numinputfields; i++) {
2301d102b48SJeremy L Thompson     // Skip active input
2311d102b48SJeremy L Thompson     if (skipactive) {
2321d102b48SJeremy L Thompson       CeedVector vec;
2331d102b48SJeremy L Thompson       ierr = CeedOperatorFieldGetVector(opinputfields[i], &vec); CeedChk(ierr);
2341d102b48SJeremy L Thompson       if (vec == CEED_VECTOR_ACTIVE)
2351d102b48SJeremy L Thompson         continue;
2361d102b48SJeremy L Thompson     }
2374d537eeaSYohann     // Get elemsize, emode, size
238d1bcdac9Sjeremylt     ierr = CeedOperatorFieldGetElemRestriction(opinputfields[i], &Erestrict);
239d1bcdac9Sjeremylt     CeedChk(ierr);
240d1bcdac9Sjeremylt     ierr = CeedElemRestrictionGetElementSize(Erestrict, &elemsize);
241d1bcdac9Sjeremylt     CeedChk(ierr);
242d1bcdac9Sjeremylt     ierr = CeedQFunctionFieldGetEvalMode(qfinputfields[i], &emode);
243d1bcdac9Sjeremylt     CeedChk(ierr);
2444d537eeaSYohann     ierr = CeedQFunctionFieldGetSize(qfinputfields[i], &size); CeedChk(ierr);
245885ac19cSjeremylt     // Basis action
246885ac19cSjeremylt     switch(emode) {
247885ac19cSjeremylt     case CEED_EVAL_NONE:
248aedaa0e5Sjeremylt       ierr = CeedVectorSetArray(impl->qvecsin[i], CEED_MEM_HOST,
249aedaa0e5Sjeremylt                                 CEED_USE_POINTER,
2504d537eeaSYohann                                 &impl->edata[i][e*Q*size]); CeedChk(ierr);
251885ac19cSjeremylt       break;
252885ac19cSjeremylt     case CEED_EVAL_INTERP:
253d1bcdac9Sjeremylt       ierr = CeedOperatorFieldGetBasis(opinputfields[i], &basis); CeedChk(ierr);
25491703d3fSjeremylt       ierr = CeedVectorSetArray(impl->evecsin[i], CEED_MEM_HOST,
255aedaa0e5Sjeremylt                                 CEED_USE_POINTER,
2564d537eeaSYohann                                 &impl->edata[i][e*elemsize*size]);
257aedaa0e5Sjeremylt       CeedChk(ierr);
258d1bcdac9Sjeremylt       ierr = CeedBasisApply(basis, 1, CEED_NOTRANSPOSE,
25991703d3fSjeremylt                             CEED_EVAL_INTERP, impl->evecsin[i],
260aedaa0e5Sjeremylt                             impl->qvecsin[i]); CeedChk(ierr);
261885ac19cSjeremylt       break;
262885ac19cSjeremylt     case CEED_EVAL_GRAD:
263d1bcdac9Sjeremylt       ierr = CeedOperatorFieldGetBasis(opinputfields[i], &basis); CeedChk(ierr);
2644d537eeaSYohann       ierr = CeedBasisGetDimension(basis, &dim); CeedChk(ierr);
26591703d3fSjeremylt       ierr = CeedVectorSetArray(impl->evecsin[i], CEED_MEM_HOST,
266aedaa0e5Sjeremylt                                 CEED_USE_POINTER,
2674d537eeaSYohann                                 &impl->edata[i][e*elemsize*size/dim]);
268aedaa0e5Sjeremylt       CeedChk(ierr);
269d1bcdac9Sjeremylt       ierr = CeedBasisApply(basis, 1, CEED_NOTRANSPOSE,
27091703d3fSjeremylt                             CEED_EVAL_GRAD, impl->evecsin[i],
271aedaa0e5Sjeremylt                             impl->qvecsin[i]); CeedChk(ierr);
272885ac19cSjeremylt       break;
273885ac19cSjeremylt     case CEED_EVAL_WEIGHT:
274885ac19cSjeremylt       break;  // No action
275885ac19cSjeremylt     case CEED_EVAL_DIV:
2761d102b48SJeremy L Thompson     case CEED_EVAL_CURL: {
2771d102b48SJeremy L Thompson       // LCOV_EXCL_START
2781d102b48SJeremy L Thompson       ierr = CeedOperatorFieldGetBasis(opinputfields[i], &basis);
2791d102b48SJeremy L Thompson       CeedChk(ierr);
2801d102b48SJeremy L Thompson       Ceed ceed;
2811d102b48SJeremy L Thompson       ierr = CeedBasisGetCeed(basis, &ceed); CeedChk(ierr);
2821d102b48SJeremy L Thompson       return CeedError(ceed, 1, "Ceed evaluation mode not implemented");
2831d102b48SJeremy L Thompson       // LCOV_EXCL_STOP
2844d537eeaSYohann       break; // Not implemented
285885ac19cSjeremylt     }
286885ac19cSjeremylt     }
287885ac19cSjeremylt   }
2881d102b48SJeremy L Thompson   return 0;
289885ac19cSjeremylt }
290885ac19cSjeremylt 
291*f10650afSjeremylt //------------------------------------------------------------------------------
292*f10650afSjeremylt // Output Basis Action
293*f10650afSjeremylt //------------------------------------------------------------------------------
2941d102b48SJeremy L Thompson static inline int CeedOperatorOutputBasis_Ref(CeedInt e, CeedInt Q,
2951d102b48SJeremy L Thompson     CeedQFunctionField *qfoutputfields, CeedOperatorField *opoutputfields,
2961d102b48SJeremy L Thompson     CeedInt numinputfields, CeedInt numoutputfields, CeedOperator op,
2971d102b48SJeremy L Thompson     CeedOperator_Ref *impl) {
2981d102b48SJeremy L Thompson   CeedInt ierr;
2991d102b48SJeremy L Thompson   CeedInt dim, elemsize, size;
3001d102b48SJeremy L Thompson   CeedElemRestriction Erestrict;
3011d102b48SJeremy L Thompson   CeedEvalMode emode;
3021d102b48SJeremy L Thompson   CeedBasis basis;
3031d102b48SJeremy L Thompson 
3041a4ead9bSjeremylt   for (CeedInt i=0; i<numoutputfields; i++) {
3054d537eeaSYohann     // Get elemsize, emode, size
306d1bcdac9Sjeremylt     ierr = CeedOperatorFieldGetElemRestriction(opoutputfields[i], &Erestrict);
307d1bcdac9Sjeremylt     CeedChk(ierr);
308d1bcdac9Sjeremylt     ierr = CeedElemRestrictionGetElementSize(Erestrict, &elemsize);
309d1bcdac9Sjeremylt     CeedChk(ierr);
310d1bcdac9Sjeremylt     ierr = CeedQFunctionFieldGetEvalMode(qfoutputfields[i], &emode);
311d1bcdac9Sjeremylt     CeedChk(ierr);
3124d537eeaSYohann     ierr = CeedQFunctionFieldGetSize(qfoutputfields[i], &size); CeedChk(ierr);
313885ac19cSjeremylt     // Basis action
314885ac19cSjeremylt     switch(emode) {
315885ac19cSjeremylt     case CEED_EVAL_NONE:
316885ac19cSjeremylt       break; // No action
317885ac19cSjeremylt     case CEED_EVAL_INTERP:
318d1bcdac9Sjeremylt       ierr = CeedOperatorFieldGetBasis(opoutputfields[i], &basis);
319d1bcdac9Sjeremylt       CeedChk(ierr);
32091703d3fSjeremylt       ierr = CeedVectorSetArray(impl->evecsout[i], CEED_MEM_HOST,
321aedaa0e5Sjeremylt                                 CEED_USE_POINTER,
3224d537eeaSYohann                                 &impl->edata[i + numinputfields][e*elemsize*size]);
323de686571SJeremy L Thompson       CeedChk(ierr);
324aedaa0e5Sjeremylt       ierr = CeedBasisApply(basis, 1, CEED_TRANSPOSE,
325aedaa0e5Sjeremylt                             CEED_EVAL_INTERP, impl->qvecsout[i],
32691703d3fSjeremylt                             impl->evecsout[i]); CeedChk(ierr);
327885ac19cSjeremylt       break;
328885ac19cSjeremylt     case CEED_EVAL_GRAD:
329d1bcdac9Sjeremylt       ierr = CeedOperatorFieldGetBasis(opoutputfields[i], &basis);
330d1bcdac9Sjeremylt       CeedChk(ierr);
3314d537eeaSYohann       ierr = CeedBasisGetDimension(basis, &dim); CeedChk(ierr);
33291703d3fSjeremylt       ierr = CeedVectorSetArray(impl->evecsout[i], CEED_MEM_HOST,
333aedaa0e5Sjeremylt                                 CEED_USE_POINTER,
3344d537eeaSYohann                                 &impl->edata[i + numinputfields][e*elemsize*size/dim]);
335de686571SJeremy L Thompson       CeedChk(ierr);
336aedaa0e5Sjeremylt       ierr = CeedBasisApply(basis, 1, CEED_TRANSPOSE,
337aedaa0e5Sjeremylt                             CEED_EVAL_GRAD, impl->qvecsout[i],
33891703d3fSjeremylt                             impl->evecsout[i]); CeedChk(ierr);
339885ac19cSjeremylt       break;
3404ce2993fSjeremylt     case CEED_EVAL_WEIGHT: {
341c042f62fSJeremy L Thompson       // LCOV_EXCL_START
3424ce2993fSjeremylt       Ceed ceed;
3434ce2993fSjeremylt       ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
3441d102b48SJeremy L Thompson       return CeedError(ceed, 1, "CEED_EVAL_WEIGHT cannot be an output "
3451d102b48SJeremy L Thompson                        "evaluation mode");
346c042f62fSJeremy L Thompson       // LCOV_EXCL_STOP
347885ac19cSjeremylt       break; // Should not occur
3484ce2993fSjeremylt     }
349885ac19cSjeremylt     case CEED_EVAL_DIV:
3501d102b48SJeremy L Thompson     case CEED_EVAL_CURL: {
3511d102b48SJeremy L Thompson       // LCOV_EXCL_START
3521d102b48SJeremy L Thompson       Ceed ceed;
3531d102b48SJeremy L Thompson       ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
3541d102b48SJeremy L Thompson       return CeedError(ceed, 1, "Ceed evaluation mode not implemented");
3551d102b48SJeremy L Thompson       // LCOV_EXCL_STOP
3568c91a0c9SJeremy L Thompson       break; // Not implemented
357885ac19cSjeremylt     }
358885ac19cSjeremylt     }
359885ac19cSjeremylt   }
3601d102b48SJeremy L Thompson   return 0;
3611d102b48SJeremy L Thompson }
3621d102b48SJeremy L Thompson 
363*f10650afSjeremylt //------------------------------------------------------------------------------
364*f10650afSjeremylt // Restore Input Vectors
365*f10650afSjeremylt //------------------------------------------------------------------------------
3661d102b48SJeremy L Thompson static inline int CeedOperatorRestoreInputs_Ref(CeedInt numinputfields,
3671d102b48SJeremy L Thompson     CeedQFunctionField *qfinputfields, CeedOperatorField *opinputfields,
3681d102b48SJeremy L Thompson     const bool skipactive, CeedOperator_Ref *impl) {
3691d102b48SJeremy L Thompson   CeedInt ierr;
3701d102b48SJeremy L Thompson   CeedEvalMode emode;
3711d102b48SJeremy L Thompson 
3721d102b48SJeremy L Thompson   for (CeedInt i=0; i<numinputfields; i++) {
3731d102b48SJeremy L Thompson     // Skip active inputs
3741d102b48SJeremy L Thompson     if (skipactive) {
3751d102b48SJeremy L Thompson       CeedVector vec;
3761d102b48SJeremy L Thompson       ierr = CeedOperatorFieldGetVector(opinputfields[i], &vec); CeedChk(ierr);
3771d102b48SJeremy L Thompson       if (vec == CEED_VECTOR_ACTIVE)
3781d102b48SJeremy L Thompson         continue;
3791d102b48SJeremy L Thompson     }
3801d102b48SJeremy L Thompson     // Restore input
3811d102b48SJeremy L Thompson     ierr = CeedQFunctionFieldGetEvalMode(qfinputfields[i], &emode);
3821d102b48SJeremy L Thompson     CeedChk(ierr);
3831d102b48SJeremy L Thompson     if (emode == CEED_EVAL_WEIGHT) { // Skip
3841d102b48SJeremy L Thompson     } else {
3851d102b48SJeremy L Thompson       ierr = CeedVectorRestoreArrayRead(impl->evecs[i],
3861d102b48SJeremy L Thompson                                         (const CeedScalar **) &impl->edata[i]);
3871d102b48SJeremy L Thompson       CeedChk(ierr);
3881d102b48SJeremy L Thompson     }
3891d102b48SJeremy L Thompson   }
3901d102b48SJeremy L Thompson   return 0;
3911d102b48SJeremy L Thompson }
3921d102b48SJeremy L Thompson 
393*f10650afSjeremylt //------------------------------------------------------------------------------
394*f10650afSjeremylt // Operator Apply
395*f10650afSjeremylt //------------------------------------------------------------------------------
3961d102b48SJeremy L Thompson static int CeedOperatorApply_Ref(CeedOperator op, CeedVector invec,
3971d102b48SJeremy L Thompson                                  CeedVector outvec, CeedRequest *request) {
3981d102b48SJeremy L Thompson   int ierr;
3991d102b48SJeremy L Thompson   CeedOperator_Ref *impl;
4001d102b48SJeremy L Thompson   ierr = CeedOperatorGetData(op, (void *)&impl); CeedChk(ierr);
4011d102b48SJeremy L Thompson   CeedQFunction qf;
4021d102b48SJeremy L Thompson   ierr = CeedOperatorGetQFunction(op, &qf); CeedChk(ierr);
4031d102b48SJeremy L Thompson   CeedInt Q, numelements, numinputfields, numoutputfields, size;
4041d102b48SJeremy L Thompson   ierr = CeedOperatorGetNumQuadraturePoints(op, &Q); CeedChk(ierr);
4051d102b48SJeremy L Thompson   ierr = CeedOperatorGetNumElements(op, &numelements); CeedChk(ierr);
4061d102b48SJeremy L Thompson   ierr= CeedQFunctionGetNumArgs(qf, &numinputfields, &numoutputfields);
4071d102b48SJeremy L Thompson   CeedChk(ierr);
4081d102b48SJeremy L Thompson   CeedTransposeMode lmode;
4091d102b48SJeremy L Thompson   CeedOperatorField *opinputfields, *opoutputfields;
4101d102b48SJeremy L Thompson   ierr = CeedOperatorGetFields(op, &opinputfields, &opoutputfields);
4111d102b48SJeremy L Thompson   CeedChk(ierr);
4121d102b48SJeremy L Thompson   CeedQFunctionField *qfinputfields, *qfoutputfields;
4131d102b48SJeremy L Thompson   ierr = CeedQFunctionGetFields(qf, &qfinputfields, &qfoutputfields);
4141d102b48SJeremy L Thompson   CeedChk(ierr);
4151d102b48SJeremy L Thompson   CeedEvalMode emode;
4161d102b48SJeremy L Thompson   CeedVector vec;
4171d102b48SJeremy L Thompson   CeedElemRestriction Erestrict;
4181d102b48SJeremy L Thompson 
4191d102b48SJeremy L Thompson   // Setup
4201d102b48SJeremy L Thompson   ierr = CeedOperatorSetup_Ref(op); CeedChk(ierr);
4211d102b48SJeremy L Thompson 
4221d102b48SJeremy L Thompson   // Input Evecs and Restriction
4231d102b48SJeremy L Thompson   ierr = CeedOperatorSetupInputs_Ref(numinputfields, qfinputfields,
4241d102b48SJeremy L Thompson                                      opinputfields, invec, false, impl,
4251d102b48SJeremy L Thompson                                      request); CeedChk(ierr);
4261d102b48SJeremy L Thompson 
4271d102b48SJeremy L Thompson   // Output Evecs
4281d102b48SJeremy L Thompson   for (CeedInt i=0; i<numoutputfields; i++) {
4291d102b48SJeremy L Thompson     ierr = CeedVectorGetArray(impl->evecs[i+impl->numein], CEED_MEM_HOST,
4301d102b48SJeremy L Thompson                               &impl->edata[i + numinputfields]); CeedChk(ierr);
4311d102b48SJeremy L Thompson   }
4321d102b48SJeremy L Thompson 
4331d102b48SJeremy L Thompson   // Loop through elements
4341d102b48SJeremy L Thompson   for (CeedInt e=0; e<numelements; e++) {
4351d102b48SJeremy L Thompson     // Output pointers
4361d102b48SJeremy L Thompson     for (CeedInt i=0; i<numoutputfields; i++) {
4371d102b48SJeremy L Thompson       ierr = CeedQFunctionFieldGetEvalMode(qfoutputfields[i], &emode);
4381d102b48SJeremy L Thompson       CeedChk(ierr);
4391d102b48SJeremy L Thompson       if (emode == CEED_EVAL_NONE) {
4401d102b48SJeremy L Thompson         ierr = CeedQFunctionFieldGetSize(qfoutputfields[i], &size);
4411d102b48SJeremy L Thompson         CeedChk(ierr);
4421d102b48SJeremy L Thompson         ierr = CeedVectorSetArray(impl->qvecsout[i], CEED_MEM_HOST,
4431d102b48SJeremy L Thompson                                   CEED_USE_POINTER,
4441d102b48SJeremy L Thompson                                   &impl->edata[i + numinputfields][e*Q*size]);
4451d102b48SJeremy L Thompson         CeedChk(ierr);
4461d102b48SJeremy L Thompson       }
4471d102b48SJeremy L Thompson     }
4481d102b48SJeremy L Thompson 
44916911fdaSjeremylt     // Input basis apply
45016911fdaSjeremylt     ierr = CeedOperatorInputBasis_Ref(e, Q, qfinputfields, opinputfields,
45116911fdaSjeremylt                                       numinputfields, false, impl);
45216911fdaSjeremylt     CeedChk(ierr);
45316911fdaSjeremylt 
4541d102b48SJeremy L Thompson     // Q function
45516911fdaSjeremylt     if (!impl->identityqf) {
45616911fdaSjeremylt       ierr = CeedQFunctionApply(qf, Q, impl->qvecsin, impl->qvecsout);
45716911fdaSjeremylt       CeedChk(ierr);
45816911fdaSjeremylt     }
4591d102b48SJeremy L Thompson 
4601d102b48SJeremy L Thompson     // Output basis apply
4611d102b48SJeremy L Thompson     ierr = CeedOperatorOutputBasis_Ref(e, Q, qfoutputfields, opoutputfields,
4621d102b48SJeremy L Thompson                                        numinputfields, numoutputfields, op, impl);
4631d102b48SJeremy L Thompson     CeedChk(ierr);
4641d102b48SJeremy L Thompson   }
465885ac19cSjeremylt 
466885ac19cSjeremylt   // Output restriction
4671a4ead9bSjeremylt   for (CeedInt i=0; i<numoutputfields; i++) {
468a2b73c81Sjeremylt     // Restore evec
469a2b73c81Sjeremylt     ierr = CeedVectorRestoreArray(impl->evecs[i+impl->numein],
470d1bcdac9Sjeremylt                                   &impl->edata[i + numinputfields]);
471d1bcdac9Sjeremylt     CeedChk(ierr);
472d1bcdac9Sjeremylt     // Get output vector
473d1bcdac9Sjeremylt     ierr = CeedOperatorFieldGetVector(opoutputfields[i], &vec); CeedChk(ierr);
474668048e2SJed Brown     // Active
475d1bcdac9Sjeremylt     if (vec == CEED_VECTOR_ACTIVE)
476d1bcdac9Sjeremylt       vec = outvec;
4777ca8db16Sjeremylt     // Restrict
478d1bcdac9Sjeremylt     ierr = CeedOperatorFieldGetElemRestriction(opoutputfields[i], &Erestrict);
479d1bcdac9Sjeremylt     CeedChk(ierr);
4804dccadb6Sjeremylt     ierr = CeedOperatorFieldGetLMode(opoutputfields[i], &lmode); CeedChk(ierr);
481d1bcdac9Sjeremylt     ierr = CeedElemRestrictionApply(Erestrict, CEED_TRANSPOSE,
482d1bcdac9Sjeremylt                                     lmode, impl->evecs[i+impl->numein], vec,
483668048e2SJed Brown                                     request); CeedChk(ierr);
484885ac19cSjeremylt   }
485885ac19cSjeremylt 
4867ca8db16Sjeremylt   // Restore input arrays
4871d102b48SJeremy L Thompson   ierr = CeedOperatorRestoreInputs_Ref(numinputfields, qfinputfields,
4881d102b48SJeremy L Thompson                                        opinputfields, false, impl);
489d1bcdac9Sjeremylt   CeedChk(ierr);
4907ca8db16Sjeremylt 
49121617c04Sjeremylt   return 0;
49221617c04Sjeremylt }
49321617c04Sjeremylt 
494*f10650afSjeremylt //------------------------------------------------------------------------------
4951d102b48SJeremy L Thompson // Assemble Linear QFunction
496*f10650afSjeremylt //------------------------------------------------------------------------------
4971d102b48SJeremy L Thompson static int CeedOperatorAssembleLinearQFunction_Ref(CeedOperator op,
4981d102b48SJeremy L Thompson     CeedVector *assembled, CeedElemRestriction *rstr, CeedRequest *request) {
4991d102b48SJeremy L Thompson   int ierr;
5001d102b48SJeremy L Thompson   CeedOperator_Ref *impl;
5011d102b48SJeremy L Thompson   ierr = CeedOperatorGetData(op, (void *)&impl); CeedChk(ierr);
5021d102b48SJeremy L Thompson   CeedQFunction qf;
5031d102b48SJeremy L Thompson   ierr = CeedOperatorGetQFunction(op, &qf); CeedChk(ierr);
5041d102b48SJeremy L Thompson   CeedInt Q, numelements, numinputfields, numoutputfields, size;
5051d102b48SJeremy L Thompson   ierr = CeedOperatorGetNumQuadraturePoints(op, &Q); CeedChk(ierr);
5061d102b48SJeremy L Thompson   ierr = CeedOperatorGetNumElements(op, &numelements); CeedChk(ierr);
5071d102b48SJeremy L Thompson   ierr= CeedQFunctionGetNumArgs(qf, &numinputfields, &numoutputfields);
5081d102b48SJeremy L Thompson   CeedChk(ierr);
5091d102b48SJeremy L Thompson   CeedOperatorField *opinputfields, *opoutputfields;
5101d102b48SJeremy L Thompson   ierr = CeedOperatorGetFields(op, &opinputfields, &opoutputfields);
5111d102b48SJeremy L Thompson   CeedChk(ierr);
5121d102b48SJeremy L Thompson   CeedQFunctionField *qfinputfields, *qfoutputfields;
5131d102b48SJeremy L Thompson   ierr = CeedQFunctionGetFields(qf, &qfinputfields, &qfoutputfields);
5141d102b48SJeremy L Thompson   CeedChk(ierr);
5151d102b48SJeremy L Thompson   CeedVector vec;
5161d102b48SJeremy L Thompson   CeedInt numactivein = 0, numactiveout = 0;
51742ea3801Sjeremylt   CeedVector *activein = NULL;
5181d102b48SJeremy L Thompson   CeedScalar *a, *tmp;
5195107b09fSJeremy L Thompson   Ceed ceed, ceedparent;
5201d102b48SJeremy L Thompson   ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
5215107b09fSJeremy L Thompson   ierr = CeedGetOperatorFallbackParentCeed(ceed, &ceedparent); CeedChk(ierr);
5225107b09fSJeremy L Thompson   ceedparent = ceedparent ? ceedparent : ceed;
5231d102b48SJeremy L Thompson 
5241d102b48SJeremy L Thompson   // Setup
5251d102b48SJeremy L Thompson   ierr = CeedOperatorSetup_Ref(op); CeedChk(ierr);
5261d102b48SJeremy L Thompson 
52716911fdaSjeremylt   // Check for identity
52816911fdaSjeremylt   if (impl->identityqf)
52916911fdaSjeremylt     // LCOV_EXCL_START
53067db23e4Sjeremylt     return CeedError(ceed, 1, "Assembling identity qfunctions not supported");
53116911fdaSjeremylt   // LCOV_EXCL_STOP
53216911fdaSjeremylt 
5331d102b48SJeremy L Thompson   // Input Evecs and Restriction
5341d102b48SJeremy L Thompson   ierr = CeedOperatorSetupInputs_Ref(numinputfields, qfinputfields,
53516911fdaSjeremylt                                      opinputfields, NULL, true, impl, request);
53616911fdaSjeremylt   CeedChk(ierr);
5371d102b48SJeremy L Thompson 
5381d102b48SJeremy L Thompson   // Count number of active input fields
5391d102b48SJeremy L Thompson   for (CeedInt i=0; i<numinputfields; i++) {
5401d102b48SJeremy L Thompson     // Get input vector
5411d102b48SJeremy L Thompson     ierr = CeedOperatorFieldGetVector(opinputfields[i], &vec); CeedChk(ierr);
5421d102b48SJeremy L Thompson     // Check if active input
5431d102b48SJeremy L Thompson     if (vec == CEED_VECTOR_ACTIVE) {
5441d102b48SJeremy L Thompson       ierr = CeedQFunctionFieldGetSize(qfinputfields[i], &size); CeedChk(ierr);
5451d102b48SJeremy L Thompson       ierr = CeedVectorSetValue(impl->qvecsin[i], 0.0); CeedChk(ierr);
5461d102b48SJeremy L Thompson       ierr = CeedVectorGetArray(impl->qvecsin[i], CEED_MEM_HOST, &tmp);
5471d102b48SJeremy L Thompson       CeedChk(ierr);
5481d102b48SJeremy L Thompson       ierr = CeedRealloc(numactivein + size, &activein); CeedChk(ierr);
5491d102b48SJeremy L Thompson       for (CeedInt field=0; field<size; field++) {
55042ea3801Sjeremylt         ierr = CeedVectorCreate(ceed, Q, &activein[numactivein+field]);
55142ea3801Sjeremylt         CeedChk(ierr);
55242ea3801Sjeremylt         ierr = CeedVectorSetArray(activein[numactivein+field], CEED_MEM_HOST,
55342ea3801Sjeremylt                                   CEED_USE_POINTER, &tmp[field*Q]);
554112e3f70Sjeremylt         CeedChk(ierr);
5551d102b48SJeremy L Thompson       }
5561d102b48SJeremy L Thompson       numactivein += size;
5571d102b48SJeremy L Thompson       ierr = CeedVectorRestoreArray(impl->qvecsin[i], &tmp); CeedChk(ierr);
5581d102b48SJeremy L Thompson     }
5591d102b48SJeremy L Thompson   }
5601d102b48SJeremy L Thompson 
5611d102b48SJeremy L Thompson   // Count number of active output fields
5621d102b48SJeremy L Thompson   for (CeedInt i=0; i<numoutputfields; i++) {
5631d102b48SJeremy L Thompson     // Get output vector
5641d102b48SJeremy L Thompson     ierr = CeedOperatorFieldGetVector(opoutputfields[i], &vec); CeedChk(ierr);
5651d102b48SJeremy L Thompson     // Check if active output
5661d102b48SJeremy L Thompson     if (vec == CEED_VECTOR_ACTIVE) {
5671d102b48SJeremy L Thompson       ierr = CeedQFunctionFieldGetSize(qfoutputfields[i], &size); CeedChk(ierr);
5681d102b48SJeremy L Thompson       numactiveout += size;
5691d102b48SJeremy L Thompson     }
5701d102b48SJeremy L Thompson   }
5711d102b48SJeremy L Thompson 
5721d102b48SJeremy L Thompson   // Check sizes
5731d102b48SJeremy L Thompson   if (!numactivein || !numactiveout)
574138d4072Sjeremylt     // LCOV_EXCL_START
575138d4072Sjeremylt     return CeedError(ceed, 1, "Cannot assemble QFunction without active inputs "
576138d4072Sjeremylt                      "and outputs");
577138d4072Sjeremylt   // LCOV_EXCL_STOP
5781d102b48SJeremy L Thompson 
5791d102b48SJeremy L Thompson   // Create output restriction
5805107b09fSJeremy L Thompson   ierr = CeedElemRestrictionCreateIdentity(ceedparent, numelements, Q,
5817f823360Sjeremylt          numelements*Q, numactivein*numactiveout, rstr); CeedChk(ierr);
5821d102b48SJeremy L Thompson   // Create assembled vector
5835107b09fSJeremy L Thompson   ierr = CeedVectorCreate(ceedparent, numelements*Q*numactivein*numactiveout,
5841d102b48SJeremy L Thompson                           assembled); CeedChk(ierr);
5851d102b48SJeremy L Thompson   ierr = CeedVectorSetValue(*assembled, 0.0); CeedChk(ierr);
5861d102b48SJeremy L Thompson   ierr = CeedVectorGetArray(*assembled, CEED_MEM_HOST, &a); CeedChk(ierr);
5871d102b48SJeremy L Thompson 
5881d102b48SJeremy L Thompson   // Loop through elements
5891d102b48SJeremy L Thompson   for (CeedInt e=0; e<numelements; e++) {
5901d102b48SJeremy L Thompson     // Input basis apply
5911d102b48SJeremy L Thompson     ierr = CeedOperatorInputBasis_Ref(e, Q, qfinputfields, opinputfields,
5921d102b48SJeremy L Thompson                                       numinputfields, true, impl);
5931d102b48SJeremy L Thompson     CeedChk(ierr);
5941d102b48SJeremy L Thompson 
5951d102b48SJeremy L Thompson     // Assemble QFunction
5961d102b48SJeremy L Thompson     for (CeedInt in=0; in<numactivein; in++) {
5971d102b48SJeremy L Thompson       // Set Inputs
59842ea3801Sjeremylt       ierr = CeedVectorSetValue(activein[in], 1.0); CeedChk(ierr);
59942ea3801Sjeremylt       if (numactivein > 1) {
60042ea3801Sjeremylt         ierr = CeedVectorSetValue(activein[(in+numactivein-1)%numactivein],
60142ea3801Sjeremylt                                   0.0); CeedChk(ierr);
60242ea3801Sjeremylt       }
6031d102b48SJeremy L Thompson       // Set Outputs
6041d102b48SJeremy L Thompson       for (CeedInt out=0; out<numoutputfields; out++) {
6051d102b48SJeremy L Thompson         // Get output vector
6061d102b48SJeremy L Thompson         ierr = CeedOperatorFieldGetVector(opoutputfields[out], &vec);
6071d102b48SJeremy L Thompson         CeedChk(ierr);
6081d102b48SJeremy L Thompson         // Check if active output
6091d102b48SJeremy L Thompson         if (vec == CEED_VECTOR_ACTIVE) {
6101d102b48SJeremy L Thompson           CeedVectorSetArray(impl->qvecsout[out], CEED_MEM_HOST,
6111d102b48SJeremy L Thompson                              CEED_USE_POINTER, a); CeedChk(ierr);
6121d102b48SJeremy L Thompson           ierr = CeedQFunctionFieldGetSize(qfoutputfields[out], &size);
6131d102b48SJeremy L Thompson           CeedChk(ierr);
6141d102b48SJeremy L Thompson           a += size*Q; // Advance the pointer by the size of the output
6151d102b48SJeremy L Thompson         }
6161d102b48SJeremy L Thompson       }
6171d102b48SJeremy L Thompson       // Apply QFunction
6181d102b48SJeremy L Thompson       ierr = CeedQFunctionApply(qf, Q, impl->qvecsin, impl->qvecsout);
6191d102b48SJeremy L Thompson       CeedChk(ierr);
6201d102b48SJeremy L Thompson     }
6211d102b48SJeremy L Thompson   }
6221d102b48SJeremy L Thompson 
6231d102b48SJeremy L Thompson   // Un-set output Qvecs to prevent accidental overwrite of Assembled
6241d102b48SJeremy L Thompson   for (CeedInt out=0; out<numoutputfields; out++) {
6251d102b48SJeremy L Thompson     // Get output vector
6261d102b48SJeremy L Thompson     ierr = CeedOperatorFieldGetVector(opoutputfields[out], &vec);
6271d102b48SJeremy L Thompson     CeedChk(ierr);
6281d102b48SJeremy L Thompson     // Check if active output
6291d102b48SJeremy L Thompson     if (vec == CEED_VECTOR_ACTIVE) {
6301d102b48SJeremy L Thompson       CeedVectorSetArray(impl->qvecsout[out], CEED_MEM_HOST, CEED_COPY_VALUES,
6311d102b48SJeremy L Thompson                          NULL); CeedChk(ierr);
6321d102b48SJeremy L Thompson     }
6331d102b48SJeremy L Thompson   }
6341d102b48SJeremy L Thompson 
6351d102b48SJeremy L Thompson   // Restore input arrays
6361d102b48SJeremy L Thompson   ierr = CeedOperatorRestoreInputs_Ref(numinputfields, qfinputfields,
6371d102b48SJeremy L Thompson                                        opinputfields, true, impl);
6381d102b48SJeremy L Thompson   CeedChk(ierr);
6391d102b48SJeremy L Thompson 
6401d102b48SJeremy L Thompson   // Restore output
6411d102b48SJeremy L Thompson   ierr = CeedVectorRestoreArray(*assembled, &a); CeedChk(ierr);
6421d102b48SJeremy L Thompson 
6431d102b48SJeremy L Thompson   // Cleanup
64442ea3801Sjeremylt   for (CeedInt i=0; i<numactivein; i++) {
64542ea3801Sjeremylt     ierr = CeedVectorDestroy(&activein[i]); CeedChk(ierr);
64642ea3801Sjeremylt   }
6471d102b48SJeremy L Thompson   ierr = CeedFree(&activein); CeedChk(ierr);
6481d102b48SJeremy L Thompson 
6491d102b48SJeremy L Thompson   return 0;
6501d102b48SJeremy L Thompson }
6511d102b48SJeremy L Thompson 
652*f10650afSjeremylt //------------------------------------------------------------------------------
6537172caadSjeremylt // Assemble Linear Diagonal
654*f10650afSjeremylt //------------------------------------------------------------------------------
6557172caadSjeremylt static int CeedOperatorAssembleLinearDiagonal_Ref(CeedOperator op,
6567172caadSjeremylt     CeedVector *assembled, CeedRequest *request) {
6577172caadSjeremylt   int ierr;
6587172caadSjeremylt 
6597172caadSjeremylt   // Assemble QFunction
6607172caadSjeremylt   CeedQFunction qf;
6617172caadSjeremylt   ierr = CeedOperatorGetQFunction(op, &qf); CeedChk(ierr);
6627172caadSjeremylt   CeedInt numinputfields, numoutputfields;
6637172caadSjeremylt   ierr= CeedQFunctionGetNumArgs(qf, &numinputfields, &numoutputfields);
6647172caadSjeremylt   CeedChk(ierr);
6657172caadSjeremylt   CeedVector assembledqf;
6667172caadSjeremylt   CeedElemRestriction rstr;
6677172caadSjeremylt   ierr = CeedOperatorAssembleLinearQFunction(op,  &assembledqf, &rstr, request);
6687172caadSjeremylt   CeedChk(ierr);
6697172caadSjeremylt   ierr = CeedElemRestrictionDestroy(&rstr); CeedChk(ierr);
6707172caadSjeremylt 
6717172caadSjeremylt   // Determine active input basis
6727172caadSjeremylt   CeedOperatorField *opfields;
6737172caadSjeremylt   CeedQFunctionField *qffields;
6747172caadSjeremylt   ierr = CeedOperatorGetFields(op, &opfields, NULL); CeedChk(ierr);
6757172caadSjeremylt   ierr = CeedQFunctionGetFields(qf, &qffields, NULL); CeedChk(ierr);
6767172caadSjeremylt   CeedInt numemodein = 0, ncomp, dim = 1;
6777172caadSjeremylt   CeedEvalMode *emodein = NULL;
6787172caadSjeremylt   CeedBasis basisin = NULL;
6797172caadSjeremylt   CeedElemRestriction rstrin = NULL;
6807172caadSjeremylt   for (CeedInt i=0; i<numinputfields; i++) {
6817172caadSjeremylt     CeedVector vec;
6827172caadSjeremylt     ierr = CeedOperatorFieldGetVector(opfields[i], &vec); CeedChk(ierr);
6837172caadSjeremylt     if (vec == CEED_VECTOR_ACTIVE) {
6847172caadSjeremylt       ierr = CeedOperatorFieldGetBasis(opfields[i], &basisin);
6857172caadSjeremylt       CeedChk(ierr);
6867172caadSjeremylt       ierr = CeedBasisGetNumComponents(basisin, &ncomp); CeedChk(ierr);
6877172caadSjeremylt       ierr = CeedBasisGetDimension(basisin, &dim); CeedChk(ierr);
6887172caadSjeremylt       ierr = CeedOperatorFieldGetElemRestriction(opfields[i], &rstrin);
6897172caadSjeremylt       CeedChk(ierr);
6907172caadSjeremylt       CeedEvalMode emode;
6917172caadSjeremylt       ierr = CeedQFunctionFieldGetEvalMode(qffields[i], &emode);
6927172caadSjeremylt       CeedChk(ierr);
6937172caadSjeremylt       switch (emode) {
6947172caadSjeremylt       case CEED_EVAL_NONE:
6957172caadSjeremylt       case CEED_EVAL_INTERP:
6967172caadSjeremylt         ierr = CeedRealloc(numemodein + 1, &emodein); CeedChk(ierr);
6977172caadSjeremylt         emodein[numemodein] = emode;
6987172caadSjeremylt         numemodein += 1;
6997172caadSjeremylt         break;
7007172caadSjeremylt       case CEED_EVAL_GRAD:
7017172caadSjeremylt         ierr = CeedRealloc(numemodein + dim, &emodein); CeedChk(ierr);
7027172caadSjeremylt         for (CeedInt d=0; d<dim; d++)
7037172caadSjeremylt           emodein[numemodein+d] = emode;
7047172caadSjeremylt         numemodein += dim;
7057172caadSjeremylt         break;
7067172caadSjeremylt       case CEED_EVAL_WEIGHT:
7077172caadSjeremylt       case CEED_EVAL_DIV:
7087172caadSjeremylt       case CEED_EVAL_CURL:
7097172caadSjeremylt         break; // Caught by QF Assembly
7107172caadSjeremylt       }
7117172caadSjeremylt     }
7127172caadSjeremylt   }
7137172caadSjeremylt 
7147172caadSjeremylt   // Determine active output basis
7157172caadSjeremylt   ierr = CeedOperatorGetFields(op, NULL, &opfields); CeedChk(ierr);
7167172caadSjeremylt   ierr = CeedQFunctionGetFields(qf, NULL, &qffields); CeedChk(ierr);
7177172caadSjeremylt   CeedInt numemodeout = 0;
7187172caadSjeremylt   CeedEvalMode *emodeout = NULL;
7197172caadSjeremylt   CeedBasis basisout = NULL;
7207172caadSjeremylt   CeedElemRestriction rstrout = NULL;
7217172caadSjeremylt   CeedTransposeMode lmodeout = CEED_NOTRANSPOSE;
7227172caadSjeremylt   for (CeedInt i=0; i<numoutputfields; i++) {
7237172caadSjeremylt     CeedVector vec;
7247172caadSjeremylt     ierr = CeedOperatorFieldGetVector(opfields[i], &vec); CeedChk(ierr);
7257172caadSjeremylt     if (vec == CEED_VECTOR_ACTIVE) {
7267172caadSjeremylt       ierr = CeedOperatorFieldGetBasis(opfields[i], &basisout);
7277172caadSjeremylt       CeedChk(ierr);
7287172caadSjeremylt       ierr = CeedOperatorFieldGetElemRestriction(opfields[i], &rstrout);
7297172caadSjeremylt       CeedChk(ierr);
7307172caadSjeremylt       ierr = CeedOperatorFieldGetLMode(opfields[i], &lmodeout);
7317172caadSjeremylt       CeedChk(ierr);
7327172caadSjeremylt       CeedEvalMode emode;
7337172caadSjeremylt       ierr = CeedQFunctionFieldGetEvalMode(qffields[i], &emode);
7347172caadSjeremylt       CeedChk(ierr);
7357172caadSjeremylt       switch (emode) {
7367172caadSjeremylt       case CEED_EVAL_NONE:
7377172caadSjeremylt       case CEED_EVAL_INTERP:
7387172caadSjeremylt         ierr = CeedRealloc(numemodeout + 1, &emodeout); CeedChk(ierr);
7397172caadSjeremylt         emodeout[numemodeout] = emode;
7407172caadSjeremylt         numemodeout += 1;
7417172caadSjeremylt         break;
7427172caadSjeremylt       case CEED_EVAL_GRAD:
7437172caadSjeremylt         ierr = CeedRealloc(numemodeout + dim, &emodeout); CeedChk(ierr);
7447172caadSjeremylt         for (CeedInt d=0; d<dim; d++)
7457172caadSjeremylt           emodeout[numemodeout+d] = emode;
7467172caadSjeremylt         numemodeout += dim;
7477172caadSjeremylt         break;
7487172caadSjeremylt       case CEED_EVAL_WEIGHT:
7497172caadSjeremylt       case CEED_EVAL_DIV:
7507172caadSjeremylt       case CEED_EVAL_CURL:
7517172caadSjeremylt         break; // Caught by QF Assembly
7527172caadSjeremylt       }
7537172caadSjeremylt     }
7547172caadSjeremylt   }
7557172caadSjeremylt 
7567172caadSjeremylt   // Create diagonal vector
7577172caadSjeremylt   CeedVector elemdiag;
7587172caadSjeremylt   ierr = CeedElemRestrictionCreateVector(rstrin, assembled, &elemdiag);
7597172caadSjeremylt   CeedChk(ierr);
7607172caadSjeremylt 
7617172caadSjeremylt   // Assemble element operator diagonals
7627172caadSjeremylt   CeedScalar *elemdiagarray, *assembledqfarray;
7637172caadSjeremylt   ierr = CeedVectorSetValue(elemdiag, 0.0); CeedChk(ierr);
7647172caadSjeremylt   ierr = CeedVectorGetArray(elemdiag, CEED_MEM_HOST, &elemdiagarray);
7657172caadSjeremylt   CeedChk(ierr);
7667172caadSjeremylt   ierr = CeedVectorGetArray(assembledqf, CEED_MEM_HOST, &assembledqfarray);
7677172caadSjeremylt   CeedChk(ierr);
7687172caadSjeremylt   CeedInt nelem, nnodes, nqpts;
7697172caadSjeremylt   ierr = CeedElemRestrictionGetNumElements(rstrin, &nelem); CeedChk(ierr);
7707172caadSjeremylt   ierr = CeedBasisGetNumNodes(basisin, &nnodes); CeedChk(ierr);
7717172caadSjeremylt   ierr = CeedBasisGetNumQuadraturePoints(basisin, &nqpts); CeedChk(ierr);
7727172caadSjeremylt   // Compute the diagonal of B^T D B
7737172caadSjeremylt   // Each node, qpt pair
77400f91b2bSjeremylt   CeedScalar *interpin, *interpout, *gradin, *gradout;
77500f91b2bSjeremylt   ierr = CeedBasisGetInterp(basisin, &interpin); CeedChk(ierr);
77600f91b2bSjeremylt   ierr = CeedBasisGetInterp(basisout, &interpout); CeedChk(ierr);
77700f91b2bSjeremylt   ierr = CeedBasisGetGrad(basisin, &gradin); CeedChk(ierr);
77800f91b2bSjeremylt   ierr = CeedBasisGetGrad(basisout, &gradout); CeedChk(ierr);
77900f91b2bSjeremylt   // Each element and component
780cdb3667fSjeremylt   CeedPragmaSIMD
78100f91b2bSjeremylt   for (CeedInt e=0; e<nelem; e++) {
782cdb3667fSjeremylt     CeedPragmaSIMD
7837172caadSjeremylt     for (CeedInt q=0; q<nqpts; q++) {
784cdb3667fSjeremylt       CeedPragmaSIMD
78500f91b2bSjeremylt       for (CeedInt n=0; n<nnodes; n++) {
7867172caadSjeremylt         CeedInt dout = -1;
7877172caadSjeremylt         // Each basis eval mode pair
7887172caadSjeremylt         for (CeedInt eout=0; eout<numemodeout; eout++) {
78900f91b2bSjeremylt           CeedScalar bt = 0.0;
7907172caadSjeremylt           if (emodeout[eout] == CEED_EVAL_GRAD)
7917172caadSjeremylt             dout += 1;
79200f91b2bSjeremylt           switch (emodeout[eout]) {
79300f91b2bSjeremylt           case CEED_EVAL_NONE:
79400f91b2bSjeremylt             if (n == q)
79500f91b2bSjeremylt               bt = 1.0;
79600f91b2bSjeremylt             break;
79700f91b2bSjeremylt           case CEED_EVAL_INTERP:
79800f91b2bSjeremylt             bt = interpout[q*nnodes+n];
79900f91b2bSjeremylt             break;
80000f91b2bSjeremylt           case CEED_EVAL_GRAD:
80100f91b2bSjeremylt             bt = gradout[(dout*nqpts+q)*nnodes+n];
80200f91b2bSjeremylt             break;
80300f91b2bSjeremylt           case CEED_EVAL_WEIGHT:
80400f91b2bSjeremylt           case CEED_EVAL_DIV:
80500f91b2bSjeremylt           case CEED_EVAL_CURL:
806cdb3667fSjeremylt             break; // Caught by QF Assembly
80700f91b2bSjeremylt           }
8087172caadSjeremylt           CeedInt din = -1;
8097172caadSjeremylt           for (CeedInt ein=0; ein<numemodein; ein++) {
8107172caadSjeremylt             CeedScalar b = 0.0;
8117172caadSjeremylt             if (emodein[ein] == CEED_EVAL_GRAD)
8127172caadSjeremylt               din += 1;
81300f91b2bSjeremylt             switch (emodein[ein]) {
81400f91b2bSjeremylt             case CEED_EVAL_NONE:
81500f91b2bSjeremylt               if (n == q)
81600f91b2bSjeremylt                 b = 1.0;
81700f91b2bSjeremylt               break;
81800f91b2bSjeremylt             case CEED_EVAL_INTERP:
81900f91b2bSjeremylt               b = interpin[q*nnodes+n];
82000f91b2bSjeremylt               break;
82100f91b2bSjeremylt             case CEED_EVAL_GRAD:
82200f91b2bSjeremylt               b = gradin[(din*nqpts+q)*nnodes+n];
82300f91b2bSjeremylt               break;
82400f91b2bSjeremylt             case CEED_EVAL_WEIGHT:
82500f91b2bSjeremylt             case CEED_EVAL_DIV:
82600f91b2bSjeremylt             case CEED_EVAL_CURL:
827cdb3667fSjeremylt               break; // Caught by QF Assembly
82800f91b2bSjeremylt             }
8297172caadSjeremylt             for (CeedInt cout=0; cout<ncomp; cout++) {
8307172caadSjeremylt               CeedScalar db = 0.0;
8317172caadSjeremylt               for (CeedInt cin=0; cin<ncomp; cin++)
8327172caadSjeremylt                 db += assembledqfarray[((((e*numemodein+ein)*ncomp+cin)*
8337172caadSjeremylt                                          numemodeout+eout)*ncomp+cout)*nqpts+q]*b;
8347172caadSjeremylt               elemdiagarray[(e*ncomp+cout)*nnodes+n] += bt * db;
8357172caadSjeremylt             }
8367172caadSjeremylt           }
8377172caadSjeremylt         }
8387172caadSjeremylt       }
83900f91b2bSjeremylt     }
84000f91b2bSjeremylt   }
8417172caadSjeremylt 
8427172caadSjeremylt   ierr = CeedVectorRestoreArray(elemdiag, &elemdiagarray); CeedChk(ierr);
8437172caadSjeremylt   ierr = CeedVectorRestoreArray(assembledqf, &assembledqfarray); CeedChk(ierr);
8447172caadSjeremylt 
8457172caadSjeremylt   // Assemble local operator diagonal
8467172caadSjeremylt   ierr = CeedVectorSetValue(*assembled, 0.0); CeedChk(ierr);
8477172caadSjeremylt   ierr = CeedElemRestrictionApply(rstrout, CEED_TRANSPOSE, lmodeout, elemdiag,
8487172caadSjeremylt                                   *assembled, request); CeedChk(ierr);
8497172caadSjeremylt 
8507172caadSjeremylt   // Cleanup
8517172caadSjeremylt   ierr = CeedVectorDestroy(&assembledqf); CeedChk(ierr);
8527172caadSjeremylt   ierr = CeedVectorDestroy(&elemdiag); CeedChk(ierr);
8537172caadSjeremylt   ierr = CeedFree(&emodein); CeedChk(ierr);
8547172caadSjeremylt   ierr = CeedFree(&emodeout); CeedChk(ierr);
8557172caadSjeremylt 
8567172caadSjeremylt   return 0;
8577172caadSjeremylt }
8587172caadSjeremylt 
859*f10650afSjeremylt //------------------------------------------------------------------------------
860713f43c3Sjeremylt // Create FDM Element Inverse
861*f10650afSjeremylt //------------------------------------------------------------------------------
862713f43c3Sjeremylt int CeedOperatorCreateFDMElementInverse_Ref(CeedOperator op,
863713f43c3Sjeremylt     CeedOperator *fdminv,
864713f43c3Sjeremylt     CeedRequest *request) {
865713f43c3Sjeremylt   int ierr;
866713f43c3Sjeremylt   Ceed ceed, ceedparent;
867713f43c3Sjeremylt   ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
868713f43c3Sjeremylt   ierr = CeedGetOperatorFallbackParentCeed(ceed, &ceedparent); CeedChk(ierr);
869713f43c3Sjeremylt   ceedparent = ceedparent ? ceedparent : ceed;
870713f43c3Sjeremylt   CeedQFunction qf;
871713f43c3Sjeremylt   ierr = CeedOperatorGetQFunction(op, &qf); CeedChk(ierr);
872713f43c3Sjeremylt 
873713f43c3Sjeremylt   // Determine active input basis
874713f43c3Sjeremylt   bool interp = false, grad = false;
875713f43c3Sjeremylt   CeedBasis basis = NULL;
876713f43c3Sjeremylt   CeedElemRestriction rstr = NULL;
877713f43c3Sjeremylt   CeedOperatorField *opfields;
878713f43c3Sjeremylt   CeedQFunctionField *qffields;
879713f43c3Sjeremylt   ierr = CeedOperatorGetFields(op, &opfields, NULL); CeedChk(ierr);
880713f43c3Sjeremylt   ierr = CeedQFunctionGetFields(qf, &qffields, NULL); CeedChk(ierr);
881713f43c3Sjeremylt   CeedInt numinputfields;
882713f43c3Sjeremylt   ierr = CeedQFunctionGetNumArgs(qf, &numinputfields, NULL); CeedChk(ierr);
883713f43c3Sjeremylt   for (CeedInt i=0; i<numinputfields; i++) {
884713f43c3Sjeremylt     CeedVector vec;
885713f43c3Sjeremylt     ierr = CeedOperatorFieldGetVector(opfields[i], &vec); CeedChk(ierr);
886713f43c3Sjeremylt     if (vec == CEED_VECTOR_ACTIVE) {
887713f43c3Sjeremylt       CeedEvalMode emode;
888713f43c3Sjeremylt       ierr = CeedQFunctionFieldGetEvalMode(qffields[i], &emode); CeedChk(ierr);
889713f43c3Sjeremylt       interp = interp || emode == CEED_EVAL_INTERP;
890713f43c3Sjeremylt       grad = grad || emode == CEED_EVAL_GRAD;
891713f43c3Sjeremylt       ierr = CeedOperatorFieldGetBasis(opfields[i], &basis); CeedChk(ierr);
892713f43c3Sjeremylt       ierr = CeedOperatorFieldGetElemRestriction(opfields[i], &rstr);
893713f43c3Sjeremylt       CeedChk(ierr);
894713f43c3Sjeremylt     }
895713f43c3Sjeremylt   }
896713f43c3Sjeremylt   if (!basis)
897713f43c3Sjeremylt     return CeedError(ceed, 1, "No active field set");
898713f43c3Sjeremylt   CeedInt P1d, Q1d, elemsize, nqpts, dim, ncomp = 1, nelem = 1, nnodes = 1;
899713f43c3Sjeremylt   ierr = CeedBasisGetNumNodes1D(basis, &P1d); CeedChk(ierr);
900713f43c3Sjeremylt   ierr = CeedBasisGetNumNodes(basis, &elemsize); CeedChk(ierr);
901713f43c3Sjeremylt   ierr = CeedBasisGetNumQuadraturePoints1D(basis, &Q1d); CeedChk(ierr);
902713f43c3Sjeremylt   ierr = CeedBasisGetNumQuadraturePoints(basis, &nqpts); CeedChk(ierr);
903713f43c3Sjeremylt   ierr = CeedBasisGetDimension(basis, &dim); CeedChk(ierr);
904713f43c3Sjeremylt   ierr = CeedBasisGetNumComponents(basis, &ncomp); CeedChk(ierr);
905713f43c3Sjeremylt   ierr = CeedElemRestrictionGetNumElements(rstr, &nelem); CeedChk(ierr);
906713f43c3Sjeremylt   ierr = CeedElemRestrictionGetNumNodes(rstr, &nnodes); CeedChk(ierr);
907713f43c3Sjeremylt 
908713f43c3Sjeremylt   // Build and diagonalize 1D Mass and Laplacian
909713f43c3Sjeremylt   bool tensorbasis;
910713f43c3Sjeremylt   ierr = CeedBasisGetTensorStatus(basis, &tensorbasis); CeedChk(ierr);
911713f43c3Sjeremylt   if (!tensorbasis)
912713f43c3Sjeremylt     return CeedError(ceed, 1, "FDMElementInverse only supported for tensor "
913713f43c3Sjeremylt                      "bases");
914713f43c3Sjeremylt   CeedScalar *work, *mass, *laplace, *x, *x2, *lambda;
915713f43c3Sjeremylt   ierr = CeedMalloc(Q1d*P1d, &work); CeedChk(ierr);
916713f43c3Sjeremylt   ierr = CeedMalloc(P1d*P1d, &mass); CeedChk(ierr);
917713f43c3Sjeremylt   ierr = CeedMalloc(P1d*P1d, &laplace); CeedChk(ierr);
918713f43c3Sjeremylt   ierr = CeedMalloc(P1d*P1d, &x); CeedChk(ierr);
919713f43c3Sjeremylt   ierr = CeedMalloc(P1d*P1d, &x2); CeedChk(ierr);
920713f43c3Sjeremylt   ierr = CeedMalloc(P1d, &lambda); CeedChk(ierr);
921713f43c3Sjeremylt   // -- Mass
922713f43c3Sjeremylt   CeedScalar *interp1d, *grad1d, *qweight1d;
923713f43c3Sjeremylt   ierr = CeedBasisGetInterp1D(basis, &interp1d); CeedChk(ierr);
924713f43c3Sjeremylt   ierr = CeedBasisGetGrad1D(basis, &grad1d); CeedChk(ierr);
925713f43c3Sjeremylt   ierr = CeedBasisGetQWeights(basis, &qweight1d); CeedChk(ierr);
926713f43c3Sjeremylt   for (CeedInt i=0; i<Q1d; i++)
927713f43c3Sjeremylt     for (CeedInt j=0; j<P1d; j++)
928713f43c3Sjeremylt       work[i+j*Q1d] = interp1d[i*P1d+j]*qweight1d[i];
9299289e5bfSjeremylt   ierr = CeedMatrixMultiply(ceed, (const CeedScalar *)work,
9309289e5bfSjeremylt                             (const CeedScalar *)interp1d, mass, P1d, P1d, Q1d);
931713f43c3Sjeremylt   CeedChk(ierr);
932713f43c3Sjeremylt   // -- Laplacian
933713f43c3Sjeremylt   for (CeedInt i=0; i<Q1d; i++)
934713f43c3Sjeremylt     for (CeedInt j=0; j<P1d; j++)
935713f43c3Sjeremylt       work[i+j*Q1d] = grad1d[i*P1d+j]*qweight1d[i];
9369289e5bfSjeremylt   ierr = CeedMatrixMultiply(ceed, (const CeedScalar *)work,
9379289e5bfSjeremylt                             (const CeedScalar *)grad1d, laplace, P1d, P1d, Q1d);
938713f43c3Sjeremylt   CeedChk(ierr);
939713f43c3Sjeremylt   // -- Diagonalize
940713f43c3Sjeremylt   ierr = CeedSimultaneousDiagonalization(ceed, laplace, mass, x, lambda, P1d);
941713f43c3Sjeremylt   CeedChk(ierr);
942713f43c3Sjeremylt   ierr = CeedFree(&work); CeedChk(ierr);
943713f43c3Sjeremylt   ierr = CeedFree(&mass); CeedChk(ierr);
944713f43c3Sjeremylt   ierr = CeedFree(&laplace); CeedChk(ierr);
945713f43c3Sjeremylt   for (CeedInt i=0; i<P1d; i++)
946713f43c3Sjeremylt     for (CeedInt j=0; j<P1d; j++)
947713f43c3Sjeremylt       x2[i+j*P1d] = x[j+i*P1d];
948713f43c3Sjeremylt   ierr = CeedFree(&x); CeedChk(ierr);
949713f43c3Sjeremylt 
950713f43c3Sjeremylt   // Assemble QFunction
951713f43c3Sjeremylt   CeedVector assembled;
952713f43c3Sjeremylt   CeedElemRestriction rstr_qf;
953713f43c3Sjeremylt   ierr =  CeedOperatorAssembleLinearQFunction(op, &assembled, &rstr_qf,
954713f43c3Sjeremylt           request); CeedChk(ierr);
955713f43c3Sjeremylt   ierr = CeedElemRestrictionDestroy(&rstr_qf); CeedChk(ierr);
956713f43c3Sjeremylt 
957713f43c3Sjeremylt   // Calculate element averages
958713f43c3Sjeremylt   CeedInt nfields = ((interp?1:0) + (grad?dim:0))*((interp?1:0) + (grad?dim:0));
959713f43c3Sjeremylt   CeedScalar *elemavg;
960713f43c3Sjeremylt   const CeedScalar *assembledarray, *qweightsarray;
961713f43c3Sjeremylt   CeedVector qweights;
962713f43c3Sjeremylt   ierr = CeedVectorCreate(ceedparent, nqpts, &qweights); CeedChk(ierr);
963713f43c3Sjeremylt   ierr = CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, CEED_EVAL_WEIGHT,
964713f43c3Sjeremylt                         CEED_VECTOR_NONE, qweights); CeedChk(ierr);
965713f43c3Sjeremylt   ierr = CeedVectorGetArrayRead(assembled, CEED_MEM_HOST, &assembledarray);
966713f43c3Sjeremylt   CeedChk(ierr);
967713f43c3Sjeremylt   ierr = CeedVectorGetArrayRead(qweights, CEED_MEM_HOST, &qweightsarray);
968713f43c3Sjeremylt   CeedChk(ierr);
969713f43c3Sjeremylt   ierr = CeedCalloc(nelem, &elemavg); CeedChk(ierr);
970713f43c3Sjeremylt   for (CeedInt e=0; e<nelem; e++) {
971713f43c3Sjeremylt     CeedInt count = 0;
972713f43c3Sjeremylt     for (CeedInt q=0; q<nqpts; q++)
973713f43c3Sjeremylt       for (CeedInt i=0; i<ncomp*ncomp*nfields; i++)
974713f43c3Sjeremylt         if (fabs(assembledarray[e*nelem*nqpts*ncomp*ncomp*nfields +
975713f43c3Sjeremylt                                                                   i*nqpts + q]) > CEED_EPSILON) {
976713f43c3Sjeremylt           elemavg[e] += assembledarray[e*nelem*nqpts*ncomp*ncomp*nfields +
977713f43c3Sjeremylt                                        i*nqpts + q] / qweightsarray[q];
978713f43c3Sjeremylt           count++;
979713f43c3Sjeremylt         }
980713f43c3Sjeremylt     if (count)
981713f43c3Sjeremylt       elemavg[e] /= count;
982713f43c3Sjeremylt   }
983713f43c3Sjeremylt   ierr = CeedVectorRestoreArrayRead(assembled, &assembledarray); CeedChk(ierr);
984713f43c3Sjeremylt   ierr = CeedVectorDestroy(&assembled); CeedChk(ierr);
985713f43c3Sjeremylt   ierr = CeedVectorRestoreArrayRead(qweights, &qweightsarray); CeedChk(ierr);
986713f43c3Sjeremylt   ierr = CeedVectorDestroy(&qweights); CeedChk(ierr);
987713f43c3Sjeremylt 
988713f43c3Sjeremylt   // Build FDM diagonal
989713f43c3Sjeremylt   CeedVector qdata;
990713f43c3Sjeremylt   CeedScalar *qdataarray;
991713f43c3Sjeremylt   ierr = CeedVectorCreate(ceedparent, nelem*ncomp*nnodes, &qdata); CeedChk(ierr);
992713f43c3Sjeremylt   ierr = CeedVectorSetArray(qdata, CEED_MEM_HOST, CEED_COPY_VALUES, NULL);
993713f43c3Sjeremylt   CeedChk(ierr);
994713f43c3Sjeremylt   ierr = CeedVectorGetArray(qdata, CEED_MEM_HOST, &qdataarray); CeedChk(ierr);
995713f43c3Sjeremylt   for (CeedInt e=0; e<nelem; e++)
996713f43c3Sjeremylt     for (CeedInt c=0; c<ncomp; c++)
997713f43c3Sjeremylt       for (CeedInt n=0; n<nnodes; n++) {
998713f43c3Sjeremylt         if (interp)
999713f43c3Sjeremylt           qdataarray[(e*ncomp+c)*nnodes+n] = 1;
1000713f43c3Sjeremylt         if (grad)
1001713f43c3Sjeremylt           for (CeedInt d=0; d<dim; d++) {
1002713f43c3Sjeremylt             CeedInt i = (n / CeedIntPow(P1d, d)) % P1d;
1003713f43c3Sjeremylt             qdataarray[(e*ncomp+c)*nnodes+n] += lambda[i];
1004713f43c3Sjeremylt           }
1005713f43c3Sjeremylt         qdataarray[(e*ncomp+c)*nnodes+n] = 1 / (elemavg[e] *
1006713f43c3Sjeremylt                                                 qdataarray[(e*ncomp+c)*nnodes+n]);
1007713f43c3Sjeremylt       }
1008713f43c3Sjeremylt   ierr = CeedFree(&elemavg); CeedChk(ierr);
1009713f43c3Sjeremylt   ierr = CeedVectorRestoreArray(qdata, &qdataarray); CeedChk(ierr);
1010713f43c3Sjeremylt 
1011713f43c3Sjeremylt   // Setup FDM operator
1012713f43c3Sjeremylt   // -- Basis
1013713f43c3Sjeremylt   CeedBasis fdm_basis;
1014713f43c3Sjeremylt   CeedScalar *graddummy, *qrefdummy, *qweightdummy;
1015713f43c3Sjeremylt   ierr = CeedCalloc(P1d*P1d, &graddummy); CeedChk(ierr);
1016713f43c3Sjeremylt   ierr = CeedCalloc(P1d, &qrefdummy); CeedChk(ierr);
1017713f43c3Sjeremylt   ierr = CeedCalloc(P1d, &qweightdummy); CeedChk(ierr);
1018713f43c3Sjeremylt   ierr = CeedBasisCreateTensorH1(ceedparent, dim, ncomp, P1d, P1d, x2,
1019713f43c3Sjeremylt                                  graddummy, qrefdummy, qweightdummy,
1020713f43c3Sjeremylt                                  &fdm_basis); CeedChk(ierr);
1021713f43c3Sjeremylt   ierr = CeedFree(&graddummy); CeedChk(ierr);
1022713f43c3Sjeremylt   ierr = CeedFree(&qrefdummy); CeedChk(ierr);
1023713f43c3Sjeremylt   ierr = CeedFree(&qweightdummy); CeedChk(ierr);
1024713f43c3Sjeremylt   ierr = CeedFree(&x2); CeedChk(ierr);
1025713f43c3Sjeremylt   ierr = CeedFree(&lambda); CeedChk(ierr);
1026713f43c3Sjeremylt 
1027713f43c3Sjeremylt   // -- Restriction
1028713f43c3Sjeremylt   CeedElemRestriction rstr_i;
1029713f43c3Sjeremylt   ierr = CeedElemRestrictionCreateIdentity(ceedparent, nelem, nnodes,
1030713f43c3Sjeremylt          nnodes*nelem, ncomp, &rstr_i); CeedChk(ierr);
1031713f43c3Sjeremylt   // -- QFunction
1032713f43c3Sjeremylt   CeedQFunction mass_qf;
1033713f43c3Sjeremylt   ierr = CeedQFunctionCreateInteriorByName(ceedparent, "MassApply", &mass_qf);
1034713f43c3Sjeremylt   CeedChk(ierr);
1035713f43c3Sjeremylt   // -- Operator
1036713f43c3Sjeremylt   ierr = CeedOperatorCreate(ceedparent, mass_qf, NULL, NULL, fdminv);
1037713f43c3Sjeremylt   CeedChk(ierr);
1038713f43c3Sjeremylt   CeedOperatorSetField(*fdminv, "u", rstr_i, CEED_NOTRANSPOSE,
1039713f43c3Sjeremylt                        fdm_basis, CEED_VECTOR_ACTIVE); CeedChk(ierr);
1040713f43c3Sjeremylt   CeedOperatorSetField(*fdminv, "qdata", rstr_i, CEED_NOTRANSPOSE,
1041713f43c3Sjeremylt                        CEED_BASIS_COLLOCATED, qdata); CeedChk(ierr);
1042713f43c3Sjeremylt   CeedOperatorSetField(*fdminv, "v", rstr_i, CEED_NOTRANSPOSE,
1043713f43c3Sjeremylt                        fdm_basis, CEED_VECTOR_ACTIVE); CeedChk(ierr);
1044713f43c3Sjeremylt 
1045713f43c3Sjeremylt   // Cleanup
1046713f43c3Sjeremylt   ierr = CeedVectorDestroy(&qdata); CeedChk(ierr);
1047713f43c3Sjeremylt   ierr = CeedBasisDestroy(&fdm_basis); CeedChk(ierr);
1048713f43c3Sjeremylt   ierr = CeedElemRestrictionDestroy(&rstr_i); CeedChk(ierr);
1049713f43c3Sjeremylt   ierr = CeedQFunctionDestroy(&mass_qf); CeedChk(ierr);
1050713f43c3Sjeremylt 
1051713f43c3Sjeremylt   return 0;
1052713f43c3Sjeremylt }
1053713f43c3Sjeremylt 
1054*f10650afSjeremylt //------------------------------------------------------------------------------
1055*f10650afSjeremylt // Operator Destroy
1056*f10650afSjeremylt //------------------------------------------------------------------------------
1057*f10650afSjeremylt static int CeedOperatorDestroy_Ref(CeedOperator op) {
1058*f10650afSjeremylt   int ierr;
1059*f10650afSjeremylt   CeedOperator_Ref *impl;
1060*f10650afSjeremylt   ierr = CeedOperatorGetData(op, (void *)&impl); CeedChk(ierr);
1061*f10650afSjeremylt 
1062*f10650afSjeremylt   for (CeedInt i=0; i<impl->numein+impl->numeout; i++) {
1063*f10650afSjeremylt     ierr = CeedVectorDestroy(&impl->evecs[i]); CeedChk(ierr);
1064*f10650afSjeremylt   }
1065*f10650afSjeremylt   ierr = CeedFree(&impl->evecs); CeedChk(ierr);
1066*f10650afSjeremylt   ierr = CeedFree(&impl->edata); CeedChk(ierr);
1067*f10650afSjeremylt   ierr = CeedFree(&impl->inputstate); CeedChk(ierr);
1068*f10650afSjeremylt 
1069*f10650afSjeremylt   for (CeedInt i=0; i<impl->numein; i++) {
1070*f10650afSjeremylt     ierr = CeedVectorDestroy(&impl->evecsin[i]); CeedChk(ierr);
1071*f10650afSjeremylt     ierr = CeedVectorDestroy(&impl->qvecsin[i]); CeedChk(ierr);
1072*f10650afSjeremylt   }
1073*f10650afSjeremylt   ierr = CeedFree(&impl->evecsin); CeedChk(ierr);
1074*f10650afSjeremylt   ierr = CeedFree(&impl->qvecsin); CeedChk(ierr);
1075*f10650afSjeremylt 
1076*f10650afSjeremylt   for (CeedInt i=0; i<impl->numeout; i++) {
1077*f10650afSjeremylt     ierr = CeedVectorDestroy(&impl->evecsout[i]); CeedChk(ierr);
1078*f10650afSjeremylt     ierr = CeedVectorDestroy(&impl->qvecsout[i]); CeedChk(ierr);
1079*f10650afSjeremylt   }
1080*f10650afSjeremylt   ierr = CeedFree(&impl->evecsout); CeedChk(ierr);
1081*f10650afSjeremylt   ierr = CeedFree(&impl->qvecsout); CeedChk(ierr);
1082*f10650afSjeremylt 
1083*f10650afSjeremylt   ierr = CeedFree(&impl); CeedChk(ierr);
1084*f10650afSjeremylt   return 0;
1085*f10650afSjeremylt }
1086*f10650afSjeremylt 
1087*f10650afSjeremylt //------------------------------------------------------------------------------
1088713f43c3Sjeremylt // Operator Create
1089*f10650afSjeremylt //------------------------------------------------------------------------------
109021617c04Sjeremylt int CeedOperatorCreate_Ref(CeedOperator op) {
109121617c04Sjeremylt   int ierr;
1092fe2413ffSjeremylt   Ceed ceed;
1093fe2413ffSjeremylt   ierr = CeedOperatorGetCeed(op, &ceed); CeedChk(ierr);
10944ce2993fSjeremylt   CeedOperator_Ref *impl;
109521617c04Sjeremylt 
109621617c04Sjeremylt   ierr = CeedCalloc(1, &impl); CeedChk(ierr);
1097de686571SJeremy L Thompson   ierr = CeedOperatorSetData(op, (void *)&impl); CeedChk(ierr);
1098fe2413ffSjeremylt 
10991d102b48SJeremy L Thompson   ierr = CeedSetBackendFunction(ceed, "Operator", op, "AssembleLinearQFunction",
11001d102b48SJeremy L Thompson                                 CeedOperatorAssembleLinearQFunction_Ref);
11011d102b48SJeremy L Thompson   CeedChk(ierr);
11027172caadSjeremylt   ierr = CeedSetBackendFunction(ceed, "Operator", op, "AssembleLinearDiagonal",
11037172caadSjeremylt                                 CeedOperatorAssembleLinearDiagonal_Ref);
11047172caadSjeremylt   CeedChk(ierr);
1105713f43c3Sjeremylt   ierr = CeedSetBackendFunction(ceed, "Operator", op, "CreateFDMElementInverse",
1106713f43c3Sjeremylt                                 CeedOperatorCreateFDMElementInverse_Ref);
1107713f43c3Sjeremylt   CeedChk(ierr);
1108cae8b89aSjeremylt   ierr = CeedSetBackendFunction(ceed, "Operator", op, "ApplyAdd",
1109fe2413ffSjeremylt                                 CeedOperatorApply_Ref); CeedChk(ierr);
1110fe2413ffSjeremylt   ierr = CeedSetBackendFunction(ceed, "Operator", op, "Destroy",
1111fe2413ffSjeremylt                                 CeedOperatorDestroy_Ref); CeedChk(ierr);
111221617c04Sjeremylt   return 0;
111321617c04Sjeremylt }
1114*f10650afSjeremylt //------------------------------------------------------------------------------
1115