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