11e35832bSjeremylt /// @file 2d1d35e2fSjeremylt /// Test creation, evaluation, and destruction for QFunction 3d1d35e2fSjeremylt /// \test Test creation, evaluation, and destruction for QFunction 41e35832bSjeremylt #include <ceed.h> 51e35832bSjeremylt #include <math.h> 61e35832bSjeremylt 74d537eeaSYohann #include "t401-qfunction.h" 81e35832bSjeremylt 91e35832bSjeremylt int main(int argc, char **argv) { 101e35832bSjeremylt Ceed ceed; 111e35832bSjeremylt CeedVector in[16], out[16]; 12d1d35e2fSjeremylt CeedVector Q_data, W, U, V; 131e35832bSjeremylt CeedQFunction qf_setup, qf_mass; 14777ff853SJeremy L Thompson CeedQFunctionContext ctx; 151e35832bSjeremylt CeedInt Q = 8; 161e35832bSjeremylt const CeedScalar *vv; 17*732e4b6cSJeremy L Thompson CeedScalar w[Q], u[Q], v[Q], ctx_data[5] = {1, 2, 3, 4, 5}; 181e35832bSjeremylt 191e35832bSjeremylt CeedInit(argv[1], &ceed); 201e35832bSjeremylt 214d537eeaSYohann CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_setup); 2284e209c4Sjeremylt CeedQFunctionAddInput(qf_setup, "w", 1, CEED_EVAL_WEIGHT); 2384e209c4Sjeremylt CeedQFunctionAddOutput(qf_setup, "qdata", 1, CEED_EVAL_NONE); 241e35832bSjeremylt 254d537eeaSYohann CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_mass); 2684e209c4Sjeremylt CeedQFunctionAddInput(qf_mass, "qdata", 1, CEED_EVAL_NONE); 271e35832bSjeremylt CeedQFunctionAddInput(qf_mass, "u", 1, CEED_EVAL_INTERP); 281e35832bSjeremylt CeedQFunctionAddOutput(qf_mass, "v", 1, CEED_EVAL_INTERP); 291e35832bSjeremylt 30777ff853SJeremy L Thompson CeedQFunctionContextCreate(ceed, &ctx); 31777ff853SJeremy L Thompson CeedQFunctionContextSetData(ctx, CEED_MEM_HOST, CEED_USE_POINTER, 32*732e4b6cSJeremy L Thompson sizeof(ctx_data), &ctx_data); 33777ff853SJeremy L Thompson CeedQFunctionSetContext(qf_mass, ctx); 341e35832bSjeremylt 351e35832bSjeremylt for (CeedInt i=0; i<Q; i++) { 361e35832bSjeremylt CeedScalar x = 2.*i/(Q-1) - 1; 371e35832bSjeremylt w[i] = 1 - x*x; 381e35832bSjeremylt u[i] = 2 + 3*x + 5*x*x; 391e35832bSjeremylt v[i] = w[i] * u[i]; 401e35832bSjeremylt } 411e35832bSjeremylt 421e35832bSjeremylt CeedVectorCreate(ceed, Q, &W); 43288c0443SJeremy L Thompson CeedVectorSetArray(W, CEED_MEM_HOST, CEED_USE_POINTER, w); 441e35832bSjeremylt CeedVectorCreate(ceed, Q, &U); 45288c0443SJeremy L Thompson CeedVectorSetArray(U, CEED_MEM_HOST, CEED_USE_POINTER, u); 461e35832bSjeremylt CeedVectorCreate(ceed, Q, &V); 471e35832bSjeremylt CeedVectorSetValue(V, 0); 48d1d35e2fSjeremylt CeedVectorCreate(ceed, Q, &Q_data); 49d1d35e2fSjeremylt CeedVectorSetValue(Q_data, 0); 501e35832bSjeremylt 511e35832bSjeremylt { 521e35832bSjeremylt in[0] = W; 53d1d35e2fSjeremylt out[0] = Q_data; 541e35832bSjeremylt CeedQFunctionApply(qf_setup, Q, in, out); 551e35832bSjeremylt } 561e35832bSjeremylt { 571e35832bSjeremylt in[0] = W; 581e35832bSjeremylt in[1] = U; 591e35832bSjeremylt out[0] = V; 601e35832bSjeremylt CeedQFunctionApply(qf_mass, Q, in, out); 611e35832bSjeremylt } 621e35832bSjeremylt 631e35832bSjeremylt CeedVectorGetArrayRead(V, CEED_MEM_HOST, &vv); 64a2546046Sjeremylt for (CeedInt i=0; i<Q; i++) 65*732e4b6cSJeremy L Thompson if (fabs(ctx_data[4] * v[i] - vv[i]) > 100.*CEED_EPSILON) 66a2546046Sjeremylt // LCOV_EXCL_START 671e35832bSjeremylt printf("[%d] v %f != vv %f\n",i, v[i], vv[i]); 68de996c55Sjeremylt // LCOV_EXCL_STOP 691e35832bSjeremylt CeedVectorRestoreArrayRead(V, &vv); 701e35832bSjeremylt 711e35832bSjeremylt CeedVectorDestroy(&W); 721e35832bSjeremylt CeedVectorDestroy(&U); 731e35832bSjeremylt CeedVectorDestroy(&V); 74d1d35e2fSjeremylt CeedVectorDestroy(&Q_data); 751e35832bSjeremylt CeedQFunctionDestroy(&qf_setup); 761e35832bSjeremylt CeedQFunctionDestroy(&qf_mass); 77777ff853SJeremy L Thompson CeedQFunctionContextDestroy(&ctx); 781e35832bSjeremylt CeedDestroy(&ceed); 791e35832bSjeremylt return 0; 801e35832bSjeremylt } 81