xref: /libCEED/tests/t401-qfunction.c (revision 288c044332e33f37503f09b6484fec9d0a55fba1)
11e35832bSjeremylt /// @file
21e35832bSjeremylt /// Test creation, evaluation, and destruction for qfunction
31e35832bSjeremylt /// \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];
121e35832bSjeremylt   CeedVector Qdata, W, U, V;
131e35832bSjeremylt   CeedQFunction qf_setup, qf_mass;
141e35832bSjeremylt   CeedInt Q = 8;
151e35832bSjeremylt   const CeedScalar *vv;
161e35832bSjeremylt   CeedScalar w[Q], u[Q], v[Q], ctx[5] = {1, 2, 3, 4, 5};
171e35832bSjeremylt 
181e35832bSjeremylt 
191e35832bSjeremylt   CeedInit(argv[1], &ceed);
201e35832bSjeremylt 
214d537eeaSYohann   CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_setup);
221e35832bSjeremylt   CeedQFunctionAddInput(qf_setup, "w", 1, CEED_EVAL_INTERP);
231e35832bSjeremylt   CeedQFunctionAddOutput(qf_setup, "qdata", 1, CEED_EVAL_INTERP);
241e35832bSjeremylt 
254d537eeaSYohann   CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_mass);
261e35832bSjeremylt   CeedQFunctionAddInput(qf_mass, "qdata", 1, CEED_EVAL_INTERP);
271e35832bSjeremylt   CeedQFunctionAddInput(qf_mass, "u", 1, CEED_EVAL_INTERP);
281e35832bSjeremylt   CeedQFunctionAddOutput(qf_mass, "v", 1, CEED_EVAL_INTERP);
291e35832bSjeremylt 
301e35832bSjeremylt   CeedQFunctionSetContext(qf_mass, &ctx, sizeof(ctx));
311e35832bSjeremylt 
321e35832bSjeremylt   for (CeedInt i=0; i<Q; i++) {
331e35832bSjeremylt     CeedScalar x = 2.*i/(Q-1) - 1;
341e35832bSjeremylt     w[i] = 1 - x*x;
351e35832bSjeremylt     u[i] = 2 + 3*x + 5*x*x;
361e35832bSjeremylt     v[i] = w[i] * u[i];
371e35832bSjeremylt   }
381e35832bSjeremylt 
391e35832bSjeremylt   CeedVectorCreate(ceed, Q, &W);
40*288c0443SJeremy L Thompson   CeedVectorSetArray(W, CEED_MEM_HOST, CEED_USE_POINTER, w);
411e35832bSjeremylt   CeedVectorCreate(ceed, Q, &U);
42*288c0443SJeremy L Thompson   CeedVectorSetArray(U, CEED_MEM_HOST, CEED_USE_POINTER, u);
431e35832bSjeremylt   CeedVectorCreate(ceed, Q, &V);
441e35832bSjeremylt   CeedVectorSetValue(V, 0);
451e35832bSjeremylt   CeedVectorCreate(ceed, Q, &Qdata);
461e35832bSjeremylt   CeedVectorSetValue(Qdata, 0);
471e35832bSjeremylt 
481e35832bSjeremylt   {
491e35832bSjeremylt     in[0] = W;
501e35832bSjeremylt     out[0] = Qdata;
511e35832bSjeremylt     CeedQFunctionApply(qf_setup, Q, in, out);
521e35832bSjeremylt   }
531e35832bSjeremylt   {
541e35832bSjeremylt     in[0] = W;
551e35832bSjeremylt     in[1] = U;
561e35832bSjeremylt     out[0] = V;
571e35832bSjeremylt     CeedQFunctionApply(qf_mass, Q, in, out);
581e35832bSjeremylt   }
591e35832bSjeremylt 
601e35832bSjeremylt   CeedVectorGetArrayRead(V, CEED_MEM_HOST, &vv);
61a2546046Sjeremylt   for (CeedInt i=0; i<Q; i++)
621e35832bSjeremylt     if (fabs(ctx[4] * v[i] - vv[i]) > 1.e-14)
63a2546046Sjeremylt       // LCOV_EXCL_START
641e35832bSjeremylt       printf("[%d] v %f != vv %f\n",i, v[i], vv[i]);
65de996c55Sjeremylt   // LCOV_EXCL_STOP
661e35832bSjeremylt   CeedVectorRestoreArrayRead(V, &vv);
671e35832bSjeremylt 
681e35832bSjeremylt   CeedVectorDestroy(&W);
691e35832bSjeremylt   CeedVectorDestroy(&U);
701e35832bSjeremylt   CeedVectorDestroy(&V);
711e35832bSjeremylt   CeedVectorDestroy(&Qdata);
721e35832bSjeremylt   CeedQFunctionDestroy(&qf_setup);
731e35832bSjeremylt   CeedQFunctionDestroy(&qf_mass);
741e35832bSjeremylt   CeedDestroy(&ceed);
751e35832bSjeremylt   return 0;
761e35832bSjeremylt }
77