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