173d26085Sjeremylt /// @file 273d26085Sjeremylt /// Test creation, use, and destruction of an interlaced multicomponent element restriction 373d26085Sjeremylt /// \test Test creation, use, and destruction of an interlaced multicomponent element restriction 473d26085Sjeremylt #include <ceed.h> 5ec3da8bcSJed Brown #include <ceed/backend.h> 673d26085Sjeremylt 773d26085Sjeremylt int main(int argc, char **argv) { 873d26085Sjeremylt Ceed ceed; 973d26085Sjeremylt CeedVector x, y; 10d1d35e2fSjeremylt CeedInt num_elem = 3; 11d1d35e2fSjeremylt CeedInt ind[2 * num_elem]; 1249fd234cSJeremy L Thompson CeedInt layout[3]; 13d1d35e2fSjeremylt CeedScalar a[2 * (num_elem + 1)]; 1473d26085Sjeremylt const CeedScalar *yy; 1573d26085Sjeremylt CeedElemRestriction r; 1673d26085Sjeremylt 1773d26085Sjeremylt CeedInit(argv[1], &ceed); 1873d26085Sjeremylt 1973d26085Sjeremylt // Setup 20d1d35e2fSjeremylt CeedVectorCreate(ceed, 2 * (num_elem + 1), &x); 21d1d35e2fSjeremylt for (CeedInt i = 0; i < num_elem + 1; i++) { 2273d26085Sjeremylt a[2 * i] = 10 + i; 2373d26085Sjeremylt a[2 * i + 1] = 20 + i; 2473d26085Sjeremylt } 2573d26085Sjeremylt CeedVectorSetArray(x, CEED_MEM_HOST, CEED_USE_POINTER, a); 2673d26085Sjeremylt 27d1d35e2fSjeremylt for (CeedInt i = 0; i < num_elem; i++) { 28d979a051Sjeremylt ind[2 * i + 0] = 2 * i; 29d979a051Sjeremylt ind[2 * i + 1] = 2 * (i + 1); 3073d26085Sjeremylt } 31*2b730f8bSJeremy L Thompson CeedElemRestrictionCreate(ceed, num_elem, 2, 2, 1, 2 * (num_elem + 1), CEED_MEM_HOST, CEED_USE_POINTER, ind, &r); 32d1d35e2fSjeremylt CeedVectorCreate(ceed, 2 * (num_elem * 2), &y); 3373d26085Sjeremylt 3473d26085Sjeremylt // Restrict 35a8d32208Sjeremylt CeedElemRestrictionApply(r, CEED_NOTRANSPOSE, x, y, CEED_REQUEST_IMMEDIATE); 3673d26085Sjeremylt 3773d26085Sjeremylt // Check 3873d26085Sjeremylt CeedVectorGetArrayRead(y, CEED_MEM_HOST, &yy); 3949fd234cSJeremy L Thompson CeedElemRestrictionGetELayout(r, &layout); 40*2b730f8bSJeremy L Thompson for (CeedInt i = 0; i < 2; i++) { // Node 41*2b730f8bSJeremy L Thompson for (CeedInt j = 0; j < 2; j++) { // Component 42*2b730f8bSJeremy L Thompson for (CeedInt k = 0; k < num_elem; k++) { // Element 43*2b730f8bSJeremy L Thompson if (yy[i * layout[0] + j * layout[1] + k * layout[2]] != a[ind[i + k * 2] + j]) { 44a2546046Sjeremylt // LCOV_EXCL_START 45*2b730f8bSJeremy L Thompson printf("Error in restricted array y[%" CeedInt_FMT "][%" CeedInt_FMT "][%" CeedInt_FMT "] = %f != %f\n", i, j, k, 46*2b730f8bSJeremy L Thompson (CeedScalar)yy[i * layout[0] + j * layout[1] + k * layout[2]], a[ind[i + k * 2] + j * (num_elem + 1)]); 47de996c55Sjeremylt // LCOV_EXCL_STOP 48*2b730f8bSJeremy L Thompson } 49*2b730f8bSJeremy L Thompson } 50*2b730f8bSJeremy L Thompson } 51*2b730f8bSJeremy L Thompson } 5273d26085Sjeremylt 5373d26085Sjeremylt CeedVectorRestoreArrayRead(y, &yy); 5473d26085Sjeremylt CeedVectorDestroy(&x); 5573d26085Sjeremylt CeedVectorDestroy(&y); 5673d26085Sjeremylt CeedElemRestrictionDestroy(&r); 5773d26085Sjeremylt CeedDestroy(&ceed); 5873d26085Sjeremylt return 0; 5973d26085Sjeremylt } 60