1 /// @file 2 /// Test GetDiv and BasisApply for a 2D Quad non-tensor H(div) basis 3 /// \test Test GetDiv and BasisApply for a 2D Quad non-tensor H(div) basis 4 #include <ceed.h> 5 #include <math.h> 6 7 #include "t330-basis.h" 8 9 int main(int argc, char **argv) { 10 Ceed ceed; 11 const CeedInt num_nodes = 4, q = 3, dim = 2, num_qpts = q * q; 12 CeedInt num_comp = 1; // one vector component 13 CeedInt p = dim * num_nodes; // DoF per element 14 CeedBasis basis; 15 CeedScalar q_ref[dim * num_qpts], q_weights[num_qpts]; 16 CeedScalar div[p * num_qpts], interp[p * dim * num_qpts]; 17 CeedVector u, v; 18 19 CeedInit(argv[1], &ceed); 20 21 BuildHdivQuadrilateral(q, q_ref, q_weights, interp, div, CEED_GAUSS); 22 CeedBasisCreateHdiv(ceed, CEED_TOPOLOGY_QUAD, num_comp, p, num_qpts, interp, div, q_ref, q_weights, &basis); 23 24 // Test GetDiv 25 { 26 const CeedScalar *div_2; 27 28 CeedBasisGetDiv(basis, &div_2); 29 for (CeedInt i = 0; i < p * num_qpts; i++) { 30 if (fabs(div[i] - div_2[i]) > 100. * CEED_EPSILON) printf("%f != %f\n", div[i], div_2[i]); 31 } 32 } 33 34 CeedVectorCreate(ceed, p, &u); 35 CeedVectorSetValue(u, 1); 36 CeedVectorCreate(ceed, num_qpts, &v); 37 CeedVectorSetValue(v, 0); 38 39 // BasisApply for H(div): CEED_EVAL_DIV, NOTRANSPOSE case 40 CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, CEED_EVAL_DIV, u, v); 41 42 { 43 const CeedScalar *v_array; 44 45 CeedVectorGetArrayRead(v, CEED_MEM_HOST, &v_array); 46 for (CeedInt i = 0; i < num_qpts; i++) { 47 if (fabs(p * 0.25 - v_array[i]) > 100. * CEED_EPSILON) printf("%f != %f\n", 2.0, v_array[i]); 48 } 49 CeedVectorRestoreArrayRead(v, &v_array); 50 } 51 52 CeedVectorSetValue(v, 1.0); 53 CeedVectorSetValue(u, 0.0); 54 55 // BasisApply for H(div): CEED_EVAL_DIV, TRANSPOSE case 56 CeedBasisApply(basis, 1, CEED_TRANSPOSE, CEED_EVAL_DIV, v, u); 57 58 { 59 const CeedScalar *u_array; 60 61 CeedVectorGetArrayRead(u, CEED_MEM_HOST, &u_array); 62 for (CeedInt i = 0; i < p; i++) { 63 if (fabs(num_qpts * 0.25 - u_array[i]) > 100. * CEED_EPSILON) printf("%f != %f\n", 2.0, u_array[i]); 64 } 65 CeedVectorRestoreArrayRead(u, &u_array); 66 } 67 68 CeedBasisDestroy(&basis); 69 CeedVectorDestroy(&u); 70 CeedVectorDestroy(&v); 71 CeedDestroy(&ceed); 72 return 0; 73 } 74