1 /// @file
2 /// Test grad with a 2D Simplex non-tensor H^1 basis
3 /// \test Test grad with a 2D Simplex non-tensor H^1 basis
4 #include <ceed.h>
5 #include <math.h>
6 #include <stdio.h>
7
8 #include "t320-basis.h"
9
10 // polynomial eval helper
Eval(CeedScalar x1,CeedScalar x2)11 static CeedScalar Eval(CeedScalar x1, CeedScalar x2) { return x1 * x1 + x2 * x2 + x1 * x2 + 1; }
12
13 // polynomial derivative helper
EvalGrad(CeedScalar x1,CeedScalar x2)14 CeedScalar EvalGrad(CeedScalar x1, CeedScalar x2) { return 2 * x1 + x2; }
15
16 // main test
main(int argc,char ** argv)17 int main(int argc, char **argv) {
18 Ceed ceed;
19 CeedVector u, v;
20 const CeedInt p = 6, q = 4, dim = 2;
21 CeedBasis basis;
22 CeedScalar q_ref[dim * q], q_weight[q];
23 CeedScalar interp[p * q], grad[dim * p * q];
24 CeedScalar x_q[] = {0.2, 0.6, 1. / 3., 0.2, 0.2, 0.2, 1. / 3., 0.6};
25 CeedScalar x_ref[] = {0., 0.5, 1., 0., 0.5, 0., 0., 0., 0., 0.5, 0.5, 1.};
26
27 CeedInit(argv[1], &ceed);
28
29 CeedVectorCreate(ceed, p, &u);
30 {
31 CeedScalar u_array[p];
32
33 // Interpolate function to quadrature points
34 for (int i = 0; i < p; i++) u_array[i] = Eval(x_ref[0 * p + i], x_ref[1 * p + i]);
35 CeedVectorSetArray(u, CEED_MEM_HOST, CEED_COPY_VALUES, u_array);
36 }
37 CeedVectorCreate(ceed, q * dim, &v);
38 CeedVectorSetValue(v, 0);
39
40 Build2DSimplex(q_ref, q_weight, interp, grad);
41 CeedBasisCreateH1(ceed, CEED_TOPOLOGY_TRIANGLE, 1, p, q, interp, grad, q_ref, q_weight, &basis);
42
43 CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, CEED_EVAL_GRAD, u, v);
44
45 // Check values at quadrature points
46 {
47 const CeedScalar *v_array;
48
49 CeedVectorGetArrayRead(v, CEED_MEM_HOST, &v_array);
50 for (int i = 0; i < q; i++) {
51 CeedScalar dfx = EvalGrad(x_q[0 * q + i], x_q[1 * q + i]);
52 if (fabs(v_array[0 * q + i] - dfx) > 100. * CEED_EPSILON) printf("[%" CeedInt_FMT "] %f != %f\n", i, v_array[0 * q + i], dfx);
53 dfx = EvalGrad(x_q[1 * q + i], x_q[0 * q + i]);
54 if (fabs(v_array[1 * q + i] - dfx) > 100. * CEED_EPSILON) printf("[%" CeedInt_FMT "] %f != %f\n", i, v_array[1 * q + i], dfx);
55 }
56 CeedVectorRestoreArrayRead(v, &v_array);
57 }
58
59 CeedVectorDestroy(&u);
60 CeedVectorDestroy(&v);
61 CeedBasisDestroy(&basis);
62 CeedDestroy(&ceed);
63 return 0;
64 }
65