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