1*ccec5866SJeremy L Thompson /// @file 2*ccec5866SJeremy L Thompson /// Test QFunction helper macro 3*ccec5866SJeremy L Thompson /// \test Test QFunction helper macro 4*ccec5866SJeremy L Thompson #include <ceed.h> 5*ccec5866SJeremy L Thompson #include <math.h> 6*ccec5866SJeremy L Thompson #include <stdlib.h> 7*ccec5866SJeremy L Thompson #include <string.h> 8*ccec5866SJeremy L Thompson #include "t406-qfunction-helper.h" 9*ccec5866SJeremy L Thompson #include "t406-qfunction.h" 10*ccec5866SJeremy L Thompson 11*ccec5866SJeremy L Thompson int main(int argc, char **argv) { 12*ccec5866SJeremy L Thompson Ceed ceed; 13*ccec5866SJeremy L Thompson CeedVector in[16], out[16]; 14*ccec5866SJeremy L Thompson CeedVector Q_data, W, U, V; 15*ccec5866SJeremy L Thompson CeedQFunction qf_setup, qf_mass; 16*ccec5866SJeremy L Thompson CeedInt Q = 8; 17*ccec5866SJeremy L Thompson const CeedScalar *vv; 18*ccec5866SJeremy L Thompson CeedScalar w[Q], u[Q], v[Q]; 19*ccec5866SJeremy L Thompson 20*ccec5866SJeremy L Thompson CeedInit(argv[1], &ceed); 21*ccec5866SJeremy L Thompson 22*ccec5866SJeremy L Thompson CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_setup); 23*ccec5866SJeremy L Thompson CeedQFunctionAddInput(qf_setup, "w", 1, CEED_EVAL_WEIGHT); 24*ccec5866SJeremy L Thompson CeedQFunctionAddOutput(qf_setup, "qdata", 1, CEED_EVAL_NONE); 25*ccec5866SJeremy L Thompson 26*ccec5866SJeremy L Thompson CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_mass); 27*ccec5866SJeremy L Thompson CeedQFunctionAddInput(qf_mass, "qdata", 1, CEED_EVAL_NONE); 28*ccec5866SJeremy L Thompson CeedQFunctionAddInput(qf_mass, "u", 1, CEED_EVAL_INTERP); 29*ccec5866SJeremy L Thompson CeedQFunctionAddOutput(qf_mass, "v", 1, CEED_EVAL_INTERP); 30*ccec5866SJeremy L Thompson 31*ccec5866SJeremy L Thompson for (CeedInt i=0; i<Q; i++) { 32*ccec5866SJeremy L Thompson CeedScalar x = 2.*i/(Q-1) - 1; 33*ccec5866SJeremy L Thompson w[i] = 1 - x*x; 34*ccec5866SJeremy L Thompson u[i] = 2 + 3*x + 5*x*x; 35*ccec5866SJeremy L Thompson v[i] = w[i] * u[i]; 36*ccec5866SJeremy L Thompson } 37*ccec5866SJeremy L Thompson 38*ccec5866SJeremy L Thompson CeedVectorCreate(ceed, Q, &W); 39*ccec5866SJeremy L Thompson CeedVectorSetArray(W, CEED_MEM_HOST, CEED_USE_POINTER, w); 40*ccec5866SJeremy L Thompson CeedVectorCreate(ceed, Q, &U); 41*ccec5866SJeremy L Thompson CeedVectorSetArray(U, CEED_MEM_HOST, CEED_USE_POINTER, u); 42*ccec5866SJeremy L Thompson CeedVectorCreate(ceed, Q, &V); 43*ccec5866SJeremy L Thompson CeedVectorSetValue(V, 0); 44*ccec5866SJeremy L Thompson CeedVectorCreate(ceed, Q, &Q_data); 45*ccec5866SJeremy L Thompson CeedVectorSetValue(Q_data, 0); 46*ccec5866SJeremy L Thompson 47*ccec5866SJeremy L Thompson { 48*ccec5866SJeremy L Thompson in[0] = W; 49*ccec5866SJeremy L Thompson out[0] = Q_data; 50*ccec5866SJeremy L Thompson CeedQFunctionApply(qf_setup, Q, in, out); 51*ccec5866SJeremy L Thompson } 52*ccec5866SJeremy L Thompson { 53*ccec5866SJeremy L Thompson in[0] = W; 54*ccec5866SJeremy L Thompson in[1] = U; 55*ccec5866SJeremy L Thompson out[0] = V; 56*ccec5866SJeremy L Thompson CeedQFunctionApply(qf_mass, Q, in, out); 57*ccec5866SJeremy L Thompson } 58*ccec5866SJeremy L Thompson 59*ccec5866SJeremy L Thompson CeedVectorGetArrayRead(V, CEED_MEM_HOST, &vv); 60*ccec5866SJeremy L Thompson for (CeedInt i=0; i<Q; i++) 61*ccec5866SJeremy L Thompson if (fabs(5*v[i]*sqrt(2.) - vv[i]) > 1E3*CEED_EPSILON) 62*ccec5866SJeremy L Thompson // LCOV_EXCL_START 63*ccec5866SJeremy L Thompson printf("[%d] v %f != vv %f\n",i, 5*v[i]*sqrt(2.), vv[i]); 64*ccec5866SJeremy L Thompson // LCOV_EXCL_STOP 65*ccec5866SJeremy L Thompson CeedVectorRestoreArrayRead(V, &vv); 66*ccec5866SJeremy L Thompson 67*ccec5866SJeremy L Thompson CeedVectorDestroy(&W); 68*ccec5866SJeremy L Thompson CeedVectorDestroy(&U); 69*ccec5866SJeremy L Thompson CeedVectorDestroy(&V); 70*ccec5866SJeremy L Thompson CeedVectorDestroy(&Q_data); 71*ccec5866SJeremy L Thompson CeedQFunctionDestroy(&qf_setup); 72*ccec5866SJeremy L Thompson CeedQFunctionDestroy(&qf_mass); 73*ccec5866SJeremy L Thompson CeedDestroy(&ceed); 74*ccec5866SJeremy L Thompson return 0; 75*ccec5866SJeremy L Thompson } 76