14411cf47Sjeremylt /// @file 2d1d35e2fSjeremylt /// Test creation, evaluation, and destruction for QFunction 3d1d35e2fSjeremylt /// \test Test creation, evaluation, and destruction for QFunction 44d537eeaSYohann #include "t400-qfunction.h" 557c64913Sjeremylt 6*2b730f8bSJeremy L Thompson #include <ceed.h> 7*2b730f8bSJeremy L Thompson 857c64913Sjeremylt int main(int argc, char **argv) { 957c64913Sjeremylt Ceed ceed; 10aedaa0e5Sjeremylt CeedVector in[16], out[16]; 11d1d35e2fSjeremylt CeedVector Q_data, W, U, V; 1257c64913Sjeremylt CeedQFunction qf_setup, qf_mass; 1357c64913Sjeremylt CeedInt Q = 8; 14aedaa0e5Sjeremylt const CeedScalar *vv; 15aedaa0e5Sjeremylt CeedScalar w[Q], u[Q], v[Q]; 1657c64913Sjeremylt 1757c64913Sjeremylt CeedInit(argv[1], &ceed); 18aedaa0e5Sjeremylt 194d537eeaSYohann CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_setup); 2084e209c4Sjeremylt CeedQFunctionAddInput(qf_setup, "w", 1, CEED_EVAL_WEIGHT); 2184e209c4Sjeremylt CeedQFunctionAddOutput(qf_setup, "qdata", 1, CEED_EVAL_NONE); 2257c64913Sjeremylt 234d537eeaSYohann CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_mass); 2484e209c4Sjeremylt CeedQFunctionAddInput(qf_mass, "qdata", 1, CEED_EVAL_NONE); 2557c64913Sjeremylt CeedQFunctionAddInput(qf_mass, "u", 1, CEED_EVAL_INTERP); 2657c64913Sjeremylt CeedQFunctionAddOutput(qf_mass, "v", 1, CEED_EVAL_INTERP); 2757c64913Sjeremylt 2857c64913Sjeremylt for (CeedInt i = 0; i < Q; i++) { 2957c64913Sjeremylt CeedScalar x = 2. * i / (Q - 1) - 1; 3057c64913Sjeremylt w[i] = 1 - x * x; 3157c64913Sjeremylt u[i] = 2 + 3 * x + 5 * x * x; 3257c64913Sjeremylt v[i] = w[i] * u[i]; 3357c64913Sjeremylt } 34aedaa0e5Sjeremylt 35aedaa0e5Sjeremylt CeedVectorCreate(ceed, Q, &W); 36288c0443SJeremy L Thompson CeedVectorSetArray(W, CEED_MEM_HOST, CEED_USE_POINTER, w); 37aedaa0e5Sjeremylt CeedVectorCreate(ceed, Q, &U); 38288c0443SJeremy L Thompson CeedVectorSetArray(U, CEED_MEM_HOST, CEED_USE_POINTER, u); 39aedaa0e5Sjeremylt CeedVectorCreate(ceed, Q, &V); 40aedaa0e5Sjeremylt CeedVectorSetValue(V, 0); 41d1d35e2fSjeremylt CeedVectorCreate(ceed, Q, &Q_data); 42d1d35e2fSjeremylt CeedVectorSetValue(Q_data, 0); 43aedaa0e5Sjeremylt 4457c64913Sjeremylt { 45aedaa0e5Sjeremylt in[0] = W; 46d1d35e2fSjeremylt out[0] = Q_data; 4757c64913Sjeremylt CeedQFunctionApply(qf_setup, Q, in, out); 4857c64913Sjeremylt } 4957c64913Sjeremylt { 50aedaa0e5Sjeremylt in[0] = W; 51aedaa0e5Sjeremylt in[1] = U; 52aedaa0e5Sjeremylt out[0] = V; 5357c64913Sjeremylt CeedQFunctionApply(qf_mass, Q, in, out); 5457c64913Sjeremylt } 55aedaa0e5Sjeremylt 56aedaa0e5Sjeremylt CeedVectorGetArrayRead(V, CEED_MEM_HOST, &vv); 57*2b730f8bSJeremy L Thompson for (CeedInt i = 0; i < Q; i++) { 58*2b730f8bSJeremy L Thompson if (v[i] != vv[i]) printf("[%" CeedInt_FMT "] v %f != vv %f\n", i, v[i], vv[i]); 59*2b730f8bSJeremy L Thompson } 60aedaa0e5Sjeremylt CeedVectorRestoreArrayRead(V, &vv); 61aedaa0e5Sjeremylt 62aedaa0e5Sjeremylt CeedVectorDestroy(&W); 63aedaa0e5Sjeremylt CeedVectorDestroy(&U); 64aedaa0e5Sjeremylt CeedVectorDestroy(&V); 65d1d35e2fSjeremylt CeedVectorDestroy(&Q_data); 6657c64913Sjeremylt CeedQFunctionDestroy(&qf_setup); 6757c64913Sjeremylt CeedQFunctionDestroy(&qf_mass); 6857c64913Sjeremylt CeedDestroy(&ceed); 6957c64913Sjeremylt return 0; 7057c64913Sjeremylt } 71