xref: /libCEED/tests/t324-basis.c (revision 4fee36f0a30516a0b5ad51bf7eb3b32d83efd623)
1 /// @file
2 /// Test grad transpose with a 2D Simplex non-tensor H1 basis
3 /// \test Test grad transpose with a 2D Simplex non-tensor H1 basis
4 #include <ceed.h>
5 #include <math.h>
6 
7 #include "t320-basis.h"
8 
9 int main(int argc, char **argv) {
10   Ceed          ceed;
11   CeedVector    u, v;
12   const CeedInt p = 6, q = 4, dim = 2;
13   CeedBasis     basis;
14   CeedScalar    q_ref[dim * q], q_weight[q];
15   CeedScalar    interp[p * q], grad[dim * p * q];
16   CeedScalar    column_sum[p];
17 
18   CeedInit(argv[1], &ceed);
19 
20   CeedVectorCreate(ceed, q * dim, &u);
21   CeedVectorSetValue(u, 1);
22   CeedVectorCreate(ceed, p, &v);
23   CeedVectorSetValue(v, 0);
24 
25   Build2DSimplex(q_ref, q_weight, interp, grad);
26   CeedBasisCreateH1(ceed, CEED_TOPOLOGY_TRIANGLE, 1, p, q, interp, grad, q_ref, q_weight, &basis);
27 
28   CeedBasisApply(basis, 1, CEED_TRANSPOSE, CEED_EVAL_GRAD, u, v);
29 
30   // Check values at quadrature points
31   for (int i = 0; i < p; i++) {
32     column_sum[i] = 0;
33     for (int j = 0; j < q * dim; j++) {
34       column_sum[i] += grad[i + j * p];
35     }
36   }
37   {
38     const CeedScalar *v_array;
39 
40     CeedVectorGetArrayRead(v, CEED_MEM_HOST, &v_array);
41     for (int i = 0; i < p; i++) {
42       if (fabs(column_sum[i] - v_array[i]) > 100. * CEED_EPSILON) printf("[%" CeedInt_FMT "] %f != %f\n", i, v_array[i], column_sum[i]);
43     }
44     CeedVectorRestoreArrayRead(v, &v_array);
45   }
46 
47   CeedVectorDestroy(&u);
48   CeedVectorDestroy(&v);
49   CeedBasisDestroy(&basis);
50   CeedDestroy(&ceed);
51   return 0;
52 }
53