xref: /libCEED/tests/t209-elemrestriction.c (revision a2546046f5f3e60916f9cd7f16707c5c85ac2e89)
1 /// @file
2 /// Test calculation of dof multiplicity in element restriction
3 /// \test Test calculation of dof multiplicity in element restriction
4 #include <ceed.h>
5 
6 int main(int argc, char **argv) {
7   Ceed ceed;
8   CeedVector mult;
9   CeedInt ne = 3;
10   CeedInt ind[4*ne];
11   const CeedScalar *mm;
12   CeedElemRestriction r;
13 
14   CeedInit(argv[1], &ceed);
15 
16   CeedVectorCreate(ceed, 3*ne+1, &mult);
17   CeedVectorSetValue(mult, 0); // Allocates array
18 
19   for (CeedInt i=0; i<ne; i++) {
20     ind[4*i+0] = i*3+0;
21     ind[4*i+1] = i*3+1;
22     ind[4*i+2] = i*3+2;
23     ind[4*i+3] = i*3+3;
24   }
25   CeedElemRestrictionCreate(ceed, ne, 4, 3*ne+1, 1, CEED_MEM_HOST,
26                             CEED_USE_POINTER, ind, &r);
27 
28   CeedElemRestrictionGetMultiplicity(r, mult);
29 
30   CeedVectorGetArrayRead(mult, CEED_MEM_HOST, &mm);
31   for (CeedInt i=0; i<3*ne+1; i++)
32     if ((1 + (i > 0 && i < 3*ne && (i%3==0) ? 1 : 0)) != mm[i])
33       // LCOV_EXCL_START
34       printf("Error in multiplicity vector: mult[%d] = %f\n",
35              i, (double)mm[i]);
36       // LCOV_EXCL_END
37   CeedVectorRestoreArrayRead(mult, &mm);
38 
39   CeedVectorDestroy(&mult);
40   CeedElemRestrictionDestroy(&r);
41   CeedDestroy(&ceed);
42   return 0;
43 }
44