1 /// @file
2 /// Test integration with a 2D Simplex non-tensor H^1 basis
3 /// \test Test integration 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 // main test
main(int argc,char ** argv)14 int main(int argc, char **argv) {
15 Ceed ceed;
16 CeedVector u, v, w;
17 const CeedInt p = 6, q = 4, dim = 2;
18 CeedBasis basis;
19 CeedScalar q_ref[dim * q], q_weight[q];
20 CeedScalar interp[p * q], grad[dim * p * q];
21 CeedScalar x_ref[] = {0., 0.5, 1., 0., 0.5, 0., 0., 0., 0., 0.5, 0.5, 1.};
22
23 CeedInit(argv[1], &ceed);
24
25 CeedVectorCreate(ceed, p, &u);
26 {
27 CeedScalar u_array[p];
28
29 // Interpolate function to quadrature points
30 for (int i = 0; i < p; i++) u_array[i] = Eval(x_ref[0 * p + i], x_ref[1 * p + i]);
31 CeedVectorSetArray(u, CEED_MEM_HOST, CEED_COPY_VALUES, u_array);
32 }
33 CeedVectorCreate(ceed, q, &v);
34 CeedVectorSetValue(v, 0);
35 CeedVectorCreate(ceed, q, &w);
36 CeedVectorSetValue(w, 0);
37
38 Build2DSimplex(q_ref, q_weight, interp, grad);
39 CeedBasisCreateH1(ceed, CEED_TOPOLOGY_TRIANGLE, 1, p, q, interp, grad, q_ref, q_weight, &basis);
40
41 CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, CEED_EVAL_INTERP, u, v);
42 CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, CEED_EVAL_WEIGHT, CEED_VECTOR_NONE, w);
43
44 // Check values at quadrature points
45 {
46 const CeedScalar *v_array, *w_array;
47 CeedScalar sum = 0;
48
49 CeedVectorGetArrayRead(v, CEED_MEM_HOST, &v_array);
50 CeedVectorGetArrayRead(w, CEED_MEM_HOST, &w_array);
51 for (int i = 0; i < q; i++) sum += v_array[i] * w_array[i];
52 if (fabs(sum - 17. / 24.) > 100. * CEED_EPSILON) printf("%f != %f\n", sum, 17. / 24.);
53 CeedVectorRestoreArrayRead(v, &v_array);
54 CeedVectorRestoreArrayRead(w, &w_array);
55 }
56
57 CeedVectorDestroy(&u);
58 CeedVectorDestroy(&v);
59 CeedVectorDestroy(&w);
60 CeedBasisDestroy(&basis);
61 CeedDestroy(&ceed);
62 return 0;
63 }
64