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